#!/bin/bash

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

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

#

#   runtest.sh of /CoreOS/lftp/Sanity/sftp

#   Description: Basic test for SFTP file transfer

#   Author: Martin Frodl <mfrodl@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="lftp"

rlJournalStart

    rlPhaseStartSetup

        rlAssertRpm "$PACKAGES"

        rlRun "useradd lftp" 0 "Creating user lftp"

        rlRun "passwd --stdin lftp <<< lftp" 0 "Setting password"

        rlRun "REMOTE=\$(mktemp -d)" 0 "Creating remote directory"

        rlRun "touch ${REMOTE}/file1" 0 "Creating remote file"

        rlRun "mkdir ${REMOTE}/dir" 0 "Creating remote subdirectory"

        rlRun "touch ${REMOTE}/dir/file2" 0 "Creating another remote file"

        rlRun "chmod -R 0755 ${REMOTE}" 0 "Allowing access to remote directory for user lftp"

        rlRun "LOCAL=\$(mktemp -d)" 0 "Creating local directory"

        rlRun "pushd ${LOCAL}"

        if ! rlIsRHEL '<8'; then

            if ! test -e ~/.ssh/known_hosts; then

                rlRun "mkdir -p ~/.ssh" 0 "Create ~/.ssh"

            fi

            rlRun "ssh-keyscan $(hostname) >> ~/.ssh/known_hosts" 0 "Add localhost to known_hosts"

        fi

    rlPhaseEnd

    rlPhaseStartTest "Mirroring files and directories"

        rlRun "lftp -u lftp,lftp -e 'mirror ${REMOTE} ${LOCAL}; exit' sftp://${HOSTNAME}" \

            0 "Mirroring ${REMOTE} to ${LOCAL}"

        rlAssertExists "file1"

        rlAssertExists "dir"

        rlAssertExists "dir/file2"

        rlRun "rm -rf *" 0 "Cleaning local directory"

    rlPhaseEnd

    rlPhaseStartTest "Mirroring symlinks [BZ#1305235]"

        rlRun "TARGET=\$(mktemp -d)" 0 "Creating another directory"

        rlRun "touch ${TARGET}/file3" 0 "Creating file inside it"

        rlRun "ln -s ${TARGET} ${REMOTE}/symlink1" 0 "Creating symlink to it"

        rlRun "chmod -R 0755 ${TARGET}" 0 "Allowing access to remote directory for user lftp"

        rlRun "lftp -u lftp,lftp -e 'mirror -L ${REMOTE} ${LOCAL}; exit' sftp://${HOSTNAME}" \

            0 "Mirroring ${REMOTE} to ${LOCAL} (follow symlinks)"

        rlAssertExists "symlink1"

        rlAssertExists "symlink1/file3"

        rlRun "rm -rf *" 0 "Cleaning local directory"

    rlPhaseEnd

    rlPhaseStartTest "Mirroring circular symlinks"

        rlRun "ln -s ${REMOTE} ${TARGET}/symlink2" 0 "Creating symlink cycle"

        rlRun "lftp -u lftp,lftp  -e 'mirror -L ${REMOTE} ${LOCAL}; exit' sftp://${HOSTNAME}" \

            0,41,1 "Mirroring ${REMOTE} to ${LOCAL} (follow symlinks)"

        rlAssertExists "symlink1"

        rlAssertExists "symlink1/file3"

        rlAssertExists "symlink1/symlink2"

        rlAssertExists "symlink1/symlink2/file1"

        rlRun "rm -rf *" 0 "Cleaning local directory"

    rlPhaseEnd

    rlPhaseStartCleanup

        rlRun "popd"

        rlRun "rm -r ${REMOTE} ${LOCAL} ${TARGET}" 0 "Removing test directories"

        rlRun "userdel -rf lftp" 0 "Deleting user lftp"

    rlPhaseEnd

rlJournalPrintText

rlJournalEnd