def test_defaults(self):
args = main.parse_arguments(["a", "b", "c"])
self.assertEqual(args.name, ["a", "b", "c"])
self.assertEqual(args.excludes, [])
self.assertEqual(args.path, [])
self.assertEqual(args.node_type, main.NodeType.MODULE)
self.assertEqual(args.output_format, main.OutputFormat.HTML)
self.assertEqual(args.output_file, None)
def test_output_format(self):
args = main.parse_arguments(["--format", "html", "a"])
self.assertEqual(args.output_format, main.OutputFormat.HTML)
args = main.parse_arguments(["-f", "html", "a"])
self.assertEqual(args.output_format, main.OutputFormat.HTML)
args = main.parse_arguments(["--format", "dot", "a"])
self.assertEqual(args.output_format, main.OutputFormat.GRAPHVIZ)
args = main.parse_arguments(["-f", "dot", "a"])
self.assertEqual(args.output_format, main.OutputFormat.GRAPHVIZ)
with captured_output() as (stdout, stderr):
self.assertRaises(SystemExit, main.parse_arguments,
["-f", "noformat", "a"])
self.assertIn(
"error: argument -f/--format: invalid choice: 'noformat' (choose from 'html', 'dot')", # noqa: B950
stderr.getvalue(),
)
def test_node_type(self):
args = main.parse_arguments(["-s", "a"])
self.assertEqual(args.node_type, main.NodeType.SCRIPT)
args = main.parse_arguments(["--script", "a"])
self.assertEqual(args.node_type, main.NodeType.SCRIPT)
args = main.parse_arguments(["-m", "a"])
self.assertEqual(args.node_type, main.NodeType.MODULE)
args = main.parse_arguments(["--module", "a"])
self.assertEqual(args.node_type, main.NodeType.MODULE)
args = main.parse_arguments(["-d", "a"])
self.assertEqual(args.node_type, main.NodeType.DISTRIBUTION)
args = main.parse_arguments(["--distribution", "a"])
self.assertEqual(args.node_type, main.NodeType.DISTRIBUTION)
def test_output_path(self):
args = main.parse_arguments(["--output", "filename", "a"])
self.assertEqual(args.output_file, "filename")
args = main.parse_arguments(["-o", "filename", "a"])
self.assertEqual(args.output_file, "filename")
def test_excludes(self):
args = main.parse_arguments(["-x", "name1", "--exclude", "name2", "a"])
self.assertEqual(args.excludes, ["name1", "name2"])
def test_path(self):
args = main.parse_arguments(["-p", "path1", "--path", "path2", "a"])
self.assertEqual(args.path, ["path1", "path2"])