test expire session

Author: uralbash

sqlalchemy_mptt/events.py

@@ -406,10 +406,8 @@ def after_flush_postexec(self, session, context):
             parent = self.get_parent_value(instance)
             while parent != NO_VALUE and parent is not None:
                 self.instances.discard(parent)
-                session.expire(parent, ['left', 'right'])
+                session.expire(parent, ['left', 'right', 'tree_id'])
                 parent = self.get_parent_value(parent)
-            else:
-                session.expire(instance, ['tree_id', ])
 
     @staticmethod
     def get_parent_value(instance):

sqlalchemy_mptt/tests/tree_testing_base.py

@@ -9,6 +9,7 @@
 
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker
+
 from sqlalchemy_mptt import mptt_sessionmaker
 
 
@@ -1719,13 +1720,13 @@ def test_leftsibling_in_level(self):
     def test_session_expire_for_move_after_to_new_tree(self):
         """https://github.com/ITCase/sqlalchemy_mptt/issues/33"""
         node = self.session.query(self.model).filter(self.model.ppk == 4).one()
+        children = self.session.query(self.model)\
+            .filter(self.model.ppk.in_((5, 6))).all()
         node.move_after('1')
         self.session.flush()
+
         self.assertEqual(node.tree_id, 2)
         self.assertEqual(node.parent_id, None)
-
-        children = self.session.query(self.model)\
-            .filter(self.model.ppk.in_((5, 6))).all()
         self.assertEqual(children[0].tree_id, 2)
         self.assertEqual(children[1].tree_id, 2)
         self.assertEqual(children[0].parent_id, 4)