From: Bob Halley <halley@dnspython.org>
Date: Fri, 26 Jun 2020 13:38:24 +0000 (-0700)
Subject: test class factory and update from_wire()
X-Git-Tag: v2.0.0rc2~61^2~8
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=39d50fd7b63425a358bf10f886399a72b8347d02;p=thirdparty%2Fdnspython.git

test class factory and update from_wire()
---

diff --git a/tests/test_message.py b/tests/test_message.py
index a45ccf68..a1234095 100644
--- a/tests/test_message.py
+++ b/tests/test_message.py
@@ -105,6 +105,10 @@ Königsgäßchen. 3600 IN NS Königsgäßchen.
 
 class MessageTestCase(unittest.TestCase):
 
+    def test_class(self):
+        m = dns.message.from_text(query_text)
+        self.assertTrue(isinstance(m, dns.message.QueryMessage))
+
     def test_comparison_eq1(self):
         q1 = dns.message.from_text(query_text)
         q2 = dns.message.from_text(query_text)
diff --git a/tests/test_update.py b/tests/test_update.py
index 41b3b8d1..49e1d3e2 100644
--- a/tests/test_update.py
+++ b/tests/test_update.py
@@ -110,10 +110,17 @@ class UpdateTestCase(unittest.TestCase):
 
     def test_from_text1(self): # type: () -> None
         update = dns.message.from_text(update_text)
+        self.assertTrue(isinstance(update, dns.update.UpdateMessage))
         w = update.to_wire(origin=dns.name.from_text('example'),
                            want_shuffle=False)
         self.assertEqual(w, goodwire)
 
+    def test_from_wire(self):
+        origin = dns.name.from_text('example')
+        u1 = dns.message.from_wire(goodwire, origin=origin)
+        u2 = dns.message.from_text(update_text, origin=origin)
+        self.assertEqual(u1, u2)
+
     def test_TSIG(self):
         keyring = dns.tsigkeyring.from_text({
             'keyname.' : 'NjHwPsMKjdN++dOfE5iAiQ=='