source test-header.sh

TESTLIB=${TESTLIB:-C:/Python26/Lib/test}

PYARMOR="${PYTHON} pyarmor.py"

TESTROOT=$(pwd)

ALLENTRIES="test_concurrent_futures.py test_multiprocessing_forkserver.py test_weakref.py"

RUNTIMEFILES="pytransform.py _pytransform.* pyshield.* product.key license.lic"

csih_inform "Python is $PYTHON"

csih_inform "Tested Package: $pkgfile"

csih_inform "PyArmor is $PYARMOR"

csih_inform "Make workpath ${workpath}"

rm -rf ${workpath}

mkdir -p ${workpath} || csih_error "Make workpath FAILED"

cd ${workpath}

[[ ${pkgfile} == *.zip ]] && unzip ${pkgfile} > /dev/null 2>&1

[[ ${pkgfile} == *.tar.bz2 ]] && tar xjf ${pkgfile}

cd pyarmor-$version || csih_error "Invalid pyarmor package file"

[[ -d src ]] && cd src/

csih_inform "Add execute permission to dynamic library"

[[ -d platforms/windows32 ]] && chmod +x platforms/windows32/_pytransform.dll

[[ -d platforms/windows64 ]] && chmod +x platforms/windows64/_pytransform.dll

csih_inform "Prepare for python features testing at $(pwd)"

echo ""

csih_inform "Copy $TESTLIB to lib/test"

[[ -d $TESTLIB ]] || csih_error "No test lib found: '$TESTLIB'"

mkdir -p ./lib

cp -a $TESTLIB ./lib

csih_inform "Find test entries"

TESTENTRIES="regrtest.py"

for s in $ALLENTRIES ; do

[[ -f $TESTLIB/$s ]] && TESTENTRIES="$TESTENTRIES,$s"

echo "$TESTENTRIES"

csih_inform "Move $TESTLIB to ${TESTLIB}.bak"

[[ -d $TESTLIB.bak ]] && csih_error "$TESTLIB.bak has been exists!"

mv $TESTLIB $TESTLIB.bak

if [[ "$PLATFORM" == win* && "$PYTHON" == *Python26* ]] ; then

csih_inform "Convert scripts to unix format"

which dos2unix >/dev/null 2>&1 && \

for s in $(find ./lib/test -name test_*.py) ; do

dos2unix $s >>result.log 2>&1

done

echo ""

echo "-------------------- Start testing ------------------------------"

echo ""

csih_inform "Show help and import pytransform"

$PYARMOR --help >>result.log 2>&1 || csih_error "Bootstrap FAILED"

csih_inform "Create project at projects/pytest"

$PYARMOR init --src=lib/test --entry="$TESTENTRIES" projects/pytest >>result.log 2>&1

csih_inform "Change current path to projects/pytest"

cd projects/pytest

csih_inform "Config project to obfuscate all test scripts"

$ARMOR config --manifest="global-include test_*.py, exclude doctest_*.py" >result.log 2>&1

csih_inform "Obfuscate scripts"

$ARMOR build >>result.log 2>&1

csih_inform "Copy runtime files to ../../lib/test"

cp dist/test/* ../../lib/test

csih_inform "Copy runtime files to $TESTLIB/.."

cp dist/* $TESTLIB/..

csih_inform "Copy entry script to ../../lib/test"

cp dist/test/regrtest.py ../../lib/test

csih_inform "Modify entry script, add extra path"

BOOSTRAPCODE="import sys\ntry:\n from pytransform import pyarmor_runtime\nexcept ImportError:\n sys.path.append(r'$TESTLIB')\n from pytransform import pyarmor_runtime"

sed -i -e "s?from pytransform import pyarmor_runtime?$BOOSTRAPCODE?g" ../../lib/test/regrtest.py

csih_inform "Copy obfuscated scripts to ../../lib/test"

for s in $(find dist/test -name test_*.py) ; do

cp $s ${s/dist\/test\//..\/..\/lib\/test\/}

csih_inform "Move ../../lib/test to $TESTLIB"

mv ../../lib/test $TESTLIB

NOTESTS="test_argparse test_profilehooks test_sys_setprofile test_sys_settrace test_cprofile"

csih_inform "Run obfuscated test scripts without $NOTESTS"

(cd $TESTLIB; $PYTHON regrtest.py -x $NOTESTS)

csih_inform "Move obfuscated test scripts to ../../lib/test"

mv $TESTLIB ../../lib/test

csih_inform "Restore original test scripts"

mv $TESTLIB.bak $TESTLIB

csih_inform "Remove runtime files from $TESTLIB/.."

(cd ${TESTLIB}/..; rm -f ${RUNTIMEFILES})

csih_inform "Change current path to test root: ${TESTROOT}"

cd ${TESTROOT}

echo "----------------------------------------------------------------"

echo ""

csih_inform "Test finished for ${PYTHON}"

(( ${_bug_counter} == 0 )) || csih_error "${_bug_counter} bugs found"

echo "" && \

csih_inform "Remove workpath ${workpath}" \

&& echo "" \

&& rm -rf ${workpath} \

&& csih_inform "Congratulations, there is no bug found"