def test_designate(self, mock_keystone_get_session, version=None): fake_designate = fakes.FakeDesignateClient() mock_designate = mock.Mock() mock_designate.client.Client.return_value = fake_designate mock_keystone_get_session.return_value = ("fake_session", "fake_auth_plugin") self.assertNotIn("designate", self.clients.cache) with mock.patch.dict("sys.modules", {"designateclient": mock_designate}): if version is not None: client = self.clients.designate(version=version) else: client = self.clients.designate() self.assertEqual(fake_designate, client) self.service_catalog.url_for.assert_called_once_with( service_type="dns", region_name=self.credential.region_name ) default = version or "1" # Check that we append /v<version> url = self.service_catalog.url_for.return_value url.__iadd__.assert_called_once_with("/v%s" % default) mock_keystone_get_session.assert_called_once_with() if version == "2": mock_designate.client.Client.assert_called_once_with( version, endpoint_override=url.__iadd__.return_value, session="fake_session") elif version == "1": mock_designate.client.Client.assert_called_once_with( version, endpoint=url.__iadd__.return_value, session="fake_session") key = "designate" if version is not None: key += "%s" % {"version": version} self.assertEqual(fake_designate, self.clients.cache[key])
def test_designate(self, mock_designate__get_session, version=None): fake_designate = fakes.FakeDesignateClient() mock_designate = mock.Mock() mock_designate.client.Client.return_value = fake_designate mock_designate__get_session.return_value = self.fake_keystone.session self.assertNotIn("designate", self.clients.cache) with mock.patch.dict("sys.modules", {"designateclient": mock_designate}): if version is not None: client = self.clients.designate(version=version) else: client = self.clients.designate() self.assertEqual(fake_designate, client) self.service_catalog.url_for.assert_called_once_with( service_type="dns", endpoint_type=consts.EndpointType.PUBLIC, region_name=self.endpoint.region_name) default = version or "1" # Check that we append /v<version> url = self.service_catalog.url_for.return_value url.__iadd__.assert_called_once_with("/v%s" % default) mock_designate__get_session.assert_called_once_with( endpoint=url.__iadd__.return_value) mock_designate.client.Client.assert_called_once_with( default, session=self.fake_keystone.session) key = "designate" if version is not None: key += "%s" % {"version": version} self.assertEqual(fake_designate, self.clients.cache[key])