From 58a404ab191241fd9f1237b674e44aca554da202 Mon Sep 17 00:00:00 2001
From: Bob Halley <halley@dnspython.org>
Date: Mon, 20 Jul 2020 19:52:20 -0700
Subject: Set EDNS default payload to 1232.

---
 tests/test_message.py  | 6 +++++-
 tests/test_resolver.py | 2 +-
 2 files changed, 6 insertions(+), 2 deletions(-)

(limited to 'tests')

diff --git a/tests/test_message.py b/tests/test_message.py
index e64578b..bfc137f 100644
--- a/tests/test_message.py
+++ b/tests/test_message.py
@@ -16,7 +16,6 @@
 # ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
 # OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 
-import os
 import unittest
 import binascii
 
@@ -337,6 +336,11 @@ class MessageTestCase(unittest.TestCase):
         self.assertEqual(m.edns, 0)
         self.assertTrue(m.ednsflags & dns.flags.DO)
 
+    def test_EDNS_default_payload_is_1232(self):
+        m = dns.message.make_query('foo', 'A')
+        m.use_edns()
+        self.assertEqual(m.payload, dns.message.DEFAULT_EDNS_PAYLOAD)
+
     def test_from_file(self):
         m = dns.message.from_file(here('query'))
         expected = dns.message.from_text(query_text)
diff --git a/tests/test_resolver.py b/tests/test_resolver.py
index 85abeb2..171f319 100644
--- a/tests/test_resolver.py
+++ b/tests/test_resolver.py
@@ -166,7 +166,7 @@ class BaseResolverTests(unittest.TestCase):
         self.assertEqual(r.timeout, 1)
         self.assertEqual(r.ndots, 2)
         self.assertEqual(r.edns, 0)
-        self.assertEqual(r.payload, 1280)
+        self.assertEqual(r.payload, dns.message.DEFAULT_EDNS_PAYLOAD)
 
     def testReadOptionsBadTimeouts(self):
         f = StringIO(bad_timeout_1)
-- 
cgit v1.2.1