0e01835b7f
1.Cyborg should support filtering queries. This patch is the implementation of client. 2.Add UT for cyborgclient. Change-Id: Ia4a98be93f2a32e7e818d4831a24652c78b97788
280 lines
11 KiB
Python
280 lines
11 KiB
Python
# Copyright 2015 NEC Corporation. All rights reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import re
|
|
import sys
|
|
|
|
import fixtures
|
|
from keystoneauth1 import fixture
|
|
import mock
|
|
import six
|
|
from testtools import matchers
|
|
|
|
from cyborgclient import exceptions
|
|
import cyborgclient.shell
|
|
from cyborgclient.tests.unit import utils
|
|
|
|
FAKE_ENV = {'OS_USERNAME': 'username',
|
|
'OS_PASSWORD': 'password',
|
|
'OS_PROJECT_NAME': 'project_name',
|
|
'OS_AUTH_URL': 'http://no.where/v2.0'}
|
|
|
|
FAKE_ENV2 = {'OS_USER_ID': 'user_id',
|
|
'OS_PASSWORD': 'password',
|
|
'OS_PROJECT_ID': 'project_id',
|
|
'OS_AUTH_URL': 'http://no.where/v2.0'}
|
|
|
|
FAKE_ENV3 = {'OS_USERNAME': 'username',
|
|
'OS_PASSWORD': 'password',
|
|
'OS_PROJECT_ID': 'project_id',
|
|
'OS_AUTH_URL': 'http://no.where/v2.0'}
|
|
|
|
FAKE_ENV4 = {'OS_USERNAME': 'username',
|
|
'OS_PASSWORD': 'password',
|
|
'OS_PROJECT_ID': 'project_id',
|
|
'OS_USER_DOMAIN_NAME': 'Default',
|
|
'OS_PROJECT_DOMAIN_NAME': 'Default',
|
|
'OS_AUTH_URL': 'http://no.where/v3'}
|
|
|
|
|
|
def _create_ver_list(versions):
|
|
return {'versions': {'values': versions}}
|
|
|
|
|
|
class ShellTest(utils.TestCase):
|
|
AUTH_URL = utils.FAKE_ENV['OS_AUTH_URL']
|
|
|
|
_msg_no_tenant_project = ("You must provide a project name or project id"
|
|
" via --os-project-name, --os-project-id,"
|
|
" env[OS_PROJECT_NAME] or env[OS_PROJECT_ID]")
|
|
|
|
def setUp(self):
|
|
super(ShellTest, self).setUp()
|
|
self.nc_util = mock.patch(
|
|
'cyborgclient.common.cliutils.isunauthenticated').start()
|
|
self.nc_util.return_value = False
|
|
|
|
def test_help_unknown_command(self):
|
|
self.assertRaises(exceptions.CommandError, self.shell, 'help foofoo')
|
|
|
|
def test_help(self):
|
|
required = [
|
|
'.*?^usage: ',
|
|
'.*?^See "cyborg help COMMAND" for help on a specific command',
|
|
]
|
|
stdout, stderr = self.shell('help')
|
|
for r in required:
|
|
self.assertThat((stdout + stderr),
|
|
matchers.MatchesRegex(r, re.DOTALL | re.MULTILINE))
|
|
|
|
def test_help_no_options(self):
|
|
required = [
|
|
'.*?^usage: ',
|
|
'.*?^See "cyborg help COMMAND" for help on a specific command.',
|
|
]
|
|
stdout, stderr = self.shell('')
|
|
for r in required:
|
|
self.assertThat((stdout + stderr),
|
|
matchers.MatchesRegex(r, re.DOTALL | re.MULTILINE))
|
|
|
|
def test_bash_completion(self):
|
|
stdout, stderr = self.shell('bash-completion')
|
|
# just check we have some output
|
|
required = [
|
|
'.*--fields',
|
|
'.*help',
|
|
'.*accelerator-list',
|
|
'.*--help']
|
|
for r in required:
|
|
self.assertThat((stdout + stderr),
|
|
matchers.MatchesRegex(r, re.DOTALL | re.MULTILINE))
|
|
|
|
def test_no_username(self):
|
|
required = ('You must provide a username via either'
|
|
' --os-username or via env[OS_USERNAME]')
|
|
self.make_env(exclude='OS_USERNAME')
|
|
try:
|
|
self.shell('accelerator-list')
|
|
except exceptions.CommandError as exc:
|
|
self.assertEqual(required, exc.message)
|
|
else:
|
|
self.fail('CommandError not raised')
|
|
|
|
def test_no_user_id(self):
|
|
required = ('You must provide a username via'
|
|
' either --os-username or via env[OS_USERNAME]')
|
|
self.make_env(exclude='OS_USER_ID', fake_env=FAKE_ENV2)
|
|
try:
|
|
self.shell('accelerator-list')
|
|
except exceptions.CommandError as exc:
|
|
self.assertEqual(required, exc.message)
|
|
else:
|
|
self.fail('CommandError not raised')
|
|
|
|
def test_no_project_name(self):
|
|
required = self._msg_no_tenant_project
|
|
self.make_env(exclude='OS_PROJECT_NAME')
|
|
try:
|
|
self.shell('accelerator-list')
|
|
except exceptions.CommandError as exc:
|
|
self.assertEqual(required, exc.message)
|
|
else:
|
|
self.fail('CommandError not raised')
|
|
|
|
def test_no_project_id(self):
|
|
required = self._msg_no_tenant_project
|
|
self.make_env(exclude='OS_PROJECT_ID', fake_env=FAKE_ENV3)
|
|
try:
|
|
self.shell('accelerator-list')
|
|
except exceptions.CommandError as exc:
|
|
self.assertEqual(required, exc.message)
|
|
else:
|
|
self.fail('CommandError not raised')
|
|
|
|
def test_no_auth_url(self):
|
|
required = ('You must provide an auth url'
|
|
' via either --os-auth-url or via env[OS_AUTH_URL]')
|
|
self.make_env(exclude='OS_AUTH_URL')
|
|
try:
|
|
self.shell('accelerator-list')
|
|
except exceptions.CommandError as exc:
|
|
self.assertEqual(required, exc.message)
|
|
else:
|
|
self.fail('CommandError not raised')
|
|
|
|
@mock.patch('cyborgclient.v1.client.Client')
|
|
def test_service_type(self, mock_client):
|
|
self.make_env()
|
|
self.shell('accelerator-list')
|
|
_, client_kwargs = mock_client.call_args_list[0]
|
|
self.assertEqual('accelerator', client_kwargs['service_type'])
|
|
|
|
@mock.patch('cyborgclient.v1.client.Client')
|
|
def test_insecure(self, mock_client):
|
|
self.make_env()
|
|
self.shell('--insecure accelerator-list')
|
|
_, session_kwargs = mock_client.call_args_list[0]
|
|
self.assertEqual(True, session_kwargs['insecure'])
|
|
|
|
@mock.patch('sys.stdin', side_effect=mock.MagicMock)
|
|
@mock.patch('getpass.getpass', side_effect=EOFError)
|
|
def test_no_password(self, mock_getpass, mock_stdin):
|
|
required = ('You must provide a password via either --os-password, '
|
|
'env[OS_PASSWORD], or prompted response')
|
|
self.make_env(exclude='OS_PASSWORD')
|
|
try:
|
|
self.shell('accelerator-list')
|
|
except exceptions.CommandError as exc:
|
|
self.assertEqual(required, exc.message)
|
|
else:
|
|
self.fail('CommandError not raised')
|
|
|
|
@mock.patch('sys.argv', ['cyborg'])
|
|
@mock.patch('sys.stdout', six.StringIO())
|
|
@mock.patch('sys.stderr', six.StringIO())
|
|
def test_main_noargs(self):
|
|
# Ensure that main works with no command-line arguments
|
|
try:
|
|
cyborgclient.shell.main()
|
|
except SystemExit:
|
|
self.fail('Unexpected SystemExit')
|
|
|
|
# We expect the normal usage as a result
|
|
self.assertIn('Command-line interface to the OpenStack Cyborg API',
|
|
sys.stdout.getvalue())
|
|
|
|
@mock.patch('cyborgclient.v1.client.Client')
|
|
def _test_main_region(self, command, expected_region_name, mock_client):
|
|
self.shell(command)
|
|
mock_client.assert_called_once_with(
|
|
api_version='latest', auth_token=None,
|
|
auth_url=self.AUTH_URL, cloud=None, cyborg_url=None,
|
|
insecure=False, interface='public', password='password',
|
|
project_domain_id=None, project_domain_name=None, project_id=None,
|
|
project_name='project_name', region_name=expected_region_name,
|
|
service_type='accelerator', user_domain_id=None,
|
|
user_domain_name=None, user_id=None, username='username')
|
|
|
|
def test_main_option_region(self):
|
|
self.make_env()
|
|
self._test_main_region('--os-region-name=myregion accelerator-list',
|
|
'myregion')
|
|
|
|
def test_main_env_region(self):
|
|
fake_env = dict(utils.FAKE_ENV, OS_REGION_NAME='myregion')
|
|
self.make_env(fake_env=fake_env)
|
|
self._test_main_region('accelerator-list', 'myregion')
|
|
|
|
def test_main_no_region(self):
|
|
self.make_env()
|
|
self._test_main_region('accelerator-list', None)
|
|
|
|
@mock.patch('cyborgclient.v1.client.Client')
|
|
def test_main_endpoint_public(self, mock_client):
|
|
self.make_env()
|
|
self.shell('--endpoint-type publicURL accelerator-list')
|
|
mock_client.assert_called_once_with(
|
|
username='username', password='password',
|
|
interface='public', project_id=None,
|
|
project_name='project_name', auth_url=self.AUTH_URL,
|
|
service_type='accelerator', region_name=None,
|
|
project_domain_id=None, project_domain_name=None,
|
|
user_domain_id=None, user_domain_name=None,
|
|
insecure=False, user_id=None, api_version='latest',
|
|
auth_token=None, cyborg_url=None, cloud=None)
|
|
|
|
@mock.patch('cyborgclient.v1.client.Client')
|
|
def test_main_endpoint_internal(self, mock_client):
|
|
self.make_env()
|
|
self.shell('--endpoint-type internalURL accelerator-list')
|
|
mock_client.assert_called_once_with(
|
|
api_version='latest', auth_token=None,
|
|
auth_url=self.AUTH_URL, cloud=None, cyborg_url=None,
|
|
insecure=False, interface='internal', password='password',
|
|
project_domain_id=None, project_domain_name=None, project_id=None,
|
|
project_name='project_name', region_name=None,
|
|
service_type='accelerator', user_domain_id=None,
|
|
user_domain_name=None, user_id=None, username='username')
|
|
|
|
|
|
class ShellTestKeystoneV3(ShellTest):
|
|
AUTH_URL = 'http://no.where/v3'
|
|
|
|
def make_env(self, exclude=None, fake_env=FAKE_ENV):
|
|
if 'OS_AUTH_URL' in fake_env:
|
|
fake_env.update({'OS_AUTH_URL': self.AUTH_URL})
|
|
env = dict((k, v) for k, v in fake_env.items() if k != exclude)
|
|
self.useFixture(fixtures.MonkeyPatch('os.environ', env))
|
|
|
|
def register_keystone_discovery_fixture(self, mreq):
|
|
v3_url = "http://no.where/v3"
|
|
v3_version = fixture.V3Discovery(v3_url)
|
|
mreq.register_uri(
|
|
'GET', v3_url, json=_create_ver_list([v3_version]),
|
|
status_code=200)
|
|
|
|
@mock.patch('cyborgclient.v1.client.Client')
|
|
def test_main_endpoint_public(self, mock_client):
|
|
self.make_env(fake_env=FAKE_ENV4)
|
|
self.shell('--endpoint-type publicURL accelerator-list')
|
|
mock_client.assert_called_once_with(
|
|
username='username', password='password',
|
|
interface='public', project_id='project_id',
|
|
project_name=None, auth_url=self.AUTH_URL,
|
|
service_type='accelerator', region_name=None,
|
|
project_domain_id=None, project_domain_name='Default',
|
|
user_domain_id=None, user_domain_name='Default',
|
|
insecure=False, user_id=None, api_version='latest',
|
|
auth_token=None, cyborg_url=None, cloud=None)
|