Merge pull request #1400 from alphagov/ken-allow-include_jobs-arg-get_all_notis

Allow include jobs request param on v2 get all notifications

tests/app/v2/notifications/test_get_notifications.py

@@ -266,7 +266,8 @@ def test_get_notification_doesnt_have_delivery_estimate_for_non_letters(
     assert 'estimated_delivery' not in json.loads(response.get_data(as_text=True))
 
 
-def test_get_all_notifications_returns_200(client, sample_template):
+def test_get_all_notifications_except_job_notifications_returns_200(client, sample_template, sample_job):
+    create_notification(template=sample_template, job=sample_job)  # should not return this job notification
     notifications = [create_notification(template=sample_template) for _ in range(2)]
     notification = notifications[-1]
 
@@ -295,6 +296,34 @@ def test_get_all_notifications_returns_200(client, sample_template):
     assert not json_response['notifications'][0]['scheduled_for']
 
 
+def test_get_all_notifications_with_include_jobs_arg_returns_200(
+    client, sample_template, sample_job
+):
+    notifications = [
+        create_notification(template=sample_template, job=sample_job),
+        create_notification(template=sample_template)
+    ]
+    notification = notifications[-1]
+
+    auth_header = create_authorization_header(service_id=notification.service_id)
+    response = client.get(
+        path='/v2/notifications?include_jobs=true',
+        headers=[('Content-Type', 'application/json'), auth_header])
+
+    json_response = json.loads(response.get_data(as_text=True))
+
+    assert response.status_code == 200
+    assert json_response['links']['current'].endswith("/v2/notifications?include_jobs=true")
+    assert 'next' in json_response['links'].keys()
+    assert len(json_response['notifications']) == 2
+
+    assert json_response['notifications'][0]['id'] == str(notification.id)
+    assert json_response['notifications'][0]['status'] == notification.status
+    assert json_response['notifications'][0]['phone_number'] == notification.to
+    assert json_response['notifications'][0]['type'] == notification.template.template_type
+    assert not json_response['notifications'][0]['scheduled_for']
+
+
 def test_get_all_notifications_no_notifications_if_no_notifications(client, sample_service):
     auth_header = create_authorization_header(service_id=sample_service.id)
     response = client.get(

tests/app/v2/notifications/test_notification_schemas.py

@@ -5,6 +5,7 @@ from flask import json
 from freezegun import freeze_time
 from jsonschema import ValidationError
 
+from app.models import NOTIFICATION_CREATED, EMAIL_TYPE
 from app.schema_validation import validate
 from app.v2.notifications.notification_schemas import (
     get_notifications_request,
@@ -13,6 +14,22 @@ from app.v2.notifications.notification_schemas import (
 )
 
 
+valid_get_json = {}
+
+valid_get_with_optionals_json = {
+    "reference": "test reference",
+    "status": [NOTIFICATION_CREATED],
+    "template_type": [EMAIL_TYPE],
+    "include_jobs": "true",
+    "older_than": "a5149c32-f03b-4711-af49-ad6993797d45"
+}
+
+
+@pytest.mark.parametrize("input", [valid_get_json, valid_get_with_optionals_json])
+def test_get_notifications_valid_json(input):
+    assert validate(input, get_notifications_request) == input
+
+
 @pytest.mark.parametrize('invalid_statuses, valid_statuses', [
     # one invalid status
     (["elephant"], []),