def test_decorator_partial__bad_sig(rf, partial_message):
partial_message["sig"] = "not-valid"
request = rf.post(
"/sms/incoming",
content_type="application/json",
data=json.dumps(partial_message),
)
view = MagicMock(return_value=sentinel.response)
assert (len(models.SMSMessagePart.objects.filter(concat_ref="78")) == 0
), "There should be no parts stored before the test."
validating = d.sms_webhook(validate_signature=True)(view)
response = validating(request)
assert view.mock_calls == [], "Wrapped view should not be called."
assert (response.status_code == 403
), "Decorator should return 403, permission denied."
assert (len(models.SMSMessagePart.objects.filter(
concat_ref="78")) == 0), "No part should be saved."
nonvalidating = d.sms_webhook(validate_signature=False)(view)
response = nonvalidating(request)
assert (response.status_code == 200
), "OK response should be returned by the decorator."
assert view.mock_calls == [], "Wrapped view should not be called."
assert (len(models.SMSMessagePart.objects.filter(
concat_ref="78")) == 1), "Part should be saved."
def test_decorator_partials(rf, partial_message):
""" Ensure partial sms messages are handled corrrectly. """
request = rf.post(
"/sms/incoming",
content_type="application/json",
data=json.dumps(partial_message),
)
view = MagicMock()
decorated = d.sms_webhook()(view)
# Before doing anything, make sure we don't already have any parts saved:
assert (len(models.SMSMessagePart.objects.filter(concat_ref="78")) == 0
), "There should be no parts stored before the test."
response = decorated(request)
assert response.status_code == 200, "Request should be successful"
assert (len(models.SMSMessagePart.objects.filter(
concat_ref="78")) == 1), "Part should be saved."
# Ensure a duplicate part doesn't also get saved:
response = decorated(request)
assert response.status_code == 200, "Request should be successful"
assert (len(models.SMSMessagePart.objects.filter(
concat_ref="78")) == 1), "No duplicate part should be saved."
assert view.mock_calls == [], "Wrapped view should not be called."
def test_decorator_complete(rf, complete_message):
request = rf.post(
"/sms/incoming",
content_type="application/json",
data=json.dumps(complete_message),
)
view = MagicMock(return_value=sentinel.response)
decorated = d.sms_webhook()(view)
response = decorated(request)
assert (len(models.SMSMessagePart.objects.filter(
concat_ref="78")) == 0), "Message should not be saved."
assert view.mock_calls == [call(request)
], "Underlying view should be called."
assert (response == sentinel.response
), "Result should be response from the underlying view."
def test_decorator_multipart(rf):
parts = ["This", " is", " a ", "multipart", " message"]
view = MagicMock(return_value=sentinel.response)
webhook = d.sms_webhook(validate_signature=False)(view)
requests = [
rf.post(
"/sms/incoming",
content_type="application/json",
data=json.dumps({
"concat": "true",
"concat-part": f"{index + 1}",
"concat-ref": "78",
"concat-total": f"{len(parts)}",
"keyword": "LOREM",
"message-timestamp": "2018-04-24 14:05:19",
"messageId": f"0B000000D0EBB58{index}",
"msisdn": "447700900419",
"text": part,
"timestamp": "1524578719",
"to": "447700900996",
"type": "unicode",
}),
) for index, part in enumerate(parts)
]
shuffle(requests)
for request in requests[:-1]:
response = webhook(request)
assert response.status_code == 200
response = webhook(requests[-1])
assert response is sentinel.response
assert len(view.mock_calls) == 1
name, args, kwargs = view.mock_calls[0]
assert args[0].sms.text == 'This is a multipart message'