summaryrefslogtreecommitdiff
path: root/test/dialect/oracle/test_dialect.py
diff options
context:
space:
mode:
authorMike Bayer <mike_mp@zzzcomputing.com>2019-01-06 01:14:26 -0500
committermike bayer <mike_mp@zzzcomputing.com>2019-01-06 17:34:50 +0000
commit1e1a38e7801f410f244e4bbb44ec795ae152e04e (patch)
tree28e725c5c8188bd0cfd133d1e268dbca9b524978 /test/dialect/oracle/test_dialect.py
parent404e69426b05a82d905cbb3ad33adafccddb00dd (diff)
downloadsqlalchemy-1e1a38e7801f410f244e4bbb44ec795ae152e04e.tar.gz
Run black -l 79 against all source files
This is a straight reformat run using black as is, with no edits applied at all. The black run will format code consistently, however in some cases that are prevalent in SQLAlchemy code it produces too-long lines. The too-long lines will be resolved in the following commit that will resolve all remaining flake8 issues including shadowed builtins, long lines, import order, unused imports, duplicate imports, and docstring issues. Change-Id: I7eda77fed3d8e73df84b3651fd6cfcfe858d4dc9
Diffstat (limited to 'test/dialect/oracle/test_dialect.py')
-rw-r--r--test/dialect/oracle/test_dialect.py285
1 files changed, 150 insertions, 135 deletions
diff --git a/test/dialect/oracle/test_dialect.py b/test/dialect/oracle/test_dialect.py
index 9f38a515a..cfb90c25f 100644
--- a/test/dialect/oracle/test_dialect.py
+++ b/test/dialect/oracle/test_dialect.py
@@ -3,14 +3,25 @@
from sqlalchemy.testing import eq_
from sqlalchemy import exc
-from sqlalchemy.testing import (fixtures,
- AssertsExecutionResults,
- AssertsCompiledSQL)
+from sqlalchemy.testing import (
+ fixtures,
+ AssertsExecutionResults,
+ AssertsCompiledSQL,
+)
from sqlalchemy import testing
from sqlalchemy import create_engine
from sqlalchemy import bindparam, outparam
-from sqlalchemy import text, Float, Integer, String, select, literal_column,\
- Unicode, UnicodeText, Sequence
+from sqlalchemy import (
+ text,
+ Float,
+ Integer,
+ String,
+ select,
+ literal_column,
+ Unicode,
+ UnicodeText,
+ Sequence,
+)
from sqlalchemy.util import u
from sqlalchemy.testing import assert_raises, assert_raises_message
from sqlalchemy.dialects.oracle import cx_oracle, base as oracle
@@ -24,45 +35,41 @@ class DialectTest(fixtures.TestBase):
def test_cx_oracle_version_parse(self):
dialect = cx_oracle.OracleDialect_cx_oracle()
- eq_(
- dialect._parse_cx_oracle_ver("5.2"),
- (5, 2)
- )
+ eq_(dialect._parse_cx_oracle_ver("5.2"), (5, 2))
- eq_(
- dialect._parse_cx_oracle_ver("5.0.1"),
- (5, 0, 1)
- )
+ eq_(dialect._parse_cx_oracle_ver("5.0.1"), (5, 0, 1))
- eq_(
- dialect._parse_cx_oracle_ver("6.0b1"),
- (6, 0)
- )
+ eq_(dialect._parse_cx_oracle_ver("6.0b1"), (6, 0))
def test_minimum_version(self):
with mock.patch(
- "sqlalchemy.dialects.oracle.cx_oracle.OracleDialect_cx_oracle."
- "_parse_cx_oracle_ver", lambda self, vers: (5, 1, 5)):
+ "sqlalchemy.dialects.oracle.cx_oracle.OracleDialect_cx_oracle."
+ "_parse_cx_oracle_ver",
+ lambda self, vers: (5, 1, 5),
+ ):
assert_raises_message(
exc.InvalidRequestError,
"cx_Oracle version 5.2 and above are supported",
cx_oracle.OracleDialect_cx_oracle,
- dbapi=Mock()
+ dbapi=Mock(),
)
with mock.patch(
- "sqlalchemy.dialects.oracle.cx_oracle.OracleDialect_cx_oracle."
- "_parse_cx_oracle_ver", lambda self, vers: (5, 3, 1)):
+ "sqlalchemy.dialects.oracle.cx_oracle.OracleDialect_cx_oracle."
+ "_parse_cx_oracle_ver",
+ lambda self, vers: (5, 3, 1),
+ ):
cx_oracle.OracleDialect_cx_oracle(dbapi=Mock())
class OutParamTest(fixtures.TestBase, AssertsExecutionResults):
- __only_on__ = 'oracle+cx_oracle'
+ __only_on__ = "oracle+cx_oracle"
__backend__ = True
@classmethod
def setup_class(cls):
- testing.db.execute("""
+ testing.db.execute(
+ """
create or replace procedure foo(x_in IN number, x_out OUT number,
y_out OUT number, z_out OUT varchar) IS
retval number;
@@ -72,19 +79,24 @@ class OutParamTest(fixtures.TestBase, AssertsExecutionResults):
y_out := x_in * 15;
z_out := NULL;
end;
- """)
+ """
+ )
def test_out_params(self):
- result = testing.db.execute(text('begin foo(:x_in, :x_out, :y_out, '
- ':z_out); end;',
- bindparams=[bindparam('x_in', Float),
- outparam('x_out', Integer),
- outparam('y_out', Float),
- outparam('z_out', String)]),
- x_in=5)
- eq_(result.out_parameters,
- {'x_out': 10, 'y_out': 75, 'z_out': None})
- assert isinstance(result.out_parameters['x_out'], int)
+ result = testing.db.execute(
+ text(
+ "begin foo(:x_in, :x_out, :y_out, " ":z_out); end;",
+ bindparams=[
+ bindparam("x_in", Float),
+ outparam("x_out", Integer),
+ outparam("y_out", Float),
+ outparam("z_out", String),
+ ],
+ ),
+ x_in=5,
+ )
+ eq_(result.out_parameters, {"x_out": 10, "y_out": 75, "z_out": None})
+ assert isinstance(result.out_parameters["x_out"], int)
@classmethod
def teardown_class(cls):
@@ -93,69 +105,61 @@ class OutParamTest(fixtures.TestBase, AssertsExecutionResults):
class QuotedBindRoundTripTest(fixtures.TestBase):
- __only_on__ = 'oracle'
+ __only_on__ = "oracle"
__backend__ = True
@testing.provide_metadata
def test_table_round_trip(self):
- oracle.RESERVED_WORDS.remove('UNION')
+ oracle.RESERVED_WORDS.remove("UNION")
metadata = self.metadata
- table = Table("t1", metadata,
- Column("option", Integer),
- Column("plain", Integer, quote=True),
- # test that quote works for a reserved word
- # that the dialect isn't aware of when quote
- # is set
- Column("union", Integer, quote=True))
+ table = Table(
+ "t1",
+ metadata,
+ Column("option", Integer),
+ Column("plain", Integer, quote=True),
+ # test that quote works for a reserved word
+ # that the dialect isn't aware of when quote
+ # is set
+ Column("union", Integer, quote=True),
+ )
metadata.create_all()
- table.insert().execute(
- {"option": 1, "plain": 1, "union": 1}
- )
- eq_(
- testing.db.execute(table.select()).first(),
- (1, 1, 1)
- )
+ table.insert().execute({"option": 1, "plain": 1, "union": 1})
+ eq_(testing.db.execute(table.select()).first(), (1, 1, 1))
table.update().values(option=2, plain=2, union=2).execute()
- eq_(
- testing.db.execute(table.select()).first(),
- (2, 2, 2)
- )
+ eq_(testing.db.execute(table.select()).first(), (2, 2, 2))
def test_numeric_bind_round_trip(self):
eq_(
testing.db.scalar(
- select([
- literal_column("2", type_=Integer()) +
- bindparam("2_1", value=2)])
+ select(
+ [
+ literal_column("2", type_=Integer())
+ + bindparam("2_1", value=2)
+ ]
+ )
),
- 4
+ 4,
)
@testing.provide_metadata
def test_numeric_bind_in_crud(self):
- t = Table(
- "asfd", self.metadata,
- Column("100K", Integer)
- )
+ t = Table("asfd", self.metadata, Column("100K", Integer))
t.create()
testing.db.execute(t.insert(), {"100K": 10})
- eq_(
- testing.db.scalar(t.select()), 10
- )
+ eq_(testing.db.scalar(t.select()), 10)
class CompatFlagsTest(fixtures.TestBase, AssertsCompiledSQL):
-
def _dialect(self, server_version, **kw):
def server_version_info(conn):
return server_version
dialect = oracle.dialect(
- dbapi=Mock(version="0.0.0", paramstyle="named"),
- **kw)
+ dbapi=Mock(version="0.0.0", paramstyle="named"), **kw
+ )
dialect._get_server_version_info = server_version_info
dialect._check_unicode_returns = Mock()
dialect._check_unicode_description = Mock()
@@ -219,15 +223,19 @@ class CompatFlagsTest(fixtures.TestBase, AssertsCompiledSQL):
class ExecuteTest(fixtures.TestBase):
- __only_on__ = 'oracle'
+ __only_on__ = "oracle"
__backend__ = True
def test_basic(self):
- eq_(testing.db.execute('/*+ this is a comment */ SELECT 1 FROM '
- 'DUAL').fetchall(), [(1, )])
+ eq_(
+ testing.db.execute(
+ "/*+ this is a comment */ SELECT 1 FROM " "DUAL"
+ ).fetchall(),
+ [(1,)],
+ )
def test_sequences_are_integers(self):
- seq = Sequence('foo_seq')
+ seq = Sequence("foo_seq")
seq.create(testing.db)
try:
val = testing.db.execute(seq)
@@ -242,25 +250,26 @@ class ExecuteTest(fixtures.TestBase):
# oracle can't actually do the ROWNUM thing with FOR UPDATE
# very well.
- t = Table('t1',
- metadata,
- Column('id', Integer, primary_key=True),
- Column('data', Integer))
+ t = Table(
+ "t1",
+ metadata,
+ Column("id", Integer, primary_key=True),
+ Column("data", Integer),
+ )
metadata.create_all()
t.insert().execute(
- {'id': 1, 'data': 1},
- {'id': 2, 'data': 7},
- {'id': 3, 'data': 12},
- {'id': 4, 'data': 15},
- {'id': 5, 'data': 32},
+ {"id": 1, "data": 1},
+ {"id": 2, "data": 7},
+ {"id": 3, "data": 12},
+ {"id": 4, "data": 15},
+ {"id": 5, "data": 32},
)
# here, we can't use ORDER BY.
eq_(
t.select(for_update=True).limit(2).execute().fetchall(),
- [(1, 1),
- (2, 7)]
+ [(1, 1), (2, 7)],
)
# here, its impossible. But we'd prefer it to raise ORA-02014
@@ -268,69 +277,71 @@ class ExecuteTest(fixtures.TestBase):
assert_raises_message(
exc.DatabaseError,
"ORA-02014",
- t.select(for_update=True).limit(2).offset(3).execute
+ t.select(for_update=True).limit(2).offset(3).execute,
)
class UnicodeSchemaTest(fixtures.TestBase):
- __only_on__ = 'oracle'
+ __only_on__ = "oracle"
__backend__ = True
@testing.provide_metadata
def test_quoted_column_non_unicode(self):
metadata = self.metadata
- table = Table("atable", metadata,
- Column("_underscorecolumn",
- Unicode(255),
- primary_key=True))
+ table = Table(
+ "atable",
+ metadata,
+ Column("_underscorecolumn", Unicode(255), primary_key=True),
+ )
metadata.create_all()
- table.insert().execute(
- {'_underscorecolumn': u('’é')},
- )
+ table.insert().execute({"_underscorecolumn": u("’é")})
result = testing.db.execute(
- table.select().where(table.c._underscorecolumn == u('’é'))
+ table.select().where(table.c._underscorecolumn == u("’é"))
).scalar()
- eq_(result, u('’é'))
+ eq_(result, u("’é"))
@testing.provide_metadata
def test_quoted_column_unicode(self):
metadata = self.metadata
- table = Table("atable", metadata,
- Column(u("méil"), Unicode(255), primary_key=True))
+ table = Table(
+ "atable",
+ metadata,
+ Column(u("méil"), Unicode(255), primary_key=True),
+ )
metadata.create_all()
- table.insert().execute(
- {u('méil'): u('’é')},
- )
+ table.insert().execute({u("méil"): u("’é")})
result = testing.db.execute(
- table.select().where(table.c[u('méil')] == u('’é'))
+ table.select().where(table.c[u("méil")] == u("’é"))
).scalar()
- eq_(result, u('’é'))
+ eq_(result, u("’é"))
class CXOracleConnectArgsTest(fixtures.TestBase):
- __only_on__ = 'oracle+cx_oracle'
+ __only_on__ = "oracle+cx_oracle"
__backend__ = True
def test_cx_oracle_service_name(self):
- url_string = 'oracle+cx_oracle://scott:tiger@host/?service_name=hr'
+ url_string = "oracle+cx_oracle://scott:tiger@host/?service_name=hr"
eng = create_engine(url_string, _initialize=False)
cargs, cparams = eng.dialect.create_connect_args(eng.url)
- assert 'SERVICE_NAME=hr' in cparams['dsn']
- assert 'SID=hr' not in cparams['dsn']
+ assert "SERVICE_NAME=hr" in cparams["dsn"]
+ assert "SID=hr" not in cparams["dsn"]
def test_cx_oracle_service_name_bad(self):
- url_string = 'oracle+cx_oracle://scott:tiger@host/hr1?service_name=hr2'
+ url_string = "oracle+cx_oracle://scott:tiger@host/hr1?service_name=hr2"
assert_raises(
exc.InvalidRequestError,
- create_engine, url_string,
- _initialize=False
+ create_engine,
+ url_string,
+ _initialize=False,
)
def _test_db_opt(self, url_string, key, value):
import cx_Oracle
+
url_obj = url.make_url(url_string)
dialect = cx_oracle.dialect(dbapi=cx_Oracle)
arg, kw = dialect.create_connect_args(url_obj)
@@ -338,6 +349,7 @@ class CXOracleConnectArgsTest(fixtures.TestBase):
def _test_db_opt_unpresent(self, url_string, key):
import cx_Oracle
+
url_obj = url.make_url(url_string)
dialect = cx_oracle.dialect(dbapi=cx_Oracle)
arg, kw = dialect.create_connect_args(url_obj)
@@ -345,10 +357,12 @@ class CXOracleConnectArgsTest(fixtures.TestBase):
def _test_dialect_param_from_url(self, url_string, key, value):
import cx_Oracle
+
url_obj = url.make_url(url_string)
dialect = cx_oracle.dialect(dbapi=cx_Oracle)
with testing.expect_deprecated(
- "cx_oracle dialect option %r should" % key):
+ "cx_oracle dialect option %r should" % key
+ ):
arg, kw = dialect.create_connect_args(url_obj)
eq_(getattr(dialect, key), value)
@@ -358,32 +372,32 @@ class CXOracleConnectArgsTest(fixtures.TestBase):
def test_mode(self):
import cx_Oracle
+
self._test_db_opt(
- 'oracle+cx_oracle://scott:tiger@host/?mode=sYsDBA',
+ "oracle+cx_oracle://scott:tiger@host/?mode=sYsDBA",
"mode",
- cx_Oracle.SYSDBA
+ cx_Oracle.SYSDBA,
)
self._test_db_opt(
- 'oracle+cx_oracle://scott:tiger@host/?mode=SYSOPER',
+ "oracle+cx_oracle://scott:tiger@host/?mode=SYSOPER",
"mode",
- cx_Oracle.SYSOPER
+ cx_Oracle.SYSOPER,
)
def test_int_mode(self):
self._test_db_opt(
- 'oracle+cx_oracle://scott:tiger@host/?mode=32767',
- "mode",
- 32767
+ "oracle+cx_oracle://scott:tiger@host/?mode=32767", "mode", 32767
)
@testing.requires.cxoracle6_or_greater
def test_purity(self):
import cx_Oracle
+
self._test_db_opt(
- 'oracle+cx_oracle://scott:tiger@host/?purity=attr_purity_new',
+ "oracle+cx_oracle://scott:tiger@host/?purity=attr_purity_new",
"purity",
- cx_Oracle.ATTR_PURITY_NEW
+ cx_Oracle.ATTR_PURITY_NEW,
)
def test_encoding(self):
@@ -391,44 +405,45 @@ class CXOracleConnectArgsTest(fixtures.TestBase):
"oracle+cx_oracle://scott:tiger@host/"
"?encoding=AMERICAN_AMERICA.UTF8",
"encoding",
- "AMERICAN_AMERICA.UTF8"
+ "AMERICAN_AMERICA.UTF8",
)
def test_threaded(self):
self._test_db_opt(
- 'oracle+cx_oracle://scott:tiger@host/?threaded=true',
+ "oracle+cx_oracle://scott:tiger@host/?threaded=true",
"threaded",
- True
+ True,
)
self._test_db_opt_unpresent(
- 'oracle+cx_oracle://scott:tiger@host/',
- "threaded"
+ "oracle+cx_oracle://scott:tiger@host/", "threaded"
)
def test_events(self):
self._test_db_opt(
- 'oracle+cx_oracle://scott:tiger@host/?events=true',
- "events",
- True
+ "oracle+cx_oracle://scott:tiger@host/?events=true", "events", True
)
def test_threaded_deprecated_at_dialect_level(self):
with testing.expect_deprecated(
- "The 'threaded' parameter to the cx_oracle dialect"):
+ "The 'threaded' parameter to the cx_oracle dialect"
+ ):
dialect = cx_oracle.dialect(threaded=False)
arg, kw = dialect.create_connect_args(
- url.make_url("oracle+cx_oracle://scott:tiger@dsn"))
- eq_(kw['threaded'], False)
+ url.make_url("oracle+cx_oracle://scott:tiger@dsn")
+ )
+ eq_(kw["threaded"], False)
def test_deprecated_use_ansi(self):
self._test_dialect_param_from_url(
- 'oracle+cx_oracle://scott:tiger@host/?use_ansi=False',
- 'use_ansi', False
+ "oracle+cx_oracle://scott:tiger@host/?use_ansi=False",
+ "use_ansi",
+ False,
)
def test_deprecated_auto_convert_lobs(self):
self._test_dialect_param_from_url(
- 'oracle+cx_oracle://scott:tiger@host/?auto_convert_lobs=False',
- 'auto_convert_lobs', False
- ) \ No newline at end of file
+ "oracle+cx_oracle://scott:tiger@host/?auto_convert_lobs=False",
+ "auto_convert_lobs",
+ False,
+ )