run-tests.sh

#!/bin/bash

# set of constants to colorize output
FAILED_OUT="\033[0;31m"
YELLOW_OUT="\033[1;33m"
NONE_OUT="\033[0m"


pretty-output() {
      t="$@xxxx"
      c=${replace:--}
      echo -e ${t//?/$c}
      echo -e "$c $@ $c"
      echo -e ${t//?/$c}
}


helper() {
    echo "Tool allows to simplify run of automated tests for POM sample project."
    echo ""
    echo "Available actions:"
    echo -e " - smoke\t\t run automated smoke tests"
    echo -e " - unittest\t\t run automated unittest tests"
    echo -e " - all\t\t\t run all automated tests"
    echo -e " - help\t\t\t display help message"
    echo ""
    echo -e "Note:\t\t help will be provided in case of no input parameters"
}


raise-error-message() {
    echo -e "Invalid ${FAILED_OUT}$1${NONE_OUT} parameter is provided!"
    echo -e "Please use ${YELLOW_OUT}smoke${NONE_OUT} or ${YELLOW_OUT}unit${NONE_OUT} or ${YELLOW_OUT}help${NONE_OUT} keys as a flag."
    exit 1
}


clear-trash() {
    local trash='.pytest_cache'
    [[ -d "$trash" ]] && echo "removing ${trash} testing trash" && rm -rf ${trash} && echo "environment is cleared"
}


unit() {
    pytest -m unit
}


smoke() {
    pytest -m smoke
}


all() {
    pytest
}


run-tests() {
    local arg=$1
    if [[ ${arg} == "smoke" ]] || [[ ${arg} == "unit" ]];
        then pretty-output "Running ${arg} suite" && pytest -m "${arg}"; clear-trash
    elif [[ ${arg} == "all" ]]
        then pretty-output "Running ${arg} suite" && pytest; clear-trash
    elif [[ $# -eq 0 ]] || [[ ${arg} == "help" ]] || [[ ${arg} == "--help" ]] || [[ ${arg} == "-h" ]]
        then helper
    else raise-error-message "${arg}"
    fi
}

run-tests "$1"