marshmallow schemas no longer return _status_enum column

now return `status`, as they should

tests/app/job/test_rest.py

@@ -447,6 +447,25 @@ def test_get_all_notifications_for_job_filtered_by_status(
         assert response.status_code == 200
 
 
+def test_get_all_notifications_for_job_returns_correct_format(
+    client,
+    sample_notification_with_job
+):
+    service_id = sample_notification_with_job.service_id
+    job_id = sample_notification_with_job.job_id
+    response = client.get(
+        path='/service/{}/job/{}/notifications'.format(service_id, job_id),
+        headers=[create_authorization_header()]
+    )
+    assert response.status_code == 200
+    resp = json.loads(response.get_data(as_text=True))
+    assert len(resp['notifications']) == 1
+    assert resp['notifications'][0]['id'] == str(sample_notification_with_job.id)
+    assert resp['notifications'][0]['status'] == sample_notification_with_job.status
+    assert '_status_fkey' not in resp['notifications'][0]
+    assert '_status_enum' not in resp['notifications'][0]
+
+
 def test_get_job_by_id(notify_api, sample_job):
     job_id = str(sample_job.id)
     service_id = sample_job.service.id

tests/app/test_schemas.py

@@ -1,5 +1,4 @@
 import pytest
-
 from marshmallow import ValidationError
 from sqlalchemy import desc
 
@@ -33,6 +32,22 @@ def test_notification_schema_adds_api_key_name(sample_notification_with_api_key)
     assert data['key_name'] == 'Test key'
 
 
+@pytest.mark.parametrize('schema_name', [
+    'notification_with_template_schema',
+    'notification_schema',
+    'notification_with_template_schema',
+    'notification_with_personalisation_schema',
+])
+def test_notification_schema_has_correct_status(sample_notification, schema_name):
+    from app import schemas
+
+    data = getattr(schemas, schema_name).dump(sample_notification).data
+
+    assert data['status'] == sample_notification.status
+    assert '_status_enum' not in data
+    assert '_status_fkey' not in data
+
+
 @pytest.mark.parametrize('user_attribute, user_value', [
     ('name', 'New User'),
     ('email_address', 'newuser@mail.com'),