tests.protocols.osp.test_osp_start_scan

Complexity

Total Complexity

5

Size/Duplication

Total Lines	103
Duplicated Lines	0 %

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	OSPStartScanTestCase.test_start_scan_legacy()	0	9	1
A	OSPStartScanTestCase.test_start_scan_without_target()	0	3	2
A	OSPStartScanTestCase.setUp()	0	3	1
A	OSPStartScanTestCase.test_start_scan()	0	35	1

# -*- coding: utf-8 -*-
# Copyright (C) 2018 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 gvm.errors import RequiredArgument
from gvm.protocols.ospv1 import Osp

from .. import MockConnection

class OSPStartScanTestCase(unittest.TestCase):
    def setUp(self):
        self.connection = MockConnection()
        self.osp = Osp(self.connection)

    def test_start_scan(self):
        scanner_params = dict()
        scanner_params['key1'] = 'value1'

        targets = list()
        _target1 = dict()
        _target1['hosts'] = 'localhost'
        _target1['ports'] = '22,80'
        targets.append(_target1)

        _target2 = dict()
        _target2['hosts'] = '192.168.10.1'
        _target2['ports'] = '443'
        _credential1 = dict()
        _credential1 = {'smb': {'username': 'username',
                               'password': 'pass',

Wrong continued indentation (add 1 space).

                               'port': 'port',

Wrong continued indentation (add 1 space).

                               'type': 'type'}}

Wrong continued indentation (add 1 space).

        _target2['credentials'] = _credential1
        targets.append(_target2)

        vts = dict()
        vts['vt1'] = {'value_id': 'value'}
        vts['vt_groups'] = ['family=A', 'family=B']

        self.osp.start_scan(
            scan_id='123-456',
            parallel='10',
            scanner_params=scanner_params,
            targets=targets,
            vt_selection=vts
        )

        self.connection.send.has_been_called_with(

The Method send does not seem to have a member named has_been_called_with.

            '<start_scan scan_id="123-456" parallel="10">'
            '<scanner_params key1="value1"/>'
            '<targets><target><hosts>localhost</hosts>'
            '<ports>22,80</ports></target>'
            '<target><hosts>192.168.10.1</hosts>'
            '<ports>443</ports>'
            '<credentials>'
            '<credential port="port" service="smb" type="type">'
            '<username>username</username>'
            '<password>pass</password>'
            '</credential></credentials>'
            '</target></targets><vt_selection>'
            '<vt_single id="vt1">'
            '<vt_value id="value_id">value</vt_value></vt_single>'
            '<vt_group filter="family=A"/><vt_group filter="family=B"/>'
            '</vt_selection></start_scan>'
        )

    def test_start_scan_without_target(self):
        with self.assertRaises(RequiredArgument):
             self.osp.start_scan()

The indentation here looks off. 12 spaces were expected, but 13 were found.

    def test_start_scan_legacy(self):
        self.osp.start_scan(
            scan_id='123-456',
            parallel='10',
            target="localhost",
            ports="22"
        )
        self.connection.send.has_been_called_with(

The Method send does not seem to have a member named has_been_called_with.

            '<start_scan scan_id="123-456" parallel="10" '
            'target="localhost" ports="22">'
            '<scanner_params/></start_scan>'
        )


if __name__ == '__main__':
    unittest.main()