PYTHON-2806 Fix test_aggregate_raw_transaction (mongodb#673)

Author: ShaneHarvey

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]))