mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-08-18 13:39:41 -04:00
32 lines
789 B
Python
32 lines
789 B
Python
|
import pytest
|
||
|
|
|
||
|
|
from app.models.job import Job
|
||
|
|
from tests import job_json, user_json
|
||
|
|
from tests.conftest import SERVICE_ONE_ID
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
'job_status, num_notifications_created, expected_still_processing',
|
||
|
|
[
|
||
|
|
('scheduled', 0, True),
|
||
|
|
('cancelled', 10, True),
|
||
|
|
('finished', 5, True),
|
||
|
|
('finished', 10, False),
|
||
|
|
]
|
||
|
|
)
|
||
|
|
def test_still_processing(
|
||
|
|
notify_admin,
|
||
|
|
job_status,
|
||
|
|
num_notifications_created,
|
||
|
|
expected_still_processing
|
||
|
|
):
|
||
|
|
json = job_json(
|
||
|
|
service_id=SERVICE_ONE_ID,
|
||
|
|
created_by=user_json(),
|
||
|
|
notification_count=10,
|
||
|
|
notifications_requested=num_notifications_created,
|
||
|
|
job_status=job_status
|
||
|
|
)
|
||
|
|
job = Job(json)
|
||
|
|
assert job.still_processing == expected_still_processing
|