#!/bin/bash -e
#
# Perform a quick sanity test on the specific testnet
#

cd "$(dirname "$0")/.."
source multinode-demo/common.sh

NET_URL=$1
if [[ -z $NET_URL ]]; then
  NET_URL=testnet.solana.com
fi

EXPECTED_NODE_COUNT=$2
if [[ -z $EXPECTED_NODE_COUNT ]]; then
  EXPECTED_NODE_COUNT=50
fi

echo "--- $NET_URL: verify ledger"
# Note: here we assume this script is actually running on the leader node...
sudo solana.ledger-tool --ledger /var/snap/solana/current/config/ledger verify

echo "--- $NET_URL: wallet sanity"
(
  set -x
  multinode-demo/test/wallet-sanity.sh $NET_URL
)

echo "--- $NET_URL: node count"
if [[ $NET_URL = testnet.solana.com ]]; then
  echo "TODO: Remove this block when a release > 0.7.0 is deployed"
else
  if [[ -n "$USE_SNAP" ]]; then
    # TODO: Merge client.sh functionality into solana-bench-tps proper and
    #       remove this USE_SNAP case
    cmd=$solana_bench_tps
  else
    cmd=multinode-demo/client.sh
  fi

  (
    set -x
    $cmd $NET_URL $EXPECTED_NODE_COUNT -c
  )
fi

# TODO: sanity test with validator
# echo "--- $NET_URL: validator sanity"
# (
#   export USE_SNAP=1
#   ./multinode-demo/setup.sh -t validator
#   set -e pipefail
#   timeout 10s ./multinode-demo/validator.sh "$NET_URL" 2>&1 | tee log
# )
#
# (
#   set +e
#   panic=$(timeout 10s tail -f /var/snap/solana/current/validator/current | grep -C100 panic)
#
#   if [[ -n $panic ]]; then
#       echo Panic observed: "$panic"
#       exit 1
#   fi
# )

exit 0