From 00a20dd075fc20ab2830534b01fc932376c04d12 Mon Sep 17 00:00:00 2001 From: Mike Bayer Date: Sat, 15 Oct 2005 20:47:19 +0000 Subject: [PATCH] --- test/testbase.py | 25 +++++++++++++------------ 1 file changed, 13 insertions(+), 12 deletions(-) diff --git a/test/testbase.py b/test/testbase.py index 2c5bc2318b..7f059670db 100644 --- a/test/testbase.py +++ b/test/testbase.py @@ -49,23 +49,24 @@ class AssertMixin(PersistTest): class EngineAssert(object): def __init__(self, engine): - self.__dict__['engine'] = engine - self.__dict__['realexec'] = engine.execute + self.engine = engine + self.realexec = engine.execute + engine.execute = self.execute + self.echo = engine.echo self.set_assert_list(None, None) def __getattr__(self, key): return getattr(self.engine, key) - def __setattr__(self, key, value): - setattr(self.__dict__['engine'], key, value) def set_assert_list(self, unittest, list): - self.__dict__['unittest'] = unittest - self.__dict__['assert_list'] = list + self.unittest = unittest + self.assert_list = list + if list is not None: + self.assert_list.reverse() def execute(self, statement, parameters, **kwargs): - # TODO: get this to work - if self.assert_list is None: - return - item = self.assert_list.pop() - (query, params) = item - self.unittest.assert_(statement == query and params == parameters) + self.engine.echo = self.echo + if self.assert_list is not None: + item = self.assert_list.pop() + (query, params) = item + self.unittest.assert_(statement == query and params == parameters, query + repr(params) + statement + repr(parameters)) return self.realexec(statement, parameters, **kwargs) def runTests(suite): -- 2.47.2