Skip to content

Commit 54fe7c2

Browse files
committed
fix flaky test
1 parent ae4e571 commit 54fe7c2

File tree

2 files changed

+8
-4
lines changed

2 files changed

+8
-4
lines changed

test/asynchronous/test_encryption.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3481,7 +3481,6 @@ class TestNoSessionsSupport(AsyncEncryptionIntegrationTest):
34813481
mongocryptd_client: AsyncMongoClient
34823482
MONGOCRYPTD_PORT = 27020
34833483

3484-
@flaky # PYTHON-4982
34853484
async def asyncSetUp(self) -> None:
34863485
await super().asyncSetUp()
34873486
start_mongocryptd(self.MONGOCRYPTD_PORT)
@@ -3494,7 +3493,6 @@ async def asyncSetUp(self) -> None:
34943493
hello = await self.mongocryptd_client.db.command("hello")
34953494
self.assertNotIn("logicalSessionTimeoutMinutes", hello)
34963495

3497-
@flaky # PYTHON-4982
34983496
async def test_implicit_session_ignored_when_unsupported(self):
34993497
self.listener.reset()
35003498
with self.assertRaises(OperationFailure):
@@ -3507,6 +3505,8 @@ async def test_implicit_session_ignored_when_unsupported(self):
35073505

35083506
self.assertNotIn("lsid", self.listener.started_events[1].command)
35093507

3508+
await self.mongocryptd_client.close()
3509+
35103510
async def test_explicit_session_errors_when_unsupported(self):
35113511
self.listener.reset()
35123512
async with self.mongocryptd_client.start_session() as s:
@@ -3519,6 +3519,8 @@ async def test_explicit_session_errors_when_unsupported(self):
35193519
):
35203520
await self.mongocryptd_client.db.test.insert_one({"x": 1}, session=s)
35213521

3522+
await self.mongocryptd_client.close()
3523+
35223524

35233525
if __name__ == "__main__":
35243526
unittest.main()

test/test_encryption.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3463,7 +3463,6 @@ class TestNoSessionsSupport(EncryptionIntegrationTest):
34633463
mongocryptd_client: MongoClient
34643464
MONGOCRYPTD_PORT = 27020
34653465

3466-
@flaky # PYTHON-4982
34673466
def setUp(self) -> None:
34683467
super().setUp()
34693468
start_mongocryptd(self.MONGOCRYPTD_PORT)
@@ -3476,7 +3475,6 @@ def setUp(self) -> None:
34763475
hello = self.mongocryptd_client.db.command("hello")
34773476
self.assertNotIn("logicalSessionTimeoutMinutes", hello)
34783477

3479-
@flaky # PYTHON-4982
34803478
def test_implicit_session_ignored_when_unsupported(self):
34813479
self.listener.reset()
34823480
with self.assertRaises(OperationFailure):
@@ -3489,6 +3487,8 @@ def test_implicit_session_ignored_when_unsupported(self):
34893487

34903488
self.assertNotIn("lsid", self.listener.started_events[1].command)
34913489

3490+
self.mongocryptd_client.close()
3491+
34923492
def test_explicit_session_errors_when_unsupported(self):
34933493
self.listener.reset()
34943494
with self.mongocryptd_client.start_session() as s:
@@ -3501,6 +3501,8 @@ def test_explicit_session_errors_when_unsupported(self):
35013501
):
35023502
self.mongocryptd_client.db.test.insert_one({"x": 1}, session=s)
35033503

3504+
self.mongocryptd_client.close()
3505+
35043506

35053507
if __name__ == "__main__":
35063508
unittest.main()

0 commit comments

Comments
 (0)