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

Add a load only and dump only for the created_by attribute of the JobSchema

Browse Source
This commit is contained in:
Rebecca Law
2016-05-11 12:03:25 +01:00
parent b8017dcf02
commit aff0abb78c
2 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/schemas.py
View File

@@ -143,6 +143,10 @@ class ApiKeySchema(BaseSchema):
class JobSchema(BaseSchema): class JobSchema(BaseSchema):
created_by_user = fields.Nested(UserSchema, attribute="created_by",
dump_to="created_by", only=["id", "name"], dump_only=True)
created_by = field_for(models.Job, 'created_by', required=True, load_only=True)
class Meta: class Meta:
model = models.Job model = models.Job

1
tests/app/job/test_rest.py
View File

@@ -71,6 +71,7 @@ def test_get_job_by_id(notify_api, sample_job):
assert response.status_code == 200 assert response.status_code == 200
resp_json = json.loads(response.get_data(as_text=True)) resp_json = json.loads(response.get_data(as_text=True))
assert resp_json['data']['id'] == job_id assert resp_json['data']['id'] == job_id
assert resp_json['data']['created_by']['name'] == 'Test User'
def test_create_job(notify_api, sample_template, mocker, fake_uuid): def test_create_job(notify_api, sample_template, mocker, fake_uuid):