mirror of
https://github.com/github/octodns.git
synced 2024-05-11 05:55:00 +00:00
Constellix provider full Dynamic support added
This commit is contained in:
@@ -195,7 +195,7 @@ The above command pulled the existing data out of Route53 and placed the results
|
|||||||
| [AzureProvider](/octodns/provider/azuredns.py) | azure-identity, azure-mgmt-dns, azure-mgmt-trafficmanager | A, AAAA, CAA, CNAME, MX, NS, PTR, SRV, TXT | Alpha (A, AAAA, CNAME) | |
|
| [AzureProvider](/octodns/provider/azuredns.py) | azure-identity, azure-mgmt-dns, azure-mgmt-trafficmanager | A, AAAA, CAA, CNAME, MX, NS, PTR, SRV, TXT | Alpha (A, AAAA, CNAME) | |
|
||||||
| [Akamai](/octodns/provider/edgedns.py) | edgegrid-python | A, AAAA, CNAME, MX, NAPTR, NS, PTR, SPF, SRV, SSHFP, TXT | No | |
|
| [Akamai](/octodns/provider/edgedns.py) | edgegrid-python | A, AAAA, CNAME, MX, NAPTR, NS, PTR, SPF, SRV, SSHFP, TXT | No | |
|
||||||
| [CloudflareProvider](/octodns/provider/cloudflare.py) | | A, AAAA, ALIAS, CAA, CNAME, LOC, MX, NS, PTR, SPF, SRV, TXT | No | CAA tags restricted |
|
| [CloudflareProvider](/octodns/provider/cloudflare.py) | | A, AAAA, ALIAS, CAA, CNAME, LOC, MX, NS, PTR, SPF, SRV, TXT | No | CAA tags restricted |
|
||||||
| [ConstellixProvider](/octodns/provider/constellix.py) | | A, AAAA, ALIAS (ANAME), CAA, CNAME, MX, NS, PTR, SPF, SRV, TXT | No | CAA tags restricted |
|
| [ConstellixProvider](/octodns/provider/constellix.py) | | A, AAAA, ALIAS (ANAME), CAA, CNAME, MX, NS, PTR, SPF, SRV, TXT | Yes | CAA tags restricted |
|
||||||
| [DigitalOceanProvider](/octodns/provider/digitalocean.py) | | A, AAAA, CAA, CNAME, MX, NS, TXT, SRV | No | CAA tags restricted |
|
| [DigitalOceanProvider](/octodns/provider/digitalocean.py) | | A, AAAA, CAA, CNAME, MX, NS, TXT, SRV | No | CAA tags restricted |
|
||||||
| [DnsMadeEasyProvider](/octodns/provider/dnsmadeeasy.py) | | A, AAAA, ALIAS (ANAME), CAA, CNAME, MX, NS, PTR, SPF, SRV, TXT | No | CAA tags restricted |
|
| [DnsMadeEasyProvider](/octodns/provider/dnsmadeeasy.py) | | A, AAAA, ALIAS (ANAME), CAA, CNAME, MX, NS, PTR, SPF, SRV, TXT | No | CAA tags restricted |
|
||||||
| [DnsimpleProvider](/octodns/provider/dnsimple.py) | | All | No | CAA tags restricted |
|
| [DnsimpleProvider](/octodns/provider/dnsimple.py) | | All | No | CAA tags restricted |
|
||||||
|
|||||||
@@ -9,6 +9,7 @@ from collections import defaultdict
|
|||||||
from requests import Session
|
from requests import Session
|
||||||
from base64 import b64encode
|
from base64 import b64encode
|
||||||
from six import string_types
|
from six import string_types
|
||||||
|
from pycountry_convert import country_alpha2_to_continent_code
|
||||||
import hashlib
|
import hashlib
|
||||||
import hmac
|
import hmac
|
||||||
import logging
|
import logging
|
||||||
@@ -53,7 +54,8 @@ class ConstellixClient(object):
|
|||||||
self._sess = Session()
|
self._sess = Session()
|
||||||
self._sess.headers.update({'x-cnsdns-apiKey': self.api_key})
|
self._sess.headers.update({'x-cnsdns-apiKey': self.api_key})
|
||||||
self._domains = None
|
self._domains = None
|
||||||
self._pools = None
|
self._pools = {'A': None, 'AAAA': None, 'CNAME': None}
|
||||||
|
self._geofilters = None
|
||||||
|
|
||||||
def _current_time(self):
|
def _current_time(self):
|
||||||
return str(int(time.time() * 1000))
|
return str(int(time.time() * 1000))
|
||||||
@@ -100,19 +102,29 @@ class ConstellixClient(object):
|
|||||||
zone_id = self.domains.get(name, False)
|
zone_id = self.domains.get(name, False)
|
||||||
if not zone_id:
|
if not zone_id:
|
||||||
raise ConstellixClientNotFound()
|
raise ConstellixClientNotFound()
|
||||||
path = '/domains/{}'.format(zone_id)
|
path = f'/domains/{zone_id}'
|
||||||
return self._request('GET', path).json()
|
return self._request('GET', path).json()
|
||||||
|
|
||||||
def domain_create(self, name):
|
def domain_create(self, name):
|
||||||
resp = self._request('POST', '/domains', data={'names': [name]})
|
resp = self._request('POST', '/domains', data={'names': [name]})
|
||||||
# Add newly created zone to domain cache
|
# Add newly created zone to domain cache
|
||||||
self._domains['{}.'.format(name)] = resp.json()[0]['id']
|
self._domains[f'{name}.'] = resp.json()[0]['id']
|
||||||
|
|
||||||
|
def domain_enable_geoip(self, domain_name):
|
||||||
|
domain = self.domain(domain_name)
|
||||||
|
if domain['hasGeoIP'] is False:
|
||||||
|
domain_id = self.domains[domain_name]
|
||||||
|
self._request(
|
||||||
|
'PUT',
|
||||||
|
f'/domains/{domain_id}',
|
||||||
|
data={'hasGeoIP': True}
|
||||||
|
)
|
||||||
|
|
||||||
def _absolutize_value(self, value, zone_name):
|
def _absolutize_value(self, value, zone_name):
|
||||||
if value == '':
|
if value == '':
|
||||||
value = zone_name
|
value = zone_name
|
||||||
elif not value.endswith('.'):
|
elif not value.endswith('.'):
|
||||||
value = '{}.{}'.format(value, zone_name)
|
value = f'{value}.{zone_name}'
|
||||||
|
|
||||||
return value
|
return value
|
||||||
|
|
||||||
@@ -120,7 +132,7 @@ class ConstellixClient(object):
|
|||||||
zone_id = self.domains.get(zone_name, False)
|
zone_id = self.domains.get(zone_name, False)
|
||||||
if not zone_id:
|
if not zone_id:
|
||||||
raise ConstellixClientNotFound()
|
raise ConstellixClientNotFound()
|
||||||
path = '/domains/{}/records'.format(zone_id)
|
path = f'/domains/{zone_id}/records'
|
||||||
|
|
||||||
resp = self._request('GET', path).json()
|
resp = self._request('GET', path).json()
|
||||||
for record in resp:
|
for record in resp:
|
||||||
@@ -147,7 +159,7 @@ class ConstellixClient(object):
|
|||||||
record_type = 'ANAME'
|
record_type = 'ANAME'
|
||||||
|
|
||||||
zone_id = self.domains.get(zone_name, False)
|
zone_id = self.domains.get(zone_name, False)
|
||||||
path = '/domains/{}/records/{}'.format(zone_id, record_type)
|
path = f'/domains/{zone_id}/records/{record_type}'
|
||||||
|
|
||||||
self._request('POST', path, data=params)
|
self._request('POST', path, data=params)
|
||||||
|
|
||||||
@@ -157,18 +169,17 @@ class ConstellixClient(object):
|
|||||||
record_type = 'ANAME'
|
record_type = 'ANAME'
|
||||||
|
|
||||||
zone_id = self.domains.get(zone_name, False)
|
zone_id = self.domains.get(zone_name, False)
|
||||||
path = '/domains/{}/records/{}/{}'.format(zone_id, record_type,
|
path = f'/domains/{zone_id}/records/{record_type}/{record_id}'
|
||||||
record_id)
|
|
||||||
self._request('DELETE', path)
|
self._request('DELETE', path)
|
||||||
|
|
||||||
def pools(self, pool_type):
|
def pools(self, pool_type):
|
||||||
if self._pools is None:
|
if self._pools[pool_type] is None:
|
||||||
self._pools = {}
|
self._pools[pool_type] = {}
|
||||||
path = '/pools/{}'.format(pool_type)
|
path = f'/pools/{pool_type}'
|
||||||
response = self._request('GET', path).json()
|
response = self._request('GET', path).json()
|
||||||
for pool in response:
|
for pool in response:
|
||||||
self._pools[pool['id']] = pool
|
self._pools[pool_type][pool['id']] = pool
|
||||||
return self._pools.values()
|
return self._pools[pool_type].values()
|
||||||
|
|
||||||
def pool(self, pool_type, pool_name):
|
def pool(self, pool_type, pool_name):
|
||||||
pools = self.pools(pool_type)
|
pools = self.pools(pool_type)
|
||||||
@@ -186,11 +197,11 @@ class ConstellixClient(object):
|
|||||||
def pool_create(self, data):
|
def pool_create(self, data):
|
||||||
path = '/pools/{}'.format(data.get('type'))
|
path = '/pools/{}'.format(data.get('type'))
|
||||||
# This returns a list of items, we want the first one
|
# This returns a list of items, we want the first one
|
||||||
response = self._request('POST', path, data=data).json()[0]
|
response = self._request('POST', path, data=data).json()
|
||||||
|
|
||||||
# Invalidate our cache
|
# Update our cache
|
||||||
self._pools = None
|
self._pools[data.get('type')][response[0]['id']] = response[0]
|
||||||
return response
|
return response[0]
|
||||||
|
|
||||||
def pool_update(self, pool_id, data):
|
def pool_update(self, pool_id, data):
|
||||||
path = '/pools/{}/{}'.format(data.get('type'), pool_id)
|
path = '/pools/{}/{}'.format(data.get('type'), pool_id)
|
||||||
@@ -203,6 +214,63 @@ class ConstellixClient(object):
|
|||||||
raise e
|
raise e
|
||||||
return data
|
return data
|
||||||
|
|
||||||
|
def pool_delete(self, pool_type, pool_id):
|
||||||
|
path = f'/pools/{pool_type}/{pool_id}'
|
||||||
|
self._request('DELETE', path)
|
||||||
|
|
||||||
|
# Update our cache
|
||||||
|
if self._pools[pool_type] is not None:
|
||||||
|
self._pools[pool_type].pop(pool_id, None)
|
||||||
|
|
||||||
|
def geofilters(self):
|
||||||
|
if self._geofilters is None:
|
||||||
|
self._geofilters = {}
|
||||||
|
path = '/geoFilters'
|
||||||
|
response = self._request('GET', path).json()
|
||||||
|
for geofilter in response:
|
||||||
|
self._geofilters[geofilter['id']] = geofilter
|
||||||
|
return self._geofilters.values()
|
||||||
|
|
||||||
|
def geofilter(self, geofilter_name):
|
||||||
|
geofilters = self.geofilters()
|
||||||
|
for geofilter in geofilters:
|
||||||
|
if geofilter['name'] == geofilter_name:
|
||||||
|
return geofilter
|
||||||
|
return None
|
||||||
|
|
||||||
|
def geofilter_by_id(self, geofilter_id):
|
||||||
|
geofilters = self.geofilters()
|
||||||
|
for geofilter in geofilters:
|
||||||
|
if geofilter['id'] == geofilter_id:
|
||||||
|
return geofilter
|
||||||
|
|
||||||
|
def geofilter_create(self, data):
|
||||||
|
path = '/geoFilters'
|
||||||
|
response = self._request('POST', path, data=data).json()
|
||||||
|
|
||||||
|
# Update our cache
|
||||||
|
self._geofilters[response[0]['id']] = response[0]
|
||||||
|
return response[0]
|
||||||
|
|
||||||
|
def geofilter_update(self, geofilter_id, data):
|
||||||
|
path = f'/geoFilters/{geofilter_id}'
|
||||||
|
try:
|
||||||
|
self._request('PUT', path, data=data).json()
|
||||||
|
|
||||||
|
except ConstellixClientBadRequest as e:
|
||||||
|
message = str(e)
|
||||||
|
if not message or "no changes to save" not in message:
|
||||||
|
raise e
|
||||||
|
return data
|
||||||
|
|
||||||
|
def geofilter_delete(self, geofilter_id):
|
||||||
|
path = f'/geoFilters/{geofilter_id}'
|
||||||
|
self._request('DELETE', path)
|
||||||
|
|
||||||
|
# Update our cache
|
||||||
|
if self._geofilters is not None:
|
||||||
|
self._geofilters.pop(geofilter_id, None)
|
||||||
|
|
||||||
|
|
||||||
class ConstellixProvider(BaseProvider):
|
class ConstellixProvider(BaseProvider):
|
||||||
'''
|
'''
|
||||||
@@ -225,7 +293,7 @@ class ConstellixProvider(BaseProvider):
|
|||||||
|
|
||||||
def __init__(self, id, api_key, secret_key, ratelimit_delay=0.0,
|
def __init__(self, id, api_key, secret_key, ratelimit_delay=0.0,
|
||||||
*args, **kwargs):
|
*args, **kwargs):
|
||||||
self.log = logging.getLogger('ConstellixProvider[{}]'.format(id))
|
self.log = logging.getLogger(f'ConstellixProvider[{id}]')
|
||||||
self.log.debug('__init__: id=%s, api_key=***, secret_key=***', id)
|
self.log.debug('__init__: id=%s, api_key=***, secret_key=***', id)
|
||||||
super(ConstellixProvider, self).__init__(id, *args, **kwargs)
|
super(ConstellixProvider, self).__init__(id, *args, **kwargs)
|
||||||
self._client = ConstellixClient(api_key, secret_key, ratelimit_delay)
|
self._client = ConstellixClient(api_key, secret_key, ratelimit_delay)
|
||||||
@@ -234,20 +302,42 @@ class ConstellixProvider(BaseProvider):
|
|||||||
def _data_for_multiple(self, _type, records):
|
def _data_for_multiple(self, _type, records):
|
||||||
record = records[0]
|
record = records[0]
|
||||||
if record['recordOption'] == 'pools':
|
if record['recordOption'] == 'pools':
|
||||||
return self._data_for_pool(_type, record)
|
return self._data_for_pool(_type, records)
|
||||||
return {
|
return {
|
||||||
'ttl': record['ttl'],
|
'ttl': record['ttl'],
|
||||||
'type': _type,
|
'type': _type,
|
||||||
'values': record['value']
|
'values': record['value']
|
||||||
}
|
}
|
||||||
|
|
||||||
def _data_for_pool(self, _type, record):
|
def _data_for_pool(self, _type, records):
|
||||||
|
default_values = []
|
||||||
|
fallback_pool_name = None
|
||||||
|
pools = {}
|
||||||
|
rules = []
|
||||||
|
|
||||||
|
for record in records:
|
||||||
|
# fetch record pool data
|
||||||
pool_id = record['pools'][0]
|
pool_id = record['pools'][0]
|
||||||
pool = self._client.pool_by_id(_type, pool_id)
|
pool = self._client.pool_by_id(_type, pool_id)
|
||||||
|
|
||||||
|
geofilter_id = 1
|
||||||
|
if 'geolocation' in record.keys() \
|
||||||
|
and record['geolocation'] is not None:
|
||||||
|
# fetch record geofilter data
|
||||||
|
geofilter_id = record['geolocation']['geoipFilter']
|
||||||
|
geofilter = self._client.geofilter_by_id(geofilter_id)
|
||||||
|
|
||||||
pool_name = pool['name'].split(':')[-1]
|
pool_name = pool['name'].split(':')[-1]
|
||||||
pools = {}
|
|
||||||
values = []
|
# fetch default values from the World Default pool
|
||||||
|
if geofilter_id == 1:
|
||||||
|
fallback_pool_name = pool_name
|
||||||
|
for value in pool['values']:
|
||||||
|
default_values.append(value['value'])
|
||||||
|
|
||||||
|
# populate pools
|
||||||
pools[pool_name] = {
|
pools[pool_name] = {
|
||||||
|
'fallback': None,
|
||||||
'values': []
|
'values': []
|
||||||
}
|
}
|
||||||
for value in pool['values']:
|
for value in pool['values']:
|
||||||
@@ -255,18 +345,52 @@ class ConstellixProvider(BaseProvider):
|
|||||||
'value': value['value'],
|
'value': value['value'],
|
||||||
'weight': value['weight']
|
'weight': value['weight']
|
||||||
})
|
})
|
||||||
values.append(value['value'])
|
|
||||||
return {
|
# populate rules
|
||||||
|
if geofilter_id == 1:
|
||||||
|
rules.append({'pool': pool_name})
|
||||||
|
else:
|
||||||
|
geos = []
|
||||||
|
|
||||||
|
if 'geoipContinents' in geofilter.keys():
|
||||||
|
for continent_code in geofilter['geoipContinents']:
|
||||||
|
geos.append(continent_code)
|
||||||
|
|
||||||
|
if 'geoipCountries' in geofilter.keys():
|
||||||
|
for country_code in geofilter['geoipCountries']:
|
||||||
|
geos.append('{}-{}'.format(
|
||||||
|
country_alpha2_to_continent_code(country_code),
|
||||||
|
country_code
|
||||||
|
))
|
||||||
|
|
||||||
|
if 'regions' in geofilter.keys():
|
||||||
|
for region in geofilter['regions']:
|
||||||
|
geos.append('{}-{}-{}'.format(
|
||||||
|
region['continentCode'],
|
||||||
|
region['countryCode'],
|
||||||
|
region['regionCode']))
|
||||||
|
|
||||||
|
rules.append({
|
||||||
|
'pool': pool_name,
|
||||||
|
'geos': sorted(geos)
|
||||||
|
})
|
||||||
|
|
||||||
|
# set fallback pool
|
||||||
|
for pool_name in pools:
|
||||||
|
if pool_name != fallback_pool_name:
|
||||||
|
pools[pool_name]['fallback'] = fallback_pool_name
|
||||||
|
|
||||||
|
res = {
|
||||||
'ttl': record['ttl'],
|
'ttl': record['ttl'],
|
||||||
'type': _type,
|
'type': _type,
|
||||||
'dynamic': {
|
'dynamic': {
|
||||||
'pools': pools,
|
'pools': dict(
|
||||||
'rules': [{
|
sorted(pools.items(), key=lambda t: t[0])),
|
||||||
'pool': pool_name
|
'rules': sorted(rules, key=lambda t: t['pool'])
|
||||||
}]
|
|
||||||
},
|
},
|
||||||
'value': values
|
'values': default_values
|
||||||
}
|
}
|
||||||
|
return res
|
||||||
|
|
||||||
_data_for_A = _data_for_multiple
|
_data_for_A = _data_for_multiple
|
||||||
_data_for_AAAA = _data_for_multiple
|
_data_for_AAAA = _data_for_multiple
|
||||||
@@ -381,7 +505,7 @@ class ConstellixProvider(BaseProvider):
|
|||||||
before = len(zone.records)
|
before = len(zone.records)
|
||||||
for name, types in values.items():
|
for name, types in values.items():
|
||||||
for _type, records in types.items():
|
for _type, records in types.items():
|
||||||
data_for = getattr(self, '_data_for_{}'.format(_type))
|
data_for = getattr(self, f'_data_for_{_type}')
|
||||||
record = Record.new(zone, name, data_for(_type, records),
|
record = Record.new(zone, name, data_for(_type, records),
|
||||||
source=self, lenient=lenient)
|
source=self, lenient=lenient)
|
||||||
zone.add_record(record, lenient=lenient)
|
zone.add_record(record, lenient=lenient)
|
||||||
@@ -491,21 +615,17 @@ class ConstellixProvider(BaseProvider):
|
|||||||
def _handle_pools(self, record):
|
def _handle_pools(self, record):
|
||||||
# If we don't have dynamic, then there's no pools
|
# If we don't have dynamic, then there's no pools
|
||||||
if not getattr(record, 'dynamic', False):
|
if not getattr(record, 'dynamic', False):
|
||||||
return None
|
return []
|
||||||
|
|
||||||
# Get our first entry in the rules that references a pool
|
res_pools = []
|
||||||
rules = list(filter(
|
|
||||||
lambda rule: 'pool' in rule.data,
|
|
||||||
record.dynamic.rules
|
|
||||||
))
|
|
||||||
|
|
||||||
pool_name = rules[0].data.get('pool')
|
|
||||||
|
|
||||||
|
for i, rule in enumerate(record.dynamic.rules):
|
||||||
|
pool_name = rule.data.get('pool')
|
||||||
pool = record.dynamic.pools.get(pool_name)
|
pool = record.dynamic.pools.get(pool_name)
|
||||||
values = pool.data.get('values')
|
values = pool.data.get('values')
|
||||||
|
|
||||||
# Make a pool name based on zone, record, type and name
|
# Make a pool name based on zone, record, type and name
|
||||||
pool_name = '{}:{}:{}:{}'.format(
|
generated_pool_name = '{}:{}:{}:{}'.format(
|
||||||
record.zone.name,
|
record.zone.name,
|
||||||
record.name,
|
record.name,
|
||||||
record._type,
|
record._type,
|
||||||
@@ -513,13 +633,53 @@ class ConstellixProvider(BaseProvider):
|
|||||||
)
|
)
|
||||||
|
|
||||||
# OK, pool is valid, let's create it or update it
|
# OK, pool is valid, let's create it or update it
|
||||||
return self._create_update_pool(
|
self.log.debug("Creating pool %s", generated_pool_name)
|
||||||
pool_name = pool_name,
|
pool_obj = self._create_update_pool(
|
||||||
|
pool_name = generated_pool_name,
|
||||||
pool_type = record._type,
|
pool_type = record._type,
|
||||||
ttl = record.ttl,
|
ttl = record.ttl,
|
||||||
values = values
|
values = values
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# Now will crate GeoFilter for the pool
|
||||||
|
continents = []
|
||||||
|
countries = []
|
||||||
|
regions = []
|
||||||
|
|
||||||
|
for geo in rule.data.get('geos', []):
|
||||||
|
codes = geo.split('-')
|
||||||
|
n = len(geo)
|
||||||
|
if n == 2:
|
||||||
|
continents.append(geo)
|
||||||
|
elif n == 5:
|
||||||
|
countries.append(codes[1])
|
||||||
|
else:
|
||||||
|
regions.append({
|
||||||
|
'continentCode': codes[0],
|
||||||
|
'countryCode': codes[1],
|
||||||
|
'regionCode': codes[2]
|
||||||
|
})
|
||||||
|
|
||||||
|
if len(continents) == 0 and \
|
||||||
|
len(countries) == 0 and \
|
||||||
|
len(regions) == 0:
|
||||||
|
pool_obj['geofilter'] = 1
|
||||||
|
else:
|
||||||
|
self.log.debug(
|
||||||
|
"Creating geofilter %s",
|
||||||
|
generated_pool_name
|
||||||
|
)
|
||||||
|
geofilter_obj = self._create_update_geofilter(
|
||||||
|
generated_pool_name,
|
||||||
|
continents,
|
||||||
|
countries,
|
||||||
|
regions
|
||||||
|
)
|
||||||
|
pool_obj['geofilter'] = geofilter_obj['id']
|
||||||
|
|
||||||
|
res_pools.append(pool_obj)
|
||||||
|
return res_pools
|
||||||
|
|
||||||
def _create_update_pool(self, pool_name, pool_type, ttl, values):
|
def _create_update_pool(self, pool_name, pool_type, ttl, values):
|
||||||
pool = {
|
pool = {
|
||||||
'name': pool_name,
|
'name': pool_name,
|
||||||
@@ -538,29 +698,173 @@ class ConstellixProvider(BaseProvider):
|
|||||||
updated_pool['id'] = pool_id
|
updated_pool['id'] = pool_id
|
||||||
return updated_pool
|
return updated_pool
|
||||||
|
|
||||||
def _apply_Create(self, change):
|
def _create_update_geofilter(
|
||||||
|
self,
|
||||||
|
geofilter_name,
|
||||||
|
continents,
|
||||||
|
countries,
|
||||||
|
regions):
|
||||||
|
geofilter = {
|
||||||
|
'filterRulesLimit': 100,
|
||||||
|
'name': geofilter_name,
|
||||||
|
'geoipContinents': continents,
|
||||||
|
'geoipCountries': countries,
|
||||||
|
'regions': regions
|
||||||
|
}
|
||||||
|
if len(regions) == 0:
|
||||||
|
geofilter.pop('regions', None)
|
||||||
|
|
||||||
|
existing_geofilter = self._client.geofilter(geofilter_name)
|
||||||
|
if not existing_geofilter:
|
||||||
|
return self._client.geofilter_create(geofilter)
|
||||||
|
|
||||||
|
geofilter_id = existing_geofilter['id']
|
||||||
|
updated_geofilter = self._client.geofilter_update(
|
||||||
|
geofilter_id, geofilter)
|
||||||
|
updated_geofilter['id'] = geofilter_id
|
||||||
|
return updated_geofilter
|
||||||
|
|
||||||
|
def _apply_Create(self, change, domain_name):
|
||||||
new = change.new
|
new = change.new
|
||||||
params_for = getattr(self, '_params_for_{}'.format(new._type))
|
params_for = getattr(self, '_params_for_{}'.format(new._type))
|
||||||
pool = self._handle_pools(new)
|
pools = self._handle_pools(new)
|
||||||
|
|
||||||
for params in params_for(new):
|
for params in params_for(new):
|
||||||
if pool:
|
if len(pools) == 0:
|
||||||
|
self._client.record_create(new.zone.name, new._type, params)
|
||||||
|
elif len(pools) == 1:
|
||||||
|
params['pools'] = [pools[0]['id']]
|
||||||
|
params['recordOption'] = 'pools'
|
||||||
|
params.pop('roundRobin', None)
|
||||||
|
self.log.debug(
|
||||||
|
"Creating record %s %s",
|
||||||
|
new.zone.name,
|
||||||
|
new._type
|
||||||
|
)
|
||||||
|
self._client.record_create(
|
||||||
|
new.zone.name,
|
||||||
|
new._type,
|
||||||
|
params
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
# To use GeoIPFilter feature we need to enable it for domain
|
||||||
|
self.log.debug("Enabling domain %s geo support", domain_name)
|
||||||
|
self._client.domain_enable_geoip(domain_name)
|
||||||
|
|
||||||
|
# First we need to create World Default (1) Record
|
||||||
|
for pool in pools:
|
||||||
|
if pool['geofilter'] != 1:
|
||||||
|
continue
|
||||||
params['pools'] = [pool['id']]
|
params['pools'] = [pool['id']]
|
||||||
params['recordOption'] = 'pools'
|
params['recordOption'] = 'pools'
|
||||||
self._client.record_create(new.zone.name, new._type, params)
|
params['geolocation'] = {
|
||||||
|
'geoipUserRegion': [pool['geofilter']]
|
||||||
|
}
|
||||||
|
params.pop('roundRobin', None)
|
||||||
|
self.log.debug(
|
||||||
|
"Creating record %s %s",
|
||||||
|
new.zone.name,
|
||||||
|
new._type)
|
||||||
|
self._client.record_create(
|
||||||
|
new.zone.name,
|
||||||
|
new._type,
|
||||||
|
params
|
||||||
|
)
|
||||||
|
|
||||||
def _apply_Update(self, change):
|
# Now we can create the rest of records
|
||||||
self._apply_Delete(change)
|
for pool in pools:
|
||||||
self._apply_Create(change)
|
if pool['geofilter'] == 1:
|
||||||
|
continue
|
||||||
|
params['pools'] = [pool['id']]
|
||||||
|
params['recordOption'] = 'pools'
|
||||||
|
params['geolocation'] = {
|
||||||
|
'geoipUserRegion': [pool['geofilter']]
|
||||||
|
}
|
||||||
|
params.pop('roundRobin', None)
|
||||||
|
self.log.debug(
|
||||||
|
"Creating record %s %s",
|
||||||
|
new.zone.name,
|
||||||
|
new._type)
|
||||||
|
self._client.record_create(
|
||||||
|
new.zone.name,
|
||||||
|
new._type,
|
||||||
|
params)
|
||||||
|
|
||||||
def _apply_Delete(self, change):
|
def _apply_Update(self, change, domain_name):
|
||||||
|
self._apply_Delete(change, domain_name)
|
||||||
|
self._apply_Create(change, domain_name)
|
||||||
|
|
||||||
|
def _apply_Delete(self, change, domain_name):
|
||||||
existing = change.existing
|
existing = change.existing
|
||||||
zone = existing.zone
|
zone = existing.zone
|
||||||
|
|
||||||
|
# if it is dynamic pools record, we need to delete World Default last
|
||||||
|
world_default_record = None
|
||||||
|
|
||||||
for record in self.zone_records(zone):
|
for record in self.zone_records(zone):
|
||||||
if existing.name == record['name'] and \
|
if existing.name == record['name'] and \
|
||||||
existing._type == record['type']:
|
existing._type == record['type']:
|
||||||
self._client.record_delete(zone.name, record['type'],
|
|
||||||
|
# handle dynamic record
|
||||||
|
if record['recordOption'] == 'pools':
|
||||||
|
if record['geolocation'] is None:
|
||||||
|
world_default_record = record
|
||||||
|
else:
|
||||||
|
if record['geolocation']['geoipFilter'] == 1:
|
||||||
|
world_default_record = record
|
||||||
|
else:
|
||||||
|
# delete record
|
||||||
|
self.log.debug(
|
||||||
|
"Deleting record %s %s",
|
||||||
|
zone.name,
|
||||||
|
record['type'])
|
||||||
|
self._client.record_delete(
|
||||||
|
zone.name,
|
||||||
|
record['type'],
|
||||||
record['id'])
|
record['id'])
|
||||||
|
# delete geofilter
|
||||||
|
self.log.debug(
|
||||||
|
"Deleting geofilter %s",
|
||||||
|
zone.name)
|
||||||
|
self._client.geofilter_delete(
|
||||||
|
record['geolocation']['geoipFilter'])
|
||||||
|
|
||||||
|
# delete pool
|
||||||
|
self.log.debug(
|
||||||
|
"Deleting pool %s %s",
|
||||||
|
zone.name,
|
||||||
|
record['type'])
|
||||||
|
self._client.pool_delete(
|
||||||
|
record['type'],
|
||||||
|
record['pools'][0])
|
||||||
|
|
||||||
|
# for all the rest records
|
||||||
|
else:
|
||||||
|
self._client.record_delete(
|
||||||
|
zone.name, record['type'], record['id'])
|
||||||
|
# delete World Default
|
||||||
|
if world_default_record:
|
||||||
|
# delete record
|
||||||
|
self.log.debug(
|
||||||
|
"Deleting record %s %s",
|
||||||
|
zone.name,
|
||||||
|
world_default_record['type']
|
||||||
|
)
|
||||||
|
self._client.record_delete(
|
||||||
|
zone.name,
|
||||||
|
world_default_record['type'],
|
||||||
|
world_default_record['id']
|
||||||
|
)
|
||||||
|
# delete pool
|
||||||
|
self.log.debug(
|
||||||
|
"Deleting pool %s %s",
|
||||||
|
zone.name,
|
||||||
|
world_default_record['type']
|
||||||
|
)
|
||||||
|
self._client.pool_delete(
|
||||||
|
world_default_record['type'],
|
||||||
|
world_default_record['pools'][0]
|
||||||
|
)
|
||||||
|
|
||||||
def _apply(self, plan):
|
def _apply(self, plan):
|
||||||
desired = plan.desired
|
desired = plan.desired
|
||||||
@@ -576,7 +880,9 @@ class ConstellixProvider(BaseProvider):
|
|||||||
|
|
||||||
for change in changes:
|
for change in changes:
|
||||||
class_name = change.__class__.__name__
|
class_name = change.__class__.__name__
|
||||||
getattr(self, '_apply_{}'.format(class_name))(change)
|
getattr(self, f'_apply_{class_name}')(
|
||||||
|
change,
|
||||||
|
desired.name)
|
||||||
|
|
||||||
# Clear out the cache if any
|
# Clear out the cache if any
|
||||||
self._zone_records.pop(desired.name, None)
|
self._zone_records.pop(desired.name, None)
|
||||||
|
|||||||
34
tests/fixtures/constellix-geofilters.json
vendored
Normal file
34
tests/fixtures/constellix-geofilters.json
vendored
Normal file
@@ -0,0 +1,34 @@
|
|||||||
|
[
|
||||||
|
{
|
||||||
|
"id": 6303,
|
||||||
|
"name": "some.other",
|
||||||
|
"filterRulesLimit": 100,
|
||||||
|
"createdTs": "2021-08-19T14:47:47Z",
|
||||||
|
"modifiedTs": "2021-08-19T14:47:47Z",
|
||||||
|
"geoipContinents": ["AS", "OC"],
|
||||||
|
"geoipCountries": ["ES", "SE", "UA"],
|
||||||
|
"regions": [
|
||||||
|
{
|
||||||
|
"continentCode": "NA",
|
||||||
|
"countryCode": "CA",
|
||||||
|
"regionCode": "NL"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"id": 5303,
|
||||||
|
"name": "unit.tests.:www.dynamic:A:one",
|
||||||
|
"filterRulesLimit": 100,
|
||||||
|
"createdTs": "2021-08-19T14:47:47Z",
|
||||||
|
"modifiedTs": "2021-08-19T14:47:47Z",
|
||||||
|
"geoipContinents": ["AS", "OC"],
|
||||||
|
"geoipCountries": ["ES", "SE", "UA"],
|
||||||
|
"regions": [
|
||||||
|
{
|
||||||
|
"continentCode": "NA",
|
||||||
|
"countryCode": "CA",
|
||||||
|
"regionCode": "NL"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
]
|
||||||
30
tests/fixtures/constellix-pools.json
vendored
30
tests/fixtures/constellix-pools.json
vendored
@@ -28,5 +28,35 @@
|
|||||||
"weight": 1
|
"weight": 1
|
||||||
}
|
}
|
||||||
]
|
]
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"id": 1808522,
|
||||||
|
"name": "unit.tests.:www.dynamic:A:one",
|
||||||
|
"type": "A",
|
||||||
|
"numReturn": 1,
|
||||||
|
"minAvailableFailover": 1,
|
||||||
|
"createdTs": "2020-09-12T00:44:35Z",
|
||||||
|
"modifiedTs": "2020-09-12T00:44:35Z",
|
||||||
|
"appliedDomains": [
|
||||||
|
{
|
||||||
|
"id": 123123,
|
||||||
|
"name": "unit.tests",
|
||||||
|
"recordOption": "pools"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"appliedTemplates": null,
|
||||||
|
"unlinkedDomains": [],
|
||||||
|
"unlinkedTemplates": null,
|
||||||
|
"itoEnabled": false,
|
||||||
|
"values": [
|
||||||
|
{
|
||||||
|
"value": "1.2.3.6",
|
||||||
|
"weight": 1
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"value": "1.2.3.7",
|
||||||
|
"weight": 1
|
||||||
|
}
|
||||||
|
]
|
||||||
}
|
}
|
||||||
]
|
]
|
||||||
156
tests/fixtures/constellix-records.json
vendored
156
tests/fixtures/constellix-records.json
vendored
@@ -64,62 +64,6 @@
|
|||||||
"roundRobinFailover": [],
|
"roundRobinFailover": [],
|
||||||
"pools": [],
|
"pools": [],
|
||||||
"poolsDetail": []
|
"poolsDetail": []
|
||||||
}, {
|
|
||||||
"id": 1898527,
|
|
||||||
"type": "SRV",
|
|
||||||
"recordType": "srv",
|
|
||||||
"name": "_imap._tcp",
|
|
||||||
"recordOption": "roundRobin",
|
|
||||||
"noAnswer": false,
|
|
||||||
"note": "",
|
|
||||||
"ttl": 600,
|
|
||||||
"gtdRegion": 1,
|
|
||||||
"parentId": 123123,
|
|
||||||
"parent": "domain",
|
|
||||||
"source": "Domain",
|
|
||||||
"modifiedTs": 1565149714387,
|
|
||||||
"value": [{
|
|
||||||
"value": ".",
|
|
||||||
"priority": 0,
|
|
||||||
"weight": 0,
|
|
||||||
"port": 0,
|
|
||||||
"disableFlag": false
|
|
||||||
}],
|
|
||||||
"roundRobin": [{
|
|
||||||
"value": ".",
|
|
||||||
"priority": 0,
|
|
||||||
"weight": 0,
|
|
||||||
"port": 0,
|
|
||||||
"disableFlag": false
|
|
||||||
}]
|
|
||||||
}, {
|
|
||||||
"id": 1898528,
|
|
||||||
"type": "SRV",
|
|
||||||
"recordType": "srv",
|
|
||||||
"name": "_pop3._tcp",
|
|
||||||
"recordOption": "roundRobin",
|
|
||||||
"noAnswer": false,
|
|
||||||
"note": "",
|
|
||||||
"ttl": 600,
|
|
||||||
"gtdRegion": 1,
|
|
||||||
"parentId": 123123,
|
|
||||||
"parent": "domain",
|
|
||||||
"source": "Domain",
|
|
||||||
"modifiedTs": 1565149714387,
|
|
||||||
"value": [{
|
|
||||||
"value": ".",
|
|
||||||
"priority": 0,
|
|
||||||
"weight": 0,
|
|
||||||
"port": 0,
|
|
||||||
"disableFlag": false
|
|
||||||
}],
|
|
||||||
"roundRobin": [{
|
|
||||||
"value": ".",
|
|
||||||
"priority": 0,
|
|
||||||
"weight": 0,
|
|
||||||
"port": 0,
|
|
||||||
"disableFlag": false
|
|
||||||
}]
|
|
||||||
}, {
|
}, {
|
||||||
"id": 1808527,
|
"id": 1808527,
|
||||||
"type": "SRV",
|
"type": "SRV",
|
||||||
@@ -160,6 +104,62 @@
|
|||||||
"port": 30,
|
"port": 30,
|
||||||
"disableFlag": false
|
"disableFlag": false
|
||||||
}]
|
}]
|
||||||
|
}, {
|
||||||
|
"id": 1808527,
|
||||||
|
"type": "SRV",
|
||||||
|
"recordType": "srv",
|
||||||
|
"name": "_imap._tcp",
|
||||||
|
"recordOption": "roundRobin",
|
||||||
|
"noAnswer": false,
|
||||||
|
"note": "",
|
||||||
|
"ttl": 600,
|
||||||
|
"gtdRegion": 1,
|
||||||
|
"parentId": 123123,
|
||||||
|
"parent": "domain",
|
||||||
|
"source": "Domain",
|
||||||
|
"modifiedTs": 1565149714387,
|
||||||
|
"value": [{
|
||||||
|
"value": ".",
|
||||||
|
"priority": 0,
|
||||||
|
"weight": 0,
|
||||||
|
"port": 0,
|
||||||
|
"disableFlag": false
|
||||||
|
}],
|
||||||
|
"roundRobin": [{
|
||||||
|
"value": ".",
|
||||||
|
"priority": 0,
|
||||||
|
"weight": 0,
|
||||||
|
"port": 0,
|
||||||
|
"disableFlag": false
|
||||||
|
}]
|
||||||
|
}, {
|
||||||
|
"id": 1808527,
|
||||||
|
"type": "SRV",
|
||||||
|
"recordType": "srv",
|
||||||
|
"name": "_pop3._tcp",
|
||||||
|
"recordOption": "roundRobin",
|
||||||
|
"noAnswer": false,
|
||||||
|
"note": "",
|
||||||
|
"ttl": 600,
|
||||||
|
"gtdRegion": 1,
|
||||||
|
"parentId": 123123,
|
||||||
|
"parent": "domain",
|
||||||
|
"source": "Domain",
|
||||||
|
"modifiedTs": 1565149714387,
|
||||||
|
"value": [{
|
||||||
|
"value": ".",
|
||||||
|
"priority": 0,
|
||||||
|
"weight": 0,
|
||||||
|
"port": 0,
|
||||||
|
"disableFlag": false
|
||||||
|
}],
|
||||||
|
"roundRobin": [{
|
||||||
|
"value": ".",
|
||||||
|
"priority": 0,
|
||||||
|
"weight": 0,
|
||||||
|
"port": 0,
|
||||||
|
"disableFlag": false
|
||||||
|
}]
|
||||||
}, {
|
}, {
|
||||||
"id": 1808515,
|
"id": 1808515,
|
||||||
"type": "AAAA",
|
"type": "AAAA",
|
||||||
@@ -630,7 +630,9 @@
|
|||||||
"modifiedTs": 1565150090588,
|
"modifiedTs": 1565150090588,
|
||||||
"value": [],
|
"value": [],
|
||||||
"roundRobin": [],
|
"roundRobin": [],
|
||||||
"geolocation": null,
|
"geolocation": {
|
||||||
|
"geoipFilter": 1
|
||||||
|
},
|
||||||
"recordFailover": {
|
"recordFailover": {
|
||||||
"disabled": false,
|
"disabled": false,
|
||||||
"failoverType": 1,
|
"failoverType": 1,
|
||||||
@@ -651,4 +653,44 @@
|
|||||||
"id": 1808521,
|
"id": 1808521,
|
||||||
"name": "unit.tests.:www.dynamic:A:two"
|
"name": "unit.tests.:www.dynamic:A:two"
|
||||||
}]
|
}]
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"id": 1808521,
|
||||||
|
"type": "A",
|
||||||
|
"recordType": "a",
|
||||||
|
"name": "www.dynamic",
|
||||||
|
"recordOption": "pools",
|
||||||
|
"noAnswer": false,
|
||||||
|
"note": "",
|
||||||
|
"ttl": 300,
|
||||||
|
"gtdRegion": 1,
|
||||||
|
"parentId": 123123,
|
||||||
|
"parent": "domain",
|
||||||
|
"source": "Domain",
|
||||||
|
"modifiedTs": 1565150090588,
|
||||||
|
"value": [],
|
||||||
|
"roundRobin": [],
|
||||||
|
"geolocation": {
|
||||||
|
"geoipFilter": 5303
|
||||||
|
},
|
||||||
|
"recordFailover": {
|
||||||
|
"disabled": false,
|
||||||
|
"failoverType": 1,
|
||||||
|
"failoverTypeStr": "Normal (always lowest level)",
|
||||||
|
"values": []
|
||||||
|
},
|
||||||
|
"failover": {
|
||||||
|
"disabled": false,
|
||||||
|
"failoverType": 1,
|
||||||
|
"failoverTypeStr": "Normal (always lowest level)",
|
||||||
|
"values": []
|
||||||
|
},
|
||||||
|
"roundRobinFailover": [],
|
||||||
|
"pools": [
|
||||||
|
1808522
|
||||||
|
],
|
||||||
|
"poolsDetail": [{
|
||||||
|
"id": 1808522,
|
||||||
|
"name": "unit.tests.:www.dynamic:A:one"
|
||||||
|
}]
|
||||||
}]
|
}]
|
||||||
|
|||||||
@@ -42,17 +42,11 @@ class TestConstellixProvider(TestCase):
|
|||||||
'value': 'aname.unit.tests.'
|
'value': 'aname.unit.tests.'
|
||||||
}))
|
}))
|
||||||
|
|
||||||
expected.add_record(Record.new(expected, 'sub', {
|
|
||||||
'ttl': 1800,
|
|
||||||
'type': 'ALIAS',
|
|
||||||
'value': 'aname.unit.tests.'
|
|
||||||
}))
|
|
||||||
|
|
||||||
# Add a dynamic record
|
# Add a dynamic record
|
||||||
expected.add_record(Record.new(expected, 'www.dynamic', {
|
expected.add_record(Record.new(expected, 'www.dynamic', {
|
||||||
'ttl': 300,
|
'ttl': 300,
|
||||||
'type': 'A',
|
'type': 'A',
|
||||||
'value': [
|
'values': [
|
||||||
'1.2.3.4',
|
'1.2.3.4',
|
||||||
'1.2.3.5'
|
'1.2.3.5'
|
||||||
],
|
],
|
||||||
@@ -79,6 +73,78 @@ class TestConstellixProvider(TestCase):
|
|||||||
expected._remove_record(record)
|
expected._remove_record(record)
|
||||||
break
|
break
|
||||||
|
|
||||||
|
expected_dynamic = Zone('unit.tests.', [])
|
||||||
|
source = YamlProvider('test', join(dirname(__file__), 'config'))
|
||||||
|
source.populate(expected_dynamic)
|
||||||
|
|
||||||
|
# Our test suite differs a bit, add our NS and remove the simple one
|
||||||
|
expected_dynamic.add_record(Record.new(expected_dynamic, 'under', {
|
||||||
|
'ttl': 3600,
|
||||||
|
'type': 'NS',
|
||||||
|
'values': [
|
||||||
|
'ns1.unit.tests.',
|
||||||
|
'ns2.unit.tests.',
|
||||||
|
]
|
||||||
|
}))
|
||||||
|
|
||||||
|
# Add some ALIAS records
|
||||||
|
expected_dynamic.add_record(Record.new(expected_dynamic, '', {
|
||||||
|
'ttl': 1800,
|
||||||
|
'type': 'ALIAS',
|
||||||
|
'value': 'aname.unit.tests.'
|
||||||
|
}))
|
||||||
|
|
||||||
|
# Add a dynamic record
|
||||||
|
expected_dynamic.add_record(Record.new(expected_dynamic, 'www.dynamic', {
|
||||||
|
'ttl': 300,
|
||||||
|
'type': 'A',
|
||||||
|
'values': [
|
||||||
|
'1.2.3.4',
|
||||||
|
'1.2.3.5'
|
||||||
|
],
|
||||||
|
'dynamic': {
|
||||||
|
'pools': {
|
||||||
|
'one': {
|
||||||
|
'fallback': 'two',
|
||||||
|
'values': [{
|
||||||
|
'value': '1.2.3.6',
|
||||||
|
'weight': 1
|
||||||
|
}, {
|
||||||
|
'value': '1.2.3.7',
|
||||||
|
'weight': 1
|
||||||
|
}],
|
||||||
|
},
|
||||||
|
'two': {
|
||||||
|
'values': [{
|
||||||
|
'value': '1.2.3.4',
|
||||||
|
'weight': 1
|
||||||
|
}, {
|
||||||
|
'value': '1.2.3.5',
|
||||||
|
'weight': 1
|
||||||
|
}],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
'rules': [{
|
||||||
|
'geos': [
|
||||||
|
'AS',
|
||||||
|
'EU-ES',
|
||||||
|
'EU-UA',
|
||||||
|
'EU-SE',
|
||||||
|
'NA-CA-NL',
|
||||||
|
'OC'
|
||||||
|
],
|
||||||
|
'pool': 'one'
|
||||||
|
}, {
|
||||||
|
'pool': 'two',
|
||||||
|
}],
|
||||||
|
}
|
||||||
|
}))
|
||||||
|
|
||||||
|
for record in list(expected_dynamic.records):
|
||||||
|
if record.name == 'sub' and record._type == 'NS':
|
||||||
|
expected_dynamic._remove_record(record)
|
||||||
|
break
|
||||||
|
|
||||||
def test_populate(self):
|
def test_populate(self):
|
||||||
provider = ConstellixProvider('test', 'api', 'secret')
|
provider = ConstellixProvider('test', 'api', 'secret')
|
||||||
|
|
||||||
@@ -126,24 +192,28 @@ class TestConstellixProvider(TestCase):
|
|||||||
with requests_mock() as mock:
|
with requests_mock() as mock:
|
||||||
base = 'https://api.dns.constellix.com/v1'
|
base = 'https://api.dns.constellix.com/v1'
|
||||||
with open('tests/fixtures/constellix-domains.json') as fh:
|
with open('tests/fixtures/constellix-domains.json') as fh:
|
||||||
mock.get('{}{}'.format(base, '/domains'), text=fh.read())
|
mock.get('{}{}'.format(base, '/domains'),
|
||||||
|
text=fh.read())
|
||||||
with open('tests/fixtures/constellix-records.json') as fh:
|
with open('tests/fixtures/constellix-records.json') as fh:
|
||||||
mock.get('{}{}'.format(base, '/domains/123123/records'),
|
mock.get('{}{}'.format(base, '/domains/123123/records'),
|
||||||
text=fh.read())
|
text=fh.read())
|
||||||
with open('tests/fixtures/constellix-pools.json') as fh:
|
with open('tests/fixtures/constellix-pools.json') as fh:
|
||||||
mock.get('{}{}'.format(base, '/pools/A'),
|
mock.get('{}{}'.format(base, '/pools/A'),
|
||||||
text=fh.read())
|
text=fh.read())
|
||||||
|
with open('tests/fixtures/constellix-geofilters.json') as fh:
|
||||||
|
mock.get('{}{}'.format(base, '/geoFilters'),
|
||||||
|
text=fh.read())
|
||||||
|
|
||||||
zone = Zone('unit.tests.', [])
|
zone = Zone('unit.tests.', [])
|
||||||
provider.populate(zone)
|
provider.populate(zone)
|
||||||
self.assertEquals(16, len(zone.records))
|
self.assertEquals(17, len(zone.records))
|
||||||
changes = self.expected.changes(zone, provider)
|
changes = self.expected_dynamic.changes(zone, provider)
|
||||||
self.assertEquals(0, len(changes))
|
self.assertEquals(0, len(changes))
|
||||||
|
|
||||||
# 2nd populate makes no network calls/all from cache
|
# 2nd populate makes no network calls/all from cache
|
||||||
again = Zone('unit.tests.', [])
|
again = Zone('unit.tests.', [])
|
||||||
provider.populate(again)
|
provider.populate(again)
|
||||||
self.assertEquals(16, len(again.records))
|
self.assertEquals(17, len(again.records))
|
||||||
|
|
||||||
# bust the cache
|
# bust the cache
|
||||||
del provider._zone_records[zone.name]
|
del provider._zone_records[zone.name]
|
||||||
@@ -225,7 +295,7 @@ class TestConstellixProvider(TestCase):
|
|||||||
}),
|
}),
|
||||||
])
|
])
|
||||||
|
|
||||||
self.assertEquals(21, provider._client._request.call_count)
|
self.assertEquals(22, provider._client._request.call_count)
|
||||||
|
|
||||||
provider._client._request.reset_mock()
|
provider._client._request.reset_mock()
|
||||||
|
|
||||||
@@ -262,6 +332,7 @@ class TestConstellixProvider(TestCase):
|
|||||||
"id": 1808520,
|
"id": 1808520,
|
||||||
"type": "A",
|
"type": "A",
|
||||||
"name": "www.dynamic",
|
"name": "www.dynamic",
|
||||||
|
"geolocation": None,
|
||||||
"recordOption": "pools",
|
"recordOption": "pools",
|
||||||
"ttl": 300,
|
"ttl": 300,
|
||||||
"value": [],
|
"value": [],
|
||||||
@@ -289,8 +360,16 @@ class TestConstellixProvider(TestCase):
|
|||||||
|
|
||||||
# Domain exists, we don't care about return
|
# Domain exists, we don't care about return
|
||||||
resp.json.side_effect = [
|
resp.json.side_effect = [
|
||||||
['{}'],
|
[], # no domains returned during populate
|
||||||
['{}'],
|
[{
|
||||||
|
'id': 123123,
|
||||||
|
'name': 'unit.tests'
|
||||||
|
}], # domain created in apply
|
||||||
|
[], # No pools returned during populate
|
||||||
|
[{
|
||||||
|
"id": 1808521,
|
||||||
|
"name": "unit.tests.:www.dynamic:A:one"
|
||||||
|
}] # pool created in apply
|
||||||
]
|
]
|
||||||
|
|
||||||
wanted = Zone('unit.tests.', [])
|
wanted = Zone('unit.tests.', [])
|
||||||
@@ -303,7 +382,7 @@ class TestConstellixProvider(TestCase):
|
|||||||
wanted.add_record(Record.new(wanted, 'www.dynamic', {
|
wanted.add_record(Record.new(wanted, 'www.dynamic', {
|
||||||
'ttl': 300,
|
'ttl': 300,
|
||||||
'type': 'A',
|
'type': 'A',
|
||||||
'value': [
|
'values': [
|
||||||
'1.2.3.4'
|
'1.2.3.4'
|
||||||
],
|
],
|
||||||
'dynamic': {
|
'dynamic': {
|
||||||
@@ -340,17 +419,393 @@ class TestConstellixProvider(TestCase):
|
|||||||
'numReturn': 1,
|
'numReturn': 1,
|
||||||
'minAvailableFailover': 1,
|
'minAvailableFailover': 1,
|
||||||
'ttl': 300,
|
'ttl': 300,
|
||||||
'id': 1808521,
|
|
||||||
'values': [{
|
'values': [{
|
||||||
"value": "1.2.3.4",
|
"value": "1.2.3.4",
|
||||||
"weight": 1
|
"weight": 1
|
||||||
}]
|
}],
|
||||||
|
'id': 1808521,
|
||||||
|
'geofilter': 1
|
||||||
}),
|
}),
|
||||||
call('DELETE', '/domains/123123/records/A/11189897'),
|
call('DELETE', '/domains/123123/records/A/11189897'),
|
||||||
call('DELETE', '/domains/123123/records/A/11189898'),
|
call('DELETE', '/domains/123123/records/A/11189898'),
|
||||||
call('DELETE', '/domains/123123/records/ANAME/11189899'),
|
call('DELETE', '/domains/123123/records/ANAME/11189899'),
|
||||||
], any_order=True)
|
], any_order=True)
|
||||||
|
|
||||||
|
def test_apply_dunamic(self):
|
||||||
|
provider = ConstellixProvider('test', 'api', 'secret')
|
||||||
|
|
||||||
|
resp = Mock()
|
||||||
|
resp.json = Mock()
|
||||||
|
provider._client._request = Mock(return_value=resp)
|
||||||
|
|
||||||
|
# non-existent domain, create everything
|
||||||
|
resp.json.side_effect = [
|
||||||
|
[], # no domains returned during populate
|
||||||
|
[{
|
||||||
|
'id': 123123,
|
||||||
|
'name': 'unit.tests'
|
||||||
|
}], # domain created in apply
|
||||||
|
[], # No pools returned during populate
|
||||||
|
[{
|
||||||
|
"id": 1808521,
|
||||||
|
"name": "unit.tests.:www.dynamic:A:one"
|
||||||
|
}], # pool created in apply
|
||||||
|
[], # no geofilters returned during populate
|
||||||
|
[{
|
||||||
|
"id": 5303,
|
||||||
|
"name": "unit.tests.:www.dynamic:A:one",
|
||||||
|
"filterRulesLimit": 100,
|
||||||
|
"geoipContinents": ["AS", "OC"],
|
||||||
|
"geoipCountries": ["ES", "SE", "UA"],
|
||||||
|
"regions": [
|
||||||
|
{
|
||||||
|
"continentCode": "NA",
|
||||||
|
"countryCode": "CA",
|
||||||
|
"regionCode": "NL"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}], # geofilters created in applly
|
||||||
|
[{
|
||||||
|
"id": 1808520,
|
||||||
|
"name": "unit.tests.:www.dynamic:A:two",
|
||||||
|
}], # pool created in apply
|
||||||
|
{
|
||||||
|
'id': 123123,
|
||||||
|
'name': 'unit.tests',
|
||||||
|
'hasGeoIP': False
|
||||||
|
}, # domain listed for enabling geo
|
||||||
|
[] # enabling geo
|
||||||
|
]
|
||||||
|
|
||||||
|
plan = provider.plan(self.expected_dynamic)
|
||||||
|
|
||||||
|
# No root NS, no ignored, no excluded, no unsupported
|
||||||
|
n = len(self.expected_dynamic.records) - 8
|
||||||
|
self.assertEquals(n, len(plan.changes))
|
||||||
|
self.assertEquals(n, provider.apply(plan))
|
||||||
|
|
||||||
|
provider._client._request.assert_has_calls([
|
||||||
|
# get all domains to build the cache
|
||||||
|
call('GET', '/domains'),
|
||||||
|
# created the domain
|
||||||
|
call('POST', '/domains', data={'names': ['unit.tests']})
|
||||||
|
])
|
||||||
|
#
|
||||||
|
# Check we tried to get our pool
|
||||||
|
provider._client._request.assert_has_calls([
|
||||||
|
call('GET', '/pools/A'),
|
||||||
|
call('POST', '/pools/A', data={
|
||||||
|
'name': 'unit.tests.:www.dynamic:A:one',
|
||||||
|
'type': 'A',
|
||||||
|
'numReturn': 1,
|
||||||
|
'minAvailableFailover': 1,
|
||||||
|
'ttl': 300,
|
||||||
|
'values': [{
|
||||||
|
'value': '1.2.3.6',
|
||||||
|
'weight': 1
|
||||||
|
}, {
|
||||||
|
'value': '1.2.3.7',
|
||||||
|
'weight': 1}]
|
||||||
|
}),
|
||||||
|
call('GET', '/geoFilters'),
|
||||||
|
call('POST', '/geoFilters', data={
|
||||||
|
'filterRulesLimit': 100,
|
||||||
|
'name': 'unit.tests.:www.dynamic:A:one',
|
||||||
|
'geoipContinents': ['AS', 'OC'],
|
||||||
|
'geoipCountries': ['ES', 'SE', 'UA'],
|
||||||
|
'regions': [{
|
||||||
|
'continentCode': 'NA',
|
||||||
|
'countryCode': 'CA',
|
||||||
|
'regionCode': 'NL'}]
|
||||||
|
}),
|
||||||
|
call('POST', '/pools/A', data={
|
||||||
|
'name': 'unit.tests.:www.dynamic:A:two',
|
||||||
|
'type': 'A',
|
||||||
|
'numReturn': 1,
|
||||||
|
'minAvailableFailover': 1,
|
||||||
|
'ttl': 300,
|
||||||
|
'values': [{
|
||||||
|
'value': '1.2.3.4',
|
||||||
|
'weight': 1
|
||||||
|
}, {
|
||||||
|
'value': '1.2.3.5',
|
||||||
|
'weight': 1}]
|
||||||
|
})
|
||||||
|
])
|
||||||
|
|
||||||
|
# These two checks are broken up so that ordering doesn't break things.
|
||||||
|
# Python3 doesn't make the calls in a consistent order so different
|
||||||
|
# things follow the GET / on different runs
|
||||||
|
provider._client._request.assert_has_calls([
|
||||||
|
call('POST', '/domains/123123/records/SRV', data={
|
||||||
|
'roundRobin': [{
|
||||||
|
'priority': 10,
|
||||||
|
'weight': 20,
|
||||||
|
'value': 'foo-1.unit.tests.',
|
||||||
|
'port': 30
|
||||||
|
}, {
|
||||||
|
'priority': 12,
|
||||||
|
'weight': 20,
|
||||||
|
'value': 'foo-2.unit.tests.',
|
||||||
|
'port': 30
|
||||||
|
}],
|
||||||
|
'name': '_srv._tcp',
|
||||||
|
'ttl': 600,
|
||||||
|
}),
|
||||||
|
])
|
||||||
|
|
||||||
|
self.assertEquals(28, provider._client._request.call_count)
|
||||||
|
|
||||||
|
provider._client._request.reset_mock()
|
||||||
|
|
||||||
|
provider._client.records = Mock(return_value=[
|
||||||
|
{
|
||||||
|
'id': 11189897,
|
||||||
|
'type': 'A',
|
||||||
|
'name': 'www',
|
||||||
|
'ttl': 300,
|
||||||
|
'recordOption': 'roundRobin',
|
||||||
|
'value': [
|
||||||
|
'1.2.3.4',
|
||||||
|
'2.2.3.4',
|
||||||
|
]
|
||||||
|
}, {
|
||||||
|
'id': 11189898,
|
||||||
|
'type': 'A',
|
||||||
|
'name': 'ttl',
|
||||||
|
'ttl': 600,
|
||||||
|
'recordOption': 'roundRobin',
|
||||||
|
'value': [
|
||||||
|
'3.2.3.4'
|
||||||
|
]
|
||||||
|
}, {
|
||||||
|
'id': 11189899,
|
||||||
|
'type': 'ALIAS',
|
||||||
|
'name': 'alias',
|
||||||
|
'ttl': 600,
|
||||||
|
'recordOption': 'roundRobin',
|
||||||
|
'value': [{
|
||||||
|
'value': 'aname.unit.tests.'
|
||||||
|
}]
|
||||||
|
}, {
|
||||||
|
"id": 1808520,
|
||||||
|
"type": "A",
|
||||||
|
"name": "www.dynamic",
|
||||||
|
"geolocation": {
|
||||||
|
"geoipFilter": 1
|
||||||
|
},
|
||||||
|
"recordOption": "pools",
|
||||||
|
"ttl": 300,
|
||||||
|
"value": [],
|
||||||
|
"pools": [
|
||||||
|
1808521
|
||||||
|
]
|
||||||
|
}, {
|
||||||
|
"id": 1808521,
|
||||||
|
"type": "A",
|
||||||
|
"name": "www.dynamic",
|
||||||
|
"geolocation": {
|
||||||
|
"geoipFilter": 5303
|
||||||
|
},
|
||||||
|
"recordOption": "pools",
|
||||||
|
"ttl": 300,
|
||||||
|
"value": [],
|
||||||
|
"pools": [
|
||||||
|
1808522
|
||||||
|
]
|
||||||
|
}
|
||||||
|
])
|
||||||
|
|
||||||
|
provider._client.pools = Mock(return_value=[
|
||||||
|
{
|
||||||
|
"id": 1808521,
|
||||||
|
"name": "unit.tests.:www.dynamic:A:two",
|
||||||
|
"type": "A",
|
||||||
|
"values": [
|
||||||
|
{
|
||||||
|
"value": "1.2.3.4",
|
||||||
|
"weight": 1
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"value": "1.2.3.5",
|
||||||
|
"weight": 1
|
||||||
|
}
|
||||||
|
]
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"id": 1808522,
|
||||||
|
"name": "unit.tests.:www.dynamic:A:one",
|
||||||
|
"type": "A",
|
||||||
|
"values": [
|
||||||
|
{
|
||||||
|
"value": "1.2.3.6",
|
||||||
|
"weight": 1
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"value": "1.2.3.7",
|
||||||
|
"weight": 1
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
])
|
||||||
|
|
||||||
|
provider._client.geofilters = Mock(return_value=[
|
||||||
|
{
|
||||||
|
"id": 5303,
|
||||||
|
"name": "unit.tests.:www.dynamic:A:one",
|
||||||
|
"filterRulesLimit": 100,
|
||||||
|
"geoipContinents": ["AS", "OC"],
|
||||||
|
"geoipCountries": ["ES", "SE", "UA"],
|
||||||
|
"regions": [
|
||||||
|
{
|
||||||
|
"continentCode": "NA",
|
||||||
|
"countryCode": "CA",
|
||||||
|
"regionCode": "NL"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
])
|
||||||
|
|
||||||
|
# Domain exists, we don't care about return
|
||||||
|
resp.json.side_effect = [
|
||||||
|
[],
|
||||||
|
[],
|
||||||
|
[],
|
||||||
|
[],
|
||||||
|
{
|
||||||
|
'id': 123123,
|
||||||
|
'name': 'unit.tests',
|
||||||
|
'hasGeoIP': True
|
||||||
|
} # domain listed for enabling geo
|
||||||
|
]
|
||||||
|
|
||||||
|
wanted = Zone('unit.tests.', [])
|
||||||
|
wanted.add_record(Record.new(wanted, 'ttl', {
|
||||||
|
'ttl': 300,
|
||||||
|
'type': 'A',
|
||||||
|
'value': '3.2.3.4'
|
||||||
|
}))
|
||||||
|
|
||||||
|
wanted.add_record(Record.new(wanted, 'www.dynamic', {
|
||||||
|
'ttl': 300,
|
||||||
|
'type': 'A',
|
||||||
|
'values': [
|
||||||
|
'1.2.3.4'
|
||||||
|
],
|
||||||
|
'dynamic': {
|
||||||
|
'pools': {
|
||||||
|
'one': {
|
||||||
|
'fallback': 'two',
|
||||||
|
'values': [{
|
||||||
|
'value': '1.2.3.6',
|
||||||
|
'weight': 1
|
||||||
|
}, {
|
||||||
|
'value': '1.2.3.7',
|
||||||
|
'weight': 1
|
||||||
|
}],
|
||||||
|
},
|
||||||
|
'two': {
|
||||||
|
'values': [{
|
||||||
|
'value': '1.2.3.4',
|
||||||
|
'weight': 1
|
||||||
|
}],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
'rules': [{
|
||||||
|
'geos': [
|
||||||
|
'AS',
|
||||||
|
'EU-ES',
|
||||||
|
'EU-UA',
|
||||||
|
'EU-SE',
|
||||||
|
'NA-CA-NL',
|
||||||
|
'OC'
|
||||||
|
],
|
||||||
|
'pool': 'one'
|
||||||
|
}, {
|
||||||
|
'pool': 'two',
|
||||||
|
}],
|
||||||
|
},
|
||||||
|
}))
|
||||||
|
|
||||||
|
plan = provider.plan(wanted)
|
||||||
|
self.assertEquals(4, len(plan.changes))
|
||||||
|
self.assertEquals(4, provider.apply(plan))
|
||||||
|
|
||||||
|
# recreate for update, and deletes for the 2 parts of the other
|
||||||
|
provider._client._request.assert_has_calls([
|
||||||
|
call('POST', '/domains/123123/records/A', data={
|
||||||
|
'roundRobin': [{
|
||||||
|
'value': '3.2.3.4'
|
||||||
|
}],
|
||||||
|
'name': 'ttl',
|
||||||
|
'ttl': 300
|
||||||
|
}),
|
||||||
|
|
||||||
|
call('DELETE', '/domains/123123/records/A/1808521'),
|
||||||
|
call('DELETE', '/geoFilters/5303'),
|
||||||
|
call('DELETE', '/pools/A/1808522'),
|
||||||
|
call('DELETE', '/domains/123123/records/A/1808520'),
|
||||||
|
call('DELETE', '/pools/A/1808521'),
|
||||||
|
call('DELETE', '/domains/123123/records/ANAME/11189899'),
|
||||||
|
|
||||||
|
call('PUT', '/pools/A/1808522', data={
|
||||||
|
'name': 'unit.tests.:www.dynamic:A:one',
|
||||||
|
'type': 'A',
|
||||||
|
'numReturn': 1,
|
||||||
|
'minAvailableFailover': 1,
|
||||||
|
'ttl': 300,
|
||||||
|
'values': [
|
||||||
|
{'value': '1.2.3.6', 'weight': 1},
|
||||||
|
{'value': '1.2.3.7', 'weight': 1}],
|
||||||
|
'id': 1808522,
|
||||||
|
'geofilter': 5303
|
||||||
|
}),
|
||||||
|
|
||||||
|
call('PUT', '/geoFilters/5303', data={
|
||||||
|
'filterRulesLimit': 100,
|
||||||
|
'name': 'unit.tests.:www.dynamic:A:one',
|
||||||
|
'geoipContinents': ['AS', 'OC'],
|
||||||
|
'geoipCountries': ['ES', 'SE', 'UA'],
|
||||||
|
'regions': [{
|
||||||
|
'continentCode': 'NA',
|
||||||
|
'countryCode': 'CA',
|
||||||
|
'regionCode': 'NL'}],
|
||||||
|
'id': 5303
|
||||||
|
}),
|
||||||
|
|
||||||
|
call('PUT', '/pools/A/1808521', data={
|
||||||
|
'name': 'unit.tests.:www.dynamic:A:two',
|
||||||
|
'type': 'A',
|
||||||
|
'numReturn': 1,
|
||||||
|
'minAvailableFailover': 1,
|
||||||
|
'ttl': 300,
|
||||||
|
'values': [{'value': '1.2.3.4', 'weight': 1}],
|
||||||
|
'id': 1808521,
|
||||||
|
'geofilter': 1
|
||||||
|
}),
|
||||||
|
|
||||||
|
call('GET', '/domains/123123'),
|
||||||
|
call('POST', '/domains/123123/records/A', data={
|
||||||
|
'name': 'www.dynamic',
|
||||||
|
'ttl': 300,
|
||||||
|
'pools': [1808522],
|
||||||
|
'recordOption': 'pools',
|
||||||
|
'geolocation': {
|
||||||
|
'geoipUserRegion': [5303]
|
||||||
|
}
|
||||||
|
}),
|
||||||
|
|
||||||
|
call('POST', '/domains/123123/records/A', data={
|
||||||
|
'name': 'www.dynamic',
|
||||||
|
'ttl': 300,
|
||||||
|
'pools': [1808522],
|
||||||
|
'recordOption': 'pools',
|
||||||
|
'geolocation': {
|
||||||
|
'geoipUserRegion': [5303]
|
||||||
|
}
|
||||||
|
})
|
||||||
|
], any_order=True)
|
||||||
|
|
||||||
def test_dynamic_record_failures(self):
|
def test_dynamic_record_failures(self):
|
||||||
provider = ConstellixProvider('test', 'api', 'secret')
|
provider = ConstellixProvider('test', 'api', 'secret')
|
||||||
|
|
||||||
@@ -367,6 +822,7 @@ class TestConstellixProvider(TestCase):
|
|||||||
"id": 1808520,
|
"id": 1808520,
|
||||||
"type": "A",
|
"type": "A",
|
||||||
"name": "www.dynamic",
|
"name": "www.dynamic",
|
||||||
|
"geolocation": None,
|
||||||
"recordOption": "pools",
|
"recordOption": "pools",
|
||||||
"ttl": 300,
|
"ttl": 300,
|
||||||
"value": [],
|
"value": [],
|
||||||
@@ -388,6 +844,8 @@ class TestConstellixProvider(TestCase):
|
|||||||
]
|
]
|
||||||
}])
|
}])
|
||||||
|
|
||||||
|
provider._client.geofilters = Mock(return_value=[])
|
||||||
|
|
||||||
wanted = Zone('unit.tests.', [])
|
wanted = Zone('unit.tests.', [])
|
||||||
|
|
||||||
resp.json.side_effect = [
|
resp.json.side_effect = [
|
||||||
@@ -397,7 +855,7 @@ class TestConstellixProvider(TestCase):
|
|||||||
wanted.add_record(Record.new(wanted, 'www.dynamic', {
|
wanted.add_record(Record.new(wanted, 'www.dynamic', {
|
||||||
'ttl': 300,
|
'ttl': 300,
|
||||||
'type': 'A',
|
'type': 'A',
|
||||||
'value': [
|
'values': [
|
||||||
'1.2.3.4'
|
'1.2.3.4'
|
||||||
],
|
],
|
||||||
'dynamic': {
|
'dynamic': {
|
||||||
@@ -428,16 +886,33 @@ class TestConstellixProvider(TestCase):
|
|||||||
"id": 1808520,
|
"id": 1808520,
|
||||||
"type": "A",
|
"type": "A",
|
||||||
"name": "www.dynamic",
|
"name": "www.dynamic",
|
||||||
|
"geolocation": {
|
||||||
|
"geoipFilter": 1
|
||||||
|
},
|
||||||
"recordOption": "pools",
|
"recordOption": "pools",
|
||||||
"ttl": 300,
|
"ttl": 300,
|
||||||
"value": [],
|
"value": [],
|
||||||
"pools": [
|
"pools": [
|
||||||
1808521
|
1808521
|
||||||
]
|
]
|
||||||
|
}, {
|
||||||
|
"id": 1808521,
|
||||||
|
"type": "A",
|
||||||
|
"name": "www.dynamic",
|
||||||
|
"geolocation": {
|
||||||
|
"geoipFilter": 5303
|
||||||
|
},
|
||||||
|
"recordOption": "pools",
|
||||||
|
"ttl": 300,
|
||||||
|
"value": [],
|
||||||
|
"pools": [
|
||||||
|
1808522
|
||||||
|
]
|
||||||
}
|
}
|
||||||
])
|
])
|
||||||
|
|
||||||
provider._client.pools = Mock(return_value=[{
|
provider._client.pools = Mock(return_value=[
|
||||||
|
{
|
||||||
"id": 1808521,
|
"id": 1808521,
|
||||||
"name": "unit.tests.:www.dynamic:A:two",
|
"name": "unit.tests.:www.dynamic:A:two",
|
||||||
"type": "A",
|
"type": "A",
|
||||||
@@ -445,27 +920,99 @@ class TestConstellixProvider(TestCase):
|
|||||||
{
|
{
|
||||||
"value": "1.2.3.4",
|
"value": "1.2.3.4",
|
||||||
"weight": 1
|
"weight": 1
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"value": "1.2.3.5",
|
||||||
|
"weight": 1
|
||||||
}
|
}
|
||||||
]
|
]
|
||||||
}])
|
},
|
||||||
|
{
|
||||||
|
"id": 1808522,
|
||||||
|
"name": "unit.tests.:www.dynamic:A:one",
|
||||||
|
"type": "A",
|
||||||
|
"values": [
|
||||||
|
{
|
||||||
|
"value": "1.2.3.6",
|
||||||
|
"weight": 1
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"value": "1.2.3.7",
|
||||||
|
"weight": 1
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
])
|
||||||
|
|
||||||
|
provider._client.geofilters = Mock(return_value=[
|
||||||
|
{
|
||||||
|
"id": 6303,
|
||||||
|
"name": "some.other",
|
||||||
|
"filterRulesLimit": 100,
|
||||||
|
"createdTs": "2021-08-19T14:47:47Z",
|
||||||
|
"modifiedTs": "2021-08-19T14:47:47Z",
|
||||||
|
"geoipContinents": ["AS", "OC"],
|
||||||
|
"geoipCountries": ["ES", "SE", "UA"],
|
||||||
|
"regions": [
|
||||||
|
{
|
||||||
|
"continentCode": "NA",
|
||||||
|
"countryCode": "CA",
|
||||||
|
"regionCode": "NL"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}, {
|
||||||
|
"id": 5303,
|
||||||
|
"name": "unit.tests.:www.dynamic:A:one",
|
||||||
|
"filterRulesLimit": 100,
|
||||||
|
"geoipContinents": ["AS", "OC"],
|
||||||
|
"geoipCountries": ["ES", "SE", "UA"],
|
||||||
|
"regions": [
|
||||||
|
{
|
||||||
|
"continentCode": "NA",
|
||||||
|
"countryCode": "CA",
|
||||||
|
"regionCode": "NL"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
])
|
||||||
|
|
||||||
wanted = Zone('unit.tests.', [])
|
wanted = Zone('unit.tests.', [])
|
||||||
|
|
||||||
wanted.add_record(Record.new(wanted, 'www.dynamic', {
|
wanted.add_record(Record.new(wanted, 'www.dynamic', {
|
||||||
'ttl': 300,
|
'ttl': 300,
|
||||||
'type': 'A',
|
'type': 'A',
|
||||||
'value': [
|
'values': [
|
||||||
'1.2.3.4'
|
'1.2.3.4'
|
||||||
],
|
],
|
||||||
'dynamic': {
|
'dynamic': {
|
||||||
'pools': {
|
'pools': {
|
||||||
|
'one': {
|
||||||
|
'fallback': 'two',
|
||||||
|
'values': [{
|
||||||
|
'value': '1.2.3.6',
|
||||||
|
'weight': 1
|
||||||
|
}, {
|
||||||
|
'value': '1.2.3.7',
|
||||||
|
'weight': 1
|
||||||
|
}],
|
||||||
|
},
|
||||||
'two': {
|
'two': {
|
||||||
'values': [{
|
'values': [{
|
||||||
'value': '1.2.3.5'
|
'value': '1.2.3.4',
|
||||||
|
'weight': 1
|
||||||
}],
|
}],
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
'rules': [{
|
'rules': [{
|
||||||
|
'geos': [
|
||||||
|
'AS',
|
||||||
|
'EU-ES',
|
||||||
|
'EU-UA',
|
||||||
|
'EU-SE',
|
||||||
|
'OC'
|
||||||
|
],
|
||||||
|
'pool': 'one'
|
||||||
|
}, {
|
||||||
'pool': 'two',
|
'pool': 'two',
|
||||||
}],
|
}],
|
||||||
},
|
},
|
||||||
@@ -473,13 +1020,25 @@ class TestConstellixProvider(TestCase):
|
|||||||
|
|
||||||
# Try an error we can handle
|
# Try an error we can handle
|
||||||
with requests_mock() as mock:
|
with requests_mock() as mock:
|
||||||
mock.get(ANY, status_code=200,
|
mock.get(
|
||||||
text='{}')
|
"https://api.dns.constellix.com/v1/domains",
|
||||||
|
status_code=200,
|
||||||
|
text='[{"id": 1234, "name": "unit.tests", "hasGeoIP": true}]')
|
||||||
|
mock.get(
|
||||||
|
"https://api.dns.constellix.com/v1/domains/1234",
|
||||||
|
status_code=200,
|
||||||
|
text='{"id": 1234, "name": "unit.tests", "hasGeoIP": true}')
|
||||||
mock.delete(ANY, status_code=200,
|
mock.delete(ANY, status_code=200,
|
||||||
text='{}')
|
text='{}')
|
||||||
mock.put("https://api.dns.constellix.com/v1/pools/A/1808521",
|
mock.put("https://api.dns.constellix.com/v1/pools/A/1808521",
|
||||||
status_code=400,
|
status_code=400,
|
||||||
text='{"errors": [\"no changes to save\"]}')
|
text='{"errors": [\"no changes to save\"]}')
|
||||||
|
mock.put("https://api.dns.constellix.com/v1/pools/A/1808522",
|
||||||
|
status_code=400,
|
||||||
|
text='{"errors": [\"no changes to save\"]}')
|
||||||
|
mock.put("https://api.dns.constellix.com/v1/geoFilters/5303",
|
||||||
|
status_code=400,
|
||||||
|
text='{"errors": [\"no changes to save\"]}')
|
||||||
mock.post(ANY, status_code=200,
|
mock.post(ANY, status_code=200,
|
||||||
text='[{"id": 1234}]')
|
text='[{"id": 1234}]')
|
||||||
|
|
||||||
@@ -487,13 +1046,87 @@ class TestConstellixProvider(TestCase):
|
|||||||
self.assertEquals(1, len(plan.changes))
|
self.assertEquals(1, len(plan.changes))
|
||||||
self.assertEquals(1, provider.apply(plan))
|
self.assertEquals(1, provider.apply(plan))
|
||||||
|
|
||||||
|
provider._client.geofilters = Mock(return_value=[
|
||||||
|
{
|
||||||
|
"id": 5303,
|
||||||
|
"name": "unit.tests.:www.dynamic:A:one",
|
||||||
|
"filterRulesLimit": 100,
|
||||||
|
"regions": [
|
||||||
|
{
|
||||||
|
"continentCode": "NA",
|
||||||
|
"countryCode": "CA",
|
||||||
|
"regionCode": "NL"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
])
|
||||||
|
|
||||||
|
plan = provider.plan(wanted)
|
||||||
|
self.assertEquals(1, len(plan.changes))
|
||||||
|
self.assertEquals(1, provider.apply(plan))
|
||||||
|
|
||||||
|
provider._client.geofilters = Mock(return_value=[
|
||||||
|
{
|
||||||
|
"id": 5303,
|
||||||
|
"name": "unit.tests.:www.dynamic:A:one",
|
||||||
|
"filterRulesLimit": 100,
|
||||||
|
"geoipContinents": ["AS", "OC"],
|
||||||
|
}
|
||||||
|
])
|
||||||
|
|
||||||
|
plan = provider.plan(wanted)
|
||||||
|
self.assertEquals(1, len(plan.changes))
|
||||||
|
self.assertEquals(1, provider.apply(plan))
|
||||||
|
|
||||||
# Now what happens if an error happens that we can't handle
|
# Now what happens if an error happens that we can't handle
|
||||||
|
# geofilter case
|
||||||
with requests_mock() as mock:
|
with requests_mock() as mock:
|
||||||
mock.get(ANY, status_code=200,
|
mock.get(
|
||||||
text='{}')
|
"https://api.dns.constellix.com/v1/domains",
|
||||||
|
status_code=200,
|
||||||
|
text='[{"id": 1234, "name": "unit.tests", "hasGeoIP": true}]')
|
||||||
|
mock.get(
|
||||||
|
"https://api.dns.constellix.com/v1/domains/1234",
|
||||||
|
status_code=200,
|
||||||
|
text='{"id": 1234, "name": "unit.tests", "hasGeoIP": true}')
|
||||||
mock.delete(ANY, status_code=200,
|
mock.delete(ANY, status_code=200,
|
||||||
text='{}')
|
text='{}')
|
||||||
mock.put("https://api.dns.constellix.com/v1/pools/A/1808521",
|
mock.put("https://api.dns.constellix.com/v1/pools/A/1808521",
|
||||||
|
status_code=400,
|
||||||
|
text='{"errors": [\"no changes to save\"]}')
|
||||||
|
mock.put("https://api.dns.constellix.com/v1/pools/A/1808522",
|
||||||
|
status_code=400,
|
||||||
|
text='{"errors": [\"no changes to save\"]}')
|
||||||
|
mock.put("https://api.dns.constellix.com/v1/geoFilters/5303",
|
||||||
|
status_code=400,
|
||||||
|
text='{"errors": [\"generic error\"]}')
|
||||||
|
mock.post(ANY, status_code=200,
|
||||||
|
text='[{"id": 1234}]')
|
||||||
|
|
||||||
|
plan = provider.plan(wanted)
|
||||||
|
self.assertEquals(1, len(plan.changes))
|
||||||
|
with self.assertRaises(ConstellixClientBadRequest):
|
||||||
|
provider.apply(plan)
|
||||||
|
|
||||||
|
# Now what happens if an error happens that we can't handle
|
||||||
|
with requests_mock() as mock:
|
||||||
|
mock.get(
|
||||||
|
"https://api.dns.constellix.com/v1/domains",
|
||||||
|
status_code=200,
|
||||||
|
text='[{"id": 1234, "name": "unit.tests", "hasGeoIP": true}]')
|
||||||
|
mock.get(
|
||||||
|
"https://api.dns.constellix.com/v1/domains/1234",
|
||||||
|
status_code=200,
|
||||||
|
text='{"id": 1234, "name": "unit.tests", "hasGeoIP": true}')
|
||||||
|
mock.delete(ANY, status_code=200,
|
||||||
|
text='{}')
|
||||||
|
mock.put("https://api.dns.constellix.com/v1/pools/A/1808521",
|
||||||
|
status_code=400,
|
||||||
|
text='{"errors": [\"generic error\"]}')
|
||||||
|
mock.put("https://api.dns.constellix.com/v1/pools/A/1808522",
|
||||||
|
status_code=400,
|
||||||
|
text='{"errors": [\"generic error\"]}')
|
||||||
|
mock.put("https://api.dns.constellix.com/v1/geoFilters/5303",
|
||||||
status_code=400,
|
status_code=400,
|
||||||
text='{"errors": [\"generic error\"]}')
|
text='{"errors": [\"generic error\"]}')
|
||||||
mock.post(ANY, status_code=200,
|
mock.post(ANY, status_code=200,
|
||||||
|
|||||||
Reference in New Issue
Block a user