You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
70 lines
2.0 KiB
70 lines
2.0 KiB
#
|
|
#
|
|
#
|
|
|
|
from __future__ import absolute_import, division, print_function, \
|
|
unicode_literals
|
|
|
|
from unittest import TestCase
|
|
|
|
from octodns.processor.awsacm import AwsAcmMangingProcessor
|
|
from octodns.record import Record
|
|
from octodns.zone import Zone
|
|
|
|
zone = Zone('unit.tests.', [])
|
|
records = {
|
|
'root': Record.new(zone, '_deadbeef', {
|
|
'ttl': 30,
|
|
'type': 'CNAME',
|
|
'value': '_0123456789abcdef.acm-validations.aws.',
|
|
}),
|
|
'sub': Record.new(zone, '_deadbeef.sub', {
|
|
'ttl': 30,
|
|
'type': 'CNAME',
|
|
'value': '_0123456789abcdef.acm-validations.aws.',
|
|
}),
|
|
'not-cname': Record.new(zone, '_deadbeef.not-cname', {
|
|
'ttl': 30,
|
|
'type': 'AAAA',
|
|
'value': '::1',
|
|
}),
|
|
'not-acm': Record.new(zone, '_not-acm', {
|
|
'ttl': 30,
|
|
'type': 'CNAME',
|
|
'value': 'localhost.unit.tests.',
|
|
}),
|
|
}
|
|
|
|
|
|
class TestAwsAcmMangingProcessor(TestCase):
|
|
|
|
def test_process_zones(self):
|
|
acm = AwsAcmMangingProcessor('acm')
|
|
|
|
source = Zone(zone.name, [])
|
|
# Unrelated stuff that should be untouched
|
|
source.add_record(records['not-cname'])
|
|
source.add_record(records['not-acm'])
|
|
# ACM records that should be ignored
|
|
source.add_record(records['root'])
|
|
source.add_record(records['sub'])
|
|
|
|
got = acm.process_source_zone(source)
|
|
self.assertEqual([
|
|
'_deadbeef.not-cname',
|
|
'_not-acm',
|
|
], sorted([r.name for r in got.records]))
|
|
|
|
existing = Zone(zone.name, [])
|
|
# Unrelated stuff that should be untouched
|
|
existing.add_record(records['not-cname'])
|
|
existing.add_record(records['not-acm'])
|
|
# Stuff that will be ignored
|
|
existing.add_record(records['root'])
|
|
existing.add_record(records['sub'])
|
|
|
|
got = acm.process_target_zone(existing)
|
|
self.assertEqual([
|
|
'_deadbeef.not-cname',
|
|
'_not-acm'
|
|
], sorted([r.name for r in got.records]))
|