es-integration-test.sh

#!/bin/bash

set -euxf -o pipefail

usage() {
  echo $"Usage: $0 <es_version>"
  exit 1
}

check_arg() {
  if [ ! $# -eq 1 ]; then
    echo "ERROR: need exactly one argument"
    usage
  fi
}

setup_es() {
  local tag=$1
  local image=docker.elastic.co/elasticsearch/elasticsearch
  local params=(
    --rm
    --detach
    --publish 9200:9200
    --env "http.host=0.0.0.0"
    --env "transport.host=127.0.0.1"
    --env "xpack.security.enabled=false"
    --env "xpack.monitoring.enabled=false"
  )
  local cid=$(docker run ${params[@]} ${image}:${tag})
  echo ${cid}
}

setup_query() {
  local arch=$(go env GOARCH)
  local params=(
    --es.tls.enabled=false
    --es.version=7
    --es.server-urls=http://127.0.0.1:9200
    --query.bearer-token-propagation=true
  )
  SPAN_STORAGE_TYPE=elasticsearch ./cmd/query/query-linux-${arch} ${params[@]}
}

teardown_es() {
  local cid=$1
  docker kill ${cid}
}

teardown_query() {
  local pid=$1
  kill -9 ${pid}
}

build_query() {
  make build-crossdock-ui-placeholder
  GOOS=linux make build-query
}

run_integration_test() {
  local es_version=$1
  local cid=$(setup_es ${es_version})
  STORAGE=elasticsearch make storage-integration-test
  make index-cleaner-integration-test
  make index-rollover-integration-test
  teardown_es ${cid}
}

run_token_propagation_test() {
  build_query
  make test-compile-es-scripts
  setup_query &
  local pid=$!
  make token-propagation-integration-test
  teardown_query ${pid}
}

main() {
  check_arg "$@"

  echo "Executing integration test for elasticsearch $1"
  run_integration_test "$1"
  echo "Executing token propagation test"
  run_token_propagation_test
}

main "$@"