61842b86e0
* Add flags to output cmd results in json(--json) and pretty print(--pprint) format Change-Id: Id4331761e4f9c4bb6735745ed8461eaa671d0945 Closes-Bug: #1341501
213 lines
10 KiB
Python
213 lines
10 KiB
Python
# Copyright 2013: Mirantis Inc.
|
|
# 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 os
|
|
|
|
import mock
|
|
|
|
from rally.cmd.commands import deployment
|
|
from rally import exceptions
|
|
from tests import test
|
|
|
|
|
|
class DeploymentCommandsTestCase(test.TestCase):
|
|
def setUp(self):
|
|
super(DeploymentCommandsTestCase, self).setUp()
|
|
self.deployment = deployment.DeploymentCommands()
|
|
|
|
@mock.patch.dict(os.environ, {'RALLY_DEPLOYMENT': 'my_deploy_id'})
|
|
@mock.patch('rally.cmd.commands.deployment.DeploymentCommands.list')
|
|
@mock.patch('rally.cmd.commands.deployment.api.create_deploy')
|
|
@mock.patch('rally.cmd.commands.deployment.open',
|
|
mock.mock_open(read_data='{"some": "json"}'),
|
|
create=True)
|
|
def test_create(self, mock_create, mock_list):
|
|
self.deployment.create('fake_deploy', False, 'path_to_config.json')
|
|
mock_create.assert_called_once_with({'some': 'json'}, 'fake_deploy')
|
|
|
|
@mock.patch.dict(os.environ, {'OS_AUTH_URL': 'fake_auth_url',
|
|
'OS_USERNAME': 'fake_username',
|
|
'OS_PASSWORD': 'fake_password',
|
|
'OS_TENANT_NAME': 'fake_tenant_name',
|
|
'OS_REGION_NAME': 'fake_region_name',
|
|
'RALLY_DEPLOYMENT': 'fake_deployment_id'})
|
|
@mock.patch('rally.cmd.commands.deployment.api.create_deploy')
|
|
@mock.patch('rally.cmd.commands.deployment.DeploymentCommands.list')
|
|
def test_createfromenv(self, mock_list, mock_create):
|
|
self.deployment.create('from_env', True)
|
|
mock_create.assert_called_once_with(
|
|
{
|
|
"type": "ExistingCloud",
|
|
"endpoint": {
|
|
"auth_url": 'fake_auth_url',
|
|
"username": 'fake_username',
|
|
"password": 'fake_password',
|
|
"tenant_name": 'fake_tenant_name',
|
|
"region_name": 'fake_region_name'
|
|
}
|
|
},
|
|
'from_env'
|
|
)
|
|
|
|
@mock.patch('rally.cmd.commands.deployment.DeploymentCommands.list')
|
|
@mock.patch('rally.cmd.commands.use.UseCommands.deployment')
|
|
@mock.patch('rally.cmd.commands.deployment.api.create_deploy',
|
|
return_value=dict(uuid='uuid'))
|
|
@mock.patch('rally.cmd.commands.deployment.open',
|
|
mock.mock_open(read_data='{"uuid": "uuid"}'),
|
|
create=True)
|
|
def test_create_and_use(self, mock_create, mock_use_deployment,
|
|
mock_list):
|
|
self.deployment.create('fake_deploy', False, 'path_to_config.json',
|
|
True)
|
|
mock_create.assert_called_once_with({'uuid': 'uuid'}, 'fake_deploy')
|
|
mock_use_deployment.assert_called_once_with('uuid')
|
|
|
|
@mock.patch('rally.cmd.commands.deployment.api.recreate_deploy')
|
|
def test_recreate(self, mock_recreate):
|
|
deploy_id = '43924f8b-9371-4152-af9f-4cf02b4eced4'
|
|
self.deployment.recreate(deploy_id)
|
|
mock_recreate.assert_called_once_with(deploy_id)
|
|
|
|
@mock.patch('rally.cmd.commands.deployment.envutils.get_global')
|
|
def test_recreate_no_deploy_id(self, mock_default):
|
|
mock_default.side_effect = exceptions.InvalidArgumentsException
|
|
self.assertRaises(exceptions.InvalidArgumentsException,
|
|
self.deployment.recreate, None)
|
|
|
|
@mock.patch('rally.cmd.commands.deployment.api.destroy_deploy')
|
|
def test_destroy(self, mock_destroy):
|
|
deploy_id = '53fd0273-60ce-42e5-a759-36f1a683103e'
|
|
self.deployment.destroy(deploy_id)
|
|
mock_destroy.assert_called_once_with(deploy_id)
|
|
|
|
@mock.patch('rally.cmd.commands.deployment.envutils.get_global')
|
|
def test_destroy_no_deploy_id(self, mock_default):
|
|
mock_default.side_effect = exceptions.InvalidArgumentsException
|
|
self.assertRaises(exceptions.InvalidArgumentsException,
|
|
self.deployment.destroy, None)
|
|
|
|
@mock.patch('rally.cmd.commands.deployment.common_cliutils.print_list')
|
|
@mock.patch('rally.cmd.commands.deployment.utils.Struct')
|
|
@mock.patch('rally.cmd.commands.deployment.envutils.get_global')
|
|
@mock.patch('rally.cmd.commands.deployment.db.deployment_list')
|
|
def test_list_different_deploy_id(self, mock_deployments,
|
|
mock_default, mock_struct,
|
|
mock_print_list):
|
|
current_deploy_id = '26a3ce76-0efa-40e4-86e5-514574bd1ff6'
|
|
mock_default.return_value = current_deploy_id
|
|
fake_deployment_list = [
|
|
{'uuid': 'fa34aea2-ae2e-4cf7-a072-b08d67466e3e',
|
|
'created_at': '03-12-2014',
|
|
'name': 'dep1',
|
|
'status': 'deploy->started',
|
|
'active': 'False'}]
|
|
|
|
mock_deployments.return_value = fake_deployment_list
|
|
self.deployment.list()
|
|
|
|
fake_deployment = fake_deployment_list[0]
|
|
fake_deployment['active'] = ''
|
|
mock_struct.assert_called_once_with(**fake_deployment)
|
|
|
|
headers = ['uuid', 'created_at', 'name', 'status', 'active']
|
|
mock_print_list.assert_called_once_with([mock_struct()], headers,
|
|
sortby_index=headers.index(
|
|
'created_at'))
|
|
|
|
@mock.patch('rally.cmd.commands.deployment.common_cliutils.print_list')
|
|
@mock.patch('rally.cmd.commands.deployment.utils.Struct')
|
|
@mock.patch('rally.cmd.commands.deployment.envutils.get_global')
|
|
@mock.patch('rally.cmd.commands.deployment.db.deployment_list')
|
|
def test_list_current_deploy_id(self, mock_deployments,
|
|
mock_default, mock_struct,
|
|
mock_print_list):
|
|
current_deploy_id = '64258e84-ffa1-4011-9e4c-aba07bdbcc6b'
|
|
mock_default.return_value = current_deploy_id
|
|
fake_deployment_list = [{'uuid': current_deploy_id,
|
|
'created_at': '13-12-2014',
|
|
'name': 'dep2',
|
|
'status': 'deploy->finished',
|
|
'active': 'True'}]
|
|
mock_deployments.return_value = fake_deployment_list
|
|
self.deployment.list()
|
|
|
|
fake_deployment = fake_deployment_list[0]
|
|
fake_deployment['active'] = '*'
|
|
mock_struct.assert_called_once_with(**fake_deployment)
|
|
|
|
headers = ['uuid', 'created_at', 'name', 'status', 'active']
|
|
mock_print_list.assert_called_once_with([mock_struct()], headers,
|
|
sortby_index=headers.index(
|
|
'created_at'))
|
|
|
|
@mock.patch('rally.cmd.commands.deployment.db.deployment_get')
|
|
@mock.patch('json.dumps')
|
|
def test_config_default(self, mock_json_dumps, mock_deployment):
|
|
deploy_id = 'fa4a423e-f15d-4d83-971a-89574f892999'
|
|
value = {'config': 'config'}
|
|
mock_deployment.return_value = value
|
|
self.deployment.config(deploy_id)
|
|
mock_json_dumps.assert_called_once_with(value['config'])
|
|
mock_deployment.assert_called_once_with(deploy_id)
|
|
|
|
@mock.patch('rally.cmd.commands.deployment.db.deployment_get')
|
|
@mock.patch('json.dumps')
|
|
def test_config_json(self, mock_json_dumps, mock_deployment):
|
|
deploy_id = '25c5f6d3-56ce-4273-834c-1ae5e1c2599c'
|
|
value = {'config': 'config'}
|
|
mock_deployment.return_value = value
|
|
self.deployment.config(deploy_id, output_json=True)
|
|
mock_json_dumps.assert_called_once_with(value['config'])
|
|
mock_deployment.assert_called_once_with(deploy_id)
|
|
|
|
@mock.patch('rally.cmd.commands.deployment.db.deployment_get')
|
|
@mock.patch('pprint.pprint')
|
|
def test_config_pprint(self, mock_pprint, mock_deployment):
|
|
deploy_id = '840a0144-4634-46fd-8cf8-b84caa0dba67'
|
|
value = {'config': 'config'}
|
|
mock_deployment.return_value = value
|
|
self.deployment.config(deploy_id, output_pprint=True)
|
|
mock_pprint.assert_called_once_with(value['config'])
|
|
mock_deployment.assert_called_once_with(deploy_id)
|
|
|
|
@mock.patch('rally.cmd.commands.deployment.envutils.get_global')
|
|
def test_config_no_deploy_id(self, mock_default):
|
|
mock_default.side_effect = exceptions.InvalidArgumentsException
|
|
self.assertRaises(exceptions.InvalidArgumentsException,
|
|
self.deployment.config, None)
|
|
|
|
@mock.patch('rally.cmd.commands.deployment.common_cliutils.print_list')
|
|
@mock.patch('rally.cmd.commands.deployment.utils.Struct')
|
|
@mock.patch('rally.cmd.commands.deployment.db.deployment_get')
|
|
def test_endpoint(self, mock_deployment, mock_struct, mock_print_list):
|
|
deploy_id = 'b1a6153e-a314-4cb3-b63b-cf08c1a416c3'
|
|
value = {'endpoints': [{}]}
|
|
mock_deployment.return_value = value
|
|
self.deployment.endpoint(deploy_id)
|
|
mock_deployment.assert_called_once_with(deploy_id)
|
|
|
|
headers = ['auth_url', 'username', 'password', 'tenant_name',
|
|
'region_name', 'use_public_urls', 'admin_port']
|
|
fake_data = ['', '', '', '', '', '', '']
|
|
mock_struct.assert_called_once_with(**dict(zip(headers, fake_data)))
|
|
mock_print_list.assert_called_once_with([mock_struct()], headers)
|
|
|
|
@mock.patch('rally.cmd.commands.deployment.envutils.get_global')
|
|
def test_deploy_no_deploy_id(self, mock_default):
|
|
mock_default.side_effect = exceptions.InvalidArgumentsException
|
|
self.assertRaises(exceptions.InvalidArgumentsException,
|
|
self.deployment.endpoint, None)
|