diff options
Diffstat (limited to 'test')
-rwxr-xr-x | test/stress_test.sh | 75 |
1 files changed, 75 insertions, 0 deletions
diff --git a/test/stress_test.sh b/test/stress_test.sh new file mode 100755 index 0000000..ceac401 --- /dev/null +++ b/test/stress_test.sh @@ -0,0 +1,75 @@ +#!/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 "$@" |