From 029289413c0f022d66d04c97d0bd21e57700fb4a Mon Sep 17 00:00:00 2001
From: Benjamin Sergeant <bsergean@gmail.com>
Date: Wed, 29 Jul 2020 17:46:37 -0700
Subject: [PATCH] ws test shell script / add option so tune how large sent file
 will be

---
 ws/test_ws.sh | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ws/test_ws.sh b/ws/test_ws.sh
index 198559f3..cd9d7155 100644
--- a/ws/test_ws.sh
+++ b/ws/test_ws.sh
@@ -14,6 +14,7 @@ function cleanup_and_exit {
 }
 
 WITH_TLS=${WITH_TLS:-0}
+BLOCKS=${BLOCKS:-20000}
 
 rm -rf /tmp/ws_test
 mkdir -p /tmp/ws_test
@@ -57,7 +58,7 @@ ws receive "${protocol}127.0.0.1:8090" ${delay} --pidfile /tmp/ws_test/pidfile.r
 
 mkdir -p /tmp/ws_test/send
 cd /tmp/ws_test/send
-dd if=/dev/urandom of=/tmp/ws_test/send/20M_file count=20000 bs=1024
+dd if=/dev/urandom of=/tmp/ws_test/send/20M_file count=$BLOCKS bs=1024
 
 # Start the sender job
 ws send ${client_tls} --pidfile /tmp/ws_test/pidfile.send "${protocol}127.0.0.1:8090" /tmp/ws_test/send/20M_file