def test_integer_bytes_conversion2(self):
# test integer to bytes, then bytes to integer == original
key = Fernet.generate_key()
fernet = Fernet(key)
for i in range(10000):
assert fernet._bytes_to_integer(fernet._integer_to_bytes(i)) == i
def test_integer_bytes_conversion(self):
# test bytes to integer, then integer to bytes == original
key = Fernet.generate_key()
fernet = Fernet(key)
for i in range(10000):
b = os.urandom(16)
assert fernet._integer_to_bytes(fernet._bytes_to_integer(b)) == b
def test_AES_ECB(self):
# AES_ECB_128 test with RFC test vectors
key = Fernet.generate_key()
fernet = Fernet(key)
KEY = '2b7e151628aed2a6abf7158809cf4f3c'.decode('hex')
AES_128 = '7df76b0c1ab899b33e42f047b91b546f'.decode('hex')
CONST_ZERO = 0x00000000000000000000000000000000
assert fernet._AES_ECB(fernet._integer_to_bytes(CONST_ZERO), KEY) == AES_128
def test_big_integer(self):
# test _integer_to_bytes when given too big an integer
key = Fernet.generate_key()
fernet = Fernet(key)
with pytest.raises(AssertionError) as e:
fernet._integer_to_bytes(2**128)
