#!/bin/bash
#
# This script runs all tests for the root CDK project, as well as any microservices, Lambda functions, or dependency
# source code packages. These include unit tests, integration tests, and snapshot tests.
#
# This script is called by the ../initialize-repo.sh file and the buildspec.yml file. It is important that this script
# be tested and validated to ensure that all available test fixtures are run.
#
# The if/then blocks are for error handling. They will cause the script to stop executing if an error is thrown from the
# node process running the test case(s). Removing them or not using them for additional calls with result in the
# script continuing to execute despite an error being thrown.

[ "$DEBUG" == 'true' ] && set -x
set -e

setup_python_env() {
	if [ -d "./.venv-test" ]; then
		echo "Reusing already setup python venv in ./.venv-test. Delete ./.venv-test if you want a fresh one created."
		return
	fi

  echo "Setting up python venv"
	python3 -m venv .venv-test
	echo "Initiating virtual environment"
	source .venv-test/bin/activate

    echo "Installing python packages"
    # install test dependencies in the python virtual environment
	pip3 install -r test/requirements-test.txt
	# pip3 install -r requirements.txt --target .

	echo "deactivate virtual environment"
	deactivate
}

run_python_test() {
	local component_path=$1
	local component_name=$2

	echo "------------------------------------------------------------------------------"
	echo "[Test] Run python unit test with coverage for $component_path $component_name"
	echo "------------------------------------------------------------------------------"
	cd $component_path

	if [ "${CLEAN:-true}" = "true" ]; then
        rm -fr .venv-test
    fi

	setup_python_env

	echo "Initiating virtual environment"
	source .venv-test/bin/activate

	# setup coverage report path
	mkdir -p $source_dir/tests/coverage-reports
	coverage_report_path=$source_dir/tests/coverage-reports/$component_name.coverage.xml
	echo "coverage report path set to $coverage_report_path"

	# Use -vv for debugging
	python3 -m pytest --cov --cov-report=term-missing --cov-report "xml:$coverage_report_path"

    # The pytest --cov with its parameters and .coveragerc generates a xml cov-report with `coverage/sources` list
    # with absolute path for the source directories. To avoid dependencies of tools (such as SonarQube) on different
    # absolute paths for source directories, this substitution is used to convert each absolute source directory
    # path to the corresponding project relative path. The $source_dir holds the absolute path for source directory.
    sed -i -e "s,<source>$source_dir,<source>source,g" $coverage_report_path

	echo "deactivate virtual environment"
	deactivate

	if [ "${CLEAN:-true}" = "true" ]; then
		rm -fr .venv-test
		rm .coverage
		rm -fr .pytest_cache
		rm -fr __pycache__ test/__pycache__
	fi
}

prepare_jest_coverage_report() {
	local component_name=$1

    if [ ! -d "coverage" ]; then
        echo "ValidationError: Missing required directory coverage after running unit tests"
        exit 129
    fi

	# prepare coverage reports
    rm -fr coverage/lcov-report
    mkdir -p $coverage_reports_top_path/jest
    coverage_report_path=$coverage_reports_top_path/jest/$component_name
    rm -fr $coverage_report_path
    mv coverage $coverage_report_path
}

run_javascript_test() {
    local component_path=$1
	local component_name=$2

    echo "------------------------------------------------------------------------------"
    echo "[Test] Run javascript unit test with coverage for $component_path $component_name"
    echo "------------------------------------------------------------------------------"
    echo "cd $component_path"
    cd $component_path

	# install and build for unit testing
	npm install

    # run unit tests
    npm run test

    # prepare coverage reports
	prepare_jest_coverage_report $component_name
}

run_cdk_project_test() {
	local component_path=$1
  local component_name=solutions-constructs

	echo "------------------------------------------------------------------------------"
	echo "[Test] $component_name"
	echo "------------------------------------------------------------------------------"
    cd $component_path

	# install and build for unit testing
	npm install

	## Option to suppress the Override Warning messages while synthesizing using CDK
	# export overrideWarningsEnabled=false

	# run unit tests
	npm run test -- -u

    # prepare coverage reports
	prepare_jest_coverage_report $component_name
}

# Get reference for source folder
source_dir="$(cd $PWD; pwd -P)"
echo $source_dir

# Download MaxMind database
if [ ! -e $source_dir/constructs/lambda/plugin/standard/assets/GeoLite2-City.mmdb ]; then
  echo "Download MaxMind database file"
  curl --create-dirs -o $source_dir/constructs/lambda/plugin/standard/assets/GeoLite2-City.mmdb https://aws-gcr-solutions-assets.s3.amazonaws.com/maxmind/GeoLite2-City.mmdb
fi

# Run unit tests
echo "Running unit tests"

coverage_reports_top_path=$source_dir/test/coverage-reports

portal_dir=$source_dir/portal
cd $portal_dir
npm install
npm run build

construct_dir=$source_dir/constructs

cd $construct_dir

rm -vf ./lambda/pipeline/service/log-processor/boto3_client.py

rm -vf ./lambda/api/app_pipeline/common.py
rm -vf ./lambda/api/app_log_ingestion/common.py
rm -vf ./lambda/api/app_log_ingestion/util/aws_svc_mgr.py
rm -vf ./lambda/api/app_log_ingestion/aws_svc_mgr.py
rm -vf ./lambda/api/pipeline/aws_svc_mgr.py
rm -vf ./lambda/api/log_agent_status/aws_svc_mgr.py
rm -vf ./lambda/api/instance_meta/aws_svc_mgr.py
rm -vf ./lambda/api/instance_group/aws_svc_mgr.py
rm -vf ./lambda/api/resource/aws_svc_mgr.py
rm -vf ./lambda/api/eks_cluster/aws_svc_mgr.py
rm -vf ./lambda/main/cfnHelper/aws_svc_mgr.py


