tests/dbtestcase.py
branchdevel
changeset 135 f8640c663e3e
parent 134 fae3fdfece65
child 138 57c4450c8ba6
--- a/tests/dbtestcase.py	Mon Feb 06 14:39:33 2012 +0100
+++ b/tests/dbtestcase.py	Tue Feb 07 01:56:59 2012 +0100
@@ -6,7 +6,7 @@
 
 from ngdatabase.mysql import Server, createConfig, Database
 
-from iro.model import schema
+from iro.model import schema, setEngine
 from iro.model.utils import WithSession
 from iro.model.schema import Base
 
@@ -42,7 +42,7 @@
         self.tdir = mkdtemp(prefix='iro-mysql-')
         self.server = Server('%s/my.cnf'%self.tdir)
         self.db = SampleDatabase("test","test",'%s/my.cnf'%self.tdir)
-        self.engine = create_engine('mysql://test:test@localhost/test?unix_socket=%s/socket'%self.tdir,
+        self.engine = create_engine('mysql://test:test@localhost/test?unix_socket=%s/socket'%self.tdir, 
                 poolclass = pool.SingletonThreadPool,  pool_size=dbPool.maxthreads, )#echo=True)
 
     def setUp(self):
@@ -52,6 +52,7 @@
         self.server.start()
         self.db.create()
         Base.metadata.create_all(self.engine)
+        setEngine(self.engine)
     
     def tearDown(self):
         self.server.stop()