Fixed bug when an output type handler is used and the value of

cursor.prefetchrows exceeds cursor.arraysize (#173).

Author: anthony-tuininga

doc/src/release_notes.rst

@@ -41,6 +41,9 @@ Thin Mode Changes
 #) Fixed bug when a connection is discarded from the pool during
  :meth:`ConnectionPool.acquire()` and the ping check fails due to the
  connection being dead.
+#) Fixed bug when an output type handler is used and the value of
+ cursor.prefetchrows exceeds cursor.arraysize
+ (`issue 173 <https://github.com/oracle/python-oracledb/issues/173>`__).
 
 Thick Mode Changes
 ++++++++++++++++++

src/oracledb/impl/base/cursor.pyx

@@ -155,6 +155,9 @@ cdef class BaseCursorImpl:
  self._verify_var(var)
  var_impl = var._impl
  var_impl._fetch_info = fetch_info
+ if var_impl.num_elements < self._fetch_array_size:
+ var_impl.num_elements = self._fetch_array_size
+ var_impl._finalize_init()
  self.fetch_vars[pos] = var
  self.fetch_var_impls[pos] = var_impl
  return 0

tests/test_3900_cursor_execute.py

@@ -372,5 +372,16 @@ def __getitem__(self, index):
  order by IntCol""")
  self.assertEqual(self.cursor.fetchall(), expected_data)
 
+ def test_3929_output_type_handler_with_prefetch_gt_arraysize(self):
+ "3929 - test an output type handler with prefetch > arraysize"
+ def type_handler(cursor, name, default_type, size, precision, scale):
+ return cursor.var(default_type, arraysize=cursor.arraysize)
+ self.cursor.arraysize = 2
+ self.cursor.prefetchrows = 3
+ self.cursor.outputtypehandler = type_handler
+ self.cursor.execute("select level from dual connect by level <= 5")
+ self.assertEqual(self.cursor.fetchall(),
+ [(1,), (2,), (3,), (4,), (5,)])
+
 if __name__ == "__main__":
  test_env.run_test_cases()