cp -vf ./lambda/pipeline/service/log-processor/../../common/custom-resource/boto3_client.py ./lambda/pipeline/service/log-processor/boto3_client.py
cp -vf ./lambda/api/app_pipeline/../common/common.py ./lambda/api/app_pipeline/common.py
cp -vf ./lambda/api/app_log_ingestion/../common/common.py ./lambda/api/app_log_ingestion/common.py
cp -vf ./lambda/api/pipeline/../common/aws_svc_mgr.py ./lambda/api/pipeline/aws_svc_mgr.py
cp -vf ./lambda/api/instance_group/../common/aws_svc_mgr.py ./lambda/api/instance_group/aws_svc_mgr.py
cp -vf ./lambda/api/app_log_ingestion/../common/aws_svc_mgr.py ./lambda/api/app_log_ingestion/aws_svc_mgr.py
cp -vf ./lambda/api/log_agent_status/../common/aws_svc_mgr.py ./lambda/api/log_agent_status/aws_svc_mgr.py
cp -vf ./lambda/api/instance_meta/../common/aws_svc_mgr.py ./lambda/api/instance_meta/aws_svc_mgr.py
cp -vf ./lambda/api/resource/../common/aws_svc_mgr.py ./lambda/api/resource/aws_svc_mgr.py
cp -vf ./lambda/api/eks_cluster/../common/aws_svc_mgr.py ./lambda/api/eks_cluster/aws_svc_mgr.py
cp -vf ./lambda/main/cfnHelper/../../api/common/aws_svc_mgr.py ./lambda/main/cfnHelper/aws_svc_mgr.py

# Add test_aws_svc_mgr.py for UT
cp -vf ./lambda/api/eks_cluster/../common/test/test_aws_svc_mgr.py ./lambda/api/eks_cluster/test/test_aws_svc_mgr.py
cp -vf ./lambda/api/resource/../common/test/test_aws_svc_mgr.py ./lambda/api/resource/test/test_aws_svc_mgr.py
cp -vf ./lambda/api/pipeline/../common/test/test_aws_svc_mgr.py ./lambda/api/pipeline/test/test_aws_svc_mgr.py
cp -vf ./lambda/api/instance_group/../common/test/test_aws_svc_mgr.py ./lambda/api/instance_group/test/test_aws_svc_mgr.py
cp -vf ./lambda/api/app_log_ingestion/../common/test/test_aws_svc_mgr.py ./lambda/api/app_log_ingestion/test/test_aws_svc_mgr.py
cp -vf ./lambda/api/log_agent_status/../common/test/test_aws_svc_mgr.py ./lambda/api/log_agent_status/test/test_aws_svc_mgr.py
cp -vf ./lambda/api/instance_meta/../common/test/test_aws_svc_mgr.py ./lambda/api/instance_meta/test/test_aws_svc_mgr.py
cp -vf ./lambda/main/cfnHelper/../../api/common/test/test_aws_svc_mgr.py ./lambda/main/cfnHelper/test/test_aws_svc_mgr.py

# Test the CDK project
run_cdk_project_test $construct_dir

# Test the attached Lambda function
run_python_test $construct_dir/lambda/pipeline/service/log-processor svc-log-processor
run_python_test $construct_dir/lambda/pipeline/app/log-processor app-log-processor
run_python_test $construct_dir/lambda/pipeline/common/opensearch-helper opensearch-helper
run_python_test $construct_dir/lambda/plugin/standard plugin
run_python_test $construct_dir/lambda/api/resource resource-api
run_python_test $construct_dir/lambda/api/instance_meta instance-meta-api
run_python_test $construct_dir/lambda/api/pipeline svc-pipeline-api
run_python_test $construct_dir/lambda/api/log_agent_status log_agent_status
run_python_test $construct_dir/lambda/main/cfnHelper cfnHelper
run_python_test $construct_dir/lambda/main/sfnHelper sfnHelper
run_python_test $construct_dir/lambda/custom-resource custom-resource
run_python_test $construct_dir/lambda/api/app_log_ingestion app_log_ingestion
run_python_test $construct_dir/lambda/api/eks_cluster eks_cluster
run_python_test $construct_dir/lambda/api/log_source log_source
run_python_test $construct_dir/lambda/api/instance_group instance_group
run_python_test $construct_dir/lambda/api/log_conf log_conf
run_python_test $construct_dir/lambda/api/app_pipeline app_pipeline
run_python_test $construct_dir/lambda/api/cross_account cross_account
run_python_test $construct_dir/lambda/api/common common
run_python_test $construct_dir/lambda/api/cluster aos_cluster
run_python_test $construct_dir/lambda/pipeline/common/custom-resource custom-resource2
run_python_test $construct_dir/lambda/api/app_pipeline_flow app_pipeline_flow

# Remove test_aws_svc_mgr.py
echo "Remove UT file"
rm -rf $construct_dir/lambda/api/eks_cluster/test/test_aws_svc_mgr.py
rm -rf $construct_dir/lambda/api/resource/test/test_aws_svc_mgr.py
rm -rf $construct_dir/lambda/api/pipeline/test/test_aws_svc_mgr.py
rm -rf $construct_dir/lambda/api/instance_group/test/test_aws_svc_mgr.py
rm -rf $construct_dir/lambda/api/app_log_ingestion/test/test_aws_svc_mgr.py
rm -rf $construct_dir/lambda/api/log_agent_status/test/test_aws_svc_mgr.py
rm -rf $construct_dir/lambda/api/instance_meta/test/test_aws_svc_mgr.py
rm -rf $construct_dir/lambda/main/cfnHelper/test/test_aws_svc_mgr.py

# Return to the source/ level
cd $source_dir