marshmallow schemas no longer return _status_enum column

now return `status`, as they should
This commit is contained in:
Leo Hemsted
2017-05-10 17:30:09 +01:00
parent 348183d93b
commit 466c833c39
3 changed files with 41 additions and 3 deletions

View File

@@ -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'),