From 120add0e82cd604ae93f469c2e28e9bdce045dd1 Mon Sep 17 00:00:00 2001
From: Michael Vines <mvines@gmail.com>
Date: Fri, 20 Jul 2018 17:01:36 -0700
Subject: [PATCH] Add support for a client node running continuous transactions
 on the net

---
 ci/testnet-deploy.sh | 48 +++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 47 insertions(+), 1 deletion(-)

diff --git a/ci/testnet-deploy.sh b/ci/testnet-deploy.sh
index b58c6f8ea7..080b72c30c 100755
--- a/ci/testnet-deploy.sh
+++ b/ci/testnet-deploy.sh
@@ -84,6 +84,14 @@ findVms "name=$leaderName"
   exit 1
 }
 
+echo "Client node:"
+findVms "name=$leaderName-client"
+clientVm=
+if [[ ${#vmlist[@]} = 2 ]]; then
+  clientVm=${vmlist[1]}
+  unset 'vmlist[1]'
+fi
+
 echo "Validator nodes:"
 findVms "name~^$leaderName-validator-"
 
@@ -96,7 +104,7 @@ if ! $ROLLING_UPDATE; then
     echo "--- Shutting down $vmName in zone $vmZone $nodePosition"
     gcloud compute ssh "$vmName" --zone "$vmZone" \
       --ssh-flag="-o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null" \
-      --command="echo sudo snap remove solana" &
+      --command="sudo snap remove solana" &
 
     if [[ $((count % 5)) = 0 ]]; then
       #  Slow down deployment to avoid triggering GCP login
@@ -110,6 +118,29 @@ if ! $ROLLING_UPDATE; then
   wait
 fi
 
+
+client_run() {
+  declare message=$1
+  declare cmd=$2
+  [[ -n $clientVm ]] || return 0;
+  vmName=${clientVm%:*}
+  vmZone=${clientVm#*:}
+  echo "--- $message $vmName in zone $vmZone"
+  gcloud compute ssh "$vmName" --zone "$vmZone" \
+    --ssh-flag="-o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null" \
+    --command="$cmd"
+}
+
+client_run \
+  "Shutting down" \
+  "\
+    set -x;
+    tmux list-sessions; \
+    tmux capture-pane -t solana -p; \
+    tmux kill-session -t solana; \
+    sudo snap remove solana; \
+  "
+
 echo "--- Refreshing leader"
 leader=true
 pids=()
@@ -188,4 +219,19 @@ echo "--- $publicUrl sanity test"
   USE_SNAP=1 ci/testnet-sanity.sh $publicUrl ${#vmlist[@]}
 )
 
+client_run \
+  "Starting client on " \
+  "\
+    set -x;
+    sudo snap install solana --$SOLANA_SNAP_CHANNEL --devmode; \
+    snap info solana; \
+    tmux new -s solana -d \" \
+        /snap/bin/solana.client-demo $SOLANA_NET_URL ${#vmlist[@]} --loop; \
+        echo Error: client-demo should never exit; \
+        bash \
+      \"; \
+    sleep 2; \
+    tmux capture-pane -t solana -p -S -100 \
+  "
+
 exit 0