#!/bin/bash

. $(dirname $0)/../include.rc
. $(dirname $0)/../volume.rc
. $(dirname $0)/../nfs.rc

cleanup;

QDD=$(dirname $0)/quota
# compile the test write program and run it
build_tester $(dirname $0)/quota.c -o $QDD

TESTS_EXPECTED_IN_LOOP=9

TEST glusterd

# --------------------------------------------------
# Create, start and mount a volume with single brick
# --------------------------------------------------

TEST $CLI volume create $V0 replica 2 $H0:$B0/${V0}{1,2};

TEST $CLI volume start $V0

TEST $GFS -s $H0 --volfile-id $V0 $M0
TEST mkdir -p $M0/test_dir

#--------------------------------------------------------
# Enable quota of the volume and set hard and soft timeout
#------------------------------------------------------

TEST $CLI volume quota $V0 enable
EXPECT 'on' volinfo_field $V0 'features.quota'
TEST $CLI volume quota $V0 soft-timeout 0
EXPECT '0' volinfo_field $V0 'features.soft-timeout'
TEST $CLI volume quota $V0 hard-timeout 0
EXPECT '0' volinfo_field $V0 'features.hard-timeout'

#-------------------------------------------------------
# Set quota limits on the directory and
# verify if the limits are being reflected properly
#------------------------------------------------------

TEST $CLI volume quota $V0 limit-objects /test_dir 10
EXPECT "10" quota_object_list_field "/test_dir" 2

TEST $CLI volume quota $V0 limit-usage /test_dir 100MB
EXPECT "100.0MB" quota_list_field "/test_dir" 2

#------------------------------------------------------
# Check the quota enforcement mechanism for object count
#-------------------------------------------------------

# Try creating 9 files and it should succeed as object limit
# is set to 10, since directory where limit is set is accounted
# as well.

for i in {1..9}; do
        #TEST_IN_LOOP touch $M0/test_dir/test$i.txt
        TEST_IN_LOOP $QDD $M0/test_dir/test$i.txt 256 4
done
EXPECT_WITHIN $MARKER_UPDATE_TIMEOUT "9" quota_object_list_field "/test_dir" 4
EXPECT_WITHIN $MARKER_UPDATE_TIMEOUT "9.0MB" quotausage "/test_dir"

# Check available limit
EXPECT_WITHIN $MARKER_UPDATE_TIMEOUT "0" quota_object_list_field "/test_dir" 6
EXPECT_WITHIN $MARKER_UPDATE_TIMEOUT "91.0MB" quota_list_field "/test_dir" 5

# Check if hard-limit exceeded
EXPECT "Yes" quota_object_list_field "/test_dir" 8

# Check if soft-limit exceeded
EXPECT "Yes" quota_object_list_field "/test_dir" 7

# Creation of 11th file should throw out an error
TEST ! touch $M0/test_dir/test11.txt

#-------------------------------------------------------
# remove  quota limits on the directory and
# verify if the limit show 'N/A' and displayes only the usage
#------------------------------------------------------
TEST $CLI volume quota $V0 remove-objects /test_dir
EXPECT "N/A" quota_object_list_field "/test_dir" 2
EXPECT "9" quota_object_list_field "/test_dir" 4

TEST $CLI volume quota $V0 remove /test_dir
EXPECT "N/A" quota_list_field "/test_dir" 2
EXPECT "9.0MB" quotausage "/test_dir" 4

# Set back the limits
TEST $CLI volume quota $V0 limit-objects /test_dir 10
EXPECT "10" quota_object_list_field "/test_dir" 2

# Remove all files and verify the file count
TEST rm -rf $M0/test_dir/test*
EXPECT_WITHIN $MARKER_UPDATE_TIMEOUT "0" quota_object_list_field "/test_dir" 4

rm -f $QDD
cleanup;
#G_TESTDEF_TEST_STATUS_NETBSD7=BAD_TEST,BUG=1332021