#
# Copyright (c) 2020 Alibaba Group Holding Limited. All Rights Reserved.
# DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
#
# This code is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License version 2 only, as
# published by the Free Software Foundation. Alibaba designates this
# particular file as subject to the "Classpath" exception as provided
# by Oracle in the LICENSE file that accompanied this code.
#
# This code is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
# FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
# version 2 for more details (a copy is included in the LICENSE file that
# accompanied this code).
#
# You should have received a copy of the GNU General Public License version
# 2 along with this work; if not, write to the Free Software Foundation,
# Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
#

#!/usr/bin/env bash

#
# @test TestJGroupDebugMode.sh
# @requires os.family == "Linux"
# @requires os.arch == "amd64"
# @summary test debugging mode of JGroup native implementation
# @run shell/timeout=300 TestJGroupDebugMode.sh
#

set -x

if [ "${TESTSRC}" = "" ]
then
    TESTSRC=${PWD}
    echo "TESTSRC not set.  Using "${TESTSRC}" as default"
fi
echo "TESTSRC=${TESTSRC}"
FS=/
JAVA=${TESTJAVA}${FS}bin${FS}java
JAVAC=${TESTJAVA}${FS}bin${FS}javac
TEST_CLASS="Test"
TEST_OPTS="-XX:+MultiTenant -XX:+TenantCpuThrottling -XX:+UseG1GC"

# generate and compile Java snippet for testing
cat >> ${TEST_CLASS}.java << EOF
import com.alibaba.tenant.*;
public class ${TEST_CLASS} {
    public static void main(String[] args) throws TenantException {
        TenantConfiguration config = new TenantConfiguration()
            .limitCpuShares(1024);
        TenantContainer tenant = TenantContainer.create(config);
        tenant.run(()-> {
            // empty!
        });
    }
}
EOF

${JAVAC} -cp . ${TEST_CLASS}.java
if [ $? != 0 ]; then
    echo "Failed to compile ${TEST_CLASS}.java"
    exit 1
fi

# Test envrionment variable debugging options

unset JGROUP_DEBUG

# disable debugging
export JGROUP_DEBUG=""
if [ ! -z "$(${JAVA} ${TEST_OPTS} -cp ${PWD} ${TEST_CLASS} | grep 'cgroup initialized successfully')" ]; then
    echo "Failed in non-debug mode"
    exit 1
fi

# enable debugging
export JGROUP_DEBUG="TRue"
if [ -z "$(${JAVA} ${TEST_OPTS} -cp ${PWD} ${TEST_CLASS} | grep 'cgroup initialized successfully')" ]; then
    echo "Failed in debug mode"
    exit 1
fi

export JGROUP_DEBUG="trUe"
if [ -z "$(${JAVA} ${TEST_OPTS} -cp ${PWD} ${TEST_CLASS} | grep 'cgroup initialized successfully')" ]; then
    echo "Failed in debug mode"
    exit 1
fi

# Test Java property debugging options
if [ ! -z "$(${JAVA} ${TEST_OPTS} -cp ${PWD} ${TEST_CLASS} | grep 'Created group with standard controllers:')" ]; then
    echo "Failed in non-debug mode"
    exit 1
fi

if [ ! -z "$(${JAVA} ${TEST_OPTS} -Dcom.alibaba.tenant.debugJGroup=false -cp ${PWD} ${TEST_CLASS} | grep 'Created group with standard controllers')" ]; then
    echo "Failed in debug mode"
    exit 1
fi

# enable debugging
if [ -z "$(${JAVA} ${TEST_OPTS} -Dcom.alibaba.tenant.debugJGroup=tRue -cp ${PWD} ${TEST_CLASS} | grep 'Created group with standard controllers')" ]; then
    echo "Failed in debug mode"
    exit 1
fi

if [ -z "$(${JAVA} ${TEST_OPTS} -Dcom.alibaba.tenant.debugJGroup=trUE -cp ${PWD} ${TEST_CLASS} | grep 'Created group with standard controllers')" ]; then
    echo "Failed in debug mode"
    exit 1
fi