# Source library for shell script tests
#
# Copyright (C) 2011 Colin Walters <walters@verbum.org>
#
# 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, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
#
# Author: Colin Walters <walters@verbum.org>

TMPDIR=${TMPDIR:-/tmp}
export TMPDIR
test_tmpdir=`mktemp -d "$TMPDIR/ostree-tests.XXXXXXXXXX"`
cd "$test_tmpdir"
touch "$test_tmpdir/.test$$"

die () {
    if test -z "$HT_TESTS_SAVE_TEMPS"; then
        test -f "$test_tmpdir/.test$$" && rm -rf "$test_tmpdir"
    else
        echo "Temporary files saved in $test_tmpdir"
    fi
}

assert_has_file () {
    test -f "$1" || (echo "Couldn't find '$1'"; exit 1)
}

assert_not_has_file () {
    if test -f "$1"; then
	echo "File '$1' exists"; exit 1
    fi
}

assert_file_has_content () {
    if ! grep -q "$2" "$1"; then
	echo "File '$1' doesn't match regexp '$2'"; exit 1
    fi
}

setup_test_repository1 () {
    mkdir files
    cd files
    ht_files=`pwd`
    export ht_files
    echo first > firstfile
    echo second > secondfile

    mkdir ../repo
    ot_repo="--repo=../repo"
    export ot_repo
    ostree init $ot_repo
    ostree commit $ot_repo -s "Test Commit 1" -b "Commit body first" --add=firstfile
    ostree commit $ot_repo -s "Test Commit 2" -b "Commit body second" --add=secondfile
    ostree fsck -q $ot_repo
}

setup_test_repository2 () {
    mkdir files
    cd files
    ht_files=`pwd`
    export ht_files
    echo first > firstfile
    mkdir baz
    echo moo > baz/cow
    echo alien > baz/saucer
    mkdir baz/deeper
    echo hi > baz/deeper/ohyeah
    mkdir baz/another/
    echo x > baz/another/y

    cd ..
    mkdir repo
    cd repo
    ot_repo="--repo=`pwd`"
    cd ../files
    export ot_repo
    ostree init $ot_repo
    ostree commit $ot_repo -s "Test Commit 1" -b "Commit body first" --add=firstfile
    ostree commit $ot_repo -s "Test Commit 2" -b "Commit body second" --add=baz/cow  --add=baz/saucer --add=baz/deeper/ohyeah --add=baz/another/y
    ostree fsck -q $ot_repo
}

trap 'die' EXIT