class ConnBaseTest(unittest.TestCase):
dbname = os.environ.get('QC_DBNAME', "ConnTest")
table = 'queue_classic_jobs'
q_name = 'default'
base_dsn = {}
def Conn(self, *args, **kwargs):
return Conn(dbname=self.dbname, *args, **dict(self.base_dsn, **kwargs))
@classmethod
def setUpClass(cls):
cls.conn = Conn(dbname="postgres", **cls.base_dsn)
cls.conn.execute('CREATE DATABASE "%s";' % cls.dbname)
cls.db_conn = Conn(dbname=cls.dbname, **cls.base_dsn)
cls.db_conn.execute("CREATE LANGUAGE plpgsql;")
cls.db_queue = Queue(cls.db_conn, cls.table, cls.q_name)
cls.db_queue.create("""\
method varchar(255),
args text""")
def setUp(self):
self.conn = Conn(dbname=self.dbname, **self.base_dsn)
def tearDown(self):
self.conn.disconnect()
@classmethod
def tearDownClass(cls):
cls.db_queue.drop()
cls.db_conn.disconnect()
cls.conn.execute("DROP DATABASE \"%s\"" % cls.dbname)
cls.conn.disconnect()
def setUp(self):
self.conn = Conn(dbname=self.dbname, **self.base_dsn)