#!/bin/bash
. /usr/share/beakerlib/beakerlib.sh || exit 1
. ../../images.sh || exit 1

rlJournalStart
    rlPhaseStartSetup
        rlRun "PROVISION_HOW=${PROVISION_HOW:-container}"
        rlRun "IMAGE_MODE=${IMAGE_MODE:-no}"

        if [ "$PROVISION_HOW" = "container" ]; then
            build_container_image "ubi/8/upstream\:latest"
            build_container_image "centos/7/upstream\:latest"
            build_container_image "fedora/latest/upstream\:latest"
        fi

        rlRun "pushd data"
    rlPhaseEnd

    tmt="tmt run -vv --all --remove provision --how $PROVISION_HOW"
    basic="plan --name 'mixed|weird'"
    debuginfo="plan --name debuginfo"

    # Verify against the default provision image (skip for image mode, no default image)
    # TODO: re-enable once default images supported with image mode, see #4806"
    if [[ "$IMAGE_MODE" != "yes" ]]; then
        rlPhaseStartTest "Test the default image ($PROVISION_HOW)"
            rlRun -s "$tmt $basic"
            rlAssertGrep "Recommended packages failed to install, continuing regardless:" $rlRun_LOG
            rlAssertGrep "forest: recommended by: /test/mixed" $rlRun_LOG
            rlAssertGrep "weird-package: recommended by: /test/weird" $rlRun_LOG
            rlAssertNotGrep "dconf: recommended by: /test/mixed" $rlRun_LOG
        rlPhaseEnd
    fi

    # Check CentOS images for container provision
    if [[ "$PROVISION_HOW" == "container" ]]; then
        for image in $TEST_IMAGE_PREFIX/centos/7/upstream:latest $TEST_IMAGE_PREFIX/ubi/8/upstream:latest; do
            rlPhaseStartTest "Test $image ($PROVISION_HOW)"
	        rlRun -s "$tmt --image $image $basic"
		rlAssertGrep "Recommended packages failed to install, continuing regardless:" $rlRun_LOG
		rlAssertGrep "forest: recommended by: /test/mixed" $rlRun_LOG
		rlAssertGrep "weird-package: recommended by: /test/weird" $rlRun_LOG
		rlAssertNotGrep "dconf: recommended by: /test/mixed" $rlRun_LOG
            rlPhaseEnd
        done
    fi

    # Check debuginfo install (only for supported distros)
    # https://bugzilla.redhat.com/show_bug.cgi?id=1964505
    if [[ "$PROVISION_HOW" == "container" ]]; then
        for image in $TEST_IMAGE_PREFIX/fedora/latest/upstream:latest $TEST_IMAGE_PREFIX/centos/7/upstream:latest; do
            rlPhaseStartTest "Test $image ($PROVISION_HOW) [debuginfo]"
	        rlRun -s "$tmt --image $image $debuginfo"
		rlAssertGrep "Recommended packages failed to install, continuing regardless:" $rlRun_LOG
		rlAssertGrep "forest: recommended by: /test/debuginfo" $rlRun_LOG
            rlPhaseEnd
        done
    fi

    # Run against image mode images
    if [[ "$IMAGE_MODE" == "yes" ]]; then
        while IFS= read -r image; do
            rlPhaseStartTest "Test $image (image-mode)"
                rlRun -s "$tmt --image $image $basic"
                rlAssertGrep "Recommended packages failed to install, continuing regardless:" $rlRun_LOG
                rlAssertGrep "forest: recommended by: /test/mixed" $rlRun_LOG
                rlAssertGrep "weird-package: recommended by: /test/weird" $rlRun_LOG
                rlAssertNotGrep "dconf: recommended by: /test/mixed" $rlRun_LOG
            rlPhaseEnd
        done <<< "$TEST_IMAGE_MODE_IMAGES"
    fi

    # Add one extra CoreOS run for virtual provision (not image mode)
    if [[ "$PROVISION_HOW" == "virtual" ]] && [[ "$IMAGE_MODE" != "yes" ]]; then
        rlPhaseStartTest "Test fedora-coreos ($PROVISION_HOW)"
	    rlRun -s "$tmt --image fedora-coreos $basic"
	    rlAssertGrep "Recommended packages failed to install, continuing regardless:" $rlRun_LOG
	    rlAssertGrep "forest: recommended by: /test/mixed" $rlRun_LOG
	    rlAssertGrep "weird-package: recommended by: /test/weird" $rlRun_LOG
	    rlAssertNotGrep "dconf: recommended by: /test/mixed" $rlRun_LOG
        rlPhaseEnd
    fi

    rlPhaseStartCleanup
        rlRun "popd"
    rlPhaseEnd
rlJournalEnd