test.sh

#!/bin/bash

source test.env
export FICAI_LISTEN FICAI_DB_HOST FICAI_DB_PORT FICAI_DB_USERNAME FICAI_DB_PASSWORD FICAI_DB_DATABASE FICAI_PWD_PEPPER FICAI_DOMAIN FICAI_BETA_KEY

TEST_TS="$( date +%s )"
TEST_EMAIL1="${TEST_TS}.1@example.com"
TEST_EMAIL2="${TEST_TS}.2@example.com"
TEST_URL="https://forums.sufficientvelocity.com/threads/$TEST_TS/"

DEBUG=no

request() {
  local CURL_ARGS
  local CURL_RESULT
  if [[ "$DEBUG" == "yes" ]]; then
    CURL_ARGS="-v"
  else
    CURL_ARGS="-s"
  fi
  curl $CURL_ARGS -D "$SHUNIT_TMPDIR/headers" -o "$SHUNIT_TMPDIR/out" \
    --cookie test.cookies --cookie-jar test.cookies \
    "$@"
  CURL_RESULT="$?"
  if [[ "$DEBUG" == "yes" ]]; then
    show_headers
    show_output
    show_cookies
  fi
  return "$CURL_RESULT"
}

request_get() {
  request "http://$FICAI_LISTEN/v1/signals" \
    -G --data-urlencode "url=$TEST_URL"
}

build_patch_body() {
  local URL=$1
  shift
  local JSON="{\"url\":\"$URL\"}"
  local DST=""
  local TAG=""
  while [[ $# -gt 0 ]]; do
    DST=""
    case "$1" in
      +*) DST="add";;
      -*) DST="rm";;
      %*) DST="erase";;
      *)
        echo "unexpected patch argument: $1" >&2
        return 1
    esac
    TAG="${1:1}"
    shift
    JSON="$( jq -c ". + {\"$DST\": (.$DST + [\"$TAG\"])}" <<<"$JSON" )"
  done
  echo "$JSON"
}

test_build_patch_body() {
  assertEquals \
    '{"url":"my-url","add":["a","b b","e"],"rm":["a","c"],"erase":["d"]}' \
    "$( build_patch_body my-url +a "+b b" -a -c %d +e )"
}

request_patch() {
  local JSON="$( build_patch_body "$@" )"
  request "http://$FICAI_LISTEN/v1/signals" \
    -X PATCH -H "Content-Type: application/json" --data-binary "$JSON"
}

show_headers() {
  cat "$SHUNIT_TMPDIR/headers"
}

show_output() {
  cat "$SHUNIT_TMPDIR/out"
}

show_cookies() {
  cat test.cookies
}

extractSignal() {
  <"$SHUNIT_TMPDIR/out" jq -r ".tags[]|select(.tag==\"$1\")"
}

assertSignal() {
  local TAG="$( extractSignal "$1" )"
  assertEquals "$1" "$2" "$(echo "$TAG" | jq -r '.signal')"
  assertEquals "$1" "$3" "$(echo "$TAG" | jq -r '.signalsFor')"
  assertEquals "$1" "$4" "$(echo "$TAG" | jq -r '.signalsAgainst')"
}

assertNoSignal() {
  assertEquals "$1" "" "$( extractSignal "$1" )"
}

oneTimeSetUp() {
  cargo build || return 1
  nohup "${CARGO_TARGET_DIR:-./target}/debug/ficai-signals-server" >test.log 2>&1 &
  echo $! >test.pid
  echo "server process pid: $(cat test.pid)"
  for i in {1..5} ; do
    sleep 0.1s
    pgrep -F test.pid >/dev/null && break
    [[ "$i" -eq 5 ]] && echo "tired of waiting for server to start" && exit 1
  done
  # todo: wait for server initialization
  sleep 1

  rm -f test.cookies
}

oneTimeTearDown() {
  # shunit2 2.1.8 runs teardown once more at the end of execution - see https://github.com/kward/shunit2/issues/112
  [[ ! -e test.pid ]] && return 0

  echo "taking down server process $(cat test.pid)..."
  pkill -F test.pid

  rm test.pid
}

headers_line() {
  head -n "$1" "$SHUNIT_TMPDIR/headers" | tail -n 1 | tr -d $'\r'
}

