Add canonical_name() method to resolver.canonical_name

1 files changed, 14 insertions, 0 deletions

diff --git a/tests/test_resolver.py b/tests/test_resolver.py
index cadf224..83e8d66 100644
--- a/tests/test_resolver.py
+++ b/tests/test_resolver.py
@@ -475,6 +475,20 @@ class LiveResolverTests(unittest.TestCase):
         answer2 = res.resolve('dns.google.', 'A')
         self.assertIs(answer2, answer1)
 
+    def testCanonicalNameNoCNAME(self):
+        cname = dns.name.from_text('www.google.com')
+        self.assertTrue(dns.resolver.canonical_name('www.google.com') == cname)
+
+    def testCanonicalNameCNAME(self):
+        name = dns.name.from_text('www.dnspython.org')
+        cname = dns.name.from_text('dmfrjf4ips8xa.cloudfront.net')
+        self.assertTrue(dns.resolver.canonical_name(name) == cname)
+
+    def testCanonicalNameDangling(self):
+        name = dns.name.from_text('dangling-cname.dnspython.org')
+        cname = dns.name.from_text('dangling-target.dnspython.org')
+        self.assertTrue(dns.resolver.canonical_name(name) == cname)
+
 class PollingMonkeyPatchMixin(object):
     def setUp(self):
         self.__native_selector_class = dns.query._selector_class