def _init_client(self, auth_client):
kwargs = {
'project': self.config['project'],
'api_version': self.config.get('api_version', 'v1'),
'auth_client': auth_client
}
return gdns.GDNSClient(**kwargs)
def _init_client(self, auth_client):
kwargs = {
'project': self.config['project'],
'api_version': self.config.get('api_version', 'v1'),
'auth_client': auth_client,
'default_zone_prefix': self.config.get('default_zone_prefix', '')
}
return gdns.GDNSClient(**kwargs)
def client(mocker, create_mock_coro):
auth_client = mocker.Mock(auth.GAuthClient)
creds = mocker.Mock()
auth_client.creds = creds
session = aiohttp.ClientSession()
client = gdns.GDNSClient(
'a-project', auth_client=auth_client, session=session)
# mock out inherited methods from http.AIOConnection
get_json_mock, get_json_coro = create_mock_coro()
request_post_mock, post_request_coro = create_mock_coro()
client.get_json = get_json_coro
client._get_json_mock = get_json_mock
client.request = post_request_coro
client._request_post_mock = request_post_mock
yield client
# test teardown
client._session.close()
def _init_dns_client(self):
auth_client = self._init_auth_client()
return gdns.GDNSClient(self.config['project'],
auth_client,
default_zone_prefix=self.config.get(
'default_zone_prefix', ''))
def _init_dns_client(self):
return gdns.GDNSClient(self.config['project'], self._init_auth())
def _init_dns_client(self):
auth_client = self._init_auth_client()
return gdns.GDNSClient(self.config['project'], auth_client)
def test_get_managed_zone_with_prefix(mocker, dns_zone, prefix, expected):
client = gdns.GDNSClient('a-project', auth_client=mocker.Mock(),
session=mocker.Mock(), default_zone_prefix=prefix)
assert expected == client.get_managed_zone(dns_zone)
