1
0
mirror of https://github.com/github/octodns.git synced 2024-05-11 05:55:00 +00:00

Real testing of register_type

This commit is contained in:
Ross McFarland
2022-03-16 12:26:29 -07:00
parent 89e31e34db
commit ea5000e191

View File

@@ -12,7 +12,8 @@ from octodns.record import ARecord, AaaaRecord, AliasRecord, CaaRecord, \
LocValue, MxRecord, MxValue, NaptrRecord, NaptrValue, NsRecord, \
PtrRecord, Record, RecordException, SshfpRecord, SshfpValue, SpfRecord, \
SrvRecord, SrvValue, TxtRecord, Update, UrlfwdRecord, UrlfwdValue, \
ValidationError, _Dynamic, _DynamicPool, _DynamicRule
ValidationError, _Dynamic, _DynamicPool, _DynamicRule, _NsValue, \
_ValuesMixin
from octodns.zone import Zone
from helpers import DynamicProvider, GeoProvider, SimpleProvider
@@ -27,6 +28,18 @@ class TestRecord(TestCase):
self.assertEqual('Type "A" already registered by '
'octodns.record.ARecord', str(ctx.exception))
class AaRecord(_ValuesMixin, Record):
_type = 'AA'
_value_type = _NsValue
Record.register_type('AA', AaRecord)
aa = Record.new(self.zone, 'registered', {
'ttl': 360,
'type': 'AA',
'value': 'does.not.matter.',
})
self.assertEqual(AaRecord, aa.__class__)
def test_lowering(self):
record = ARecord(self.zone, 'MiXeDcAsE', {
'ttl': 30,