--- a/tests/dbtestcase.py Mon Mar 05 23:52:29 2012 +0100
+++ b/tests/dbtestcase.py Sat Mar 10 19:00:12 2012 +0100
@@ -7,7 +7,7 @@
import atexit
from ngdatabase.mysql import Server, createConfig, Database
-from iro.model import schema, setEngine, setPool
+from iro.model import setEngine, setPool
from iro.model.utils import WithSession
from iro.model.schema import Base
@@ -33,8 +33,8 @@
def __cleanDB(self):
'''cleaning database'''
with self.session() as session:
- for table in schema.__tables__:
- session.query(getattr(schema,table)).delete()
+ for table in reversed(Base.metadata.sorted_tables):
+ session.execute(table.delete())
class SampleDatabase(Database):
@@ -55,12 +55,13 @@
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.dburl = 'mysql://test:test@localhost/test?unix_socket=%s/socket'%self.tdir
+ self.engine = create_engine(self.dburl,
poolclass = pool.SingletonThreadPool, pool_size=dbPool.maxthreads, )#echo=True)
- self.vaild = False
+ self.valid = False
def setUp(self):
- if not self.vaild:
+ if not self.valid:
with open('%s/my.cnf'%self.tdir,'w') as cnf:
cnf.write(createConfig(self.tdir))
self.server.create()
@@ -69,7 +70,7 @@
Base.metadata.create_all(self.engine)
setEngine(self.engine)
setPool(dbPool)
- self.vaild = True
+ self.valid = True
md=ModuleData()