Add integration test for trust authentication

Author: CommanderKeynes

.circleci/pgcat.toml

@@ -49,6 +49,7 @@ tls_private_key = ".circleci/server.key"
 # Connecting to that database allows running commands like `SHOW POOLS`, `SHOW DATABASES`, etc..
 admin_username = "admin_user"
 admin_password = "admin_pass"
+admin_auth_type = "md5"
 
 # pool
 # configs are structured as pool.<pool_name>

.circleci/pgcat_trust.toml

@@ -0,0 +1,22 @@
+
+[general]
+
+host = "0.0.0.0"
+port = 6432
+admin_username = "admin_user"
+admin_password = ""
+admin_auth_type = "trust"
+
+[pools.sharded_db.users.0]
+username = "sharding_user"
+password = "sharding_user"
+auth_type = "trust"
+pool_size = 10
+min_pool_size = 1
+pool_mode = "transaction"
+
+[pools.sharded_db.shards.0]
+servers = [
+  [ "127.0.0.1", 5432, "primary" ],
+]
+database = "shard0"

tests/python/tests.py

@@ -16,6 +16,11 @@ def pgcat_start():
     os.system("./target/debug/pgcat .circleci/pgcat.toml &")
     time.sleep(2)
 
+def pgcat_trust_start():
+    pg_cat_send_signal(signal.SIGTERM)
+    os.system("./target/debug/pgcat .circleci/pgcat_trust.toml &")
+    time.sleep(2)
+
 
 def pg_cat_send_signal(signal: signal.Signals):
     try:
@@ -33,6 +38,28 @@ def pg_cat_send_signal(signal: signal.Signals):
             raise Exception("pgcat not closed after SIGTERM")
 
 
+def connect_db_trust(
+    autocommit: bool = True,
+    admin: bool = False,
+) -> Tuple[psycopg2.extensions.connection, psycopg2.extensions.cursor]:
+
+    if admin:
+        user = "admin_user"
+        db = "pgcat"
+    else:
+        user = "sharding_user"
+        db = "sharded_db"
+
+    conn = psycopg2.connect(
+        f"postgres://{user}:@{PGCAT_HOST}:{PGCAT_PORT}/{db}?application_name=testing_pgcat",
+        connect_timeout=2,
+    )
+    conn.autocommit = autocommit
+    cur = conn.cursor()
+
+    return (conn, cur)
+
+
 def connect_db(
     autocommit: bool = True,
     admin: bool = False,
@@ -62,6 +89,29 @@ def cleanup_conn(conn: psycopg2.extensions.connection, cur: psycopg2.extensions.
     conn.close()
 
 
+def test_admin_trust_auth():
+    conn, cur = connect_db_trust(admin=True)
+    cur.execute("SHOW POOLS")
+    res = cur.fetchall()
+    print(res)
+    cleanup_conn(conn, cur)
+
+
+def test_normal_trust_auth():
+    conn, cur = connect_db_trust(autocommit=False)
+    cur.execute("SELECT 1")
+    res = cur.fetchall()
+    print(res)
+    cleanup_conn(conn, cur)
+
+
+def test_trust():
+    pgcat_trust_start()
+    test_admin_trust_auth()
+    test_normal_trust_auth()
+    pg_cat_send_signal(signal.SIGTERM)
+
+
 def test_normal_db_access():
     pgcat_start()
     conn, cur = connect_db(autocommit=False)
@@ -71,6 +121,7 @@ def test_normal_db_access():
     cleanup_conn(conn, cur)
 
 
+
 def test_admin_db_access():
     conn, cur = connect_db(admin=True)
 
@@ -317,3 +368,5 @@ def test_shutdown_logic():
 test_normal_db_access()
 test_admin_db_access()
 test_shutdown_logic()
+
+test_trust()