runtests.sh

#!/usr/bin/env bash
set -e

if [ ! -d venv ]; then
    virtualenv venv
fi

venv/bin/pip install -r requirements/development.txt --exists-action w --quiet

while getopts "1il" opt; do
    case $opt in
        1) RUN_ONCE=1;;
        i) INTEGRATION=1;;
        l) LOCAL=1;;
    esac
done

[ -z $INTEGRATION ] && TEST_SUITE="unit" || TEST_SUITE="integration"
[ -z $LOCAL ] || TEST_SUITE="integration_local"
[ -z $INTEGRATION ] && TIME_OUT="--process-timeout=30" || TIME_OUT="--process-timeout=3600"
[ -z $LOCAL ] || TIME_OUT="--process-timeout=1200"

# Do not use the nose.proxy otherwise we can't run Popen with unbuffered output
# fixes: nose.proxy.AttributeError: '_io.StringIO' object has no attribute 'buffer'
[ -z $INTEGRATION ] && NO_CAPTURE="" || NO_CAPTURE="--nocapture"  

shift $((OPTIND - 1))

if [ -z $INTEGRATION ]; then
    if [ -e "/proc/cpuinfo" ]; then
        numprocs=$(cat /proc/cpuinfo  | grep processor | wc -l | cut -d ' ' -f 1)
    elif [ "x$(uname)" = "xDarwin" ]; then
        numprocs=$(sysctl -n hw.ncpu)
    else
        numprocs=1
    fi
else
    echo "Testing sudo so we can start Dockers in the tests"
    sudo echo "Could sudo. Ok!"
    numprocs=1
fi

# Don't write .pyc files
export PYTHONDONTWRITEBYTECODE=1  
# Remove existing .pyc files
find . -type f -name *.pyc -delete

test_cmd="
    echo 'Running puppetfiles $TEST_SUITE tests';
    venv/bin/nosetests --processes=$numprocs tests/$TEST_SUITE $TIME_OUT $NO_CAPTURE;
    echo 'Checking PEP8';
    venv/bin/autopep8 -r --diff tests;
"

if [ -z $RUN_ONCE ]; then
    LC_NUMERIC="en_US.UTF-8" watch -c -n 0.1 -- "$test_cmd"
else
    sh -ec "$test_cmd"
fi