docker/pgcat
1
0
Fork 0
mirror of https://github.com/postgresml/pgcat.git synced 2026-03-23 17:36:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Reverted LDAP changes, Cleaned up tests

Browse Source
This commit is contained in:
CommanderKeynes
2024-09-10 06:51:52 -05:00
parent d8ccf4babb
commit fba40eba2f
4 changed files with 78 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/python/utils.py
View File

@@ -2,6 +2,7 @@ import os
import signal
import time
from typing import Tuple
import tempfile
import psutil
import psycopg2
@@ -20,8 +21,11 @@ def pgcat_start():
_pgcat_start(config_path='.circleci/pgcat.toml')
def pgcat_trust_start():
_pgcat_start(config_path='.circleci/pgcat_trust.toml')
def pgcat_generic_start(config: str):
tmp = tempfile.NamedTemporaryFile()
with open(tmp.name, 'w') as f:
f.write(config)
_pgcat_start(config_path=tmp.name)
def glauth_send_signal(signal: signal.Signals):
@@ -43,7 +47,7 @@ def glauth_send_signal(signal: signal.Signals):
def pg_cat_send_signal(signal: signal.Signals):
try:
for proc in psutil.process_iter(["pid", "name"]):
if proc.name() == "pgcat":
if "pgcat" == proc.name():
os.kill(proc.pid, signal)
except Exception as e:
# The process can be gone when we send this signal