#!/bin/bash
#This bash script aims to automate the test process of all Candela Technologies's test_* scripts in the lanforge-scripts directory. The script can be run 2 ways and may include (via user input) the "start_num" and "stop_num" variables to select which tests should be run.
# OPTION ONE: ./test_all_scripts.sh : this command runs all the scripts in the array "testCommands"
# OPTION TWO: ./test_all_scripts.sh 4 5 :  this command runs py-script commands (in testCommands array) that include the py-script options beginning with 4 and 5 (inclusive) in case function ret_case_num.
#Variables
NUM_STA=4
SSID_USED="jedway-wpa2-x2048-5-3"
PASSWD_USED="jedway-wpa2-x2048-5-3"
RADIO_USED="wiphy0"
SECURITY="wpa2"
COL_NAMES="name,tx_bytes,rx_bytes,dropped"

START_NUM=0
CURR_TEST_NUM=0
CURR_TEST_NAME="BLANK"
STOP_NUM=9

DATA_DIR="${TEST_DIR}"
REPORT_DIR="/home/lanforge/html-reports"
BASE_DIR="/home/lanforge/Documents/lanforge-scripts"
#set -vex

#Test array
testCommands=(
    ".$BASE_DIR/cpu_stats.py --duration 15"
    ".$BASE_DIR/py-scripts/example_security_connection.py --num_stations $NUM_STA --ssid jedway-wpa-1 --passwd jedway-wpa-1 --radio $RADIO_USED --security wpa --debug"
    ".$BASE_DIR/py-scripts/example_security_connection.py --num_stations $NUM_STA --ssid $SSID_USED --passwd $SSID_USED --radio $RADIO_USED --security wpa2 --debug"
    ".$BASE_DIR/py-scripts/example_security_connection.py --num_stations $NUM_STA --ssid jedway-wep-48 --passwd 0123456789 --radio $RADIO_USED --security wep --debug"
    ".$BASE_DIR/py-scripts/example_security_connection.py --num_stations $NUM_STA --ssid jedway-wpa3-1 --passwd jedway-wpa3-1 --radio $RADIO_USED --security wpa3 --debug"
    ".$BASE_DIR/py-scripts/sta_connect2.py --dut_ssid $SSID_USED --dut_passwd $PASSWD_USED --dut_security $SECURITY"
    #"./$BASE_DIR/py-scripts/test_fileio.py --macvlan_parent eth2 --num_ports 3 --use_macvlans --first_mvlan_ip 192.168.92.13 --netmask 255.255.255.0 --gateway 192.168.92.1" # Better tested on Kelly, where VRF is turned off
    ".$BASE_DIR/py-scripts/test_generic.py --radio $RADIO_USED --ssid $SSID_USED --passwd $PASSWD_USED --num_stations $NUM_STA --type lfping --dest 10.40.0.1 --security $SECURITY --debug"
    ".$BASE_DIR/py-scripts/test_generic.py --radio $RADIO_USED --ssid $SSID_USED --passwd $PASSWD_USED --num_stations $NUM_STA --type speedtest --speedtest_min_up 20 --speedtest_min_dl 20 --speedtest_max_ping 150 --security $SECURITY --debug"
    ".$BASE_DIR/py-scripts/testgroup.py --group_name group1 --add_group --list_groups --debug"
    ".$BASE_DIR/py-scripts/testgroup.py --group_name group1 --add_group --add_cx l3_test1,l3_test2 --remove_cx l3_test3 --list_groups --debug"
    ".$BASE_DIR/py-scripts/test_ipv4_connection.py --radio $RADIO_USED --num_stations $NUM_STA --ssid $SSID_USED --passwd $PASSWD_USED --security $SECURITY --debug"
    ".$BASE_DIR/py-scripts/test_ipv4_l4_urls_per_ten.py --radio $RADIO_USED --num_stations $NUM_STA --security $SECURITY --ssid $SSID_USED --passwd $PASSWD_USED --num_tests 1 --requests_per_ten 600 --target_per_ten 600 --debug"
    ".$BASE_DIR/py-scripts/test_ipv4_l4_wifi.py --radio $RADIO_USED --num_stations $NUM_STA --security $SECURITY --ssid $SSID_USED --passwd $PASSWD_USED --test_duration 15s --debug"
    ".$BASE_DIR/py-scripts/test_ipv4_l4.py --radio $RADIO_USED --num_stations 4 --security $SECURITY --ssid $SSID_USED --passwd $PASSWD_USED --test_duration 15s --debug"
    ".$BASE_DIR/py-scripts/test_ipv4_variable_time.py --radio $RADIO_USED --ssid $SSID_USED --passwd $PASSWD_USED --security $SECURITY --test_duration 15s --output_format excel --layer3_cols $COL_NAMES --debug"
    ".$BASE_DIR/py-scripts/test_ipv4_variable_time.py --radio $RADIO_USED --ssid $SSID_USED --passwd $PASSWD_USED --security $SECURITY --test_duration 15s --output_format csv --layer3_cols $COL_NAMES --debug"
    ".$BASE_DIR/py-scripts/test_ipv4_l4_ftp_upload.py --upstream_port eth1 --radio RADIO_USED --num_stations $NUM_STA --security $SECURITY --ssid $SSID_USED --debug"
    ".$BASE_DIR/py-scripts/test_ipv6_connection.py --radio $RADIO_USED --ssid $SSID_USED --passwd $PASSWD_USED --security $SECURITY --debug"
    ".$BASE_DIR/py-scripts/test_ipv6_variable_time.py --radio $RADIO_USED --ssid $SSID_USED --passwd $PASSWD_USED --security $SECURITY --test_duration 15s --cx_type tcp6 --debug"
    ".$BASE_DIR/py-scripts/test_l3_longevity.py --test_duration 4m --endp_type \"lf_tcp lf_udp mc_udp\" --tos \"BK VI\" --upstream_port eth1
    --radio \"radio==wiphy0 stations==4 ssid==$SSID_USED ssid_pw==$PASSWD_USED security==$SECURITY\"
    --radio \"radio==wiphy1 stations==4 ssid==$SSID_USED ssid_pw==$PASSWD_USED security==$SECURITY\""
    ".$BASE_DIR/py-scripts/test_l3_powersave_traffic.py --radio $RADIO_USED --ssid $SSID_USED --passwd $PASSWD_USED --security $SECURITY --debug"
    ".$BASE_DIR/py-scripts/test_l3_scenario_throughput.py -t 15 -sc BLANK"
    ".$BASE_DIR/py-scripts/test_status_msg.py --debug " #this is all which is needed to run
    #"./test_wanlink.py --debug"
    #".$BASE_DIR/py-scripts/ws_generic_monitor_test.py"
    ".$BASE_DIR/py-json/ws-sta-monitor.py --debug"
    ".$BASE_DIR/py-scripts/create_bridge.py --radio wiphy1 --upstream_port eth1 --target_device sta0000 --debug"
    ".$BASE_DIR/py-scripts/create_l3.py --radio wiphy1 --ssid $SSID_USED --passwd $PASSWD_USED --security $SECURITY --debug"
    ".$BASE_DIR/py-scripts/create_l4.py --radio wiphy1 --ssid $SSID_USED --passwd $PASSWD_USED --security $SECURITY --debug"
    ".$BASE_DIR/py-scripts/create_macvlan.py --radio wiphy1 --macvlan_parent eth1 --debug"
    ".$BASE_DIR/py-scripts/create_station.py --radio wiphy1 --ssid $SSID_USED --passwd $PASSWD_USED --security $SECURITY --debug"
    ".$BASE_DIR/py-scripts/create_vap.py --radio wiphy1 --ssid $SSID_USED --passwd $PASSWD_USED --security $SECURITY --debug"
    ".$BASE_DIR/py-scripts/wlan_capacity_calculator.py -sta 11abg -t Voice -p 48 -m 106 -e WEP -q Yes -b 1 2 5.5 11 -pre Long -s N/A -co G.711 -r Yes -c Yes"
    ".$BASE_DIR/py-scripts/wlan_capacity_calculator.py -sta 11n -t Voice -d 17 -ch 40 -gu 800 -high 9 -e WEP -q Yes -ip 5 -mc 42 -b 6 9 12 24 -m 1538 -co G.729 -pl Greenfield -cw 15 -r Yes -c Yes"
    ".$BASE_DIR/py-scripts/wlan_capacity_calculator.py -sta 11ac -t Voice -d 9 -spa 3 -ch 20 -gu 800 -high 1 -e TKIP -q Yes -ip 3 -mc 0 -b 6 12 24 54 -m 1518 -co Greenfield -cw 15 -rc Yes"
)
declare -A name_to_num
name_to_num=(
    ["example_security_connection"]=1
    ["test_ipv4_connection"]=2
    ["test_generic"]=3
    ["test_ipv4_l4_urls_per_ten"]=4
    ["test_ipv4_l4_wifi"]=5
    ["test_ipv4_l4"]=6
    ["test_ipv4_variable_time"]=7
    ["create_bridge"]=8
    ["create_l3"]=9
    ["create_l4"]=10
    ["create_macvlan"]=10
    ["create_station"]=11
    ["create_vap"]=12
    ["cpu_stats"]=13
    ["test_fileio"]=14
    ["testgroup"]=15
    ["test_ipv6_connection"]=16
    ["test_ipv6_variable_time"]=17
    ["test_l3_longevity"]=18
    ["test_l3_powersave_traffic"]=19
    ["test_l3_scenario_throughput"]=20
    ["test_status_msg"]=21
    ["test_wanlink"]=22
    ["../$BASE_DIR/py-json/wlan_theoretical_sta"]=23
    ["ws_generic_monitor_test"]=24
    ["sta_connect2"]=25
    ["wlan_capacity_calculator"]=26
)

