192 lines
7.7 KiB
Python
192 lines
7.7 KiB
Python
# Copyright 2017 Red Hat, Inc.
|
|
#
|
|
# 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 tripleo_common.exception import NotFound
|
|
from tripleoclient.tests.v1.overcloud_deploy import fakes
|
|
from tripleoclient.v1 import overcloud_roles
|
|
|
|
|
|
class TestOvercloudRolesListAvailable(fakes.TestDeployOvercloud):
|
|
|
|
def setUp(self):
|
|
super(TestOvercloudRolesListAvailable, self).setUp()
|
|
self.cmd = overcloud_roles.RoleList(self.app, None)
|
|
|
|
@mock.patch('tripleo_common.utils.roles.get_roles_list_from_directory')
|
|
@mock.patch('os.path.realpath')
|
|
def test_action(self, realpath_mock, get_roles_mock):
|
|
realpath_mock.return_value = '/foo'
|
|
get_roles_mock.return_value = ['a', 'b']
|
|
self.cmd._get_roles = get_roles_mock
|
|
|
|
arglist = []
|
|
verifylist = [
|
|
('roles_path', '/usr/share/openstack-tripleo-heat-templates/roles')
|
|
]
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
|
|
self.cmd.take_action(parsed_args)
|
|
get_roles_mock.assert_called_once_with('/foo')
|
|
|
|
@mock.patch('tripleo_common.utils.roles.get_roles_list_from_directory')
|
|
@mock.patch('os.path.realpath')
|
|
def test_action_role_path(self, realpath_mock, get_roles_mock):
|
|
realpath_mock.return_value = '/tmp'
|
|
get_roles_mock.return_value = ['a', 'b']
|
|
self.cmd._get_roles = get_roles_mock
|
|
|
|
arglist = ['--roles-path', '/tmp']
|
|
verifylist = [
|
|
('roles_path', '/tmp')
|
|
]
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
|
|
self.cmd.take_action(parsed_args)
|
|
get_roles_mock.assert_called_once_with('/tmp')
|
|
|
|
|
|
class TestOvercloudRolesGenerateData(fakes.TestDeployOvercloud):
|
|
|
|
def setUp(self):
|
|
super(TestOvercloudRolesGenerateData, self).setUp()
|
|
self.cmd = overcloud_roles.RolesGenerate(self.app, None)
|
|
|
|
@mock.patch(
|
|
'tripleo_common.utils.roles.generate_roles_data_from_directory')
|
|
@mock.patch('tripleo_common.utils.roles.check_role_exists')
|
|
@mock.patch('tripleo_common.utils.roles.get_roles_list_from_directory')
|
|
@mock.patch('os.path.realpath')
|
|
def test_action(self, realpath_mock, get_roles_mock, check_mock,
|
|
generate_roles_mock):
|
|
realpath_mock.return_value = '/tmp'
|
|
get_roles_mock.return_value = ['Controller', 'Compute']
|
|
generate_roles_mock.return_value = 'foo'
|
|
capture_mock = mock.MagicMock()
|
|
self.cmd._capture_output = capture_mock
|
|
stop_capture_mock = mock.MagicMock()
|
|
self.cmd._stop_capture_output = stop_capture_mock
|
|
|
|
arglist = ['--roles-path', '/tmp', 'Controller', 'Compute']
|
|
verifylist = [
|
|
('roles_path', '/tmp'),
|
|
('roles', ['Controller', 'Compute'])
|
|
]
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
|
|
self.cmd.take_action(parsed_args)
|
|
get_roles_mock.assert_called_once_with('/tmp')
|
|
check_mock.assert_called_once_with(['Controller', 'Compute'],
|
|
['Controller', 'Compute'])
|
|
capture_mock.assert_called_once_with(None)
|
|
generate_roles_mock.assert_called_once_with('/tmp',
|
|
['Controller', 'Compute'],
|
|
True)
|
|
stop_capture_mock.assert_called_once_with(None)
|
|
|
|
@mock.patch(
|
|
'tripleo_common.utils.roles.generate_roles_data_from_directory')
|
|
@mock.patch('tripleo_common.utils.roles.check_role_exists')
|
|
@mock.patch('tripleo_common.utils.roles.get_roles_list_from_directory')
|
|
@mock.patch('os.path.realpath')
|
|
def test_action_with_outputfile(self, realpath_mock, get_roles_mock,
|
|
check_mock, generate_roles_mock):
|
|
realpath_mock.return_value = '/tmp'
|
|
get_roles_mock.return_value = ['Controller', 'Compute']
|
|
generate_roles_mock.return_value = 'foo'
|
|
capture_mock = mock.MagicMock()
|
|
self.cmd._capture_output = capture_mock
|
|
stop_capture_mock = mock.MagicMock()
|
|
self.cmd._stop_capture_output = stop_capture_mock
|
|
|
|
arglist = ['--roles-path', '/tmp', '-o', 'foo.yaml',
|
|
'Controller', 'Compute']
|
|
verifylist = [
|
|
('output_file', 'foo.yaml'),
|
|
('roles_path', '/tmp'),
|
|
('roles', ['Controller', 'Compute'])
|
|
]
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
|
|
self.cmd.take_action(parsed_args)
|
|
get_roles_mock.assert_called_once_with('/tmp')
|
|
check_mock.assert_called_once_with(['Controller', 'Compute'],
|
|
['Controller', 'Compute'])
|
|
capture_mock.assert_called_once_with('foo.yaml')
|
|
generate_roles_mock.assert_called_once_with('/tmp',
|
|
['Controller', 'Compute'],
|
|
True)
|
|
stop_capture_mock.assert_called_once_with('foo.yaml')
|
|
|
|
@mock.patch('tripleo_common.utils.roles.get_roles_list_from_directory')
|
|
@mock.patch('os.path.realpath')
|
|
def test_action_with_invald_roles(self, realpath_mock, get_roles_mock):
|
|
realpath_mock.return_value = '/tmp'
|
|
get_roles_mock.return_value = ['Controller', 'Compute']
|
|
|
|
arglist = ['--roles-path', '/tmp', 'Foo', 'Bar']
|
|
verifylist = [
|
|
('roles_path', '/tmp'),
|
|
('roles', ['Foo', 'Bar'])
|
|
]
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
|
|
self.assertRaises(NotFound, self.cmd.take_action, parsed_args)
|
|
get_roles_mock.assert_called_once_with('/tmp')
|
|
|
|
|
|
class TestOvercloudRoleShow(fakes.TestDeployOvercloud):
|
|
|
|
def setUp(self):
|
|
super(TestOvercloudRoleShow, self).setUp()
|
|
self.cmd = overcloud_roles.RoleShow(self.app, None)
|
|
|
|
@mock.patch('yaml.safe_load')
|
|
@mock.patch('tripleoclient.v1.overcloud_roles.open')
|
|
@mock.patch('os.path.realpath')
|
|
def test_action(self, realpath_mock, open_mock, yaml_mock):
|
|
realpath_mock.return_value = '/tmp'
|
|
yaml_mock.return_value = [{'name': 'foo', 'Services': ['a', 'b']}]
|
|
|
|
arglist = ['--roles-path', '/tmp', 'foo']
|
|
verifylist = [
|
|
('roles_path', '/tmp'),
|
|
('role', 'foo')
|
|
]
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
|
|
self.cmd.take_action(parsed_args)
|
|
open_mock.assert_called_once_with('/tmp/foo.yaml', 'r')
|
|
|
|
@mock.patch('tripleo_common.utils.roles.get_roles_list_from_directory')
|
|
@mock.patch('tripleoclient.v1.overcloud_roles.open')
|
|
@mock.patch('os.path.realpath')
|
|
def test_action_invalid_role(self, realpath_mock, open_mock,
|
|
get_roles_mock):
|
|
realpath_mock.return_value = '/tmp'
|
|
open_mock.side_effect = IOError('bar')
|
|
get_roles_mock.return_value = ['Controller', 'Compute']
|
|
|
|
arglist = ['--roles-path', '/tmp', 'foo']
|
|
verifylist = [
|
|
('roles_path', '/tmp'),
|
|
('role', 'foo')
|
|
]
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
|
|
self.assertRaises(NotFound, self.cmd.take_action, parsed_args)
|
|
open_mock.assert_called_once_with('/tmp/foo.yaml', 'r')
|