def test_encrypt(self):
provider = StubProvider()
kf = KeyFile('asdf.txt')
kf._KeyFile__key = b'x' * 32
kf._get_provider = MagicMock(return_value=provider)
secret = kf.encrypt('hello', 'test')
provider.encrypt.assert_called_once_with(b'hello')
kf._get_provider.assert_called_once_with('test')
assert secret == SecureValue('test', b'ciphertext')
def test_encrypt_nokey(self):
kf = KeyFile('asdf.txt')
with pytest.raises(TypeError):
kf.encrypt(b'hello')
