def test_domain_incrementing_notified_serial_from_yesterday(self):
test = Domain(0, 'domain.name', notified_serial=2013121003)
self.assertEqual(test.notified_serial, 2013121003)
test.touch_serial()
self.assertEqual(test.notified_serial, 2013121101)
test.touch_serial()
self.assertEqual(test.notified_serial, 2013121102)
def test_domain_incrementing_notified_serial_from_None(self):
test = Domain(0, 'domain.name', notified_serial=None)
self.assertIsNone(test.notified_serial)
test.touch_serial()
self.assertEqual(test.notified_serial, 2013121101)
test.touch_serial()
self.assertEqual(test.notified_serial, 2013121102)
def test_domain_incrementing_notified_serial_from_yesterday(self):
test = Domain(0, 'domain.name', notified_serial=2013121003)
self.assertEqual(test.notified_serial, 2013121003)
test.touch_serial()
self.assertEqual(test.notified_serial, 2013121101)
test.touch_serial()
self.assertEqual(test.notified_serial, 2013121102)
def test_domain_incrementing_notified_serial_from_None(self):
test = Domain(0, 'domain.name', notified_serial=None)
self.assertIsNone(test.notified_serial)
test.touch_serial()
self.assertEqual(test.notified_serial, 2013121101)
test.touch_serial()
self.assertEqual(test.notified_serial, 2013121102)
def setup_mock_pdns(session):
"""
Setups a mock PDNS installation with existing domains and records using
the provided session
@param session: the sqlalchemy session
@return: a namedtuple of the PDNS database in the form (domains, records)
holding the domains in the form of:
stable_a stable_ptr_134 stable_ptr_135
testing_a testing_ptr_132 testing_ptr_133
and with records in the form of:
stable_a_134 stable_a_135 stable_ptr_134 stable_ptr_135
testing_a_132 testing_a_133 testing_ptr_132 testing_ptr_133
cname_record record_with_cname txt_record record_with_txt
"""
CurrentPdns = collections.namedtuple('CurrentPdns', 'domains records')
PdnsDomainRows = collections.namedtuple('PdnsDomainRows',
'tld_a '
'stable_a testing_a '
'super_stable_a super_testing_a '
'testing_ptr_132 testing_ptr_133 '
'stable_ptr_134 stable_ptr_135 '
'wide_domain_ptr_10_10 wide_domain_ptr_10')
current_domains = PdnsDomainRows(
Domain(0, 'tld'),
Domain(1, 'stable.tld'),
Domain(2, 'test.tld'),
Domain(3, 'super.stable.tld'),
Domain(4, 'super.test.tld'),
Domain(5, '132.168.192.in-addr.arpa'),
Domain(6, '133.168.192.in-addr.arpa'),
Domain(7, '134.168.192.in-addr.arpa'),
Domain(8, '135.168.192.in-addr.arpa'),
Domain(9, '10.10.in-addr.arpa'),
Domain(10, '10.in-addr.arpa'),
)
current_domains = current_domains
PdnsRecordRows = collections.namedtuple('PdnsRecordRows',
'stable_a_134 stable_a_135 '
'stable_ptr_134 stable_ptr_135 '
'testing_a_132 testing_a_133 '
'testing_ptr_132 testing_ptr_133 '
'cname_record record_with_cname '
'txt_record record_with_txt tld_a tld_ptr')
current_records = PdnsRecordRows(Record(0, current_domains.stable_a.id, 'test_existing.stable.tld',
'A', '192.168.134.2'),
Record(1, current_domains.stable_a.id, 'test_existing2.stable.tld',
'A', '192.168.135.2'),
Record(2, current_domains.stable_ptr_134.id, '2.134.168.192.in-addr.arpa',
'PTR', 'test_existing.stable.tld'),
Record(3, current_domains.stable_ptr_135.id, '2.135.168.192.in-addr.arpa',
'PTR', 'test_existing2.stable.tld'),
Record(4, current_domains.testing_a.id, 'test_existing.test.tld',
'A', '192.168.132.2'),
Record(5, current_domains.testing_a.id, 'test_existing2.test.tld',
'A', '192.168.133.2'),
Record(6, current_domains.testing_ptr_132.id, '2.132.168.192.in-addr.arpa',
'PTR', 'test_existing.test.tld'),
Record(7, current_domains.testing_ptr_133.id, '2.133.168.192.in-addr.arpa',
'PTR', 'test_existing2.test.tld'),
Record(21, current_domains.testing_a.id, 'cnamer.test.tld',
'CNAME', 'cnamee.test.tld'),
Record(22, current_domains.testing_a.id, 'cnamee.test.tld',
'A', '192.168.133.57'),
Record(23, current_domains.testing_a.id, 'text.test.tld',
'TXT', 'this is a text record'),
Record(24, current_domains.testing_a.id, 'text.test.tld',
'A', '192.168.133.61'),
Record(25, current_domains.tld_a.id, 'big.domain.tld',
'A', '10.10.111.61'),
Record(26, current_domains.wide_domain_ptr_10_10.id, '61.111.10.10.in-addr.arpa',
'PTR', 'big.domain.tld')
)
Base.metadata.create_all(session.bind)
# the deepcopy allows the original records to still be used as
# comparisons when deletes happen on the actual db
for domain in deepcopy(current_domains):
session.add(domain)
session.commit()
for record in deepcopy(current_records):
session.add(record)
session.commit()
return CurrentPdns(current_domains, current_records)
