def test_main_loads_inventory(self):
self.parser.parse_args = True
self.parser_args.data_path = "path"
main()
self.assertEqual(
self.inventory.assert_called_with("path"),
None,
)
def test_unused_subcommand(self, reportMock):
self.parser_args.subcommand = "unused"
main()
self.assertTrue(self.inventory.return_value.load.called)
self.assertEqual(
reportMock.assert_called_with(self.inventory.return_value),
None,
)
self.assertEqual(
reportMock.return_value.output.assert_called_with(),
None,
)
def test_active_subcommand(self, reportMock):
self.parser_args.resource_type = "ec2"
self.parser_args.subcommand = "active"
main()
self.assertTrue(self.inventory.return_value.load.called)
self.assertEqual(
reportMock.assert_called_with(self.inventory.return_value),
None,
)
self.assertEqual(
reportMock.return_value.output.assert_called_with("ec2"),
None,
)
def test_search_subcommand(self, reportMock):
self.parser_args.subcommand = "search"
self.parser_args.search_string = "inst"
main()
self.assertTrue(self.inventory.return_value.load.called)
self.assertEqual(
reportMock.assert_called_with(self.inventory.return_value),
None,
)
self.assertEqual(
reportMock.return_value.output.assert_called_with("inst"),
None,
)
def test_monitor_subcommand(self, reportMock):
self.parser_args.subcommand = "monitor"
self.parser_args.monitor_status = "true"
main()
self.assertTrue(self.inventory.return_value.load.called)
self.assertEqual(
reportMock.assert_called_with(self.inventory.return_value),
None,
)
self.assertEqual(
reportMock.return_value.output.assert_called_with("true"),
None,
)
def test_export_subcommand(self, reportMock):
self.parser_args.subcommand = "export"
self.parser_args.export_path = "file.ods"
main()
self.assertTrue(self.inventory.return_value.load.called)
self.assertEqual(
reportMock.assert_called_with(self.inventory.return_value),
None,
)
self.assertEqual(
reportMock.return_value.output.assert_called_with("file.ods"),
None,
)
def test_generate_subcommand_with_resource_type(self):
self.parser_args.subcommand = "generate"
self.parser_args.resource_type = "ec2"
main()
self.inventory.return_value.generate.assert_called_with("ec2")
def test_generate_subcommand(self):
self.parser_args.subcommand = "generate"
main()
self.assertTrue(self.inventory.return_value.generate.called)
def test_main_loads_logger(self, loggerMock):
self.parser.parse_args = True
main()
self.assertTrue(loggerMock.called)
def test_main_loads_parser(self):
self.parser.parse_args = True
main()
self.assertTrue(self.parser.called)
