james/pgmon
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 3 Wiki Activity

Format python using black

Browse Source
This commit is contained in:
James Campbell 2025-05-15 02:04:50 -04:00
parent c872fc6b90
commit 8928bba337
Signed by: james
GPG Key ID: 2287C33A40DC906A
1 changed files with 22 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
tests/sql-tests.py
View File

@ -5,20 +5,22 @@ import requests
import yaml
import sys
pg_versions=[9.2, 9.6, 10, 11, 12, 13, 14, 15, 16 17]
pg_versions = [9.2, 9.6, 10, 11, 12, 13, 14, 15, 16, 17]
pgmon_port=93849
pgmon_port = 93849
tests = {}
container = None
def std_version(version):
if version[0] == '9':
if version[0] == "9":
return int(f"{version[0]}0{version[1]}00")
else
else:
return int(f"{version}0000")
def run_test(metric, params, status, check):
"""
Validate the return code and restults of a query
@ -32,16 +34,19 @@ def run_test(metric, params, status, check):
result = requests.get(f"http://localhost:{pgmon_port}/{metric}", params=params)
if result.status_code != status:
print(f"FAIL: {metric}[{params}] returned wrong status code: {result.status_code}")
print(
f"FAIL: {metric}[{params}] returned wrong status code: {result.status_code}"
)
return False
if re.match(check, result.text):
print(f"SUCCESS: {metric}[{params}]")
return True
else
else:
print(f"FAIL: {metric}[{params}] result is invalid, got:\n {result.text}")
return False
def run_all_tests(version):
"""
Run all defined tests against the current running instance
@ -52,22 +57,22 @@ def run_all_tests(version):
errors = 0
# Convert versions like 12 to 120000
version_num = std_version(version)
version_num = std_version(version)
# Loop through all of the metrics to test.
for metric in tests.keys():
params = metric.get('params', {})
params = metric.get("params", {})
status = 200
check = ''
check = ""
# Find the correct version of the status and check parameters (assuming there are any).
# If there are any check conditions, find the highest version that does not exceed the version we're currently testing against.
# To do this, we order the keys (versions) in reverse, so we start with the highest.
for v in reversed(sorted(metric.get('expect', {}).keys())):
for v in reversed(sorted(metric.get("expect", {}).keys())):
# If we've reached a version <= the one we're testing use it.
if int(v) <= version_num:
status = metric['expect'][v]['status']
check = metric['expect'][v]['check']
status = metric["expect"][v]["status"]
check = metric["expect"][v]["check"]
break
if not run_test(metric, metrics[metric].get(params, {}), status, check):
@ -75,16 +80,15 @@ def run_all_tests(version):
return errors
def start_test_db(version):
#container = PostgresContainer()
# container = PostgresContainer()
pass
# Read the test script
try:
with open("metric_tests.yml", 'r') as f:
with open("metric_tests.yml", "r") as f:
tests = yaml.safe_load(f)
except yaml.parser.ParserError as e:
sys.exit("Failed to parse metrics_test.yml: {e}")
# Run