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

PYTHON-2806 Fix test_aggregate_raw_transaction (#673)

Browse Source
This commit is contained in:
Shane Harvey 2021-07-12 16:54:50 -07:00 committed by GitHub
parent 6d1ebf4597
commit da49bd88a2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
test/test_cursor.py
View File

@ -1483,10 +1483,13 @@ class TestRawBatchCursor(IntegrationTest):
session=session).sort('_id'))
cmd = listener.results['started'][0]
self.assertEqual(cmd.command_name, 'find')
self.assertEqual(cmd.command['$clusterTime'],
decode_all(session.cluster_time.raw)[0])
self.assertIn('$clusterTime', cmd.command)
self.assertEqual(cmd.command['startTransaction'], True)
self.assertEqual(cmd.command['txnNumber'], 1)
# Ensure we update $clusterTime from the command response.
last_cmd = listener.results['succeeded'][-1]
self.assertEqual(last_cmd.reply['$clusterTime']['clusterTime'],
session.cluster_time['clusterTime'])
self.assertEqual(1, len(batches))
self.assertEqual(docs, decode_all(batches[0]))
@ -1677,9 +1680,13 @@ class TestRawBatchCommandCursor(IntegrationTest):
[{'$sort': {'_id': 1}}], session=session))
cmd = listener.results['started'][0]
self.assertEqual(cmd.command_name, 'aggregate')
self.assertEqual(cmd.command['$clusterTime'], session.cluster_time)
self.assertIn('$clusterTime', cmd.command)
self.assertEqual(cmd.command['startTransaction'], True)
self.assertEqual(cmd.command['txnNumber'], 1)
# Ensure we update $clusterTime from the command response.
last_cmd = listener.results['succeeded'][-1]
self.assertEqual(last_cmd.reply['$clusterTime']['clusterTime'],
session.cluster_time['clusterTime'])
self.assertEqual(1, len(batches))
self.assertEqual(docs, decode_all(batches[0]))