def setUp(test): testing.setUp(test) conn = test.globs['conn'] with conn.cursor() as cur: cur.execute("CREATE TABLE mq (id SERIAL PRIMARY KEY, data JSONB)") cur.execute("INSERT INTO mq (data) VALUES " + ", ".join(jformat(datum) for datum in dataset)) conn.commit()
def setUp(test): testing.setUp(test) conn = test.globs['conn'] with conn.cursor() as cur: cur.execute("CREATE TABLE ser (id SERIAL PRIMARY KEY, data JSONB)") for dt in DTIMES: obj = dict(dtime=dt) cur.execute("INSERT INTO ser (data) VALUES (%s)", (jformat(obj), )) conn.commit()
def setUp(test): testing.setUp(test) conn = test.globs['conn'] with conn.cursor() as cur: cur.execute("CREATE TABLE ser (id SERIAL PRIMARY KEY, data JSONB)") for dt in DTIMES: obj = dict(dtime=dt) cur.execute("INSERT INTO ser (data) VALUES (%s)", (jformat(obj),)) conn.commit()
def setUp(test): testing.setUp(test) # silence this, otherwise half-baked objects raise exceptions # on trying to __repr__ missing attributes test.orig_DEBUG_EXCEPTION_FORMATTER = \ exceptionformatter.DEBUG_EXCEPTION_FORMATTER exceptionformatter.DEBUG_EXCEPTION_FORMATTER = 0 def fetchone(table, clause=None): qry = 'SELECT * FROM ' + table if clause: qry += ' WHERE ' + clause with test.globs['dm'].getCursor(False) as cur: cur.execute(qry) return cur.fetchone() test.globs['fetchone'] = fetchone