def test_split_task_helper( self, mock_task_get, mock_task_get_max_task_id_for_project, mock_task_create, mock_task_delete, mock_project_get, mock_project_save, mock_project_tasks, mock_instructions, ): if self.skip_tests: return # arrange task_stub = Task() task_stub.id = 1 task_stub.project_id = 1 task_stub.task_status = 1 task_stub.locked_by = 1234 task_stub.lock_holder = 1234 task_stub.is_square = True task_stub.x = 16856 task_stub.y = 17050 task_stub.zoom = 15 task_stub.geometry = shape.from_shape( Polygon( [ (5.1855468740711421, 7.2970875628719796), (5.1855468740711421, 7.3079847788619219), (5.1965332021941588, 7.3079847788619219), (5.1965332021941588, 7.2970875628719796), (5.1855468740711421, 7.2970875628719796), ] ) ) mock_task_get.return_value = task_stub mock_task_get_max_task_id_for_project.return_value = 1 mock_project_get.return_value = Project() mock_project_tasks.return_value = [task_stub] splitTaskDTO = SplitTaskDTO() splitTaskDTO.user_id = 1234 splitTaskDTO.project_id = 1 splitTaskDTO.task_id = 1 # act result = SplitService.split_task(splitTaskDTO) # assert self.assertEqual(4, len(result.tasks))
def test_record_auto_unlock_adds_autounlocked_action( self, mock_update, mock_set_task_history, mock_get_last_action, mock_get_last_status, ): mock_history = MagicMock() mock_last_action = MagicMock() mock_last_action.action = "LOCKED_FOR_MAPPING" mock_get_last_action.return_value = mock_last_action mock_get_last_status.return_value = TaskStatus.READY mock_set_task_history.return_value = mock_history test_task = Task() test_task.locked_by = "testuser" lock_duration = "02:00" test_task.record_auto_unlock(lock_duration) mock_set_task_history.assert_called_with( action=TaskAction.AUTO_UNLOCKED_FOR_MAPPING, user_id="testuser") self.assertEqual(mock_history.action_text, lock_duration) self.assertEqual(test_task.locked_by, None) mock_last_action.delete.assert_called()