#!/bin/bash

# This file is used to run integration tests for developers(self-test).

set -eu

OUT_DIR=/tmp/tidb_cdc_test
CUR=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
source "$CUR/_utils/test_prepare"
export PATH=$PATH:$CUR/_utils:$CUR/../bin:$CUR/../../bin:$CUR/../../scripts/bin
export TICDC_NEWARCH=true
export CDC_BINARY=cdc.test
export NEXT_GEN=${NEXT_GEN:-0}

mkdir -p $OUT_DIR || true

if [ "${1-}" = '--debug' ]; then
	WORK_DIR=$OUT_DIR/debug
	trap 'stop_test $WORK_DIR' EXIT
	rm -rf $WORK_DIR && mkdir -p $WORK_DIR

	PATH="$CUR/../bin:$CUR/_utils:$PATH" \
		LD_LIBRARY_PATH="$CUR/../bin:$CUR/_utils:$PATH" \
		OUT_DIR=$OUT_DIR \
		TEST_NAME="debug" \
		start_tidb_cluster --workdir $WORK_DIR

	cdc server --log-file $WORK_DIR/cdc.log --log-level debug --addr 127.0.0.1:8300 >$WORK_DIR/stdout.log 2>&1 &
	sleep 1
	cdc_cli_changefeed create --sink-uri="mysql://normal:123456@127.0.0.1:3306/"

	echo 'You may now debug from another terminal. Press [ENTER] to exit.'
	read line
	exit 0
fi

run_case() {
	# cleanup test binaries and data, preserve logs, if we debug one case,
	# these files will be preserved since no more case will be run.
	find /tmp/tidb_cdc_test/*/* -type d | xargs rm -rf || true
	local case=$1
	local script=$2
	local sink_type=$3
	echo "=================>> Running test $case $script using Sink-Type: $sink_type... <<================="
	local start_ts
	local end_ts
	local duration
	start_ts=$(date +%s)
	PATH="$CUR/../bin:$CUR/_utils:$PATH" \
		LD_LIBRARY_PATH="$CUR/../bin:$CUR/_utils:$PATH" \
		OUT_DIR=$OUT_DIR \
		TEST_NAME=$case \
		bash "$script" "$sink_type"
	end_ts=$(date +%s)
	duration=$((end_ts - start_ts))
	echo "=================>> Finished test $case $script in ${duration}s <<================="
}

sink_type=$1

set +eu
test_case=$2

if [ -z "$test_case" ]; then
	test_case="*"
fi

start_at=$3
run_test="no"
if [ -z "$start_at" ]; then
	run_test="yes"
else
	test_case="*"
fi

# Print environment variables.
env
set -eu

if [ "$test_case" == "*" ]; then
	for script in $CUR/*/run.sh; do
		test_name="$(basename "$(dirname "$script")")"
		if [ "$run_test" == "yes" ] || [ "$start_at" == "$test_name" ]; then
			run_test="yes"
			run_case $test_name $script $sink_type
		fi
	done
else
	for name in $test_case; do
		script="$CUR/$name/run.sh"
		run_case $name $script $sink_type
	done
fi

# with color
echo "\033[0;36m<<< Run all test success >>>\033[0m"