#!/bin/bash -
# libguestfs
# Copyright (C) 2010-2017 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, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

# Engage in some montecarlo testing of virt-make-fs.  This test is
# copied from the original Perl tool virt-make-fs, on the basis that
# the new C tool should be able to pass the same tests.

set -e

$TEST_FUNCTIONS
skip_if_skipped

# Check which filesystems are supported by the appliance.
eval $(
perl -MSys::Guestfs -e '
  $g = Sys::Guestfs->new();
  $g->add_drive ("/dev/null");
  $g->launch ();
  $g->feature_available (["ntfs3g"]) and print "ntfs3g_available=yes\n";
  $g->feature_available (["ntfsprogs"]) and print "ntfsprogs_available=yes\n";
  $g->feature_available (["btrfs"]) and print "btrfs_available=yes\n";
')

# Allow btrfs to be disabled when btrfs is broken (eg. RHBZ#863978).
if [ -n "$SKIP_TEST_VIRT_MAKE_FS_BTRFS" ]; then
    btrfs_available=
fi

# UML backend doesn't support qcow2.
if [ "$(guestfish get-backend)" != "uml" ]; then
    qcow2_supported=yes
fi

declare -a choices

# Return a random element from the array 'choices'.
function random_choice
{
    echo "${choices[$((RANDOM % ${#choices[*]}))]}"
}

# Can't test vfat because we cannot create a tar archive
# where files are owned by UID:GID 0:0.  As a result, tar
# in the appliance fails when trying to change the UID of
# the files to some non-zero value (not supported by FAT).
choices=(--type=ext2 --type=ext3 --type=ext4)
if [ "$ntfs3g_available" = "yes" -a "$ntfsprogs_available" = "yes" ]; then
    choices[${#choices[*]}]="--type=ntfs"
fi
if [ "$btrfs_available" = "yes" ]; then
    choices[${#choices[*]}]="--type=btrfs"
fi
type=`random_choice`

if [ "$qcow2_supported" = "yes" ]; then
    choices=("" --format=raw --format=qcow2)
    format=`random_choice`
else
    format="--format=raw"
fi

choices=(--partition --partition=gpt --partition=mbr)
partition=`random_choice`

choices=("" --size=+1M)
size=`random_choice`

choices=("" --label=FOO)
label=`random_choice`

if [ -n "$LIBGUESTFS_DEBUG" ]; then debug=--debug; fi

params="$type $format $partition $size $label $debug"
echo "test-virt-make-fs: parameters: $params"

rm -f test.file test.tar output.img

tarsize=$((RANDOM & 8191))
echo "test-virt-make-fs: size of test file: $tarsize KB"
dd if=/dev/zero of=test.file bs=1024 count=$tarsize
tar -c -f test.tar test.file
rm test.file

$VG virt-make-fs $params -- test.tar output.img

rm test.tar output.img
