testCorrectness.sh

#!/usr/bin/env bash
set -euo pipefail
source bashUtilFunctions.sh

pids=()
write_urls=("https://mattermost.db.in.tum.de/cbdp202223/channels/url-shortener" "https://www.tum.de/en/about-tum" "https://chat.openai.com/")
write_urls_results=()
read_urls_fail=("www.read.operation/will/fail" "https://gitlab.db.in.tum.de/" "www.google.com")

function cleanup {
  # Kill all background processes
  for pid in "${pids[@]}"; do
    kill -s TERM "$pid" > /dev/null 2>&1
  done

  # Remove files
  rm -f loadBalancerOutput.txt
  rm -f raft*Output.txt
  rm -f raft_*.db
  rm -f dataset.csv

  if [[ $? -ne 0 ]]; then
    echo "incorrect"
  else
    echo "correct"
  fi
}

trap cleanup EXIT


# Spawn raft nodes
for i in {0..3}; do
  cmake-build-debug/src/raft "$i" >"raft${i}Output.txt" &
  pids+=($!)
  sleep 2
done

# Spawn the load balancer
cmake-build-debug/src/loadBalancer >"loadBalancerOutput.txt" &
pids+=($!)

# Wait for them to select leader
sleep 15

# Read (should return failed)
for url in "${read_urls_fail[@]}"; do
  output=$(cmake-build-debug/src/client "read" "$url")

  # Check if output has two rows
  if [ $(echo "$output" | wc -l) -ne 1 ]; then
    exit 1
  fi
done

# Write (should return success)
for url in "${write_urls[@]}"; do
  client_output=$(cmake-build-debug/src/client "write" "$url")
  short_url=$(check_write_request_output "$client_output" "$url")
  write_urls_results+=("$short_url")
done

# Wait leader to replicate log to other raft nodes
sleep 20

# Read (should return success)
for ((i = 0; i < ${#write_urls[@]}; i++)); do
  # input=long_url output=short_url
  output=$(cmake-build-debug/src/client "read" "${write_urls[i]}")
  check_read_request_output "$output" "${write_urls[i]}" "${write_urls_results[i]}"

  # input=short_url output=long_url
  output=$(cmake-build-debug/src/client "read" "${write_urls_results[i]}")
  check_read_request_output "$output" "${write_urls_results[i]}" "${write_urls[i]}"
done