tests: add flock tests

Signed-off-by: Karel Zak <kzak@redhat.com>

diff --git a/tests/commands.sh b/tests/commands.sh
index e0a9b0d13831de5c0285a2bcff689cfa256f4737..ac4a490cf80d561701c9f3d68609431472742318 100644 (file)
--- a/tests/commands.sh
+++ b/tests/commands.sh
@@ -48,6 +48,7 @@ TS_CMD_COLUMN=${TS_CMD_COLUMN:-"$top_builddir/column"}
 TS_CMD_EJECT=${TS_CMD_EJECT-"$top_builddir/eject"}
 TS_CMD_FALLOCATE=${TS_CMD_FALLOCATE-"$top_builddir/fallocate"}
 TS_CMD_FDISK=${TS_CMD_FDISK-"$top_builddir/fdisk"}
+TS_CMD_FLOCK=${TS_CMD_FLOCK-"$top_builddir/flock"}
 TS_CMD_SFDISK=${TS_CMD_SFDISK-"$top_builddir/sfdisk"}
 TS_CMD_FINDMNT=${TS_CMD_FINDMNT-"$top_builddir/findmnt"}
 TS_CMD_FSCKCRAMFS=${TS_CMD_FSCKCRAMFS:-"$top_builddir/fsck.cramfs"}

diff --git a/tests/expected/misc/flock b/tests/expected/misc/flock
new file mode 100644 (file)
index 0000000..9c97396
--- /dev/null
+++ b/tests/expected/misc/flock
@@ -0,0 +1,3 @@
+Locking
+Unlocking
+Unlocked

diff --git a/tests/expected/misc/flock-exclusive b/tests/expected/misc/flock-exclusive
new file mode 100644 (file)
index 0000000..3582111
--- /dev/null
+++ b/tests/expected/misc/flock-exclusive
@@ -0,0 +1 @@
+Success

diff --git a/tests/expected/misc/flock-no-fork b/tests/expected/misc/flock-no-fork
new file mode 100644 (file)
index 0000000..3582111
--- /dev/null
+++ b/tests/expected/misc/flock-no-fork
@@ -0,0 +1 @@
+Success

diff --git a/tests/expected/misc/flock-non-block b/tests/expected/misc/flock-non-block
new file mode 100644 (file)
index 0000000..3582111
--- /dev/null
+++ b/tests/expected/misc/flock-non-block
@@ -0,0 +1 @@
+Success

diff --git a/tests/expected/misc/flock-shared b/tests/expected/misc/flock-shared
new file mode 100644 (file)
index 0000000..011f325
--- /dev/null
+++ b/tests/expected/misc/flock-shared
@@ -0,0 +1,2 @@
+Have shared lock
+Success

diff --git a/tests/expected/misc/flock-time-check b/tests/expected/misc/flock-time-check
new file mode 100644 (file)
index 0000000..2e9ba47
--- /dev/null
+++ b/tests/expected/misc/flock-time-check
@@ -0,0 +1 @@
+success

diff --git a/tests/expected/misc/flock-timeout b/tests/expected/misc/flock-timeout
new file mode 100644 (file)
index 0000000..b4cd5a4
--- /dev/null
+++ b/tests/expected/misc/flock-timeout
@@ -0,0 +1,2 @@
+After timeout.
+Success

diff --git a/tests/ts/misc/flock b/tests/ts/misc/flock
new file mode 100755 (executable)
index 0000000..b2fc772
--- /dev/null
+++ b/tests/ts/misc/flock
@@ -0,0 +1,95 @@
+#!/bin/bash
+
+# This file is part of util-linux.
+#
+# This file 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 file 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.
+
+TS_TOPDIR="${0%/*}/../.."
+TS_DESC="flock"
+
+. $TS_TOPDIR/functions.sh
+ts_init "$*"
+
+ts_check_test_command "$TS_CMD_FLOCK"
+
+
+function do_lock {
+       local opts="$1"
+       local expected_rc="$2"
+       local mesg="$3"
+
+       $TS_CMD_FLOCK $1 $TS_OUTDIR/lockfile \
+               echo "$mesg" \
+               > $TS_OUTPUT 2>&1
+
+       local rc="$?"
+
+       if [  "$rc" == "$expected_rc" ]; then
+               ts_log "Success"
+       else
+               ts_log "Failed [rc=$rc]"
+       fi
+}
+
+# general lock
+GEN_OUTPUT="$TS_OUTPUT"
+START=$(date '+%s')
+$TS_CMD_FLOCK --shared $TS_OUTDIR/lockfile \
+       bash -c 'echo "Locking"; sleep 3; echo "Unlocking"' \
+       > $GEN_OUTPUT 2>&1 &
+
+
+ts_init_subtest "non-block"
+do_lock "--nonblock --conflict-exit-code 123" 123 "You will never see this!"
+ts_finalize_subtest
+
+
+ts_init_subtest "no-fork"
+do_lock "--no-fork --nonblock --conflict-exit-code 123" 123 "You will never see this!"
+ts_finalize_subtest
+
+
+ts_init_subtest "shared"
+do_lock "--shared" 0 "Have shared lock"
+ts_finalize_subtest
+
+
+# this is the same as non-block test (exclusive lock is the default), but here
+# we explicitly specify --exclusive on command line
+ts_init_subtest "exclusive"
+do_lock "--nonblock --exclusive --conflict-exit-code 123" 123 "You will never see this!"
+ts_finalize_subtest
+
+
+ts_init_subtest "timeout"
+do_lock "--timeout 5 --conflict-exit-code 5" 0 "After timeout."
+END=$(date '+%s')
+ts_finalize_subtest
+
+
+# expected is 3 seconds (see "sleep 3" for the general lock), but we should not
+# rely on exact number due to scheduler, machine load, etc. Let's check for
+# inmterval <3,5>.
+# 
+ts_init_subtest "time-check"
+TIMEDIFF=$(( $END - $START ))
+if [ $TIMEDIFF -lt 3 ]; then
+       ts_log "general lock failed [$TIMEDIFF sec]"
+elif [ $TIMEDIFF -gt 5 ]; then
+       ts_log "wait too long [$TIMEDIFF sec]"
+else
+       ts_log "success"
+fi
+ts_finalize_subtest "diff ${TIMEDIFF} sec"
+
+
+echo "Unlocked" >> $GEN_OUTPUT 
+ts_finalize