#!/bin/bash

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

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

#

#   runtest.sh of /CoreOS/postgresql/Sanity/pg_repack

#   Description: Sanity test for pg_repackaudit extension

#   Author: Honza Horak <hhorak@redhat.com>

#

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

#

#   Copyright (c) 2015 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

PACKAGE="postgresql"

rlJournalStart

    rlPhaseStart FAIL "Setup"

        rlRun "rlImport database/postgresql"

        PACKAGES="$PACKAGES ${postgresqlPackagePrefix}pg_repack \

                  ${postgresqlPackagePrefix}${PACKAGE} \

                  ${postgresqlPackagePrefix}${PACKAGE}-contrib \

                  ${postgresqlPackagePrefix}${PACKAGE}-server";

        for PACKAGE in $PACKAGES; do

            rlAssertRpm $PACKAGE

        done

        rlRun "postgresqlStop";

        [[ -d ${postgresqlDataDir} ]] && rlFileBackup "${postgresqlDataDir}"

        rlRun "postgresqlCleanup"

        rlRun "TmpDir=\`mktemp -d\`" 0 "Creating tmp directory"

        rlRun "pushd $TmpDir"

        rlRun "postgresqlStart"

    rlPhaseEnd

    rlPhaseStartTest

        rlRun "postgresqlQuery \"DROP EXTENSION IF EXISTS pg_repack;\"";

        rlRun "postgresqlQuery \"CREATE DATABASE testrepack;\"" 0;

        rlRun "postgresqlQuery \"CREATE EXTENSION pg_repack;\" testrepack" 0;

        rlRun "postgresqlExec 'pgbench -i -s 50 testrepack' postgres"

        rlRun "postgresqlExec 'pg_repack --no-order --table pgbench_accounts testrepack -e' postgres"

        rlRun "postgresqlQuery \"DROP EXTENSION pg_repack;\" testrepack";

        rlRun "postgresqlQuery \"DROP DATABASE testrepack;\"" 0;

    rlPhaseEnd

    rlPhaseStart WARN "Cleanup";

        rm -f $rlRun_LOG;

        rlRun "postgresqlCleanup";

        rlFileRestore;

        rlRun "popd"

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

    rlPhaseEnd

rlJournalPrintText

rlJournalEnd