#!/bin/bash

# vim: dict+=/usr/share/beakerlib/dictionary.vim cpt=.,w,b,u,t,i,k

# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

#

#   runtest.sh of /CoreOS/bash/Sanity/login-shell

#   Description: the test creates user with bash as login shell and verified the user can log in and get th

#   Author: Martin Kyral <mkyral@redhat.com>

#

# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

#

#   Copyright (c) 2016 Red Hat, Inc.

#

#   This program is free software: you can redistribute it and/or

#   modify it under the terms of the GNU General Public License as

#   published by the Free Software Foundation, either version 2 of

#   the License, or (at your option) any later version.

#

#   This program is distributed in the hope that it will be

#   useful, but WITHOUT ANY WARRANTY; without even the implied

#   warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR

#   PURPOSE.  See the GNU General Public License for more details.

#

#   You should have received a copy of the GNU General Public License

#   along with this program. If not, see http://www.gnu.org/licenses/.

#

# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

# Include Beaker environment

. /usr/share/beakerlib/beakerlib.sh || exit 1

PACKAGES=${PACKAGES:-"bash"}

SH_BIN=${SH_BIN:-"bash"}

rlJournalStart

    rlPhaseStartSetup

        rlAssertRpm --all

        rlAssertBinaryOrigin $SH_BIN

        rlRun "TmpDir=\$(mktemp -d)" 0 "Creating tmp directory"

        rlRun "cp script.exp ssh.exp tst.sh $TmpDir"

        rlRun "pushd $TmpDir"

        #USER=$(cat /dev/urandom | tr -dc 'a-zA-Z0-9' | fold -w 8 | head -n 1)

        USER="user${RANDOM}"

        rlRun "useradd -s /bin/${SH_BIN} -d /home/${USER} ${USER}"

        rlRun "echo krava | passwd --stdin ${USER}"

        rlRun "sed -i 's/shelluser/${USER}/g' script.exp"

        rlRun "sed -i 's/shelluser/${USER}/g' ssh.exp"

        rlRun "sed -i 's/shelluser/${USER}/g' tst.sh"

        rlRun "sed -i 's/shellbin/${SH_BIN}/g' script.exp"

        rlRun "sed -i 's/shellbin/${SH_BIN}/g' ssh.exp"

        rlRun "sed -i 's/shellbin/${SH_BIN}/g' tst.sh"

        rlRun "cp tst.sh /home/${USER}/"

        rlRun "chown ${USER} /home/${USER}/tst.sh"

        rlRun "mv /etc/profile.d/tps-cd.sh /root/tps-cd.sh" 0,1 "Backing tps-cd.sh up"

    rlPhaseEnd

    rlPhaseStartTest

        rlLog "::::: 'su -c' test :::::"

        rlRun "su ${USER} -c 'echo something' | tee test1.log" 0 "echo something test"

        rlLog "test1.log: $(cat test1.log)"

        rlAssertGrep "something" "test1.log"

        rlRun "su ${USER} -c 'echo \$0 | tee /home/${USER}/ps1.log'" 0 "shell detection test"

        rlLog "ps1.log: $(cat /home/${USER}/ps1.log)"

        rlAssertGrep "$SH_BIN" "/home/${USER}/ps1.log"

        # the following does not work with pdksh for some reason

        # not a big deal, ssh test can substitute this

        [ "$SH_BIN" != "pdksh" ] && {

        rlLog "::::: 'su -l' test :::::"

        expect script.exp &

        sleep 20

        pgrep expect && pkill expect

        rlLog "test2.log: $(cat /home/${USER}/test2.log)"

        rlAssertGrep "something" "/home/${USER}/test2.log"

        rlLog "ps2.log: $(cat /home/${USER}/ps2.log)"

        rlAssertGrep "$SH_BIN" "/home/${USER}/ps2.log"

        }

        rlLog "::::: ssh login test :::::"

        expect ssh.exp &

        sleep 20

        pgrep expect && pkill expect

        rlLog "ps3.log: $(cat /home/${USER}/ps3.log)"

        rlAssertGrep "$SH_BIN" "/home/${USER}/ps3.log"

        rlAssertExists "/home/${USER}/tst.dat"

    rlPhaseEnd

    rlPhaseStartCleanup

        rlRun "mv /root/tps-cd.sh /etc/profile.d/tps-cd.sh" 0,1 "Restoring tps-cd.sh"

        rlRun "su -c 'kill -9 -1' ${USER}" 0 "Killing all processes of ${USER}"

        rlRun "userdel -r ${USER}"

        rlRun "popd"

        rlRun "rm -r $TmpDir" 0 "Removing tmp directory"

    rlPhaseEnd

rlJournalPrintText

rlJournalEnd