#!/bin/bash

source /home/scott/bin/color_vars.sh

ROOT=/home/scott/lib/python_modules
DOCTEST=0
UNITTEST=0
INTEGRATION=0
FAILURES=0

dup() {
    if [ $# -ne 2 ]; then
        echo "Usage: dup <string> <count>"
        return
    fi
    local times=$(seq 1 $2)
    for x in ${times}; do
        echo -n "$1"
    done
}

make_header() {
    if [ $# -ne 2 ]; then
        echo "Usage: make_header <required title> <color>"
        return
    fi
    local title="$1"
    local title_len=${#title}
    title_len=$((title_len + 4))
    local width=70
    local left=4
    local right=$(($width-($title_len+$left)))
    local color="$2"
    dup '-' $left
    echo -ne "[ ${color}${title}${NC} ]"
    dup '-' $right
    echo
}

function usage() {
    echo "Usage: $0 [-a]|[-i][-u][-d]"
    echo
    echo "Runs tests under ${ROOT}.  Options control which test types:"
    echo
    echo "    -a | --all . . . . . . . . . . . . Run all types of tests"
    echo "    -d | --doctests  . . . . . . . . . Run doctests"
    echo "    -u | --unittests . . . . . . . . . Run unittests"
    echo "    -i | --integration . . . . . . . . Run integration tests"
    echo
    exit 1
}

while [[ $# -gt 0 ]]; do
    key="$1"
    case $key in
        -a|--all)
            DOCTEST=1
            UNITTEST=1
            INTEGRATION=1
            ;;
        -d|--doctests)
            DOCTEST=1
            ;;
        -u|--unittests)
            UNITTEST=1
            ;;
        -i|--integration)
            INTEGRATION=1
            ;;
        *)    # unknown option
            echo "Argument $key was not recognized."
            echo
            usage
            exit 1
            ;;
    esac
    shift
done

if [ $(expr ${DOCTEST} + ${UNITTEST} + ${INTEGRATION}) -eq 0 ]; then
    usage
    exit 2
fi


FAILED_TESTS=""
if [ ${DOCTEST} -eq 1 ]; then
    for doctest in $(grep -lR doctest ${ROOT}/*); do
        if [[ ${doctest} == *.py ]]; then
            BASE=$(basename ${doctest})
            BASE="${BASE} (doctest)"
            make_header "${BASE}" "${CYAN}"
            OUT=$( python3 ${doctest} 2>&1 )
            FAILED=$( echo "${OUT}" | grep '\*\*\*Test Failed\*\*\*' | wc -l )
            if [ $FAILED == 0 ]; then
                echo "OK"
            else
                echo -e "${FAILED}"
                FAILURES=$((FAILURES+1))
                FAILED_TESTS="${FAILED_TESTS},${BASE} (python3 ${doctest})"
            fi
        fi
    done
fi

if [ ${UNITTEST} -eq 1 ]; then
    for test in $(find ${ROOT} -name "*_test.py" -print); do
        BASE=$(basename ${test})
        BASE="${BASE} (unittest)"
        make_header "${BASE}" "${GREEN}"
        ${test}
        if [ $? -ne 0 ]; then
            FAILURES=$((FAILURES+1))
            FAILED_TESTS="${FAILED_TESTS},${BASE} (python3 ${test})"
        fi
    done
fi

if [ ${INTEGRATION} -eq 1 ]; then
    for test in $(find ${ROOT} -name "*_itest.py" -print); do
        BASE=$(basename ${test})
        BASE="${BASE} (integration test)"
        make_header "${BASE}" "${ORANGE}"
        ${test}
        if [ $? -ne 0 ]; then
            FAILURES=$((FAILURES+1))
            FAILED_TESTS="${FAILED_TESTS},${BASE} (python3 ${test})"
        fi
    done
fi

if [ ${FAILURES} -ne 0 ]; then
    FAILED_TESTS=$(echo ${FAILED_TESTS} | sed 's/^,/__/g')
    FAILED_TESTS=$(echo ${FAILED_TESTS} | sed 's/,/\n__/g')
    if [ ${FAILURES} -eq 1 ]; then
        echo -e "${RED}There was ${FAILURES} failure:"
    else
        echo -e "${RED}There were ${FAILURES} failures:"
    fi
    echo "${FAILED_TESTS}"
    echo -e "${NC}"
else
    echo -e "${BLACK}${ON_GREEN}Everything looks good.${NC}"
fi