def test_get_command_must_skip_on_exception_loading_module(self, importlib_mock):
cmd = BaseCommand(cmd_packages=self.packages)
importlib_mock.import_module = Mock()
importlib_mock.import_module.side_effect = ImportError()
result = cmd.get_command(None, "cmd1")
self.assertEqual(result, None, "must not return a command")
def test_get_command_must_skip_on_absence_of_cli_method(self, importlib_mock):
cmd = BaseCommand(cmd_packages=self.packages)
importlib_mock.import_module = Mock()
importlib_mock.import_module.return_value = {} # Returned Module does *not* have 'cli' property
result = cmd.get_command(None, "cmd1")
self.assertEqual(result, None, "must not return a command")
def test_get_command_must_skip_on_absence_of_cli_method(self, importlib_mock):
cmd = BaseCommand(cmd_packages=self.packages)
importlib_mock.import_module = Mock()
importlib_mock.import_module.return_value = {} # Returned Module does *not* have 'cli' property
result = cmd.get_command(None, "cmd1")
self.assertEquals(result, None, "must not return a command")
def test_get_command_must_skip_on_exception_loading_module(self, importlib_mock):
cmd = BaseCommand(cmd_packages=self.packages)
importlib_mock.import_module = Mock()
importlib_mock.import_module.side_effect = ImportError()
result = cmd.get_command(None, "cmd1")
self.assertEquals(result, None, "must not return a command")
def test_set_commands_must_extract_command_name(self):
expected = {
"cmd1": "a.b.cmd1",
"cmd2": "foo.cmd2",
"cmd3": "cmd3"
}
result = BaseCommand._set_commands(self.packages)
self.assertEquals(result, expected)
def test_get_command_must_return_command_module(self, importlib_mock):
module_mock = Mock()
module_mock.cli = Mock()
importlib_mock.import_module = Mock()
importlib_mock.import_module.return_value = module_mock
cmd = BaseCommand(cmd_packages=self.packages)
result = cmd.get_command(None, "cmd1")
self.assertEqual(result, module_mock.cli)
result = cmd.get_command(None, "cmd2")
self.assertEqual(result, module_mock.cli)
result = cmd.get_command(None, "cmd3")
self.assertEqual(result, module_mock.cli)
# Library to import the modules must be called three times
importlib_mock.import_module.assert_has_calls([call("a.b.cmd1"), call("foo.cmd2"), call("cmd3")])
def test_get_command_must_return_command_module(self, importlib_mock):
module_mock = Mock()
module_mock.cli = Mock()
importlib_mock.import_module = Mock()
importlib_mock.import_module.return_value = module_mock
cmd = BaseCommand(cmd_packages=self.packages)
result = cmd.get_command(None, "cmd1")
self.assertEquals(result, module_mock.cli)
result = cmd.get_command(None, "cmd2")
self.assertEquals(result, module_mock.cli)
result = cmd.get_command(None, "cmd3")
self.assertEquals(result, module_mock.cli)
# Library to import the modules must be called three times
importlib_mock.import_module.assert_has_calls([
call("a.b.cmd1"),
call("foo.cmd2"),
call("cmd3")
])
def test_get_command_must_skip_unknown_commands(self):
cmd = BaseCommand(cmd_packages=self.packages)
result = cmd.get_command(None, "unknown_command")
self.assertEqual(result, None, "must not return a command")
def test_list_commands_must_return_commands_name(self):
expected = ["cmd1", "cmd2", "cmd3"].sort()
cmd = BaseCommand(cmd_packages=self.packages)
result = cmd.list_commands(ctx=None)
self.assertEqual(result.sort(), expected)
def test_set_commands_must_extract_command_name(self):
expected = {"cmd1": "a.b.cmd1", "cmd2": "foo.cmd2", "cmd3": "cmd3"}
result = BaseCommand._set_commands(self.packages)
self.assertEqual(result, expected)
def test_must_inherit(self):
cmd = BaseCommand()
self.assertTrue(isinstance(cmd, click.MultiCommand))
def test_get_command_must_skip_unknown_commands(self):
cmd = BaseCommand(cmd_packages=self.packages)
result = cmd.get_command(None, "unknown_command")
self.assertEquals(result, None, "must not return a command")
def test_list_commands_must_return_commands_name(self):
expected = ["cmd1", "cmd2", "cmd3"].sort()
cmd = BaseCommand(cmd_packages=self.packages)
result = cmd.list_commands(ctx=None)
self.assertEquals(result.sort(), expected)
