| #!/usr/bin/env bash |
| |
| RED='\033[0;31m' |
| GREEN='\033[0;32m' |
| NC='\033[0m' # No Color |
| [ "$(uname -s)" == "Darwin" ] && { realpath(){ echo "$(cd $(dirname $1);pwd -P)/$(basename $1)"; }; } |
| AIDEGEN_DIR=$(dirname $(realpath $0)) |
| ASUITE_DIR="$(dirname $AIDEGEN_DIR)" |
| CORE_DIR="$(dirname $ASUITE_DIR)/tradefederation/core" |
| ATEST_DIR="$CORE_DIR/atest" |
| rc_file=${AIDEGEN_DIR}/.coveragerc |
| |
| function get_python_path() { |
| echo "$PYTHONPATH:$CORE_DIR:$ATEST_DIR:$ASUITE_DIR" |
| } |
| |
| function print_summary() { |
| local test_results=$1 |
| local tmp_dir=$(mktemp -d) |
| PYTHONPATH=$(get_python_path) python3 -m coverage report |
| PYTHONPATH=$(get_python_path) python3 -m coverage html -d $tmp_dir --rcfile=$rc_file |
| echo "coverage report available at file://${tmp_dir}/index.html" |
| |
| if [[ $test_results -eq 0 ]]; then |
| echo -e "${GREEN}All unittests pass${NC}!" |
| else |
| echo -e "${RED}Unittest failure found${NC}!" |
| fi |
| } |
| |
| function run_unittests() { |
| local specified_tests=$@ |
| local rc=0 |
| |
| # Get all unit tests under tools/acloud. |
| local all_tests=$(find $AIDEGEN_DIR -type f -name "*_unittest.py"); |
| local tests_to_run=$all_tests |
| |
| PYTHONPATH=$(get_python_path) python3 -m coverage erase |
| for t in $tests_to_run; |
| do |
| echo "Test" $t |
| if ! PYTHONPATH=$(get_python_path) python3 -m coverage run --append --rcfile=$rc_file $t; then |
| rc=1 |
| echo -e "${RED}$t failed${NC}" |
| fi |
| done |
| |
| print_summary $rc |
| cleanup |
| } |
| |
| function check_env() { |
| if [ -z "$ANDROID_BUILD_TOP" ]; then |
| echo "Missing ANDROID_BUILD_TOP env variable. Run 'lunch' first." |
| exit 1 |
| fi |
| } |
| |
| function cleanup() { |
| # Search for *.pyc and delete them. |
| find $AIDEGEN_DIR -name "*.pyc" -exec rm -f {} \; |
| |
| # Delete the generated .coverage files too. |
| find $ASUITE_DIR -name "*.coverage" -exec rm -f {} \; |
| } |
| |
| check_env |
| cleanup |
| run_unittests "$@" |