118 lines
4.2 KiB
Python
118 lines
4.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright 2015 Mirantis, 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.
|
|
|
|
|
|
from nailgun import consts
|
|
from nailgun import errors
|
|
from nailgun.policy.merge import NetworkRoleMergePolicy
|
|
from nailgun.policy.name_match import ExactMatchingPolicy
|
|
from nailgun.policy.name_match import NameMatchingPolicy
|
|
from nailgun.policy.name_match import PatternMatchingPolicy
|
|
from nailgun.test.base import BaseUnitTest
|
|
|
|
|
|
class TestNetworkRoleMergePolicy(BaseUnitTest):
|
|
def setUp(self):
|
|
super(TestNetworkRoleMergePolicy, self).setUp()
|
|
self.policy = NetworkRoleMergePolicy()
|
|
|
|
@staticmethod
|
|
def _make_plugin_network_role(**kwargs):
|
|
properties = {
|
|
'subnet': True,
|
|
'gateway': False,
|
|
'vip': []
|
|
}
|
|
|
|
properties.update(kwargs)
|
|
|
|
return {
|
|
'id': 'test_network_role',
|
|
'default_mapping': consts.NETWORKS.public,
|
|
'properties': properties
|
|
}
|
|
|
|
def test_apply_path(self):
|
|
target = self._make_plugin_network_role(vip=[{'name': 'test_vip_a'}])
|
|
patch = self._make_plugin_network_role(vip=[{'name': 'test_vip_b'}])
|
|
self.policy.apply_patch(target, patch)
|
|
expected = self._make_plugin_network_role(
|
|
vip=[{'name': 'test_vip_a'}, {'name': 'test_vip_b'}]
|
|
)
|
|
|
|
self.assertDictEqual(expected, target)
|
|
|
|
def test_apply_patch_vips_without_duplicates(self):
|
|
target = self._make_plugin_network_role(
|
|
vip=[{'name': 'test_vip_a'}, {'name': 'test_vip_b'}]
|
|
)
|
|
patch = self._make_plugin_network_role(vip=[{'name': 'test_vip_a'}])
|
|
self.policy.apply_patch(target, patch)
|
|
self.assertItemsEqual(
|
|
[{'name': 'test_vip_a'}, {'name': 'test_vip_b'}],
|
|
target['properties']['vip']
|
|
)
|
|
|
|
def test_apply_patch_vips_reset(self):
|
|
target = self._make_plugin_network_role(
|
|
vip=[{'name': 'test_vip_a'}, {'name': 'test_vip_b'}]
|
|
)
|
|
|
|
patch = self._make_plugin_network_role(vip=[])
|
|
self.policy.apply_patch(target, patch)
|
|
|
|
self.assertItemsEqual([], target['properties']['vip'])
|
|
|
|
def test_apply_patch_fail_if_conflict(self):
|
|
with self.assertRaisesRegexp(errors.UnresolvableConflict, 'subnet'):
|
|
self.policy.apply_patch(
|
|
self._make_plugin_network_role(subnet=True),
|
|
self._make_plugin_network_role(subnet=False)
|
|
)
|
|
|
|
with self.assertRaisesRegexp(errors.UnresolvableConflict, 'prop1'):
|
|
self.policy.apply_patch(
|
|
self._make_plugin_network_role(prop1=0.1),
|
|
self._make_plugin_network_role(prop1=1)
|
|
)
|
|
|
|
def test_apply_path_fail_if_vip_conflict(self):
|
|
with self.assertRaisesRegexp(
|
|
errors.UnresolvableConflict,
|
|
"VIP 'test' conflicts with existing one"):
|
|
self.policy.apply_patch(
|
|
self._make_plugin_network_role(
|
|
vip=[{"name": "test", "value": 1}]
|
|
),
|
|
self._make_plugin_network_role(
|
|
vip=[{"name": "test", "value": 2}]
|
|
)
|
|
)
|
|
|
|
|
|
class TestNameMatchingPolicy(BaseUnitTest):
|
|
def test_exact_match(self):
|
|
match_policy = NameMatchingPolicy.create("controller")
|
|
self.assertIsInstance(match_policy, ExactMatchingPolicy)
|
|
self.assertTrue(match_policy.match("controller"))
|
|
self.assertFalse(match_policy.match("controller1"))
|
|
|
|
def test_pattern_match(self):
|
|
match_policy = NameMatchingPolicy.create("/controller/")
|
|
self.assertIsInstance(match_policy, PatternMatchingPolicy)
|
|
self.assertTrue(match_policy.match("controller"))
|
|
self.assertTrue(match_policy.match("controller1"))
|