devilbox/.tests/.lib.sh
2017-05-16 20:43:29 +02:00

492 lines
11 KiB
Bash

#!/bin/sh -eu
DEVILBOX_PATH="$( echo "${1}"| sed 's/\/*$//' )" # remove last slash(es): /
################################################################################
#
# H E L P E R
#
################################################################################
run() {
_cmd="${1}"
_debug="0"
_red="\033[0;31m"
_green="\033[0;32m"
_reset="\033[0m"
_user="$(whoami)"
# If 2nd argument is set and enabled, allow debug command
if [ "${#}" = "2" ]; then
if [ "${2}" = "1" ]; then
_debug="1"
fi
fi
if [ "${_debug}" = "1" ]; then
printf "${_red}%s \$ ${_green}${_cmd}${_reset}\n" "${_user}"
fi
sh -c "LANG=C LC_ALL=C ${_cmd}"
}
runsu() {
_cmd="${1}"
_debug="0"
_red="\033[0;31m"
_green="\033[0;32m"
_reset="\033[0m"
_user="$(whoami)"
# If 2nd argument is set and enabled, allow debug command
if [ "${#}" = "2" ]; then
if [ "${2}" = "1" ]; then
_debug="1"
fi
fi
if [ "${_debug}" = "1" ]; then
printf "${_red}%s \$ ${_green}sudo ${_cmd}${_reset}\n" "${_user}"
fi
sudo "LANG=C LC_ALL=C ${_cmd}"
}
print_h1() {
_headline="${1}"
_blue="\033[0;34m"
_reset="\033[0m"
printf "${_blue}%s${_reset}\n" "################################################################################"
printf "${_blue}%s${_reset}\n" "#"
printf "${_blue}%s %s${_reset}\n" "#" "${_headline}"
printf "${_blue}%s${_reset}\n" "#"
printf "${_blue}%s${_reset}\n" "################################################################################"
}
print_h2() {
_headline="${1}"
_blue="\033[0;34m"
_reset="\033[0m"
printf "${_blue}%s${_reset}\n" "############################################################"
printf "${_blue}%s %s${_reset}\n" "#" "${_headline}"
printf "${_blue}%s${_reset}\n" "############################################################"
}
wait_for() {
_time="${1}"
_debug="0"
# Sleep with debug output
if [ "${#}" = "2" ]; then
if [ "${2}" = "1" ]; then
printf "wait "
# shellcheck disable=SC2034
for i in $(seq 1 "${_time}"); do
sleep 1
printf "."
done
printf "\n"
return 0
fi
fi
# Sleep silently
sleep "${_time}"
}
################################################################################
#
# G E T D E F A U L T S
#
################################################################################
###
### Get newline separated default data mount directories (from .env file)
###
get_data_mounts() {
_mounts="$( grep -E '^.*_DATADIR=' "${DEVILBOX_PATH}/env-example" | sed 's/^.*=//g' )"
_data=""
IFS='
'
for _mount in ${_mounts}; do
_prefix="$( echo "${_mount}" | cut -c-1 )"
# Relative path?
if [ "${_prefix}" = "." ]; then
_mount="$( echo "${_mount}" | sed 's/^\.//g' )" # Remove leading dot: .
_mount="$( echo "${_mount}" | sed 's/^\///' )" # Remove leading slash: /
_mount="${DEVILBOX_PATH}/${_mount}"
fi
# newline Append
if [ "${_data}" = "" ]; then
_data="${_mount}"
else
_data="${_data}\n${_mount}"
fi
done
echo "${_data}"
}
###
### Default enabled Docker Versions
###
get_enabled_version_httpd() {
_default="$( grep -E '^HTTPD_SERVER=' "${DEVILBOX_PATH}/.env" | sed 's/^.*=//g' )"
echo "${_default}"
}
get_enabled_version_mysql() {
_default="$( grep -E '^MYSQL_SERVER=' "${DEVILBOX_PATH}/.env" | sed 's/^.*=//g' )"
echo "${_default}"
}
get_enabled_version_postgres() {
_default="$( grep -E '^POSTGRES_SERVER=' "${DEVILBOX_PATH}/.env" | sed 's/^.*=//g' )"
echo "${_default}"
}
get_enabled_version_php() {
_default="$( grep -E '^PHP_SERVER=' "${DEVILBOX_PATH}/.env" | sed 's/^.*=//g' )"
echo "${_default}"
}
################################################################################
#
# S E T / R E S E T F U N C T I O N S
#
################################################################################
###
### Recreate .env file from env-example
###
reset_env_file() {
# Re-create .env file
if [ -f "${DEVILBOX_PATH}/.env" ]; then
rm -f "${DEVILBOX_PATH}/.env"
fi
cp "${DEVILBOX_PATH}/env-example" "${DEVILBOX_PATH}/.env"
}
###
### Comment out all docker versions
###
comment_all_dockers() {
# Comment out all enabled docker versions
run "sed -i'' \"s/^HTTPD_SERVER=/#HTTPD_SERVER=/g\" \"${DEVILBOX_PATH}/.env\""
run "sed -i'' \"s/^MYSQL_SERVER=/#MYSQL_SERVER=/g\" \"${DEVILBOX_PATH}/.env\""
run "sed -i'' \"s/^POSTGRES_SERVER=/#POSTGRES_SERVER=/g\" \"${DEVILBOX_PATH}/.env\""
run "sed -i'' \"s/^PHP_SERVER=/#PHP_SERVER=/g\" \"${DEVILBOX_PATH}/.env\""
}
###
### Eenable desired docker version
###
enable_docker_httpd() {
_docker_version="${1}"
run "sed -i'' \"s/#HTTPD_SERVER=${_docker_version}/HTTPD_SERVER=${_docker_version}/g\" \"${DEVILBOX_PATH}/.env\""
}
enable_docker_mysql() {
_docker_version="${1}"
run "sed -i'' \"s/#MYSQL_SERVER=${_docker_version}/MYSQL_SERVER=${_docker_version}/g\" \"${DEVILBOX_PATH}/.env\""
}
enable_docker_pgsql() {
_docker_version="${1}"
run "sed -i'' \"s/#POSTGRES_SERVER=${_docker_version}/POSTGRES_SERVER=${_docker_version}/g\" \"${DEVILBOX_PATH}/.env\""
}
enable_docker_php() {
_docker_version="${1}"
run "sed -i'' \"s/#PHP_SERVER=${_docker_version}/PHP_SERVER=${_docker_version}/g\" \"${DEVILBOX_PATH}/.env\""
}
################################################################################
#
# S T A R T / S T O P T H E D E V I L B O X
#
################################################################################
devilbox_start() {
_srv1="${1}"
_ver1="${2}"
_srv2="${3}"
_ver2="${4}"
# Default values for remaining servers
_def_php="php-fpm-7.0"
_def_httpd="nginx-stable"
_def_mysql="mariadb-10.0"
_def_pgsql="9.6"
# Specific enabled servers
_set_php=""
_set_httpd=""
_set_mysql=""
_set_pgsql=""
# Print Headline
print_h1 "${_srv1}-${_ver1} vs ${_srv2}-${_ver2}"
# Adjust .env
comment_all_dockers
# Enable Type 1
if [ "${_srv1}" = "HTTPD" ]; then
_set_httpd="${_ver1}"
elif [ "${_srv1}" = "MYSQL" ]; then
_set_mysql="${_ver1}"
elif [ "${_srv1}" = "PGSQL" ]; then
_set_pgsql="${_ver1}"
elif [ "${_srv1}" = "PHP" ]; then
_set_php="${_ver1}"
else
echo "Invalid server: ${_srv1}"
exit 1
fi
# Enable Type 2
if [ "${_srv2}" = "HTTPD" ]; then
_set_httpd="${_ver2}"
elif [ "${_srv2}" = "MYSQL" ]; then
_set_mysql="${_ver2}"
elif [ "${_srv2}" = "PGSQL" ]; then
_set_pgsql="${_ver2}"
elif [ "${_srv2}" = "PHP" ]; then
_set_php="${_ver2}"
else
echo "Invalid server: ${_srv2}"
exit 1
fi
# Enable remaining onces
if [ "${_set_php}" = "" ]; then
_set_php="${_def_php}"
fi
if [ "${_set_httpd}" = "" ]; then
_set_httpd="${_def_httpd}"
fi
if [ "${_set_mysql}" = "" ]; then
_set_mysql="${_def_mysql}"
fi
if [ "${_set_pgsql}" = "" ]; then
_set_pgsql="${_def_pgsql}"
fi
# Set versions in .env file
enable_docker_php "${_set_php}"
enable_docker_httpd "${_set_httpd}"
enable_docker_mysql "${_set_mysql}"
enable_docker_pgsql "${_set_pgsql}"
# Show Server settings
echo "Enable SERVERs"
echo "--------------"
grep '^[A-Za-z0-9]*_SERVER' "${DEVILBOX_PATH}/.env" | column -t -s '='
echo
# Show all other settings
echo "Enable Settings"
echo "---------------"
grep -E '^[-_A-Za-z0-9]*=' "${DEVILBOX_PATH}/.env" | grep -v 'SERVER_' | column -t -s '='
echo
# Run
docker-compose up -d
# Wait for it to come up
wait_for 30 1
# Show log/info
docker-compose logs
#docker-compose ps
}
devilbox_stop() {
# Stop existing dockers
cd "${DEVILBOX_PATH}" || exit 1
docker-compose down > /dev/null 2>&1 || true
docker-compose stop > /dev/null 2>&1 || true
docker-compose kill > /dev/null 2>&1 || true
docker-compose rm -f || true
# Delete existing data dirs
_data_dirs="$( get_data_mounts )"
IFS='
'
for d in ${_data_dirs}; do
runsu "rm -rf ${d}"
done
}
################################################################################
#
# T E S T T H E D E V I L B O X
#
################################################################################
debilbox_test() {
###
### Variables
###
_ret=0 # Final exit code
_oks=4 # Require this many [OK]'s on the page
###
### 1. Show Info
###
print_h2 "1. Info"
# Show wanted versions
echo ".env settings"
echo "------------------------------------------------------------"
echo "HTTPD: $(get_enabled_version_httpd)"
echo "PHP: $(get_enabled_version_php)"
echo "MySQL: $(get_enabled_version_mysql)"
echo "PgSQL: $(get_enabled_version_postgres)"
echo
# Get actual versions
echo "http://localhost settings"
echo "------------------------------------------------------------"
curl -q http://localhost/index.php 2>/dev/null | grep -E '<h3>.*</h3>' | sed 's/.*<h3>//g' | sed 's/<\/h3>//g'
echo
###
### 2. Test docker-compose
###
print_h2 "2. docker-compose"
echo "docker-compose ps"
echo "------------------------------------------------------------"
if _test_docker_compose >/dev/null 2>&1; then
echo "[OK]: All running"
else
echo "[ERR]: Broken"
_ret="$(( _ret + 1 ))"
fi
###
### 3. Show Curl output
###
print_h2 "3. Test status via curl"
echo "Count [OK]'s on curl-ed url"
echo "------------------------------------------------------------"
if _cnt="$( _test_curled_oks "${_oks}" )"; then
echo "[OK]: ${_cnt} of ${_oks}"
else
echo "[ERR]: ${_cnt} of ${_oks}"
_ret="$(( _ret + 1 ))"
fi
echo
###
### Final return
###
if [ "${_ret}" != "0" ]; then
print_h2 "4. Error output"
echo "Curl"
echo "------------------------------------------------------------"
curl -vv http://localhost/index.php || true
echo
echo "docker-compose ps"
echo "------------------------------------------------------------"
docker-compose ps
echo
echo "docker-compose logs"
echo "------------------------------------------------------------"
docker-compose logs
echo
echo "log files"
echo "------------------------------------------------------------"
ls -lap log/
sudo find log -type f -exec sh -c 'echo "{}:\n-----------------"; cat "{}"; echo "\n\n"' \;
return 1
fi
return 0
}
###
### Test against stopped containers
###
_test_docker_compose() {
_broken="$( docker-compose ps | grep -c 'Exit' || true )"
_running="$( docker-compose ps | grep -c 'Up' || true )"
_total="$( docker-compose ps -q | grep -c '' || true )"
if [ "${_broken}" != "0" ]; then
return 1
fi
if [ "${_running}" != "${_total}" ]; then
return 1
fi
return 0
}
###
### Test [OK]'s found on website
###
_test_curled_oks() {
_oks="${1}"
max="20"
i=0
while [ $i -lt $max ]; do
if [ "$( curl -s -o /dev/null -w '%{http_code}' http://localhost/index.php )" != "404" ]; then
break;
fi
sleep 1s
i=$(( i + 1 ))
done
# sleep (in case hhvm segfaulted and needs to be restarted)
sleep 10
_count="$( curl -q http://localhost/index.php 2>/dev/null | grep -c 'OK' || true )"
echo "${_count}"
if [ "${_count}" != "${_oks}" ]; then
return 1
else
return 0
fi
}