Merge pull request #105 from encukou/py3-tests

Py3: Add missing tests

4 files changed, 244 insertions, 0 deletions

diff --git a/tests/test_grange.py b/tests/test_grange.py
new file mode 100644
index 0000000..aee9653
--- /dev/null
+++ b/tests/test_grange.py
@@ -0,0 +1,86 @@
+# Copyright (C) 2003-2007, 2009-2011 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 unittest
+
+import dns
+import dns.exception
+import dns.grange
+
+
+class GRangeTestCase(unittest.TestCase):
+
+    def testFromText1(self):
+        start, stop, step = dns.grange.from_text('1-1')
+        self.assertEqual(start, 1)
+        self.assertEqual(stop, 1)
+        self.assertEqual(step, 1)
+
+    def testFromText2(self):
+        start, stop, step = dns.grange.from_text('1-4')
+        self.assertEqual(start, 1)
+        self.assertEqual(stop, 4)
+        self.assertEqual(step, 1)
+
+    def testFromText3(self):
+        start, stop, step = dns.grange.from_text('4-255')
+        self.assertEqual(start, 4)
+        self.assertEqual(stop, 255)
+        self.assertEqual(step, 1)
+
+    def testFromText4(self):
+        start, stop, step = dns.grange.from_text('1-1/1')
+        self.assertEqual(start, 1)
+        self.assertEqual(stop, 1)
+        self.assertEqual(step, 1)
+
+    def testFromText5(self):
+        start, stop, step = dns.grange.from_text('1-4/2')
+        self.assertEqual(start, 1)
+        self.assertEqual(stop, 4)
+        self.assertEqual(step, 2)
+
+    def testFromText6(self):
+        start, stop, step = dns.grange.from_text('4-255/77')
+        self.assertEqual(start, 4)
+        self.assertEqual(stop, 255)
+        self.assertEqual(step, 77)
+
+    def testFailFromText1(self):
+        def bad():
+            start = 2
+            stop = 1
+            step = 1
+            dns.grange.from_text('%d-%d/%d' % (start, stop, step))
+        self.assertRaises(AssertionError, bad)
+
+    def testFailFromText2(self):
+        def bad():
+            start = '-1'
+            stop = 3
+            step = 1
+            dns.grange.from_text('%s-%d/%d' % (start, stop, step))
+        self.assertRaises(dns.exception.SyntaxError, bad)
+
+    def testFailFromText2(self):
+        def bad():
+            start = 1
+            stop = 4
+            step = '-2'
+            dns.grange.from_text('%d-%d/%s' % (start, stop, step))
+        self.assertRaises(dns.exception.SyntaxError, bad)
+
+if __name__ == '__main__':
+    unittest.main()
diff --git a/tests/test_rdata.py b/tests/test_rdata.py
new file mode 100644
index 0000000..69e64e1
--- /dev/null
+++ b/tests/test_rdata.py
@@ -0,0 +1,33 @@
+# Copyright (C) 2006, 2007, 2009-2011 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 unittest
+
+import dns.rdata
+import dns.rdataclass
+import dns.rdatatype
+
+class RdataTestCase(unittest.TestCase):
+
+    def test_str(self):
+        rdata = dns.rdata.from_text(dns.rdataclass.IN, dns.rdatatype.A, "1.2.3.4")
+        self.failUnless(rdata.address == "1.2.3.4")
+
+    def test_unicode(self):
+        rdata = dns.rdata.from_text(dns.rdataclass.IN, dns.rdatatype.A, u"1.2.3.4")
+        self.failUnless(rdata.address == "1.2.3.4")
+
+if __name__ == '__main__':
+    unittest.main()
diff --git a/tests/test_rdtypeanyloc.py b/tests/test_rdtypeanyloc.py
new file mode 100644
index 0000000..7595b9f
--- /dev/null
+++ b/tests/test_rdtypeanyloc.py
@@ -0,0 +1,68 @@
+# Copyright (C) 2014 Red Hat, Inc.
+# Author: Petr Spacek <pspacek@redhat.com>
+#
+# 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 RED HAT 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 unittest
+
+import dns.rrset
+import dns.rdtypes.ANY.LOC
+
+class RdtypeAnyLocTestCase(unittest.TestCase):
+
+    def testEqual1(self):
+        '''Test default values for size, horizontal and vertical precision.'''
+        r1 = dns.rrset.from_text('foo', 300, 'IN', 'LOC',
+                                 '49 11 42.400 N 16 36 29.600 E 227.64m')
+        r2 = dns.rrset.from_text('FOO', 600, 'in', 'loc',
+                                 '49 11 42.400 N 16 36 29.600 E 227.64m '
+                                 '1.00m 10000.00m 10.00m')
+        self.assertTrue(r1 == r2, '"%s" != "%s"' % (r1, r2))
+
+    def testEqual2(self):
+        '''Test default values for size, horizontal and vertical precision.'''
+        r1 = dns.rdtypes.ANY.LOC.LOC(1, 29, (49, 11, 42, 400),
+                                     (16, 36, 29, 600), 22764.0) # centimeters
+        r2 = dns.rdtypes.ANY.LOC.LOC(1, 29, (49, 11, 42, 400),
+                                     (16, 36, 29, 600), 22764.0, # centimeters
+                                     100.0, 1000000.00, 1000.0)  # centimeters
+        self.assertTrue(r1 == r2, '"%s" != "%s"' % (r1, r2))
+
+    def testEqual3(self):
+        '''Test size, horizontal and vertical precision parsers: 100 cm == 1 m.
+
+        Parsers in from_text() and __init__() have to produce equal results.'''
+        r1 = dns.rdtypes.ANY.LOC.LOC(1, 29, (49, 11, 42, 400),
+                                     (16, 36, 29, 600), 22764.0,
+                                     200.0, 1000.00, 200.0)      # centimeters
+        r2 = dns.rrset.from_text('FOO', 600, 'in', 'loc',
+                                 '49 11 42.400 N 16 36 29.600 E 227.64m '
+                                 '2.00m 10.00m 2.00m')[0]
+        self.assertTrue(r1 == r2, '"%s" != "%s"' % (r1, r2))
+
+    def testEqual4(self):
+        '''Test size, horizontal and vertical precision parsers without unit.
+
+        Parsers in from_text() and __init__() have produce equal result
+        for values with and without trailing "m".'''
+        r1 = dns.rdtypes.ANY.LOC.LOC(1, 29, (49, 11, 42, 400),
+                                     (16, 36, 29, 600), 22764.0,
+                                     200.0, 1000.00, 200.0)      # centimeters
+        r2 = dns.rrset.from_text('FOO', 600, 'in', 'loc',
+                                 '49 11 42.400 N 16 36 29.600 E 227.64 '
+                                 '2 10 2')[0] # meters without explicit unit
+        self.assertTrue(r1 == r2, '"%s" != "%s"' % (r1, r2))
+
+if __name__ == '__main__':
+    unittest.main()
diff --git a/tests/test_resolver.py b/tests/test_resolver.py
index 2f09463..f3c0d35 100644
--- a/tests/test_resolver.py
+++ b/tests/test_resolver.py
@@ -47,6 +47,10 @@ example. 1 IN A 10.0.0.1
 ;ADDITIONAL
 """
 
+class FakeAnswer(object):
+    def __init__(self, expiration):
+        self.expiration = expiration
+
 class BaseResolverTests(object):
 
     if sys.platform != 'win32':
@@ -102,6 +106,59 @@ class BaseResolverTests(object):
             zname = dns.resolver.zone_for_name(name)
         self.assertRaises(dns.resolver.NotAbsolute, bad)
 
+    def testLRUReplace(self):
+        cache = dns.resolver.LRUCache(4)
+        for i in range(0, 5):
+            name = dns.name.from_text('example%d.' % i)
+            answer = FakeAnswer(time.time() + 1)
+            cache.put((name, dns.rdatatype.A, dns.rdataclass.IN), answer)
+        for i in range(0, 5):
+            name = dns.name.from_text('example%d.' % i)
+            if i == 0:
+                self.assertTrue(cache.get((name, dns.rdatatype.A,
+                                           dns.rdataclass.IN))
+                                is None)
+            else:
+                self.assertTrue(not cache.get((name, dns.rdatatype.A,
+                                               dns.rdataclass.IN))
+                                is None)
+
+    def testLRUDoesLRU(self):
+        cache = dns.resolver.LRUCache(4)
+        for i in range(0, 4):
+            name = dns.name.from_text('example%d.' % i)
+            answer = FakeAnswer(time.time() + 1)
+            cache.put((name, dns.rdatatype.A, dns.rdataclass.IN), answer)
+        name = dns.name.from_text('example0.')
+        cache.get((name, dns.rdatatype.A, dns.rdataclass.IN))
+        # The LRU is now example1.
+        name = dns.name.from_text('example4.')
+        answer = FakeAnswer(time.time() + 1)
+        cache.put((name, dns.rdatatype.A, dns.rdataclass.IN), answer)
+        for i in range(0, 5):
+            name = dns.name.from_text('example%d.' % i)
+            if i == 1:
+                self.assertTrue(cache.get((name, dns.rdatatype.A,
+                                           dns.rdataclass.IN))
+                                is None)
+            else:
+                self.assertTrue(not cache.get((name, dns.rdatatype.A,
+                                               dns.rdataclass.IN))
+                                is None)
+
+    def testLRUExpiration(self):
+        cache = dns.resolver.LRUCache(4)
+        for i in range(0, 4):
+            name = dns.name.from_text('example%d.' % i)
+            answer = FakeAnswer(time.time() + 1)
+            cache.put((name, dns.rdatatype.A, dns.rdataclass.IN), answer)
+        time.sleep(2)
+        for i in range(0, 4):
+            name = dns.name.from_text('example%d.' % i)
+            self.assertTrue(cache.get((name, dns.rdatatype.A,
+                                       dns.rdataclass.IN))
+                            is None)
+
 class PollingMonkeyPatchMixin(object):
     def setUp(self):
         self.__native_polling_backend = dns.query._polling_backend