From: Brian Wellington <bwelling@xbill.org>
Date: Wed, 17 Jun 2020 22:13:12 +0000 (-0700)
Subject: Test message encoding with EDNS options.
X-Git-Tag: v2.0.0rc1~52
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=2e50d55ede849f0150264864595336eef0cbd52f;p=thirdparty%2Fdnspython.git

Test message encoding with EDNS options.
---

diff --git a/tests/test_message.py b/tests/test_message.py
index ab40a05e..4e80e01e 100644
--- a/tests/test_message.py
+++ b/tests/test_message.py
@@ -144,6 +144,15 @@ class MessageTestCase(unittest.TestCase):
         m = dns.message.from_wire(goodwire3)
         self.assertEqual(str(m), query_text_2)
 
+    def test_EDNS_options_wire(self):
+        m = dns.message.make_query('foo', 'A')
+        opt = dns.edns.GenericOption(3, b'data')
+        m.use_edns(options=[opt])
+        m2 = dns.message.from_wire(m.to_wire())
+        self.assertEqual(m2.edns, 0)
+        self.assertEqual(len(m2.options), 1)
+        self.assertEqual(m2.options[0], opt)
+
     def test_TooBig(self):
         def bad():
             q = dns.message.from_text(query_text)