Set EDNS default payload to 1232.

2 files changed, 6 insertions, 2 deletions

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)