privatevoid.net/depot
1
1
Fork 0
Code Issues 23 Pull requests 2 Projects 3 Releases Packages Wiki Activity Actions

cluster/services/patroni: adjust test for different numbers of replicas

Browse source
This commit is contained in:
Max Headroom 2024-11-30 21:20:03 +01:00
parent f2cbe6676b
commit ac73a38060
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
cluster/services/patroni/simulacrum/test.nix
View file

@ -1,8 +1,9 @@
{ cluster, ... }: { cluster, lib, ... }:
let let
clusterName = "poseidon"; clusterName = "poseidon";
link = cluster.config.links.patroni-pg-access; link = cluster.config.links.patroni-pg-access;
expectedReplicas = (lib.length cluster.config.services.patroni.nodes.worker) - 1;
in in
{ {
defaults = { depot, pkgs, ... }: { defaults = { depot, pkgs, ... }: {
@ -26,7 +27,7 @@ in
def booted(nodes): def booted(nodes):
return filter(lambda node: node.booted, nodes) return filter(lambda node: node.booted, nodes)
def wait_for_all_nodes_ready(expected_replicas=2): def wait_for_all_nodes_ready(expected_replicas=${toString expectedReplicas}):
booted_nodes = booted(nodes) booted_nodes = booted(nodes)
for node in booted_nodes: for node in booted_nodes:
node.wait_for_unit("patroni.service") node.wait_for_unit("patroni.service")
@ -71,7 +72,7 @@ in
for node in nodes: for node in nodes:
node.crash() node.crash()
wait_for_all_nodes_ready(1) wait_for_all_nodes_ready(${toString (expectedReplicas - 1)})
# Execute some queries while a node is down. # Execute some queries while a node is down.
run_dummy_queries() run_dummy_queries()