From 2e50d55ede849f0150264864595336eef0cbd52f Mon Sep 17 00:00:00 2001
From: Brian Wellington <bwelling@xbill.org>
Date: Wed, 17 Jun 2020 15:13:12 -0700
Subject: [PATCH] Test message encoding with EDNS options.

---
 tests/test_message.py | 9 +++++++++
 1 file changed, 9 insertions(+)

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)
-- 
2.47.3