Files
pgcat/tests/python/tests.py

23 lines
595 B
Python
Raw Normal View History

2022-02-03 18:02:50 -08:00
import psycopg2
def test_normal_db_access():
conn = psycopg2.connect("postgres://sharding_user:sharding_user@127.0.0.1:6432/sharded_db?application_name=testing_pgcat")
cur = conn.cursor()
2022-02-03 18:02:50 -08:00
cur.execute("SELECT 1")
res = cur.fetchall()
print(res)
2022-02-03 18:02:50 -08:00
2022-02-03 18:13:36 -08:00
def test_admin_db_access():
conn = psycopg2.connect("postgres://admin_user:admin_pass@127.0.0.1:6432/pgcat")
conn.autocommit = True # BEGIN/COMMIT is not supported by admin db
cur = conn.cursor()
cur.execute("SHOW POOLS")
res = cur.fetchall()
print(res)
test_normal_db_access()
test_admin_db_access()