diff options
Diffstat (limited to 'test/orm/test_hasparent.py')
| -rw-r--r-- | test/orm/test_hasparent.py | 67 |
1 files changed, 44 insertions, 23 deletions
diff --git a/test/orm/test_hasparent.py b/test/orm/test_hasparent.py index 38dd722c2..fe5e05a17 100644 --- a/test/orm/test_hasparent.py +++ b/test/orm/test_hasparent.py @@ -2,11 +2,17 @@ from sqlalchemy.testing import assert_raises, assert_raises_message -from sqlalchemy import Integer, String, ForeignKey, Sequence, \ - exc as sa_exc +from sqlalchemy import Integer, String, ForeignKey, Sequence, exc as sa_exc from sqlalchemy.testing.schema import Table, Column -from sqlalchemy.orm import mapper, relationship, create_session, \ - sessionmaker, class_mapper, backref, Session +from sqlalchemy.orm import ( + mapper, + relationship, + create_session, + sessionmaker, + class_mapper, + backref, + Session, +) from sqlalchemy.orm import attributes, exc as orm_exc from sqlalchemy import testing from sqlalchemy.testing import eq_ @@ -23,24 +29,33 @@ class ParentRemovalTest(fixtures.MappedTest): raised. """ + run_inserts = None @classmethod def define_tables(cls, metadata): - if testing.against('oracle'): - fk_args = dict(deferrable=True, initially='deferred') - elif testing.against('mysql'): + if testing.against("oracle"): + fk_args = dict(deferrable=True, initially="deferred") + elif testing.against("mysql"): fk_args = {} else: - fk_args = dict(onupdate='cascade') - - Table('users', metadata, - Column('id', Integer, primary_key=True, - test_needs_autoincrement=True)) - Table('addresses', metadata, - Column('id', Integer, primary_key=True, - test_needs_autoincrement=True), - Column('user_id', Integer, ForeignKey('users.id', **fk_args))) + fk_args = dict(onupdate="cascade") + + Table( + "users", + metadata, + Column( + "id", Integer, primary_key=True, test_needs_autoincrement=True + ), + ) + Table( + "addresses", + metadata, + Column( + "id", Integer, primary_key=True, test_needs_autoincrement=True + ), + Column("user_id", Integer, ForeignKey("users.id", **fk_args)), + ) @classmethod def setup_classes(cls): @@ -53,11 +68,15 @@ class ParentRemovalTest(fixtures.MappedTest): @classmethod def setup_mappers(cls): mapper(cls.classes.Address, cls.tables.addresses) - mapper(cls.classes.User, cls.tables.users, properties={ - 'addresses': relationship(cls.classes.Address, - cascade='all, delete-orphan'), - - }) + mapper( + cls.classes.User, + cls.tables.users, + properties={ + "addresses": relationship( + cls.classes.Address, cascade="all, delete-orphan" + ) + }, + ) def _assert_hasparent(self, a1): assert attributes.has_parent(self.classes.User, a1, "addresses") @@ -132,7 +151,8 @@ class ParentRemovalTest(fixtures.MappedTest): assert_raises_message( orm_exc.StaleDataError, "can't be sure this is the most recent parent.", - u1.addresses.remove, a1 + u1.addresses.remove, + a1, ) # u1.addresses wasn't actually impacted, because the event was @@ -185,7 +205,8 @@ class ParentRemovalTest(fixtures.MappedTest): assert_raises_message( orm_exc.StaleDataError, "can't be sure this is the most recent parent.", - u1.addresses.remove, a1 + u1.addresses.remove, + a1, ) s.flush() |