1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
# SPDX-FileCopyrightText: 2021-2024 Greenbone AG
#
# SPDX-License-Identifier: AGPL-3.0-or-later
"""Test module for command line arguments."""
import tempfile
import unittest
from pathlib import Path
from typing import List
from notus.scanner.cli.parser import Arguments, CliParser
from notus.scanner.config import (
DEFAULT_LOG_LEVEL,
DEFAULT_MQTT_BROKER_ADDRESS,
DEFAULT_MQTT_BROKER_PORT,
DEFAULT_PID_FILE,
DEFAULT_PRODUCTS_DIRECTORY,
)
class CliParserTestCase(unittest.TestCase):
def setUp(self):
self.parser = CliParser()
def parse_args(self, args: List[str]) -> Arguments:
return self.parser.parse_arguments(args)
def test_mqtt_broker(self):
args = self.parse_args(["--mqtt-broker-address=localhost"])
self.assertEqual("localhost", args.mqtt_broker_address)
args = self.parse_args(["-b", "localhost"])
self.assertEqual("localhost", args.mqtt_broker_address)
def test_mqtt_broker_port(self):
args = self.parse_args(["--mqtt-broker-port=12345"])
self.assertEqual(args.mqtt_broker_port, 12345)
args = self.parse_args(["-p", "12345"])
self.assertEqual(args.mqtt_broker_port, 12345)
def test_correct_upper_case_log_level(self):
args = self.parse_args(["--log-level=ERROR"])
self.assertEqual("ERROR", args.log_level)
def test_correct_lower_case_log_level(self):
args = self.parse_args(["-L", "info"])
self.assertEqual("INFO", args.log_level)
def test_advisories_directory(self):
args = self.parse_args(["--products-directory=/tmp"])
self.assertEqual(Path("/tmp"), args.products_directory)
def test_pid_file(self):
args = self.parse_args(["--pid-file=/foo/bar"])
self.assertEqual(args.pid_file, "/foo/bar")
def test_log_file(self):
args = self.parse_args(["--log-file=/foo/bar"])
self.assertEqual(args.log_file, "/foo/bar")
args = self.parse_args(["-l", "/foo/bar"])
self.assertEqual(args.log_file, "/foo/bar")
def test_foreground(self):
args = self.parse_args(["--foreground"])
self.assertTrue(args.foreground)
args = self.parse_args(["-f"])
self.assertTrue(args.foreground)
def test_disable_hashsum_verification(self):
args = self.parse_args(["--disable-hashsum-verification=true"])
self.assertTrue(args.disable_hashsum_verification)
def test_defaults(self):
args = self.parse_args([])
self.assertEqual(
args.products_directory, Path(DEFAULT_PRODUCTS_DIRECTORY)
)
self.assertIsNone(args.config)
self.assertIsNone(args.log_file)
self.assertEqual(args.log_level, DEFAULT_LOG_LEVEL)
self.assertEqual(args.mqtt_broker_port, DEFAULT_MQTT_BROKER_PORT)
self.assertEqual(args.mqtt_broker_address, DEFAULT_MQTT_BROKER_ADDRESS)
self.assertEqual(args.pid_file, DEFAULT_PID_FILE)
self.assertEqual(args.disable_hashsum_verification, False)
self.assertFalse(args.foreground)
def test_config_file_provide_mqtt_broker_address(self):
with tempfile.NamedTemporaryFile() as fp:
fp.write(b"[notus-scanner]\nmqtt-broker-address='1.2.3.4'")
fp.flush()
args = self.parse_args(["-c", fp.name])
self.assertEqual(args.mqtt_broker_address, "1.2.3.4")
def test_config_file(self):
with tempfile.NamedTemporaryFile() as fp:
fp.write(
b"""[notus-scanner]
mqtt-broker-address="1.2.3.4"
mqtt-broker-port="123"
products-directory="/tmp"
pid-file="foo.bar"
log-file="foo.log"
log-level="DEBUG"
"""
)
fp.flush()
args = self.parse_args(["-c", fp.name])
self.assertEqual(args.mqtt_broker_address, "1.2.3.4")
self.assertEqual(args.mqtt_broker_port, 123)
self.assertEqual(args.products_directory, Path("/tmp"))
self.assertEqual(args.pid_file, "foo.bar")
self.assertEqual(args.log_file, "foo.log")
self.assertEqual(args.log_level, "DEBUG")
|
