def test_set_task_state(self):
self.cache_mock.get.return_value = {"course_id": "date"}
state = TaskState("task_name", "course_id2")
state.set_task_state()
self.cache_mock.set.assert_called_with(key=state.key, value={"course_id": "date", "course_id2": ANY})
self.cache_mock.get.return_value = None
state.set_task_state()
self.cache_mock.set.assert_called_with(key=state.key, value={"course_id2": ANY})
def test_set_task_state(self):
self.cache_mock.get.return_value = {"course_id": "date"}
state = TaskState("task_name", "course_id2")
state.set_task_state()
self.cache_mock.set.assert_called_with(key=state.key, value={
'course_id': 'date', 'course_id2': ANY})
self.cache_mock.get.return_value = None
state.set_task_state()
self.cache_mock.set.assert_called_with(key=state.key, value={
'course_id2': ANY})
