mirror of
https://github.com/github/octodns.git
synced 2024-05-11 05:55:00 +00:00
Add: New tests.
This commit is contained in:
Matt
@@ -18,6 +18,17 @@ from octodns.provider.yaml import YamlProvider
|
|||||||
from octodns.zone import Zone
|
from octodns.zone import Zone
|
||||||
|
|
||||||
|
|
||||||
|
def set_record_proxied_flag(record, proxied):
|
||||||
|
try:
|
||||||
|
record._octodns['cloudflare']['proxied'] = proxied
|
||||||
|
except KeyError:
|
||||||
|
record._octodns['cloudflare'] = {
|
||||||
|
'proxied': proxied
|
||||||
|
}
|
||||||
|
|
||||||
|
return record
|
||||||
|
|
||||||
|
|
||||||
class TestCloudflareProvider(TestCase):
|
class TestCloudflareProvider(TestCase):
|
||||||
expected = Zone('unit.tests.', [])
|
expected = Zone('unit.tests.', [])
|
||||||
source = YamlProvider('test', join(dirname(__file__), 'config'))
|
source = YamlProvider('test', join(dirname(__file__), 'config'))
|
||||||
@@ -757,3 +768,463 @@ class TestCloudflareProvider(TestCase):
|
|||||||
|
|
||||||
plan = provider.plan(wanted)
|
plan = provider.plan(wanted)
|
||||||
self.assertEquals(False, hasattr(plan, 'changes'))
|
self.assertEquals(False, hasattr(plan, 'changes'))
|
||||||
|
|
||||||
|
def test_proxied_ignore_ttl(self):
|
||||||
|
provider = CloudflareProvider('test', 'email', 'token')
|
||||||
|
|
||||||
|
provider.zone_records = Mock(return_value=[
|
||||||
|
{
|
||||||
|
"id": "fc12ab34cd5611334422ab3322997642",
|
||||||
|
"type": "A",
|
||||||
|
"name": "unit.tests.a",
|
||||||
|
"content": "1.2.3.4",
|
||||||
|
"proxiable": True,
|
||||||
|
"proxied": True,
|
||||||
|
"ttl": 1,
|
||||||
|
"locked": False,
|
||||||
|
"zone_id": "ff12ab34cd5611334422ab3322997650",
|
||||||
|
"zone_name": "unit.tests",
|
||||||
|
"modified_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"created_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"meta": {
|
||||||
|
"auto_added": False
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"id": "fc12ab34cd5611334422ab3322997643",
|
||||||
|
"type": "AAAA",
|
||||||
|
"name": "unit.tests.aaaa",
|
||||||
|
"content": "::1",
|
||||||
|
"proxiable": True,
|
||||||
|
"proxied": True,
|
||||||
|
"ttl": 1,
|
||||||
|
"locked": False,
|
||||||
|
"zone_id": "ff12ab34cd5611334422ab3322997650",
|
||||||
|
"zone_name": "unit.tests",
|
||||||
|
"modified_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"created_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"meta": {
|
||||||
|
"auto_added": False
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"id": "fc12ab34cd5611334422ab3322997644",
|
||||||
|
"type": "CNAME",
|
||||||
|
"name": "unit.tests.cname",
|
||||||
|
"content": "www.unit.tests",
|
||||||
|
"proxiable": True,
|
||||||
|
"proxied": True,
|
||||||
|
"ttl": 1,
|
||||||
|
"locked": False,
|
||||||
|
"zone_id": "ff12ab34cd5611334422ab3322997650",
|
||||||
|
"zone_name": "unit.tests",
|
||||||
|
"modified_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"created_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"meta": {
|
||||||
|
"auto_added": False
|
||||||
|
}
|
||||||
|
},
|
||||||
|
])
|
||||||
|
|
||||||
|
zone = Zone('unit.tests.', [])
|
||||||
|
provider.populate(zone)
|
||||||
|
|
||||||
|
wanted = Zone('unit.tests.', [])
|
||||||
|
wanted.add_record(
|
||||||
|
set_record_proxied_flag(
|
||||||
|
Record.new(wanted, 'unit.tests.a', {
|
||||||
|
'ttl': 120,
|
||||||
|
'type': 'A',
|
||||||
|
'value': '1.2.3.4'
|
||||||
|
}), True
|
||||||
|
)
|
||||||
|
)
|
||||||
|
wanted.add_record(
|
||||||
|
set_record_proxied_flag(
|
||||||
|
Record.new(wanted, 'unit.tests.aaaa', {
|
||||||
|
'ttl': 250,
|
||||||
|
'type': 'AAAA',
|
||||||
|
'value': '::1'
|
||||||
|
}), True
|
||||||
|
)
|
||||||
|
)
|
||||||
|
wanted.add_record(
|
||||||
|
set_record_proxied_flag(
|
||||||
|
Record.new(wanted, 'unit.tests.cname', {
|
||||||
|
'ttl': 300,
|
||||||
|
'type': 'CNAME',
|
||||||
|
'value': 'www.unit.tests.'
|
||||||
|
}), True
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
plan = provider.plan(wanted)
|
||||||
|
self.assertEquals(False, hasattr(plan, 'changes'))
|
||||||
|
|
||||||
|
def test_enable_proxied(self):
|
||||||
|
provider = CloudflareProvider('test', 'email', 'token')
|
||||||
|
|
||||||
|
provider.zone_records = Mock(return_value=[
|
||||||
|
{
|
||||||
|
"id": "fc12ab34cd5611334422ab3322997642",
|
||||||
|
"type": "A",
|
||||||
|
"name": "unit.tests.a",
|
||||||
|
"content": "1.2.3.4",
|
||||||
|
"proxiable": True,
|
||||||
|
"proxied": False,
|
||||||
|
"ttl": 120,
|
||||||
|
"locked": False,
|
||||||
|
"zone_id": "ff12ab34cd5611334422ab3322997650",
|
||||||
|
"zone_name": "unit.tests",
|
||||||
|
"modified_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"created_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"meta": {
|
||||||
|
"auto_added": False
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"id": "fc12ab34cd5611334422ab3322997643",
|
||||||
|
"type": "AAAA",
|
||||||
|
"name": "unit.tests.aaaa",
|
||||||
|
"content": "::1",
|
||||||
|
"proxiable": True,
|
||||||
|
"proxied": False,
|
||||||
|
"ttl": 250,
|
||||||
|
"locked": False,
|
||||||
|
"zone_id": "ff12ab34cd5611334422ab3322997650",
|
||||||
|
"zone_name": "unit.tests",
|
||||||
|
"modified_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"created_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"meta": {
|
||||||
|
"auto_added": False
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"id": "fc12ab34cd5611334422ab3322997644",
|
||||||
|
"type": "CNAME",
|
||||||
|
"name": "unit.tests.cname",
|
||||||
|
"content": "www.unit.tests",
|
||||||
|
"proxiable": True,
|
||||||
|
"proxied": False,
|
||||||
|
"ttl": 300,
|
||||||
|
"locked": False,
|
||||||
|
"zone_id": "ff12ab34cd5611334422ab3322997650",
|
||||||
|
"zone_name": "unit.tests",
|
||||||
|
"modified_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"created_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"meta": {
|
||||||
|
"auto_added": False
|
||||||
|
}
|
||||||
|
},
|
||||||
|
])
|
||||||
|
|
||||||
|
zone = Zone('unit.tests.', [])
|
||||||
|
provider.populate(zone)
|
||||||
|
|
||||||
|
wanted = Zone('unit.tests.', [])
|
||||||
|
wanted.add_record(
|
||||||
|
set_record_proxied_flag(
|
||||||
|
Record.new(wanted, 'unit.tests.a', {
|
||||||
|
'ttl': 120,
|
||||||
|
'type': 'A',
|
||||||
|
'value': '1.2.3.4'
|
||||||
|
}), True
|
||||||
|
)
|
||||||
|
)
|
||||||
|
wanted.add_record(
|
||||||
|
set_record_proxied_flag(
|
||||||
|
Record.new(wanted, 'unit.tests.aaaa', {
|
||||||
|
'ttl': 250,
|
||||||
|
'type': 'AAAA',
|
||||||
|
'value': '::1'
|
||||||
|
}), True
|
||||||
|
)
|
||||||
|
)
|
||||||
|
wanted.add_record(
|
||||||
|
set_record_proxied_flag(
|
||||||
|
Record.new(wanted, 'unit.tests.cname', {
|
||||||
|
'ttl': 300,
|
||||||
|
'type': 'CNAME',
|
||||||
|
'value': 'www.unit.tests.'
|
||||||
|
}), True
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
plan = provider.plan(wanted)
|
||||||
|
self.assertEquals(True, hasattr(plan, 'changes'))
|
||||||
|
|
||||||
|
def test_disable_proxied(self):
|
||||||
|
provider = CloudflareProvider('test', 'email', 'token')
|
||||||
|
|
||||||
|
provider.zone_records = Mock(return_value=[
|
||||||
|
{
|
||||||
|
"id": "fc12ab34cd5611334422ab3322997642",
|
||||||
|
"type": "A",
|
||||||
|
"name": "unit.tests.a",
|
||||||
|
"content": "1.2.3.4",
|
||||||
|
"proxiable": True,
|
||||||
|
"proxied": True,
|
||||||
|
"ttl": 120,
|
||||||
|
"locked": False,
|
||||||
|
"zone_id": "ff12ab34cd5611334422ab3322997650",
|
||||||
|
"zone_name": "unit.tests",
|
||||||
|
"modified_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"created_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"meta": {
|
||||||
|
"auto_added": False
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"id": "fc12ab34cd5611334422ab3322997643",
|
||||||
|
"type": "AAAA",
|
||||||
|
"name": "unit.tests.aaaa",
|
||||||
|
"content": "::1",
|
||||||
|
"proxiable": True,
|
||||||
|
"proxied": True,
|
||||||
|
"ttl": 250,
|
||||||
|
"locked": False,
|
||||||
|
"zone_id": "ff12ab34cd5611334422ab3322997650",
|
||||||
|
"zone_name": "unit.tests",
|
||||||
|
"modified_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"created_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"meta": {
|
||||||
|
"auto_added": False
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"id": "fc12ab34cd5611334422ab3322997644",
|
||||||
|
"type": "CNAME",
|
||||||
|
"name": "unit.tests.cname",
|
||||||
|
"content": "www.unit.tests",
|
||||||
|
"proxiable": True,
|
||||||
|
"proxied": True,
|
||||||
|
"ttl": 300,
|
||||||
|
"locked": False,
|
||||||
|
"zone_id": "ff12ab34cd5611334422ab3322997650",
|
||||||
|
"zone_name": "unit.tests",
|
||||||
|
"modified_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"created_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"meta": {
|
||||||
|
"auto_added": False
|
||||||
|
}
|
||||||
|
},
|
||||||
|
])
|
||||||
|
|
||||||
|
zone = Zone('unit.tests.', [])
|
||||||
|
provider.populate(zone)
|
||||||
|
|
||||||
|
wanted = Zone('unit.tests.', [])
|
||||||
|
wanted.add_record(
|
||||||
|
set_record_proxied_flag(
|
||||||
|
Record.new(wanted, 'unit.tests.a', {
|
||||||
|
'ttl': 120,
|
||||||
|
'type': 'A',
|
||||||
|
'value': '1.2.3.4'
|
||||||
|
}), False
|
||||||
|
)
|
||||||
|
)
|
||||||
|
wanted.add_record(
|
||||||
|
set_record_proxied_flag(
|
||||||
|
Record.new(wanted, 'unit.tests.aaaa', {
|
||||||
|
'ttl': 250,
|
||||||
|
'type': 'AAAA',
|
||||||
|
'value': '::1'
|
||||||
|
}), False
|
||||||
|
)
|
||||||
|
)
|
||||||
|
wanted.add_record(
|
||||||
|
set_record_proxied_flag(
|
||||||
|
Record.new(wanted, 'unit.tests.cname', {
|
||||||
|
'ttl': 300,
|
||||||
|
'type': 'CNAME',
|
||||||
|
'value': 'www.unit.tests.'
|
||||||
|
}), False
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
plan = provider.plan(wanted)
|
||||||
|
self.assertEquals(True, hasattr(plan, 'changes'))
|
||||||
|
|
||||||
|
def test_leave_proxied_disabled(self):
|
||||||
|
provider = CloudflareProvider('test', 'email', 'token')
|
||||||
|
|
||||||
|
provider.zone_records = Mock(return_value=[
|
||||||
|
{
|
||||||
|
"id": "fc12ab34cd5611334422ab3322997642",
|
||||||
|
"type": "A",
|
||||||
|
"name": "unit.tests.a",
|
||||||
|
"content": "1.2.3.4",
|
||||||
|
"proxiable": True,
|
||||||
|
"proxied": False,
|
||||||
|
"ttl": 120,
|
||||||
|
"locked": False,
|
||||||
|
"zone_id": "ff12ab34cd5611334422ab3322997650",
|
||||||
|
"zone_name": "unit.tests",
|
||||||
|
"modified_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"created_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"meta": {
|
||||||
|
"auto_added": False
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"id": "fc12ab34cd5611334422ab3322997643",
|
||||||
|
"type": "AAAA",
|
||||||
|
"name": "unit.tests.aaaa",
|
||||||
|
"content": "::1",
|
||||||
|
"proxiable": True,
|
||||||
|
"proxied": False,
|
||||||
|
"ttl": 250,
|
||||||
|
"locked": False,
|
||||||
|
"zone_id": "ff12ab34cd5611334422ab3322997650",
|
||||||
|
"zone_name": "unit.tests",
|
||||||
|
"modified_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"created_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"meta": {
|
||||||
|
"auto_added": False
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"id": "fc12ab34cd5611334422ab3322997644",
|
||||||
|
"type": "CNAME",
|
||||||
|
"name": "unit.tests.cname",
|
||||||
|
"content": "www.unit.tests",
|
||||||
|
"proxiable": True,
|
||||||
|
"proxied": False,
|
||||||
|
"ttl": 300,
|
||||||
|
"locked": False,
|
||||||
|
"zone_id": "ff12ab34cd5611334422ab3322997650",
|
||||||
|
"zone_name": "unit.tests",
|
||||||
|
"modified_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"created_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"meta": {
|
||||||
|
"auto_added": False
|
||||||
|
}
|
||||||
|
},
|
||||||
|
])
|
||||||
|
|
||||||
|
zone = Zone('unit.tests.', [])
|
||||||
|
provider.populate(zone)
|
||||||
|
|
||||||
|
wanted = Zone('unit.tests.', [])
|
||||||
|
wanted.add_record(
|
||||||
|
set_record_proxied_flag(
|
||||||
|
Record.new(wanted, 'unit.tests.a', {
|
||||||
|
'ttl': 120,
|
||||||
|
'type': 'A',
|
||||||
|
'value': '1.2.3.4'
|
||||||
|
}), False
|
||||||
|
)
|
||||||
|
)
|
||||||
|
wanted.add_record(
|
||||||
|
set_record_proxied_flag(
|
||||||
|
Record.new(wanted, 'unit.tests.aaaa', {
|
||||||
|
'ttl': 250,
|
||||||
|
'type': 'AAAA',
|
||||||
|
'value': '::1'
|
||||||
|
}), False
|
||||||
|
)
|
||||||
|
)
|
||||||
|
wanted.add_record(
|
||||||
|
set_record_proxied_flag(
|
||||||
|
Record.new(wanted, 'unit.tests.cname', {
|
||||||
|
'ttl': 300,
|
||||||
|
'type': 'CNAME',
|
||||||
|
'value': 'www.unit.tests.'
|
||||||
|
}), False
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
plan = provider.plan(wanted)
|
||||||
|
self.assertEquals(False, hasattr(plan, 'changes'))
|
||||||
|
|
||||||
|
def test_leave_proxied_enabled(self):
|
||||||
|
provider = CloudflareProvider('test', 'email', 'token')
|
||||||
|
|
||||||
|
provider.zone_records = Mock(return_value=[
|
||||||
|
{
|
||||||
|
"id": "fc12ab34cd5611334422ab3322997642",
|
||||||
|
"type": "A",
|
||||||
|
"name": "unit.tests.a",
|
||||||
|
"content": "1.2.3.4",
|
||||||
|
"proxiable": True,
|
||||||
|
"proxied": True,
|
||||||
|
"ttl": 120,
|
||||||
|
"locked": False,
|
||||||
|
"zone_id": "ff12ab34cd5611334422ab3322997650",
|
||||||
|
"zone_name": "unit.tests",
|
||||||
|
"modified_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"created_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"meta": {
|
||||||
|
"auto_added": False
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"id": "fc12ab34cd5611334422ab3322997643",
|
||||||
|
"type": "AAAA",
|
||||||
|
"name": "unit.tests.aaaa",
|
||||||
|
"content": "::1",
|
||||||
|
"proxiable": True,
|
||||||
|
"proxied": True,
|
||||||
|
"ttl": 250,
|
||||||
|
"locked": False,
|
||||||
|
"zone_id": "ff12ab34cd5611334422ab3322997650",
|
||||||
|
"zone_name": "unit.tests",
|
||||||
|
"modified_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"created_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"meta": {
|
||||||
|
"auto_added": False
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"id": "fc12ab34cd5611334422ab3322997644",
|
||||||
|
"type": "CNAME",
|
||||||
|
"name": "unit.tests.cname",
|
||||||
|
"content": "www.unit.tests",
|
||||||
|
"proxiable": True,
|
||||||
|
"proxied": True,
|
||||||
|
"ttl": 300,
|
||||||
|
"locked": False,
|
||||||
|
"zone_id": "ff12ab34cd5611334422ab3322997650",
|
||||||
|
"zone_name": "unit.tests",
|
||||||
|
"modified_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"created_on": "2017-03-11T18:01:43.420689Z",
|
||||||
|
"meta": {
|
||||||
|
"auto_added": False
|
||||||
|
}
|
||||||
|
},
|
||||||
|
])
|
||||||
|
|
||||||
|
zone = Zone('unit.tests.', [])
|
||||||
|
provider.populate(zone)
|
||||||
|
|
||||||
|
wanted = Zone('unit.tests.', [])
|
||||||
|
wanted.add_record(
|
||||||
|
set_record_proxied_flag(
|
||||||
|
Record.new(wanted, 'unit.tests.a', {
|
||||||
|
'ttl': 120,
|
||||||
|
'type': 'A',
|
||||||
|
'value': '1.2.3.4'
|
||||||
|
}), True
|
||||||
|
)
|
||||||
|
)
|
||||||
|
wanted.add_record(
|
||||||
|
set_record_proxied_flag(
|
||||||
|
Record.new(wanted, 'unit.tests.aaaa', {
|
||||||
|
'ttl': 250,
|
||||||
|
'type': 'AAAA',
|
||||||
|
'value': '::1'
|
||||||
|
}), True
|
||||||
|
)
|
||||||
|
)
|
||||||
|
wanted.add_record(
|
||||||
|
set_record_proxied_flag(
|
||||||
|
Record.new(wanted, 'unit.tests.cname', {
|
||||||
|
'ttl': 300,
|
||||||
|
'type': 'CNAME',
|
||||||
|
'value': 'www.unit.tests.'
|
||||||
|
}), True
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
plan = provider.plan(wanted)
|
||||||
|
self.assertEquals(False, hasattr(plan, 'changes'))
|
||||||
|
|||||||
Reference in New Issue
Block a user