387 lines
14 KiB
Python
387 lines
14 KiB
Python
# Copyright 2015 Huawei Technologies Co.,LTD.
|
|
#
|
|
# 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 importlib
|
|
from unittest import mock
|
|
|
|
from magnum.api import validation as v
|
|
from magnum.common import exception
|
|
import magnum.conf
|
|
from magnum import objects
|
|
from magnum.tests import base
|
|
from magnum.tests.unit.objects import utils as obj_utils
|
|
|
|
CONF = magnum.conf.CONF
|
|
|
|
|
|
class TestValidation(base.BaseTestCase):
|
|
|
|
def _test_enforce_cluster_type_supported(
|
|
self, mock_cluster_template_get, mock_cluster_get_by_uuid,
|
|
mock_pecan_request, cluster_type, assert_raised=False):
|
|
|
|
@v.enforce_cluster_type_supported()
|
|
def test(self, cluster):
|
|
pass
|
|
|
|
server_type, cluster_distro, coe = cluster_type
|
|
cluster_template = obj_utils.get_test_cluster_template(
|
|
mock_pecan_request.context, uuid='cluster_template_id',
|
|
coe=coe, cluster_distro=cluster_distro, server_type=server_type)
|
|
mock_cluster_template_get.return_value = cluster_template
|
|
|
|
cluster = mock.MagicMock()
|
|
cluster.cluster_template_id = 'cluster_template_id'
|
|
cluster.cluster_template = cluster_template
|
|
mock_cluster_get_by_uuid.return_value = cluster
|
|
|
|
if assert_raised:
|
|
return self.assertRaises(
|
|
exception.ClusterTypeNotSupported, test, self, cluster)
|
|
else:
|
|
self.assertIsNone(test(self, cluster))
|
|
|
|
@mock.patch('pecan.request')
|
|
@mock.patch('magnum.objects.Cluster.get_by_uuid')
|
|
@mock.patch('magnum.objects.ClusterTemplate.get')
|
|
def test_enforce_cluster_type_supported(
|
|
self, mock_cluster_template_get, mock_cluster_get_by_uuid,
|
|
mock_pecan_request):
|
|
|
|
cluster_type = ('vm', 'fedora-coreos', 'kubernetes')
|
|
self._test_enforce_cluster_type_supported(
|
|
mock_cluster_template_get, mock_cluster_get_by_uuid,
|
|
mock_pecan_request, cluster_type)
|
|
|
|
@mock.patch('pecan.request')
|
|
@mock.patch('magnum.objects.Cluster.get_by_uuid')
|
|
@mock.patch('magnum.objects.ClusterTemplate.get')
|
|
def test_enforce_cluster_type_not_supported(
|
|
self, mock_cluster_template_get, mock_cluster_get_by_uuid,
|
|
mock_pecan_request):
|
|
|
|
cluster_type = ('vm', 'foo', 'kubernetes')
|
|
exc = self._test_enforce_cluster_type_supported(
|
|
mock_cluster_template_get, mock_cluster_get_by_uuid,
|
|
mock_pecan_request, cluster_type, assert_raised=True)
|
|
self.assertEqual('Cluster type (vm, foo, kubernetes) not supported.',
|
|
exc.message)
|
|
|
|
def _test_enforce_network_driver_types_create(
|
|
self,
|
|
network_driver_type,
|
|
validator_allowed_network_drivers=None,
|
|
validator_default_network_driver=None,
|
|
coe="kubernetes",
|
|
assert_raised=False,
|
|
):
|
|
@v.enforce_network_driver_types_create()
|
|
def test(self, cluster_template):
|
|
pass
|
|
|
|
cluster_template = mock.MagicMock()
|
|
cluster_template.name = 'test_cluster_template'
|
|
cluster_template.network_driver = network_driver_type
|
|
cluster_template.coe = coe
|
|
|
|
# NOTE(dalees): Patch the validator class variables directly, so the
|
|
# changes are removed after the test.
|
|
with mock.patch.multiple(
|
|
v.K8sValidator,
|
|
supported_network_drivers=["flannel", "type1", "type2"],
|
|
allowed_network_drivers=validator_allowed_network_drivers
|
|
or v.K8sValidator.allowed_network_drivers,
|
|
default_network_driver=validator_default_network_driver
|
|
or v.K8sValidator.default_network_driver,
|
|
):
|
|
if assert_raised:
|
|
self.assertRaises(
|
|
exception.InvalidParameterValue,
|
|
test,
|
|
self,
|
|
cluster_template,
|
|
)
|
|
else:
|
|
test(self, cluster_template)
|
|
return cluster_template
|
|
|
|
def test_enforce_network_driver_types_one_allowed_create(self):
|
|
self._test_enforce_network_driver_types_create(
|
|
network_driver_type="type1",
|
|
validator_allowed_network_drivers=["type1"],
|
|
)
|
|
|
|
def test_enforce_network_driver_types_two_allowed_create(self):
|
|
self._test_enforce_network_driver_types_create(
|
|
network_driver_type="type1",
|
|
validator_allowed_network_drivers=["type1", "type2"],
|
|
)
|
|
|
|
def test_enforce_network_driver_types_not_allowed_create(self):
|
|
self._test_enforce_network_driver_types_create(
|
|
network_driver_type="type1",
|
|
validator_allowed_network_drivers=["type2"],
|
|
assert_raised=True,
|
|
)
|
|
|
|
def test_enforce_network_driver_types_all_allowed_create(self):
|
|
for driver in ['flannel', 'type1', 'type2']:
|
|
self._test_enforce_network_driver_types_create(
|
|
network_driver_type=driver,
|
|
validator_allowed_network_drivers=["all"],
|
|
)
|
|
|
|
def test_enforce_network_driver_types_invalid_coe_create(self):
|
|
self._test_enforce_network_driver_types_create(
|
|
network_driver_type="flannel",
|
|
coe="invalid_coe_type",
|
|
assert_raised=True,
|
|
)
|
|
|
|
def test_enforce_network_driver_types_default_create(self):
|
|
cluster_template = self._test_enforce_network_driver_types_create(
|
|
network_driver_type=None
|
|
)
|
|
self.assertEqual("flannel", cluster_template.network_driver)
|
|
|
|
def test_enforce_network_driver_types_default_config_create(self):
|
|
cluster_template = self._test_enforce_network_driver_types_create(
|
|
network_driver_type=None,
|
|
validator_allowed_network_drivers=["type1"],
|
|
validator_default_network_driver="type1",
|
|
)
|
|
self.assertEqual("type1", cluster_template.network_driver)
|
|
|
|
def test_enforce_network_driver_types_default_invalid_create(self):
|
|
self._test_enforce_network_driver_types_create(
|
|
network_driver_type=None,
|
|
validator_default_network_driver="invalid_driver",
|
|
assert_raised=True,
|
|
)
|
|
|
|
@mock.patch('pecan.request')
|
|
@mock.patch('magnum.api.utils.get_resource')
|
|
def _test_enforce_network_driver_types_update(
|
|
self,
|
|
mock_get_resource,
|
|
mock_pecan_request,
|
|
network_driver_type,
|
|
validator_allowed_network_drivers=None,
|
|
assert_raised=False,
|
|
):
|
|
@v.enforce_network_driver_types_update()
|
|
def test(self, cluster_template_ident, patch):
|
|
pass
|
|
|
|
cluster_template_ident = "test_uuid_or_name"
|
|
|
|
patch = [{'path': '/network_driver', 'value': network_driver_type,
|
|
'op': 'replace'}]
|
|
context = mock_pecan_request.context
|
|
cluster_template = obj_utils.get_test_cluster_template(
|
|
context, uuid=cluster_template_ident, coe='kubernetes')
|
|
cluster_template.network_driver = network_driver_type
|
|
mock_get_resource.return_value = cluster_template
|
|
|
|
# NOTE(dalees): Patch the validator class variables directly, so the
|
|
# changes are removed after the test.
|
|
with mock.patch.multiple(
|
|
v.K8sValidator,
|
|
supported_network_drivers=["flannel", "type1", "type2"],
|
|
allowed_network_drivers=validator_allowed_network_drivers
|
|
or v.K8sValidator.allowed_network_drivers,
|
|
):
|
|
if assert_raised:
|
|
self.assertRaises(
|
|
exception.InvalidParameterValue,
|
|
test,
|
|
self,
|
|
cluster_template_ident,
|
|
patch,
|
|
)
|
|
else:
|
|
test(self, cluster_template_ident, patch)
|
|
mock_get_resource.assert_called_once_with(
|
|
"ClusterTemplate", cluster_template_ident
|
|
)
|
|
|
|
def test_enforce_network_driver_types_one_allowed_update(self):
|
|
self._test_enforce_network_driver_types_update(
|
|
network_driver_type="type1",
|
|
validator_allowed_network_drivers=["type1"],
|
|
)
|
|
|
|
def test_enforce_network_driver_types_two_allowed_update(self):
|
|
self._test_enforce_network_driver_types_update(
|
|
network_driver_type="type1",
|
|
validator_allowed_network_drivers=["type1", "type2"],
|
|
)
|
|
|
|
def test_enforce_network_driver_types_not_allowed_update(self):
|
|
self._test_enforce_network_driver_types_update(
|
|
network_driver_type="type1",
|
|
validator_allowed_network_drivers=["type2"],
|
|
assert_raised=True,
|
|
)
|
|
|
|
def test_enforce_network_driver_types_all_allowed_update(self):
|
|
for driver in ['flannel', 'type1', 'type2']:
|
|
self._test_enforce_network_driver_types_update(
|
|
network_driver_type=driver,
|
|
validator_allowed_network_drivers=["all"],
|
|
)
|
|
|
|
def _test_enforce_volume_driver_types_create(
|
|
self,
|
|
volume_driver_type,
|
|
coe='kubernetes',
|
|
assert_raised=False):
|
|
|
|
@v.enforce_volume_driver_types_create()
|
|
def test(self, cluster_template):
|
|
pass
|
|
|
|
cluster_template = obj_utils.get_test_cluster_template(
|
|
{}, name='test_cluster_template', coe=coe,
|
|
volume_driver=volume_driver_type)
|
|
|
|
if assert_raised:
|
|
self.assertRaises(exception.InvalidParameterValue,
|
|
test, self, cluster_template)
|
|
else:
|
|
test(self, cluster_template)
|
|
|
|
def test_enforce_volume_driver_types_valid_create(self):
|
|
self._test_enforce_volume_driver_types_create(
|
|
volume_driver_type='cinder')
|
|
|
|
def test_enforce_volume_driver_types_invalid_create(self):
|
|
self._test_enforce_volume_driver_types_create(
|
|
volume_driver_type='type',
|
|
assert_raised=True)
|
|
|
|
def _test_enforce_server_type(
|
|
self,
|
|
server_type,
|
|
coe='kubernetes',
|
|
assert_raised=False):
|
|
|
|
@v.enforce_server_type()
|
|
def test(self, cluster_template):
|
|
pass
|
|
|
|
cluster_template = obj_utils.get_test_cluster_template(
|
|
{}, name='test_cluster_template', coe=coe,
|
|
server_type=server_type)
|
|
|
|
if assert_raised:
|
|
self.assertRaises(exception.InvalidParameterValue,
|
|
test, self, cluster_template)
|
|
else:
|
|
test(self, cluster_template)
|
|
|
|
def test_enforce_server_type_valid_vm(self):
|
|
self._test_enforce_server_type(
|
|
server_type='vm')
|
|
|
|
def test_enforce_server_type_valid_bm(self):
|
|
self._test_enforce_server_type(
|
|
server_type='bm')
|
|
|
|
def test_enforce_server_type_invalid(self):
|
|
self._test_enforce_server_type(
|
|
server_type='invalid',
|
|
assert_raised=True)
|
|
|
|
@mock.patch('pecan.request')
|
|
@mock.patch('magnum.api.utils.get_resource')
|
|
def _test_enforce_volume_driver_types_update(
|
|
self,
|
|
mock_get_resource,
|
|
mock_pecan_request,
|
|
volume_driver_type,
|
|
op,
|
|
assert_raised=False):
|
|
|
|
@v.enforce_volume_driver_types_update()
|
|
def test(self, cluster_template_ident, patch):
|
|
pass
|
|
|
|
cluster_template_ident = 'test_uuid_or_name'
|
|
patch = [{'path': '/volume_driver', 'value': volume_driver_type,
|
|
'op': op}]
|
|
context = mock_pecan_request.context
|
|
cluster_template = obj_utils.get_test_cluster_template(
|
|
context, uuid=cluster_template_ident, coe='kubernetes')
|
|
mock_get_resource.return_value = cluster_template
|
|
|
|
# Reload the validator module so that ClusterTemplate configs are
|
|
# re-evaluated.
|
|
importlib.reload(v)
|
|
validator = v.K8sValidator
|
|
|
|
with mock.patch.multiple(
|
|
validator, supported_volume_driver=["cinder"]
|
|
):
|
|
if assert_raised:
|
|
self.assertRaises(
|
|
exception.InvalidParameterValue,
|
|
test,
|
|
self,
|
|
cluster_template_ident,
|
|
patch,
|
|
)
|
|
else:
|
|
test(self, cluster_template_ident, patch)
|
|
mock_get_resource.assert_called_once_with(
|
|
"ClusterTemplate", cluster_template_ident
|
|
)
|
|
|
|
def test_enforce_volume_driver_types_supported_replace_update(self):
|
|
self._test_enforce_volume_driver_types_update(
|
|
volume_driver_type='cinder',
|
|
op='replace')
|
|
|
|
def test_enforce_volume_driver_types_not_supported_replace_update(self):
|
|
self._test_enforce_volume_driver_types_update(
|
|
volume_driver_type='type1',
|
|
op='replace',
|
|
assert_raised=True)
|
|
|
|
def test_enforce_volume_driver_types_supported_add_update(self):
|
|
self._test_enforce_volume_driver_types_update(
|
|
volume_driver_type='cinder',
|
|
op='add')
|
|
|
|
def test_enforce_volume_driver_types_not_supported_add_update(self):
|
|
self._test_enforce_volume_driver_types_update(
|
|
volume_driver_type='type1',
|
|
op='add',
|
|
assert_raised=True)
|
|
|
|
def test_enforce_volume_driver_types_remove_update(self):
|
|
self._test_enforce_volume_driver_types_update(
|
|
volume_driver_type='cinder',
|
|
op='remove')
|
|
|
|
def test_validate_cluster_properties(self):
|
|
allowed_properties = v.cluster_update_allowed_properties
|
|
for field in objects.Cluster.fields:
|
|
if field in allowed_properties:
|
|
v.validate_cluster_properties(set([field]))
|
|
else:
|
|
self.assertRaises(exception.InvalidParameterValue,
|
|
v.validate_cluster_properties, set([field]))
|