test.sh

# This file is to be executed using `bash_unit`

DOCKER=${DOCKER:-$(command -v docker || echo false)}
SKOPEO=${SKOPEO:-$(command -v skopeo || echo false)}
CRICTL=${CRICTL:-$(command -v crictl || echo false)}
CURL=${CURL:-$(command -v curl || echo false)}
HARDLINK=${HARDLINK:-$(command -v hardlink || echo false)}

IMAGE=${IMAGE-nicolast/static-container-registry:test}
IMAGES=${IMAGES-/tmp/images}
CONTAINER_NAME=${CONTAINER_NAME-static-container-registry-test}
REGISTRY_HOST=${REGISTRY_HOST-127.0.0.1}
REGISTRY_PORT=${REGISTRY_PORT-5000}
REGISTRY="$REGISTRY_HOST:$REGISTRY_PORT"

TEST_DOCKER=${TEST_DOCKER:-1}
TEST_CONTAINERD=${TEST_CONTAINERD:-1}
TEST_CRIO=${TEST_CRIO:-1}
TEST_SKOPEO=${TEST_SKOPEO:-1}

if [ "$TEST_DOCKER" -eq 1 ]; then
test_docker() {
        for image in ${AVAILABLE_IMAGES[*]}; do
                assert "$DOCKER pull '$REGISTRY/$image'"
        done
}
fi

if [ "$TEST_CONTAINERD" -eq 1 ]; then
test_containerd() {
        for image in ${AVAILABLE_IMAGES[*]}; do
                assert "sudo ${CRICTL} --image-endpoint unix:///run/containerd/containerd.sock pull '$REGISTRY/$image'"
        done
}
fi

if [ "$TEST_CRIO" -eq 1 ]; then
test_crio() {
        for image in ${AVAILABLE_IMAGES[*]}; do
                assert "sudo ${CRICTL} --image-endpoint unix:///run/crio/crio.sock pull '$REGISTRY/$image'"
        done
}
fi

if [ "$TEST_SKOPEO" -eq 1 ]; then
test_skopeo() {
        for image in ${AVAILABLE_IMAGES[*]}; do
                assert "$SKOPEO --debug inspect --tls-verify=false 'docker://$REGISTRY/$image'"
        done
}
fi

setup_suite() {
        if [ "x$IMAGE" != "x" ]; then
                assert build_project_image
        fi
        if [ "x$IMAGES" != "x" ]; then
                assert create_images_directory
        fi
}

teardown_suite() {
        if [ "x$IMAGES" != "x" ]; then
                remove_images_directory
        fi
        if [ "x$IMAGE" != "x" ]; then
                delete_project_image
        fi
}

setup() {
        if [ "x$IMAGE" != "x" ]; then
                $DOCKER run \
                        -d \
                        -p "$REGISTRY:80" \
                        -v "$IMAGES:/var/lib/images:ro" \
                        --name "$CONTAINER_NAME" \
                        "$IMAGE" > /dev/null
        fi

        local i=600
        while [ $i -gt 0 ]; do
                local ok
                ok=$($CURL --silent "http://$REGISTRY/v2/" 2>/dev/null)
                if [ "x$ok" = 'xok' ]; then
                        i=0
                else
                        sleep 0.1
                        i=$((i - 1))
                fi
        done
}

teardown() {
        if [ "x$IMAGE" != "x" ]; then
                $DOCKER stop "$CONTAINER_NAME" > /dev/null
                $DOCKER rm "$CONTAINER_NAME" > /dev/null
        fi
}

build_project_image() {
        $DOCKER build -t "$IMAGE" .
}

delete_project_image() {
        $DOCKER rmi "$IMAGE" > /dev/null
}

AVAILABLE_IMAGES=(
    'alpine:3.9.3'
    'alpine:3.9'
    'alpine:3.8.4'
    'metalk8s-keepalived:latest'
)
create_images_directory() {
        mkdir "$IMAGES" "$IMAGES/alpine" "$IMAGES/metalk8s-keepalived"
        $SKOPEO copy --format v2s2 --dest-compress \
                docker://docker.io/alpine:3.9.3 \
                "dir:$IMAGES/alpine/3.9.3"
        $SKOPEO copy --format v2s2 --dest-compress \
                docker://docker.io/alpine:3.9 \
                "dir:$IMAGES/alpine/3.9"
        $DOCKER pull docker.io/alpine:3.8.4
        $SKOPEO copy --format v2s2 --dest-compress \
                docker-daemon:alpine:3.8.4 \
                "dir:$IMAGES/alpine/3.8.4"
        $DOCKER rmi docker.io/alpine:3.8.4
        $SKOPEO copy --format v2s2 --dest-compress \
                docker://docker.io/nicolast/metalk8s-keepalived:latest \
                "dir:$IMAGES/metalk8s-keepalived/latest"

        $HARDLINK -c -vv "${IMAGES}"
}

remove_images_directory() {
        rm -rf "$IMAGES"
}