182 lines
6.0 KiB
Bash
Executable File
182 lines
6.0 KiB
Bash
Executable File
#!/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 -v $(pwd)/tests/entrypoint.sh:/entrypoint.sh --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
|
|
}
|
|
|
|
test_rejoin() {
|
|
run_test_container test1-orig test1 --init
|
|
run_test_container test2-orig test2 --join test1-orig --rejoin 5
|
|
|
|
sleep 3
|
|
|
|
docker exec test1-orig bash -c "ip l s down eth0"
|
|
sleep 70
|
|
|
|
docker exec test2-orig grep test1 /etc/hosts && (docker logs test1-orig; docker logs test2-orig; false)
|
|
docker exec test1-orig bash -c "ip l s up eth0"
|
|
|
|
sleep 10
|
|
|
|
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_fixed_address() {
|
|
run_test_container test1-orig test1 --init --overlay-net 10.10.0.1/16
|
|
run_test_container test2-orig test2 --join test1-orig --overlay-net 10.10.0.0/16
|
|
|
|
sleep 3
|
|
|
|
docker exec test1-orig ping -c1 -W1 test2 || (docker logs test1-orig; docker logs test2-orig; false)
|
|
docker exec test2-orig ping -c1 -W1 10.10.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 '\<test_.*$'); do
|
|
echo "--- Running $test_func:"
|
|
$test_func
|
|
echo "--- OK"
|
|
done
|