Test improvements.

Author: anthony-tuininga

tests/test_1100_connection.py

@@ -1,5 +1,5 @@
 # -----------------------------------------------------------------------------
-# Copyright (c) 2020, 2024, Oracle and/or its affiliates.
+# Copyright (c) 2020, 2025, Oracle and/or its affiliates.
 #
 # This software is dual-licensed to you under the Universal Permissive License
 # (UPL) 1.0 as shown at https://oss.oracle.com/licenses/upl and Apache License
@@ -954,6 +954,30 @@ def hook(params):
         conn = test_env.get_connection()
         self.assertEqual(conn.stmtcachesize, orig_stmtcachesize)
 
+    def test_1157(self):
+        "1157 - test connect() with multiple parameters hooks"
+
+        def hook1(params):
+            order.append("first")
+
+        def hook2(params):
+            order.append("second")
+
+        def hook3(params):
+            order.append("third")
+
+        oracledb.register_params_hook(hook1)
+        oracledb.register_params_hook(hook2)
+        oracledb.register_params_hook(hook3)
+        try:
+            order = []
+            test_env.get_connection()
+            self.assertEqual(order, ["first", "second", "third"])
+        finally:
+            oracledb.unregister_params_hook(hook1)
+            oracledb.unregister_params_hook(hook2)
+            oracledb.unregister_params_hook(hook3)
+
 
 if __name__ == "__main__":
     test_env.run_test_cases()

tests/test_3400_soda_collection.py

@@ -1,5 +1,5 @@
 # -----------------------------------------------------------------------------
-# Copyright (c) 2020, 2024, Oracle and/or its affiliates.
+# Copyright (c) 2020, 2025, Oracle and/or its affiliates.
 #
 # This software is dual-licensed to you under the Universal Permissive License
 # (UPL) 1.0 as shown at https://oss.oracle.com/licenses/upl and Apache License
@@ -360,6 +360,10 @@ def test_3413(self):
         coll.truncate()
         self.assertEqual(coll.find().count(), 0)
 
+    @unittest.skipIf(
+        test_env.get_client_version() < (19, 11),
+        "client version not supported.. min required 19.11",
+    )
     def test_3414(self):
         "3414 - verify hints are reflected in the executed SQL statement"
         soda_db = self.get_soda_database()
@@ -755,6 +759,10 @@ def test_3434(self):
         with self.assertRaisesFullCode("ORA-40734"):
             coll.find().keys(keys).replaceOneAndGet({"data": "new"})
 
+    @unittest.skipIf(
+        test_env.get_client_version() < (19, 9),
+        "client version not supported.. min required 19.9",
+    )
     def test_3435(self):
         "3435 - test writting a read-only collection"
         soda_db = self.get_soda_database()

tests/test_6300_cursor_other_async.py

@@ -575,7 +575,7 @@ def type_handler(cursor, metadata):
 
     @unittest.skipIf(test_env.get_is_drcp(), "not supported with DRCP")
     async def test_6335(self):
-        "4535 - kill connection with open cursor"
+        "6335 - kill connection with open cursor"
         admin_conn = await test_env.get_admin_connection_async()
         conn = await test_env.get_connection_async()
         self.assertEqual(conn.is_healthy(), True)
@@ -864,7 +864,7 @@ async def test_6350(self):
         self.assertEqual(fetched_value, value)
 
     async def test_6351(self):
-        "4360 - fetch JSON columns as Python objects"
+        "6351 - fetch JSON columns as Python objects"
         expected_data = [
             (1, [1, 2, 3], [4, 5, 6], [7, 8, 9]),
             (2, None, None, None),

tests/test_8000_dataframe.py

@@ -476,6 +476,50 @@ def test_8017(self):
             DATASET_2, batch_size=len(DATASET_2), num_batches=1
         )
 
+    def test_8018(self):
+        "8018 - verify get_column() returns the correct value"
+        self.__check_interop()
+        self.__populate_table(DATASET_1)
+        statement = "select * from TestDataFrame order by Id"
+        ora_df = self.conn.fetch_df_all(statement)
+        array = pyarrow.array(ora_df.get_column(1))
+        self.assertEqual(array.to_pylist(), ["John", "Big"])
+
+    def test_8019(self):
+        "8019 - verify OracleColumn and get_buffers"
+        self.__populate_table(DATASET_1)
+        statement = "select * from TestDataFrame order by Id"
+        ora_df = self.conn.fetch_df_all(statement)
+        ora_col = ora_df.get_column(1)
+        self.assertEqual(ora_col.num_chunks(), 1)
+        self.assertEqual(ora_col.size(), 2)
+
+        buffers = ora_col.get_buffers()
+        self.assertEqual(len(buffers), 3)
+        self.assertIsNotNone(buffers["data"])
+        self.assertIsNotNone(buffers["offsets"])
+        self.assertIsNone(buffers["validity"])
+
+    def test_8020(self):
+        "8020 - verify  OracleColumn Attributes"
+        self.__populate_table(DATASET_2)
+        statement = "select * from TestDataFrame order by Id"
+        ora_df = self.conn.fetch_df_all(statement)
+
+        ora_col = ora_df.get_column(0)
+        self.assertEqual(ora_col.describe_null[0], 0)
+        self.assertEqual(ora_col.dtype[0], 0)
+        metadata = {"name": "ID", "size": 2, "num_chunks": 1}
+        self.assertEqual(metadata, ora_col.metadata)
+        self.assertEqual(ora_col.null_count, 0)
+
+        ora_col = ora_df.get_column(4)
+        self.assertEqual(ora_col.describe_null[0], 3)
+        self.assertEqual(ora_col.dtype[0], 21)
+        metadata = {"name": "COUNTRY", "size": 2, "num_chunks": 1}
+        self.assertEqual(metadata, ora_col.metadata)
+        self.assertEqual(ora_col.null_count, 1)
+
 
 if __name__ == "__main__":
     test_env.run_test_cases()