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_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")
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")
