#!/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

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