def test_empty_bytes(self): # test _bytes_to_integer function when given empty bytes key = Fernet.generate_key() fernet = Fernet(key) b = os.urandom(0) with pytest.raises(AssertionError) as e: fernet._bytes_to_integer(b)
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