PYTHON-3198 Fix NameError: name sys is not defined (#920)

Author: ShaneHarvey

pymongo/uri_parser.py

@@ -15,6 +15,7 @@
 
 """Tools to parse and validate a MongoDB URI."""
 import re
+import sys
 import warnings
 from typing import Any, Dict, List, Mapping, MutableMapping, Optional, Tuple, Union
 from urllib.parse import unquote_plus
@@ -615,7 +616,6 @@ def _parse_kms_tls_options(kms_tls_options):
 
 if __name__ == "__main__":
     import pprint
-    import sys
 
     try:
         pprint.pprint(parse_uri(sys.argv[1]))

test/test_client.py

@@ -1689,6 +1689,11 @@ def test_srv_max_hosts_kwarg(self):
         )
         self.assertEqual(len(client.topology_description.server_descriptions()), 2)
 
+    @unittest.skipIf(_HAVE_DNSPYTHON, "dnspython must not be installed")
+    def test_srv_no_dnspython_error(self):
+        with self.assertRaisesRegex(ConfigurationError, 'The "dnspython" module must be'):
+            MongoClient("mongodb+srv://test1.test.build.10gen.cc/")
+
     @unittest.skipIf(
         client_context.load_balancer or client_context.serverless,
         "loadBalanced clients do not run SDAM",