mirror of
https://github.com/GSA/notifications-api.git
synced 2026-02-03 18:01:08 -05:00
Fix for job urls that should have been within context of a service.
The service id path parameter is captured in the url prefix for the job blueprint.
This commit is contained in:
@@ -2,9 +2,9 @@ import uuid
|
||||
|
||||
from app.dao.jobs_dao import (
|
||||
save_job,
|
||||
get_job_by_id,
|
||||
get_job,
|
||||
get_jobs_by_service,
|
||||
get_jobs
|
||||
_get_jobs
|
||||
)
|
||||
|
||||
from app.models import Job
|
||||
@@ -35,17 +35,39 @@ def test_save_job(notify_db, notify_db_session, sample_template):
|
||||
|
||||
|
||||
def test_get_job_by_id(notify_db, notify_db_session, sample_job):
|
||||
job_from_db = get_job_by_id(sample_job.id)
|
||||
job_from_db = get_job(sample_job.service.id, sample_job.id)
|
||||
assert sample_job == job_from_db
|
||||
|
||||
|
||||
def test_get_jobs_for_service(notify_db, notify_db_session, sample_job):
|
||||
def test_get_jobs_for_service(notify_db, notify_db_session, sample_template):
|
||||
|
||||
service_id = sample_job.service_id
|
||||
job_from_db = get_jobs_by_service(service_id)
|
||||
from tests.app.conftest import sample_job as create_job
|
||||
from tests.app.conftest import sample_service as create_service
|
||||
from tests.app.conftest import sample_template as create_template
|
||||
from tests.app.conftest import sample_user as create_user
|
||||
|
||||
assert len(job_from_db) == 1
|
||||
assert sample_job == job_from_db[0]
|
||||
one_job = create_job(notify_db, notify_db_session, sample_template.service,
|
||||
sample_template)
|
||||
|
||||
other_user = create_user(notify_db, notify_db_session,
|
||||
email="test@digital.cabinet-office.gov.uk")
|
||||
other_service = create_service(notify_db, notify_db_session,
|
||||
user=other_user)
|
||||
other_template = create_template(notify_db, notify_db_session,
|
||||
service=other_service)
|
||||
other_job = create_job(notify_db, notify_db_session, service=other_service,
|
||||
template=other_template)
|
||||
|
||||
one_job_from_db = get_jobs_by_service(one_job.service_id)
|
||||
other_job_from_db = get_jobs_by_service(other_job.service_id)
|
||||
|
||||
assert len(one_job_from_db) == 1
|
||||
assert one_job == one_job_from_db[0]
|
||||
|
||||
assert len(other_job_from_db) == 1
|
||||
assert other_job == other_job_from_db[0]
|
||||
|
||||
assert one_job_from_db != other_job_from_db
|
||||
|
||||
|
||||
def test_get_all_jobs(notify_db, notify_db_session, sample_template):
|
||||
@@ -55,5 +77,5 @@ def test_get_all_jobs(notify_db, notify_db_session, sample_template):
|
||||
notify_db_session,
|
||||
sample_template.service,
|
||||
sample_template)
|
||||
jobs_from_db = get_jobs()
|
||||
jobs_from_db = _get_jobs()
|
||||
assert len(jobs_from_db) == 5
|
||||
|
||||
@@ -9,12 +9,15 @@ from tests.app.conftest import sample_job as create_job
|
||||
def test_get_jobs(notify_api, notify_db, notify_db_session, sample_template):
|
||||
_setup_jobs(notify_db, notify_db_session, sample_template)
|
||||
|
||||
service_id = sample_template.service.id
|
||||
|
||||
with notify_api.test_request_context():
|
||||
with notify_api.test_client() as client:
|
||||
auth_header = create_authorization_header(service_id=sample_template.service.id,
|
||||
path=url_for('job.get_job'),
|
||||
path = url_for('job.get_job_for_service', service_id=service_id)
|
||||
auth_header = create_authorization_header(service_id=service_id,
|
||||
path=path,
|
||||
method='GET')
|
||||
response = client.get(url_for('job.get_job'), headers=[auth_header])
|
||||
response = client.get(path, headers=[auth_header])
|
||||
assert response.status_code == 200
|
||||
resp_json = json.loads(response.get_data(as_text=True))
|
||||
assert len(resp_json['data']) == 5
|
||||
@@ -23,12 +26,14 @@ def test_get_jobs(notify_api, notify_db, notify_db_session, sample_template):
|
||||
def test_get_job_with_invalid_id_returns400(notify_api, notify_db,
|
||||
notify_db_session,
|
||||
sample_template):
|
||||
service_id = sample_template.service.id
|
||||
with notify_api.test_request_context():
|
||||
with notify_api.test_client() as client:
|
||||
path = url_for('job.get_job_for_service', job_id='invalid_id', service_id=service_id)
|
||||
auth_header = create_authorization_header(service_id=sample_template.service.id,
|
||||
path=url_for('job.get_job', job_id='invalid_id'),
|
||||
path=path,
|
||||
method='GET')
|
||||
response = client.get(url_for('job.get_job', job_id='invalid_id'), headers=[auth_header])
|
||||
response = client.get(path, headers=[auth_header])
|
||||
assert response.status_code == 400
|
||||
resp_json = json.loads(response.get_data(as_text=True))
|
||||
assert resp_json == {'message': 'Invalid job id',
|
||||
@@ -39,12 +44,14 @@ def test_get_job_with_unknown_id_returns404(notify_api, notify_db,
|
||||
notify_db_session,
|
||||
sample_template):
|
||||
random_id = str(uuid.uuid4())
|
||||
service_id = sample_template.service.id
|
||||
with notify_api.test_request_context():
|
||||
with notify_api.test_client() as client:
|
||||
path = url_for('job.get_job_for_service', job_id=random_id, service_id=service_id)
|
||||
auth_header = create_authorization_header(service_id=sample_template.service.id,
|
||||
path=url_for('job.get_job', job_id=random_id),
|
||||
path=path,
|
||||
method='GET')
|
||||
response = client.get(url_for('job.get_job', job_id=random_id), headers=[auth_header])
|
||||
response = client.get(path, headers=[auth_header])
|
||||
assert response.status_code == 404
|
||||
resp_json = json.loads(response.get_data(as_text=True))
|
||||
assert resp_json == {'message': 'Job not found', 'result': 'error'}
|
||||
@@ -53,12 +60,14 @@ def test_get_job_with_unknown_id_returns404(notify_api, notify_db,
|
||||
def test_get_job_by_id(notify_api, notify_db, notify_db_session,
|
||||
sample_job):
|
||||
job_id = str(sample_job.id)
|
||||
service_id = sample_job.service.id
|
||||
with notify_api.test_request_context():
|
||||
with notify_api.test_client() as client:
|
||||
path = url_for('job.get_job_for_service', job_id=job_id, service_id=service_id)
|
||||
auth_header = create_authorization_header(service_id=sample_job.service.id,
|
||||
path=url_for('job.get_job', job_id=job_id),
|
||||
path=path,
|
||||
method='GET')
|
||||
response = client.get(url_for('job.get_job', job_id=job_id), headers=[auth_header])
|
||||
response = client.get(path, headers=[auth_header])
|
||||
assert response.status_code == 200
|
||||
resp_json = json.loads(response.get_data(as_text=True))
|
||||
assert resp_json['data']['id'] == job_id
|
||||
@@ -81,13 +90,14 @@ def test_post_job(notify_api, notify_db, notify_db_session, sample_template):
|
||||
}
|
||||
with notify_api.test_request_context():
|
||||
with notify_api.test_client() as client:
|
||||
path = url_for('job.create_job', service_id=service_id)
|
||||
auth_header = create_authorization_header(service_id=sample_template.service.id,
|
||||
path=url_for('job.create_job'),
|
||||
path=path,
|
||||
method='POST',
|
||||
request_body=json.dumps(data))
|
||||
headers = [('Content-Type', 'application/json'), auth_header]
|
||||
response = client.post(
|
||||
url_for('job.create_job'),
|
||||
path,
|
||||
data=json.dumps(data),
|
||||
headers=headers)
|
||||
assert response.status_code == 201
|
||||
|
||||
Reference in New Issue
Block a user