diff --git a/tests/integration_tests/reports/scheduler_tests.py b/tests/integration_tests/reports/scheduler_tests.py index 3284ee9772469..29dd58273a2c2 100644 --- a/tests/integration_tests/reports/scheduler_tests.py +++ b/tests/integration_tests/reports/scheduler_tests.py @@ -171,7 +171,7 @@ def test_execute_task(update_state_mock, command_mock, init_mock, owners): init_mock.return_value = None command_mock.side_effect = ReportScheduleUnexpectedError("Unexpected error") with freeze_time("2020-01-01T09:00:00Z"): - execute(report_schedule.id, "2020-01-01T09:00:00Z") + execute(report_schedule.id) update_state_mock.assert_called_with(state="FAILURE") db.session.delete(report_schedule) @@ -199,7 +199,7 @@ def test_execute_task_with_command_exception( init_mock.return_value = None command_mock.side_effect = CommandException("Unexpected error") with freeze_time("2020-01-01T09:00:00Z"): - execute(report_schedule.id, "2020-01-01T09:00:00Z") + execute(report_schedule.id) update_state_mock.assert_called_with(state="FAILURE") logger_mock.exception.assert_called_with( "A downstream exception occurred while generating a report: None. Unexpected error",