PYTHON-3050 Add session support to find_raw_batches (#41)

Author: blink1073

bindings/python/pymongoarrow/api.py

@@ -58,7 +58,7 @@ def find_arrow_all(collection, query, *, schema, **kwargs):
     context = PyMongoArrowContext.from_schema(
         schema, codec_options=collection.codec_options)
 
-    for opt in ('session', 'cursor_type'):
+    for opt in ('cursor_type',):
         if kwargs.pop(opt, None):
             warnings.warn(
                 f'Ignoring option {opt!r} as it is not supported by '

bindings/python/test/test_arrow.py

@@ -72,7 +72,7 @@ def test_find_simple(self):
         self.assertEqual(find_cmd.command['projection'],
                          {'_id': True, 'data': True})
 
-    def test_find_projection(self):
+    def test_find_with_projection(self):
         expected = Table.from_pydict(
             {'_id': [4, 3], 'data': [None, 60]},
             ArrowSchema([('_id', int32()), ('data', int64())]))
@@ -87,6 +87,17 @@ def test_find_projection(self):
         self.assertEqual(find_cmd.command_name, 'find')
         self.assertEqual(find_cmd.command['projection'], projection)
 
+    def test_find_with_session(self):
+        with self.client.start_session() as session:
+            self.assertIsNone(session.operation_time)
+            last_use = session._server_session.last_use
+            expected = Table.from_pydict(
+            {'_id': [1, 2, 3, 4], 'data': [10, 20, 30, None]},
+            ArrowSchema([('_id', int32()), ('data', int64())]))
+            table = self.run_find({}, schema=self.schema, session=session)
+            self.assertEqual(table, expected)
+            self.assertIsNotNone(session.operation_time)
+
     def test_find_multiple_batches(self):
         orig_method = self.coll.find_raw_batches