From 897e8a30ed5b292df9d987f5f566e58a71394ce9 Mon Sep 17 00:00:00 2001
From: Egor Tensin <Egor.Tensin@gmail.com>
Date: Sun, 12 Mar 2023 16:46:43 +0100
Subject: test: refactoring

In preparation to adding LinuxServer tests.
---
 test/check_api.sh |  33 +++++++++
 test/host/test.sh | 147 ++++++++++++++++++++++++++++++++++++++++
 test/test.sh      | 196 ------------------------------------------------------
 3 files changed, 180 insertions(+), 196 deletions(-)
 create mode 100755 test/check_api.sh
 create mode 100755 test/host/test.sh
 delete mode 100755 test/test.sh

(limited to 'test')

diff --git a/test/check_api.sh b/test/check_api.sh
new file mode 100755
index 0000000..4dcc199
--- /dev/null
+++ b/test/check_api.sh
@@ -0,0 +1,33 @@
+#!/usr/bin/env bash
+
+set -o errexit -o nounset -o pipefail
+shopt -s inherit_errexit lastpipe
+
+run_curl() {
+    curl -sS -D - --connect-timeout 3 http://192.168.177.1:1234/ "$@"
+}
+
+run_curl_api() {
+    run_curl -H 'Content-Type: application/json' "$@"
+}
+
+call_api_method() {
+    local method
+    for method; do
+        echo ------------------------------------------------------------------
+        echo "Checking API method: $method"
+        echo ------------------------------------------------------------------
+        run_curl_api -d '{"jsonrpc": "2.0", "method": "'"$method"'", "params": {}}'
+    done
+}
+
+check_api() {
+    call_api_method ListPeers
+    call_api_method GetDeviceInfo
+}
+
+main() {
+    check_api
+}
+
+main "$@"
diff --git a/test/host/test.sh b/test/host/test.sh
new file mode 100755
index 0000000..25d5a87
--- /dev/null
+++ b/test/host/test.sh
@@ -0,0 +1,147 @@
+#!/usr/bin/env bash
+
+set -o errexit -o nounset -o pipefail
+shopt -s inherit_errexit lastpipe
+
+script_dir="$( dirname -- "${BASH_SOURCE[0]}" )"
+script_dir="$( cd -- "$script_dir" && pwd )"
+readonly script_dir
+
+base_dir="$( mktemp -d )"
+readonly base_dir
+
+readonly subnet_base=192.168.166
+ip_counter=1
+port_counter=561
+
+add_device() {
+    local name
+    for name; do
+        local dir
+        dir="$base_dir/devices/$name"
+        mkdir -p -- "$dir"
+
+        local ip
+        ip="$subnet_base.$ip_counter"
+        ip_counter=$((ip_counter + 1))
+        echo "$ip" > "$dir/ip"
+
+        local port
+        port="$port_counter"
+        port_counter=$((port_counter + 1))
+        echo "$port" > "$dir/port"
+
+        wg genkey | tee "$dir/private" | wg pubkey > "$dir/public"
+        ip link add dev "$name" type wireguard
+        ip addr add "$ip/24" dev "$name"
+        wg set "$name" private-key "$dir/private"
+        wg set "$name" listen-port "$port"
+    done
+}
+
+connect_devices() {
+    if [ "$#" -ne 2 ]; then
+        echo "usage: ${FUNCNAME[0]} DEV1 DEV2" >&2
+        return 1
+    fi
+
+    local dev1="$1"
+    local dev2="$2"
+
+    local dev1_dir
+    dev1_dir="$base_dir/devices/$dev1"
+    local dev2_dir
+    dev2_dir="$base_dir/devices/$dev2"
+
+    local pubkey1
+    pubkey1="$( cat -- "$dev1_dir/public" )"
+    local port
+    port="$( cat -- "$dev1_dir/port" )"
+    local pubkey2
+    pubkey2="$( cat -- "$dev2_dir/public" )"
+    local ip
+    ip="$( cat -- "$dev2_dir/ip" )"
+
+    wg set "$dev1" peer "$pubkey2" allowed-ips "$ip/32"
+    wg set "$dev2" peer "$pubkey1" allowed-ips "$subnet_base.0/24" endpoint "127.0.0.1:$port" persistent-keepalive 25
+}
+
+up_device() {
+    local name
+    for name; do
+        ip link set "$name" up
+    done
+}
+
+show_device() {
+    local name
+    for name; do
+        echo ------------------------------------------------------------------
+        echo "Device: $name"
+        echo ------------------------------------------------------------------
+        wg show "$name"
+        echo
+    done
+}
+
+add_devices() {
+    add_device server
+    add_device client1
+    add_device client2
+    add_device client3
+    connect_devices server client1
+    connect_devices server client2
+    connect_devices server client3
+    up_device server client1 client2 client3
+    sleep 2
+    show_device server client1 client2 client3
+}
+
+build_services() {
+    echo ------------------------------------------------------------------
+    echo Pull third-party images
+    echo ------------------------------------------------------------------
+    docker-compose pull api
+
+    echo ------------------------------------------------------------------
+    echo Build wg-api-web
+    echo ------------------------------------------------------------------
+    docker-compose build --force-rm --progress plain --pull web
+
+    echo ------------------------------------------------------------------
+    echo docker-compose up
+    echo ------------------------------------------------------------------
+    WG_IFACE=server docker-compose up -d
+}
+
+cleanup() {
+    echo ------------------------------------------------------------------
+    echo Cleaning up
+    echo ------------------------------------------------------------------
+
+    if [ -d "$base_dir/devices" ]; then
+        local name
+        find "$base_dir/devices" -mindepth 1 -maxdepth 1 -type d -printf '%P\0' \
+                | while IFS= read -d '' -r name; do
+            echo "Removing device: $name"
+            ip link delete "$name" type wireguard || true
+        done
+    fi
+
+    echo "Removing $base_dir"
+    rm -rf -- "$base_dir"
+
+    echo "Brining down containers..."
+    docker-compose down -v --remove-orphans
+}
+
+main() {
+    cd -- "$script_dir/.."
+    trap cleanup EXIT
+
+    add_devices
+    build_services
+    "$script_dir/../check_api.sh"
+}
+
+main "$@"
diff --git a/test/test.sh b/test/test.sh
deleted file mode 100755
index c8839bb..0000000
--- a/test/test.sh
+++ /dev/null
@@ -1,196 +0,0 @@
-#!/usr/bin/env bash
-
-set -o errexit -o nounset -o pipefail
-shopt -s inherit_errexit lastpipe
-
-script_dir="$( dirname -- "${BASH_SOURCE[0]}" )"
-script_dir="$( cd -- "$script_dir" && pwd )"
-readonly script_dir
-
-devices_only=
-
-base_dir="$( mktemp -d )"
-readonly base_dir
-
-readonly subnet_base=192.168.166
-ip_counter=1
-port_counter=561
-
-add_device() {
-    local name
-    for name; do
-        local dir
-        dir="$base_dir/devices/$name"
-        mkdir -p -- "$dir"
-
-        local ip
-        ip="$subnet_base.$ip_counter"
-        ip_counter=$((ip_counter + 1))
-        echo "$ip" > "$dir/ip"
-
-        local port
-        port="$port_counter"
-        port_counter=$((port_counter + 1))
-        echo "$port" > "$dir/port"
-
-        wg genkey | tee "$dir/private" | wg pubkey > "$dir/public"
-        ip link add dev "$name" type wireguard
-        ip addr add "$ip/24" dev "$name"
-        wg set "$name" private-key "$dir/private"
-        wg set "$name" listen-port "$port"
-    done
-}
-
-connect_devices() {
-    if [ "$#" -ne 2 ]; then
-        echo "usage: ${FUNCNAME[0]} DEV1 DEV2" >&2
-        return 1
-    fi
-
-    local dev1="$1"
-    local dev2="$2"
-
-    local dev1_dir
-    dev1_dir="$base_dir/devices/$dev1"
-    local dev2_dir
-    dev2_dir="$base_dir/devices/$dev2"
-
-    local pubkey1
-    pubkey1="$( cat -- "$dev1_dir/public" )"
-    local port
-    port="$( cat -- "$dev1_dir/port" )"
-    local pubkey2
-    pubkey2="$( cat -- "$dev2_dir/public" )"
-    local ip
-    ip="$( cat -- "$dev2_dir/ip" )"
-
-    wg set "$dev1" peer "$pubkey2" allowed-ips "$ip/32"
-    wg set "$dev2" peer "$pubkey1" allowed-ips "$subnet_base.0/24" endpoint "127.0.0.1:$port" persistent-keepalive 25
-}
-
-up_device() {
-    local name
-    for name; do
-        ip link set "$name" up
-    done
-}
-
-show_device() {
-    local name
-    for name; do
-        echo ------------------------------------------------------------------
-        echo "Device: $name"
-        echo ------------------------------------------------------------------
-        wg show "$name"
-        echo
-    done
-}
-
-add_devices() {
-    add_device server
-    add_device client1
-    add_device client2
-    add_device client3
-    connect_devices server client1
-    connect_devices server client2
-    connect_devices server client3
-    up_device server client1 client2 client3
-    sleep 2
-    show_device server client1 client2 client3
-}
-
-build_services() {
-    echo ------------------------------------------------------------------
-    echo Pull third-party images
-    echo ------------------------------------------------------------------
-    docker-compose pull api
-
-    echo ------------------------------------------------------------------
-    echo Build wg-api-web
-    echo ------------------------------------------------------------------
-    docker-compose build --force-rm --progress plain --pull web
-
-    echo ------------------------------------------------------------------
-    echo docker-compose up
-    echo ------------------------------------------------------------------
-    WG_IFACE=server docker-compose up -d
-}
-
-run_curl() {
-    curl -sS -D - --connect-timeout 3 http://192.168.177.1:1234/ "$@"
-}
-
-run_curl_api() {
-    run_curl -H 'Content-Type: application/json' "$@"
-}
-
-call_api_method() {
-    local method
-    for method; do
-        echo ------------------------------------------------------------------
-        echo "Checking API method: $method"
-        echo ------------------------------------------------------------------
-        run_curl_api -d '{"jsonrpc": "2.0", "method": "'"$method"'", "params": {}}'
-    done
-}
-
-check_api() {
-    call_api_method ListPeers
-    call_api_method GetDeviceInfo
-}
-
-cleanup() {
-    echo ------------------------------------------------------------------
-    echo Cleaning up
-    echo ------------------------------------------------------------------
-
-    if [ -d "$base_dir/devices" ]; then
-        local name
-        find "$base_dir/devices" -mindepth 1 -maxdepth 1 -type d -printf '%P\0' \
-                | while IFS= read -d '' -r name; do
-            echo "Removing device: $name"
-            ip link delete "$name" type wireguard || true
-        done
-    fi
-
-    echo "Removing $base_dir"
-    rm -rf -- "$base_dir"
-
-    if [ -z "$devices_only" ]; then
-        echo "Brining down containers..."
-        docker-compose down -v --remove-orphans
-    fi
-}
-
-main() {
-    cd -- "$script_dir/.."
-    trap cleanup EXIT
-
-    local opt
-    while getopts ':i' opt "$@"; do
-        case "$opt" in
-            i)
-                devices_only=1
-                ;;
-            :)
-                echo "usage error: required argument missing for option -$OPTARG" >&2
-                exit 1;
-                ;;
-            *)
-                echo "usage error: invalid option -$OPTARG" >&2;
-                exit 1;
-                ;;
-        esac
-    done
-
-    add_devices
-
-    if [ -z "$devices_only" ]; then
-        build_services
-        check_api
-    else
-        while true; do sleep 1; done
-    fi
-}
-
-main "$@"
-- 
cgit v1.2.3