Support running ssl-enabled integration tests

Author: blink1073

.github/workflows/test-python.yml

@@ -180,15 +180,19 @@ jobs:
           python-version: "3.9"
       - name: Install just
         run: uv tool install rust-just
-      - id: setup-mongodb
-        uses: mongodb-labs/drivers-evergreen-tools@master
-        with:
-          version: "8.0"
       - name: Install dependencies
-        run: just install
+        run: |
+          just install
+      - name: Start server
+        run: just run-server
       - name: Run tests
         run:  |
           just integration-tests
+      - name: Start server with SSL
+        run: just run-server --ssl
+      - name: Run tests
+        run: |
+          SSL=ssl just integration-tests
 
   make_sdist:
     runs-on: ubuntu-latest

CONTRIBUTING.md

@@ -419,6 +419,8 @@ The `integration_tests` directory has a set of scripts that verify the usage of
 
 To run the tests, use `just integration_tests`.
 
+The tests should be able to run with and without SSL enabled.
+
 ## Specification Tests
 
 The MongoDB [specifications repository](https://github.com/mongodb/specifications)

integration_tests/README.md

@@ -6,14 +6,37 @@ Each test uses [PEP 723 inline metadata](https://packaging.python.org/en/latest/
 
 The `run.sh` convenience script can be used to run all of the files using `uv`.
 
-When creating a new script, use the following snippet to ensure that the local version of PyMongo is used:
+Here is an example header for the script with the inline dependencies:
+
+```python
+# /// script
+# dependencies = [
+#   "uvloop>=0.18"
+# ]
+# requires-python = ">=3.10"
+# ///
+```
+
+Here is an example of using the test helper function to create a configured client for the test:
 
 
 ```python
-# Use pymongo from parent directory.
+import asyncio
 import sys
 from pathlib import Path
 
+# Use pymongo from parent directory.
 root = Path(__file__).parent.parent
 sys.path.insert(0, str(root))
+
+from test.asynchronous import async_simple_test_client  # noqa: E402
+
+
+async def main():
+    async with async_simple_test_client() as client:
+        result = await client.admin.command("ping")
+        assert result["ok"]
+
+
+asyncio.run(main())
 ```

integration_tests/test_uv_loop.py

@@ -15,13 +15,13 @@
 root = Path(__file__).parent.parent
 sys.path.insert(0, str(root))
 
-from pymongo import AsyncMongoClient  # noqa: E402
+from test.asynchronous import async_simple_test_client  # noqa: E402
 
 
 async def main():
-    client = AsyncMongoClient()
-    result = await client.admin.command("ping")
-    assert result["ok"]
+    async with async_simple_test_client() as client:
+        result = await client.admin.command("ping")
+        assert result["ok"]
 
 
 uvloop.run(main())

test/__init__.py

@@ -1248,6 +1248,13 @@ def teardown():
     print_running_clients()
 
 
+@contextmanager
+def async_simple_test_client():
+    client_context.init()
+    yield client_context.client
+    client_context.client.close()
+
+
 def test_cases(suite):
     """Iterator over all TestCases within a TestSuite."""
     for suite_or_case in suite._tests:

test/asynchronous/__init__.py

@@ -1264,6 +1264,13 @@ async def async_teardown():
     print_running_clients()
 
 
+@asynccontextmanager
+async def async_simple_test_client():
+    await async_client_context.init()
+    yield async_client_context.client
+    await async_client_context.client.close()
+
+
 def test_cases(suite):
     """Iterator over all TestCases within a TestSuite."""
     for suite_or_case in suite._tests: