def test_delete_task_state(self, state_mock): base = BaseProgressReportTask() base._delete_task_state(args=["task_name", "course_id"]) state_mock.assert_called_with(ANY, "course_id") base._delete_task_state(args=None) state_mock.assert_called_with(ANY, "AllCourses")
def test_on_failure(self, del_mock): base = BaseProgressReportTask() base.on_failure("exc", "task_id", "args", "kwargs", "einfo") del_mock.assert_called_once_with("args")
def test_on_success(self, del_mock): base = BaseProgressReportTask() base.on_success("retval", "task_id", "args", "kwargs") del_mock.assert_called_once_with("args")