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()
