class TestProcessMessageStatus(TestCase):
def setUp(self):
super(TestProcessMessageStatus, self).setUp()
self.backend = SQLTelerivetBackend(
name='TELERIVET',
is_global=True,
hq_api_id=SQLTelerivetBackend.get_api_id())
self.backend.set_extra_fields(
webhook_secret='The chamber of webhook secrets')
self.backend.save()
self.sms = SMS(
text='I am testy',
custom_metadata={'case_id': '123'},
)
self.sms.save()
def tearDown(self):
self.sms.delete()
super(TestProcessMessageStatus, self).tearDown()
def test_status_delivered(self):
process_message_status(self.sms, DELIVERED)
sms = SMS.objects.get(couch_id=self.sms.couch_id)
self.assertTrue('gateway_delivered' in sms.custom_metadata.keys())
def test_error_status(self):
message_id = self.sms.couch_id
process_message_status(self.sms,
FAILED,
error_message='Somthing went wrong')
sms = SMS.objects.get(couch_id=message_id)
self.assertTrue('gateway_delivered' not in sms.custom_metadata.keys())
class TelerivetViewTestCase(TestCase):
view_path = 'telerivet_message_status'
def setUp(self):
self.backend = SQLTelerivetBackend(
name='TELERIVET',
is_global=True,
hq_api_id=SQLTelerivetBackend.get_api_id())
self.backend.set_extra_fields(
webhook_secret='The chamber of webhook secrets')
self.backend.save()
self.sms = SMS(phone_number='27845698785', text="Fame is fickle")
self.sms.save()
def tearDown(self):
self.backend.delete()
self.sms.delete()
def test_message_status_successful_without_prior_metadata(self):
data = {
'status': 'delivered',
'secret': self.backend.config.webhook_secret
}
response = self.client.post(
reverse(self.view_path, kwargs={'message_id': self.sms.couch_id}),
data)
self.assertTrue(response.status_code, 200)
self.sms.refresh_from_db()
expected_custom_metadata = {'gateway_delivered': True}
self.assertEqual(self.sms.custom_metadata, expected_custom_metadata)
def test_message_status_successful_with_metadata(self):
self.sms.custom_metadata = {'custom': 'data'}
self.sms.save()
data = {
'status': 'delivered',
'secret': self.backend.config.webhook_secret
}
response = self.client.post(
reverse(self.view_path, kwargs={'message_id': self.sms.couch_id}),
data)
self.assertTrue(response.status_code, 200)
self.sms.refresh_from_db()
expected_custom_metadata = {
'custom': 'data',
'gateway_delivered': True
}
self.assertEqual(self.sms.custom_metadata, expected_custom_metadata)
def test_message_status_with_wrong_message_id(self):
self.sms.custom_metadata = {}
self.sms.save()
data = {
'status': 'delivered',
'secret': self.backend.config.webhook_secret
}
response = self.client.post(
reverse(self.view_path,
kwargs={'message_id': f'{self.sms.couch_id}123'}), data)
self.assertTrue(response.status_code == 404)
def test_message_status_with_invalid_secret(self):
self.sms.custom_metadata = {}
self.sms.save()
data = {'status': 'delivered', 'secret': 'not a secret'}
response = self.client.post(
reverse(self.view_path, kwargs={'message_id': self.sms.couch_id}),
data)
self.assertTrue(response.status_code == 403)
