pvl/hosts/tests.py
changeset 462 6d699c76d75d
parent 461 e3bddc5eeff5
child 463 2cbdb2435487
equal deleted inserted replaced
461:e3bddc5eeff5 462:6d699c76d75d
     1 import ipaddr
     1 import ipaddr
     2 import unittest
     2 import unittest
     3 
     3 
     4 from pvl.hosts import config
     4 from pvl.hosts import config, zone
     5 from StringIO import StringIO
     5 from StringIO import StringIO
     6 
     6 
     7 class Options(object):
     7 class Options(object):
     8     hosts_charset   = 'utf-8'
     8     hosts_charset   = 'utf-8'
     9     hosts_domain    = None
     9     hosts_domain    = None
    82             config.apply_host(self.options, 'foo', 'test', {
    82             config.apply_host(self.options, 'foo', 'test', {
    83                 'ethernet': 'foo',
    83                 'ethernet': 'foo',
    84                 'ethernet.eth0': 'bar',
    84                 'ethernet.eth0': 'bar',
    85             })
    85             })
    86 
    86 
       
    87 class TestForwardZone(unittest.TestCase):
       
    88     def setUp(self):
       
    89         self.options = Options()
       
    90         self.options.add_origin = False
       
    91 
       
    92     def testResolve(self):
       
    93         self.assertEquals(zone.resolve(None, None, 'host'), 'host.')
       
    94         self.assertEquals(zone.resolve(None, 'domain', 'host'), 'host.domain.')
       
    95 
       
    96         with self.assertRaises(zone.HostZoneError):
       
    97             zone.resolve('origin', 'domain', 'host')
       
    98 
       
    99         self.assertEquals(zone.resolve('domain', 'domain', 'host'), 'host')
       
   100         self.assertEquals(zone.resolve('origin', 'domain.origin', 'host'), 'host.domain')
       
   101         
       
   102         with self.assertRaises(zone.HostZoneError):
       
   103             zone.resolve('origin', 'domainorigin', 'host')
       
   104 
       
   105         with self.assertRaises(zone.HostZoneError):
       
   106             zone.resolve('origin', None, 'host.domain')
       
   107 
    87 if __name__ == '__main__':
   108 if __name__ == '__main__':
    88     unittest.main()
   109     unittest.main()