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.
125 lines
3.6 KiB
125 lines
3.6 KiB
#
|
|
#
|
|
#
|
|
|
|
from unittest import TestCase
|
|
from unittest.mock import call, patch
|
|
|
|
from octodns.processor.templating import Templating
|
|
from octodns.record import Record
|
|
from octodns.zone import Zone
|
|
|
|
|
|
def _find(zone, name):
|
|
return next(r for r in zone.records if r.name == name)
|
|
|
|
|
|
class TemplatingTest(TestCase):
|
|
def test_cname(self):
|
|
templ = Templating('test')
|
|
|
|
zone = Zone('unit.tests.', [])
|
|
cname = Record.new(
|
|
zone,
|
|
'cname',
|
|
{
|
|
'type': 'CNAME',
|
|
'ttl': 42,
|
|
'value': '_cname.{zone_name}something.else.',
|
|
},
|
|
lenient=True,
|
|
)
|
|
zone.add_record(cname)
|
|
noop = Record.new(
|
|
zone,
|
|
'noop',
|
|
{
|
|
'type': 'CNAME',
|
|
'ttl': 42,
|
|
'value': '_noop.nothing_to_do.something.else.',
|
|
},
|
|
lenient=True,
|
|
)
|
|
zone.add_record(noop)
|
|
|
|
got = templ.process_source_zone(zone, None)
|
|
cname = _find(got, 'cname')
|
|
self.assertEqual('_cname.unit.tests.something.else.', cname.value)
|
|
noop = _find(got, 'noop')
|
|
self.assertEqual('_noop.nothing_to_do.something.else.', noop.value)
|
|
|
|
def test_txt(self):
|
|
templ = Templating('test')
|
|
|
|
zone = Zone('unit.tests.', [])
|
|
txt = Record.new(
|
|
zone,
|
|
'txt',
|
|
{
|
|
'type': 'TXT',
|
|
'ttl': 42,
|
|
'value': 'There are {zone_num_records} record(s) in {zone_name}',
|
|
},
|
|
)
|
|
zone.add_record(txt)
|
|
noop = Record.new(
|
|
zone,
|
|
'noop',
|
|
{'type': 'TXT', 'ttl': 43, 'value': 'Nothing to template here.'},
|
|
)
|
|
zone.add_record(noop)
|
|
|
|
got = templ.process_source_zone(zone, None)
|
|
txt = _find(got, 'txt')
|
|
self.assertEqual('There are 2 record(s) in unit.tests.', txt.values[0])
|
|
noop = _find(got, 'noop')
|
|
self.assertEqual('Nothing to template here.', noop.values[0])
|
|
|
|
@patch('octodns.record.TxtValue.template')
|
|
def test_params(self, mock_template):
|
|
templ = Templating('test')
|
|
|
|
class DummySource:
|
|
|
|
def __init__(self, id):
|
|
self.id = id
|
|
|
|
zone = Zone('unit.tests.', [])
|
|
record_source = DummySource('record')
|
|
txt = Record.new(
|
|
zone,
|
|
'txt',
|
|
{
|
|
'type': 'TXT',
|
|
'ttl': 42,
|
|
'value': 'There are {zone_num_records} record(s) in {zone_name}',
|
|
},
|
|
source=record_source,
|
|
)
|
|
zone.add_record(txt)
|
|
|
|
templ.process_source_zone(
|
|
zone, sources=[record_source, DummySource('other')]
|
|
)
|
|
mock_template.assert_called_once()
|
|
self.assertEqual(
|
|
call(
|
|
{
|
|
'record_name': 'txt',
|
|
'record_decoded_name': 'txt',
|
|
'record_encoded_name': 'txt',
|
|
'record_fqdn': 'txt.unit.tests.',
|
|
'record_decoded_fqdn': 'txt.unit.tests.',
|
|
'record_encoded_fqdn': 'txt.unit.tests.',
|
|
'record_type': 'TXT',
|
|
'record_ttl': 42,
|
|
'record_source_id': 'record',
|
|
'zone_name': 'unit.tests.',
|
|
'zone_decoded_name': 'unit.tests.',
|
|
'zone_encoded_name': 'unit.tests.',
|
|
'zone_num_records': 1,
|
|
'zone_source_ids': 'record, other',
|
|
}
|
|
),
|
|
mock_template.call_args,
|
|
)
|