function blank_db() {
    echo "Loading blank scenario..." >>~/test_all_output_file.txt
    ./scenario.py --load BLANK >>~/test_all_output_file.txt
    #check_blank.py
}
function echo_print() {
    echo "Beginning $CURR_TEST_NAME test..." >>~/test_all_output_file.txt
}
results=()
detailedresults=()
NOW=$(date +"%Y-%m-%d-%H-%M")
NOW="${NOW/:/-}"
TEST_DIR="/home/lanforge/report-data/${NOW}"
URL="../report-data/${NOW}"
mkdir "$TEST_DIR"
function run_test() {
    for i in "${testCommands[@]}"; do
        .$BASE_DIR/py-scripts/scenario.py --load BLANK
        NAME=$(cat /dev/urandom | tr -dc 'a-zA-Z0-9' | fold -w 32 | head -n 1)
        CURR_TEST_NAME=${i%%.py*}
        CURR_TEST_NAME=${CURR_TEST_NAME#./*}
        CURR_TEST_NUM="${name_to_num[$CURR_TEST_NAME]}"

        #if (( $CURR_TEST_NUM > $STOP_NUM )) || (( $STOP_NUM == $CURR_TEST_NUM )) && (( $STOP_NUM != 0 )); then
        #    exit 1
        #fi
        echo ""
        echo "Test $CURR_TEST_NUM: $CURR_TEST_NAME"

        if (( $CURR_TEST_NUM > $START_NUM )) || (( $CURR_TEST_NUM == $START_NUM )); then
            echo_print
            echo "$i"
            $i > "${TEST_DIR}/${NAME}.txt" 2> "${TEST_DIR}/${NAME}_stderr.txt"
            retval=$?
            grep -i fail "${TEST_DIR}/${NAME}.txt" && retval=1
            chmod 664 "${TEST_DIR}/${NAME}.txt"
            if (( $retval == 0 )); then
                results+=("<tr><td>${CURR_TEST_NAME}</td><td class='scriptdetails'>${i}</td>
                          <td class='success'>Success</td>
                          <td><a href=\"${URL}/${NAME}.txt\" target=\"_blank\">STDOUT</a></td>
                          <td></td></tr>")
            else
                results+=("<tr><td>${CURR_TEST_NAME}</td><td class='scriptdetails'>${i}</td>
                          <td class='failure'>Failure</td>
                          <td><a href=\"${URL}/${NAME}.txt\" target=\"_blank\">STDOUT</a></td>
                          <td><a href=\"${URL}/${NAME}_stderr.txt\" target=\"_blank\">STDERR</a></td></tr>")

            fi
        fi
    done
    echo $results
}
function check_args() {
    if [ ! -z $1 ]; then
        START_NUM=$1
    fi
    if [ ! -z $2 ]; then
        STOP_NUM=$2
    fi
}
function html_generator() {
    header="<html>
		<head>
		<title>Test All Scripts Results $NOW</title>
		<style>
		.success {
			background-color:green;
		}
		.failure {
			background-color:red;
		}
		table {
			border: 1px solid gray;
		}
		td {
			margin: 0;
			padding: 2px;
			font-family: 'Courier New',courier,sans-serif;
		}
		h1, h2, h3, h4 {
			font-family: 'Century Gothic',Arial,sans,sans-serif;
		}
		.scriptdetails {
			font-size: 10px;
		}
		</style>
		<script src=\"sortabletable.js\"></script>
		</head>
		<body>
		<h1>Test All Scripts Results</h1>
		<h4>$NOW</h4>
		<table border ='1' id='myTable2'>
		<tr>
        <th onclick=\"sortTable(0)\">Command Name</th>
        <th onclick=\"sortTable(1)\">Command</th>
        <th onclick=\"sortTable(2)\">Status</th>
        <th onclick=\"sortTable(3)\">STDOUT</th>
        <th onclick=\"sortTable(4)\">STDERR</th>
    </tr>
		"
    tail="</body>
		</html>"

    fname="/home/lanforge/html-reports/test_all_output_file-${NOW}.html"
    echo "$header"  >> $fname
    echo "${results[@]}"  >> $fname
    echo "</table>" >> $fname
    echo "$tail" >> $fname
}

#true >~/test_all_output_file.txt
check_args $1 $2
run_test
echo "${detailedresults}"
html_generator
#test generic and fileio are for macvlans