def test_subkey_gen_bad_length(self):
# test subkey generation function when given bad input length
key = Fernet.generate_key()
fernet = Fernet(key)
KEY = os.urandom(15)
with pytest.raises(AssertionError) as e:
fernet._generate_subkey(KEY)
def test_subkey_generation(self):
# test subkey generation using RFC test vectors
key = Fernet.generate_key()
fernet = Fernet(key)
KEY = '2b7e151628aed2a6abf7158809cf4f3c'.decode('hex')
AES_128 = '7df76b0c1ab899b33e42f047b91b546f'.decode('hex')
K1 = 'fbeed618357133667c85e08f7236a8de'.decode('hex')
K2 = 'f7ddac306ae266ccf90bc11ee46d513b'.decode('hex')
K1_, K2_ = fernet._generate_subkey(KEY)
assert K1 == K1_ and K2 == K2_
