1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243
|
#------------------------------------------------------------------------------
# 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.
#------------------------------------------------------------------------------
"""
2100 - Module for testing NCHAR variables
"""
import cx_Oracle as oracledb
import test_env
class TestCase(test_env.BaseTestCase):
def setUp(self):
super().setUp()
self.raw_data = []
self.data_by_key = {}
for i in range(1, 11):
unicode_col = "Unicode \u3042 %d" % i
fixed_char_col = ("Fixed Unicode %d" % i).ljust(40)
if i % 2:
nullable_col = "Nullable %d" % i
else:
nullable_col = None
data_tuple = (i, unicode_col, fixed_char_col, nullable_col)
self.raw_data.append(data_tuple)
self.data_by_key[i] = data_tuple
def test_2100_unicode_length(self):
"2100 - test value length"
return_value = self.cursor.var(int)
self.cursor.execute("""
begin
:retval := LENGTH(:value);
end;""",
value="InVal \u3042",
retval=return_value)
self.assertEqual(return_value.getvalue(), 7)
def test_2101_bind_unicode(self):
"2101 - test binding in a unicode"
self.cursor.setinputsizes(value=oracledb.DB_TYPE_NVARCHAR)
self.cursor.execute("""
select * from TestUnicodes
where UnicodeCol = :value""",
value="Unicode \u3042 5")
self.assertEqual(self.cursor.fetchall(), [self.data_by_key[5]])
def test_2102_bind_different_var(self):
"2102 - test binding a different variable on second execution"
retval_1 = self.cursor.var(oracledb.DB_TYPE_NVARCHAR, 30)
retval_2 = self.cursor.var(oracledb.DB_TYPE_NVARCHAR, 30)
self.cursor.execute(r"begin :retval := unistr('Called \3042'); end;",
retval=retval_1)
self.assertEqual(retval_1.getvalue(), "Called \u3042")
self.cursor.execute("begin :retval := 'Called'; end;", retval=retval_2)
self.assertEqual(retval_2.getvalue(), "Called")
def test_2103_bind_unicode_after_number(self):
"2103 - test binding in a string after setting input sizes to a number"
unicode_val = self.cursor.var(oracledb.DB_TYPE_NVARCHAR)
unicode_val.setvalue(0, "Unicode \u3042 6")
self.cursor.setinputsizes(value=oracledb.NUMBER)
self.cursor.execute("""
select * from TestUnicodes
where UnicodeCol = :value""",
value=unicode_val)
self.assertEqual(self.cursor.fetchall(), [self.data_by_key[6]])
def test_2104_bind_unicode_array_direct(self):
"2104 - test binding in a unicode array"
return_value = self.cursor.var(oracledb.NUMBER)
array = [r[1] for r in self.raw_data]
array_var = self.cursor.arrayvar(oracledb.DB_TYPE_NVARCHAR, array)
statement = """
begin
:retval := pkg_TestUnicodeArrays.TestInArrays(
:integer_value, :array);
end;"""
self.cursor.execute(statement, retval=return_value, integer_value=5,
array=array_var)
self.assertEqual(return_value.getvalue(), 116)
array = ["Unicode - \u3042 %d" % i for i in range(15)]
array_var = self.cursor.arrayvar(oracledb.DB_TYPE_NVARCHAR, array)
self.cursor.execute(statement, integer_value=8, array=array_var)
self.assertEqual(return_value.getvalue(), 208)
def test_2105_bind_unicode_array_by_sizes(self):
"2105 - test binding in a unicode array (with setinputsizes)"
return_value = self.cursor.var(oracledb.NUMBER)
self.cursor.setinputsizes(array = [oracledb.DB_TYPE_NVARCHAR, 10])
array = [r[1] for r in self.raw_data]
self.cursor.execute("""
begin
:retval := pkg_TestUnicodeArrays.TestInArrays(:integer_value,
:array);
end;""",
retval=return_value,
integer_value=6,
array=array)
self.assertEqual(return_value.getvalue(), 117)
def test_2106_bind_unicode_array_by_var(self):
"2106 - test binding in a unicode array (with arrayvar)"
return_value = self.cursor.var(oracledb.NUMBER)
array = self.cursor.arrayvar(oracledb.DB_TYPE_NVARCHAR, 10, 20)
array.setvalue(0, [r[1] for r in self.raw_data])
self.cursor.execute("""
begin
:retval := pkg_TestUnicodeArrays.TestInArrays(:integer_value,
:array);
end;""",
retval=return_value,
integer_value=7,
array=array)
self.assertEqual(return_value.getvalue(), 118)
def test_2107_bind_in_out_unicode_array_by_var(self):
"2107 - test binding in/out a unicode array (with arrayvar)"
array = self.cursor.arrayvar(oracledb.DB_TYPE_NVARCHAR, 10, 100)
original_data = [r[1] for r in self.raw_data]
fmt = "Converted element \u3042 # %d originally had length %d"
expected_data = [fmt % (i, len(original_data[i - 1])) \
for i in range(1, 6)] + original_data[5:]
array.setvalue(0, original_data)
self.cursor.execute("""
begin
pkg_TestUnicodeArrays.TestInOutArrays(:numElems, :array);
end;""",
numElems = 5,
array = array)
self.assertEqual(array.getvalue(), expected_data)
def test_2108_bind_out_unicode_array_by_var(self):
"2108 - test binding out a unicode array (with arrayvar)"
array = self.cursor.arrayvar(oracledb.DB_TYPE_NVARCHAR, 6, 100)
fmt = "Test out element \u3042 # %d"
expected_data = [fmt % i for i in range(1, 7)]
self.cursor.execute("""
begin
pkg_TestUnicodeArrays.TestOutArrays(:numElems, :array);
end;""",
numElems = 6,
array = array)
self.assertEqual(array.getvalue(), expected_data)
def test_2109_bind_null(self):
"2109 - test binding in a null"
self.cursor.execute("""
select * from TestUnicodes
where UnicodeCol = :value""",
value = None)
self.assertEqual(self.cursor.fetchall(), [])
def test_2110_bind_out_set_input_sizes_by_type(self):
"2110 - test binding out with set input sizes defined (by type)"
bind_vars = self.cursor.setinputsizes(value=oracledb.DB_TYPE_NVARCHAR)
self.cursor.execute(r"""
begin
:value := unistr('TSI \3042');
end;""")
self.assertEqual(bind_vars["value"].getvalue(), "TSI \u3042")
def test_2111_bind_in_out_set_input_sizes_by_type(self):
"2111 - test binding in/out with set input sizes defined (by type)"
bind_vars = self.cursor.setinputsizes(value=oracledb.DB_TYPE_NVARCHAR)
self.cursor.execute(r"""
begin
:value := :value || unistr(' TSI \3042');
end;""",
value = "InVal \u3041")
self.assertEqual(bind_vars["value"].getvalue(),
"InVal \u3041 TSI \u3042")
def test_2112_bind_out_var(self):
"2112 - test binding out with cursor.var() method"
var = self.cursor.var(oracledb.DB_TYPE_NVARCHAR)
self.cursor.execute(r"""
begin
:value := unistr('TSI (VAR) \3042');
end;""",
value=var)
self.assertEqual(var.getvalue(), "TSI (VAR) \u3042")
def test_2113_bind_in_out_var_direct_set(self):
"2113 - test binding in/out with cursor.var() method"
var = self.cursor.var(oracledb.DB_TYPE_NVARCHAR)
var.setvalue(0, "InVal \u3041")
self.cursor.execute(r"""
begin
:value := :value || unistr(' TSI (VAR) \3042');
end;""",
value = var)
self.assertEqual(var.getvalue(), "InVal \u3041 TSI (VAR) \u3042")
def test_2114_cursor_description(self):
"2114 - test cursor description is accurate"
self.cursor.execute("select * from TestUnicodes")
varchar_ratio, nvarchar_ratio = test_env.get_charset_ratios()
expected_value = [
('INTCOL', oracledb.DB_TYPE_NUMBER, 10, None, 9, 0, False),
('UNICODECOL', oracledb.DB_TYPE_NVARCHAR, 20, 20 * nvarchar_ratio,
None, None, False),
('FIXEDUNICODECOL', oracledb.DB_TYPE_NCHAR, 40,
40 * nvarchar_ratio, None, None, False),
('NULLABLECOL', oracledb.DB_TYPE_NVARCHAR, 50, 50 * nvarchar_ratio,
None, None, True)
]
self.assertEqual(self.cursor.description, expected_value)
def test_2115_fetchall(self):
"2115 - test that fetching all of the data returns the correct results"
self.cursor.execute("select * From TestUnicodes order by IntCol")
self.assertEqual(self.cursor.fetchall(), self.raw_data)
self.assertEqual(self.cursor.fetchall(), [])
def test_2116_fetchmany(self):
"2116 - test that fetching data in chunks returns the correct results"
self.cursor.execute("select * From TestUnicodes order by IntCol")
self.assertEqual(self.cursor.fetchmany(3), self.raw_data[0:3])
self.assertEqual(self.cursor.fetchmany(2), self.raw_data[3:5])
self.assertEqual(self.cursor.fetchmany(4), self.raw_data[5:9])
self.assertEqual(self.cursor.fetchmany(3), self.raw_data[9:])
self.assertEqual(self.cursor.fetchmany(3), [])
def test_2117_fetchone(self):
"2117 - test that fetching a single row returns the correct results"
self.cursor.execute("""
select *
from TestUnicodes
where IntCol in (3, 4)
order by IntCol""")
self.assertEqual(self.cursor.fetchone(), self.data_by_key[3])
self.assertEqual(self.cursor.fetchone(), self.data_by_key[4])
self.assertEqual(self.cursor.fetchone(), None)
if __name__ == "__main__":
test_env.run_test_cases()
|