darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-03 09:51:11 -05:00
Code Issues Packages Projects Releases Wiki Activity

add test for populate_go_live

Browse Source
This commit is contained in:
Kenneth Kehl
2024-10-23 12:28:24 -07:00
parent b59a71ec54
commit 430dd37f0e
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/app/test_commands.py
View File

@@ -534,7 +534,7 @@ def test_populate_go_live_success(notify_api, mocker):
mock_logger.info.assert_any_call("Populate go live user and date") mock_logger.info.assert_any_call("Populate go live user and date")
def test_process_row_from_job_success(mocker): def test_process_row_from_job_success(notify_api, mocker):
mock_current_app = mocker.patch("app.commands.current_app") mock_current_app = mocker.patch("app.commands.current_app")
mock_logger = mock_current_app.logger mock_logger = mock_current_app.logger
mock_dao_get_job_by_id = mocker.patch("app.commands.dao_get_job_by_id") mock_dao_get_job_by_id = mocker.patch("app.commands.dao_get_job_by_id")
@@ -559,10 +559,14 @@ def test_process_row_from_job_success(mocker):
mock_dao_get_template_by_id.return_value = mock_template mock_dao_get_template_by_id.return_value = mock_template
mock_get_job_from_s3.return_value = "some_csv_content" mock_get_job_from_s3.return_value = "some_csv_content"
mock_process_row.return_value = "notification_123" mock_process_row.return_value = "notification_123"
process_row_from_job("job_456", 2)
notify_api.test_cli_runner().invoke(
process_row_from_job,
["-j", "job_456", "-n", "2"],
)
mock_dao_get_job_by_id.assert_called_once_with("job_456") mock_dao_get_job_by_id.assert_called_once_with("job_456")
mock_dao_get_template_by_id.assert_called_once_with( mock_dao_get_template_by_id.assert_called_once_with(
mock_job.tempalte_id, mock_job.template_version mock_job.template_id, mock_job.template_version
) )
mock_get_job_from_s3.assert_called_once_with( mock_get_job_from_s3.assert_called_once_with(
str(mock_job.service_id), str(mock_job.id) str(mock_job.service_id), str(mock_job.id)