# SPDX-FileCopyrightText: 2023-2024 Greenbone AG # # SPDX-License-Identifier: GPL-3.0-or-later # # pylint: disable=line-too-long, too-many-lines import io import sys import unittest from contextlib import redirect_stderr, redirect_stdout from pathlib import Path from unittest.mock import MagicMock, patch from pontos.testing import temp_file from greenbone.feed.sync.config import ( DEFAULT_CONFIG_FILE, DEFAULT_DESTINATION_PREFIX, DEFAULT_ENTERPRISE_KEY_PATH, DEFAULT_FEED_RELEASE, DEFAULT_GVMD_LOCK_FILE_PATH, DEFAULT_OPENVAS_LOCK_FILE_PATH, DEFAULT_USER_CONFIG_FILE, ) from greenbone.feed.sync.errors import ConfigFileError from greenbone.feed.sync.helper import DEFAULT_FLOCK_WAIT_INTERVAL from greenbone.feed.sync.parser import CliParser, feed_type from greenbone.feed.sync.rsync import ( DEFAULT_RSYNC_COMPRESSION_LEVEL, DEFAULT_RSYNC_URL, ) class FeedTypeTestCase(unittest.TestCase): def test_nvt(self): self.assertEqual("nvt", feed_type("nvt")) self.assertEqual("nvt", feed_type("nvts")) self.assertEqual("nvt", feed_type("NVT")) self.assertEqual("nvt", feed_type("NVTS")) def test_report_format(self): self.assertEqual("report-format", feed_type("report-format")) self.assertEqual("report-format", feed_type("report-formats")) self.assertEqual("report-format", feed_type("report_format")) self.assertEqual("report-format", feed_type("report_formats")) self.assertEqual("report-format", feed_type("REPORT_FORMAT")) self.assertEqual("report-format", feed_type("REPORT_FORMATS")) self.assertEqual("report-format", feed_type("REPORT-FORMAT")) self.assertEqual("report-format", feed_type("REPORT-FORMATS")) def test_port_list(self): self.assertEqual("port-list", feed_type("port-list")) self.assertEqual("port-list", feed_type("port-lists")) self.assertEqual("port-list", feed_type("port_list")) self.assertEqual("port-list", feed_type("port_lists")) self.assertEqual("port-list", feed_type("PORT_LIST")) self.assertEqual("port-list", feed_type("PORT_LISTS")) self.assertEqual("port-list", feed_type("PORT-LIST")) self.assertEqual("port-list", feed_type("PORT-LISTS")) def test_scan_config(self): self.assertEqual("scan-config", feed_type("scan-config")) self.assertEqual("scan-config", feed_type("scan-configs")) self.assertEqual("scan-config", feed_type("scan_config")) self.assertEqual("scan-config", feed_type("scan_configs")) self.assertEqual("scan-config", feed_type("SCAN_CONFIG")) self.assertEqual("scan-config", feed_type("SCAN_CONFIGS")) self.assertEqual("scan-config", feed_type("SCAN-CONFIG")) self.assertEqual("scan-config", feed_type("SCAN-CONFIGS")) def test_gvmd_data(self): self.assertEqual("gvmd-data", feed_type("gvmd-data")) self.assertEqual("gvmd-data", feed_type("gvmd_data")) self.assertEqual("gvmd-data", feed_type("GVMD-DATA")) self.assertEqual("gvmd-data", feed_type("GVMD_DATA")) class CliParserTestCase(unittest.TestCase): def test_output_group(self): parser = CliParser() with redirect_stderr(io.StringIO()) as f, self.assertRaises(SystemExit): parser.parse_arguments(["-vvv", "--quiet"]) self.assertIn( "argument --quiet: not allowed with argument --verbose/-v", f.getvalue(), ) def test_wait_group(self): parser = CliParser() with redirect_stderr(io.StringIO()) as f, self.assertRaises(SystemExit): parser.parse_arguments(["--no-wait", "--wait-interval", "20"]) self.assertIn( "argument --wait-interval: not allowed with argument --no-wait", f.getvalue(), ) def test_defaults(self): parser = CliParser() args = parser.parse_arguments([]) self.assertEqual(args.type, "all") self.assertEqual( args.destination_prefix, Path(DEFAULT_DESTINATION_PREFIX) ) self.assertEqual(args.feed_url, DEFAULT_RSYNC_URL) self.assertEqual( args.gvmd_data_destination, Path(DEFAULT_DESTINATION_PREFIX) / "gvm" / "data-objects" / "gvmd", ) self.assertEqual( args.gvmd_data_url, f"{DEFAULT_RSYNC_URL}/data-feed/{DEFAULT_FEED_RELEASE}/", ) self.assertEqual( args.notus_destination, Path(DEFAULT_DESTINATION_PREFIX) / "notus", ) self.assertEqual( args.notus_url, f"{DEFAULT_RSYNC_URL}/vulnerability-feed/{DEFAULT_FEED_RELEASE}/vt-data/notus/", ) self.assertEqual( args.nasl_destination, Path(DEFAULT_DESTINATION_PREFIX) / "openvas" / "plugins", ) self.assertEqual( args.nasl_url, f"{DEFAULT_RSYNC_URL}/vulnerability-feed/{DEFAULT_FEED_RELEASE}/vt-data/nasl/", ) self.assertEqual( args.scap_data_destination, Path(DEFAULT_DESTINATION_PREFIX) / "gvm" / "scap-data", ) self.assertEqual( args.scap_data_url, f"{DEFAULT_RSYNC_URL}/vulnerability-feed/{DEFAULT_FEED_RELEASE}/scap-data/", ) self.assertEqual( args.cert_data_destination, Path(DEFAULT_DESTINATION_PREFIX) / "gvm" / "cert-data", ) self.assertEqual( args.cert_data_url, f"{DEFAULT_RSYNC_URL}/vulnerability-feed/{DEFAULT_FEED_RELEASE}/cert-data/", ) self.assertEqual( args.report_formats_destination, Path(DEFAULT_DESTINATION_PREFIX) / "gvm" / "data-objects" / "gvmd" / "report-formats", ) self.assertEqual( args.report_formats_url, f"{DEFAULT_RSYNC_URL}/data-feed/{DEFAULT_FEED_RELEASE}/report-formats/", ) self.assertEqual( args.scan_configs_destination, Path(DEFAULT_DESTINATION_PREFIX) / "gvm" / "data-objects" / "gvmd" / "scan-configs", ) self.assertEqual( args.scan_configs_url, f"{DEFAULT_RSYNC_URL}/data-feed/{DEFAULT_FEED_RELEASE}/scan-configs/", ) self.assertEqual( args.port_lists_destination, Path(DEFAULT_DESTINATION_PREFIX) / "gvm" / "data-objects" / "gvmd" / "port-lists", ) self.assertEqual( args.port_lists_url, f"{DEFAULT_RSYNC_URL}/data-feed/{DEFAULT_FEED_RELEASE}/port-lists/", ) self.assertEqual( args.gvmd_lock_file, Path(DEFAULT_DESTINATION_PREFIX) / DEFAULT_GVMD_LOCK_FILE_PATH, ) self.assertEqual( args.openvas_lock_file, Path(DEFAULT_DESTINATION_PREFIX) / DEFAULT_OPENVAS_LOCK_FILE_PATH, ) self.assertEqual(args.wait_interval, DEFAULT_FLOCK_WAIT_INTERVAL) self.assertFalse(args.no_wait) self.assertEqual( args.compression_level, DEFAULT_RSYNC_COMPRESSION_LEVEL ) self.assertIsNone(args.private_directory) self.assertIsNone(args.verbose) self.assertFalse(args.fail_fast) self.assertIsNone(args.rsync_timeout) self.assertEqual( args.greenbone_enterprise_feed_key, Path(DEFAULT_ENTERPRISE_KEY_PATH), ) self.assertEqual(args.feed_release, DEFAULT_FEED_RELEASE) def test_help(self): parser = CliParser() with ( redirect_stdout(io.StringIO()) as f, self.assertRaises(SystemExit) as cm, ): parser.parse_arguments(["--help"]) self.assertEqual(cm.exception.code, 0) self.assertTrue(f.getvalue().startswith("usage: ")) def test_verbose(self): parser = CliParser() args = parser.parse_arguments(["-v"]) self.assertEqual(args.verbose, 1) args = parser.parse_arguments(["--verbose"]) self.assertEqual(args.verbose, 1) args = parser.parse_arguments(["-vv"]) self.assertEqual(args.verbose, 2) args = parser.parse_arguments(["-vvv"]) self.assertEqual(args.verbose, 3) args = parser.parse_arguments(["-vvvv"]) self.assertEqual(args.verbose, 4) def test_quiet(self): parser = CliParser() args = parser.parse_arguments(["--quiet"]) self.assertTrue(args.quiet) def test_selftest(self): parser = CliParser() args = parser.parse_arguments(["--selftest"]) self.assertTrue(args.selftest) @patch("greenbone.feed.sync.parser.Path") def test_use_default_config_files(self, path_mock): path_mock_instance = path_mock.return_value path_mock_instance.expanduser.return_value = path_mock_instance path_mock_instance.resolve.return_value = path_mock_instance path_mock_instance.exists.return_value = False parser = CliParser() parser.parse_arguments([]) path_mock.assert_any_call(DEFAULT_USER_CONFIG_FILE) path_mock.assert_any_call(DEFAULT_CONFIG_FILE) def test_private_directory(self): parser = CliParser() args = parser.parse_arguments(["--private-directory", "foobar"]) self.assertEqual(args.private_directory, Path("foobar")) def test_compression_level(self): parser = CliParser() args = parser.parse_arguments(["--compression-level", "0"]) self.assertEqual(args.compression_level, 0) args = parser.parse_arguments(["--compression-level", "1"]) self.assertEqual(args.compression_level, 1) args = parser.parse_arguments(["--compression-level", "2"]) self.assertEqual(args.compression_level, 2) args = parser.parse_arguments(["--compression-level", "3"]) self.assertEqual(args.compression_level, 3) args = parser.parse_arguments(["--compression-level", "4"]) self.assertEqual(args.compression_level, 4) args = parser.parse_arguments(["--compression-level", "5"]) self.assertEqual(args.compression_level, 5) args = parser.parse_arguments(["--compression-level", "6"]) self.assertEqual(args.compression_level, 6) args = parser.parse_arguments(["--compression-level", "7"]) self.assertEqual(args.compression_level, 7) args = parser.parse_arguments(["--compression-level", "8"]) self.assertEqual(args.compression_level, 8) args = parser.parse_arguments(["--compression-level", "9"]) self.assertEqual(args.compression_level, 9) with redirect_stderr(io.StringIO()) as f, self.assertRaises(SystemExit): parser.parse_arguments(["--compression-level", "10"]) self.assertIn( "error: argument --compression-level: invalid choice:", f.getvalue(), ) def test_gvmd_data_destination(self): parser = CliParser() args = parser.parse_arguments(["--gvmd-data-destination", "foo/bar"]) self.assertEqual(args.gvmd_data_destination, Path("foo/bar")) def test_gvmd_data_url(self): parser = CliParser() args = parser.parse_arguments( ["--gvmd-data-url", "rsync://foo.bar/gvmd-data"] ) self.assertEqual(args.gvmd_data_url, "rsync://foo.bar/gvmd-data") def test_notus_destination(self): parser = CliParser() args = parser.parse_arguments(["--notus-destination", "foo/bar"]) self.assertEqual(args.notus_destination, Path("foo/bar")) def test_notus_url(self): parser = CliParser() args = parser.parse_arguments(["--notus-url", "rsync://foo.bar/notus"]) self.assertEqual(args.notus_url, "rsync://foo.bar/notus") def test_nasl_destination(self): parser = CliParser() args = parser.parse_arguments(["--nasl-destination", "foo/bar"]) self.assertEqual(args.nasl_destination, Path("foo/bar")) def test_nasl_url(self): parser = CliParser() args = parser.parse_arguments(["--nasl-url", "rsync://foo.bar/nasl"]) self.assertEqual(args.nasl_url, "rsync://foo.bar/nasl") def test_scap_data_destination(self): parser = CliParser() args = parser.parse_arguments(["--scap-data-destination", "foo/bar"]) self.assertEqual(args.scap_data_destination, Path("foo/bar")) def test_scap_data_url(self): parser = CliParser() args = parser.parse_arguments( ["--scap-data-url", "rsync://foo.bar/scap-data"] ) self.assertEqual(args.scap_data_url, "rsync://foo.bar/scap-data") def test_cert_data_destination(self): parser = CliParser() args = parser.parse_arguments(["--cert-data-destination", "foo/bar"]) self.assertEqual(args.cert_data_destination, Path("foo/bar")) def test_cert_data_url(self): parser = CliParser() args = parser.parse_arguments( ["--cert-data-url", "rsync://foo.bar/cert-data"] ) self.assertEqual(args.cert_data_url, "rsync://foo.bar/cert-data") def test_report_formats_destination(self): parser = CliParser() args = parser.parse_arguments( ["--report-formats-destination", "foo/bar"] ) self.assertEqual(args.report_formats_destination, Path("foo/bar")) def test_report_formats_url(self): parser = CliParser() args = parser.parse_arguments( ["--report-formats-url", "rsync://foo.bar/report-formats"] ) self.assertEqual( args.report_formats_url, "rsync://foo.bar/report-formats" ) def test_scan_configs_destination(self): parser = CliParser() args = parser.parse_arguments(["--scan-configs-destination", "foo/bar"]) self.assertEqual(args.scan_configs_destination, Path("foo/bar")) def test_scan_configs_url(self): parser = CliParser() args = parser.parse_arguments( ["--scan-configs-url", "rsync://foo.bar/scan-configs"] ) self.assertEqual(args.scan_configs_url, "rsync://foo.bar/scan-configs") def test_port_lists_destination(self): parser = CliParser() args = parser.parse_arguments(["--port-lists-destination", "foo/bar"]) self.assertEqual(args.port_lists_destination, Path("foo/bar")) def test_port_lists_url(self): parser = CliParser() args = parser.parse_arguments( ["--port-lists-url", "rsync://foo.bar/port-lists"] ) self.assertEqual(args.port_lists_url, "rsync://foo.bar/port-lists") def test_gvmd_lock_file(self): parser = CliParser() args = parser.parse_arguments(["--gvmd-lock-file", "/run/gvmd.lock"]) self.assertEqual(args.gvmd_lock_file, Path("/run/gvmd.lock")) def test_openvas_lock_file(self): parser = CliParser() args = parser.parse_arguments( ["--openvas-lock-file", "/run/openvas.lock"] ) self.assertEqual(args.openvas_lock_file, Path("/run/openvas.lock")) def test_fail_fast(self): parser = CliParser() args = parser.parse_arguments(["--fail-fast"]) self.assertTrue(args.fail_fast) args = parser.parse_arguments(["--failfast"]) self.assertTrue(args.fail_fast) def test_no_wait(self): parser = CliParser() args = parser.parse_arguments(["--no-wait"]) self.assertTrue(args.no_wait) def test_wait_interval(self): parser = CliParser() args = parser.parse_arguments(["--wait-interval", "100"]) self.assertEqual(args.wait_interval, 100) def test_rsync_timeout(self): parser = CliParser() args = parser.parse_arguments(["--rsync-timeout", "120"]) self.assertEqual(args.rsync_timeout, 120) def test_greenbone_enterprise_feed_key(self): parser = CliParser() args = parser.parse_arguments( ["--greenbone-enterprise-feed-key", "/tmp/some.key"] ) self.assertEqual( args.greenbone_enterprise_feed_key, Path("/tmp/some.key") ) def test_other(self): parser = CliParser() with ( redirect_stderr(io.StringIO()) as f, self.assertRaises(SystemExit) as cm, ): parser.parse_arguments(["--foo-bar", "10"]) self.assertIn("error: unrecognized arguments: --foo-bar", f.getvalue()) self.assertEqual(cm.exception.code, 2) @patch("greenbone.feed.sync.parser.Path") def test_config(self, path_mock: MagicMock): content = """[greenbone-feed-sync] verbose = 3 feed-url = "rsync://foo.bar" destination-prefix = "/usr/lib" """ path_mock_instance = path_mock.return_value path_mock_instance.absolute.return_value = "/foo/bar/foo.toml" path_mock_instance.expanduser.return_value = path_mock_instance path_mock_instance.resolve.return_value = path_mock_instance path_mock_instance.exists.return_value = True path_mock_instance.read_text.return_value = content parser = CliParser() args = parser.parse_arguments(["--config", "foo.toml"]) self.assertEqual(args.verbose, 3) self.assertEqual(args.feed_url, "rsync://foo.bar") self.assertEqual(args.destination_prefix, Path("/usr/lib")) @patch("greenbone.feed.sync.parser.Path") def test_load_from_default_config(self, path_mock: MagicMock): content = """[greenbone-feed-sync] verbose = 3 feed-url = "rsync://foo.bar" destination-prefix = "/usr/lib" """ path_mock_instance = path_mock.return_value path_mock_instance.absolute.return_value = ( "/foo/bar/.config/greenbone-feed-sync.toml" ) path_mock_instance.expanduser.return_value = path_mock_instance path_mock_instance.resolve.return_value = path_mock_instance path_mock_instance.exists.return_value = True path_mock_instance.read_text.return_value = content parser = CliParser() args = parser.parse_arguments([]) self.assertEqual(args.verbose, 3) self.assertEqual(args.feed_url, "rsync://foo.bar") self.assertEqual(args.destination_prefix, Path("/usr/lib")) @patch.dict( "os.environ", { "GREENBONE_FEED_SYNC_DESTINATION_PREFIX": "/usr/lib", "GREENBONE_FEED_SYNC_URL": "rsync://foo.bar", "GREENBONE_FEED_SYNC_VERBOSE": "3", }, ) def test_from_environment(self): parser = CliParser() args = parser.parse_arguments([]) self.assertEqual(args.verbose, 3) self.assertEqual(args.feed_url, "rsync://foo.bar") self.assertEqual(args.destination_prefix, Path("/usr/lib")) @patch.dict( "os.environ", { "GREENBONE_FEED_SYNC_DESTINATION_PREFIX": "/usr/lib", "GREENBONE_FEED_SYNC_URL": "rsync://foo.bar", "GREENBONE_FEED_SYNC_VERBOSE": "3", }, ) @patch("greenbone.feed.sync.parser.Path") def test_environment_takes_precedence(self, path_mock): content = """[greenbone-feed-sync] verbose = 9 feed-url = "rsync://lorem.ipsum" destination-prefix = "/opt/lib" wait-interval = 100 """ path_mock_instance = path_mock.return_value path_mock_instance.absolute.return_value = ( "/foo/bar/.config/greenbone-feed-sync.toml" ) path_mock_instance.expanduser.return_value = path_mock_instance path_mock_instance.resolve.return_value = path_mock_instance path_mock_instance.exists.return_value = True path_mock_instance.read_text.return_value = content parser = CliParser() args = parser.parse_arguments([]) self.assertEqual(args.verbose, 3) self.assertEqual(args.feed_url, "rsync://foo.bar") self.assertEqual(args.destination_prefix, Path("/usr/lib")) self.assertEqual(args.wait_interval, 100) @patch.dict( "os.environ", { "GREENBONE_FEED_SYNC_DESTINATION_PREFIX": "/usr/lib", "GREENBONE_FEED_SYNC_URL": "rsync://foo.bar", "GREENBONE_FEED_SYNC_VERBOSE": "3", }, ) @patch("greenbone.feed.sync.parser.Path") def test_argument_takes_precedence(self, path_mock): content = """[greenbone-feed-sync] verbose = 9 feed-url = "rsync://lorem.ipsum" destination-prefix = "/opt/lib" wait-interval = 100 """ path_mock_instance = path_mock.return_value path_mock_instance.absolute.return_value = ( "/foo/bar/.config/greenbone-feed-sync.toml" ) path_mock_instance.expanduser.return_value = path_mock_instance path_mock_instance.resolve.return_value = path_mock_instance path_mock_instance.exists.return_value = True path_mock_instance.read_text.return_value = content parser = CliParser() args = parser.parse_arguments( [ "--wait-interval", "90", ] ) self.assertEqual(args.verbose, 3) self.assertEqual(args.feed_url, "rsync://foo.bar") self.assertEqual(args.destination_prefix, Path("/usr/lib")) self.assertEqual(args.wait_interval, 90) def test_config_file_not_exists(self): parser = CliParser() with self.assertRaisesRegex( ConfigFileError, "Config file foo.bar does not exist." ): parser.parse_arguments(["--config", "foo.bar"]) def test_type(self): parser = CliParser() args = parser.parse_arguments(["--type", "nvt"]) self.assertEqual(args.type, "nvt") args = parser.parse_arguments(["--type", "nvts"]) self.assertEqual(args.type, "nvt") args = parser.parse_arguments(["--type", "NVT"]) self.assertEqual(args.type, "nvt") args = parser.parse_arguments(["--type", "NVTS"]) self.assertEqual(args.type, "nvt") args = parser.parse_arguments(["--type", "notus"]) self.assertEqual(args.type, "notus") args = parser.parse_arguments(["--type", "NOTUS"]) self.assertEqual(args.type, "notus") args = parser.parse_arguments(["--type", "NoTuS"]) self.assertEqual(args.type, "notus") args = parser.parse_arguments(["--type", "nasl"]) self.assertEqual(args.type, "nasl") args = parser.parse_arguments(["--type", "NASL"]) self.assertEqual(args.type, "nasl") args = parser.parse_arguments(["--type", "NaSl"]) self.assertEqual(args.type, "nasl") args = parser.parse_arguments(["--type", "scap"]) self.assertEqual(args.type, "scap") args = parser.parse_arguments(["--type", "SCAP"]) self.assertEqual(args.type, "scap") args = parser.parse_arguments(["--type", "ScAp"]) self.assertEqual(args.type, "scap") args = parser.parse_arguments(["--type", "cert"]) self.assertEqual(args.type, "cert") args = parser.parse_arguments(["--type", "CERT"]) self.assertEqual(args.type, "cert") args = parser.parse_arguments(["--type", "CeRt"]) self.assertEqual(args.type, "cert") args = parser.parse_arguments(["--type", "all"]) self.assertEqual(args.type, "all") args = parser.parse_arguments(["--type", "ALL"]) self.assertEqual(args.type, "all") args = parser.parse_arguments(["--type", "AlL"]) self.assertEqual(args.type, "all") args = parser.parse_arguments(["--type", "report-format"]) self.assertEqual(args.type, "report-format") args = parser.parse_arguments(["--type", "REPORT-FORMAT"]) self.assertEqual(args.type, "report-format") args = parser.parse_arguments(["--type", "report_format"]) self.assertEqual(args.type, "report-format") args = parser.parse_arguments(["--type", "REPORT_FORMAT"]) self.assertEqual(args.type, "report-format") args = parser.parse_arguments(["--type", "Report-Format"]) self.assertEqual(args.type, "report-format") args = parser.parse_arguments(["--type", "report-formats"]) self.assertEqual(args.type, "report-format") args = parser.parse_arguments(["--type", "REPORT_FORMATS"]) self.assertEqual(args.type, "report-format") args = parser.parse_arguments(["--type", "scan-config"]) self.assertEqual(args.type, "scan-config") args = parser.parse_arguments(["--type", "SCAN-CONFIG"]) self.assertEqual(args.type, "scan-config") args = parser.parse_arguments(["--type", "scan_config"]) self.assertEqual(args.type, "scan-config") args = parser.parse_arguments(["--type", "SCAN_config"]) self.assertEqual(args.type, "scan-config") args = parser.parse_arguments(["--type", "Scan-Config"]) self.assertEqual(args.type, "scan-config") args = parser.parse_arguments(["--type", "scan-configs"]) self.assertEqual(args.type, "scan-config") args = parser.parse_arguments(["--type", "SCAN_CONFIGS"]) self.assertEqual(args.type, "scan-config") args = parser.parse_arguments(["--type", "port-list"]) self.assertEqual(args.type, "port-list") args = parser.parse_arguments(["--type", "PORT-LIST"]) self.assertEqual(args.type, "port-list") args = parser.parse_arguments(["--type", "port_list"]) self.assertEqual(args.type, "port-list") args = parser.parse_arguments(["--type", "PORT_LIST"]) self.assertEqual(args.type, "port-list") args = parser.parse_arguments(["--type", "port-list"]) self.assertEqual(args.type, "port-list") args = parser.parse_arguments(["--type", "port-lists"]) self.assertEqual(args.type, "port-list") args = parser.parse_arguments(["--type", "PORT_LISTS"]) self.assertEqual(args.type, "port-list") args = parser.parse_arguments(["--type", "gvmd-data"]) self.assertEqual(args.type, "gvmd-data") args = parser.parse_arguments(["--type", "GVMD-DATA"]) self.assertEqual(args.type, "gvmd-data") args = parser.parse_arguments(["--type", "gvmd_data"]) self.assertEqual(args.type, "gvmd-data") args = parser.parse_arguments(["--type", "GVMD_DATA"]) self.assertEqual(args.type, "gvmd-data") args = parser.parse_arguments(["--type", "gvmd-data"]) self.assertEqual(args.type, "gvmd-data") def test_group(self): parser = CliParser() args = parser.parse_arguments(["--group", "some_group"]) self.assertEqual(args.group, "some_group") args = parser.parse_arguments(["--group", "123"]) self.assertEqual(args.group, 123) def test_user(self): parser = CliParser() args = parser.parse_arguments(["--user", "some_user"]) self.assertEqual(args.user, "some_user") args = parser.parse_arguments(["--user", "123"]) self.assertEqual(args.user, 123) def test_feed_url_from_enterprise_feed_key(self): parser = CliParser() content = """a_user@some.feed.server:/feed/ Lorem ipsum dolor sit amet, consetetur sadipscing elitr, sed diam nonumy eirmod tempor """ with temp_file(content=content, name="enterprise.key") as f: args = parser.parse_arguments( ["--greenbone-enterprise-feed-key", str(f)] ) self.assertEqual( args.feed_url, "ssh://a_user@some.feed.server/enterprise" ) self.assertEqual( args.gvmd_data_url, f"ssh://a_user@some.feed.server/enterprise/data-feed/{DEFAULT_FEED_RELEASE}/", ) self.assertEqual( args.port_lists_url, f"ssh://a_user@some.feed.server/enterprise/data-feed/{DEFAULT_FEED_RELEASE}/port-lists/", ) self.assertEqual( args.report_formats_url, f"ssh://a_user@some.feed.server/enterprise/data-feed/{DEFAULT_FEED_RELEASE}/report-formats/", ) self.assertEqual( args.scan_configs_url, f"ssh://a_user@some.feed.server/enterprise/data-feed/{DEFAULT_FEED_RELEASE}/scan-configs/", ) self.assertEqual( args.notus_url, f"ssh://a_user@some.feed.server/enterprise/vulnerability-feed/{DEFAULT_FEED_RELEASE}/vt-data/notus/", ) self.assertEqual( args.nasl_url, f"ssh://a_user@some.feed.server/enterprise/vulnerability-feed/{DEFAULT_FEED_RELEASE}/vt-data/nasl/", ) self.assertEqual( args.scap_data_url, f"ssh://a_user@some.feed.server/enterprise/vulnerability-feed/{DEFAULT_FEED_RELEASE}/scap-data/", ) self.assertEqual( args.cert_data_url, f"ssh://a_user@some.feed.server/enterprise/vulnerability-feed/{DEFAULT_FEED_RELEASE}/cert-data/", ) def test_ignore_non_existing_enterprise_feed_key(self): parser = CliParser() args = parser.parse_arguments( ["--greenbone-enterprise-feed-key", "/tmp/some.key"] ) self.assertEqual( args.feed_url, "rsync://feed.community.greenbone.net/community" ) self.assertEqual( args.gvmd_data_url, f"rsync://feed.community.greenbone.net/community/data-feed/{DEFAULT_FEED_RELEASE}/", ) self.assertEqual( args.port_lists_url, f"rsync://feed.community.greenbone.net/community/data-feed/{DEFAULT_FEED_RELEASE}/port-lists/", ) self.assertEqual( args.report_formats_url, f"rsync://feed.community.greenbone.net/community/data-feed/{DEFAULT_FEED_RELEASE}/report-formats/", ) self.assertEqual( args.scan_configs_url, f"rsync://feed.community.greenbone.net/community/data-feed/{DEFAULT_FEED_RELEASE}/scan-configs/", ) self.assertEqual( args.notus_url, f"rsync://feed.community.greenbone.net/community/vulnerability-feed/{DEFAULT_FEED_RELEASE}/vt-data/notus/", ) self.assertEqual( args.nasl_url, f"rsync://feed.community.greenbone.net/community/vulnerability-feed/{DEFAULT_FEED_RELEASE}/vt-data/nasl/", ) self.assertEqual( args.scap_data_url, f"rsync://feed.community.greenbone.net/community/vulnerability-feed/{DEFAULT_FEED_RELEASE}/scap-data/", ) self.assertEqual( args.cert_data_url, f"rsync://feed.community.greenbone.net/community/vulnerability-feed/{DEFAULT_FEED_RELEASE}/cert-data/", ) @patch.object(sys, "argv", ["greenbone-nvt-sync"]) def test_greenbone_nvt_sync(self): parser = CliParser() args = parser.parse_arguments([]) self.assertEqual(args.type, "nvt") @patch.object(sys, "argv", ["greenbone-scapdata-sync"]) def test_greenbone_scap_data_sync(self): parser = CliParser() args = parser.parse_arguments([]) self.assertEqual(args.type, "scap") @patch.object(sys, "argv", ["greenbone-certdata-sync"]) def test_greenbone_cert_data_sync(self): parser = CliParser() args = parser.parse_arguments([]) self.assertEqual(args.type, "cert") def test_feed_release(self): parser = CliParser() feed_release = "1.2.3" args = parser.parse_arguments(["--feed-release", feed_release]) self.assertTrue(args.feed_release, "1.2.3") self.assertEqual( args.feed_url, "rsync://feed.community.greenbone.net/community" ) self.assertEqual( args.gvmd_data_url, f"rsync://feed.community.greenbone.net/community/data-feed/{feed_release}/", ) self.assertEqual( args.port_lists_url, f"rsync://feed.community.greenbone.net/community/data-feed/{feed_release}/port-lists/", ) self.assertEqual( args.report_formats_url, f"rsync://feed.community.greenbone.net/community/data-feed/{feed_release}/report-formats/", ) self.assertEqual( args.scan_configs_url, f"rsync://feed.community.greenbone.net/community/data-feed/{feed_release}/scan-configs/", ) self.assertEqual( args.notus_url, f"rsync://feed.community.greenbone.net/community/vulnerability-feed/{feed_release}/vt-data/notus/", ) self.assertEqual( args.nasl_url, f"rsync://feed.community.greenbone.net/community/vulnerability-feed/{feed_release}/vt-data/nasl/", ) self.assertEqual( args.scap_data_url, f"rsync://feed.community.greenbone.net/community/vulnerability-feed/{feed_release}/scap-data/", ) self.assertEqual( args.cert_data_url, f"rsync://feed.community.greenbone.net/community/vulnerability-feed/{feed_release}/cert-data/", ) def test_feed_version(self): parser = CliParser() feed_release = "1.2.3" args = parser.parse_arguments(["--feed-version", feed_release]) self.assertTrue(args.feed_version, "1.2.3") self.assertTrue(args.feed_release, "1.2.3") self.assertEqual( args.feed_url, "rsync://feed.community.greenbone.net/community" ) self.assertEqual( args.gvmd_data_url, f"rsync://feed.community.greenbone.net/community/data-feed/{feed_release}/", ) self.assertEqual( args.port_lists_url, f"rsync://feed.community.greenbone.net/community/data-feed/{feed_release}/port-lists/", ) self.assertEqual( args.report_formats_url, f"rsync://feed.community.greenbone.net/community/data-feed/{feed_release}/report-formats/", ) self.assertEqual( args.scan_configs_url, f"rsync://feed.community.greenbone.net/community/data-feed/{feed_release}/scan-configs/", ) self.assertEqual( args.notus_url, f"rsync://feed.community.greenbone.net/community/vulnerability-feed/{feed_release}/vt-data/notus/", ) self.assertEqual( args.nasl_url, f"rsync://feed.community.greenbone.net/community/vulnerability-feed/{feed_release}/vt-data/nasl/", ) self.assertEqual( args.scap_data_url, f"rsync://feed.community.greenbone.net/community/vulnerability-feed/{feed_release}/scap-data/", ) self.assertEqual( args.cert_data_url, f"rsync://feed.community.greenbone.net/community/vulnerability-feed/{feed_release}/cert-data/", ) def test_feed_release_overrules_version(self): parser = CliParser() feed_release = "1.2.3" args = parser.parse_arguments( ["--feed-version", "2.3.4", "--feed-release", feed_release] ) self.assertTrue(args.feed_version, "2.3.4") self.assertTrue(args.feed_release, "1.2.3") self.assertEqual( args.feed_url, "rsync://feed.community.greenbone.net/community" ) self.assertEqual( args.gvmd_data_url, f"rsync://feed.community.greenbone.net/community/data-feed/{feed_release}/", ) self.assertEqual( args.port_lists_url, f"rsync://feed.community.greenbone.net/community/data-feed/{feed_release}/port-lists/", ) self.assertEqual( args.report_formats_url, f"rsync://feed.community.greenbone.net/community/data-feed/{feed_release}/report-formats/", ) self.assertEqual( args.scan_configs_url, f"rsync://feed.community.greenbone.net/community/data-feed/{feed_release}/scan-configs/", ) self.assertEqual( args.notus_url, f"rsync://feed.community.greenbone.net/community/vulnerability-feed/{feed_release}/vt-data/notus/", ) self.assertEqual( args.nasl_url, f"rsync://feed.community.greenbone.net/community/vulnerability-feed/{feed_release}/vt-data/nasl/", ) self.assertEqual( args.scap_data_url, f"rsync://feed.community.greenbone.net/community/vulnerability-feed/{feed_release}/scap-data/", ) self.assertEqual( args.cert_data_url, f"rsync://feed.community.greenbone.net/community/vulnerability-feed/{feed_release}/cert-data/", ) def test_destination_prefix(self): parser = CliParser() destination_prefix = "/tmp/foo/bar" args = parser.parse_arguments( ["--destination-prefix", destination_prefix] ) self.assertEqual(args.type, "all") self.assertEqual(args.destination_prefix, Path(destination_prefix)) self.assertEqual( args.gvmd_data_destination, Path(destination_prefix) / "gvm" / "data-objects" / "gvmd", ) self.assertEqual( args.notus_destination, Path(destination_prefix) / "notus", ) self.assertEqual( args.nasl_destination, Path(destination_prefix) / "openvas" / "plugins", ) self.assertEqual( args.scap_data_destination, Path(destination_prefix) / "gvm" / "scap-data", ) self.assertEqual( args.cert_data_destination, Path(destination_prefix) / "gvm" / "cert-data", ) self.assertEqual( args.report_formats_destination, Path(destination_prefix) / "gvm" / "data-objects" / "gvmd" / "report-formats", ) self.assertEqual( args.scan_configs_destination, Path(destination_prefix) / "gvm" / "data-objects" / "gvmd" / "scan-configs", ) self.assertEqual( args.port_lists_destination, Path(destination_prefix) / "gvm" / "data-objects" / "gvmd" / "port-lists", ) self.assertEqual( args.gvmd_lock_file, Path(destination_prefix) / DEFAULT_GVMD_LOCK_FILE_PATH, ) self.assertEqual( args.openvas_lock_file, Path(destination_prefix) / DEFAULT_OPENVAS_LOCK_FILE_PATH, )