diff options
Diffstat (limited to 'designateclient/functionaltests')
| -rw-r--r-- | designateclient/functionaltests/client.py | 21 | ||||
| -rw-r--r-- | designateclient/functionaltests/v2/fixtures.py | 22 | ||||
| -rw-r--r-- | designateclient/functionaltests/v2/test_shared_zone.py | 73 |
3 files changed, 115 insertions, 1 deletions
diff --git a/designateclient/functionaltests/client.py b/designateclient/functionaltests/client.py index 4d1d0b0..6edf377 100644 --- a/designateclient/functionaltests/client.py +++ b/designateclient/functionaltests/client.py @@ -348,9 +348,28 @@ class BlacklistCommands(object): return self.parsed_cmd(cmd, FieldValueModel, *args, **kwargs) +class SharedZoneCommands(object): + + def shared_zone_show(self, zone_id, shared_zone_id, *args, **kwargs): + cmd = 'zone share show {0} {1}'.format(zone_id, shared_zone_id) + return self.parsed_cmd(cmd, FieldValueModel, *args, **kwargs) + + def shared_zone_list(self, zone_id, *args, **kwargs): + cmd = 'zone share list {0}'.format(zone_id) + return self.parsed_cmd(cmd, ListModel, *args, **kwargs) + + def share_zone(self, zone_id, target_project_id, *args, **kwargs): + cmd = 'zone share create {0} {1}'.format(zone_id, target_project_id) + return self.parsed_cmd(cmd, FieldValueModel, *args, **kwargs) + + def unshare_zone(self, zone_id, shared_zone_id, *args, **kwargs): + cmd = 'zone share delete {0} {1}'.format(zone_id, shared_zone_id) + return self.parsed_cmd(cmd, FieldValueModel, *args, **kwargs) + + class DesignateCLI(base.CLIClient, ZoneCommands, ZoneTransferCommands, ZoneExportCommands, ZoneImportCommands, RecordsetCommands, - TLDCommands, BlacklistCommands): + TLDCommands, BlacklistCommands, SharedZoneCommands): # instantiate this once to minimize requests to keystone _CLIENTS = None diff --git a/designateclient/functionaltests/v2/fixtures.py b/designateclient/functionaltests/v2/fixtures.py index cc5a83b..c72a18b 100644 --- a/designateclient/functionaltests/v2/fixtures.py +++ b/designateclient/functionaltests/v2/fixtures.py @@ -228,3 +228,25 @@ class BlacklistFixture(BaseFixture): client.zone_blacklist_delete(blacklist_id) except CommandFailed: pass + + +class SharedZoneFixture(BaseFixture): + """See DesignateCLI.recordset_create for __init__ args""" + + def __init__(self, zone, *args, **kwargs): + super(SharedZoneFixture, self).__init__(*args, **kwargs) + self.zone = zone + + def _setUp(self): + super(SharedZoneFixture, self)._setUp() + self.zone_share = self.client.zone_share(zone_id=self.zone.id, + *self.args, **self.kwargs) + self.addCleanup(self.cleanup_shared_zone, self.client, self.zone.id, + self.zone_share.id) + + @classmethod + def cleanup_shared_zone(cls, client, zone_id, shared_zone_id): + try: + client.unshare_zone(zone_id, shared_zone_id) + except CommandFailed: + pass diff --git a/designateclient/functionaltests/v2/test_shared_zone.py b/designateclient/functionaltests/v2/test_shared_zone.py new file mode 100644 index 0000000..0e8b392 --- /dev/null +++ b/designateclient/functionaltests/v2/test_shared_zone.py @@ -0,0 +1,73 @@ +""" + Copyright 2020 Cloudification GmbH. All rights reserved. + +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 designateclient.functionaltests.base import BaseDesignateTest +from designateclient.functionaltests.client import DesignateCLI +from designateclient.functionaltests.datagen import random_zone_name +from designateclient.functionaltests.v2.fixtures import SharedZoneFixture +from designateclient.functionaltests.v2.fixtures import ZoneFixture + + +class TestSharedZone(BaseDesignateTest): + + def setUp(self): + super(TestSharedZone, self).setUp() + self.ensure_tld_exists('com') + fixture = self.useFixture(ZoneFixture( + name=random_zone_name(), + email='test@example.com', + )) + self.zone = fixture.zone + self.target_client = DesignateCLI.as_user('alt') + + def test_list_shared_zones(self): + shared_zone = self.useFixture(SharedZoneFixture( + zone_id=self.zone.id, + target_tenant_id=self.target_client.project_id + )).zone_share + + shared_zones = self.clients.shared_zone_list(self.zone.id) + self.assertGreater(len(shared_zones), 0) + self.assertTrue(self._is_entity_in_list(shared_zone, shared_zones)) + + def test_share_and_show_shared_zone(self): + shared_zone = self.useFixture(SharedZoneFixture( + zone_id=self.zone.id, + target_tenant_id=self.target_client.project_id + )).zone_share + + fetched_shared_zone = self.clients.shared_zone_show(self.zone.id, + shared_zone.id) + + self.assertEqual( + shared_zone.created_at, fetched_shared_zone.created_at) + self.assertEqual(shared_zone.id, fetched_shared_zone.id) + self.assertEqual( + shared_zone.project_id, fetched_shared_zone.project_id) + self.assertEqual(shared_zone.zone_id, fetched_shared_zone.zone_id) + + def test_unshare_zone(self): + shared_zone = self.useFixture(SharedZoneFixture( + zone_id=self.zone.id, + target_tenant_id=self.target_client.project_id + )).zone_share + + shared_zones = self.clients.shared_zone_list(self.zone.id) + self.assertTrue(self._is_entity_in_list(shared_zone, shared_zones)) + + self.clients.unshare_zone(self.zone.id, shared_zone.id) + + shared_zones = self.clients.shared_zone_list(self.zone.id) + self.assertFalse(self._is_entity_in_list(shared_zone, shared_zones)) |
