darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-03 01:41:05 -05:00
Code Issues Packages Projects Releases Wiki Activity

update notification rest tests

Browse Source
This commit is contained in:
Leo Hemsted
2017-03-28 13:47:08 +01:00
parent 25d1777937
commit 11919d9810
3 changed files with 33 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
tests/app/notifications/rest/test_send_notification.py
View File

@@ -1,8 +1,9 @@
import random
import string
import pytest
from datetime import datetime
from unittest.mock import ANY
import pytest
from flask import (json, current_app)
from freezegun import freeze_time
from notifications_python_client.authentication import create_jwt_token
@@ -133,6 +134,7 @@ def test_send_notification_with_placeholders_replaced(notify_api, sample_email_t
"something_encrypted",
'2016-01-01T11:09:00.061258Z'
),
kwargs=ANY,
queue="db-email"
)
assert response.status_code == 201
@@ -290,6 +292,7 @@ def test_should_allow_valid_sms_notification(notify_api, sample_template, mocker
notification_id,
"something_encrypted",
'2016-01-01T11:09:00.061258Z'),
kwargs=ANY,
queue='db-sms')
assert response.status_code == 201
assert notification_id
@@ -348,8 +351,8 @@ def test_should_allow_valid_email_notification(notify_api, sample_email_template
notification_id,
"something_encrypted",
'2016-01-01T11:00:00.000000Z'
),
kwargs=ANY,
queue="db-email"
)
@@ -555,6 +558,10 @@ def test_should_send_email_if_team_api_key_and_a_service_user(notify_api, sample
notification_id,
"something_encrypted",
"2016-01-01T11:00:00.000000Z"),
kwargs={
'api_key_id': str(api_key.id),
'key_type': KEY_TYPE_TEAM
},
queue='db-email'
)
assert response.status_code == 201
@@ -594,6 +601,10 @@ def test_should_send_sms_to_anyone_with_test_key(
notification_id = json.loads(response.data)['data']['notification']['id']
app.celery.tasks.send_sms.apply_async.assert_called_once_with(
(str(sample_template.service.id), notification_id, "something_encrypted", "2016-01-01T11:00:00.000000Z"),
kwargs={
'api_key_id': str(api_key.id),
'key_type': KEY_TYPE_TEST
},
queue='research-mode'
)
assert response.status_code == 201
@@ -635,6 +646,10 @@ def test_should_send_email_to_anyone_with_test_key(
notification_id,
"something_encrypted",
"2016-01-01T11:00:00.000000Z"),
kwargs={
'api_key_id': str(api_key.id),
'key_type': KEY_TYPE_TEST
},
queue='research-mode'
)
assert response.status_code == 201
@@ -669,6 +684,10 @@ def test_should_send_sms_if_team_api_key_and_a_service_user(notify_api, sample_t
notification_id,
"something_encrypted",
"2016-01-01T11:00:00.000000Z"),
kwargs={
'api_key_id': str(api_key.id),
'key_type': KEY_TYPE_TEAM
},
queue='db-sms'
)
assert response.status_code == 201
@@ -710,6 +729,7 @@ def test_should_persist_notification(notify_api, sample_template,
mocked.assert_called_once_with(
(str(template.service.id), notification_id, "something_encrypted", '2016-01-01T11:00:00.000000Z'),
kwargs=ANY,
queue='db-{}'.format(template_type))
assert response.status_code == 201
@@ -1037,5 +1057,7 @@ def test_send_notification_uses_priority_queue_when_template_is_marked_as_priori
assert response.status_code == 201
mocked.assert_called_once_with(
(str(sample.service_id), notification_id, "something_encrypted", '2016-01-01T11:00:00.000000Z'), queue='notify'
(str(sample.service_id), notification_id, "something_encrypted", '2016-01-01T11:00:00.000000Z'),
kwargs=ANY,
queue='notify'
)

10
tests/app/v2/notifications/test_post_notifications.py
View File

@@ -1,10 +1,14 @@
import uuid
from unittest.mock import ANY
import pytest
from flask import json
from freezegun import freeze_time
from app.models import Notification
from tests import create_authorization_header
from tests.app.conftest import sample_template as create_sample_template
from freezegun import freeze_time
@pytest.mark.parametrize("reference", [None, "reference_from_client"])
@@ -220,5 +224,7 @@ def test_send_notification_uses_priority_queue_when_template_is_marked_as_priori
print(response.data)
assert response.status_code == 201
mocked.assert_called_once_with(
(str(sample.service_id), notification_id, 'something_encrypted', '2016-01-01T11:00:00.000000Z'), queue='notify'
(str(sample.service_id), notification_id, 'something_encrypted', '2016-01-01T11:00:00.000000Z'),
kwargs=ANY,
queue='notify'
)