#!/bin/bash set -e declare -A started_containers cleanup() { if [ ${#started_containers[@]} -gt 0 ]; then echo "Stopping all remaining containers: ${started_containers[@]}" docker container rm -f ${started_containers[@]} fi echo "Removing shared network" docker network rm wesher_test } docker network create wesher_test trap cleanup EXIT run_test_container() { local name=$1 echo "Starting $name" shift local hostname=$1 shift docker run -d --cap-add=NET_ADMIN --name ${name} --hostname ${hostname} -v $(pwd):/app --network=wesher_test costela/wesher-test "$@" started_containers[$name]=$name } stop_test_container() { echo "Stopping $1" docker container rm -f $1 unset started_containers[$1] } test_3_node_up() { run_test_container test1-orig test1 --init run_test_container test2-orig test2 --join test1-orig run_test_container test3-orig test3 --join test1-orig sleep 3 docker exec test1-orig ping -c1 -W1 test2 || (docker logs test1-orig; docker logs test2-orig; false) docker exec test1-orig ping -c1 -W1 test3 || (docker logs test1-orig; docker logs test3-orig; false) stop_test_container test3-orig stop_test_container test2-orig stop_test_container test1-orig } test_5_node_up() { run_test_container test1-orig test1 --init run_test_container test2-orig test2 --join test1-orig run_test_container test3-orig test3 --join test1-orig run_test_container test4-orig test4 --join test1-orig run_test_container test5-orig test5 --join test1-orig sleep 5 docker exec test1-orig ping -c1 -W1 test2 || (docker logs test1-orig; docker logs test2-orig; false) docker exec test1-orig ping -c1 -W1 test3 || (docker logs test1-orig; docker logs test3-orig; false) docker exec test1-orig ping -c1 -W1 test4 || (docker logs test1-orig; docker logs test4-orig; false) docker exec test1-orig ping -c1 -W1 test5 || (docker logs test1-orig; docker logs test5-orig; false) stop_test_container test5-orig stop_test_container test4-orig stop_test_container test3-orig stop_test_container test2-orig stop_test_container test1-orig } test_node_restart() { run_test_container test1-orig test1 --init run_test_container test2-orig test2 --join test1-orig sleep 3 docker stop test2-orig docker start test2-orig sleep 3 docker exec test1-orig ping -c1 -W1 test2 || (docker logs test1-orig; docker logs test2-orig; false) stop_test_container test2-orig stop_test_container test1-orig } test_cluster_simultaneous_start() { run_test_container test1-orig test1 --join test2-orig,test3-orig run_test_container test2-orig test2 --join test1-orig,test3-orig run_test_container test3-orig test3 --join test1-orig,test2-orig sleep 3 docker exec test1-orig ping -c1 -W1 test2 || (docker logs test1-orig; docker logs test2-orig; false) docker exec test1-orig ping -c1 -W1 test3 || (docker logs test1-orig; docker logs test3-orig; false) stop_test_container test3-orig stop_test_container test2-orig stop_test_container test1-orig } test_multiple_clusters_restart() { cluster1='--cluster-port 7946 --wireguard-port 51820 --interface wg1 --overlay-net 10.10.0.0/16' cluster2='--cluster-port 7947 --wireguard-port 51821 --interface wg2 --overlay-net 10.11.0.0/16' run_test_container test1-orig test1 --init $cluster1 run_test_container test2-orig test2 --init $cluster2 run_test_container test3-orig test3 --join test1-orig $cluster1 docker exec -d test3-orig bash -c "/entrypoint.sh --join test2-orig $cluster2" sleep 3 docker stop test3-orig docker start test3-orig docker exec -d test3-orig bash -c "/entrypoint.sh $cluster2" sleep 3 docker exec test3-orig ping -c1 -W1 test1 || (docker logs test1-orig; docker logs test3-orig; false) docker exec test3-orig ping -c1 -W1 test2 || (docker logs test2-orig; docker logs test3-orig; false) docker exec test1-orig ping -c1 -W1 test3 || (docker logs test1-orig; docker logs test3-orig; false) docker exec test2-orig ping -c1 -W1 test3 || (docker logs test2-orig; docker logs test3-orig; false) stop_test_container test3-orig stop_test_container test2-orig stop_test_container test1-orig } test_routed_network() { run_test_container test1-orig test1 --init --routed-net 10.15.0.0/16 run_test_container test2-orig test2 --join test1-orig --routed-net 10.15.0.0/16 docker exec test2-orig bash -c "ip l a test type bridge; ip l s up test; ip a a 10.15.0.1/24 dev test" sleep 3 docker exec test1-orig ping -c1 -W1 test2 || (docker logs test1-orig; docker logs test2-orig; false) docker exec test1-orig ping -c1 -W1 10.15.0.1 || (docker logs test1-orig; docker logs test2-orig; false) stop_test_container test2-orig stop_test_container test1-orig } for test_func in $(declare -F | grep -Eo '\