def test_push_sms_encrypted_ok(pb_encrypt):
device = Mock()
device.device_iden = "123"
mock_response = Mock()
mock_response.status_code = 200
session = Mock()
session.post.return_value = mock_response
pb_encrypt.return_value = "encrypted text"
pb = PushBullet("apikey")
pb._session = session
pb._encryption_key = "123"
# When
pb.push_sms(device, "+123456789", "This is an example text")
session.post.assert_called_once_with(
pb.EPHEMERALS_URL,
data=json.dumps({
"type": "push",
"push": {
"ciphertext": "encrypted text",
"encrypted": True
},
}),
)
def test_decryption():
pb = PushBullet(API_KEY, encryption_password="******")
pb._encryption_key = a2b_base64(
"1sW28zp7CWv5TtGjlQpDHHG4Cbr9v36fG5o4f74LsKg=")
test_data = "MSfJxxY5YdjttlfUkCaKA57qU9SuCN8+ZhYg/xieI+lDnQ=="
decrypted = pb._decrypt_data(test_data)
assert decrypted == "meow!"
