From 0ef521c5165f1acd2efb70d36676e580f6427fa1 Mon Sep 17 00:00:00 2001
From: Harry Ho <hh@m-labs.hk>
Date: Fri, 18 Jun 2021 15:49:21 +0800
Subject: [PATCH] Add auto test scripts for M-Labs and Creotech setups

---
 test_creotech | 37 +++++++++++++++++++++++++++++++++++++
 test_mlabs    | 51 +++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 88 insertions(+)
 create mode 100755 test_creotech
 create mode 100755 test_mlabs

diff --git a/test_creotech b/test_creotech
new file mode 100755
index 0000000..933b54d
--- /dev/null
+++ b/test_creotech
@@ -0,0 +1,37 @@
+#!/bin/sh
+
+# Simultaneously, start testing RP's network connectivity, while
+# powering on MCH and RP using ~/power_strip/ scripts
+# Display power-on time
+TIMESTAMP_PRETTY=$(date '+%Y-%m-%d %H:%M:%S')
+echo "MCH & RP powered on at $TIMESTAMP_PRETTY."
+# Test RP network connectivity
+# NOTE: Change $RP_HOST inside `rp_ping_test` for your own setup!
+./rp_ping_test &
+# Power on MCH 
+./mch_start &
+# Wait 2 minutes before measurement
+sleep 120
+
+# Get and plot Sayma data
+# (Assumption: numpy, matplotlib and scipy have been installed on local)
+# (Assumption: directory ./creotech-raw exists)
+# NOTE: Change the RP hostname and LV/HV arguments according to
+# which two channels you are measuring!
+./get_and_plot_sayma_data creotech-raw creotech-1 1 creotech-1 2 creotech-1:LV,LV &
+# Wait 1 minute before powering off MCH and RP
+sleep 60
+
+# Issue shutdown on RP, and wait until RP is disconnected
+# NOTE: Change $RP_HOST, $RP_USER & $RP_KEY inside
+# `rp_shutdown` for your own setup!
+./rp_shutdown &
+# Wait 30 seconds
+sleep 30
+
+# Simultaneously power off MCH and RP
+# Display power-off time
+TIMESTAMP_PRETTY=$(date '+%Y-%m-%d %H:%M:%S')
+echo "MCH & RP powered off at $TIMESTAMP_PRETTY."
+# Power off MCH
+./mch_stop
diff --git a/test_mlabs b/test_mlabs
new file mode 100755
index 0000000..96518fd
--- /dev/null
+++ b/test_mlabs
@@ -0,0 +1,51 @@
+#!/bin/sh
+
+# tnetplug host
+export TNETPLUG_HOST=192.168.1.31
+export TNETPLUG_PORT=3131
+
+# Smart USB hub host
+# (uhubctl must be installed on the host)
+export HOST=rpi-3
+export HOSTPORT=22
+# Hub port where the RedPitaya is connected
+export HUBPORT=4
+
+# Simultaneously, start testing RP's network connectivity, while
+# powering on MCH using tnetplug, and RP using uhubctl
+# Display power-on time
+TIMESTAMP_PRETTY=$(date '+%Y-%m-%d %H:%M:%S')
+echo "MCH & RP powered on at $TIMESTAMP_PRETTY."
+# Test RP network connectivity
+# NOTE: Change $RP_HOST inside `rp_ping_test` for your own setup!
+./rp_ping_test &
+# Power on MCH
+echo "A" | nc -W1 $TNETPLUG_HOST $TNETPLUG_PORT &
+# Power on RedPitaya connected to the smart USB hub
+# (assume that only 1 hub is connected)
+ssh -p $HOSTPORT $HOST "uhubctl -a on -p $HUBPORT" &
+# Wait 2 minutes before measurement
+sleep 120
+
+# Get and plot Sayma data
+# NOTE: Change the RP hostname and LV/HV arguments according to
+# which two channels you are measuring!
+nix-shell -p python3Packages.matplotlib python3Packages.numpy python3Packages.scipy --run "./get_and_plot_sayma_data mlabs-raw mlabs-1 1 mlabs-1 2 mlabs-1:LV,LV" &
+# Wait 1 minute before powering off MCH and RP
+sleep 60
+
+# Issue shutdown on RP, and wait until RP is disconnected
+# NOTE: Change $RP_HOST, $RP_USER & $RP_KEY inside
+# `rp_shutdown` for your own setup!
+./rp_shutdown &
+# Wait 30 seconds
+sleep 30
+
+# Simultaneously power off MCH and RP
+# Display power-off time
+TIMESTAMP_PRETTY=$(date '+%Y-%m-%d %H:%M:%S')
+echo "MCH & RP powered off at $TIMESTAMP_PRETTY."
+# Power off MCH
+echo "a" | nc -W1 $TNETPLUG_HOST $TNETPLUG_PORT &
+# Power off RedPitaya connected to the smart USB hub
+ssh -p $HOSTPORT $HOST "uhubctl -a off -p $HUBPORT"