98 lines
2.9 KiB
Bash
98 lines
2.9 KiB
Bash
|
#!/usr/bin/env bash
|
||
|
|
# Copyright 2019, The Android Open Source Project
|
||
|
|
#
|
||
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
# you may not use this file except in compliance with the License.
|
||
|
|
# You may obtain a copy of the License at
|
||
|
|
#
|
||
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
#
|
||
|
|
# Unless required by applicable law or agreed to in writing, software
|
||
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
# See the License for the specific language governing permissions and
|
||
|
|
# limitations under the License.
|
||
|
|
|
||
|
|
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)"
|
||
|
|
ATEST_DIR="$ASUITE_DIR/atest"
|
||
|
|
RC_FILE=${AIDEGEN_DIR}/.coveragerc
|
||
|
|
MOD_COVERAGE='coverage:import coverage'
|
||
|
|
MOD_PROTOBUF='protobuf:from google import protobuf'
|
||
|
|
|
||
|
|
function get_python_path() {
|
||
|
|
echo "$PYTHONPATH:$ASUITE_DIR:$ATEST_DIR"
|
||
|
|
}
|
||
|
|
|
||
|
|
function print_summary() {
|
||
|
|
local test_results=$1
|
||
|
|
local tmp_dir=$(mktemp -d)
|
||
|
|
python3 -m coverage report
|
||
|
|
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}!"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
function run_unittests() {
|
||
|
|
local specified_tests=$@
|
||
|
|
local rc=0
|
||
|
|
|
||
|
|
# Get all unit tests under asuite/aidegen.
|
||
|
|
local all_tests=$(find $AIDEGEN_DIR -type f -name "*_unittest.py");
|
||
|
|
local tests_to_run=$all_tests
|
||
|
|
|
||
|
|
python3 -m coverage erase
|
||
|
|
for t in $tests_to_run; do
|
||
|
|
echo "Testing" $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 install_module() {
|
||
|
|
local FLAG="--user"
|
||
|
|
local module=$(echo $1| awk -F: '{print $1}')
|
||
|
|
local action=$(echo $1| awk -F: '{print $2}')
|
||
|
|
if ! python3 -c "$action" 2>/dev/null; then
|
||
|
|
echo -en "Module $RED$module$NC is missing. Start to install..."
|
||
|
|
cmd="python3 -m pip install $module $FLAG"
|
||
|
|
eval "$cmd" >/dev/null || { echo Failed to install $module; exit 1; }
|
||
|
|
echo " Done."
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
function check_env() {
|
||
|
|
if [ -z "$ANDROID_BUILD_TOP" ]; then
|
||
|
|
echo "Missing ANDROID_BUILD_TOP env variable. Run 'lunch' first."
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
# Append more necessary modules if needed.
|
||
|
|
for mod in "$MOD_COVERAGE" "$MOD_PROTOBUF"; do
|
||
|
|
install_module "$mod"
|
||
|
|
done
|
||
|
|
}
|
||
|
|
|
||
|
|
function cleanup() {
|
||
|
|
# Search for *.pyc and delete them.
|
||
|
|
find $AIDEGEN_DIR -name "*.pyc" -exec rm -f {} \;
|
||
|
|
}
|
||
|
|
|
||
|
|
check_env
|
||
|
|
cleanup
|
||
|
|
run_unittests "$@"
|