diff options
author | Mike Bayer <mike_mp@zzzcomputing.com> | 2019-01-06 01:14:26 -0500 |
---|---|---|
committer | mike bayer <mike_mp@zzzcomputing.com> | 2019-01-06 17:34:50 +0000 |
commit | 1e1a38e7801f410f244e4bbb44ec795ae152e04e (patch) | |
tree | 28e725c5c8188bd0cfd133d1e268dbca9b524978 /test/dialect/oracle/test_dialect.py | |
parent | 404e69426b05a82d905cbb3ad33adafccddb00dd (diff) | |
download | sqlalchemy-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.py | 285 |
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, + ) |