mirror of
https://github.com/github/octodns.git
synced 2024-05-11 05:55:00 +00:00
Include octodns special section in record __repr__
This commit is contained in:
@@ -328,7 +328,10 @@ class ValuesMixin(object):
|
||||
def __repr__(self):
|
||||
values = "', '".join([str(v) for v in self.values])
|
||||
klass = self.__class__.__name__
|
||||
return f"<{klass} {self._type} {self.ttl}, {self.decoded_fqdn}, ['{values}']>"
|
||||
octodns = ''
|
||||
if self._octodns:
|
||||
octodns = f', {self._octodns}'
|
||||
return f"<{klass} {self._type} {self.ttl}, {self.decoded_fqdn}, ['{values}']{octodns}>"
|
||||
|
||||
|
||||
class ValueMixin(object):
|
||||
@@ -371,4 +374,7 @@ class ValueMixin(object):
|
||||
|
||||
def __repr__(self):
|
||||
klass = self.__class__.__name__
|
||||
return f'<{klass} {self._type} {self.ttl}, {self.decoded_fqdn}, {self.value}>'
|
||||
octodns = ''
|
||||
if self._octodns:
|
||||
octodns = f', {self._octodns}'
|
||||
return f'<{klass} {self._type} {self.ttl}, {self.decoded_fqdn}, {self.value}{octodns}>'
|
||||
|
||||
@@ -654,3 +654,51 @@ class TestRecordValidation(TestCase):
|
||||
),
|
||||
)
|
||||
self.assertEqual('needle', record.context)
|
||||
|
||||
def test_values_mixin_repr(self):
|
||||
# ValuesMixin
|
||||
record = Record.new(
|
||||
self.zone,
|
||||
'www',
|
||||
{
|
||||
'ttl': 42,
|
||||
'type': 'A',
|
||||
'values': ['1.2.3.4', '2.3.4.5'],
|
||||
'octodns': {'key': 'value'},
|
||||
},
|
||||
)
|
||||
# has the octodns special section
|
||||
self.assertEqual(
|
||||
"<ARecord A 42, www.unit.tests., ['1.2.3.4', '2.3.4.5'], {'key': 'value'}>",
|
||||
record.__repr__(),
|
||||
)
|
||||
# no special section
|
||||
record._octodns = {}
|
||||
self.assertEqual(
|
||||
"<ARecord A 42, www.unit.tests., ['1.2.3.4', '2.3.4.5']>",
|
||||
record.__repr__(),
|
||||
)
|
||||
|
||||
def test_value_mixin_repr(self):
|
||||
# ValueMixin
|
||||
record = Record.new(
|
||||
self.zone,
|
||||
'pointer',
|
||||
{
|
||||
'ttl': 43,
|
||||
'type': 'CNAME',
|
||||
'value': 'unit.tests.',
|
||||
'octodns': {'key': 42},
|
||||
},
|
||||
)
|
||||
# has the octodns special section
|
||||
self.assertEqual(
|
||||
"<CnameRecord CNAME 43, pointer.unit.tests., unit.tests., {'key': 42}>",
|
||||
record.__repr__(),
|
||||
)
|
||||
# no special section
|
||||
record._octodns = {}
|
||||
self.assertEqual(
|
||||
'<CnameRecord CNAME 43, pointer.unit.tests., unit.tests.>',
|
||||
record.__repr__(),
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user