448 lines
16 KiB
Python
448 lines
16 KiB
Python
# Copyright [2015] Hewlett-Packard Development Company, L.P.
|
|
# 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 fixtures
|
|
import mock
|
|
import six
|
|
|
|
import troveclient.client
|
|
from troveclient import exceptions
|
|
import troveclient.shell
|
|
from troveclient.tests import fakes
|
|
from troveclient.tests import utils
|
|
import troveclient.v1.shell
|
|
|
|
|
|
class ShellFixture(fixtures.Fixture):
|
|
def setUp(self):
|
|
super(ShellFixture, self).setUp()
|
|
self.shell = troveclient.shell.OpenStackTroveShell()
|
|
|
|
def tearDown(self):
|
|
if hasattr(self.shell, 'cs'):
|
|
self.shell.cs.clear_callstack()
|
|
super(ShellFixture, self).tearDown()
|
|
|
|
|
|
class ShellTest(utils.TestCase):
|
|
FAKE_ENV = {
|
|
'OS_USERNAME': 'username',
|
|
'OS_PASSWORD': 'password',
|
|
'OS_PROJECT_ID': 'project_id',
|
|
'OS_AUTH_URL': 'http://no.where/v2.0',
|
|
}
|
|
|
|
def setUp(self, *args):
|
|
"""Run before each test."""
|
|
super(ShellTest, self).setUp()
|
|
|
|
for var in self.FAKE_ENV:
|
|
self.useFixture(fixtures.EnvironmentVariable(var,
|
|
self.FAKE_ENV[var]))
|
|
self.shell = self.useFixture(ShellFixture()).shell
|
|
|
|
@mock.patch('sys.stdout', new_callable=six.StringIO)
|
|
@mock.patch('troveclient.client.get_version_map',
|
|
return_value=fakes.get_version_map())
|
|
def run_command(self, cmd, mock_stdout, mock_get_version_map):
|
|
if isinstance(cmd, list):
|
|
self.shell.main(cmd)
|
|
else:
|
|
self.shell.main(cmd.split())
|
|
return mock_stdout.getvalue()
|
|
|
|
def assert_called(self, method, url, body=None, **kwargs):
|
|
return self.shell.cs.assert_called(method, url, body, **kwargs)
|
|
|
|
def assert_called_anytime(self, method, url, body=None):
|
|
return self.shell.cs.assert_called_anytime(method, url, body)
|
|
|
|
def test_instance_list(self):
|
|
self.run_command('list')
|
|
self.assert_called('GET', '/instances')
|
|
|
|
def test_instance_show(self):
|
|
self.run_command('show 1234')
|
|
self.assert_called('GET', '/instances/1234')
|
|
|
|
def test_instance_delete(self):
|
|
self.run_command('delete 1234')
|
|
self.assert_called('DELETE', '/instances/1234')
|
|
|
|
def test_instance_update(self):
|
|
self.run_command('update 1234')
|
|
self.assert_called('PATCH', '/instances/1234')
|
|
|
|
def test_resize_flavor(self):
|
|
self.run_command('resize-flavor 1234 1')
|
|
self.assert_called('POST', '/instances/1234/action')
|
|
|
|
def test_resize_volume(self):
|
|
self.run_command('resize-volume 1234 3')
|
|
self.assert_called('POST', '/instances/1234/action')
|
|
|
|
def test_restart(self):
|
|
self.run_command('restart 1234')
|
|
self.assert_called('POST', '/instances/1234/action')
|
|
|
|
def test_detach_replica(self):
|
|
self.run_command('detach-replica 1234')
|
|
self.assert_called('PATCH', '/instances/1234')
|
|
|
|
def test_promote_to_replica_source(self):
|
|
self.run_command('promote-to-replica-source 1234')
|
|
self.assert_called('POST', '/instances/1234/action')
|
|
|
|
def test_eject_replica_source(self):
|
|
self.run_command('eject-replica-source 1234')
|
|
self.assert_called('POST', '/instances/1234/action')
|
|
|
|
def test_flavor_list(self):
|
|
self.run_command('flavor-list')
|
|
self.assert_called('GET', '/flavors')
|
|
|
|
def test_flavor_list_with_datastore(self):
|
|
cmd = ('flavor-list --datastore_type mysql '
|
|
'--datastore_version_id some-version-id')
|
|
self.run_command(cmd)
|
|
self.assert_called(
|
|
'GET', '/datastores/mysql/versions/some-version-id/flavors')
|
|
|
|
def test_flavor_list_error(self):
|
|
cmd = 'flavor-list --datastore_type mysql'
|
|
exepcted_error_msg = ('Specify both <datastore_type> and '
|
|
'<datastore_version_id> to list datastore '
|
|
'version associated flavors')
|
|
self.assertRaisesRegexp(
|
|
exceptions.CommandError, exepcted_error_msg, self.run_command, cmd)
|
|
|
|
def test_flavor_show(self):
|
|
self.run_command('flavor-show 1')
|
|
self.assert_called('GET', '/flavors/1')
|
|
|
|
def test_flavor_show_uuid(self):
|
|
self.run_command('flavor-show m1.uuid')
|
|
self.assert_called('GET', '/flavors/m1.uuid')
|
|
|
|
def test_cluster_list(self):
|
|
self.run_command('cluster-list')
|
|
self.assert_called('GET', '/clusters')
|
|
|
|
def test_cluster_show(self):
|
|
self.run_command('cluster-show cls-1234')
|
|
self.assert_called('GET', '/clusters/cls-1234')
|
|
|
|
def test_cluster_instances(self):
|
|
self.run_command('cluster-instances cls-1234')
|
|
self.assert_called('GET', '/clusters/cls-1234')
|
|
|
|
def test_cluster_delete(self):
|
|
self.run_command('cluster-delete cls-1234')
|
|
self.assert_called('DELETE', '/clusters/cls-1234')
|
|
|
|
def test_boot(self):
|
|
self.run_command('create test-member-1 1 --size 1')
|
|
self.assert_called_anytime(
|
|
'POST', '/instances',
|
|
{'instance': {
|
|
'volume': {'size': 1},
|
|
'flavorRef': '1',
|
|
'name': 'test-member-1',
|
|
'replica_count': 1
|
|
}})
|
|
|
|
def test_boot_nic_error(self):
|
|
cmd = ('create test-member-1 1 --size 1 '
|
|
'--nic net-id=some-id,port-id=some-id')
|
|
self.assertRaisesRegexp(
|
|
exceptions.CommandError, 'Invalid nic argument',
|
|
self.run_command, cmd)
|
|
|
|
def test_cluster_create(self):
|
|
cmd = ('cluster-create test-clstr vertica 7.1 '
|
|
'--instance flavor_id=2,volume=2 '
|
|
'--instance flavor_id=2,volume=1')
|
|
self.run_command(cmd)
|
|
self.assert_called_anytime(
|
|
'POST', '/clusters',
|
|
{'cluster': {
|
|
'instances': [
|
|
{
|
|
'volume': {'size': '2'},
|
|
'flavorRef': '2'
|
|
},
|
|
{
|
|
'volume': {'size': '1'},
|
|
'flavorRef': '2'
|
|
}],
|
|
'datastore': {'version': '7.1', 'type': 'vertica'},
|
|
'name': 'test-clstr'}})
|
|
|
|
def test_cluster_create_error(self):
|
|
cmd = ('cluster-create test-clstr vertica 7.1 --instance volume=2 '
|
|
'--instance flavor_id=2,volume=1')
|
|
self.assertRaisesRegexp(
|
|
exceptions.CommandError, 'flavor_id is required',
|
|
self.run_command, cmd)
|
|
|
|
def test_datastore_list(self):
|
|
self.run_command('datastore-list')
|
|
self.assert_called('GET', '/datastores')
|
|
|
|
def test_datastore_show(self):
|
|
self.run_command('datastore-show d-123')
|
|
self.assert_called('GET', '/datastores/d-123')
|
|
|
|
def test_datastore_version_list(self):
|
|
self.run_command('datastore-version-list d-123')
|
|
self.assert_called('GET', '/datastores/d-123/versions')
|
|
|
|
def test_datastore_version_show(self):
|
|
self.run_command('datastore-version-show v-56 --datastore d-123')
|
|
self.assert_called('GET', '/datastores/d-123/versions/v-56')
|
|
|
|
def test_datastore_version_show_error(self):
|
|
expected_error_msg = ('The datastore name or id is required to '
|
|
'retrieve a datastore version by name.')
|
|
self.assertRaisesRegexp(exceptions.NoUniqueMatch, expected_error_msg,
|
|
self.run_command,
|
|
'datastore-version-show v-56')
|
|
|
|
def test_configuration_list(self):
|
|
self.run_command('configuration-list')
|
|
self.assert_called('GET', '/configurations')
|
|
|
|
def test_configuration_show(self):
|
|
self.run_command('configuration-show c-123')
|
|
self.assert_called('GET', '/configurations/c-123')
|
|
|
|
def test_configuration_create(self):
|
|
cmd = "configuration-create c-123 some-thing"
|
|
self.assertRaises(ValueError, self.run_command, cmd)
|
|
|
|
def test_configuration_update(self):
|
|
cmd = "configuration-update c-123 some-thing"
|
|
self.assertRaises(ValueError, self.run_command, cmd)
|
|
|
|
def test_configuration_patch(self):
|
|
cmd = "configuration-patch c-123 some-thing"
|
|
self.assertRaises(ValueError, self.run_command, cmd)
|
|
|
|
def test_configuration_parameter_list(self):
|
|
cmd = 'configuration-parameter-list v-156 --datastore d-123'
|
|
self.run_command(cmd)
|
|
self.assert_called('GET',
|
|
'/datastores/d-123/versions/v-156/parameters')
|
|
|
|
def test_configuration_parameter_list_error(self):
|
|
expected_error_msg = ('The datastore name or id is required to '
|
|
'retrieve the parameters for the configuration '
|
|
'group by name')
|
|
self.assertRaisesRegexp(
|
|
exceptions.NoUniqueMatch, expected_error_msg,
|
|
self.run_command, 'configuration-parameter-list v-156')
|
|
|
|
def test_configuration_parameter_show(self):
|
|
cmd = ('configuration-parameter-show v_56 '
|
|
'max_connections --datastore d_123')
|
|
self.run_command(cmd)
|
|
self.assert_called(
|
|
'GET',
|
|
'/datastores/d_123/versions/v_56/parameters/max_connections')
|
|
|
|
def test_configuration_instances(self):
|
|
cmd = 'configuration-instances c-123'
|
|
self.run_command(cmd)
|
|
self.assert_called('GET', '/configurations/c-123/instances')
|
|
|
|
def test_configuration_delete(self):
|
|
self.run_command('configuration-delete c-123')
|
|
self.assert_called('DELETE', '/configurations/c-123')
|
|
|
|
def test_configuration_default(self):
|
|
self.run_command('configuration-default 1234')
|
|
self.assert_called('GET', '/instances/1234/configuration')
|
|
|
|
def test_configuration_attach(self):
|
|
self.run_command('configuration-attach 1234 c-123')
|
|
self.assert_called('PUT', '/instances/1234')
|
|
|
|
def test_configuration_detach(self):
|
|
self.run_command('configuration-detach 1234')
|
|
self.assert_called('PUT', '/instances/1234')
|
|
|
|
def test_metadata_edit(self):
|
|
self.run_command('metadata-edit 1234 key-123 value-123')
|
|
self.assert_called('PATCH', '/instances/1234/metadata/key-123')
|
|
|
|
def test_metadata_update(self):
|
|
self.run_command('metadata-update 1234 key-123 key-456 value-123')
|
|
self.assert_called('PUT', '/instances/1234/metadata/key-123')
|
|
|
|
def test_metadata_delete(self):
|
|
self.run_command('metadata-delete 1234 key-123')
|
|
self.assert_called('DELETE', '/instances/1234/metadata/key-123')
|
|
|
|
def test_metadata_create(self):
|
|
self.run_command('metadata-create 1234 key123 value123')
|
|
self.assert_called_anytime(
|
|
'POST', '/instances/1234/metadata/key123',
|
|
{'metadata': {'value': 'value123'}})
|
|
|
|
def test_metadata_list(self):
|
|
self.run_command('metadata-list 1234')
|
|
self.assert_called('GET', '/instances/1234/metadata')
|
|
|
|
def test_metadata_show(self):
|
|
self.run_command('metadata-show 1234 key123')
|
|
self.assert_called('GET', '/instances/1234/metadata/key123')
|
|
|
|
def test_limit_list(self):
|
|
self.run_command('limit-list')
|
|
self.assert_called('GET', '/limits')
|
|
|
|
def test_backup_list(self):
|
|
self.run_command('backup-list')
|
|
self.assert_called('GET', '/backups')
|
|
|
|
def test_backup_show(self):
|
|
self.run_command('backup-show bk-1234')
|
|
self.assert_called('GET', '/backups/bk-1234')
|
|
|
|
def test_backup_list_instance(self):
|
|
self.run_command('backup-list-instance 1234')
|
|
self.assert_called('GET', '/instances/1234/backups')
|
|
|
|
def test_backup_delete(self):
|
|
self.run_command('backup-delete bk-1234')
|
|
self.assert_called('DELETE', '/backups/bk-1234')
|
|
|
|
def test_backup_create(self):
|
|
self.run_command('backup-create 1234 bkp_1')
|
|
self.assert_called_anytime(
|
|
'POST', '/backups',
|
|
{'backup': {
|
|
'instance': '1234',
|
|
'name': 'bkp_1'
|
|
}})
|
|
|
|
def test_backup_copy(self):
|
|
self.run_command('backup-copy new_bkp bk-1234')
|
|
self.assert_called_anytime(
|
|
'POST', '/backups',
|
|
{'backup': {
|
|
'name': 'new_bkp',
|
|
'backup': {'region': None, 'id': 'bk-1234'}
|
|
}})
|
|
|
|
def test_database_list(self):
|
|
self.run_command('database-list 1234')
|
|
self.assert_called('GET', '/instances/1234/databases')
|
|
|
|
def test_database_delete(self):
|
|
self.run_command('database-delete 1234 db_1')
|
|
self.assert_called('DELETE', '/instances/1234/databases/db_1')
|
|
|
|
def test_database_create(self):
|
|
cmd = ('database-create 1234 db_1 --character_set utf8 '
|
|
'--collate utf8_general_ci')
|
|
self.run_command(cmd)
|
|
self.assert_called_anytime(
|
|
'POST', '/instances/1234/databases',
|
|
{'databases': [{'character_set': 'utf8',
|
|
'name': 'db_1',
|
|
'collate': 'utf8_general_ci'}]})
|
|
|
|
def test_user_list(self):
|
|
self.run_command('user-list 1234')
|
|
self.assert_called('GET', '/instances/1234/users')
|
|
|
|
def test_user_show(self):
|
|
self.run_command('user-show 1234 jacob')
|
|
self.assert_called('GET', '/instances/1234/users/jacob')
|
|
|
|
def test_user_delete(self):
|
|
self.run_command('user-delete 1234 jacob')
|
|
self.assert_called('DELETE', '/instances/1234/users/jacob')
|
|
|
|
def test_user_create(self):
|
|
self.run_command('user-create 1234 jacob password')
|
|
self.assert_called_anytime(
|
|
'POST', '/instances/1234/users',
|
|
{'users': [{
|
|
'password': 'password',
|
|
'name': 'jacob',
|
|
'databases': []}]})
|
|
|
|
def test_user_show_access(self):
|
|
self.run_command('user-show-access 1234 jacob')
|
|
self.assert_called('GET', '/instances/1234/users/jacob/databases')
|
|
|
|
def test_user_update_host(self):
|
|
cmd = 'user-update-attributes 1234 jacob --new_host 10.0.0.1'
|
|
self.run_command(cmd)
|
|
self.assert_called('PUT', '/instances/1234/users/jacob')
|
|
|
|
def test_user_update_name(self):
|
|
self.run_command('user-update-attributes 1234 jacob --new_name sam')
|
|
self.assert_called('PUT', '/instances/1234/users/jacob')
|
|
|
|
def test_user_update_password(self):
|
|
cmd = 'user-update-attributes 1234 jacob --new_password new_pwd'
|
|
self.run_command(cmd)
|
|
self.assert_called('PUT', '/instances/1234/users/jacob')
|
|
|
|
def test_user_grant_access(self):
|
|
self.run_command('user-grant-access 1234 jacob db1 db2')
|
|
self.assert_called('PUT', '/instances/1234/users/jacob/databases')
|
|
|
|
def test_user_revoke_access(self):
|
|
self.run_command('user-revoke-access 1234 jacob db1')
|
|
self.assert_called('DELETE',
|
|
'/instances/1234/users/jacob/databases/db1')
|
|
|
|
def test_root_enable(self):
|
|
self.run_command('root-enable 1234')
|
|
self.assert_called_anytime('POST', '/instances/1234/root')
|
|
|
|
def test_root_show(self):
|
|
self.run_command('root-show 1234')
|
|
self.assert_called('GET', '/instances/1234/root')
|
|
|
|
def test_secgroup_list(self):
|
|
self.run_command('secgroup-list')
|
|
self.assert_called('GET', '/security-groups')
|
|
|
|
def test_secgroup_show(self):
|
|
self.run_command('secgroup-show 2')
|
|
self.assert_called('GET', '/security-groups/2')
|
|
|
|
def test_secgroup_list_rules(self):
|
|
self.run_command('secgroup-list-rules 2')
|
|
self.assert_called('GET', '/security-groups/2')
|
|
|
|
def test_secgroup_delete_rule(self):
|
|
self.run_command('secgroup-delete-rule 2')
|
|
self.assert_called('DELETE', '/security-group-rules/2')
|
|
|
|
def test_secgroup_add_rule(self):
|
|
self.run_command('secgroup-add-rule 2 15.0.0.0/24')
|
|
self.assert_called_anytime(
|
|
'POST', '/security-group-rules',
|
|
{'security_group_rule': {
|
|
'cidr': '15.0.0.0/24',
|
|
'group_id': '2',
|
|
}})
|