Test improvements.

Author: anthony-tuininga

tests/test_3400_soda_collection.py

@@ -752,6 +752,92 @@ def test_3432_drop_empty_mapped_collection(self):
         self.assertTrue(mapped_coll.drop())
         self.assertFalse(original_coll.drop())
 
+    def test_3433_replace_one_returns(self):
+        "3433 - test that replaceOne() returns a correct boolean"
+        soda_db = self.get_soda_database()
+        coll = soda_db.createCollection("TestReplaceDocReturns")
+        coll.find().remove()
+        doc = coll.insertOneAndGet({"address": {"city": "Sydney"}})
+
+        new_content = {"address": {"city": "Melbourne"}}
+        self.assertTrue(coll.find().key(doc.key).replaceOne(new_content))
+
+        unregistered_key = "DB4A2628F1E0985C891F3F4836"
+        self.assertFalse(
+            coll.find().key(unregistered_key).replaceOne(new_content)
+        )
+        self.conn.commit()
+        coll.drop()
+
+    def test_3434_replace_one_and_get_negative(self):
+        "3434 - replaceOne() and replaceOneAndGet() with invalid scenarios"
+        conn = test_env.get_connection()
+        soda_db = conn.getSodaDatabase()
+        coll = soda_db.createCollection("TestReplaceOneNegative")
+        coll.find().remove()
+        coll.insertMany([{"Wisdom": 1.7} for d in range(2)])
+        keys = [d.key for d in coll.find().getDocuments()]
+        self.assertRaisesRegex(
+            oracledb.DatabaseError,
+            "^ORA-40734:",
+            coll.find().keys(keys).replaceOne,
+            {"data": "new"},
+        )
+        self.assertRaisesRegex(
+            oracledb.DatabaseError,
+            "^ORA-40734:",
+            coll.find().keys(keys).replaceOneAndGet,
+            {"data": "new"},
+        )
+
+    def test_3435_write_read_only_coll_negative(self):
+        "3435 - test writting a read-only collection"
+        soda_db = self.get_soda_database()
+        metadata = {
+            "readOnly": True,
+        }
+        coll = soda_db.createCollection("TestCollReadOnly", metadata)
+
+        methods = [
+            coll.insertOne,
+            coll.insertOneAndGet,
+            coll.insertMany,
+            coll.insertManyAndGet,
+            coll.save,
+            coll.saveAndGet,
+        ]
+        for method in methods:
+            with self.subTest(method=method):
+                self.assertRaisesRegex(
+                    oracledb.DatabaseError,
+                    "^ORA-40663:",
+                    method,
+                    {"Song 1": "No end"},
+                )
+
+    def test_3436_create_collection_same_metadata(self):
+        "3436 - createCollection() with the same name and metadata"
+        soda_db = self.get_soda_database()
+        coll_name = "TestCollSameMetadata"
+        coll1 = soda_db.createCollection(coll_name, {"readOnly": True})
+        coll2 = soda_db.createCollection(coll_name, {"readOnly": True})
+        self.assertTrue(coll1.drop())
+        self.assertFalse(coll2.drop())
+
+    def test_3437_create_collection_different_metadata(self):
+        "3437 - createCollection() with the same name but different metadata"
+        soda_db = self.get_soda_database()
+        coll_name = "TestCollDifferentMetadata"
+        coll = soda_db.createCollection(coll_name)
+        with self.assertRaisesRegex(oracledb.DatabaseError, "^ORA-40669:"):
+            soda_db.createCollection(coll_name, {"readOnly": False})
+        coll.drop()
+
+        coll = soda_db.createCollection(coll_name, {"readOnly": True})
+        with self.assertRaisesRegex(oracledb.DatabaseError, "^ORA-40669:"):
+            soda_db.createCollection(coll_name, {"readOnly": False})
+        coll.drop()
+
 
 if __name__ == "__main__":
     test_env.run_test_cases()

tests/test_4300_cursor_other.py

@@ -987,6 +987,45 @@ def test_4361_fetch_domain_and_annotations(self):
         }
         self.assertEqual(column_2.annotations, expected_annotations)
 
+    def test_4362_statement_after_execute(self):
+        "4362 - test getting statement after it was executed"
+        cursor = self.conn.cursor()
+        sql = "select 1 from dual"
+        cursor.execute(sql)
+        self.assertEqual(cursor.statement, sql)
+
+    def test_4363_cursor_fetchvars(self):
+        "4363 - test getting cursor fetchvars"
+        cursor = self.conn.cursor()
+        self.assertIsNone(cursor.fetchvars)
+
+        cursor.execute("truncate table TestTempTable")
+        cursor.execute(
+            "insert into TestTempTable (IntCol, StringCol1) values (1, '12')",
+        )
+        cursor.execute("select IntCol, StringCol1 from TestTempTable")
+        cursor.fetchall()
+        self.assertEqual(len(cursor.fetchvars), 2)
+        self.assertEqual(cursor.fetchvars[0].getvalue(), 1)
+        self.assertEqual(cursor.fetchvars[1].getvalue(), "12")
+
+    def test_4364_fetchmany_with_cursor_arraysize(self):
+        "4364 - test fetchmany() with non-default cursor.arraysize"
+        self.cursor.arraysize = 20
+        values = [(i,) for i in range(30)]
+        self.cursor.execute("truncate table TestTempTable")
+        self.cursor.executemany(
+            "insert into TestTempTable (IntCol) values (:1)", values
+        )
+        self.cursor.execute("select IntCol from TestTempTable order by IntCol")
+        # fetch first 20 elements
+        fetched_values = self.cursor.fetchmany()
+        self.assertEqual(fetched_values, values[: self.cursor.arraysize])
+
+        # fetch missing elements
+        fetched_values = self.cursor.fetchmany()
+        self.assertEqual(fetched_values, values[self.cursor.arraysize :])
+
 
 if __name__ == "__main__":
     test_env.run_test_cases()

tests/test_env.py

@@ -70,6 +70,11 @@
 
 import oracledb
 
+# Python 3.7 doesn't have support for testing asyncio so fake it to avoid the
+# entire test suite failing!
+if not hasattr(unittest, "IsolatedAsyncioTestCase"):
+    unittest.IsolatedAsyncioTestCase = object
+
 # default values
 DEFAULT_MAIN_USER = "pythontest"
 DEFAULT_PROXY_USER = "pythontestproxy"