diff options
Diffstat (limited to 'designate/tests/test_manage')
-rw-r--r-- | designate/tests/test_manage/__init__.py | 5 | ||||
-rw-r--r-- | designate/tests/test_manage/test_update_pool.py | 65 |
2 files changed, 70 insertions, 0 deletions
diff --git a/designate/tests/test_manage/__init__.py b/designate/tests/test_manage/__init__.py new file mode 100644 index 00000000..dd671f43 --- /dev/null +++ b/designate/tests/test_manage/__init__.py @@ -0,0 +1,5 @@ +from designate.tests import TestCase + + +class DesignateManageTestCase(TestCase): + pass diff --git a/designate/tests/test_manage/test_update_pool.py b/designate/tests/test_manage/test_update_pool.py new file mode 100644 index 00000000..d5b21fbc --- /dev/null +++ b/designate/tests/test_manage/test_update_pool.py @@ -0,0 +1,65 @@ +# 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 oslo_log import log as logging + +import mock +from designate.tests.test_manage import DesignateManageTestCase +from designate.manage.pool import PoolCommands +from designate.tests import fixtures +from designate import objects + +LOG = logging.getLogger(__name__) + + +class UpdatePoolTestCase(DesignateManageTestCase): + def setUp(self): + super(DesignateManageTestCase, self).setUp() + self.stdlog = fixtures.StandardLogging() + self.useFixture(self.stdlog) + + def hydrate_pool_targets(self, targets): + pool_targets = objects.PoolTargetList() + masters = objects.PoolTargetMasterList() + for target in targets: + masters.append(target) + target = objects.PoolTarget(masters=masters) + target.masters = masters + pool_targets.append(target) + return pool_targets + + def test_update_pools_zones(self): + values = dict( + name='example.com.', + email='info@example.com', + type='PRIMARY' + ) + + zone = self.central_service.create_zone( + self.admin_context, zone=objects.Zone.from_dict(values)) + + # Ensure the correct NS Records are in place + pool = self.central_service.get_pool( + self.admin_context, zone.pool_id) + + pool.targets = self.hydrate_pool_targets([objects.PoolTargetMaster( + pool_target_id=pool.id, + host="127.0.0.1", + port="53")]) + + command = PoolCommands() + command.context = self.admin_context + command.central_api = self.central_service + + with mock.patch.object(self.central_service, + "update_zone") as mock_update_zone: + command._update_zones(pool) + mock_update_zone.assert_called_once() |