Skip to content
Snippets Groups Projects
Commit 7201e269 authored by Nicolas KAROLAK's avatar Nicolas KAROLAK
Browse files

change(test_postgresql): nicer output

parent 31afd055
No related branches found
No related tags found
No related merge requests found
......@@ -377,36 +377,40 @@ def check_ha(db_conn: dict, errors: int = 0, warnings: int = 0) -> tuple:
nodes = get_nodes(ha_conf)
# check haproxy
print("Checking local HAProxy frontends:")
if not check_listen(db_host, 54321):
print("{}HAProxy pgsql-primary frontend is not listening{}".format(RED, DEF))
print("{}✖{} HAProxy pgsql-primary frontend is not listening".format(RED, DEF))
errors += 1
else:
print("HAProxy pgsql-primary frontend is listening")
print("{}✔{} HAProxy pgsql-primary frontend is listening".format(GREEN, DEF))
if not check_listen(db_host, 54322):
print("{}HAProxy pgsql-standby frontend is not listening{}".format(RED, DEF))
print("{}✖{} HAProxy pgsql-standby frontend is not listening".format(RED, DEF))
errors += 1
else:
print("HAProxy pgsql-standby frontend is listening")
print("{}✔{} HAProxy pgsql-standby frontend is listening".format(GREEN, DEF))
# check remotes
print("Checking remote PostgreSQL nodes:")
for node in nodes:
node_host = nodes[node]["host"]
node_port = nodes[node]["port"]
if not check_listen(node_host, node_port):
print("{}Cannot bind {}:{}{}".format(RED, node_host, node_port, DEF))
print("{}✖{} Cannot bind {}:{}".format(RED, DEF, node_host, node_port))
errors += 1
else:
print("Can bind {}:{}".format(node_host, node_port))
print("{}✔{} Can bind {}:{}".format(GREEN, DEF, node_host, node_port))
# check fenced
print("Checking cluster state:")
fenced, node = check_fenced(nodes)
if fenced:
print("{}Node `{}` is fenced{}".format(RED, node, DEF))
print("{}✖{} Node `{}` is fenced".format(RED, DEF, node))
errors += 1
else:
print("No fenced node found")
print("{}✔{} No fenced node found".format(GREEN, DEF))
# check replication
print("Checking replication state:")
primary = {
"dbname": "postgres",
"user": db_user,
......@@ -422,10 +426,10 @@ def check_ha(db_conn: dict, errors: int = 0, warnings: int = 0) -> tuple:
"port": 54322,
}
if not check_replication(primary, standby):
print("{}Cannot replicate data between primary/standby{}".format(RED, DEF))
print("{}✖{} Cannot replicate data between primary/standby".format(RED, DEF))
errors += 1
else:
print("Can replicate data between primary/standby")
print("{}✔{} Can replicate data between primary/standby".format(GREEN, DEF))
return errors, warnings
......@@ -449,31 +453,44 @@ def check_local(db_conn: dict, errors: int = 0, warnings: int = 0) -> tuple:
db_pass = db_conn["pass"]
# check listen
print("Checking local PostgreSQL node:")
if not check_listen(db_host, db_port):
print("{}Cannot connect to {}:{}{}".format(RED, db_host, db_port, DEF))
print("{}✖{} Cannot connect to {}:{}".format(RED, DEF, db_host, db_port))
errors += 1
else:
print("{}✔{} Can connect to {}:{}".format(GREEN, DEF, db_host, db_port))
# check read
print("Checking read operation:")
if not check_read(db_host, db_port, db_user, db_pass):
print(
"{}Cannot read data on {}@{}:{}{}".format(
RED, db_user, db_host, db_port, DEF
"{}✖{} Cannot read data on {}@{}:{}".format(
RED, DEF, db_user, db_host, db_port
)
)
errors += 1
else:
print("Can read data on {}@{}:{}".format(db_user, db_host, db_port))
print(
"{}✔{} Can read data on {}@{}:{}".format(
GREEN, DEF, db_user, db_host, db_port
)
)
# check write
print("Checking write operation:")
if not check_write(db_host, db_port, db_user, db_pass):
print(
"{}Cannot write data on {}@{}:{}{}".format(
RED, db_user, db_host, db_port, DEF
"{}✖{} Cannot write data on {}@{}:{}".format(
RED, DEF, db_user, db_host, db_port
)
)
errors += 1
else:
print("Can write data on {}@{}:{}".format(db_user, db_host, db_port))
print(
"{}✔{} Can write data on {}@{}:{}".format(
GREEN, DEF, db_user, db_host, db_port
)
)
return errors, warnings
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment