From 458d915f1f1aa2fcd472eec2912a4f5e8dca0dc9 Mon Sep 17 00:00:00 2001 From: Maciej Urbanski Date: Wed, 15 Nov 2023 12:47:26 +0100 Subject: [PATCH] reduce entropy use in integration tests --- test/integration/helpers.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/test/integration/helpers.py b/test/integration/helpers.py index 633025ef6..2446aa7a0 100755 --- a/test/integration/helpers.py +++ b/test/integration/helpers.py @@ -66,26 +66,30 @@ BUCKET_NAME_LENGTH = BUCKET_NAME_LENGTH_RANGE[1] BUCKET_CREATED_AT_MILLIS = 'created_at_millis' +RNG = secrets.SystemRandom() +RNG_SEED = str(RNG.random()) +RNG_COUNTER = 0 + +if sys.version_info < (3, 9): + RNG.randbytes = lambda n: RNG.getrandbits(n * 8).to_bytes(n, 'little') + SSE_NONE = EncryptionSetting(mode=EncryptionMode.NONE,) SSE_B2_AES = EncryptionSetting( mode=EncryptionMode.SSE_B2, algorithm=EncryptionAlgorithm.AES256, ) +_SSE_KEY = RNG.randbytes(32) SSE_C_AES = EncryptionSetting( mode=EncryptionMode.SSE_C, algorithm=EncryptionAlgorithm.AES256, - key=EncryptionKey(secret=os.urandom(32), key_id='user-generated-key-id') + key=EncryptionKey(secret=_SSE_KEY, key_id='user-generated-key-id') ) SSE_C_AES_2 = EncryptionSetting( mode=EncryptionMode.SSE_C, algorithm=EncryptionAlgorithm.AES256, - key=EncryptionKey(secret=os.urandom(32), key_id='another-user-generated-key-id') + key=EncryptionKey(secret=_SSE_KEY, key_id='another-user-generated-key-id') ) -RNG = secrets.SystemRandom() -RNG_SEED = str(RNG.random()) -RNG_COUNTER = 0 - def bucket_name_part(length: int) -> str: assert length >= 1