mirror of
https://github.com/github/octodns.git
synced 2024-05-11 05:55:00 +00:00
Extract UltraProvider from octoDNS core
This commit is contained in:
Ross McFarland
-94
@@ -1,94 +0,0 @@
|
||||
{
|
||||
"zoneName": "octodns1.test.",
|
||||
"rrSets": [
|
||||
{
|
||||
"ownerName": "_srv._tcp.octodns1.test.",
|
||||
"rrtype": "SRV (33)",
|
||||
"ttl": 3600,
|
||||
"rdata": [
|
||||
"0 20 443 cname.octodns1.test."
|
||||
]
|
||||
},
|
||||
{
|
||||
"ownerName": "a.octodns1.test.",
|
||||
"rrtype": "A (1)",
|
||||
"ttl": 3600,
|
||||
"rdata": [
|
||||
"1.1.1.1"
|
||||
]
|
||||
},
|
||||
{
|
||||
"ownerName": "aaaa.octodns1.test.",
|
||||
"rrtype": "AAAA (28)",
|
||||
"ttl": 3600,
|
||||
"rdata": [
|
||||
"0:0:0:0:0:0:0:1"
|
||||
]
|
||||
},
|
||||
{
|
||||
"ownerName": "caa.octodns1.test.",
|
||||
"rrtype": "CAA (257)",
|
||||
"ttl": 3600,
|
||||
"rdata": [
|
||||
"0 issue \"symantec.com\""
|
||||
]
|
||||
},
|
||||
{
|
||||
"ownerName": "cname.octodns1.test.",
|
||||
"rrtype": "CNAME (5)",
|
||||
"ttl": 60,
|
||||
"rdata": [
|
||||
"a.octodns1.test."
|
||||
]
|
||||
},
|
||||
{
|
||||
"ownerName": "mail.octodns1.test.",
|
||||
"rrtype": "MX (15)",
|
||||
"ttl": 3600,
|
||||
"rdata": [
|
||||
"1 aspmx.l.google.com.",
|
||||
"5 alt1.aspmx.l.google.com."
|
||||
]
|
||||
},
|
||||
{
|
||||
"ownerName": "octodns1.test.",
|
||||
"rrtype": "NS (2)",
|
||||
"ttl": 86400,
|
||||
"rdata": [
|
||||
"pdns1.ultradns.biz.",
|
||||
"pdns1.ultradns.com.",
|
||||
"pdns1.ultradns.net.",
|
||||
"pdns1.ultradns.org."
|
||||
]
|
||||
},
|
||||
{
|
||||
"ownerName": "octodns1.test.",
|
||||
"rrtype": "SOA (6)",
|
||||
"ttl": 86400,
|
||||
"rdata": [
|
||||
"pdns1.ultradns.com. phelps.netflix.com. 2020062003 86400 86400 86400 86400"
|
||||
]
|
||||
},
|
||||
{
|
||||
"ownerName": "ptr.octodns1.test.",
|
||||
"rrtype": "PTR (12)",
|
||||
"ttl": 300,
|
||||
"rdata": [
|
||||
"foo.bar.com."
|
||||
]
|
||||
},
|
||||
{
|
||||
"ownerName": "spf.octodns1.test.",
|
||||
"rrtype": "SPF (99)",
|
||||
"ttl": 3600,
|
||||
"rdata": [
|
||||
"v=spf1 -all"
|
||||
]
|
||||
}
|
||||
],
|
||||
"resultInfo": {
|
||||
"totalCount": 13,
|
||||
"offset": 0,
|
||||
"returnedCount": 10
|
||||
}
|
||||
}
|
||||
-51
@@ -1,51 +0,0 @@
|
||||
{
|
||||
"zoneName": "octodns1.test.",
|
||||
"rrSets": [
|
||||
{
|
||||
"ownerName": "txt.octodns1.test.",
|
||||
"rrtype": "TXT (16)",
|
||||
"ttl": 3600,
|
||||
"rdata": [
|
||||
"foobar",
|
||||
"v=spf1 -all"
|
||||
]
|
||||
},
|
||||
{
|
||||
"ownerName": "octodns1.test.",
|
||||
"rrtype": "A (1)",
|
||||
"ttl": 3600,
|
||||
"rdata": [
|
||||
"1.2.3.4",
|
||||
"1.2.3.5",
|
||||
"1.2.3.6"
|
||||
],
|
||||
"profile": {
|
||||
"@context": "http://schemas.ultradns.com/RDPool.jsonschema",
|
||||
"order": "FIXED",
|
||||
"description": "octodns1.test."
|
||||
}
|
||||
},
|
||||
{
|
||||
"ownerName": "octodns1.test.",
|
||||
"rrtype": "APEXALIAS (65282)",
|
||||
"ttl": 3600,
|
||||
"rdata": [
|
||||
"www.octodns1.test."
|
||||
]
|
||||
},
|
||||
{
|
||||
"ownerName": "host1.octodns1.test.",
|
||||
"rrtype": "RRSET (70)",
|
||||
"ttl": 3600,
|
||||
"rdata": [
|
||||
"E3B0C44298FC1C149AFBF4C8996FB92427AE41E4649B934CA495991B7852B855"
|
||||
]
|
||||
}
|
||||
|
||||
],
|
||||
"resultInfo": {
|
||||
"totalCount": 13,
|
||||
"offset": 10,
|
||||
"returnedCount": 3
|
||||
}
|
||||
}
|
||||
-135
@@ -1,135 +0,0 @@
|
||||
{
|
||||
"queryInfo": {
|
||||
"q": "zone_type:PRIMARY",
|
||||
"sort": "NAME",
|
||||
"reverse": false,
|
||||
"limit": 10
|
||||
},
|
||||
"resultInfo": {
|
||||
"totalCount": 20,
|
||||
"offset": 0,
|
||||
"returnedCount": 10
|
||||
},
|
||||
"zones": [
|
||||
{
|
||||
"properties": {
|
||||
"name": "octodns1.test.",
|
||||
"accountName": "Netflix - Automation",
|
||||
"type": "PRIMARY",
|
||||
"dnssecStatus": "UNSIGNED",
|
||||
"status": "ACTIVE",
|
||||
"owner": "phelpstest",
|
||||
"resourceRecordCount": 6,
|
||||
"lastModifiedDateTime": "2020-06-19T01:05Z"
|
||||
}
|
||||
},
|
||||
{
|
||||
"properties": {
|
||||
"name": "octodns10.test.",
|
||||
"accountName": "Netflix - Automation",
|
||||
"type": "PRIMARY",
|
||||
"dnssecStatus": "UNSIGNED",
|
||||
"status": "ACTIVE",
|
||||
"owner": "phelpstest",
|
||||
"resourceRecordCount": 5,
|
||||
"lastModifiedDateTime": "2020-06-19T01:06Z"
|
||||
}
|
||||
},
|
||||
{
|
||||
"properties": {
|
||||
"name": "octodns11.test.",
|
||||
"accountName": "Netflix - Automation",
|
||||
"type": "PRIMARY",
|
||||
"dnssecStatus": "UNSIGNED",
|
||||
"status": "ACTIVE",
|
||||
"owner": "phelpstest",
|
||||
"resourceRecordCount": 5,
|
||||
"lastModifiedDateTime": "2020-06-19T01:06Z"
|
||||
}
|
||||
},
|
||||
{
|
||||
"properties": {
|
||||
"name": "octodns12.test.",
|
||||
"accountName": "Netflix - Automation",
|
||||
"type": "PRIMARY",
|
||||
"dnssecStatus": "UNSIGNED",
|
||||
"status": "ACTIVE",
|
||||
"owner": "phelpstest",
|
||||
"resourceRecordCount": 5,
|
||||
"lastModifiedDateTime": "2020-06-19T01:06Z"
|
||||
}
|
||||
},
|
||||
{
|
||||
"properties": {
|
||||
"name": "octodns13.test.",
|
||||
"accountName": "Netflix - Automation",
|
||||
"type": "PRIMARY",
|
||||
"dnssecStatus": "UNSIGNED",
|
||||
"status": "ACTIVE",
|
||||
"owner": "phelpstest",
|
||||
"resourceRecordCount": 5,
|
||||
"lastModifiedDateTime": "2020-06-19T01:06Z"
|
||||
}
|
||||
},
|
||||
{
|
||||
"properties": {
|
||||
"name": "octodns14.test.",
|
||||
"accountName": "Netflix - Automation",
|
||||
"type": "PRIMARY",
|
||||
"dnssecStatus": "UNSIGNED",
|
||||
"status": "ACTIVE",
|
||||
"owner": "phelpstest",
|
||||
"resourceRecordCount": 5,
|
||||
"lastModifiedDateTime": "2020-06-19T01:06Z"
|
||||
}
|
||||
},
|
||||
{
|
||||
"properties": {
|
||||
"name": "octodns15.test.",
|
||||
"accountName": "Netflix - Automation",
|
||||
"type": "PRIMARY",
|
||||
"dnssecStatus": "UNSIGNED",
|
||||
"status": "ACTIVE",
|
||||
"owner": "phelpstest",
|
||||
"resourceRecordCount": 5,
|
||||
"lastModifiedDateTime": "2020-06-19T01:06Z"
|
||||
}
|
||||
},
|
||||
{
|
||||
"properties": {
|
||||
"name": "octodns16.test.",
|
||||
"accountName": "Netflix - Automation",
|
||||
"type": "PRIMARY",
|
||||
"dnssecStatus": "UNSIGNED",
|
||||
"status": "ACTIVE",
|
||||
"owner": "phelpstest",
|
||||
"resourceRecordCount": 5,
|
||||
"lastModifiedDateTime": "2020-06-19T01:06Z"
|
||||
}
|
||||
},
|
||||
{
|
||||
"properties": {
|
||||
"name": "octodns17.test.",
|
||||
"accountName": "Netflix - Automation",
|
||||
"type": "PRIMARY",
|
||||
"dnssecStatus": "UNSIGNED",
|
||||
"status": "ACTIVE",
|
||||
"owner": "phelpstest",
|
||||
"resourceRecordCount": 5,
|
||||
"lastModifiedDateTime": "2020-06-19T01:06Z"
|
||||
}
|
||||
},
|
||||
{
|
||||
"properties": {
|
||||
"name": "octodns18.test.",
|
||||
"accountName": "Netflix - Automation",
|
||||
"type": "PRIMARY",
|
||||
"dnssecStatus": "UNSIGNED",
|
||||
"status": "ACTIVE",
|
||||
"owner": "phelpstest",
|
||||
"resourceRecordCount": 5,
|
||||
"lastModifiedDateTime": "2020-06-19T01:07Z"
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
-135
@@ -1,135 +0,0 @@
|
||||
{
|
||||
"queryInfo": {
|
||||
"q": "zone_type:PRIMARY",
|
||||
"sort": "NAME",
|
||||
"reverse": false,
|
||||
"limit": 10
|
||||
},
|
||||
"resultInfo": {
|
||||
"totalCount": 20,
|
||||
"offset": 10,
|
||||
"returnedCount": 10
|
||||
},
|
||||
"zones": [
|
||||
{
|
||||
"properties": {
|
||||
"name": "octodns19.test.",
|
||||
"accountName": "Netflix - Automation",
|
||||
"type": "PRIMARY",
|
||||
"dnssecStatus": "UNSIGNED",
|
||||
"status": "ACTIVE",
|
||||
"owner": "phelpstest",
|
||||
"resourceRecordCount": 5,
|
||||
"lastModifiedDateTime": "2020-06-19T01:07Z"
|
||||
}
|
||||
},
|
||||
{
|
||||
"properties": {
|
||||
"name": "octodns2.test.",
|
||||
"accountName": "Netflix - Automation",
|
||||
"type": "PRIMARY",
|
||||
"dnssecStatus": "UNSIGNED",
|
||||
"status": "ACTIVE",
|
||||
"owner": "phelpstest",
|
||||
"resourceRecordCount": 5,
|
||||
"lastModifiedDateTime": "2020-06-19T01:05Z"
|
||||
}
|
||||
},
|
||||
{
|
||||
"properties": {
|
||||
"name": "octodns20.test.",
|
||||
"accountName": "Netflix - Automation",
|
||||
"type": "PRIMARY",
|
||||
"dnssecStatus": "UNSIGNED",
|
||||
"status": "ACTIVE",
|
||||
"owner": "phelpstest",
|
||||
"resourceRecordCount": 5,
|
||||
"lastModifiedDateTime": "2020-06-19T01:07Z"
|
||||
}
|
||||
},
|
||||
{
|
||||
"properties": {
|
||||
"name": "octodns3.test.",
|
||||
"accountName": "Netflix - Automation",
|
||||
"type": "PRIMARY",
|
||||
"dnssecStatus": "UNSIGNED",
|
||||
"status": "ACTIVE",
|
||||
"owner": "phelpstest",
|
||||
"resourceRecordCount": 5,
|
||||
"lastModifiedDateTime": "2020-06-19T01:05Z"
|
||||
}
|
||||
},
|
||||
{
|
||||
"properties": {
|
||||
"name": "octodns4.test.",
|
||||
"accountName": "Netflix - Automation",
|
||||
"type": "PRIMARY",
|
||||
"dnssecStatus": "UNSIGNED",
|
||||
"status": "ACTIVE",
|
||||
"owner": "phelpstest",
|
||||
"resourceRecordCount": 5,
|
||||
"lastModifiedDateTime": "2020-06-19T01:05Z"
|
||||
}
|
||||
},
|
||||
{
|
||||
"properties": {
|
||||
"name": "octodns5.test.",
|
||||
"accountName": "Netflix - Automation",
|
||||
"type": "PRIMARY",
|
||||
"dnssecStatus": "UNSIGNED",
|
||||
"status": "ACTIVE",
|
||||
"owner": "phelpstest",
|
||||
"resourceRecordCount": 5,
|
||||
"lastModifiedDateTime": "2020-06-19T01:05Z"
|
||||
}
|
||||
},
|
||||
{
|
||||
"properties": {
|
||||
"name": "octodns6.test.",
|
||||
"accountName": "Netflix - Automation",
|
||||
"type": "PRIMARY",
|
||||
"dnssecStatus": "UNSIGNED",
|
||||
"status": "ACTIVE",
|
||||
"owner": "phelpstest",
|
||||
"resourceRecordCount": 5,
|
||||
"lastModifiedDateTime": "2020-06-19T01:05Z"
|
||||
}
|
||||
},
|
||||
{
|
||||
"properties": {
|
||||
"name": "octodns7.test.",
|
||||
"accountName": "Netflix - Automation",
|
||||
"type": "PRIMARY",
|
||||
"dnssecStatus": "UNSIGNED",
|
||||
"status": "ACTIVE",
|
||||
"owner": "phelpstest",
|
||||
"resourceRecordCount": 5,
|
||||
"lastModifiedDateTime": "2020-06-19T01:06Z"
|
||||
}
|
||||
},
|
||||
{
|
||||
"properties": {
|
||||
"name": "octodns8.test.",
|
||||
"accountName": "Netflix - Automation",
|
||||
"type": "PRIMARY",
|
||||
"dnssecStatus": "UNSIGNED",
|
||||
"status": "ACTIVE",
|
||||
"owner": "phelpstest",
|
||||
"resourceRecordCount": 5,
|
||||
"lastModifiedDateTime": "2020-06-19T01:06Z"
|
||||
}
|
||||
},
|
||||
{
|
||||
"properties": {
|
||||
"name": "octodns9.test.",
|
||||
"accountName": "Netflix - Automation",
|
||||
"type": "PRIMARY",
|
||||
"dnssecStatus": "UNSIGNED",
|
||||
"status": "ACTIVE",
|
||||
"owner": "phelpstest",
|
||||
"resourceRecordCount": 5,
|
||||
"lastModifiedDateTime": "2020-06-19T01:06Z"
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
@@ -1,514 +1,16 @@
|
||||
from __future__ import unicode_literals
|
||||
#
|
||||
#
|
||||
#
|
||||
|
||||
from __future__ import absolute_import, division, print_function, \
|
||||
unicode_literals
|
||||
|
||||
from mock import Mock, call
|
||||
from os.path import dirname, join
|
||||
from requests import HTTPError
|
||||
from requests_mock import ANY, mock as requests_mock
|
||||
from urllib.parse import parse_qs
|
||||
from unittest import TestCase
|
||||
from json import load as json_load
|
||||
|
||||
from octodns.record import Record
|
||||
from octodns.provider.ultra import UltraProvider, UltraNoZonesExistException
|
||||
from octodns.provider.yaml import YamlProvider
|
||||
from octodns.zone import Zone
|
||||
|
||||
|
||||
def _get_provider():
|
||||
'''
|
||||
Helper to return a provider after going through authentication sequence
|
||||
'''
|
||||
with requests_mock() as mock:
|
||||
mock.post('https://restapi.ultradns.com/v2/authorization/token',
|
||||
status_code=200,
|
||||
text='{"token type": "Bearer", "refresh_token": "abc", '
|
||||
'"access_token":"123", "expires_in": "3600"}')
|
||||
return UltraProvider('test', 'testacct', 'user', 'pass')
|
||||
class TestUltraShim(TestCase):
|
||||
|
||||
|
||||
class TestUltraProvider(TestCase):
|
||||
expected = Zone('unit.tests.', [])
|
||||
host = 'https://restapi.ultradns.com'
|
||||
empty_body = [{"errorCode": 70002, "errorMessage": "Data not found."}]
|
||||
|
||||
expected = Zone('unit.tests.', [])
|
||||
source = YamlProvider('test', join(dirname(__file__), 'config'))
|
||||
source.populate(expected)
|
||||
|
||||
def test_login(self):
|
||||
path = '/v2/authorization/token'
|
||||
|
||||
# Bad Auth
|
||||
with requests_mock() as mock:
|
||||
mock.post(f'{self.host}{path}', status_code=401,
|
||||
text='{"errorCode": 60001}')
|
||||
with self.assertRaises(Exception) as ctx:
|
||||
UltraProvider('test', 'account', 'user', 'wrongpass')
|
||||
self.assertEqual('Unauthorized', str(ctx.exception))
|
||||
|
||||
# Good Auth
|
||||
with requests_mock() as mock:
|
||||
headers = {'Content-Type': 'application/x-www-form-urlencoded'}
|
||||
mock.post(f'{self.host}{path}', status_code=200,
|
||||
request_headers=headers,
|
||||
text='{"token type": "Bearer", "refresh_token": "abc", '
|
||||
'"access_token":"123", "expires_in": "3600"}')
|
||||
UltraProvider('test', 'account', 'user', 'rightpass')
|
||||
self.assertEqual(1, mock.call_count)
|
||||
expected_payload = "grant_type=password&username=user&"\
|
||||
"password=rightpass"
|
||||
self.assertEqual(parse_qs(mock.last_request.text),
|
||||
parse_qs(expected_payload))
|
||||
|
||||
def test_get_zones(self):
|
||||
provider = _get_provider()
|
||||
path = "/v2/zones"
|
||||
|
||||
# Test authorization issue
|
||||
with requests_mock() as mock:
|
||||
mock.get(f'{self.host}{path}', status_code=400,
|
||||
json={"errorCode": 60004,
|
||||
"errorMessage": "Authorization Header required"})
|
||||
with self.assertRaises(HTTPError) as ctx:
|
||||
zones = provider.zones
|
||||
self.assertEqual(400, ctx.exception.response.status_code)
|
||||
|
||||
# Test no zones exist error
|
||||
with requests_mock() as mock:
|
||||
mock.get(f'{self.host}{path}', status_code=404,
|
||||
headers={'Authorization': 'Bearer 123'},
|
||||
json=self.empty_body)
|
||||
zones = provider.zones
|
||||
self.assertEqual(1, mock.call_count)
|
||||
self.assertEqual(list(), zones)
|
||||
|
||||
# Reset zone cache so they are queried again
|
||||
provider._zones = None
|
||||
|
||||
with requests_mock() as mock:
|
||||
payload = {
|
||||
"resultInfo": {
|
||||
"totalCount": 1,
|
||||
"offset": 0,
|
||||
"returnedCount": 1
|
||||
},
|
||||
"zones": [
|
||||
{
|
||||
"properties": {
|
||||
"name": "testzone123.com.",
|
||||
"accountName": "testaccount",
|
||||
"type": "PRIMARY",
|
||||
"dnssecStatus": "UNSIGNED",
|
||||
"status": "ACTIVE",
|
||||
"owner": "user",
|
||||
"resourceRecordCount": 5,
|
||||
"lastModifiedDateTime": "2020-06-19T00:47Z"
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
|
||||
mock.get(f'{self.host}{path}', status_code=200,
|
||||
headers={'Authorization': 'Bearer 123'},
|
||||
json=payload)
|
||||
zones = provider.zones
|
||||
self.assertEqual(1, mock.call_count)
|
||||
self.assertEqual(1, len(zones))
|
||||
self.assertEqual('testzone123.com.', zones[0])
|
||||
|
||||
# Test different paging behavior
|
||||
provider._zones = None
|
||||
with requests_mock() as mock:
|
||||
mock.get(f'{self.host}{path}?limit=100&q=zone_type%3APRIMARY&'
|
||||
'offset=0', status_code=200,
|
||||
json={"resultInfo": {"totalCount": 15,
|
||||
"offset": 0,
|
||||
"returnedCount": 10},
|
||||
"zones": []})
|
||||
mock.get(f'{self.host}{path}?limit=100&q=zone_type%3APRIMARY'
|
||||
'&offset=10', status_code=200,
|
||||
json={"resultInfo": {"totalCount": 15,
|
||||
"offset": 10,
|
||||
"returnedCount": 5},
|
||||
"zones": []})
|
||||
zones = provider.zones
|
||||
self.assertEqual(2, mock.call_count)
|
||||
|
||||
def test_request(self):
|
||||
provider = _get_provider()
|
||||
path = '/foo'
|
||||
payload = {'a': 1}
|
||||
|
||||
with requests_mock() as mock:
|
||||
mock.get(f'{self.host}{path}', status_code=401,
|
||||
headers={'Authorization': 'Bearer 123'}, json={})
|
||||
with self.assertRaises(Exception) as ctx:
|
||||
provider._get(path)
|
||||
self.assertEqual('Unauthorized', str(ctx.exception))
|
||||
|
||||
# Test all GET patterns
|
||||
with requests_mock() as mock:
|
||||
mock.get(f'{self.host}{path}', status_code=200,
|
||||
headers={'Authorization': 'Bearer 123'},
|
||||
json=payload)
|
||||
provider._get(path, json=payload)
|
||||
|
||||
mock.get(f'{self.host}{path}?a=1', status_code=200,
|
||||
headers={'Authorization': 'Bearer 123'})
|
||||
provider._get(path, params=payload, json_response=False)
|
||||
|
||||
# Test all POST patterns
|
||||
with requests_mock() as mock:
|
||||
mock.post(f'{self.host}{path}', status_code=200,
|
||||
headers={'Authorization': 'Bearer 123'},
|
||||
json=payload)
|
||||
provider._post(path, json=payload)
|
||||
|
||||
mock.post(f'{self.host}{path}', status_code=200,
|
||||
headers={'Authorization': 'Bearer 123'},
|
||||
text="{'a':1}")
|
||||
provider._post(path, data=payload, json_response=False)
|
||||
|
||||
# Test all PUT patterns
|
||||
with requests_mock() as mock:
|
||||
mock.put(f'{self.host}{path}', status_code=200,
|
||||
headers={'Authorization': 'Bearer 123'},
|
||||
json=payload)
|
||||
provider._put(path, json=payload)
|
||||
|
||||
# Test all DELETE patterns
|
||||
with requests_mock() as mock:
|
||||
mock.delete(f'{self.host}{path}', status_code=200,
|
||||
headers={'Authorization': 'Bearer 123'})
|
||||
provider._delete(path, json_response=False)
|
||||
|
||||
def test_zone_records(self):
|
||||
provider = _get_provider()
|
||||
zone_payload = {
|
||||
"resultInfo": {"totalCount": 1,
|
||||
"offset": 0,
|
||||
"returnedCount": 1},
|
||||
"zones": [{"properties": {"name": "octodns1.test."}}]}
|
||||
|
||||
records_payload = {
|
||||
"zoneName": "octodns1.test.",
|
||||
"rrSets": [
|
||||
{
|
||||
"ownerName": "octodns1.test.",
|
||||
"rrtype": "NS (2)",
|
||||
"ttl": 86400,
|
||||
"rdata": [
|
||||
"ns1.octodns1.test."
|
||||
]
|
||||
},
|
||||
{
|
||||
"ownerName": "octodns1.test.",
|
||||
"rrtype": "SOA (6)",
|
||||
"ttl": 86400,
|
||||
"rdata": [
|
||||
"pdns1.ultradns.com. phelps.netflix.com. 1 10 10 10 10"
|
||||
]
|
||||
},
|
||||
],
|
||||
"resultInfo": {
|
||||
"totalCount": 2,
|
||||
"offset": 0,
|
||||
"returnedCount": 2
|
||||
}
|
||||
}
|
||||
|
||||
zone_path = '/v2/zones'
|
||||
rec_path = '/v2/zones/octodns1.test./rrsets'
|
||||
with requests_mock() as mock:
|
||||
mock.get(f'{self.host}{zone_path}?limit=100&q=zone_type%3APRIMARY&'
|
||||
'offset=0', status_code=200, json=zone_payload)
|
||||
mock.get(f'{self.host}{rec_path}?offset=0&limit=100',
|
||||
status_code=200, json=records_payload)
|
||||
|
||||
zone = Zone('octodns1.test.', [])
|
||||
self.assertTrue(provider.zone_records(zone))
|
||||
self.assertEqual(mock.call_count, 2)
|
||||
|
||||
# Populate the same zone again and confirm cache is hit
|
||||
self.assertTrue(provider.zone_records(zone))
|
||||
self.assertEqual(mock.call_count, 2)
|
||||
|
||||
def test_populate(self):
|
||||
provider = _get_provider()
|
||||
|
||||
# Non-existent zone doesn't populate anything
|
||||
with requests_mock() as mock:
|
||||
mock.get(ANY, status_code=404, json=self.empty_body)
|
||||
|
||||
zone = Zone('unit.tests.', [])
|
||||
provider.populate(zone)
|
||||
self.assertEqual(set(), zone.records)
|
||||
|
||||
# re-populating the same non-existent zone uses cache and makes no
|
||||
# calls
|
||||
again = Zone('unit.tests.', [])
|
||||
provider.populate(again)
|
||||
self.assertEqual(set(), again.records)
|
||||
|
||||
# Test zones with data
|
||||
provider._zones = None
|
||||
path = '/v2/zones'
|
||||
with requests_mock() as mock:
|
||||
with open('tests/fixtures/ultra-zones-page-1.json') as fh:
|
||||
mock.get(f'{self.host}{path}?limit=100&q=zone_type%3APRIMARY&'
|
||||
'offset=0', status_code=200, text=fh.read())
|
||||
with open('tests/fixtures/ultra-zones-page-2.json') as fh:
|
||||
mock.get(f'{self.host}{path}?limit=100&q=zone_type%3APRIMARY&'
|
||||
'offset=10', status_code=200, text=fh.read())
|
||||
with open('tests/fixtures/ultra-records-page-1.json') as fh:
|
||||
rec_path = '/v2/zones/octodns1.test./rrsets'
|
||||
mock.get(f'{self.host}{rec_path}?offset=0&limit=100',
|
||||
status_code=200, text=fh.read())
|
||||
with open('tests/fixtures/ultra-records-page-2.json') as fh:
|
||||
rec_path = '/v2/zones/octodns1.test./rrsets'
|
||||
mock.get(f'{self.host}{rec_path}?offset=10&limit=100',
|
||||
status_code=200, text=fh.read())
|
||||
|
||||
zone = Zone('octodns1.test.', [])
|
||||
|
||||
self.assertTrue(provider.populate(zone))
|
||||
self.assertEqual('octodns1.test.', zone.name)
|
||||
self.assertEqual(12, len(zone.records))
|
||||
self.assertEqual(4, mock.call_count)
|
||||
|
||||
def test_apply(self):
|
||||
provider = _get_provider()
|
||||
|
||||
provider._request = Mock()
|
||||
|
||||
provider._request.side_effect = [
|
||||
UltraNoZonesExistException('No Zones'),
|
||||
None, # zone create
|
||||
] + [None] * 15 # individual record creates
|
||||
|
||||
# non-existent zone, create everything
|
||||
plan = provider.plan(self.expected)
|
||||
self.assertEqual(15, len(plan.changes))
|
||||
self.assertEqual(15, provider.apply(plan))
|
||||
self.assertFalse(plan.exists)
|
||||
|
||||
provider._request.assert_has_calls([
|
||||
# created the domain
|
||||
call('POST', '/v2/zones', json={
|
||||
'properties': {'name': 'unit.tests.',
|
||||
'accountName': 'testacct',
|
||||
'type': 'PRIMARY'},
|
||||
'primaryCreateInfo': {'createType': 'NEW'}}),
|
||||
# Validate multi-ip apex A record is correct
|
||||
call('POST', '/v2/zones/unit.tests./rrsets/A/unit.tests.', json={
|
||||
'ttl': 300,
|
||||
'rdata': ['1.2.3.4', '1.2.3.5'],
|
||||
'profile': {
|
||||
'@context':
|
||||
'http://schemas.ultradns.com/RDPool.jsonschema',
|
||||
'order': 'FIXED',
|
||||
'description': 'unit.tests.'
|
||||
}
|
||||
}),
|
||||
# make sure semicolons are not escaped when sending data
|
||||
call('POST', '/v2/zones/unit.tests./rrsets/TXT/txt.unit.tests.',
|
||||
json={'ttl': 600,
|
||||
'rdata': ['Bah bah black sheep',
|
||||
'have you any wool.',
|
||||
'v=DKIM1;k=rsa;s=email;h=sha256;'
|
||||
'p=A/kinda+of/long/string+with+numb3rs']}),
|
||||
], True)
|
||||
# expected number of total calls
|
||||
self.assertEqual(17, provider._request.call_count)
|
||||
|
||||
# Create sample rrset payload to attempt to alter
|
||||
page1 = json_load(open('tests/fixtures/ultra-records-page-1.json'))
|
||||
page2 = json_load(open('tests/fixtures/ultra-records-page-2.json'))
|
||||
mock_rrsets = list()
|
||||
mock_rrsets.extend(page1['rrSets'])
|
||||
mock_rrsets.extend(page2['rrSets'])
|
||||
|
||||
# Seed a bunch of records into a zone and verify update / delete ops
|
||||
provider._request.reset_mock()
|
||||
provider._zones = ['octodns1.test.']
|
||||
provider.zone_records = Mock(return_value=mock_rrsets)
|
||||
|
||||
provider._request.side_effect = [None] * 13
|
||||
|
||||
wanted = Zone('octodns1.test.', [])
|
||||
wanted.add_record(Record.new(wanted, '', {
|
||||
'ttl': 60, # Change TTL
|
||||
'type': 'A',
|
||||
'value': '5.6.7.8' # Change number of IPs (3 -> 1)
|
||||
}))
|
||||
wanted.add_record(Record.new(wanted, 'txt', {
|
||||
'ttl': 3600,
|
||||
'type': 'TXT',
|
||||
'values': [ # Alter TXT value
|
||||
"foobar",
|
||||
"v=spf1 include:mail.server.net ?all"
|
||||
]
|
||||
}))
|
||||
|
||||
plan = provider.plan(wanted)
|
||||
self.assertEqual(11, len(plan.changes))
|
||||
self.assertEqual(11, provider.apply(plan))
|
||||
self.assertTrue(plan.exists)
|
||||
|
||||
provider._request.assert_has_calls([
|
||||
# Validate multi-ip apex A record replaced with standard A
|
||||
call('PUT', '/v2/zones/octodns1.test./rrsets/A/octodns1.test.',
|
||||
json={'ttl': 60,
|
||||
'rdata': ['5.6.7.8']}),
|
||||
# Make sure TXT value is properly updated
|
||||
call('PUT',
|
||||
'/v2/zones/octodns1.test./rrsets/TXT/txt.octodns1.test.',
|
||||
json={'ttl': 3600,
|
||||
'rdata': ["foobar",
|
||||
"v=spf1 include:mail.server.net ?all"]}),
|
||||
# Confirm a few of the DELETE operations properly occur
|
||||
call('DELETE',
|
||||
'/v2/zones/octodns1.test./rrsets/A/a.octodns1.test.',
|
||||
json_response=False),
|
||||
call('DELETE',
|
||||
'/v2/zones/octodns1.test./rrsets/AAAA/aaaa.octodns1.test.',
|
||||
json_response=False),
|
||||
call('DELETE',
|
||||
'/v2/zones/octodns1.test./rrsets/CAA/caa.octodns1.test.',
|
||||
json_response=False),
|
||||
call('DELETE',
|
||||
'/v2/zones/octodns1.test./rrsets/CNAME/cname.octodns1.test.',
|
||||
json_response=False),
|
||||
], True)
|
||||
|
||||
def test_gen_data(self):
|
||||
provider = _get_provider()
|
||||
zone = Zone('unit.tests.', [])
|
||||
|
||||
for name, _type, expected_path, expected_payload, expected_record in (
|
||||
# A
|
||||
('a', 'A',
|
||||
'/v2/zones/unit.tests./rrsets/A/a.unit.tests.',
|
||||
{'ttl': 60, 'rdata': ['1.2.3.4']},
|
||||
Record.new(zone, 'a',
|
||||
{'ttl': 60, 'type': 'A', 'values': ['1.2.3.4']})),
|
||||
('a', 'A',
|
||||
'/v2/zones/unit.tests./rrsets/A/a.unit.tests.',
|
||||
{'ttl': 60, 'rdata': ['1.2.3.4', '5.6.7.8'],
|
||||
'profile': {'@context':
|
||||
'http://schemas.ultradns.com/RDPool.jsonschema',
|
||||
'order': 'FIXED',
|
||||
'description': 'a.unit.tests.'}},
|
||||
Record.new(zone, 'a',
|
||||
{'ttl': 60, 'type': 'A',
|
||||
'values': ['1.2.3.4', '5.6.7.8']})),
|
||||
|
||||
# AAAA
|
||||
('aaaa', 'AAAA',
|
||||
'/v2/zones/unit.tests./rrsets/AAAA/aaaa.unit.tests.',
|
||||
{'ttl': 60, 'rdata': ['::1']},
|
||||
Record.new(zone, 'aaaa',
|
||||
{'ttl': 60, 'type': 'AAAA', 'values': ['::1']})),
|
||||
('aaaa', 'AAAA',
|
||||
'/v2/zones/unit.tests./rrsets/AAAA/aaaa.unit.tests.',
|
||||
{'ttl': 60, 'rdata': ['::1', '::2'],
|
||||
'profile': {'@context':
|
||||
'http://schemas.ultradns.com/RDPool.jsonschema',
|
||||
'order': 'FIXED',
|
||||
'description': 'aaaa.unit.tests.'}},
|
||||
Record.new(zone, 'aaaa',
|
||||
{'ttl': 60, 'type': 'AAAA',
|
||||
'values': ['::1', '::2']})),
|
||||
|
||||
# CAA
|
||||
('caa', 'CAA',
|
||||
'/v2/zones/unit.tests./rrsets/CAA/caa.unit.tests.',
|
||||
{'ttl': 60, 'rdata': ['0 issue foo.com']},
|
||||
Record.new(zone, 'caa',
|
||||
{'ttl': 60, 'type': 'CAA',
|
||||
'values':
|
||||
[{'flags': 0, 'tag': 'issue', 'value': 'foo.com'}]})),
|
||||
|
||||
# CNAME
|
||||
('cname', 'CNAME',
|
||||
'/v2/zones/unit.tests./rrsets/CNAME/cname.unit.tests.',
|
||||
{'ttl': 60, 'rdata': ['netflix.com.']},
|
||||
Record.new(zone, 'cname',
|
||||
{'ttl': 60, 'type': 'CNAME',
|
||||
'value': 'netflix.com.'})),
|
||||
|
||||
|
||||
# MX
|
||||
('mx', 'MX',
|
||||
'/v2/zones/unit.tests./rrsets/MX/mx.unit.tests.',
|
||||
{'ttl': 60, 'rdata': ['1 mx1.unit.tests.', '1 mx2.unit.tests.']},
|
||||
Record.new(zone, 'mx',
|
||||
{'ttl': 60, 'type': 'MX',
|
||||
'values': [{'preference': 1,
|
||||
'exchange': 'mx1.unit.tests.'},
|
||||
{'preference': 1,
|
||||
'exchange': 'mx2.unit.tests.'}]})),
|
||||
|
||||
# NS
|
||||
('ns', 'NS',
|
||||
'/v2/zones/unit.tests./rrsets/NS/ns.unit.tests.',
|
||||
{'ttl': 60, 'rdata': ['ns1.unit.tests.', 'ns2.unit.tests.']},
|
||||
Record.new(zone, 'ns',
|
||||
{'ttl': 60, 'type': 'NS',
|
||||
'values': ['ns1.unit.tests.', 'ns2.unit.tests.']})),
|
||||
|
||||
# PTR
|
||||
('ptr', 'PTR',
|
||||
'/v2/zones/unit.tests./rrsets/PTR/ptr.unit.tests.',
|
||||
{'ttl': 60, 'rdata': ['a.unit.tests.']},
|
||||
Record.new(zone, 'ptr',
|
||||
{'ttl': 60, 'type': 'PTR',
|
||||
'value': 'a.unit.tests.'})),
|
||||
|
||||
# SPF
|
||||
('spf', 'SPF',
|
||||
'/v2/zones/unit.tests./rrsets/SPF/spf.unit.tests.',
|
||||
{'ttl': 60, 'rdata': ['v=spf1 -all']},
|
||||
Record.new(zone, 'spf',
|
||||
{'ttl': 60, 'type': 'SPF',
|
||||
'values': ['v=spf1 -all']})),
|
||||
|
||||
# SRV
|
||||
('_srv._tcp', 'SRV',
|
||||
'/v2/zones/unit.tests./rrsets/SRV/_srv._tcp.unit.tests.',
|
||||
{'ttl': 60, 'rdata': ['10 20 443 target.unit.tests.']},
|
||||
Record.new(zone, '_srv._tcp',
|
||||
{'ttl': 60, 'type': 'SRV',
|
||||
'values': [{'priority': 10,
|
||||
'weight': 20,
|
||||
'port': 443,
|
||||
'target': 'target.unit.tests.'}]})),
|
||||
|
||||
# TXT
|
||||
('txt', 'TXT',
|
||||
'/v2/zones/unit.tests./rrsets/TXT/txt.unit.tests.',
|
||||
{'ttl': 60, 'rdata': ['abc', 'def']},
|
||||
Record.new(zone, 'txt',
|
||||
{'ttl': 60, 'type': 'TXT',
|
||||
'values': ['abc', 'def']})),
|
||||
|
||||
# ALIAS
|
||||
('', 'ALIAS',
|
||||
'/v2/zones/unit.tests./rrsets/APEXALIAS/unit.tests.',
|
||||
{'ttl': 60, 'rdata': ['target.unit.tests.']},
|
||||
Record.new(zone, '',
|
||||
{'ttl': 60, 'type': 'ALIAS',
|
||||
'value': 'target.unit.tests.'})),
|
||||
|
||||
):
|
||||
# Validate path and payload based on record meet expectations
|
||||
path, payload = provider._gen_data(expected_record)
|
||||
self.assertEqual(expected_path, path)
|
||||
self.assertEqual(expected_payload, payload)
|
||||
|
||||
# Use generator for record and confirm the output matches
|
||||
rec = provider._record_for(zone, name, _type,
|
||||
expected_payload, False)
|
||||
path, payload = provider._gen_data(rec)
|
||||
self.assertEqual(expected_path, path)
|
||||
self.assertEqual(expected_payload, payload)
|
||||
def test_missing(self):
|
||||
with self.assertRaises(ModuleNotFoundError):
|
||||
from octodns.provider.ultra import UltraProvider
|
||||
UltraProvider
|
||||
|
||||
Reference in New Issue
Block a user