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_