python-cx_Oracle/test/test_2000_long_var.py

106 lines
4.1 KiB
Python

#------------------------------------------------------------------------------
# Copyright (c) 2016, 2020, Oracle and/or its affiliates. All rights reserved.
#
# Portions Copyright 2007-2015, Anthony Tuininga. All rights reserved.
#
# Portions Copyright 2001-2007, Computronix (Canada) Ltd., Edmonton, Alberta,
# Canada. All rights reserved.
#------------------------------------------------------------------------------
"""
2000 - Module for testing long and long raw variables
"""
import base
import cx_Oracle as oracledb
class TestCase(base.BaseTestCase):
def __perform_test(self, typ):
name_part = "Long" if typ is oracledb.DB_TYPE_LONG else "LongRaw"
self.cursor.execute("truncate table Test%ss" % name_part)
long_string = ""
for i in range(1, 11):
char = chr(ord('A') + i - 1)
long_string += char * 25000
self.cursor.setinputsizes(long_string=typ)
if typ is oracledb.DB_TYPE_LONG_RAW:
bind_value = long_string.encode()
else:
bind_value = long_string
self.cursor.execute("""
insert into Test%ss (
IntCol,
%sCol
) values (
:integer_value,
:long_string
)""" % (name_part, name_part),
integer_value=i,
long_string=bind_value)
self.connection.commit()
self.cursor.execute("""
select *
from Test%ss
order by IntCol""" % name_part)
long_string = ""
for integer_value, fetched_value in self.cursor:
char = chr(ord('A') + integer_value - 1)
long_string += char * 25000
if typ is oracledb.DB_TYPE_LONG_RAW:
actual_value = long_string.encode()
else:
actual_value = long_string
self.assertEqual(len(fetched_value), integer_value * 25000)
self.assertEqual(fetched_value, actual_value)
def test_2000_longs(self):
"2000 - test binding and fetching long data"
self.__perform_test(oracledb.DB_TYPE_LONG)
def test_2001_long_with_execute_many(self):
"2001 - test binding long data with executemany()"
data = []
self.cursor.execute("truncate table TestLongs")
for i in range(5):
char = chr(ord('A') + i)
long_str = char * (32768 * (i + 1))
data.append((i + 1, long_str))
self.cursor.executemany("insert into TestLongs values (:1, :2)", data)
self.connection.commit()
self.cursor.execute("select * from TestLongs order by IntCol")
fetched_data = self.cursor.fetchall()
self.assertEqual(fetched_data, data)
def test_2002_long_raws(self):
"2002 - test binding and fetching long raw data"
self.__perform_test(oracledb.DB_TYPE_LONG_RAW)
def test_2003_long_cursor_description(self):
"2003 - test cursor description is accurate for longs"
self.cursor.execute("select * from TestLongs")
expected_value = [
('INTCOL', oracledb.DB_TYPE_NUMBER, 10, None, 9, 0, 0),
('LONGCOL', oracledb.DB_TYPE_LONG, None, None, None, None, 0)
]
self.assertEqual(self.cursor.description, expected_value)
def test_2004_long_raw_cursor_description(self):
"2004 - test cursor description is accurate for long raws"
self.cursor.execute("select * from TestLongRaws")
self.assertEqual(self.cursor.description,
[ ('INTCOL', oracledb.DB_TYPE_NUMBER, 10, None, 9, 0, 0),
('LONGRAWCOL', oracledb.DB_TYPE_LONG_RAW, None, None, None,
None, 0) ])
def test_2005_array_size_too_large(self):
"2005 - test array size too large generates an exception"
self.cursor.arraysize = 268435456
self.assertRaises(oracledb.DatabaseError, self.cursor.execute,
"select * from TestLongRaws")
if __name__ == "__main__":
base.run_test_cases()