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": "{index_1b}".format(index_1b=index + 1),
"concat-ref": "78",
"concat-total": "{parts_len}".format(parts_len=len(parts)),
"keyword": "LOREM",
"message-timestamp": "2018-04-24 14:05:19",
"messageId": "0B000000D0EBB58{index}".format(index=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"