#!/bin/bash

# Runs state.show_highstate using all localized grains' combinations

if [[ $(whoami) != 'root' ]]; then
    if [[ -f /usr/bin/sudo ]]; then
        SUDO='/usr/bin/sudo'
    else
        echo 'Please install sudo first, or run this script as root'
        exit 1
    fi
fi

RUN_TEST="salt-call --local --retcode-passthrough state.show_highstate"
ALL_VIRTUAL=(
    kvm
)
ALL_OS=(
    Leap,42,3
    SLES,11,4
    SLES,12,3
)
ALL_LOCATIONS=( $(bin/get_valid_custom_grains.py) )

write_grains() {
    $SUDO sed -i -e "s/\(city:\).*/\1 $2/" -e "s/\(country:\).*/\1 $1/" -e "s/\(osfullname:\).*/\1 $4/" -e "s/\(osmajorrelease:\).*/\1 $5/" \
        -e "s/\(osrelease_info:\).*/\1 [$5, $6]/" -e "s/\(virt_cluster:\).*/\1 $3/" -e "s/\(virtual:\).*/\1 $7/" /etc/salt/grains
    echo "Grains: osfullname: $4, osmajorrelease: $5, city: $2, country: $1, virt_cluster: $3, virtual: $7"
}

for os in ${ALL_OS[@]}; do
    for location in ${ALL_LOCATIONS[@]}; do
        for virtual in ${ALL_VIRTUAL[@]}; do
            write_grains ${location//,/ } ${os//,/ } ${virtual}
            $RUN_TEST > /dev/null
            _STATUS=$(echo $?)
            # We ignore exit code 2 as it means that an empty file is produced
            # See https://github.com/saltstack/salt/issues/39172
            if [[ $_STATUS -eq 0 ]] || [[ $_STATUS -eq 2 ]]; then
                echo 'PASSED'
            else
                echo 'FAILED'
                STATUS=1
            fi
            echo
        done
    done
done

exit $STATUS