diff options
Diffstat (limited to 'test/orm/test_expire.py')
| -rw-r--r-- | test/orm/test_expire.py | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/test/orm/test_expire.py b/test/orm/test_expire.py index 083c2f465..127380fad 100644 --- a/test/orm/test_expire.py +++ b/test/orm/test_expire.py @@ -82,6 +82,24 @@ class ExpireTest(_fixtures.FixtureTest): self.assert_sql_count(testing.db, go, 0) + def test_expire_autoflush(self): + User, users = self.classes.User, self.tables.users + Address, addresses = self.classes.Address, self.tables.addresses + + mapper(User, users) + mapper(Address, addresses, properties={"user": relationship(User)}) + + s = Session() + + a1 = s.query(Address).get(2) + u1 = s.query(User).get(7) + a1.user = u1 + + s.expire(a1, ["user_id"]) + + # autoflushes + eq_(a1.user_id, 7) + def test_persistence_check(self): users, User = self.tables.users, self.classes.User @@ -1748,6 +1766,24 @@ class RefreshTest(_fixtures.FixtureTest): lambda: s.refresh(u), ) + def test_refresh_autoflush(self): + User, users = self.classes.User, self.tables.users + Address, addresses = self.classes.Address, self.tables.addresses + + mapper(User, users) + mapper(Address, addresses, properties={"user": relationship(User)}) + + s = Session() + + a1 = s.query(Address).get(2) + u1 = s.query(User).get(7) + a1.user = u1 + + s.refresh(a1, ["user_id"]) + + # autoflushes + eq_(a1.user_id, 7) + def test_refresh_expired(self): User, users = self.classes.User, self.tables.users |