#!/usr/bin/env bash

set -e

#
# Run all tests (not including functional)
#   ./test-.sh
#   ./test-unit.sh -v
#
# Run tests for one package
#   PKG=./unit ./test-unit.sh
#   PKG=ssh ./test-unit.sh
#

ARTIFACT_DIR=${ARTIFACT_DIR:=/tmp/artifacts}

export GOBIN=${PWD}/bin:${GOBIN}

# Use deps from vendor dir.
export GOFLAGS="-mod=vendor"

# Invoke ./cover for HTML output
COVER=${COVER:-"-cover"}

# https://ci-operator-configresolver-ui-ci.apps.ci.l2s4.p1.openshiftapps.com/help#env
OPENSHIFT_CI=${OPENSHIFT_CI:=false}

TESTABLE="./pkg/... ./cmd/..."
FORMATTABLE=(cmd pkg)

# user has not provided PKG override
if [ -z "${PKG}" ]; then
	TEST=${TESTABLE}
	FMT=("${FORMATTABLE[@]}")

# user has provided PKG override
else
	# strip out slashes and dots from PKG=./foo/
	TEST=${PKG//\//}
	TEST=${TEST//./}

	# only run gofmt on packages provided by user
	FMT=("${TEST[@]}")
fi

# split TEST into an array and prepend repo path to each local package
read -ra split <<<"$TEST"
TEST=("${split[@]/#/github.com/openshift/console-operator/}")

echo "Running tests..."
if [ "$OPENSHIFT_CI" = true ]; then
    go test -v "${COVER}" "$@" "${TEST[@]}" 2>&1 | tee "$ARTIFACT_DIR/test.out"
    RESULT="${PIPESTATUS[0]}"
    go-junit-report < "$ARTIFACT_DIR/test.out" > "$ARTIFACT_DIR/junit.xml"
    if [ "$RESULT" -ne 0 ]; then
        exit 255
    fi
else
	echo "go test "${COVER}" "$@" "${TEST[@]}""
    go test "${COVER}" "${TEST[@]}"
fi

echo "Checking gofmt..."
fmtRes=$(gofmt -l "${FMT[@]}")
if [ -n "${fmtRes}" ]; then
	echo -e "gofmt checking failed:\n${fmtRes}"
	exit 255
fi

echo "Checking govet..."
vetRes=$(go vet "${TEST[@]}")
if [ -n "${vetRes}" ]; then
	echo -e "govet checking failed:\n${vetRes}"
	exit 255
fi

echo "Success"