path: root/test/stress_test.sh

#!/usr/bin/env bash

# Copyright (c) 2020 Egor Tensin <Egor.Tensin@gmail.com>
# This file is part of the "math-server" project.
# For details, see https://github.com/egor-tensin/math-server.
# Distributed under the MIT License.

set -o errexit -o nounset -o pipefail

script_name="$( basename -- "${BASH_SOURCE[0]}" )"
readonly script_name
script_dir="$( dirname -- "${BASH_SOURCE[0]}" )"
script_dir="$( cd -- "$script_dir" && pwd )"
readonly script_dir

install_dir="$HOME/install"
readonly server_path='bin/math-server'
readonly client_path='bin/math-client'
readonly stress_test_path="$script_dir/stress_test.py"
readonly server_port=16666
server_pid=

dump() {
    local msg
    for msg; do
        echo "$script_name: $msg"
    done
}

kill_server() {
    dump "Killing the server with PID $server_pid..."
    kill "$server_pid"
    dump "Waiting for the server to terminate..."
    wait "$server_pid"
    dump "Done"
}

run_server() {
    dump "Running the server..."
    "$install_dir/$server_path" --port "$server_port" &
    server_pid="$!"
    dump "It's PID is $server_pid"
    trap kill_server EXIT
}

run_stress_test() {
    dump "Running stress_test.py..."
    "$stress_test_path"                      \
        --client "$install_dir/$client_path" \
        --port "$server_port"                \
        --processes 4                        \
        --expressions 1000
}

script_usage() {
    echo "usage: $script_name [INSTALL_DIR]"
}

parse_args() {
    if [ "$#" -gt 1 ]; then
        script_usage >&2
        return 1
    fi
    if [ "$#" -gt 0 ]; then
        install_dir="$1"
    fi
}

main() {
    parse_args "$@"
    run_server
    run_stress_test
}

main "$@"