reset client context after primary steps down

Author: sleepyStick

test/__init__.py

@@ -859,6 +859,12 @@ def max_message_size_bytes(self):
 client_context = ClientContext()
 
 
+def recreate_client_context():
+    global client_context
+    client_context = ClientContext()
+    client_context.init()
+
+
 class PyMongoTestCase(unittest.TestCase):
     def assertEqualCommand(self, expected, actual, msg=None):
         self.assertEqual(sanitize_cmd(expected), sanitize_cmd(actual), msg)

test/asynchronous/__init__.py

@@ -861,6 +861,12 @@ async def max_message_size_bytes(self):
 async_client_context = AsyncClientContext()
 
 
+async def recreate_client_context():
+    global async_client_context
+    async_client_context = AsyncClientContext()
+    await async_client_context.init()
+
+
 class AsyncPyMongoTestCase(unittest.IsolatedAsyncioTestCase):
     def assertEqualCommand(self, expected, actual, msg=None):
         self.assertEqual(sanitize_cmd(expected), sanitize_cmd(actual), msg)

test/asynchronous/test_connections_survive_primary_stepdown_spec.py

@@ -19,7 +19,12 @@
 
 sys.path[0:0] = [""]
 
-from test.asynchronous import AsyncIntegrationTest, async_client_context, unittest
+from test.asynchronous import (
+    AsyncIntegrationTest,
+    async_client_context,
+    recreate_client_context,
+    unittest,
+)
 from test.asynchronous.helpers import async_repl_set_step_down
 from test.utils import (
     CMAPListener,
@@ -60,6 +65,7 @@ async def _setup_class(cls):
     @classmethod
     async def _tearDown_class(cls):
         await cls.client.close()
+        await recreate_client_context()
 
     async def asyncSetUp(self):
         # Note that all ops use same write-concern as self.db (majority).

test/asynchronous/test_monitoring.py

@@ -413,6 +413,7 @@ async def test_not_primary_error(self):
         try:
             await client.pymongo_test.test.find_one_and_delete({})
         except NotPrimaryError as exc:
+            print("an error was raised")
             error = exc.errors
         started = self.listener.started_events[0]
         failed = self.listener.failed_events[0]

test/test_connections_survive_primary_stepdown_spec.py

@@ -19,7 +19,12 @@
 
 sys.path[0:0] = [""]
 
-from test import IntegrationTest, client_context, unittest
+from test import (
+    IntegrationTest,
+    client_context,
+    recreate_client_context,
+    unittest,
+)
 from test.helpers import repl_set_step_down
 from test.utils import (
     CMAPListener,
@@ -60,6 +65,7 @@ def _setup_class(cls):
     @classmethod
     def _tearDown_class(cls):
         cls.client.close()
+        recreate_client_context()
 
     def setUp(self):
         # Note that all ops use same write-concern as self.db (majority).

test/test_monitoring.py

@@ -413,6 +413,7 @@ def test_not_primary_error(self):
         try:
             client.pymongo_test.test.find_one_and_delete({})
         except NotPrimaryError as exc:
+            print("an error was raised")
             error = exc.errors
         started = self.listener.started_events[0]
         failed = self.listener.failed_events[0]