mongodb/mongo-python-driver
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

PYTHON-2214 Tolerate StaleConfig errors in test_create_collection

Browse Source
This commit is contained in:
Shane Harvey 2020-04-29 10:32:54 -07:00
parent 07c834ea3e
commit d8342367a9
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
test/test_transactions.py
View File

@ -214,10 +214,15 @@ class TestTransactions(TransactionsBase):
db = client.pymongo_test
coll = db.test_create_collection
self.addCleanup(coll.drop)
with client.start_session() as s, s.start_transaction():
coll2 = db.create_collection(coll.name, session=s)
# Use with_transaction to avoid StaleConfig errors on sharded clusters.
def create_and_insert(session):
coll2 = db.create_collection(coll.name, session=session)
self.assertEqual(coll, coll2)
coll.insert_one({}, session=s)
coll.insert_one({}, session=session)
with client.start_session() as s:
s.with_transaction(create_and_insert)
# Outside a transaction we raise CollectionInvalid on existing colls.
with self.assertRaises(CollectionInvalid):