# IRIS Source Code # Copyright (C) 2023 - DFIR-IRIS # contact@dfir-iris.org # # This program is free software; you can redistribute it and/or # modify it under the terms of the GNU Lesser General Public # License as published by the Free Software Foundation; either # version 3 of the License, or (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU # Lesser General Public License for more details. # # You should have received a copy of the GNU Lesser General Public License # along with this program; if not, write to the Free Software Foundation, # Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. from unittest import TestCase from iris import Iris _IDENTIFIER_FOR_NONEXISTENT_OBJECT = 123456789 class TestsRestTasks(TestCase): def setUp(self) -> None: self._subject = Iris() def tearDown(self): self._subject.clear_database() def test_add_task_should_return_201(self): case_identifier = self._subject.create_dummy_case() body = {'task_assignees_id': [], 'task_description': '', 'task_status_id': 1, 'task_tags': '', 'task_title': 'dummy title', 'custom_attributes': {}} response = self._subject.create(f'/api/v2/cases/{case_identifier}/tasks', body) self.assertEqual(201, response.status_code) def test_add_task_with_missing_task_title_identifier_should_return_400(self): case_identifier = self._subject.create_dummy_case() body = {'task_assignees_id': [], 'task_description': '', 'task_status_id': 1, 'task_tags': '', 'custom_attributes': {}} response = self._subject.create(f'/api/v2/cases/{case_identifier}/tasks', body) self.assertEqual(400, response.status_code) def test_create_case_with_spurious_slash_should_return_404(self): case_identifier = self._subject.create_dummy_case() body = {'task_assignees_id': [], 'task_description': '', 'task_status_id': 1, 'task_tags': '', 'task_title': 'dummy title', 'custom_attributes': {}} response = self._subject.create(f'/api/v2/cases/{case_identifier}/tasks/', body) self.assertEqual(404, response.status_code) def test_get_task_should_return_200(self): case_identifier = self._subject.create_dummy_case() body = {'task_assignees_id': [], 'task_description': '', 'task_status_id': 1, 'task_tags': '', 'task_title': 'dummy title', 'custom_attributes': {}} response = self._subject.create(f'/api/v2/cases/{case_identifier}/tasks', body).json() task_identifier = response['id'] response = self._subject.get(f'/api/v2/cases/{case_identifier}/tasks/{task_identifier}') self.assertEqual(200, response.status_code) def test_get_task_with_missing_task_identifier_should_return_error(self): case_identifier = self._subject.create_dummy_case() body = {'task_assignees_id': [], 'task_description': '', 'task_status_id': 1, 'task_tags': '', 'task_title': 'dummy title', 'custom_attributes': {}} self._subject.create(f'/api/v2/cases/{case_identifier}/tasks', body) response = self._subject.get(f'/api/v2/cases/{case_identifier}/tasks/{_IDENTIFIER_FOR_NONEXISTENT_OBJECT}') self.assertEqual(404, response.status_code) def test_delete_task_should_return_204(self): case_identifier = self._subject.create_dummy_case() body = {'task_assignees_id': [], 'task_description': '', 'task_status_id': 1, 'task_tags': '', 'task_title': 'dummy title', 'custom_attributes': {}} response = self._subject.create(f'/api/v2/cases/{case_identifier}/tasks', body).json() task_identifier = response['id'] test = self._subject.delete(f'/api/v2/cases/{case_identifier}/tasks/{task_identifier}') self.assertEqual(204, test.status_code) def test_delete_task_with_missing_task_identifier_should_return_404(self): case_identifier = self._subject.create_dummy_case() body = {'task_assignees_id': [], 'task_description': '', 'task_status_id': 1, 'task_tags': '', 'task_title': 'dummy title', 'custom_attributes': {}} self._subject.create(f'/api/v2/cases/{case_identifier}/tasks', body) test = self._subject.delete(f'/api/v2/cases/{case_identifier}/tasks/{_IDENTIFIER_FOR_NONEXISTENT_OBJECT}') self.assertEqual(404, test.status_code) def test_get_user_task_should_not_fail(self): case_identifier = self._subject.create_dummy_case() user = self._subject.create_dummy_user() body = {'task_assignees_id': [user.get_identifier()], 'task_description': '', 'task_status_id': 1, 'task_tags': '', 'task_title': 'dummy title', 'custom_attributes': {}} self._subject.create(f'/api/v2/cases/{case_identifier}/tasks', body) response = user.get('/user/tasks/list') self.assertEqual(200, response.status_code) def test_get_user_task_should_contain_task_case_field(self): case_identifier = self._subject.create_dummy_case() user = self._subject.create_dummy_user() body = {'task_assignees_id': [user.get_identifier()], 'task_description': '', 'task_status_id': 1, 'task_tags': '', 'task_title': 'dummy title', 'custom_attributes': {}} self._subject.create(f'/api/v2/cases/{case_identifier}/tasks', body) response = user.get('/user/tasks/list').json() self.assertEqual(f'#{case_identifier} - case name', response['data']['tasks'][0]['task_case']) def test_update_task_should_not_fail(self): case_identifier = self._subject.create_dummy_case() body = {'task_assignees_id': [], 'task_status_id': 1, 'task_title': 'dummy title'} response = self._subject.create(f'/api/v2/cases/{case_identifier}/tasks', body).json() identifier = response['id'] response = self._subject.update(f'/api/v2/cases/{case_identifier}/tasks/{identifier}', {'task_title': 'new title', 'task_status_id': 1, 'task_assignees_id': []}) self.assertEqual(200, response.status_code) def test_update_task_should_update_assignees(self): case_identifier = self._subject.create_dummy_case() user = self._subject.create_dummy_user() body = {'task_assignees_id': [], 'task_status_id': 1, 'task_title': 'dummy title'} response = self._subject.create(f'/api/v2/cases/{case_identifier}/tasks', body).json() identifier = response['id'] self._subject.update(f'/api/v2/cases/{case_identifier}/tasks/{identifier}', {'task_title': 'dummy title', 'task_status_id': 1, 'task_assignees_id': [user.get_identifier()]}) response = self._subject.get('/case/tasks/list', query_parameters={'cid': case_identifier}).json() self.assertEqual(user.get_identifier(), response['data']['tasks'][0]['task_assignees'][0]['id']) def test_update_task_without_task_status_id_should_return_400(self): case_identifier = self._subject.create_dummy_case() body = {'task_assignees_id': [], 'task_status_id': 1, 'task_title': 'dummy title'} response = self._subject.create(f'/api/v2/cases/{case_identifier}/tasks', body).json() identifier = response['id'] response = self._subject.update(f'/api/v2/cases/{case_identifier}/tasks/{identifier}', {'task_title': 'new title', 'task_assignees_id': []}) self.assertEqual(400, response.status_code) def test_update_task_should_return_a_task(self): case_identifier = self._subject.create_dummy_case() body = {'task_assignees_id': [], 'task_status_id': 1, 'task_title': 'dummy title'} response = self._subject.create(f'/api/v2/cases/{case_identifier}/tasks', body).json() identifier = response['id'] response = self._subject.update(f'/api/v2/cases/{case_identifier}/tasks/{identifier}', {'task_title': 'new title', 'task_status_id': 1, 'task_assignees_id': []}).json() self.assertEqual('new title', response['task_title']) def test_get_tasks_should_return_200(self): case_identifier = self._subject.create_dummy_case() response = self._subject.get(f'/api/v2/cases/{case_identifier}/tasks') self.assertEqual(200, response.status_code) def test_get_tasks_should_return_empty_list_for_field_data_when_there_are_no_tasks(self): case_identifier = self._subject.create_dummy_case() response = self._subject.get(f'/api/v2/cases/{case_identifier}/tasks').json() self.assertEqual([], response['data']) def test_get_tasks_should_return_total(self): case_identifier = self._subject.create_dummy_case() body = {'task_assignees_id': [], 'task_status_id': 1, 'task_title': 'dummy title'} self._subject.create(f'/api/v2/cases/{case_identifier}/tasks', body).json() response = self._subject.get(f'/api/v2/cases/{case_identifier}/tasks').json() self.assertEqual(1, response['total']) def test_get_tasks_should_honour_per_page_pagination_parameter(self): case_identifier = self._subject.create_dummy_case() body = {'task_assignees_id': [], 'task_status_id': 1, 'task_title': 'task1'} self._subject.create(f'/api/v2/cases/{case_identifier}/tasks', body).json() body = {'task_assignees_id': [], 'task_status_id': 1, 'task_title': 'task2'} self._subject.create(f'/api/v2/cases/{case_identifier}/tasks', body).json() body = {'task_assignees_id': [], 'task_status_id': 1, 'task_title': 'task3'} self._subject.create(f'/api/v2/cases/{case_identifier}/tasks', body).json() response = self._subject.get(f'/api/v2/cases/{case_identifier}/tasks', { 'per_page': 2 }).json() self.assertEqual(2, len(response['data'])) def test_get_tasks_should_return_current_page(self): case_identifier = self._subject.create_dummy_case() body = {'task_assignees_id': [], 'task_status_id': 1, 'task_title': 'task1'} self._subject.create(f'/api/v2/cases/{case_identifier}/tasks', body).json() body = {'task_assignees_id': [], 'task_status_id': 1, 'task_title': 'task2'} self._subject.create(f'/api/v2/cases/{case_identifier}/tasks', body).json() body = {'task_assignees_id': [], 'task_status_id': 1, 'task_title': 'task3'} self._subject.create(f'/api/v2/cases/{case_identifier}/tasks', body).json() response = self._subject.get(f'/api/v2/cases/{case_identifier}/tasks', { 'page': 2, 'per_page': 2 }).json() self.assertEqual(2, response['current_page']) def test_get_tasks_should_return_correct_task_uuid(self): case_identifier = self._subject.create_dummy_case() body = {'task_assignees_id': [], 'task_status_id': 1, 'task_title': 'title'} response = self._subject.create(f'/api/v2/cases/{case_identifier}/tasks', body).json() identifier = response['id'] response = self._subject.get(f'/api/v2/tasks/{identifier}').json() expected_uuid = response['task_uuid'] response = self._subject.get(f'/api/v2/cases/{case_identifier}/tasks').json() self.assertEqual(expected_uuid, response['data'][0]['task_uuid'])