コード例 #1
0
ファイル: common.py プロジェクト: noteed/pueuey
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()
コード例 #2
0
ファイル: common.py プロジェクト: noteed/pueuey
 def setUp(self):
     self.conn = Conn(dbname=self.dbname, **self.base_dsn)