summaryrefslogtreecommitdiff
path: root/lib/dnspython/tests/resolver.py
diff options
context:
space:
mode:
authorJelmer Vernooij <jelmer@samba.org>2010-03-29 16:08:11 +0200
committerJelmer Vernooij <jelmer@samba.org>2010-03-29 18:05:29 +0200
commit4bbc3ff037026c72f3249f59c1b5af69a6ad6d69 (patch)
tree125857a1edbf3246579e51b10f6e62ec875062bc /lib/dnspython/tests/resolver.py
parent89aa3b766b926d19dac4805a7e72433e497ce872 (diff)
downloadsamba-4bbc3ff037026c72f3249f59c1b5af69a6ad6d69.tar.gz
s4-python: Move dnspython to lib/, like the other Python modules
This also avoids it from being installed if it's already present on the system.
Diffstat (limited to 'lib/dnspython/tests/resolver.py')
-rw-r--r--lib/dnspython/tests/resolver.py105
1 files changed, 105 insertions, 0 deletions
diff --git a/lib/dnspython/tests/resolver.py b/lib/dnspython/tests/resolver.py
new file mode 100644
index 00000000000..4cacbdc79d9
--- /dev/null
+++ b/lib/dnspython/tests/resolver.py
@@ -0,0 +1,105 @@
+# Copyright (C) 2003-2007, 2009, 2010 Nominum, Inc.
+#
+# Permission to use, copy, modify, and distribute this software and its
+# documentation for any purpose with or without fee is hereby granted,
+# provided that the above copyright notice and this permission notice
+# appear in all copies.
+#
+# THE SOFTWARE IS PROVIDED "AS IS" AND NOMINUM DISCLAIMS ALL WARRANTIES
+# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL NOMINUM BE LIABLE FOR
+# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
+# OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+
+import cStringIO
+import sys
+import time
+import unittest
+
+import dns.name
+import dns.message
+import dns.name
+import dns.rdataclass
+import dns.rdatatype
+import dns.resolver
+
+resolv_conf = """
+ /t/t
+# comment 1
+; comment 2
+domain foo
+nameserver 10.0.0.1
+nameserver 10.0.0.2
+"""
+
+message_text = """id 1234
+opcode QUERY
+rcode NOERROR
+flags QR AA RD
+;QUESTION
+example. IN A
+;ANSWER
+example. 1 IN A 10.0.0.1
+;AUTHORITY
+;ADDITIONAL
+"""
+
+class ResolverTestCase(unittest.TestCase):
+
+ if sys.platform != 'win32':
+ def testRead(self):
+ f = cStringIO.StringIO(resolv_conf)
+ r = dns.resolver.Resolver(f)
+ self.failUnless(r.nameservers == ['10.0.0.1', '10.0.0.2'] and
+ r.domain == dns.name.from_text('foo'))
+
+ def testCacheExpiration(self):
+ message = dns.message.from_text(message_text)
+ name = dns.name.from_text('example.')
+ answer = dns.resolver.Answer(name, dns.rdatatype.A, dns.rdataclass.IN,
+ message)
+ cache = dns.resolver.Cache()
+ cache.put((name, dns.rdatatype.A, dns.rdataclass.IN), answer)
+ time.sleep(2)
+ self.failUnless(cache.get((name, dns.rdatatype.A, dns.rdataclass.IN))
+ is None)
+
+ def testCacheCleaning(self):
+ message = dns.message.from_text(message_text)
+ name = dns.name.from_text('example.')
+ answer = dns.resolver.Answer(name, dns.rdatatype.A, dns.rdataclass.IN,
+ message)
+ cache = dns.resolver.Cache(cleaning_interval=1.0)
+ cache.put((name, dns.rdatatype.A, dns.rdataclass.IN), answer)
+ time.sleep(2)
+ self.failUnless(cache.get((name, dns.rdatatype.A, dns.rdataclass.IN))
+ is None)
+
+ def testZoneForName1(self):
+ name = dns.name.from_text('www.dnspython.org.')
+ ezname = dns.name.from_text('dnspython.org.')
+ zname = dns.resolver.zone_for_name(name)
+ self.failUnless(zname == ezname)
+
+ def testZoneForName2(self):
+ name = dns.name.from_text('a.b.www.dnspython.org.')
+ ezname = dns.name.from_text('dnspython.org.')
+ zname = dns.resolver.zone_for_name(name)
+ self.failUnless(zname == ezname)
+
+ def testZoneForName3(self):
+ name = dns.name.from_text('dnspython.org.')
+ ezname = dns.name.from_text('dnspython.org.')
+ zname = dns.resolver.zone_for_name(name)
+ self.failUnless(zname == ezname)
+
+ def testZoneForName4(self):
+ def bad():
+ name = dns.name.from_text('dnspython.org', None)
+ zname = dns.resolver.zone_for_name(name)
+ self.failUnlessRaises(dns.resolver.NotAbsolute, bad)
+
+if __name__ == '__main__':
+ unittest.main()