84 lines
3.2 KiB
Python
84 lines
3.2 KiB
Python
# 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
|
|
|
|
from osc_lib.tests import utils
|
|
|
|
from tripleoclient.v1 import overcloud_config
|
|
|
|
|
|
class TestOvercloudConfig(utils.TestCommand):
|
|
|
|
def setUp(self):
|
|
super(TestOvercloudConfig, self).setUp()
|
|
|
|
self.cmd = overcloud_config.DownloadConfig(self.app, None)
|
|
self.app.client_manager.workflow_engine = mock.Mock()
|
|
self.app.client_manager.orchestration = mock.Mock()
|
|
self.workflow = self.app.client_manager.workflow_engine
|
|
|
|
@mock.patch('tripleoclient.v1.overcloud_config.processutils.execute')
|
|
@mock.patch('tripleoclient.v1.overcloud_config.open')
|
|
@mock.patch('tripleoclient.v1.overcloud_config.request')
|
|
@mock.patch('shutil.rmtree')
|
|
@mock.patch('tripleoclient.workflows.deployment.config_download_export')
|
|
def test_overcloud_download_config(
|
|
self, mock_config, mock_rmtree, mock_request,
|
|
mock_open, mock_execute):
|
|
arglist = ['--name', 'overcloud', '--config-dir', '/tmp']
|
|
verifylist = [
|
|
('name', 'overcloud'),
|
|
('config_dir', '/tmp'),
|
|
('preserve_config_dir', True)
|
|
]
|
|
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
self.cmd.take_action(parsed_args)
|
|
mock_config.assert_called_once_with(
|
|
self.app.client_manager, plan='overcloud', config_type=None)
|
|
mock_rmtree.assert_not_called()
|
|
mock_open.assert_called()
|
|
mock_request.urlopen.assert_called()
|
|
|
|
@mock.patch('os.path.exists')
|
|
@mock.patch('tripleoclient.v1.overcloud_config.processutils.execute')
|
|
@mock.patch('tripleoclient.v1.overcloud_config.open')
|
|
@mock.patch('tripleoclient.v1.overcloud_config.request')
|
|
@mock.patch('shutil.rmtree')
|
|
@mock.patch('tripleoclient.workflows.deployment.config_download_export')
|
|
def test_overcloud_download_config_no_preserve(
|
|
self, mock_config, mock_rmtree, mock_request,
|
|
mock_open, mock_execute, mock_exists):
|
|
arglist = ['--name', 'overcloud', '--config-dir', '/tmp',
|
|
'--no-preserve-config']
|
|
verifylist = [
|
|
('name', 'overcloud'),
|
|
('config_dir', '/tmp'),
|
|
('preserve_config_dir', False)
|
|
]
|
|
|
|
def side_effect(arg):
|
|
if arg == '/tmp/overcloud':
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
mock_exists.side_effect = side_effect
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
self.cmd.take_action(parsed_args)
|
|
mock_config.assert_called_once_with(
|
|
self.app.client_manager, plan='overcloud', config_type=None)
|
|
mock_rmtree.assert_called()
|
|
mock_open.assert_called()
|
|
mock_request.urlopen.assert_called()
|