PYTHON-5563 Fix unified test discovery (#2644)

Author: blink1073

test/asynchronous/test_auth_oidc.py

@@ -30,7 +30,7 @@
 
 sys.path[0:0] = [""]
 
-from test.asynchronous.unified_format import generate_test_classes
+from test.asynchronous.unified_format import generate_test_classes, get_test_path
 from test.utils_shared import EventListener, OvertCommandListener
 
 from bson import SON
@@ -54,14 +54,13 @@
 _IS_SYNC = False
 
 ROOT = Path(__file__).parent.parent.resolve()
-TEST_PATH = ROOT / "auth" / "unified"
 ENVIRON = os.environ.get("OIDC_ENV", "test")
 DOMAIN = os.environ.get("OIDC_DOMAIN", "")
 TOKEN_DIR = os.environ.get("OIDC_TOKEN_DIR", "")
 TOKEN_FILE = os.environ.get("OIDC_TOKEN_FILE", "")
 
 # Generate unified tests.
-globals().update(generate_test_classes(str(TEST_PATH), module=__name__))
+globals().update(generate_test_classes(get_test_path("auth", "unified"), module=__name__))
 
 pytestmark = pytest.mark.auth_oidc
 

test/asynchronous/test_auth_spec.py

@@ -27,16 +27,15 @@
 sys.path[0:0] = [""]
 
 from test import unittest
-from test.asynchronous.unified_format import generate_test_classes
+from test.asynchronous.unified_format import generate_test_classes, get_test_path
 
 from pymongo import AsyncMongoClient
 from pymongo.auth_oidc_shared import OIDCCallback
 
 pytestmark = pytest.mark.auth
 
 _IS_SYNC = False
-
-_TEST_PATH = os.path.join(os.path.dirname(os.path.realpath(__file__)), "auth")
+_TEST_PATH = get_test_path("auth")
 
 
 class TestAuthSpec(AsyncPyMongoTestCase):

test/asynchronous/test_change_stream.py

@@ -35,7 +35,7 @@
     async_client_context,
     unittest,
 )
