#!/bin/sh
# test $TEST_REBOOTS successful reboots in a row
# Author: Martin Pitt <martin.pitt@ubuntu.com>
# For bisecting/testing you can replace individual binaries in /lib/systemd
# with --copy /host/path/systemd-foo:/tmp/systemd-replace/systemd-foo
set -e

IS_SYSTEM_RUNNING_TIMEOUT=300
TEST_REBOOTS=5

. `dirname $0`/assert.sh

fail() {
    [ -n "$1" ] && echo "$1"
    set +e
    journalctl --sync
    journalctl -a > "$AUTOPKGTEST_ARTIFACTS/boot-smoke-journal.txt"
    systemctl --no-pager --no-legend list-jobs > "$AUTOPKGTEST_ARTIFACTS/boot-smoke-running-jobs.txt"
    udevadm info --export-db > "$AUTOPKGTEST_ARTIFACTS/boot-smoke-udevdb.txt"
    exit 1
}

if [ -z "$AUTOPKGTEST_REBOOT_MARK" ]; then
    # enable persistent journal
    mkdir -p /var/log/journal
    # allow X to start even on headless machines
    mkdir -p /etc/X11/xorg.conf.d/
    cat << EOF > /etc/X11/xorg.conf.d/dummy.conf
Section "Device"
    Identifier "test"
    Driver "dummy"
EndSection
EOF


    AUTOPKGTEST_REBOOT_MARK=0
    if [ -d /tmp/systemd-replace/ ]; then
        for f in /tmp/systemd-replace/*; do
            echo "Installing $f..."
            rm -f /lib/systemd/$(basename $f)
            cp $f /lib/systemd/
        done
    fi
else
    echo "waiting to boot..."
    TIMEOUT=${IS_SYSTEM_RUNNING_TIMEOUT}
    while [ $TIMEOUT -ge 0 ]; do
        state="$(systemctl is-system-running || true)"
        case $state in
            running|degraded)
                break
                ;;
            *)
                sleep 1
                TIMEOUT=$((TIMEOUT - 1))
		;;
        esac
    done

    echo "checking for running system"
    if [ "$state" = "degraded" ]; then
        systemctl --no-pager --no-legend --failed list-units > "$AUTOPKGTEST_ARTIFACTS/boot-smoke-failed-units.txt" || true
        echo "systemctl is-system-running: degraded (non-fatal)"
    elif [ "$state" != "running" ]; then
        fail "system not running after timeout $IS_SYSTEM_RUNNING_TIMEOUT, state: $state"
    fi

    echo "checking for failed unmounts for user systemd"
    # grep complete journal to catch shutdown messages
    if journalctl | grep -E "systemd\[([2-9]|[1-9][0-9]+)\].*Failed unmounting"; then
        fail "found failed unmount in journal"
    fi

    # grep only this boot's journal, earlier ones complain about missing "render" group
    echo "checking for connection timeouts"
    if journalctl -b | grep "Connection timed out"; then
        fail "found connection timeout in journal for this boot"
    fi

    echo "checking that NetworkManager runs"
    pidof NetworkManager || fail "NetworkManager was not running"
fi

if [ "$AUTOPKGTEST_REBOOT_MARK" -ge "$TEST_REBOOTS" ]; then
    exit 0
fi

echo "reboot #$AUTOPKGTEST_REBOOT_MARK"
/tmp/autopkgtest-reboot $(($AUTOPKGTEST_REBOOT_MARK + 1))
