add create service tests for dvla_org

This commit is contained in:
Leo Hemsted
2017-04-20 17:22:45 +01:00
parent f127bac600
commit 76676280ab

View File

@@ -129,9 +129,7 @@ def test_get_service_list_should_return_empty_list_if_no_services(notify_api, no
assert len(json_resp['data']) == 0 assert len(json_resp['data']) == 0
def test_get_service_by_id(notify_api, sample_service): def test_get_service_by_id(client, sample_service):
with notify_api.test_request_context():
with notify_api.test_client() as client:
auth_header = create_authorization_header() auth_header = create_authorization_header()
resp = client.get( resp = client.get(
'/service/{}'.format(sample_service.id), '/service/{}'.format(sample_service.id),
@@ -144,6 +142,7 @@ def test_get_service_by_id(notify_api, sample_service):
assert not json_resp['data']['research_mode'] assert not json_resp['data']['research_mode']
assert json_resp['data']['organisation'] is None assert json_resp['data']['organisation'] is None
assert json_resp['data']['branding'] == 'govuk' assert json_resp['data']['branding'] == 'govuk'
assert json_resp['data']['dvla_organisation'] == '001'
def test_get_service_by_id_should_404_if_no_service(notify_api, notify_db): def test_get_service_by_id_should_404_if_no_service(notify_api, notify_db):
@@ -191,9 +190,7 @@ def test_get_service_by_id_should_404_if_no_service_for_user(notify_api, sample_
assert json_resp['message'] == 'No result found' assert json_resp['message'] == 'No result found'
def test_create_service(notify_api, sample_user): def test_create_service(client, sample_user):
with notify_api.test_request_context():
with notify_api.test_client() as client:
data = { data = {
'name': 'created service', 'name': 'created service',
'user_id': str(sample_user.id), 'user_id': str(sample_user.id),
@@ -214,6 +211,7 @@ def test_create_service(notify_api, sample_user):
assert json_resp['data']['name'] == 'created service' assert json_resp['data']['name'] == 'created service'
assert json_resp['data']['email_from'] == 'created.service' assert json_resp['data']['email_from'] == 'created.service'
assert not json_resp['data']['research_mode'] assert not json_resp['data']['research_mode']
assert json_resp['data']['dvla_organisation'] == '001'
auth_header_fetch = create_authorization_header() auth_header_fetch = create_authorization_header()