tests.scripts.test_send_tasks.SendTasksTestCase.test_sent_task() - Code Metrics - greenbone/gvm-tools - Measure and Improve Code Quality continuously with Scrutinizer

SendTasksTestCase.test_sent_task() A
last analyzed 2021-02-02 09:40 UTC

↳ Parent: tests.scripts.test_send_tasks

Complexity

Conditions

Size

Total Lines	58
Code Lines	21

Duplication

Lines	0
Ratio	0 %

Importance

Changes

Metric	Value
cc	2
eloc	21
nop	2
dl	0
loc	58
rs	9.376
c	0
b	0
f	0

How to fix Long Method

# -*- coding: utf-8 -*-
# Copyright (C) 2020-2021 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
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(
            (CWD.parent.parent / '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)

        configs_file = CWD / 'get_configs.xml'
        configs = configs_file.read_text()
        mock_gmp.mock_response('get_configs', configs)

        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)

        tasks = self.send_tasks.parse_send_xml_tree(mock_gmp.gmp_protocol, task)
        self.assertEqual(tasks, ['c8ef0597-e2c1-4e23-869f-072fa2914bf2'])


1			# -- coding: utf-8 --
2			# Copyright (C) 2020-2021 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			from unittest.mock import patch, MagicMock
22			from pathlib import Path
23			from lxml import etree
24			from . import GmpMockFactory, load_script
25
26			CWD = Path(__file__).absolute().parent
27
28
29			class SendTasksTestCase(unittest.TestCase):
30			def setUp(self):
31			self.send_tasks = load_script(
32			(CWD.parent.parent / 'scripts'), 'send-tasks'
33			)
34
35			@patch('builtins.input', lambda *args: 'y')
36			@patch('gvm.protocols.latest.Gmp', new_callable=GmpMockFactory)
37			def test_sent_task(self, mock_gmp: GmpMockFactory):
38			task_xml_path = CWD / 'example_task.xml'
39			task_xml_str = task_xml_path.read_text()
40
41			self.send_tasks.numerical_option = MagicMock(return_value=1)
42
43			configs_file = CWD / 'get_configs.xml'
44			configs = configs_file.read_text()
45			mock_gmp.mock_response('get_configs', configs)
46
47			mock_gmp.mock_response(
48			'get_scanners',
49			'<get_scanners_response status="200" status_text="OK">'
50			'<scanner id="c1c85af7-0cca-4690-8ccc-c79feb5588cf">'
51			'<name>as</name>'
52			'</scanner>'
53			'<scanner id="6acd0832-df90-11e4-b9d5-28d24461215b">'
54			'<name>CVE</name>'
55			'</scanner>'
56			'<scanner id="08b69003-5fc2-4037-a479-93b440211c73">'
57			'<name>OpenVAS Default</name>'
58			'</scanner>'
59			'</get_scanners_response>',
60			)
61
62			mock_gmp.mock_response(
63			'get_targets',
64			'<get_targets_response status="200" status_text="OK">'
65			'<target id="60f95d0e-029e-4931-a13a-b1d11260517d">'
66			'<name>own</name>'
67			'</target>'
68			'<target id="ead9576c-5a4d-4081-b98d-ccd77d5d16f8">'
69			'<name>Target for xn</name>'
70			'</target>'
71			'<target id="6c9f73f5-f14c-42bf-ab44-edb8d2493dbc">'
72			'<name>Unnamed</name>'
73			'</target>'
74			'<target id="a1f478c1-27d0-4d8c-959f-150625186421">'
75			'<name>work</name>'
76			'</target>'
77			'<target id="5ca97fe1-694d-4e4a-bd4c-55529719d17e">'
78			'<name>work2</name>'
79			'</target>'
80			'</get_targets_response>',
81			)
82
83			mock_gmp.mock_response(
84			'create_task',
85			'<create_task_response status="201" status_text="OK,'
86			'resource created" id="c8ef0597-e2c1-4e23-869f-072fa2914bf2"/>',
87			)
88
89			task = etree.XML(task_xml_str)
90
91			tasks = self.send_tasks.parse_send_xml_tree(mock_gmp.gmp_protocol, task)
92			self.assertEqual(tasks, ['c8ef0597-e2c1-4e23-869f-072fa2914bf2'])
93

greenbone / gvm-tools

SendTasksTestCase.test_sent_task() A last analyzed 2021-02-02 09:40 UTC

Complexity

Size

Duplication

Importance

How to fix Long Method

Long Method

Duplication Side-by-Side

Filter issues like

SendTasksTestCase.test_sent_task() A
last analyzed 2021-02-02 09:40 UTC