253 lines
8.8 KiB
Python
253 lines
8.8 KiB
Python
|
# Copyright 2015 Google Inc. 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.
|
||
|
|
|
||
|
|
import unittest2
|
||
|
|
|
||
|
|
|
||
|
|
class TestClient(unittest2.TestCase):
|
||
|
|
|
||
|
|
PROJECT = 'PROJECT'
|
||
|
|
ZONE_NAME = 'zone-name'
|
||
|
|
|
||
|
|
def _getTargetClass(self):
|
||
|
|
from gcloud.dns.client import Client
|
||
|
|
return Client
|
||
|
|
|
||
|
|
def _makeOne(self, *args, **kw):
|
||
|
|
return self._getTargetClass()(*args, **kw)
|
||
|
|
|
||
|
|
def test_ctor(self):
|
||
|
|
from gcloud.dns.connection import Connection
|
||
|
|
creds = _Credentials()
|
||
|
|
http = object()
|
||
|
|
client = self._makeOne(project=self.PROJECT, credentials=creds,
|
||
|
|
http=http)
|
||
|
|
self.assertTrue(isinstance(client.connection, Connection))
|
||
|
|
self.assertTrue(client.connection.credentials is creds)
|
||
|
|
self.assertTrue(client.connection.http is http)
|
||
|
|
|
||
|
|
def test_quotas_defaults(self):
|
||
|
|
PATH = 'projects/%s' % (self.PROJECT,)
|
||
|
|
MANAGED_ZONES = 1234
|
||
|
|
RRS_PER_RRSET = 23
|
||
|
|
RRSETS_PER_ZONE = 345
|
||
|
|
RRSET_ADDITIONS = 456
|
||
|
|
RRSET_DELETIONS = 567
|
||
|
|
TOTAL_SIZE = 67890
|
||
|
|
DATA = {
|
||
|
|
'quota': {
|
||
|
|
'managedZones': str(MANAGED_ZONES),
|
||
|
|
'resourceRecordsPerRrset': str(RRS_PER_RRSET),
|
||
|
|
'rrsetsPerManagedZone': str(RRSETS_PER_ZONE),
|
||
|
|
'rrsetAdditionsPerChange': str(RRSET_ADDITIONS),
|
||
|
|
'rrsetDeletionsPerChange': str(RRSET_DELETIONS),
|
||
|
|
'totalRrdataSizePerChange': str(TOTAL_SIZE),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
CONVERTED = dict([(key, int(value))
|
||
|
|
for key, value in DATA['quota'].items()])
|
||
|
|
creds = _Credentials()
|
||
|
|
client = self._makeOne(self.PROJECT, creds)
|
||
|
|
conn = client.connection = _Connection(DATA)
|
||
|
|
|
||
|
|
quotas = client.quotas()
|
||
|
|
|
||
|
|
self.assertEqual(quotas, CONVERTED)
|
||
|
|
|
||
|
|
self.assertEqual(len(conn._requested), 1)
|
||
|
|
req = conn._requested[0]
|
||
|
|
self.assertEqual(req['method'], 'GET')
|
||
|
|
self.assertEqual(req['path'], '/%s' % PATH)
|
||
|
|
|
||
|
|
def test_quotas_w_kind_key(self):
|
||
|
|
PATH = 'projects/%s' % (self.PROJECT,)
|
||
|
|
MANAGED_ZONES = 1234
|
||
|
|
RRS_PER_RRSET = 23
|
||
|
|
RRSETS_PER_ZONE = 345
|
||
|
|
RRSET_ADDITIONS = 456
|
||
|
|
RRSET_DELETIONS = 567
|
||
|
|
TOTAL_SIZE = 67890
|
||
|
|
DATA = {
|
||
|
|
'quota': {
|
||
|
|
'managedZones': str(MANAGED_ZONES),
|
||
|
|
'resourceRecordsPerRrset': str(RRS_PER_RRSET),
|
||
|
|
'rrsetsPerManagedZone': str(RRSETS_PER_ZONE),
|
||
|
|
'rrsetAdditionsPerChange': str(RRSET_ADDITIONS),
|
||
|
|
'rrsetDeletionsPerChange': str(RRSET_DELETIONS),
|
||
|
|
'totalRrdataSizePerChange': str(TOTAL_SIZE),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
CONVERTED = dict([(key, int(value))
|
||
|
|
for key, value in DATA['quota'].items()])
|
||
|
|
WITH_KIND = {'quota': DATA['quota'].copy()}
|
||
|
|
WITH_KIND['quota']['kind'] = 'dns#quota'
|
||
|
|
creds = _Credentials()
|
||
|
|
client = self._makeOne(self.PROJECT, creds)
|
||
|
|
conn = client.connection = _Connection(WITH_KIND)
|
||
|
|
|
||
|
|
quotas = client.quotas()
|
||
|
|
|
||
|
|
self.assertEqual(quotas, CONVERTED)
|
||
|
|
|
||
|
|
self.assertEqual(len(conn._requested), 1)
|
||
|
|
req = conn._requested[0]
|
||
|
|
self.assertEqual(req['method'], 'GET')
|
||
|
|
self.assertEqual(req['path'], '/%s' % PATH)
|
||
|
|
|
||
|
|
def test_list_zones_defaults(self):
|
||
|
|
from gcloud.dns.zone import ManagedZone
|
||
|
|
ID_1 = '123'
|
||
|
|
ZONE_1 = 'zone_one'
|
||
|
|
DNS_1 = 'one.example.com'
|
||
|
|
ID_2 = '234'
|
||
|
|
ZONE_2 = 'zone_two'
|
||
|
|
DNS_2 = 'two.example.com'
|
||
|
|
PATH = 'projects/%s/managedZones' % (self.PROJECT,)
|
||
|
|
TOKEN = 'TOKEN'
|
||
|
|
DATA = {
|
||
|
|
'nextPageToken': TOKEN,
|
||
|
|
'managedZones': [
|
||
|
|
{'kind': 'dns#managedZone',
|
||
|
|
'id': ID_1,
|
||
|
|
'name': ZONE_1,
|
||
|
|
'dnsName': DNS_1},
|
||
|
|
{'kind': 'dns#managedZone',
|
||
|
|
'id': ID_2,
|
||
|
|
'name': ZONE_2,
|
||
|
|
'dnsName': DNS_2},
|
||
|
|
]
|
||
|
|
}
|
||
|
|
creds = _Credentials()
|
||
|
|
client = self._makeOne(self.PROJECT, creds)
|
||
|
|
conn = client.connection = _Connection(DATA)
|
||
|
|
|
||
|
|
zones, token = client.list_zones()
|
||
|
|
|
||
|
|
self.assertEqual(len(zones), len(DATA['managedZones']))
|
||
|
|
for found, expected in zip(zones, DATA['managedZones']):
|
||
|
|
self.assertTrue(isinstance(found, ManagedZone))
|
||
|
|
self.assertEqual(found.zone_id, expected['id'])
|
||
|
|
self.assertEqual(found.name, expected['name'])
|
||
|
|
self.assertEqual(found.dns_name, expected['dnsName'])
|
||
|
|
self.assertEqual(token, TOKEN)
|
||
|
|
|
||
|
|
self.assertEqual(len(conn._requested), 1)
|
||
|
|
req = conn._requested[0]
|
||
|
|
self.assertEqual(req['method'], 'GET')
|
||
|
|
self.assertEqual(req['path'], '/%s' % PATH)
|
||
|
|
|
||
|
|
def test_list_zones_explicit(self):
|
||
|
|
from gcloud.dns.zone import ManagedZone
|
||
|
|
ID_1 = '123'
|
||
|
|
ZONE_1 = 'zone_one'
|
||
|
|
DNS_1 = 'one.example.com'
|
||
|
|
ID_2 = '234'
|
||
|
|
ZONE_2 = 'zone_two'
|
||
|
|
DNS_2 = 'two.example.com'
|
||
|
|
PATH = 'projects/%s/managedZones' % (self.PROJECT,)
|
||
|
|
TOKEN = 'TOKEN'
|
||
|
|
DATA = {
|
||
|
|
'managedZones': [
|
||
|
|
{'kind': 'dns#managedZone',
|
||
|
|
'id': ID_1,
|
||
|
|
'name': ZONE_1,
|
||
|
|
'dnsName': DNS_1},
|
||
|
|
{'kind': 'dns#managedZone',
|
||
|
|
'id': ID_2,
|
||
|
|
'name': ZONE_2,
|
||
|
|
'dnsName': DNS_2},
|
||
|
|
]
|
||
|
|
}
|
||
|
|
creds = _Credentials()
|
||
|
|
client = self._makeOne(self.PROJECT, creds)
|
||
|
|
conn = client.connection = _Connection(DATA)
|
||
|
|
|
||
|
|
zones, token = client.list_zones(max_results=3, page_token=TOKEN)
|
||
|
|
|
||
|
|
self.assertEqual(len(zones), len(DATA['managedZones']))
|
||
|
|
for found, expected in zip(zones, DATA['managedZones']):
|
||
|
|
self.assertTrue(isinstance(found, ManagedZone))
|
||
|
|
self.assertEqual(found.zone_id, expected['id'])
|
||
|
|
self.assertEqual(found.name, expected['name'])
|
||
|
|
self.assertEqual(found.dns_name, expected['dnsName'])
|
||
|
|
self.assertEqual(token, None)
|
||
|
|
|
||
|
|
self.assertEqual(len(conn._requested), 1)
|
||
|
|
req = conn._requested[0]
|
||
|
|
self.assertEqual(req['method'], 'GET')
|
||
|
|
self.assertEqual(req['path'], '/%s' % PATH)
|
||
|
|
self.assertEqual(req['query_params'],
|
||
|
|
{'maxResults': 3, 'pageToken': TOKEN})
|
||
|
|
|
||
|
|
def test_zone_explicit(self):
|
||
|
|
from gcloud.dns.zone import ManagedZone
|
||
|
|
DESCRIPTION = 'DESCRIPTION'
|
||
|
|
DNS_NAME = 'test.example.com'
|
||
|
|
creds = _Credentials()
|
||
|
|
client = self._makeOne(self.PROJECT, creds)
|
||
|
|
zone = client.zone(self.ZONE_NAME, DNS_NAME, DESCRIPTION)
|
||
|
|
self.assertTrue(isinstance(zone, ManagedZone))
|
||
|
|
self.assertEqual(zone.name, self.ZONE_NAME)
|
||
|
|
self.assertEqual(zone.dns_name, DNS_NAME)
|
||
|
|
self.assertEqual(zone.description, DESCRIPTION)
|
||
|
|
self.assertTrue(zone._client is client)
|
||
|
|
|
||
|
|
def test_zone_w_dns_name_wo_description(self):
|
||
|
|
from gcloud.dns.zone import ManagedZone
|
||
|
|
DNS_NAME = 'test.example.com'
|
||
|
|
creds = _Credentials()
|
||
|
|
client = self._makeOne(self.PROJECT, creds)
|
||
|
|
zone = client.zone(self.ZONE_NAME, DNS_NAME)
|
||
|
|
self.assertTrue(isinstance(zone, ManagedZone))
|
||
|
|
self.assertEqual(zone.name, self.ZONE_NAME)
|
||
|
|
self.assertEqual(zone.dns_name, DNS_NAME)
|
||
|
|
self.assertEqual(zone.description, DNS_NAME)
|
||
|
|
self.assertTrue(zone._client is client)
|
||
|
|
|
||
|
|
def test_zone_wo_dns_name(self):
|
||
|
|
from gcloud.dns.zone import ManagedZone
|
||
|
|
creds = _Credentials()
|
||
|
|
client = self._makeOne(self.PROJECT, creds)
|
||
|
|
zone = client.zone(self.ZONE_NAME)
|
||
|
|
self.assertTrue(isinstance(zone, ManagedZone))
|
||
|
|
self.assertEqual(zone.name, self.ZONE_NAME)
|
||
|
|
self.assertEqual(zone.dns_name, None)
|
||
|
|
self.assertEqual(zone.description, None)
|
||
|
|
self.assertTrue(zone._client is client)
|
||
|
|
|
||
|
|
|
||
|
|
class _Credentials(object):
|
||
|
|
|
||
|
|
_scopes = None
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def create_scoped_required():
|
||
|
|
return True
|
||
|
|
|
||
|
|
def create_scoped(self, scope):
|
||
|
|
self._scopes = scope
|
||
|
|
return self
|
||
|
|
|
||
|
|
|
||
|
|
class _Connection(object):
|
||
|
|
|
||
|
|
def __init__(self, *responses):
|
||
|
|
self._responses = responses
|
||
|
|
self._requested = []
|
||
|
|
|
||
|
|
def api_request(self, **kw):
|
||
|
|
self._requested.append(kw)
|
||
|
|
response, self._responses = self._responses[0], self._responses[1:]
|
||
|
|
return response
|