From 693e7df70582583d417165b408b3320657c1499c Mon Sep 17 00:00:00 2001 From: Bouke Haarsma Date: Fri, 7 Dec 2012 23:42:50 +0100 Subject: [PATCH] Fix for test, all tests succeed --- tests/mvvm/viewmodel/test_wrapper.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/mvvm/viewmodel/test_wrapper.py b/tests/mvvm/viewmodel/test_wrapper.py index 2646a91..bc6eb63 100644 --- a/tests/mvvm/viewmodel/test_wrapper.py +++ b/tests/mvvm/viewmodel/test_wrapper.py @@ -2,11 +2,11 @@ import unittest import mock +from sqlalchemy import event from traits.has_traits import HasTraits from traits.traits import CTrait from common.models import Skater, Country -from app import Base from mvvm.viewmodel import wrapper from mvvm.viewmodel.wrapper import wrap, wrap_cls from tests import engine, Session @@ -55,6 +55,7 @@ def test_sqlalchemy(self): connection = engine.connect() trans = connection.begin() session = Session(bind=connection) + event.listen(session, 'after_flush', wrapper.session_flush) skater = Skater( first_name='Bouke',