Skip to content

Commit 70d5f74

Browse files
committed
bring back setUpClass and tearDownClass
1 parent dd9589e commit 70d5f74

File tree

2 files changed

+28
-14
lines changed

2 files changed

+28
-14
lines changed

test/asynchronous/test_sdam_monitoring_spec.py

Lines changed: 14 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -274,15 +274,23 @@ class TestSdamMonitoring(AsyncIntegrationTest):
274274
test_client: AsyncMongoClient
275275
coll: AsyncCollection
276276

277-
@async_client_context.require_failCommand_fail_point
278-
async def asyncSetUp(self):
279-
await super().asyncSetUp()
277+
@classmethod
278+
def setUpClass(cls):
280279
# Speed up the tests by decreasing the event publish frequency.
281-
self.knobs = client_knobs(
280+
cls.knobs = client_knobs(
282281
events_queue_frequency=0.1, heartbeat_frequency=0.1, min_heartbeat_interval=0.1
283282
)
284-
self.knobs.enable()
285-
self.listener = ServerAndTopologyEventListener()
283+
cls.knobs.enable()
284+
cls.listener = ServerAndTopologyEventListener()
285+
286+
@classmethod
287+
def tearDownClass(cls):
288+
cls.knobs.disable()
289+
290+
@async_client_context.require_failCommand_fail_point
291+
async def asyncSetUp(self):
292+
await super().asyncSetUp()
293+
286294
retry_writes = async_client_context.supports_transactions()
287295
self.test_client = await self.async_rs_or_single_client(
288296
event_listeners=[self.listener], retryWrites=retry_writes
@@ -292,7 +300,6 @@ async def asyncSetUp(self):
292300
self.listener.reset()
293301

294302
async def asyncTearDown(self):
295-
self.knobs.disable()
296303
await super().asyncTearDown()
297304

298305
async def _test_app_error(self, fail_command_opts, expected_error):

test/test_sdam_monitoring_spec.py

Lines changed: 14 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -274,15 +274,23 @@ class TestSdamMonitoring(IntegrationTest):
274274
test_client: MongoClient
275275
coll: Collection
276276

277-
@client_context.require_failCommand_fail_point
278-
def setUp(self):
279-
super().setUp()
277+
@classmethod
278+
def setUpClass(cls):
280279
# Speed up the tests by decreasing the event publish frequency.
281-
self.knobs = client_knobs(
280+
cls.knobs = client_knobs(
282281
events_queue_frequency=0.1, heartbeat_frequency=0.1, min_heartbeat_interval=0.1
283282
)
284-
self.knobs.enable()
285-
self.listener = ServerAndTopologyEventListener()
283+
cls.knobs.enable()
284+
cls.listener = ServerAndTopologyEventListener()
285+
286+
@classmethod
287+
def tearDownClass(cls):
288+
cls.knobs.disable()
289+
290+
@client_context.require_failCommand_fail_point
291+
def setUp(self):
292+
super().setUp()
293+
286294
retry_writes = client_context.supports_transactions()
287295
self.test_client = self.rs_or_single_client(
288296
event_listeners=[self.listener], retryWrites=retry_writes
@@ -292,7 +300,6 @@ def setUp(self):
292300
self.listener.reset()
293301

294302
def tearDown(self):
295-
self.knobs.disable()
296303
super().tearDown()
297304

298305
def _test_app_error(self, fail_command_opts, expected_error):

0 commit comments

Comments
 (0)