#!/bin/bash

function usage {
  echo "Usage: $0 [OPTION]..."
  echo "Run Heat's test suite(s)"
  echo ""
  echo "  -V, --virtual-env        Always use virtualenv.  Install automatically if not present"
  echo "  -N, --no-virtual-env     Don't use virtualenv.  Run tests in local environment (default)"
  echo "  -F, --force              Force a clean re-build of the virtual environment. Useful when dependencies have been added."
  echo "  -f, --func               Run functional tests"
  echo "  -u, --unit               Run unit tests (default when nothing specified)"
  echo "  -p, --pep8               Run pep8 tests"
  echo "  --all                    Run all tests"
  echo "  -c, --coverage           Generate coverage report (selects --unit)"
  echo "  -h, --help               Print this usage message"
  exit
}

# must not assign -a as an option, needed for selecting custom attributes
function process_option {
  case "$1" in
    -V|--virtual-env) always_venv=1; never_venv=0;;
    -N|--no-virtual-env) always_venv=0; never_venv=1;;
    -F|--force) force=1;;
    -f|--func) test_func=1; noseargs="$noseargs -a tag=func";;
    -u|--unit) test_unit=1; noseargs="$noseargs -a tag=unit";;
    -p|--pep8) test_pep8=1;;
    --all) test_func=1; test_unit=1; test_pep8=1; noseargs="$noseargs -a tag=func -a tag=unit";;
    -c|--coverage) coverage=1; test_unit=1; noseargs="$noseargs -a tag=unit";;
    -h|--help) usage;;
    *) noseargs="$noseargs $1"
  esac
}

venv=.venv
with_venv=tools/with_venv.sh
wrapper=""

function run_tests {
  echo 'Running tests'
  NOSETESTS="python heat/testing/runner.py $noseopts $noseargs"
  # Just run the test suites in current environment
  ${wrapper} $NOSETESTS 2> run_tests.err.log
}

function run_pep8 {
  echo "Running pep8..."
  PEP8_OPTIONS="--exclude=$PEP8_EXCLUDE --repeat"
  PEP8_INCLUDE="bin/heat-cfn bin/heat-boto bin/heat-api-cfn bin/heat-api bin/heat-engine heat tools setup.py heat/testing/runner.py"
  ${wrapper} pep8 $PEP8_OPTIONS $PEP8_INCLUDE
}

# run unit tests with pep8 when no arguments are specified
# otherwise process CLI options
if [[ $# == 0 ]]; then
    noseargs="$noseargs -a tag=unit"
    test_pep8=1
else
    for arg in "$@"; do
        process_option $arg
    done
fi

# If enabled, tell nose to collect coverage data
if [ "$coverage" == 1 ]; then
    noseopts="$noseopts --with-coverage --cover-package=heat"
fi

if [ "$never_venv" == 0 ]
then
  # Remove the virtual environment if --force used
  if [ "$force" == 1 ]; then
    echo "Cleaning virtualenv..."
    rm -rf ${venv}
  fi
  if [ -e ${venv} ]; then
    wrapper="${with_venv}"
  else
    if [ "$always_venv" == 1 ]; then
      # Automatically install the virtualenv
      python tools/install_venv.py
      wrapper="${with_venv}"
    else
      echo -e "No virtual environment found...create one? (Y/n) \c"
      read use_ve
      if [ "x$use_ve" = "xY" -o "x$use_ve" = "x" -o "x$use_ve" = "xy" ]; then
        # Install the virtualenv and run the test suite in it
        python tools/install_venv.py
        wrapper=${with_venv}
      fi
    fi
  fi
fi

# Delete old coverage data from previous runs
if [ "$coverage" == 1 ]; then
    ${wrapper} coverage erase
fi

result=0

# If functional or unit tests have been selected, run them
if [ ! -z "$noseargs" ]; then
  run_tests
  result=$?
fi

# Run pep8 if it was selected
if [ "$test_pep8" == 1 ]; then
  run_pep8
fi

# Generate coverage report
if [ "$coverage" == 1 ]; then
    echo "Generating coverage report in covhtml/"
    # Don't compute coverage for common code, which is tested elsewhere
    ${wrapper} coverage html --include='heat/*' --omit='heat/openstack/common/*' -d covhtml -i
fi

exit $result