tests.scripts.test_send_tasks.SendTasksTestCase.test_sent_task() - Code Metrics - Inspection of "Tests and refactor for the send-tasks script, scri..." - greenbone/gvm-tools - Measure and Improve Code Quality continuously with Scrutinizer

Passed

Pull Request — master (#317)

by Jaspar

created 2020-11-06 17:25 UTC

SendTasksTestCase.test_sent_task() A

↳ Parent: tests.scripts.test_send_tasks

Complexity

Conditions

Size

Total Lines	68
Code Lines	20

Duplication

Lines	0
Ratio	0 %

Importance

Changes

Metric	Value
cc	2
eloc	20
nop	2
dl	0
loc	68
rs	9.4
c	0
b	0
f	0

How to fix Long Method

# -*- coding: utf-8 -*-
# Copyright (C) 2020 Greenbone Networks GmbH
#
# SPDX-License-Identifier: GPL-3.0-or-later
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see<http://www.gnu.org/licenses/>


import unittest
import os
from unittest.mock import patch, MagicMock
from pathlib import Path
from lxml import etree
from . import GmpMockFactory, load_script


CWD = Path(__file__).absolute().parent


class SendTasksTestCase(unittest.TestCase):
    def setUp(self):
        self.send_tasks = load_script(
            os.path.join(CWD, '../../scripts'), 'send-tasks'
        )

    @patch('builtins.input', lambda *args: 'y')
    @patch('gvm.protocols.latest.Gmp', new_callable=GmpMockFactory)
    def test_sent_task(self, mock_gmp: GmpMockFactory):
        task_xml_path = CWD / 'example_task.xml'
        task_xml_str = task_xml_path.read_text()

        self.send_tasks.numerical_option = MagicMock(return_value=1)

        mock_gmp.mock_response(
            'get_configs',
            '<get_configs_response status="200" status_text="OK">'
            '<config id="d21f6c81-2b88-4ac1-b7b4-a2a9f2ad4663">'
            '<name>Base</name>'
            '</config>'
            '<config id="3fe6b460-e6ca-4af7-b712-1d7e9ea96eb0">'
            '<name>BSI TR-03116: Part 4 (Date: 10. Januar 2020)</name>'
            '</config>'
            '<config id="8715c877-47a0-438d-98a3-27c7a6ab2196">'
            '<name>Discovery</name>'
            '</config>'
            '</get_configs_response>',
        )

        mock_gmp.mock_response(
            'get_scanners',
            '<get_scanners_response status="200" status_text="OK">'
            '<scanner id="c1c85af7-0cca-4690-8ccc-c79feb5588cf">'
            '<name>as</name>'
            '</scanner>'
            '<scanner id="6acd0832-df90-11e4-b9d5-28d24461215b">'
            '<name>CVE</name>'
            '</scanner>'
            '<scanner id="08b69003-5fc2-4037-a479-93b440211c73">'
            '<name>OpenVAS Default</name>'
            '</scanner>'
            '</get_scanners_response>',
        )

        mock_gmp.mock_response(
            'get_targets',
            '<get_targets_response status="200" status_text="OK">'
            '<target id="60f95d0e-029e-4931-a13a-b1d11260517d">'
            '<name>own</name>'
            '</target>'
            '<target id="ead9576c-5a4d-4081-b98d-ccd77d5d16f8">'
            '<name>Target for xn</name>'
            '</target>'
            '<target id="6c9f73f5-f14c-42bf-ab44-edb8d2493dbc">'
            '<name>Unnamed</name>'
            '</target>'
            '<target id="a1f478c1-27d0-4d8c-959f-150625186421">'
            '<name>work</name>'
            '</target>'
            '<target id="5ca97fe1-694d-4e4a-bd4c-55529719d17e">'
            '<name>work2</name>'
            '</target>'
            '</get_targets_response>',
        )

        mock_gmp.mock_response(
            'create_task',
            '<create_task_response status="201" status_text="OK,'
            'resource created" id="c8ef0597-e2c1-4e23-869f-072fa2914bf2"/>',
        )

        task = etree.XML(task_xml_str)

        self.send_tasks.parse_send_xml_tree(mock_gmp.gmp_protocol, task)


1			# -- coding: utf-8 --
2			# Copyright (C) 2020 Greenbone Networks GmbH
3			#
4			# SPDX-License-Identifier: GPL-3.0-or-later
5			#
6			# This program is free software: you can redistribute it and/or modify
7			# it under the terms of the GNU General Public License as published by
8			# the Free Software Foundation, either version 3 of the License, or
9			# (at your option) any later version.
10			#
11			# This program is distributed in the hope that it will be useful,
12			# but WITHOUT ANY WARRANTY; without even the implied warranty of
13			# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14			# GNU General Public License for more details.
15			#
16			# You should have received a copy of the GNU General Public License
17			# along with this program. If not, see<http://www.gnu.org/licenses/>
18
19
20			import unittest
21			import os
22			from unittest.mock import patch, MagicMock
23			from pathlib import Path
24			from lxml import etree
25			from . import GmpMockFactory, load_script
26
27
28			CWD = Path(__file__).absolute().parent
29
30
31			class SendTasksTestCase(unittest.TestCase):
32			def setUp(self):
33			self.send_tasks = load_script(
34			os.path.join(CWD, '../../scripts'), 'send-tasks'
35			)
36
37			@patch('builtins.input', lambda *args: 'y')
38			@patch('gvm.protocols.latest.Gmp', new_callable=GmpMockFactory)
39			def test_sent_task(self, mock_gmp: GmpMockFactory):
40			task_xml_path = CWD / 'example_task.xml'
41			task_xml_str = task_xml_path.read_text()
42
43			self.send_tasks.numerical_option = MagicMock(return_value=1)
44
45			mock_gmp.mock_response(
46			'get_configs',
47			'<get_configs_response status="200" status_text="OK">'
48			'<config id="d21f6c81-2b88-4ac1-b7b4-a2a9f2ad4663">'
49			'<name>Base</name>'
50			'</config>'
51			'<config id="3fe6b460-e6ca-4af7-b712-1d7e9ea96eb0">'
52			'<name>BSI TR-03116: Part 4 (Date: 10. Januar 2020)</name>'
53			'</config>'
54			'<config id="8715c877-47a0-438d-98a3-27c7a6ab2196">'
55			'<name>Discovery</name>'
56			'</config>'
57			'</get_configs_response>',
58			)
59
60			mock_gmp.mock_response(
61			'get_scanners',
62			'<get_scanners_response status="200" status_text="OK">'
63			'<scanner id="c1c85af7-0cca-4690-8ccc-c79feb5588cf">'
64			'<name>as</name>'
65			'</scanner>'
66			'<scanner id="6acd0832-df90-11e4-b9d5-28d24461215b">'
67			'<name>CVE</name>'
68			'</scanner>'
69			'<scanner id="08b69003-5fc2-4037-a479-93b440211c73">'
70			'<name>OpenVAS Default</name>'
71			'</scanner>'
72			'</get_scanners_response>',
73			)
74
75			mock_gmp.mock_response(
76			'get_targets',
77			'<get_targets_response status="200" status_text="OK">'
78			'<target id="60f95d0e-029e-4931-a13a-b1d11260517d">'
79			'<name>own</name>'
80			'</target>'
81			'<target id="ead9576c-5a4d-4081-b98d-ccd77d5d16f8">'
82			'<name>Target for xn</name>'
83			'</target>'
84			'<target id="6c9f73f5-f14c-42bf-ab44-edb8d2493dbc">'
85			'<name>Unnamed</name>'
86			'</target>'
87			'<target id="a1f478c1-27d0-4d8c-959f-150625186421">'
88			'<name>work</name>'
89			'</target>'
90			'<target id="5ca97fe1-694d-4e4a-bd4c-55529719d17e">'
91			'<name>work2</name>'
92			'</target>'
93			'</get_targets_response>',
94			)
95
96			mock_gmp.mock_response(
97			'create_task',
98			'<create_task_response status="201" status_text="OK,'
99			'resource created" id="c8ef0597-e2c1-4e23-869f-072fa2914bf2"/>',
100			)
101
102			task = etree.XML(task_xml_str)
103
104			self.send_tasks.parse_send_xml_tree(mock_gmp.gmp_protocol, task)
105

greenbone / gvm-tools

Pull Request — master (#317)

SendTasksTestCase.test_sent_task() A

Complexity

Size

Duplication

Importance

How to fix Long Method

Long Method

Duplication Side-by-Side

Filter issues like