Add resolver edns options

1 files changed, 9 insertions, 1 deletions

diff --git a/tests/test_resolver.py b/tests/test_resolver.py
index 16d9065..4a6bbb9 100644
--- a/tests/test_resolver.py
+++ b/tests/test_resolver.py
@@ -21,8 +21,8 @@ import sys
 import socket
 import time
 import unittest
-
 import pytest
+from unittest.mock import patch
 
 import dns.e164
 import dns.message
@@ -647,6 +647,14 @@ class LiveResolverTests(unittest.TestCase):
         dnsgoogle = dns.name.from_text('dns.google.')
         self.assertEqual(answer[0].target, dnsgoogle)
 
+    @patch.object(dns.message.Message, 'use_edns')
+    def testResolveEdnsOptions(self, message_use_edns_mock):
+        resolver = dns.resolver.Resolver()
+        options = [dns.edns.ECSOption('1.1.1.1')]
+        resolver.use_edns(True, options=options)
+        resolver.resolve('dns.google.', 'A')
+        assert message_use_edns_mock.call_args.kwargs == {'options': options}
+
     def testResolveNodataException(self):
         def bad():
             dns.resolver.resolve('dnspython.org.', 'SRV')