run-test.sh

#!/bin/bash

# Uncomment this for verbose output
# set -x

TEST_IMAGE=$1
CID_FILE=`date +%s`$$.cid

remove_container_id_file () {
  rm -f $CID_FILE
}

remove_node_modules () {
  rm -rf test/node_modules
  rm -f test/package-lock.json
  rm -rf test/.npm
}

test_probe () {
  probe_path=$1
  echo "Testing probe path ${probe_path}"
  EXPECTED='OK'
  RESPONSE=$(curl "http://localhost:8080${probe_path}")
  if [ "${RESPONSE}" == "${EXPECTED}" ] ; then
    echo "Got expected response from probe ${probe_path} '${RESPONSE}'"
  else
    echo "Unexepected response received from probe ${probe_path} '${RESPONSE}'"
    exit 1
  fi
}

test_http () {
  echo "Testing runtime HTTP function"
  EXPECTED='{"message":"This is the test function for Node.js FaaS. Success."}'
  RESPONSE=$(curl http://localhost:8080)

  if [ "${RESPONSE}" == "${EXPECTED}" ] ; then
    echo "Got expected response '${RESPONSE}'"
  else
    echo "Unexepected response received from function '${RESPONSE}'"
    fail
  fi
}

test_events () {
  echo "Testing cloud events"
  EXPECTED='{"message":"Hello eventing!"}'
  RESPONSE=$(curl -d "{\"message\": \"Hello eventing!\"}" \
-H "ce-type: dev.knative.cronjob.event" \
-H "ce-id: 115fccc4-d65e-4e2d-a669-8f39b0672f6e" \
-H "ce-knativehistory: testfunc-channel-8gxnw.js-faas-demo.svc.cluster.local" \
-H "ce-source: /apis/v1/namespaces/js-faas-demo/cronjobsources/testfunc-cronjob-source" \
-H "ce-specversion: 0.2" \
-H "ce-time: 2019-09-19T15:10:00.000412129Z" \
-H "content-type: application/json; charset=utf-8" \
-H "content-length: 30" \
http://localhost:8080)

  if [ "${RESPONSE}" == "${EXPECTED}" ] ; then
    echo "Got expected response '${RESPONSE}'"
  else
    echo "Unexepected response received from function '${RESPONSE}'"
    fail
  fi
}

cleanup () {
  echo "Stopping ${TEST_IMAGE}..."
  docker stop $(cat ${CID_FILE})

  echo "Removing generated test files..."
  remove_container_id_file
  remove_node_modules
}

wait_for_cid() {
  wait_for_file $CID_FILE
}

wait_for_file() {
  local max_attempts=10
  local sleep_time=1
  local attempt=1
  local result=1
  while [ $attempt -le $max_attempts ]; do
    [ -f $1 ] && [ -s $1 ] && break
    echo "Waiting for $1..."
    attempt=$(( $attempt + 1 ))
    sleep $sleep_time
  done
}

fail () {
  cleanup
  exit 1
}

echo "Testing ${TEST_IMAGE}"
echo "Building image..."
docker build -t ${TEST_IMAGE} .

echo "Starting container for ${TEST_IMAGE}..."
echo "docker run --rm --cidfile ${CID_FILE} -a stdout -a stderr -v $(pwd)/test:/home/node/usr -p 8080:8080 ${TEST_IMAGE} &"
docker run --rm --cidfile ${CID_FILE} -a stdout -a stderr -v $(pwd)/test:/home/node/usr -p 8080:8080 ${TEST_IMAGE} &

echo "Giving it a few seconds to initialize..."
wait_for_cid
wait_for_file "test/package-lock.json"

test_http
test_events
test_probe "/health/readiness"
test_probe "/health/liveness"

echo "Test success. Cleaning up..."
cleanup

echo "Done."