Пример #1
0
    def test_save_autoserial_greater(self):
        saved_filename = tempfile.mkstemp()[1]
        self.zone.save(saved_filename, autoserial=True)

        z = Zone('example.com.')
        z.load_from_file(saved_filename)
        self.failUnless(z.root.soa.serial >= self.zone.root.soa.serial)
Пример #2
0
    def setUp(self):
        self.zone = Zone('example.com.')
        zone_file = os.path.join(os.path.dirname(__file__), 'files', 'example.com')
        self.zone.load_from_file(zone_file)

        self.zone.root.soa.mname = 'mname.example.com.'
        self.zone.root.soa.rname = 'rname.example.com.'
        self.zone.root.soa.serial += 1
        self.zone.root.soa.refresh = 1
        self.zone.root.soa.retry = 2
        self.zone.root.soa.expire = 3
        self.zone.root.soa.minttl = 4

        self.zone.add_name('zip.example.com.')
        self.zone.names['zip.example.com.'].records('A', create=True).add('10.9.8.7')

        mx = self.zone.names['foo.example.com.'].records('MX')
        mx.delete( (10, 'mail.example.com.') )
        mx.add( (30, 'anothermail.example.com.') )

        self.zone.names['bar.example.com.'].records('A').add('10.20.30.40')

        self.zone.delete_name('foofoo.example.com.')

        self.saved_filename = tempfile.mkstemp()[1]
        # self.saved_filename = '/var/tmp/foo.tmp'
        self.zone.save(self.saved_filename)
Пример #3
0
    def setUp(self):
        zone_file = os.path.join(os.path.dirname(__file__), 'files', 'example.com')
        self.zone = Zone('example.com.')
        self.zone.load_from_file(zone_file)

        revzone_file = os.path.join(os.path.dirname(__file__), 'files', '10.in-addr.arpa')
        self.revzone = Zone('10.in-addr.arpa.')
        self.revzone.load_from_file(revzone_file)
Пример #4
0
class ZoneModifySaveTest(unittest.TestCase):
    def setUp(self):
        self.zone = Zone('example.com.')
        zone_file = os.path.join(os.path.dirname(__file__), 'files',
                                 'example.com')
        self.zone.load_from_file(zone_file)

        self.zone.root.soa.mname = 'mname.example.com.'
        self.zone.root.soa.rname = 'rname.example.com.'
        self.zone.root.soa.serial += 1
        self.zone.root.soa.refresh = 1
        self.zone.root.soa.retry = 2
        self.zone.root.soa.expire = 3
        self.zone.root.soa.minttl = 4

        self.zone.add_name('zip.example.com.')
        self.zone.names['zip.example.com.'].records(
            'A', create=True).add('10.9.8.7')

        mx = self.zone.names['foo.example.com.'].records('MX')
        mx.delete((10, 'mail.example.com.'))
        mx.add((30, 'anothermail.example.com.'))

        self.zone.names['bar.example.com.'].records('A').add('10.20.30.40')

        self.zone.delete_name('foofoo.example.com.')

        self.saved_filename = tempfile.mkstemp()[1]
        # self.saved_filename = '/var/tmp/foo.tmp'
        self.zone.save(self.saved_filename)

    def test_file_exists(self):
        self.failUnless(os.path.exists(self.saved_filename))

    def test_file_size(self):
        size = os.stat(self.saved_filename)[6]
        self.failUnlessEqual(size, 534)

    def test_save_autoserial_greater(self):
        saved_filename = tempfile.mkstemp()[1]
        self.zone.save(saved_filename, autoserial=True)

        z = Zone('example.com.')
        z.load_from_file(saved_filename)
        self.failUnless(z.root.soa.serial >= self.zone.root.soa.serial)
