mirror of
https://github.com/GSA/notifications-api.git
synced 2026-04-20 09:20:09 -04:00
pytest fixes
This commit is contained in:
@@ -75,7 +75,9 @@ def dao_get_jobs_by_service_id(
|
||||
stmt = (
|
||||
select(Job)
|
||||
.where(*query_filter)
|
||||
.order_by(func.coalesce(Job.processing_started, Job.created_at).desc())
|
||||
.order_by(
|
||||
func.coalesce(Job.processing_started, Job.created_at).desc(), Job.id.desc()
|
||||
)
|
||||
.limit(page_size)
|
||||
.offset(offset)
|
||||
)
|
||||
|
||||
@@ -209,12 +209,18 @@ def test_get_jobs_for_service_in_processed_at_then_created_at_order(
|
||||
),
|
||||
]
|
||||
|
||||
expected_order = sorted(
|
||||
created_jobs,
|
||||
key=lambda job: ((job.processing_started or job.created_at), str(job.id)),
|
||||
reverse=True,
|
||||
)
|
||||
|
||||
jobs = dao_get_jobs_by_service_id(sample_template.service.id).items
|
||||
|
||||
assert len(jobs) == len(created_jobs)
|
||||
assert len(jobs) == len(expected_order)
|
||||
|
||||
for index in range(0, len(created_jobs)):
|
||||
assert jobs[index].id == created_jobs[index].id
|
||||
for index in range(len(expected_order)):
|
||||
assert jobs[index].id == expected_order[index].id
|
||||
|
||||
|
||||
def test_update_job(sample_job):
|
||||
|
||||
@@ -1070,16 +1070,30 @@ def test_get_jobs_should_retrieve_from_ft_notification_status_for_old_jobs(
|
||||
service_id=sample_template.service_id,
|
||||
)
|
||||
|
||||
assert resp_json["data"][0]["id"] == str(job_3.id)
|
||||
assert resp_json["data"][0]["statistics"] == []
|
||||
assert resp_json["data"][1]["id"] == str(job_2.id)
|
||||
assert resp_json["data"][1]["statistics"] == [
|
||||
{"status": NotificationStatus.CREATED, "count": 1},
|
||||
]
|
||||
assert resp_json["data"][2]["id"] == str(job_1.id)
|
||||
assert resp_json["data"][2]["statistics"] == [
|
||||
{"status": NotificationStatus.DELIVERED, "count": 6},
|
||||
]
|
||||
returned_jobs = resp_json["data"]
|
||||
|
||||
expected_jobs = [job_3, job_2, job_1]
|
||||
expected_order = sorted(
|
||||
expected_jobs,
|
||||
key=lambda job: ((job.processing_started or job.created_at), str(job.id)),
|
||||
reverse=True,
|
||||
)
|
||||
expected_ids = [str(job.id) for job in expected_order]
|
||||
returned_ids = [job["id"] for job in returned_jobs if job["id"] in expected_ids]
|
||||
assert returned_ids == expected_ids
|
||||
|
||||
for job in expected_jobs:
|
||||
idx = returned_ids.index(str(job.id))
|
||||
if job is job_3:
|
||||
assert returned_jobs[idx]["statistics"] == []
|
||||
elif job is job_2:
|
||||
assert returned_jobs[idx]["statistics"] == [
|
||||
{"status": NotificationStatus.CREATED, "count": 1},
|
||||
]
|
||||
elif job is job_1:
|
||||
assert returned_jobs[idx]["statistics"] == [
|
||||
{"status": NotificationStatus.DELIVERED, "count": 6},
|
||||
]
|
||||
|
||||
|
||||
@freeze_time("2017-07-17 07:17")
|
||||
|
||||
Reference in New Issue
Block a user