Move to test_dns

Author: aclark4life

test/test_uri_parser.py

@@ -24,7 +24,7 @@
 sys.path[0:0] = [""]
 
 from test import unittest
-from unittest.mock import MagicMock, patch
+from unittest.mock import patch
 
 from bson.binary import JAVA_LEGACY
 from pymongo import ReadPreference
@@ -37,26 +37,6 @@
 )
 
 
-def run_initial_dns_seedlist_discovery_prose_tests(self, test_cases):
-    for case in test_cases:
-        with patch("dns.resolver.resolve") as mock_resolver:
-
-            def mock_resolve(query, record_type, *args, **kwargs):
-                mock_srv = MagicMock()
-                mock_srv.target.to_text.return_value = case["mock_target"]
-                return [mock_srv]
-
-            mock_resolver.side_effect = mock_resolve
-            domain = case["query"].split("._tcp.")[1]
-            connection_string = f"mongodb+srv://{domain}"
-            try:
-                parse_uri(connection_string)
-            except ConfigurationError as e:
-                self.assertIn(case["expected_error"], str(e))
-            else:
-                self.fail(f"ConfigurationError was not raised for query: {case['query']}")
-
-
 class TestURI(unittest.TestCase):
     def test_validate_userinfo(self):
         self.assertRaises(InvalidURI, parse_userinfo, "foo@")
@@ -574,71 +554,6 @@ def test_port_with_whitespace(self):
         with self.assertRaisesRegex(ValueError, r"Port contains whitespace character: '\\n'"):
             parse_uri("mongodb://localhost:27\n017")
 
-    # Initial DNS Seedlist Discovery prose tests
-    # https://github.com/mongodb/specifications/blob/0a7a8b5/source/initial-dns-seedlist-discovery/tests/README.md#prose-tests
-
-    def test_1_allow_srv_hosts_with_fewer_than_three_dot_separated_parts(self):
-        with patch("dns.resolver.resolve"):
-            parse_uri("mongodb+srv://localhost/")
-            parse_uri("mongodb+srv://mongo.local/")
-
-    def test_2_throw_when_return_address_does_not_end_with_srv_domain(self):
-        test_cases = [
-            {
-                "query": "_mongodb._tcp.localhost",
-                "mock_target": "localhost.mongodb",
-                "expected_error": "Invalid SRV host",
-            },
-            {
-                "query": "_mongodb._tcp.blogs.mongodb.com",
-                "mock_target": "blogs.evil.com",
-                "expected_error": "Invalid SRV host",
-            },
-            {
-                "query": "_mongodb._tcp.blogs.mongo.local",
-                "mock_target": "test_1.evil.com",
-                "expected_error": "Invalid SRV host",
-            },
-        ]
-        run_initial_dns_seedlist_discovery_prose_tests(self, test_cases)
-
-    def test_3_throw_when_return_address_is_identical_to_srv_hostname(self):
-        test_cases = [
-            {
-                "query": "_mongodb._tcp.localhost",
-                "mock_target": "localhost",
-                "expected_error": "Invalid SRV host",
-            },
-            {
-                "query": "_mongodb._tcp.mongo.local",
-                "mock_target": "mongo.local",
-                "expected_error": "Invalid SRV host",
-            },
-        ]
-        run_initial_dns_seedlist_discovery_prose_tests(self, test_cases)
-
-    def test_4_throw_when_return_address_does_not_contain_dot_separating_shared_part_of_domain(
-        self
-    ):
-        test_cases = [
-            {
-                "query": "_mongodb._tcp.localhost",
-                "mock_target": "test_1.cluster_1localhost",
-                "expected_error": "Invalid SRV host",
-            },
-            {
-                "query": "_mongodb._tcp.mongo.local",
-                "mock_target": "test_1.my_hostmongo.local",
-                "expected_error": "Invalid SRV host",
-            },
-            {
-                "query": "_mongodb._tcp.blogs.mongodb.com",
-                "mock_target": "cluster.testmongodb.com",
-                "expected_error": "Invalid SRV host",
-            },
-        ]
-        run_initial_dns_seedlist_discovery_prose_tests(self, test_cases)
-
 
 if __name__ == "__main__":
     unittest.main()