#!/usr/bin/env -S bats --report-formatter junit --formatter tap -j 6
# -*-sh-*-

source ./common/test_helper_functions.bash || exit 1
source ./common/functions || exit 1

if [ -s ./common/TEST_ENV ]; then
	source ./common/TEST_ENV
fi

setup_file() {
	PKG=DIMEMAS
	MODULE=dimemas
	TESTNAME="perf-tools/Dimemas"

	export PKG MODULE TESTNAME
}

setup() {
	OUTPUT="$(mktemp)"

	export OUTPUT
}

teardown() {
	rm -f "${OUTPUT}"
}

@test "[${TESTNAME}] Verify ${MODULE} module is loaded and matches rpm version (${LMOD_FAMILY_COMPILER}/${LMOD_FAMILY_MPI})" {
	module list "${MODULE}" | grep "1) ${MODULE}" >&"${OUTPUT}" || exit 1
	run grep "${MODULE}" "${OUTPUT}"
	assert_success

	# check version against rpm
	local rpm
	rpm=$(get_rpm_name "${MODULE}")
	local version
	version="$(rpm -q --queryformat='%{VERSION}\n' "${rpm}")"
	run cat "${OUTPUT}"
	assert_output "  1) ${MODULE}/${version}"
}

@test "[${TESTNAME}] Verify ${PKG}_DIR is defined and exists (${LMOD_FAMILY_COMPILER}/${LMOD_FAMILY_MPI})" {
	PKG_DIR="${PKG}_DIR"

	if [ -z "${!PKG_DIR}" ]; then
		flunk "${PKG}_DIR directory not defined"
	fi

	if [ ! -d "${!PKG_DIR}" ]; then
		flunk "directory ${!PKG_DIR} does not exist"
	fi
}

# ----------
# Binaries
# ----------

@test "[${TESTNAME}] Verify ${PKG}_BIN is defined and exists (${LMOD_FAMILY_COMPILER}/${LMOD_FAMILY_MPI})" {
	PKG_BIN="${PKG}_BIN"

	if [ -z "${!PKG_BIN}" ]; then
		flunk "${PKG}_BIN directory not defined"
	fi

	if [ ! -d "${!PKG_BIN}" ]; then
		flunk "directory ${!PKG_BIN} does not exist"
	fi
}

@test "[${TESTNAME}] Verify availability of prv2dim binary (${LMOD_FAMILY_COMPILER}/${LMOD_FAMILY_MPI})" {
	run command -v prv2dim
	assert_success
}

@test "[${TESTNAME}] Verify availability of Dimemas binary (${LMOD_FAMILY_COMPILER}/${LMOD_FAMILY_MPI})" {
	run command -v Dimemas
	assert_success
}

# Run Dimemas
@test "[${TESTNAME}] Run Dimemas simulation (${LMOD_FAMILY_COMPILER}/${LMOD_FAMILY_MPI})" {
	rm -f prediction.prv ./*.dim ./*.pcf

	# Convert Paraver trace file to Dimemas format
	run prv2dim lulesh2_27p.prv lulesh2_27p.dim
	assert_success

	run ls lulesh2_27p.dim
	assert_success

	run ls lulesh2_27p.pcf

	# Run Dimemas simulation
	run Dimemas -S 32K -p prediction.prv MN.128.1ppn.cfg
	assert_success

	run ls prediction.prv
	assert_success

	rm -f prediction.prv ./*.dim ./*.pcf
}
