Update json structure for unit tests.

This commit is contained in:
Rebecca Law
2016-05-16 11:53:22 +01:00
parent fcf277c413
commit d4225de45e
3 changed files with 12 additions and 6 deletions

View File

@@ -57,7 +57,9 @@ def view_jobs(service_id):
@user_has_permissions('view_activity', admin_override=True) @user_has_permissions('view_activity', admin_override=True)
def view_job(service_id, job_id): def view_job(service_id, job_id):
job = job_api_client.get_job(service_id, job_id)['data'] job = job_api_client.get_job(service_id, job_id)['data']
template = service_api_client.get_service_template(service_id, job['template'], job['template_version'])['data'] template = service_api_client.get_service_template(service_id=service_id,
template_id=job['template'],
version=job['template_version'])['data']
notifications = notification_api_client.get_notifications_for_service(service_id, job_id) notifications = notification_api_client.get_notifications_for_service(service_id, job_id)
finished = job['status'] == 'finished' finished = job['status'] == 'finished'
return render_template( return render_template(

View File

@@ -53,14 +53,14 @@ def template_json(service_id,
type_="sms", type_="sms",
content="template content", content="template content",
subject=None, subject=None,
versions=['1']): versions='1'):
template = { template = {
'id': id_, 'id': id_,
'name': name, 'name': name,
'template_type': type_, 'template_type': type_,
'content': content, 'content': content,
'service': service_id, 'service': service_id,
'versions': versions 'version': versions
} }
if subject is not None: if subject is not None:
template['subject'] = subject template['subject'] = subject
@@ -83,7 +83,6 @@ def template_version_json(service_id,
created_at = datetime.utcnow().strftime('%Y-%m-%d %H:%M:%S.%f') created_at = datetime.utcnow().strftime('%Y-%m-%d %H:%M:%S.%f')
template['created_at'] = created_at template['created_at'] = created_at
template['version'] = version template['version'] = version
del template['versions']
return template return template
@@ -129,6 +128,7 @@ def job_json():
'id': job_id, 'id': job_id,
'service': 1, 'service': 1,
'template': 1, 'template': 1,
'template_version': 1,
'original_file_name': 'thisisatest.csv', 'original_file_name': 'thisisatest.csv',
'created_at': created_at, 'created_at': created_at,
'notification_count': 1, 'notification_count': 1,
@@ -143,6 +143,7 @@ def job_json_with_created_by(service_id=None, job_id=None):
'id': job_id if job_id else str(generate_uuid()), 'id': job_id if job_id else str(generate_uuid()),
'service': service_id if service_id else str(generate_uuid()), 'service': service_id if service_id else str(generate_uuid()),
'template': 1, 'template': 1,
'template_version': 1,
'original_file_name': 'thisisatest.csv', 'original_file_name': 'thisisatest.csv',
'created_at': str(datetime.now().time()), 'created_at': str(datetime.now().time()),
'notification_count': 1, 'notification_count': 1,
@@ -190,7 +191,8 @@ def notification_json(service_id,
'sent_at': sent_at, 'sent_at': sent_at,
'status': status, 'status': status,
'created_at': created_at, 'created_at': created_at,
'updated_at': updated_at 'updated_at': updated_at,
'template_version': template['version']
} for i in range(5)], } for i in range(5)],
'total': 5, 'total': 5,
'page_size': 50, 'page_size': 50,

View File

@@ -203,9 +203,11 @@ def mock_get_aggregate_service_statistics(mocker):
@pytest.fixture(scope='function') @pytest.fixture(scope='function')
def mock_get_service_template(mocker): def mock_get_service_template(mocker):
def _get(service_id, template_id): def _get(service_id, template_id, version=None):
template = template_json( template = template_json(
service_id, template_id, "Two week reminder", "sms", "Your vehicle tax is about to expire") service_id, template_id, "Two week reminder", "sms", "Your vehicle tax is about to expire")
if version:
template.update({'version': version})
return {'data': template} return {'data': template}
return mocker.patch( return mocker.patch(