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/orm/test_lockmode.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/orm/test_lockmode.py')
| -rw-r--r-- | test/orm/test_lockmode.py | 116 |
1 files changed, 61 insertions, 55 deletions
diff --git a/test/orm/test_lockmode.py b/test/orm/test_lockmode.py index 34ae52d3c..1a05ffbe4 100644 --- a/test/orm/test_lockmode.py +++ b/test/orm/test_lockmode.py @@ -54,7 +54,8 @@ class LegacyLockModeTest(_fixtures.FixtureTest): assert_raises_message( exc.ArgumentError, "Unknown with_lockmode argument: 'unknown_mode'", - sess.query(User.id).with_lockmode, 'unknown_mode' + sess.query(User.id).with_lockmode, + "unknown_mode", ) @@ -64,12 +65,20 @@ class ForUpdateTest(_fixtures.FixtureTest): User, users = cls.classes.User, cls.tables.users mapper(User, users) - def _assert(self, read=False, nowait=False, of=None, key_share=None, - assert_q_of=None, assert_sel_of=None): + def _assert( + self, + read=False, + nowait=False, + of=None, + key_share=None, + assert_q_of=None, + assert_sel_of=None, + ): User = self.classes.User s = Session() q = s.query(User).with_for_update( - read=read, nowait=nowait, of=of, key_share=key_share) + read=read, nowait=nowait, of=of, key_share=key_share + ) sel = q._compile_context().statement assert q._for_update_arg.read is read @@ -99,9 +108,7 @@ class ForUpdateTest(_fixtures.FixtureTest): def test_of_single_col(self): User, users = self.classes.User, self.tables.users self._assert( - of=User.id, - assert_q_of=[users.c.id], - assert_sel_of=[users.c.id] + of=User.id, assert_q_of=[users.c.id], assert_sel_of=[users.c.id] ) @@ -117,24 +124,21 @@ class BackendTest(_fixtures.FixtureTest): def setup_mappers(cls): User, users = cls.classes.User, cls.tables.users Address, addresses = cls.classes.Address, cls.tables.addresses - mapper(User, users, properties={ - "addresses": relationship(Address) - }) + mapper(User, users, properties={"addresses": relationship(Address)}) mapper(Address, addresses) def test_inner_joinedload_w_limit(self): User = self.classes.User sess = Session() - q = sess.query(User).options( - joinedload(User.addresses, innerjoin=True) - ).with_for_update().limit(1) + q = ( + sess.query(User) + .options(joinedload(User.addresses, innerjoin=True)) + .with_for_update() + .limit(1) + ) if testing.against("oracle"): - assert_raises_message( - exc.DatabaseError, - "ORA-02014", - q.all - ) + assert_raises_message(exc.DatabaseError, "ORA-02014", q.all) else: q.all() sess.close() @@ -162,11 +166,7 @@ class BackendTest(_fixtures.FixtureTest): q = q.limit(1) if testing.against("oracle"): - assert_raises_message( - exc.DatabaseError, - "ORA-02014", - q.all - ) + assert_raises_message(exc.DatabaseError, "ORA-02014", q.all) else: q.all() sess.close() @@ -203,15 +203,14 @@ class BackendTest(_fixtures.FixtureTest): class CompileTest(_fixtures.FixtureTest, AssertsCompiledSQL): """run some compile tests, even though these are redundant.""" + run_inserts = None @classmethod def setup_mappers(cls): User, users = cls.classes.User, cls.tables.users Address, addresses = cls.classes.Address, cls.tables.addresses - mapper(User, users, properties={ - "addresses": relationship(Address) - }) + mapper(User, users, properties={"addresses": relationship(Address)}) mapper(Address, addresses) def test_default_update(self): @@ -220,7 +219,7 @@ class CompileTest(_fixtures.FixtureTest, AssertsCompiledSQL): self.assert_compile( sess.query(User.id).with_for_update(), "SELECT users.id AS users_id FROM users FOR UPDATE", - dialect=default.DefaultDialect() + dialect=default.DefaultDialect(), ) def test_not_supported_by_dialect_should_just_use_update(self): @@ -229,24 +228,25 @@ class CompileTest(_fixtures.FixtureTest, AssertsCompiledSQL): self.assert_compile( sess.query(User.id).with_for_update(read=True), "SELECT users.id AS users_id FROM users FOR UPDATE", - dialect=default.DefaultDialect() + dialect=default.DefaultDialect(), ) def test_postgres_read(self): User = self.classes.User sess = Session() - self.assert_compile(sess.query(User.id).with_for_update(read=True), - "SELECT users.id AS users_id FROM users FOR SHARE", - dialect="postgresql") + self.assert_compile( + sess.query(User.id).with_for_update(read=True), + "SELECT users.id AS users_id FROM users FOR SHARE", + dialect="postgresql", + ) def test_postgres_read_nowait(self): User = self.classes.User sess = Session() self.assert_compile( - sess.query(User.id). - with_for_update(read=True, nowait=True), + sess.query(User.id).with_for_update(read=True, nowait=True), "SELECT users.id AS users_id FROM users FOR SHARE NOWAIT", - dialect="postgresql" + dialect="postgresql", ) def test_postgres_update(self): @@ -255,7 +255,7 @@ class CompileTest(_fixtures.FixtureTest, AssertsCompiledSQL): self.assert_compile( sess.query(User.id).with_for_update(), "SELECT users.id AS users_id FROM users FOR UPDATE", - dialect="postgresql" + dialect="postgresql", ) def test_postgres_update_of(self): @@ -264,7 +264,7 @@ class CompileTest(_fixtures.FixtureTest, AssertsCompiledSQL): self.assert_compile( sess.query(User.id).with_for_update(of=User.id), "SELECT users.id AS users_id FROM users FOR UPDATE OF users", - dialect="postgresql" + dialect="postgresql", ) def test_postgres_update_of_entity(self): @@ -273,7 +273,7 @@ class CompileTest(_fixtures.FixtureTest, AssertsCompiledSQL): self.assert_compile( sess.query(User.id).with_for_update(of=User), "SELECT users.id AS users_id FROM users FOR UPDATE OF users", - dialect="postgresql" + dialect="postgresql", ) def test_postgres_update_of_entity_list(self): @@ -282,11 +282,12 @@ class CompileTest(_fixtures.FixtureTest, AssertsCompiledSQL): sess = Session() self.assert_compile( - sess.query(User.id, Address.id). - with_for_update(of=[User, Address]), + sess.query(User.id, Address.id).with_for_update( + of=[User, Address] + ), "SELECT users.id AS users_id, addresses.id AS addresses_id " "FROM users, addresses FOR UPDATE OF users, addresses", - dialect="postgresql" + dialect="postgresql", ) def test_postgres_for_no_key_update(self): @@ -295,7 +296,7 @@ class CompileTest(_fixtures.FixtureTest, AssertsCompiledSQL): self.assert_compile( sess.query(User.id).with_for_update(key_share=True), "SELECT users.id AS users_id FROM users FOR NO KEY UPDATE", - dialect="postgresql" + dialect="postgresql", ) def test_postgres_for_no_key_nowait_update(self): @@ -304,17 +305,18 @@ class CompileTest(_fixtures.FixtureTest, AssertsCompiledSQL): self.assert_compile( sess.query(User.id).with_for_update(key_share=True, nowait=True), "SELECT users.id AS users_id FROM users FOR NO KEY UPDATE NOWAIT", - dialect="postgresql" + dialect="postgresql", ) def test_postgres_update_of_list(self): User = self.classes.User sess = Session() self.assert_compile( - sess.query(User.id) - .with_for_update(of=[User.id, User.id, User.id]), + sess.query(User.id).with_for_update( + of=[User.id, User.id, User.id] + ), "SELECT users.id AS users_id FROM users FOR UPDATE OF users", - dialect="postgresql" + dialect="postgresql", ) def test_postgres_update_skip_locked(self): @@ -323,7 +325,7 @@ class CompileTest(_fixtures.FixtureTest, AssertsCompiledSQL): self.assert_compile( sess.query(User.id).with_for_update(skip_locked=True), "SELECT users.id AS users_id FROM users FOR UPDATE SKIP LOCKED", - dialect="postgresql" + dialect="postgresql", ) def test_oracle_update(self): @@ -332,7 +334,7 @@ class CompileTest(_fixtures.FixtureTest, AssertsCompiledSQL): self.assert_compile( sess.query(User.id).with_for_update(), "SELECT users.id AS users_id FROM users FOR UPDATE", - dialect="oracle" + dialect="oracle", ) def test_oracle_update_skip_locked(self): @@ -341,7 +343,7 @@ class CompileTest(_fixtures.FixtureTest, AssertsCompiledSQL): self.assert_compile( sess.query(User.id).with_for_update(skip_locked=True), "SELECT users.id AS users_id FROM users FOR UPDATE SKIP LOCKED", - dialect="oracle" + dialect="oracle", ) def test_mysql_read(self): @@ -350,15 +352,17 @@ class CompileTest(_fixtures.FixtureTest, AssertsCompiledSQL): self.assert_compile( sess.query(User.id).with_for_update(read=True), "SELECT users.id AS users_id FROM users LOCK IN SHARE MODE", - dialect="mysql" + dialect="mysql", ) def test_for_update_on_inner_w_joinedload(self): User = self.classes.User sess = Session() self.assert_compile( - sess.query(User).options( - joinedload(User.addresses)).with_for_update().limit(1), + sess.query(User) + .options(joinedload(User.addresses)) + .with_for_update() + .limit(1), "SELECT anon_1.users_id AS anon_1_users_id, anon_1.users_name " "AS anon_1_users_name, addresses_1.id AS addresses_1_id, " "addresses_1.user_id AS addresses_1_user_id, " @@ -367,15 +371,17 @@ class CompileTest(_fixtures.FixtureTest, AssertsCompiledSQL): "FROM users LIMIT %s FOR UPDATE) AS anon_1 " "LEFT OUTER JOIN addresses AS addresses_1 " "ON anon_1.users_id = addresses_1.user_id FOR UPDATE", - dialect="mysql" + dialect="mysql", ) def test_for_update_on_inner_w_joinedload_no_render_oracle(self): User = self.classes.User sess = Session() self.assert_compile( - sess.query(User).options( - joinedload(User.addresses)).with_for_update().limit(1), + sess.query(User) + .options(joinedload(User.addresses)) + .with_for_update() + .limit(1), "SELECT anon_1.users_id AS anon_1_users_id, " "anon_1.users_name AS anon_1_users_name, " "addresses_1.id AS addresses_1_id, " @@ -386,5 +392,5 @@ class CompileTest(_fixtures.FixtureTest, AssertsCompiledSQL): "FROM users) WHERE ROWNUM <= :param_1) anon_1 " "LEFT OUTER JOIN addresses addresses_1 " "ON anon_1.users_id = addresses_1.user_id FOR UPDATE", - dialect="oracle" + dialect="oracle", ) |
