def test_debug_command_line(self): @click.command() def log(): pass from cloudify_agent.shell.main import main main.add_command(log, 'log') self._run('cfy-agent --debug log') # assert all loggers are now at debug level from cloudify_agent.api.utils import logger self.assertEqual(logger.level, logging.DEBUG)
def test_api_errors_conversion(self): @click.command() @handle_failures def _raise_api_error(): raise exceptions.DaemonError() from cloudify_agent.shell.main import main main.add_command(_raise_api_error, 'raise-error') try: self._run('cfy-agent raise-error', raise_system_exit=True) self.fail('Expected failure of command execution') except SystemExit as e: self.assertEqual(e.code, 201)
def test_api_exceptions_conversion(): @click.command() @handle_failures def _raise_api_exception(): raise exceptions.DaemonException() from cloudify_agent.shell.main import main main.add_command(_raise_api_exception, 'raise-error') try: run_agent_command('cfy-agent raise-error', raise_system_exit=True) pytest.fail('Expected failure of command execution') except SystemExit as e: assert e.code == 101