本文整理汇总了Python中st2client.shell.Shell.get_client方法的典型用法代码示例。如果您正苦于以下问题:Python Shell.get_client方法的具体用法?Python Shell.get_client怎么用?Python Shell.get_client使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类st2client.shell.Shell
的用法示例。
在下文中一共展示了Shell.get_client方法的5个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: test_automatic_auth_skipped_on_auth_command
# 需要导入模块: from st2client.shell import Shell [as 别名]
# 或者: from st2client.shell.Shell import get_client [as 别名]
def test_automatic_auth_skipped_on_auth_command(self):
self._write_mock_config()
shell = Shell()
shell._get_auth_token = mock.Mock()
argv = ['auth', 'testu', '-p', 'testp']
args = shell.parser.parse_args(args=argv)
shell.get_client(args=args)
self.assertEqual(shell._get_auth_token.call_count, 0)
示例2: test_automatic_auth_skipped_if_token_provided_as_env_variable
# 需要导入模块: from st2client.shell import Shell [as 别名]
# 或者: from st2client.shell.Shell import get_client [as 别名]
def test_automatic_auth_skipped_if_token_provided_as_env_variable(self):
self._write_mock_config()
shell = Shell()
shell._get_auth_token = mock.Mock()
os.environ['ST2_AUTH_TOKEN'] = 'fooo'
argv = ['action', 'list']
args = shell.parser.parse_args(args=argv)
shell.get_client(args=args)
self.assertEqual(shell._get_auth_token.call_count, 0)
示例3: test_automatic_auth_skipped_if_token_provided_as_cli_argument
# 需要导入模块: from st2client.shell import Shell [as 别名]
# 或者: from st2client.shell.Shell import get_client [as 别名]
def test_automatic_auth_skipped_if_token_provided_as_cli_argument(self):
self._write_mock_config()
shell = Shell()
shell._get_auth_token = mock.Mock()
argv = ['action', 'list', '--token=bar']
args = shell.parser.parse_args(args=argv)
shell.get_client(args=args)
self.assertEqual(shell._get_auth_token.call_count, 0)
argv = ['action', 'list', '-t', 'bar']
args = shell.parser.parse_args(args=argv)
shell.get_client(args=args)
self.assertEqual(shell._get_auth_token.call_count, 0)
示例4: TestShell
# 需要导入模块: from st2client.shell import Shell [as 别名]
# 或者: from st2client.shell.Shell import get_client [as 别名]
class TestShell(base.BaseCLITestCase):
capture_output = True
def __init__(self, *args, **kwargs):
super(TestShell, self).__init__(*args, **kwargs)
self.shell = Shell()
def test_endpoints_default(self):
base_url = 'http://127.0.0.1'
auth_url = 'http://127.0.0.1:9100'
api_url = 'http://127.0.0.1:9101/v1'
args = ['trigger', 'list']
parsed_args = self.shell.parser.parse_args(args)
client = self.shell.get_client(parsed_args)
self.assertEqual(client.endpoints['base'], base_url)
self.assertEqual(client.endpoints['auth'], auth_url)
self.assertEqual(client.endpoints['api'], api_url)
def test_endpoints_base_url_from_cli(self):
base_url = 'http://www.st2.com'
auth_url = 'http://www.st2.com:9100'
api_url = 'http://www.st2.com:9101/v1'
args = ['--url', base_url, 'trigger', 'list']
parsed_args = self.shell.parser.parse_args(args)
client = self.shell.get_client(parsed_args)
self.assertEqual(client.endpoints['base'], base_url)
self.assertEqual(client.endpoints['auth'], auth_url)
self.assertEqual(client.endpoints['api'], api_url)
def test_endpoints_base_url_from_env(self):
base_url = 'http://www.st2.com'
auth_url = 'http://www.st2.com:9100'
api_url = 'http://www.st2.com:9101/v1'
os.environ['ST2_BASE_URL'] = base_url
args = ['trigger', 'list']
parsed_args = self.shell.parser.parse_args(args)
client = self.shell.get_client(parsed_args)
self.assertEqual(client.endpoints['base'], base_url)
self.assertEqual(client.endpoints['auth'], auth_url)
self.assertEqual(client.endpoints['api'], api_url)
def test_endpoints_override_from_cli(self):
base_url = 'http://www.st2.com'
auth_url = 'http://www.st2.com:8888'
api_url = 'http://www.stackstorm1.com:9101/v1'
args = ['--url', base_url,
'--auth-url', auth_url,
'--api-url', api_url,
'trigger', 'list']
parsed_args = self.shell.parser.parse_args(args)
client = self.shell.get_client(parsed_args)
self.assertEqual(client.endpoints['base'], base_url)
self.assertEqual(client.endpoints['auth'], auth_url)
self.assertEqual(client.endpoints['api'], api_url)
def test_endpoints_override_from_env(self):
base_url = 'http://www.st2.com'
auth_url = 'http://www.st2.com:8888'
api_url = 'http://www.stackstorm1.com:9101/v1'
os.environ['ST2_BASE_URL'] = base_url
os.environ['ST2_AUTH_URL'] = auth_url
os.environ['ST2_API_URL'] = api_url
args = ['trigger', 'list']
parsed_args = self.shell.parser.parse_args(args)
client = self.shell.get_client(parsed_args)
self.assertEqual(client.endpoints['base'], base_url)
self.assertEqual(client.endpoints['auth'], auth_url)
self.assertEqual(client.endpoints['api'], api_url)
@mock.patch.object(
httpclient.HTTPClient, 'get',
mock.MagicMock(return_value=base.FakeResponse(json.dumps(base.RESOURCES), 200, 'OK')))
def test_exit_code_on_success(self):
argv = ['trigger', 'list']
self.assertEqual(self.shell.run(argv), 0)
@mock.patch.object(
httpclient.HTTPClient, 'get',
mock.MagicMock(return_value=base.FakeResponse(None, 500, 'INTERNAL SERVER ERROR')))
def test_exit_code_on_error(self):
argv = ['trigger', 'list']
self.assertEqual(self.shell.run(argv), 1)
def _validate_parser(self, args_list, is_subcommand=True):
for args in args_list:
ns = self.shell.parser.parse_args(args)
func = (self.shell.commands[args[0]].run_and_print
if not is_subcommand
else self.shell.commands[args[0]].commands[args[1]].run_and_print)
self.assertEqual(ns.func, func)
def test_action(self):
args_list = [
['action', 'list'],
['action', 'get', 'abc'],
['action', 'create', '/tmp/action.json'],
['action', 'update', '123', '/tmp/action.json'],
['action', 'delete', 'abc'],
['action', 'execute', '-h'],
['action', 'execute', 'remote', '-h'],
#.........这里部分代码省略.........
示例5: ShellTestCase
# 需要导入模块: from st2client.shell import Shell [as 别名]
# 或者: from st2client.shell.Shell import get_client [as 别名]
class ShellTestCase(base.BaseCLITestCase):
capture_output = True
def __init__(self, *args, **kwargs):
super(ShellTestCase, self).__init__(*args, **kwargs)
self.shell = Shell()
def setUp(self):
super(ShellTestCase, self).setUp()
if six.PY3:
# In python --version outputs to stdout and in 2.x to stderr
self.version_output = self.stdout
else:
self.version_output = self.stderr
def test_commands_usage_and_help_strings(self):
# No command, should print out user friendly usage / help string
self.assertEqual(self.shell.run([]), 2)
self.stderr.seek(0)
stderr = self.stderr.read()
self.assertTrue('Usage: ' in stderr)
self.assertTrue('For example:' in stderr)
self.assertTrue('CLI for StackStorm' in stderr)
self.assertTrue('positional arguments:' in stderr)
self.stdout.truncate()
self.stderr.truncate()
# --help should result in the same output
try:
self.assertEqual(self.shell.run(['--help']), 0)
except SystemExit as e:
self.assertEqual(e.code, 0)
self.stdout.seek(0)
stdout = self.stdout.read()
self.assertTrue('Usage: ' in stdout)
self.assertTrue('For example:' in stdout)
self.assertTrue('CLI for StackStorm' in stdout)
self.assertTrue('positional arguments:' in stdout)
self.stdout.truncate()
self.stderr.truncate()
# Sub command with no args
try:
self.assertEqual(self.shell.run(['action']), 2)
except SystemExit as e:
self.assertEqual(e.code, 2)
self.stderr.seek(0)
stderr = self.stderr.read()
self.assertTrue('usage' in stderr)
if six.PY2:
self.assertTrue('{list,get,create,update' in stderr)
self.assertTrue('error: too few arguments' in stderr)
def test_endpoints_default(self):
base_url = 'http://127.0.0.1'
auth_url = 'http://127.0.0.1:9100'
api_url = 'http://127.0.0.1:9101/v1'
stream_url = 'http://127.0.0.1:9102/v1'
args = ['trigger', 'list']
parsed_args = self.shell.parser.parse_args(args)
client = self.shell.get_client(parsed_args)
self.assertEqual(client.endpoints['base'], base_url)
self.assertEqual(client.endpoints['auth'], auth_url)
self.assertEqual(client.endpoints['api'], api_url)
self.assertEqual(client.endpoints['stream'], stream_url)
def test_endpoints_base_url_from_cli(self):
base_url = 'http://www.st2.com'
auth_url = 'http://www.st2.com:9100'
api_url = 'http://www.st2.com:9101/v1'
stream_url = 'http://www.st2.com:9102/v1'
args = ['--url', base_url, 'trigger', 'list']
parsed_args = self.shell.parser.parse_args(args)
client = self.shell.get_client(parsed_args)
self.assertEqual(client.endpoints['base'], base_url)
self.assertEqual(client.endpoints['auth'], auth_url)
self.assertEqual(client.endpoints['api'], api_url)
self.assertEqual(client.endpoints['stream'], stream_url)
def test_endpoints_base_url_from_env(self):
base_url = 'http://www.st2.com'
auth_url = 'http://www.st2.com:9100'
api_url = 'http://www.st2.com:9101/v1'
stream_url = 'http://www.st2.com:9102/v1'
os.environ['ST2_BASE_URL'] = base_url
args = ['trigger', 'list']
parsed_args = self.shell.parser.parse_args(args)
client = self.shell.get_client(parsed_args)
self.assertEqual(client.endpoints['base'], base_url)
self.assertEqual(client.endpoints['auth'], auth_url)
self.assertEqual(client.endpoints['api'], api_url)
self.assertEqual(client.endpoints['stream'], stream_url)
#.........这里部分代码省略.........