From 44c5831f3dc6b10c55b5839697ac36d6fe476dc9 Mon Sep 17 00:00:00 2001 From: Imdad Ahad Date: Mon, 10 Jul 2017 15:50:57 +0100 Subject: [PATCH] Remove old status schema references --- app/schemas.py | 4 ++-- tests/app/test_schemas.py | 2 -- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/app/schemas.py b/app/schemas.py index 937ab7812..2701512ee 100644 --- a/app/schemas.py +++ b/app/schemas.py @@ -260,7 +260,7 @@ class NotificationModelSchema(BaseSchema): class Meta: model = models.Notification strict = True - exclude = ('_personalisation', 'job', 'service', 'template', 'api_key', '_status_enum', '_status_fkey') + exclude = ('_personalisation', 'job', 'service', 'template', 'api_key',) status = fields.String(required=False) @@ -416,7 +416,7 @@ class NotificationWithTemplateSchema(BaseSchema): class Meta: model = models.Notification strict = True - exclude = ('_personalisation', '_status_enum', '_status_fkey') + exclude = ('_personalisation', ) template = fields.Nested( TemplateSchema, diff --git a/tests/app/test_schemas.py b/tests/app/test_schemas.py index 4ab15f569..18bbbca37 100644 --- a/tests/app/test_schemas.py +++ b/tests/app/test_schemas.py @@ -44,8 +44,6 @@ def test_notification_schema_has_correct_status(sample_notification, schema_name 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', [