-from test.asynchronous.unified_format import generate_test_classes
+from test.asynchronous.unified_format import generate_test_classes, get_test_path
 from test.utils_shared import (
     AllowListEventListener,
     EventListener,
@@ -1143,12 +1143,9 @@ def asyncTearDown(self):
         self.listener.reset()
 
 
-_TEST_PATH = os.path.join(os.path.dirname(os.path.realpath(__file__)), "change_streams")
-
-
 globals().update(
     generate_test_classes(
-        os.path.join(_TEST_PATH, "unified"),
+        get_test_path("change_streams", "unified"),
         module=__name__,
     )
 )

test/asynchronous/test_client_metadata.py

@@ -19,7 +19,7 @@
 import time
 import unittest
 from test.asynchronous import AsyncIntegrationTest
-from test.asynchronous.unified_format import generate_test_classes
+from test.asynchronous.unified_format import generate_test_classes, get_test_path
 from test.utils_shared import CMAPListener
 from typing import Any, Optional
 
@@ -40,16 +40,8 @@
 
 _IS_SYNC = False
 
-# Location of JSON test specifications.
-if _IS_SYNC:
-    _TEST_PATH = os.path.join(pathlib.Path(__file__).resolve().parent, "handshake", "unified")
-else:
-    _TEST_PATH = os.path.join(
-        pathlib.Path(__file__).resolve().parent.parent, "handshake", "unified"
-    )
-
 # Generate unified tests.
-globals().update(generate_test_classes(_TEST_PATH, module=__name__))
+globals().update(generate_test_classes(get_test_path("handshake", "unified"), module=__name__))
 
 
 def _get_handshake_driver_info(request):

test/asynchronous/test_collection_management.py

@@ -22,20 +22,12 @@
 sys.path[0:0] = [""]
 
 from test import unittest
-from test.asynchronous.unified_format import generate_test_classes
+from test.asynchronous.unified_format import generate_test_classes, get_test_path
 
 _IS_SYNC = False
 
-# Location of JSON test specifications.
-if _IS_SYNC:
-    _TEST_PATH = os.path.join(pathlib.Path(__file__).resolve().parent, "collection_management")
-else:
-    _TEST_PATH = os.path.join(
-        pathlib.Path(__file__).resolve().parent.parent, "collection_management"
-    )
-
 # Generate unified tests.
-globals().update(generate_test_classes(_TEST_PATH, module=__name__))
+globals().update(generate_test_classes(get_test_path("collection_management"), module=__name__))
 
 if __name__ == "__main__":
     unittest.main()

test/asynchronous/test_command_logging.py

@@ -22,20 +22,13 @@
 sys.path[0:0] = [""]
 
 from test import unittest
-from test.asynchronous.unified_format import generate_test_classes
+from test.asynchronous.unified_format import generate_test_classes, get_test_path
 
 _IS_SYNC = False
 
-# Location of JSON test specifications.
-if _IS_SYNC:
-    _TEST_PATH = os.path.join(pathlib.Path(__file__).resolve().parent, "command_logging")
-else:
-    _TEST_PATH = os.path.join(pathlib.Path(__file__).resolve().parent.parent, "command_logging")
-
-
 globals().update(
     generate_test_classes(
-        _TEST_PATH,
+        get_test_path("command_logging"),
         module=__name__,
     )
 )

test/asynchronous/test_command_monitoring.py

@@ -22,20 +22,13 @@
 sys.path[0:0] = [""]
 
 from test import unittest
-from test.asynchronous.unified_format import generate_test_classes
+from test.asynchronous.unified_format import generate_test_classes, get_test_path
 
 _IS_SYNC = False
 
-# Location of JSON test specifications.
-if _IS_SYNC:
-    _TEST_PATH = os.path.join(pathlib.Path(__file__).resolve().parent, "command_monitoring")
-else:
-    _TEST_PATH = os.path.join(pathlib.Path(__file__).resolve().parent.parent, "command_monitoring")
-
-
 globals().update(
     generate_test_classes(
-        _TEST_PATH,
+        get_test_path("command_monitoring"),
         module=__name__,
     )
 )

test/asynchronous/test_connection_logging.py

@@ -22,20 +22,13 @@
 sys.path[0:0] = [""]
 
 from test import unittest
-from test.asynchronous.unified_format import generate_test_classes
+from test.asynchronous.unified_format import generate_test_classes, get_test_path
 
 _IS_SYNC = False
 
-# Location of JSON test specifications.
-if _IS_SYNC:
-    _TEST_PATH = os.path.join(pathlib.Path(__file__).resolve().parent, "connection_logging")
-else:
-    _TEST_PATH = os.path.join(pathlib.Path(__file__).resolve().parent.parent, "connection_logging")
-
-
 globals().update(
     generate_test_classes(
-        _TEST_PATH,
+        get_test_path("connection_logging"),
         module=__name__,
     )
 )

test/asynchronous/test_crud_unified.py

@@ -22,18 +22,12 @@
 sys.path[0:0] = [""]
 
 from test import unittest
-from test.asynchronous.unified_format import generate_test_classes
+from test.asynchronous.unified_format import generate_test_classes, get_test_path
 
 _IS_SYNC = False
 
-# Location of JSON test specifications.
-if _IS_SYNC:
-    _TEST_PATH = os.path.join(pathlib.Path(__file__).resolve().parent, "crud", "unified")
-else:
-    _TEST_PATH = os.path.join(pathlib.Path(__file__).resolve().parent.parent, "crud", "unified")
-
 # Generate unified tests.
-globals().update(generate_test_classes(_TEST_PATH, module=__name__))
+globals().update(generate_test_classes(get_test_path("crud", "unified"), module=__name__))
 
 if __name__ == "__main__":
     unittest.main()

test/asynchronous/test_csot.py

@@ -22,7 +22,7 @@
 sys.path[0:0] = [""]
 
 from test.asynchronous import AsyncIntegrationTest, async_client_context, unittest
-from test.asynchronous.unified_format import generate_test_classes
+from test.asynchronous.unified_format import generate_test_classes, get_test_path
 from test.asynchronous.utils import flaky
 
 import pymongo
@@ -31,14 +31,8 @@
 
 _IS_SYNC = False
 
-# Location of JSON test specifications.
-if _IS_SYNC:
-    TEST_PATH = os.path.join(Path(__file__).resolve().parent, "csot")
-else:
-    TEST_PATH = os.path.join(Path(__file__).resolve().parent.parent, "csot")
-
 # Generate unified tests.
-globals().update(generate_test_classes(TEST_PATH, module=__name__))
+globals().update(generate_test_classes(get_test_path("csot"), module=__name__))
 
 
 class TestCSOT(AsyncIntegrationTest):