From 66efc5215323deef0929c8d66bcbe54d535c53d4 Mon Sep 17 00:00:00 2001
From: Egor Tensin <Egor.Tensin@gmail.com>
Date: Wed, 24 Mar 2021 19:06:14 +0300
Subject: add stress_test.sh as a CTest test

---
 test/CMakeLists.txt |  8 ++++++++
 test/stress_test.sh | 24 +++++++++++-------------
 2 files changed, 19 insertions(+), 13 deletions(-)

diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt
index 5b81467..a956d03 100644
--- a/test/CMakeLists.txt
+++ b/test/CMakeLists.txt
@@ -1,2 +1,10 @@
 add_subdirectory(benchmarks)
 add_subdirectory(unit_tests)
+
+if(CMAKE_HOST_UNIX)
+    # Not going to bother with bash on Windows.
+
+    # stress_test.py is a Python 3 script.
+    find_package(Python3 REQUIRED COMPONENTS Interpreter)
+    add_test(NAME stress_test COMMAND "${CMAKE_CURRENT_SOURCE_DIR}/stress_test.sh" "$<TARGET_FILE:server>" "$<TARGET_FILE:client>")
+endif()
diff --git a/test/stress_test.sh b/test/stress_test.sh
index 92187ae..1f7c9e1 100755
--- a/test/stress_test.sh
+++ b/test/stress_test.sh
@@ -13,9 +13,8 @@ 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'
+server_path=
+client_path=
 readonly stress_test_path="$script_dir/stress_test.py"
 readonly server_port=16666
 server_pid=
@@ -37,7 +36,7 @@ kill_server() {
 
 run_server() {
     dump "Running the server..."
-    "$install_dir/$server_path" --port "$server_port" &
+    "$server_path" --port "$server_port" &
     server_pid="$!"
     dump "Its PID is $server_pid"
     trap kill_server EXIT
@@ -45,25 +44,24 @@ run_server() {
 
 run_stress_test() {
     dump "Running stress_test.py..."
-    "$stress_test_path"                      \
-        --client "$install_dir/$client_path" \
-        --port "$server_port"                \
-        --processes 4                        \
+    "$stress_test_path"         \
+        --client "$client_path" \
+        --port "$server_port"   \
+        --processes 4           \
         --expressions 1000
 }
 
 script_usage() {
-    echo "usage: $script_name [INSTALL_DIR]"
+    echo "usage: $script_name SERVER_PATH CLIENT_PATH"
 }
 
 parse_args() {
-    if [ "$#" -gt 1 ]; then
+    if [ "$#" -ne 2 ]; then
         script_usage >&2
         return 1
     fi
-    if [ "$#" -gt 0 ]; then
-        install_dir="$1"
-    fi
+    server_path="$1"
+    client_path="$2"
 }
 
 main() {
-- 
cgit v1.2.3