From 2ac1db2d24e90cb0e23e7e4d6f19bd5dd156b3a1 Mon Sep 17 00:00:00 2001 From: Shane Harvey Date: Mon, 12 Jul 2021 15:27:35 -0700 Subject: [PATCH] PYTHON-2806 Fix test_aggregate_raw_transaction --- test/test_cursor.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/test/test_cursor.py b/test/test_cursor.py index 021e4d7cb4..b2de09429b 100644 --- a/test/test_cursor.py +++ b/test/test_cursor.py @@ -1483,10 +1483,13 @@ def test_find_raw_transaction(self): 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 @@ def test_aggregate_raw_transaction(self): [{'$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]))