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')