mirror of
https://github.com/postgresml/pgcat.git
synced 2026-03-23 17:36:28 +00:00
670311daf9
* Implement Trust Authentication * Remove remaining LDAP stuff * Reverted LDAP changes, Cleaned up tests --------- Co-authored-by: Andrew Jackson <andrewjackson2988@gmail.com> Co-authored-by: CommanderKeynes <andrewjackson947@gmail.coma>
72 lines
1.8 KiB
Python
72 lines
1.8 KiB
Python
import utils
|
|
import signal
|
|
|
|
class TestTrustAuth:
|
|
@classmethod
|
|
def setup_method(cls):
|
|
config= """
|
|
[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"
|
|
"""
|
|
utils.pgcat_generic_start(config)
|
|
|
|
@classmethod
|
|
def teardown_method(self):
|
|
utils.pg_cat_send_signal(signal.SIGTERM)
|
|
|
|
def test_admin_trust_auth(self):
|
|
conn, cur = utils.connect_db_trust(admin=True)
|
|
cur.execute("SHOW POOLS")
|
|
res = cur.fetchall()
|
|
print(res)
|
|
utils.cleanup_conn(conn, cur)
|
|
|
|
def test_normal_trust_auth(self):
|
|
conn, cur = utils.connect_db_trust(autocommit=False)
|
|
cur.execute("SELECT 1")
|
|
res = cur.fetchall()
|
|
print(res)
|
|
utils.cleanup_conn(conn, cur)
|
|
|
|
class TestMD5Auth:
|
|
@classmethod
|
|
def setup_method(cls):
|
|
utils.pgcat_start()
|
|
|
|
@classmethod
|
|
def teardown_method(self):
|
|
utils.pg_cat_send_signal(signal.SIGTERM)
|
|
|
|
def test_normal_db_access(self):
|
|
conn, cur = utils.connect_db(autocommit=False)
|
|
cur.execute("SELECT 1")
|
|
res = cur.fetchall()
|
|
print(res)
|
|
utils.cleanup_conn(conn, cur)
|
|
|
|
def test_admin_db_access(self):
|
|
conn, cur = utils.connect_db(admin=True)
|
|
|
|
cur.execute("SHOW POOLS")
|
|
res = cur.fetchall()
|
|
print(res)
|
|
utils.cleanup_conn(conn, cur)
|