mirror of
https://github.com/github/octodns.git
synced 2024-05-11 05:55:00 +00:00
90 lines
2.9 KiB
Python
90 lines
2.9 KiB
Python
from unittest import TestCase
|
|
|
|
from octodns.processor.restrict import (
|
|
RestrictionException,
|
|
TtlRestrictionFilter,
|
|
)
|
|
from octodns.record import Record
|
|
from octodns.zone import Zone
|
|
|
|
|
|
class TestTtlRestrictionFilter(TestCase):
|
|
zone = Zone('unit.tests.', [])
|
|
matches = Record.new(
|
|
zone, 'matches', {'type': 'A', 'ttl': 42, 'value': '1.2.3.4'}
|
|
)
|
|
zone.add_record(matches)
|
|
doesnt = Record.new(
|
|
zone, 'doesnt', {'type': 'A', 'ttl': 42, 'value': '2.3.4.5'}
|
|
)
|
|
zone.add_record(doesnt)
|
|
matchable1 = Record.new(
|
|
zone, 'start-f43ad96-end', {'type': 'A', 'ttl': 42, 'value': '3.4.5.6'}
|
|
)
|
|
zone.add_record(matchable1)
|
|
matchable2 = Record.new(
|
|
zone, 'start-a3b444c-end', {'type': 'A', 'ttl': 42, 'value': '4.5.6.7'}
|
|
)
|
|
zone.add_record(matchable2)
|
|
|
|
def test_restrict_ttl(self):
|
|
# configured values
|
|
restrictor = TtlRestrictionFilter('test', min_ttl=32, max_ttl=1024)
|
|
|
|
zone = Zone('unit.tests.', [])
|
|
good = Record.new(
|
|
zone, 'good', {'type': 'A', 'ttl': 42, 'value': '1.2.3.4'}
|
|
)
|
|
zone.add_record(good)
|
|
|
|
restricted = restrictor.process_source_zone(zone)
|
|
self.assertEqual(zone.records, restricted.records)
|
|
|
|
low = Record.new(
|
|
zone, 'low', {'type': 'A', 'ttl': 16, 'value': '1.2.3.4'}
|
|
)
|
|
copy = zone.copy()
|
|
copy.add_record(low)
|
|
with self.assertRaises(RestrictionException) as ctx:
|
|
restrictor.process_source_zone(copy)
|
|
self.assertEqual(
|
|
'low.unit.tests. ttl=16 too low, min_ttl=32', str(ctx.exception)
|
|
)
|
|
|
|
high = Record.new(
|
|
zone, 'high', {'type': 'A', 'ttl': 2048, 'value': '1.2.3.4'}
|
|
)
|
|
copy = zone.copy()
|
|
copy.add_record(high)
|
|
with self.assertRaises(RestrictionException) as ctx:
|
|
restrictor.process_source_zone(copy)
|
|
self.assertEqual(
|
|
'high.unit.tests. ttl=2048 too high, max_ttl=1024',
|
|
str(ctx.exception),
|
|
)
|
|
|
|
# defaults
|
|
restrictor = TtlRestrictionFilter('test')
|
|
low = Record.new(
|
|
zone, 'low', {'type': 'A', 'ttl': 0, 'value': '1.2.3.4'}
|
|
)
|
|
copy = zone.copy()
|
|
copy.add_record(low)
|
|
with self.assertRaises(RestrictionException) as ctx:
|
|
restrictor.process_source_zone(copy)
|
|
self.assertEqual(
|
|
'low.unit.tests. ttl=0 too low, min_ttl=1', str(ctx.exception)
|
|
)
|
|
|
|
high = Record.new(
|
|
zone, 'high', {'type': 'A', 'ttl': 999999, 'value': '1.2.3.4'}
|
|
)
|
|
copy = zone.copy()
|
|
copy.add_record(high)
|
|
with self.assertRaises(RestrictionException) as ctx:
|
|
restrictor.process_source_zone(copy)
|
|
self.assertEqual(
|
|
'high.unit.tests. ttl=999999 too high, max_ttl=604800',
|
|
str(ctx.exception),
|
|
)
|