pvl/hosts/tests.py
changeset 462 6d699c76d75d
parent 461 e3bddc5eeff5
child 463 2cbdb2435487
--- a/pvl/hosts/tests.py	Wed Feb 25 14:23:04 2015 +0200
+++ b/pvl/hosts/tests.py	Wed Feb 25 14:29:40 2015 +0200
@@ -1,7 +1,7 @@
 import ipaddr
 import unittest
 
-from pvl.hosts import config
+from pvl.hosts import config, zone
 from StringIO import StringIO
 
 class Options(object):
@@ -84,5 +84,26 @@
                 'ethernet.eth0': 'bar',
             })
 
+class TestForwardZone(unittest.TestCase):
+    def setUp(self):
+        self.options = Options()
+        self.options.add_origin = False
+
+    def testResolve(self):
+        self.assertEquals(zone.resolve(None, None, 'host'), 'host.')
+        self.assertEquals(zone.resolve(None, 'domain', 'host'), 'host.domain.')
+
+        with self.assertRaises(zone.HostZoneError):
+            zone.resolve('origin', 'domain', 'host')
+
+        self.assertEquals(zone.resolve('domain', 'domain', 'host'), 'host')
+        self.assertEquals(zone.resolve('origin', 'domain.origin', 'host'), 'host.domain')
+        
+        with self.assertRaises(zone.HostZoneError):
+            zone.resolve('origin', 'domainorigin', 'host')
+
+        with self.assertRaises(zone.HostZoneError):
+            zone.resolve('origin', None, 'host.domain')
+
 if __name__ == '__main__':
     unittest.main()