buildcache/test_scripts/run_file_lock_stresstest.sh

#!/bin/bash

# Note: This script is expected to be run from the build folder.

total_success=true

function run_test {
  TESTFILE=/tmp/bc_file_lock_stresstest_data-$$

  rm -f "$TESTFILE"

  test_type="network share safe locks"
  if [[ "$1" = "true" ]] ; then
    test_type="allow local locks"
  fi

  # Run four instances of the stresstest in parallel.
  echo "Starting four concurrent processes (${test_type})..."
  pids=""
  for i in {1..4}; do
    base/file_lock_stresstest "$TESTFILE" $1 &
    pids+=" $!"
  done

  # Wait for all processes to finish.
  got_error=false
  for p in $pids ; do
    if ! wait $p ; then
      got_error=true
    fi
  done

  # Get the result and delete the file.
  DATA=$(cat "$TESTFILE")
  rm -f "$TESTFILE"

  # Delete the lock file (if any).
  rm -f "${TESTFILE}.lock"

  # Did we have an error exit status from any of the processes?
  if $got_error ; then
    echo "*** FAIL: At least one of the processes failed."
    exit 1
  fi

  # Check the data file contents.
  EXPECTED_DATA="4000"
  if [[ "${DATA}" = "${EXPECTED_DATA}" ]] ; then
    echo "The test passed!"
  else
    echo "*** FAIL: The count should be ${EXPECTED_DATA}, but is ${DATA}."
    total_success=false
  fi
}

# Without local locks.
run_test false

# With local locks.
run_test true

# Check the test result.
if $total_success ; then
  echo "All tests passed!"
  exit 0
fi

echo "*** FAIL: At least one of the tests failed."
exit 1