test class factory and update from_wire()

diff --git a/tests/test_message.py b/tests/test_message.py
index a45ccf68fc32d7f86e82eb7b240819ff5270516e..a123409512a378b0a75025a94d86a667133d2c74 100644 (file)
--- 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 41b3b8d1cac82798bc9285fb11ad37e256ad9a26..49e1d3e2015c56fc0677e923ce5a6f172b1bca42 100644 (file)
--- 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=='