tests.test_argument_parser

Complexity

Total Complexity

13

Size/Duplication

Total Lines	89
Duplicated Lines	0 %

Importance

Changes

0

13 Methods

Rating	Name	Duplication	Size	Complexity
A	ArgumentParserTestCase.test_default_address()	0	3	1
A	ArgumentParserTestCase.test_correct_log_level()	0	3	1
A	ArgumentParserTestCase.test_address_param()	0	3	1
A	ArgumentParserTestCase.test_default_key()	0	3	1
A	ArgumentParserTestCase.test_correct_upper_case_log_level()	0	3	1
A	ArgumentParserTestCase.test_port_interval()	0	9	1
A	ArgumentParserTestCase.test_default_port()	0	3	1
A	ArgumentParserTestCase.test_non_existing_key()	0	3	1
A	ArgumentParserTestCase.test_existing_key()	0	3	1
A	ArgumentParserTestCase.test_port_as_string()	0	3	1
A	ArgumentParserTestCase.setUp()	0	2	1
A	ArgumentParserTestCase.test_default_log_level()	0	3	1
A	ArgumentParserTestCase.test_correct_lower_case_log_level()	0	3	1

# Copyright (C) 2014-2018 Greenbone Networks GmbH
#
# SPDX-License-Identifier: GPL-2.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 2
# 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, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

""" Test module for command line arguments.
"""

import logging
import unittest

from ospd import misc
from ospd.misc import create_args_parser, get_common_args


class ArgumentParserTestCase(unittest.TestCase):
    def setUp(self):
        self.parser = create_args_parser('Wrapper name')

    def test_port_interval(self):
        self.assertRaises(
            SystemExit, get_common_args, self.parser, '--port=65536'.split()
        )
        self.assertRaises(
            SystemExit, get_common_args, self.parser, '--port=0'.split()
        )
        args = get_common_args(self.parser, '--port=3353'.split())
        self.assertEqual(3353, args['port'])

    def test_port_as_string(self):
        self.assertRaises(
            SystemExit, get_common_args, self.parser, '--port=abcd'.split()
        )

    def test_default_port(self):
        args = get_common_args(self.parser, [])
        self.assertEqual(misc.PORT, args['port'])

    def test_default_address(self):
        args = get_common_args(self.parser, [])
        self.assertEqual(misc.ADDRESS, args['address'])

    def test_address_param(self):
        args = get_common_args(self.parser, '-b 1.2.3.4'.split())
        self.assertEqual('1.2.3.4', args['address'])

    def test_default_log_level(self):
        args = get_common_args(self.parser, [])
        self.assertEqual(logging.WARNING, args['log_level'])

    def test_correct_lower_case_log_level(self):
        args = get_common_args(self.parser, '-L error'.split())
        self.assertEqual(logging.ERROR, args['log_level'])

    def test_correct_upper_case_log_level(self):
        args = get_common_args(self.parser, '-L INFO'.split())
        self.assertEqual(logging.INFO, args['log_level'])

    def test_correct_log_level(self):
        self.assertRaises(
            SystemExit, get_common_args, self.parser, '-L blah'.split()
        )

    def test_non_existing_key(self):
        self.assertRaises(
            SystemExit, get_common_args, self.parser, '-k abcdef.ghijkl'.split()
        )

    def test_existing_key(self):
        args = get_common_args(self.parser, '-k /etc/passwd'.split())
        self.assertEqual('/etc/passwd', args['keyfile'])

    def test_default_key(self):
        args = get_common_args(self.parser, [])
        self.assertEqual(misc.KEY_FILE, args['keyfile'])