Fix test

Author: NoahStapp

test/asynchronous/test_retryable_reads.py

@@ -227,6 +227,8 @@ async def test_retryable_reads_are_retried_on_the_same_implicit_session(self):
             retryReads=True,
         )
 
+        await client.t.t.insert_one({"x": 1})
+
         commands = [
             ("aggregate", lambda: client.t.t.count_documents({})),
             ("aggregate", lambda: client.t.t.aggregate([{"$match": {}}])),
@@ -250,13 +252,14 @@ async def test_retryable_reads_are_retried_on_the_same_implicit_session(self):
                 await operation()
 
             #  Assert that both events occurred on the same session.
-            lsids = [
-                event.command["lsid"]
+            command_docs = [
+                event.command
                 for event in listener.started_events
                 if event.command_name == command_name
             ]
-            self.assertEqual(len(lsids), 2)
-            self.assertEqual(lsids[0], lsids[1])
+            self.assertEqual(len(command_docs), 2)
+            self.assertEqual(command_docs[0]["lsid"], command_docs[1]["lsid"])
+            self.assertIsNot(command_docs[0], command_docs[1])
 
 
 if __name__ == "__main__":

test/test_retryable_reads.py

@@ -225,6 +225,8 @@ def test_retryable_reads_are_retried_on_the_same_implicit_session(self):
             retryReads=True,
         )
 
+        client.t.t.insert_one({"x": 1})
+
         commands = [
             ("aggregate", lambda: client.t.t.count_documents({})),
             ("aggregate", lambda: client.t.t.aggregate([{"$match": {}}])),
@@ -248,13 +250,14 @@ def test_retryable_reads_are_retried_on_the_same_implicit_session(self):
                 operation()
 
             #  Assert that both events occurred on the same session.
-            lsids = [
-                event.command["lsid"]
+            command_docs = [
+                event.command
                 for event in listener.started_events
                 if event.command_name == command_name
             ]
-            self.assertEqual(len(lsids), 2)
-            self.assertEqual(lsids[0], lsids[1])
+            self.assertEqual(len(command_docs), 2)
+            self.assertEqual(command_docs[0]["lsid"], command_docs[1]["lsid"])
+            self.assertIsNot(command_docs[0], command_docs[1])
 
 
 if __name__ == "__main__":