def test_print_bash_completion_mismatch(self, mock_CONF): self.useFixture(fixtures.MonkeyPatch('sys.stdout', StringIO())) mock_CONF.category.query_category = 'bar' categories = {'foo': mock.sentinel.foo} cmd_common.print_bash_completion(categories) self.assertEqual('', sys.stdout.getvalue())
def test_print_bash_completion_no_query_category(self, mock_CONF, mock_print): mock_CONF.category.query_category = None categories = {'foo': mock.sentinel.foo, 'bar': mock.sentinel.bar} cmd_common.print_bash_completion(categories) mock_print.assert_called_once_with(' '.join(categories.keys()))
def test_print_bash_completion_no_query_category(self, mock_CONF): self.useFixture(fixtures.MonkeyPatch('sys.stdout', StringIO())) mock_CONF.category.query_category = None categories = {'foo': mock.sentinel.foo, 'bar': mock.sentinel.bar} cmd_common.print_bash_completion(categories) self.assertEqual(' '.join(categories.keys()) + '\n', sys.stdout.getvalue())
def test_print_bash_completion(self, mock_CONF, mock_print, mock_method_of): mock_CONF.category.query_category = 'foo' actions = [('f1', mock.sentinel.f1), ('f2', mock.sentinel.f2)] mock_method_of.return_value = actions mock_fn = mock.Mock() categories = {'foo': mock_fn} cmd_common.print_bash_completion(categories) mock_fn.assert_called_once_with() mock_method_of.assert_called_once_with(mock_fn.return_value) mock_print.assert_called_once_with(' '.join([k for k, v in actions]))
def test_print_bash_completion(self, mock_CONF, mock_method_of): self.useFixture(fixtures.MonkeyPatch('sys.stdout', StringIO())) mock_CONF.category.query_category = 'foo' actions = [('f1', mock.sentinel.f1), ('f2', mock.sentinel.f2)] mock_method_of.return_value = actions mock_fn = mock.Mock() categories = {'foo': mock_fn} cmd_common.print_bash_completion(categories) mock_fn.assert_called_once_with() mock_method_of.assert_called_once_with(mock_fn.return_value) self.assertEqual(' '.join([k for k, v in actions]) + '\n', sys.stdout.getvalue())
def main(): """Parse options and call the appropriate class/method.""" CONF.register_cli_opts(cli_opts) CONF.register_cli_opt(category_opt) config.parse_args(sys.argv) if CONF.category.name == "version": print(version.version_string_with_package()) return 0 if CONF.category.name == "bash-completion": cmd_common.print_bash_completion(CATEGORIES) return 0 try: fn, fn_args, fn_kwargs = cmd_common.get_action_fn() ret = fn(*fn_args, **fn_kwargs) return ret except Exception as ex: print(_("error: %s") % ex) return 1
def main(): """Parse options and call the appropriate class/method.""" CONF.register_cli_opt(category_opt) config.parse_args(sys.argv) if CONF.category.name == "version": print(version.version_string_with_package()) return 0 if CONF.category.name == "bash-completion": cmd_common.print_bash_completion(CATEGORIES) return 0 try: fn, fn_args, fn_kwargs = cmd_common.get_action_fn() ret = fn(*fn_args, **fn_kwargs) return ret except Exception: print(_('Error:\n%s') % traceback.format_exc()) # This is 255 so it's not confused with the upgrade check exit codes. return 255
def test_print_bash_completion_mismatch(self, mock_CONF, mock_print): mock_CONF.category.query_category = 'bar' categories = {'foo': mock.sentinel.foo} cmd_common.print_bash_completion(categories) self.assertFalse(mock_print.called)