from sqlalchemy import create_engine, pool

from tempfile import mkdtemp

import shutil

from ngdatabase.mysql import Server, createConfig, Database

from iro.model.schema import Base

from iro.controller.pool import dbPool

        self.engine = md.engine

class SampleDatabase(Database):

    def createPassword(self):

        self.password="test"

        return self.password

class ModuleData:

    def __init__(self):

        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,

                poolclass = pool.SingletonThreadPool,  pool_size=dbPool.maxthreads, )#echo=True)

    def setUp(self):

        with open('%s/my.cnf'%self.tdir,'w') as cnf:

            cnf.write(createConfig(self.tdir))

        self.server.create()

        self.server.start()

        self.db.create()

        Base.metadata.create_all(self.engine)

    def tearDown(self):

        self.server.stop()

        shutil.rmtree(self.tdir)

md=ModuleData()

def setUpModule():

    md.setUp()

def tearDownModule():

    md.tearDown()