def test_send_push_notification(self):
# Make a mock of the response
def request_callback(request):
payload = querystring_parser.parse(request.body)
if validate_zeropush_payload(payload):
response_body = {
"sent_count": len(payload["device_tokens"][""]),
"inactive_tokens": [],
"unregistered_tokens": [],
}
headers = {"Content-Type": "application/json"}
return (200, headers, json.dumps(response_body))
else:
return (400, {}, {})
responses.add_callback(
responses.POST, ZEROPUSH_NOTIFY_URL, callback=request_callback, content_type="application/json"
)
zeropush = ZeroPushService({"AUTH_TOKEN": "123123"})
devices = PushDeviceFactory.create_batch(10)
send = zeropush.send_push_notification(
devices,
"This is the message",
sound="annoyingSound.mp3",
badge_number=1,
payload={"extra": "payload", "in": "notification"},
expiry=timedelta(days=30),
category="test",
)
assert send is True
def test_zeropush_register_valid(self):
"""
Test if we get back True when the resposne of registering is ok
"""
responses.add_callback(
responses.POST, ZEROPUSH_REGISTER_URL, callback=request_register_callback, content_type="application/json"
)
zeropush = ZeroPushService({"AUTH_TOKEN": "123123"})
token = uuid4()
registered = zeropush.register_push_device(token)
assert registered is True
def test_settings_improperly_configure(self):
# It must give an attribute error when settings are not set
try:
zeropush = ZeroPushService({})
# This shouldn't be happening
assert zeropush is None
except ImproperlyConfigured:
pass
# It should run without problems, when valid settings are set
zeropush = ZeroPushService({
'AUTH_TOKEN': '123123'
})
assert isinstance(zeropush, ZeroPushService)
def test_zeropush_register_invalid(self):
"""
Test if we get back False when the response of registering is not ok.
"""
responses.add(responses.POST, ZEROPUSH_REGISTER_URL,
body='{"error": "invalid token"}', status=400,
content_type='application/json')
zeropush = ZeroPushService({
'AUTH_TOKEN': '123123'
})
token = uuid4()
registered = zeropush.register_push_device(token)
assert registered is False
def test_zeropush_register_invalid(self):
"""
Test if we get back False when the response of registering is not ok.
"""
responses.add(
responses.POST,
ZEROPUSH_REGISTER_URL,
body='{"error": "invalid token"}',
status=400,
content_type="application/json",
)
zeropush = ZeroPushService({"AUTH_TOKEN": "123123"})
token = uuid4()
registered = zeropush.register_push_device(token)
assert registered is False
def test_zeropush_register_valid(self):
"""
Test if we get back True when the resposne of registering is ok
"""
responses.add_callback(
responses.POST, ZEROPUSH_REGISTER_URL,
callback=request_register_callback,
content_type='application/json',
)
zeropush = ZeroPushService({
'AUTH_TOKEN': '123123'
})
token = uuid4()
registered = zeropush.register_push_device(token)
assert registered is True
def test_zeropush_datetime_expiry(self):
"""
Test if ZeroPush Service also accepts datetime objects
"""
# Make a mock of the response
def request_callback(request):
payload = querystring_parser.parse(request.body)
if validate_zeropush_payload(payload):
response_body = {
"sent_count": len(payload['device_tokens']['']),
"inactive_tokens": [],
"unregistered_tokens": []
}
headers = {
'Content-Type': 'application/json'
}
return (200, headers, json.dumps(response_body))
else:
return (400, {}, {})
responses.add_callback(
responses.POST, ZEROPUSH_NOTIFY_URL,
callback=request_callback,
content_type='application/json',
)
zeropush = ZeroPushService({
'AUTH_TOKEN': '123123'
})
devices = PushDeviceFactory.create_batch(10)
send = zeropush.send_push_notification(devices, 'This is the message',
sound="annoyingSound.mp3",
badge_number=1,
payload={
"extra": "payload",
"in": "notification"
},
expiry=datetime.now(),
category="test"
)
assert send is True
