def test_build_cipher():
key_ = b"1234523451234545" # 16 byte key
iv_ = os.urandom(16)
cipher, iv = build_cipher(key_, iv_)
assert iv == iv_
assert isinstance(cipher, CbcMode)
def test_build_cipher_mismatch_key_lenth():
key_ = b"1234523451234545" # 16 byte key
with pytest.raises(AESError):
build_cipher(key_, '', alg='aes_192_cbc')
def test_build_cipher_wrong_key_lenth():
key_ = b"1234523451234545" # 16 byte key
with pytest.raises(AESError):
build_cipher(key_, '', alg='aes_384_cbc')
def test_build_cipher_no_iv():
key_ = b"1234523451234545" # 16 byte key
cipher, iv = build_cipher(key_, '')
assert iv
assert isinstance(cipher, CbcMode)
def test_build_cipher_wrong_mode():
key_ = b"1234523451234545" # 16 byte key
with pytest.raises(AESError):
build_cipher(key_, '', alg='aes_128_abc')