def _drop_dummy_db(self, dbname):
conn = connector(host=ENV['pg_sockdir'],
port=cf.PG_PORT,
user=cf.PG_USER,
password=cf.PG_PASSWORD,
database='postgres')
conn.connect()
conn.execute("DROP DATABASE %s" % (dbname))
conn.close()
def _create_dummy_table(self, dbname, tablename):
conn = connector(host=ENV['pg_sockdir'],
port=cf.PG_PORT,
user=cf.PG_USER,
password=cf.PG_PASSWORD,
database=dbname)
conn.connect()
conn.execute("CREATE TABLE %s (id INTEGER)" % (tablename))
conn.execute("INSERT INTO %s SELECT generate_series(1, 500000)" %
(tablename))
conn.close()
def _get_duration_since_last_vacuum(self, dbname, tablename):
conn = connector(host=ENV['pg_sockdir'],
port=cf.PG_PORT,
user=cf.PG_USER,
password=cf.PG_PASSWORD,
database=dbname)
conn.connect()
conn.execute(
"SELECT coalesce(EXTRACT(EPOCH FROM (NOW() - last_vacuum)),0) AS duration FROM pg_stat_user_tables WHERE relname = '%s'"
% (tablename))
conn.close()
return list(conn.get_rows())[0]['duration']
def _exec_query(self, dbname, query):
conn = connector(
host = ENV['pg_sockdir'],
port = cf.PG_PORT,
user = cf.PG_USER,
password = cf.PG_PASSWORD,
database = dbname
)
conn.connect()
conn.execute(query)
conn.close()
return list(conn.get_rows())
def test_00_env_pg(self):
"""
[dashboard] 00: PostgreSQL instance is up & running
"""
conn = connector(host=ENV['pg_sockdir'],
port=cf.PG_PORT,
user=cf.PG_USER,
password=cf.PG_PASSWORD,
database='postgres')
try:
conn.connect()
conn.close()
global XSESSION
XSESSION = self._temboard_login()
assert True
except error as e:
assert False
def pg_add_super_user(pg_bin, pg_user, pg_host, pg_port, pg_password=''):
(ret_code, stdout, stderr) = exec_command(
[pg_bin + "/createuser", "-h", pg_host, "-p", pg_port, "-ls", pg_user])
if ret_code != 0:
raise Exception(str(stderr))
conn = connector(host=pg_host,
port=pg_port,
user=pg_user,
database='postgres')
if len(pg_password) > 0:
try:
conn.connect()
conn.execute("ALTER USER %s PASSWORD '%s'" %
(pg_user, pg_password))
conn.close()
except error as e:
raise Exception(str(e.message))
