b74ea2e4f8
This adds a new group of commands : 'rally use' the first implemented is 'rally use deployment <deploy_id>' As it is not possible in python to export globaly an env var,it saves the variable in a file at ~/.rally/deployment like : RALLY_DEPLOYMENT=1234abcd For each command that takes a deploy_id as parameter, this parameter is now optional. The algorithm to find the deploy_id is as follows : - If the --deploy-id param is set, use it - If the environ contains the key 'RALLY_DEPLOYMENT', use it - load the content of ~/.rally/deployment and if it contains the key RALLY_DEPLOYMENT, use it - Else, raise an InvalidArgumentException Note that I have not been able to use the default arg in @cliutils.args because it is called before the effective method call. Setting deploy_id=envutils.default_deployment_id() has the same problem Partially implements blueprint: rally-use-command Change-Id: I1e62e26d259bd9e81f1c008058ee9ce91cc3290c
201 lines
8.2 KiB
Python
201 lines
8.2 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 mock
|
|
import os
|
|
import uuid
|
|
|
|
from rally.cmd import main
|
|
from rally import exceptions
|
|
from rally.openstack.common import test
|
|
|
|
|
|
class TaskCommandsTestCase(test.BaseTestCase):
|
|
|
|
def setUp(self):
|
|
super(TaskCommandsTestCase, self).setUp()
|
|
self.task = main.TaskCommands()
|
|
|
|
@mock.patch('rally.cmd.main.TaskCommands.detailed')
|
|
@mock.patch('rally.orchestrator.api.create_task',
|
|
return_value=dict(uuid='fc1a9bbe-1ead-4740-92b5-0feecf421634',
|
|
created_at='2014-01-14 09:14:45.395822',
|
|
status='init',
|
|
failed=False))
|
|
@mock.patch('rally.cmd.main.api.start_task')
|
|
@mock.patch('rally.cmd.main.open',
|
|
mock.mock_open(read_data='{"some": "json"}'),
|
|
create=True)
|
|
def test_start(self, mock_api, mock_create_task,
|
|
mock_task_detailed):
|
|
deploy_id = str(uuid.uuid4())
|
|
self.task.start('path_to_config.json', deploy_id,)
|
|
mock_api.assert_called_once_with(deploy_id, {u'some': u'json'},
|
|
task=mock_create_task.return_value)
|
|
|
|
@mock.patch('rally.cmd.main.envutils.default_deployment_id')
|
|
def test_start_no_deploy_id(self, mock_default):
|
|
mock_default.side_effect = exceptions.InvalidArgumentsException
|
|
self.assertRaises(exceptions.InvalidArgumentsException,
|
|
self.task.start, 'path_to_config.json', None)
|
|
|
|
def test_abort(self):
|
|
test_uuid = str(uuid.uuid4())
|
|
with mock.patch("rally.cmd.main.api") as mock_api:
|
|
mock_api.abort_task = mock.MagicMock()
|
|
self.task.abort(test_uuid)
|
|
main.api.abort_task.assert_called_once_with(test_uuid)
|
|
|
|
def test_status(self):
|
|
test_uuid = str(uuid.uuid4())
|
|
value = {'task_id': "task", "status": "status"}
|
|
with mock.patch("rally.cmd.main.db") as mock_db:
|
|
mock_db.task_get = mock.MagicMock(return_value=value)
|
|
self.task.status(test_uuid)
|
|
mock_db.task_get.assert_called_once_with(test_uuid)
|
|
|
|
@mock.patch('rally.cmd.main.db')
|
|
def test_detailed(self, mock_db):
|
|
test_uuid = str(uuid.uuid4())
|
|
value = {'task_id': "task", "status": "status", "results": []}
|
|
mock_db.task_get_detailed = mock.MagicMock(return_value=value)
|
|
self.task.detailed(test_uuid)
|
|
mock_db.task_get_detailed.assert_called_once_with(test_uuid)
|
|
|
|
@mock.patch('rally.cmd.main.db')
|
|
def test_detailed_wrong_id(self, mock_db):
|
|
test_uuid = str(uuid.uuid4())
|
|
mock_db.task_get_detailed = mock.MagicMock(return_value=None)
|
|
self.task.detailed(test_uuid)
|
|
mock_db.task_get_detailed.assert_called_once_with(test_uuid)
|
|
|
|
@mock.patch('rally.cmd.main.db')
|
|
def test_results(self, mock_db):
|
|
test_uuid = str(uuid.uuid4())
|
|
value = [
|
|
{'key': 'key', 'data': {'raw': 'raw'}}
|
|
]
|
|
mock_db.task_result_get_all_by_uuid.return_value = value
|
|
self.task.results(test_uuid)
|
|
mock_db.task_result_get_all_by_uuid.assert_called_once_with(test_uuid)
|
|
|
|
def test_list(self):
|
|
db_response = [
|
|
{'uuid': 'a', 'created_at': 'b', 'status': 'c', 'failed': True}
|
|
]
|
|
with mock.patch("rally.cmd.main.db") as mock_db:
|
|
mock_db.task_list = mock.MagicMock(return_value=db_response)
|
|
self.task.list()
|
|
mock_db.task_list.assert_called_once_with()
|
|
|
|
def test_delete(self):
|
|
task_uuid = str(uuid.uuid4())
|
|
force = False
|
|
with mock.patch("rally.cmd.main.api") as mock_api:
|
|
mock_api.delete_task = mock.Mock()
|
|
self.task.delete(task_uuid, force)
|
|
mock_api.delete_task.assert_called_once_with(task_uuid,
|
|
force=force)
|
|
|
|
def test_plot(self):
|
|
test_uuid = str(uuid.uuid4())
|
|
mock_plot = mock.Mock()
|
|
PLOTS = {"aggregated": mock_plot}
|
|
with mock.patch("rally.cmd.main.processing.PLOTS", new=PLOTS):
|
|
self.task.plot("aggregated", "concurrent", test_uuid)
|
|
mock_plot.assert_called_once_with(test_uuid, "concurrent")
|
|
|
|
|
|
class DeploymentCommandsTestCase(test.BaseTestCase):
|
|
def setUp(self):
|
|
super(DeploymentCommandsTestCase, self).setUp()
|
|
self.deployment = main.DeploymentCommands()
|
|
|
|
@mock.patch('rally.cmd.main.api.create_deploy')
|
|
@mock.patch('rally.cmd.main.open',
|
|
mock.mock_open(read_data='{"some": "json"}'),
|
|
create=True)
|
|
def test_create(self, mock_create):
|
|
self.deployment.create('path_to_config.json', 'fake_deploy')
|
|
mock_create.assert_called_once_with({'some': 'json'}, 'fake_deploy')
|
|
|
|
@mock.patch('rally.cmd.main.api.recreate_deploy')
|
|
def test_recreate(self, mock_recreate):
|
|
deploy_id = str(uuid.uuid4())
|
|
self.deployment.recreate(deploy_id)
|
|
mock_recreate.assert_called_once_with(deploy_id)
|
|
|
|
@mock.patch('rally.cmd.main.envutils.default_deployment_id')
|
|
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.main.api.destroy_deploy')
|
|
def test_destroy(self, mock_destroy):
|
|
deploy_id = str(uuid.uuid4())
|
|
self.deployment.destroy(deploy_id)
|
|
mock_destroy.assert_called_once_with(deploy_id)
|
|
|
|
@mock.patch('rally.cmd.main.envutils.default_deployment_id')
|
|
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.main.db.deployment_get')
|
|
def test_config(self, mock_deployment):
|
|
deploy_id = str(uuid.uuid4())
|
|
value = {'config': 'config'}
|
|
mock_deployment.return_value = value
|
|
self.deployment.config(deploy_id)
|
|
mock_deployment.assert_called_once_with(deploy_id)
|
|
|
|
@mock.patch('rally.cmd.main.envutils.default_deployment_id')
|
|
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.main.db.deployment_get')
|
|
def test_endpoint(self, mock_deployment):
|
|
deploy_id = str(uuid.uuid4())
|
|
value = {'endpoint': {}}
|
|
mock_deployment.return_value = value
|
|
self.deployment.endpoint(deploy_id)
|
|
mock_deployment.assert_called_once_with(deploy_id)
|
|
|
|
@mock.patch('rally.cmd.main.envutils.default_deployment_id')
|
|
def test_deploy_no_deploy_id(self, mock_default):
|
|
mock_default.side_effect = exceptions.InvalidArgumentsException
|
|
self.assertRaises(exceptions.InvalidArgumentsException,
|
|
self.deployment.endpoint, None)
|
|
|
|
|
|
class UseCommandsTestCase(test.BaseTestCase):
|
|
def setUp(self):
|
|
super(UseCommandsTestCase, self).setUp()
|
|
self.use = main.UseCommands()
|
|
|
|
@mock.patch('os.path.exists')
|
|
@mock.patch('rally.cmd.main.fileutils.update_env_file')
|
|
def test_deployment(self, mock_file, mock_path):
|
|
deploy_id = str(uuid.uuid4())
|
|
mock_path.return_value = True
|
|
self.use.deployment(deploy_id)
|
|
mock_path.assert_called_once_with(os.path.expanduser('~/.rally/'))
|
|
mock_file.assert_called_once_with(os.path.expanduser(
|
|
'~/.rally/deployment'), 'RALLY_DEPLOYMENT', deploy_id)
|