Пример #5
0
class ZoneModifySaveTest(unittest.TestCase):
    def setUp(self):
        self.zone = Zone('example.com.')
        zone_file = os.path.join(os.path.dirname(__file__), 'files', 'example.com')
        self.zone.load_from_file(zone_file)

        self.zone.root.soa.mname = 'mname.example.com.'
        self.zone.root.soa.rname = 'rname.example.com.'
        self.zone.root.soa.serial += 1
        self.zone.root.soa.refresh = 1
        self.zone.root.soa.retry = 2
        self.zone.root.soa.expire = 3
        self.zone.root.soa.minttl = 4

        self.zone.add_name('zip.example.com.')
        self.zone.names['zip.example.com.'].records('A', create=True).add('10.9.8.7')

        mx = self.zone.names['foo.example.com.'].records('MX')
        mx.delete( (10, 'mail.example.com.') )
        mx.add( (30, 'anothermail.example.com.') )

        self.zone.names['bar.example.com.'].records('A').add('10.20.30.40')

        self.zone.delete_name('foofoo.example.com.')

        self.saved_filename = tempfile.mkstemp()[1]
        # self.saved_filename = '/var/tmp/foo.tmp'
        self.zone.save(self.saved_filename)

    def test_file_exists(self):
        self.failUnless(os.path.exists(self.saved_filename))

    def test_file_size(self):
        size = os.stat(self.saved_filename)[6]
        self.failUnlessEqual(size, 800)

    def test_save_autoserial_greater(self):
        saved_filename = tempfile.mkstemp()[1]
        self.zone.save(saved_filename, autoserial=True)

        z = Zone('example.com.')
        z.load_from_file(saved_filename)
        self.failUnless(z.root.soa.serial >= self.zone.root.soa.serial)
Пример #6
0
class ZoneLoadTest(unittest.TestCase):
    def setUp(self):
        zone_file = os.path.join(os.path.dirname(__file__), 'files', 'example.com')
        self.zone = Zone('example.com.')
        self.zone.load_from_file(zone_file)

        revzone_file = os.path.join(os.path.dirname(__file__), 'files', '10.in-addr.arpa')
        self.revzone = Zone('10.in-addr.arpa.')
        self.revzone.load_from_file(revzone_file)

    def test_type(self):
        self.failUnlessEqual(type(self.zone), Zone)

    def test_domain(self):
        self.failUnlessEqual(self.zone.domain, 'example.com.')

    def test_root_type(self):
        root = self.zone.root
        self.failUnlessEqual(type(root), Name)

    def test_root_name(self):
        root = self.zone.root
        self.failUnlessEqual(root.name, '@')

    def test_soa_type(self):
        soa = self.zone.root.soa
        self.failUnlessEqual(type(soa), SOA)

    def test_soa_mname(self):
        self.failUnlessEqual(self.zone.root.soa.mname, 'ns1.example.com.')

    def test_soa_rname(self):
        self.failUnlessEqual(self.zone.root.soa.rname, 'hostmaster.example.com.')

    def test_soa_serial(self):
        self.failUnlessEqual(self.zone.root.soa.serial, 2007012501)

    def test_soa_refresh(self):
        self.failUnlessEqual(self.zone.root.soa.refresh, 28800)

    def test_soa_retry(self):
        self.failUnlessEqual(self.zone.root.soa.retry, 7200)

    def test_soa_expire(self):
        self.failUnlessEqual(self.zone.root.soa.expire, 864000)

    def test_soa_minttl(self):
        self.failUnlessEqual(self.zone.root.soa.minttl, 86400)

    def test_root_records_A(self):
        records = self.zone.root.records('A').items
        self.failUnlessEqual(records, ['10.0.0.1'])

    def test_root_records_AAAA(self):
        records = self.zone.root.records('AAAA').items
        self.failUnlessEqual(records, ['aaaa:bbbb:cccc::1'])

    def test_root_records_NS(self):
        records = self.zone.root.records('NS').items
        self.failUnlessEqual(records, ['ns1.example.com.', 'ns2.example.com.'])

    def test_root_records_MX(self):
        records = self.zone.root.records('MX').items
        self.failUnlessEqual(records, [(10, 'mail.example.com.'), (20, 'mail2.example.com.')])

    def test_names_type(self):
        names = self.zone.names
        self.failUnlessEqual(type(names), types.DictType)

    def test_names_foo_A(self):
        records = self.zone.names['foo.example.com.'].records('A').items
        self.failUnlessEqual(records, ['10.0.0.1'])

    def test_names_foo_AAAA(self):
        records = self.zone.names['foo.example.com.'].records('AAAA').items
        self.failUnlessEqual(records, ['aaaa:bbbb:cccc::1'])

    def test_names_foo_MX(self):
        records = self.zone.names['foo.example.com.'].records('MX').items
        self.failUnlessEqual(records, [(10, 'mail.example.com.')])

    def test_names_xmpp_SRV(self):
        records = self.zone.names['_xmpp-client._tcp.example.com.'].records('SRV').items
        self.failUnlessEqual(records, [(20, 0, 5222, 'foobar.example.com.')])

    def test_names_bar_A(self):
        records = self.zone.names['bar.example.com.'].records('A').items
        self.failUnlessEqual(records, ['10.0.0.2', '10.0.0.3'])

    def test_names_foofoo_CNAME(self):
        records = self.zone.names['foofoo.example.com.'].records('CNAME').items
        self.failUnlessEqual(records, ['foo.example.com.'])

    def test_names_foo_PTR(self):
        records = self.revzone.names['1.0.0.10.in-addr.arpa.'].records('PTR').items
        self.failUnlessEqual(records, ['foo.example.com.'])

    def test_names_root_A(self):
        records = self.zone.names['example.com.'].records('A').items
        self.failUnlessEqual(records, ['10.0.0.1'])

    def test_names_root_NS(self):
        records = self.zone.names['example.com.'].records('NS').items
        self.failUnlessEqual(records, ['ns1.example.com.', 'ns2.example.com.'])

    def test_names_root_MX(self):
        records = self.zone.names['example.com.'].records('MX').items
        self.failUnlessEqual(records, [(10, 'mail.example.com.'), (20, 'mail2.example.com.')])
