test-rpms

#!/bin/bash
set -e

SELFDIR=$(dirname "$0")
SELFDIR=$(cd "$SELFDIR" && pwd)
# shellcheck source=lib/library.sh
source "$SELFDIR/lib/library.sh"

SERVER=
IMAGE=
EXPECTED_VARIANT=
RUBY_RPM_PATH=
RBENV_RPM_PATH=
DEBUG_ON_FAIL=false

function usage()
{
    echo "Usage: ./test-rpms <OPTIONS>"
    echo "Test built RPM packages. Either tests local .rpm files (which"
    echo "is the default mode), or tests packages on an actual YUM server."
    echo
    echo "Required options:"
    echo "  -S URL      Test packages on the given YUM server, not local files"
    echo "              (e.g. https://yum.fullstaqruby.org/centos-7)"
    echo "  -i IMAGE    Image name to run test in (e.g. centos:7)"
    echo "  -v VARIANT  Expected variant name (one of: normal, jemalloc, malloctrim)"
    echo
    echo "Required options (local file mode):"
    echo "  -r PATH     Path to Ruby RPM package"
    echo "  -b PATH     Path to Rbenv RPM package"
    echo
    echo "Required options (server mode):"
    echo "  -n VERSION  Ruby package version to test against"
    echo "              (e.g. 2.6; 2.6-jemalloc)"
    echo
    echo "Optional options:"
    echo "  -D          Open a debug shell on failure"
    echo "  -h          Show usage"
}

function parse_options()
{
    local OPTIND=1
    local ORIG_ARGV
    local opt
    while getopts "S:i:v:r:b:n:Dh" opt; do
        case "$opt" in
        S)
            SERVER="$OPTARG"
            ;;
        i)
            IMAGE="$OPTARG"
            ;;
        v)
            EXPECTED_VARIANT="$OPTARG"
            ;;
        r)
            RUBY_RPM_PATH=$(absolute_path "$OPTARG")
            ;;
        b)
            RBENV_RPM_PATH=$(absolute_path "$OPTARG")
            ;;
        n)
            RUBY_PACKAGE_VERSION="$OPTARG"
            ;;
        D)
            DEBUG_ON_FAIL=true
            ;;
        h)
            usage
            exit
            ;;
        *)
            return 1
            ;;
        esac
    done

    (( OPTIND -= 1 )) || true
    shift $OPTIND || true
    ORIG_ARGV=("$@")

    if [[ "$SERVER" = "" ]]; then
        if [[ "$RUBY_RPM_PATH" = "" ]]; then
            echo 'ERROR: please specify a Ruby RPM package path with -r.' >&2
            exit 1
        fi
        if [[ ! -e "$RUBY_RPM_PATH" ]]; then
            echo "ERROR: $RUBY_RPM_PATH does not exist." >&2
            exit 1
        fi
        if [[ "$RBENV_RPM_PATH" = "" ]]; then
            echo 'ERROR: please specify an Rbenv RPM package path with -b.' >&2
            exit 1
        fi
        if [[ ! -e "$RBENV_RPM_PATH" ]]; then
            echo "ERROR: $RBENV_RPM_PATH does not exist." >&2
            exit 1
        fi
        if [[ "$RUBY_PACKAGE_VERSION" != "" ]]; then
            echo 'ERROR: if -S is not given, then -n may also not be given.' >&2
            exit 1
        fi
    else
        if [[ "$RUBY_RPM_PATH" != "" ]]; then
            echo 'ERROR: is -S is given, then -r may not be given.' >&2
            exit 1
        fi
        if [[ "$RBENV_RPM_PATH" != "" ]]; then
            echo 'ERROR: is -S is given, then -b may not be given.' >&2
            exit 1
        fi
        if [[ "$RUBY_PACKAGE_VERSION" = "" ]]; then
            echo 'ERROR: please specify a Ruby package version to test with -n.' >&2
            exit 1
        fi
    fi
    if [[ "$IMAGE" = "" ]]; then
        echo 'ERROR: please specify an image name with -i.' >&2
        exit 1
    fi
    if [[ "$EXPECTED_VARIANT" != normal && "$EXPECTED_VARIANT" != jemalloc && "$EXPECTED_VARIANT" != malloctrim ]]; then
        echo 'ERROR: please specify a valid variant name with -v.' >&2
        exit 1
    fi
}

parse_options "$@"

if tty -s; then
    TTY_ARGS=(-t -i)
else
    TTY_ARGS=()
fi

RUN_ARGS=()
UTILITY_IMAGE_VERSION=$(read_single_value_file "$SELFDIR/environments/utility/image_tag")

TEMP_DIR=$(mktemp -d /tmp/test.XXXXXX)
function _cleanup()
{
    rm -rf "$TEMP_DIR"
}
mkdir -p "$TEMP_DIR/repo"

if [[ "$SERVER" = "" ]]; then
    echo '--- Entering preparation Docker container ---'
    verbose_run docker run --rm --init "${TTY_ARGS[@]}" \
        -v "$SELFDIR:/system:ro" \
        -v "$RUBY_RPM_PATH:/input/$(basename "$RUBY_RPM_PATH"):ro" \
        -v "$RBENV_RPM_PATH:/input/$(basename "$RBENV_RPM_PATH"):ro" \
        -v "$TEMP_DIR/repo:/output" \
        --user "$(id -u):$(id -g)" \
        "fullstaq/ruby-build-env-utility:$UTILITY_IMAGE_VERSION" \
        /system/container-entrypoints/test-rpms-prepare
    echo
    RUN_ARGS=()
else
    RUN_ARGS+=(
        -e "RUBY_PACKAGE_VERSION=$RUBY_PACKAGE_VERSION"
    )
fi


echo '--- Entering main Docker container ---'
verbose_run docker run --rm --init "${TTY_ARGS[@]}" \
    -v "$SELFDIR:/system:ro" \
    -v "$TEMP_DIR/repo:/input/repo:ro" \
    -e "SERVER=$SERVER" \
    -e "EXPECTED_VARIANT=$EXPECTED_VARIANT" \
    -e "DEBUG_ON_FAIL=$DEBUG_ON_FAIL" \
    "${RUN_ARGS[@]}" \
    --user root \
    --entrypoint /system/container-entrypoints/test-rpms \
    "$IMAGE"