test404() {
  curl -s -D "$SHUNIT_TMPDIR/headers" -o "$SHUNIT_TMPDIR/out" "http://$FICAI_LISTEN/derp"
  assertEquals 'HTTP/1.1 404 Not Found' "$( headers_line 1 )"
}

testUnauthorizedGet() {
  request_get
  assertEquals 'HTTP/1.1 403 Forbidden' "$( headers_line 1 )"
  assertTrue "[[ ! -s \"$SHUNIT_TMPDIR/out\" ]]"
}

testCreateUserInvalidBetaKey() {
  request "http://$FICAI_LISTEN/v1/accounts" \
    -X POST -H "Content-Type: application/json" --data-binary "{\"email\":\"$TEST_EMAIL1\",\"password\":\"pass\",\"betaKey\":\"x$FICAI_BETA_KEY\"}"

  assertEquals 'HTTP/1.1 400 Bad Request' "$( headers_line 1 )"
  assertFalse "cookie must not be set" "grep -q FicAiSession test.cookies"
}

testCreateUser() {
  request "http://$FICAI_LISTEN/v1/accounts" \
    -X POST -H "Content-Type: application/json" --data-binary "{\"email\":\"$TEST_EMAIL1\",\"password\":\"pass\",\"betaKey\":\"$FICAI_BETA_KEY\"}"

  assertEquals 'HTTP/1.1 201 Created' "$( headers_line 1 )"
  assertTrue "cookie must be set" "grep -q FicAiSession test.cookies"
}

testCreateUserSecondTime() {
  request "http://$FICAI_LISTEN/v1/accounts" \
    -X POST -H "Content-Type: application/json" --data-binary "{\"email\":\"$TEST_EMAIL1\",\"password\":\"pass\",\"betaKey\":\"$FICAI_BETA_KEY\"}"
  assertEquals 'HTTP/1.1 409 Conflict' "$( headers_line 1 )"
  assertEquals 'account already exists' "$( show_output )"
}

testGetEmptySignals() {
  request_get
  assertEquals 'HTTP/1.1 200 OK' "$( headers_line 1 )"
  assertEquals '{"tags":[]}' "$(cat "$SHUNIT_TMPDIR/out")"
}

testAdd() {
  request_patch "$TEST_URL" +worm +taylor
  request_get
  assertEquals 'HTTP/1.1 200 OK' "$( headers_line 1 )"
  assertSignal worm true 1 0
  assertSignal taylor true 1 0
}

testRm() {
  request_patch "$TEST_URL" -taylor "+taylor hebert"
  request_get
  assertEquals 'HTTP/1.1 200 OK' "$( headers_line 1 )"
  assertSignal worm true 1 0
  assertSignal taylor false 0 1
  assertSignal "taylor hebert" true 1 0
}

testErase() {
  request_patch "$TEST_URL" %taylor
  request_get
  assertEquals 'HTTP/1.1 200 OK' "$( headers_line 1 )"
  assertSignal worm true 1 0
  assertSignal "taylor hebert" true 1 0
  assertNoSignal taylor
}

testLogIn() {
  rm test.cookies
  request "http://$FICAI_LISTEN/v1/sessions" \
    -X POST -H "Content-Type: application/json" --data-binary "{\"email\":\"$TEST_EMAIL1\",\"password\":\"pass\"}"

  assertEquals 'HTTP/1.1 204 No Content' "$( headers_line 1 )"
  assertTrue "cookie must be set" "grep -q FicAiSession test.cookies"
}

testLogInWithWrongEmail() {
  request "http://$FICAI_LISTEN/v1/sessions" \
    -X POST -H "Content-Type: application/json" --data-binary "{\"email\":\"$TEST_EMAIL2\",\"password\":\"pass\"}"

  assertEquals 'HTTP/1.1 403 Forbidden' "$( headers_line 1 )"
}

testLogInWithWrongPassword() {
  request "http://$FICAI_LISTEN/v1/sessions" \
    -X POST -H "Content-Type: application/json" --data-binary "{\"email\":\"$TEST_EMAIL1\",\"password\":\"wrong pass\"}"

  assertEquals 'HTTP/1.1 403 Forbidden' "$( headers_line 1 )"
}

source shunit2