Пример #7
0
 def test_unicode_domain(self):
     zone = Zone(u'example.com')
     self.failUnlessEqual(zone.domain, u'example.com.')
Пример #8
0
 def test_missing_dot(self):
     zone = Zone('example.com')
     self.failUnlessEqual(zone.domain, 'example.com.')
Пример #9
0
class ZoneModifyTest(unittest.TestCase):
    def setUp(self):
        self.zone = Zone('example.com.')
        zone_file = os.path.join(os.path.dirname(__file__), 'files', 'example.com')
        self.zone.load_from_file(zone_file)

        revzone_file = os.path.join(os.path.dirname(__file__), 'files', '10.in-addr.arpa')
        self.revzone = Zone('10.in-addr.arpa.')
        self.revzone.load_from_file(revzone_file)

    def test_root_add_NS(self):
        # add another NS record to @
        self.zone.root.records('NS').add('ns3.example.com.')
        records = self.zone.names['example.com.'].records('NS').items
        self.failUnlessEqual(records, ['ns1.example.com.', 'ns2.example.com.', 'ns3.example.com.'])

    def test_root_add_duplicate_NS(self):
        # add a duplicate NS record to @ - has no effect
        self.zone.root.records('NS').add('ns1.example.com.')
        records = self.zone.names['example.com.'].records('NS').items
        self.failUnlessEqual(records, ['ns1.example.com.', 'ns2.example.com.'])

    def test_root_delete_NS(self):
        # delete NS record from @
        self.zone.root.records('NS').delete('ns2.example.com.')
        records = self.zone.names['example.com.'].records('NS').items
        self.failUnlessEqual(records, ['ns1.example.com.'])

    def test_root_delete_noexist_NS(self):
        # delete non-existent NS record from @
        self.failUnlessRaises(RecordsError, self.zone.root.records('NS').delete, 'ns99.example.com.')

    def test_root_add_A(self):
        # add another A record to @
        self.zone.root.records('A').add('10.2.3.4')
        records = self.zone.names['example.com.'].records('A').items
        self.failUnlessEqual(records, ['10.0.0.1', '10.2.3.4'])

    def test_root_add_AAAA(self):
        # add another AAAA record to @
        self.zone.root.records('AAAA').add('aaaa:bbbb:cccc::2')
        records = self.zone.names['example.com.'].records('AAAA').items
        self.failUnlessEqual(records, ['aaaa:bbbb:cccc::1', 'aaaa:bbbb:cccc::2'])

    def test_names_add_root_MX(self):
        # add MX record to @ via names attribute
        self.zone.names['example.com.'].records('MX').add( (30, 'mail3.example.com.') )
        records = self.zone.names['example.com.'].records('MX').items
        self.failUnlessEqual(records, [(10, 'mail.example.com.'), (20, 'mail2.example.com.'), (30, 'mail3.example.com.')])

    def test_names_delete_root_MX(self):
        # delete MX record from @ via names attribute
        self.zone.names['example.com.'].records('MX').delete( (10, 'mail.example.com.') )
        records = self.zone.names['example.com.'].records('MX').items
        self.failUnlessEqual(records, [(20, 'mail2.example.com.')])

    def test_names_add_bar_A(self):
        # add A record to bar.example.com.
        self.zone.names['bar.example.com.'].records('A').add('10.20.30.40')
        records = self.zone.names['bar.example.com.'].records('A').items
        self.failUnlessEqual(records, ['10.0.0.2', '10.0.0.3', '10.20.30.40'])

    def test_names_add_bar_AAAA(self):
        # add AAAA record to bar.example.com.
        self.zone.names['bar.example.com.'].records('AAAA').add('aaaa:bbbb:cccc::123')
        records = self.zone.names['bar.example.com.'].records('AAAA').items
        self.failUnlessEqual(records, ['aaaa:bbbb:cccc::2', 'aaaa:bbbb:cccc::3', 'aaaa:bbbb:cccc::123'])

    def test_names_delete_bar_A(self):
        # delete A record from bar.example.com.
        self.zone.names['bar.example.com.'].records('A').delete('10.0.0.2')
        records = self.zone.names['bar.example.com.'].records('A').items
        self.failUnlessEqual(records, ['10.0.0.3'])

    def test_names_delete_bar_AAAA(self):
        # delete A record from bar.example.com.
        self.zone.names['bar.example.com.'].records('AAAA').delete('aaaa:bbbb:cccc::3')
        records = self.zone.names['bar.example.com.'].records('AAAA').items
        self.failUnlessEqual(records, ['aaaa:bbbb:cccc::2'])

    def test_names_add_poppy_CNAME(self):
        # add CNAME record poppy.example.com.
        self.zone.add_name('poppy.example.com.')
        self.zone.names['poppy.example.com.'].records('CNAME', create=True).add('bar.example.com.')
        records = self.zone.names['poppy.example.com.'].records('CNAME').items
        self.failUnlessEqual(records, ['bar.example.com.'])

    def test_names_add_bar_PTR(self):
        self.revzone.add_name('2.0.0.10.in-addr.arpa.')
        self.revzone.names['2.0.0.10.in-addr.arpa.'].records('PTR', create=True).add('bar.example.com.')
        records = self.revzone.names['2.0.0.10.in-addr.arpa.'].records('PTR').items
        self.failUnlessEqual(records, ['bar.example.com.'])

    def test_names_add_xmppserver_SRV(self):
        # add SRV _xmpp-server._tcp.example.com.
        self.zone.add_name('_xmpp-server._tcp.example.com.')
        self.zone.names['_xmpp-server._tcp.example.com.'].records('SRV', create=True).add((20, 0, 5269 , 'bar.example.com.'))
        records = self.zone.names['_xmpp-server._tcp.example.com.'].records('SRV').items
        self.failUnlessEqual(records, [(20, 0, 5269, 'bar.example.com.')])

    def test_names_delete_xmppclient_SRV(self):
        # delete SRV _xmpp-client._tcp.example.com.
        self.zone.names['_xmpp-client._tcp.example.com.'].records('SRV', create=True).add((10, 0, 5222, 'bar.example.com.'))
        self.zone.names['_xmpp-client._tcp.example.com.'].records('SRV', create=True).delete((10, 0, 5222, 'bar.example.com.'))
        records = self.zone.names['_xmpp-client._tcp.example.com.'].records('SRV').items
        self.failUnlessEqual(records, [(20, 0, 5222, 'foobar.example.com.')])

    def test_names_add_bar_MX(self):
        # add MX record to bar.example.com.
        self.zone.names['bar.example.com.'].records('MX', create=True).add( (100, 'backupmx.example.com.') )
        records = self.zone.names['bar.example.com.'].records('MX').items
        self.failUnlessEqual(records, [(100, 'backupmx.example.com.')])

    def test_names_delete_foo_MX(self):
        # delete MX record from foo.example.com.
        self.zone.names['foo.example.com.'].records('MX').delete( (10, 'mail.example.com.') )
        records = self.zone.names['foo.example.com.'].records('MX').items
        self.failUnlessEqual(records, [])

    def test_names_bar_NS(self):
        # try to fetch NS records from bar.example.com. - should fail (non-root nodes can't contain NS records)
        node = self.zone.names['bar.example.com.'].records('NS')
        self.failUnlessEqual(node, None)

    def test_names_replace_foo_MX(self):
        # replace MX record for foo.example.com.
        mx = self.zone.names['foo.example.com.'].records('MX')
        mx.delete( (10, 'mail.example.com.') )
        mx.add( (30, 'anothermail.example.com.') )
        records = self.zone.names['foo.example.com.'].records('MX').items
        self.failUnlessEqual(records, [(30, 'anothermail.example.com.')])

    def test_add_name_zip_A(self):
        # add new name zip.example.com. with A record
        self.zone.add_name('zip.example.com.')
        self.zone.names['zip.example.com.'].records('A', create=True).add('10.9.8.7')
        records = self.zone.names['zip.example.com.'].records('A').items
        self.failUnlessEqual(records, ['10.9.8.7'])

    def test_delete_name_foo(self):
        # delete name foo.example.com. from zone (and hence all associated nodes for that name)
        self.zone.delete_name('foo.example.com.')
        self.failUnlessEqual(self.zone.names.keys(), ['foofoo.example.com.', '_xmpp-client._tcp.example.com.', 'bar.example.com.', 'example.com.'])

    def test_names_bar_clear_all_records(self):
        # clear all records for bar.example.com.
        self.zone.names['bar.example.com.'].clear_all_records()
        self.failUnlessEqual(self.zone.names.keys(), ['foo.example.com.', 'foofoo.example.com.', '_xmpp-client._tcp.example.com.', 'bar.example.com.', 'example.com.'])
        self.failUnlessEqual(self.zone.names['bar.example.com.'].records('A'), None)

    def test_names_foo_clear_all_records_exclude(self):
        # clear records for foo.example.com. excluding some
        self.zone.names['foo.example.com.'].clear_all_records(exclude='MX')
        self.failUnlessEqual(self.zone.names.keys(), ['foo.example.com.', 'foofoo.example.com.', '_xmpp-client._tcp.example.com.', 'bar.example.com.', 'example.com.'])
        self.failUnlessEqual(self.zone.names['foo.example.com.'].records('A'), None)
        self.failUnlessEqual(self.zone.names['foo.example.com.'].records('MX').items, [(10, 'mail.example.com.')])

    def test_names_foo_clear_all_records_include(self):
        # clear all AAAA records for bar.example.com
        self.zone.names['bar.example.com.'].clear_all_records(include='AAAA')
        self.failUnlessEqual(self.zone.names.keys(), ['foo.example.com.', 'foofoo.example.com.', '_xmpp-client._tcp.example.com.', 'bar.example.com.', 'example.com.'])
        self.failUnlessEqual(self.zone.names['bar.example.com.'].records('AAAA'), None)
        self.failUnlessEqual(self.zone.names['bar.example.com.'].records('A').items,['10.0.0.2', '10.0.0.3'])

    def test_names_add_bar_TXT_simple(self):
        # add simple TXT record to bar.example.com.
        self.zone.names['bar.example.com.'].records('TXT', create=True).add('"v=spf1 a mx ?all"')
        records = self.zone.names['bar.example.com.'].records('TXT').items
        self.failUnlessEqual(records, ['"v=spf1 a mx ?all"'])

    def test_names_add_bar_TXT_with_periods(self):
        # add TXT record to bar.example.com. containing periods
        self.zone.names['bar.example.com.'].records('TXT', create=True).add('"v=spf1 a mx include:mailseat.com include:cluster3.eu.messagelabs.com ?all"')
        records = self.zone.names['bar.example.com.'].records('TXT').items
        self.failUnlessEqual(records, ['"v=spf1 a mx include:mailseat.com include:cluster3.eu.messagelabs.com ?all"'])

    def test_names_add_bar_TXT_no_quotes(self):
        # add TXT record to bar.example.com. excluding surrounding quotes
        self.zone.names['bar.example.com.'].records('TXT', create=True).add('v=spf1 a mx include:mailseat.com include:cluster3.eu.messagelabs.com ?all')
        records = self.zone.names['bar.example.com.'].records('TXT').items
        self.failUnlessEqual(records, ['"v=spf1 a mx include:mailseat.com include:cluster3.eu.messagelabs.com ?all"'])
Пример #10
0
 def setUp(self):
     self.zone = Zone('example.com.')
     zone_file = os.path.join(os.path.dirname(__file__), 'files',
                              'example.com')
     self.zone.load_from_file(zone_file)
Пример #11
0
 def test_unicode_domain(self):
     zone = Zone('example.com')
     self.assertEqual(zone.domain, 'example.com.')
Пример #12
0
 def test_missing_dot(self):
     zone = Zone('example.com')
     self.assertEqual(zone.domain, 'example.com.')