mirror of
https://github.com/github/octodns.git
synced 2024-05-11 05:55:00 +00:00
516 lines
21 KiB
Python
516 lines
21 KiB
Python
from __future__ import 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 six import text_type
|
|
from six.moves.urllib import parse
|
|
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 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.assertEquals('Unauthorized', text_type(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.assertEquals(1, mock.call_count)
|
|
expected_payload = "grant_type=password&username=user&"\
|
|
"password=rightpass"
|
|
self.assertEquals(parse.parse_qs(mock.last_request.text),
|
|
parse.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.assertEquals(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.assertEquals(1, mock.call_count)
|
|
self.assertEquals(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.assertEquals(1, mock.call_count)
|
|
self.assertEquals(1, len(zones))
|
|
self.assertEquals('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.assertEquals(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.assertEquals('Unauthorized', text_type(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.assertEquals(mock.call_count, 2)
|
|
|
|
# Populate the same zone again and confirm cache is hit
|
|
self.assertTrue(provider.zone_records(zone))
|
|
self.assertEquals(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.assertEquals(set(), zone.records)
|
|
|
|
# re-populating the same non-existent zone uses cache and makes no
|
|
# calls
|
|
again = Zone('unit.tests.', [])
|
|
provider.populate(again)
|
|
self.assertEquals(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.assertEquals('octodns1.test.', zone.name)
|
|
self.assertEquals(12, len(zone.records))
|
|
self.assertEquals(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.assertEquals(15, len(plan.changes))
|
|
self.assertEquals(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.assertEquals(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.assertEquals(11, len(plan.changes))
|
|
self.assertEquals(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)
|