summaryrefslogtreecommitdiff
path: root/designate/tests/test_manage
diff options
context:
space:
mode:
Diffstat (limited to 'designate/tests/test_manage')
-rw-r--r--designate/tests/test_manage/__init__.py5
-rw-r--r--designate/tests/test_manage/test_update_pool.py65
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()