summaryrefslogtreecommitdiffstats
path: root/ctdb/tests/onnode/common.sh
blob: f920bc398a687cdebf3a6863d797df4a5425d450 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
# Hey Emacs, this is a -*- shell-script -*- !!!  :-)

# Set indirectly by run_tests at top level.
unset CTDB_NODES_SOCKETS

# Default to just "onnode".
: ${ONNODE:=onnode}

# Augment PATH with relevant stubs/ directories.

if [ -d "${ONNODE_TESTS_DIR}/stubs" ] ; then
    PATH="${ONNODE_TESTS_DIR}/stubs:$PATH"
fi

export ONNODE_TESTCASE_DIR=$(dirname "$0")
if [ $(basename "$ONNODE_TESTCASE_DIR") = "onnode" ] ; then
    # Just a test script, no testcase subdirectory.
    ONNODE_TESTCASE_DIR="$ONNODE_TESTS_DIR"
else
    if [ -d "${ONNODE_TESTCASE_DIR}/stubs" ] ; then
	PATH="${ONNODE_TESTCASE_DIR}/stubs:$PATH"
    fi
fi

# Find CTDB nodes file.
if [ -z "$CTDB_NODES_FILE" ] ; then
    if [ -r "${ONNODE_TESTCASE_DIR}/nodes" ] ; then
	CTDB_NODES_FILE="${ONNODE_TESTCASE_DIR}/nodes"
    elif [ -r "${ONNODE_TESTS_DIR}/nodes" ] ; then
	CTDB_NODES_FILE="${ONNODE_TESTS_DIR}/nodes"
    else
	CTDB_NODES_FILE="${CTDB_BASE:-/etc/ctdb}/nodes"
    fi
fi

export CTDB_NODES_FILE

export ONNODE_TESTS_VAR_DIR="${ONNODE_TESTS_DIR}/var"
mkdir -p "$ONNODE_TESTS_VAR_DIR"

if [ -z "$CTDB_BASE" ] ; then
    export CTDB_BASE=$(dirname "$CTDB_NODES_FILE")
fi

define_test ()
{
    _f="$0"
    _f="${_f#./}"  # strip leading ./
    _f="${_f%%/*}" # if subdir, strip off file
    _f="${_f%.sh}" # strip off .sh suffix if any

    echo "$_f $1 - $2"
}

# Set output for ctdb command.  Option 1st argument is return code.
ctdb_set_output ()
{
    _out="$ONNODE_TESTS_VAR_DIR/ctdb.out"
    cat >"$_out"

    _rc="$ONNODE_TESTS_VAR_DIR/ctdb.rc"
    echo "${1:-0}" >"$_rc"

    trap "rm -f $_out $_rc" 0
}

required_result ()
{
    required_rc="${1:-0}"
    required_output=$(cat)
}

simple_test ()
{
    _sort="cat"
    if [ "$1" = "-s" ] ; then
	shift
	_sort="sort"
    fi
    _out=$("$@" 2>&1)
    _rc=$?
    _out=$(echo "$_out" | $_sort )

    if [ "$_out" = "$required_output" -a $_rc = $required_rc ] ; then
	echo "PASSED"
    else
	cat <<EOF
CTDB_NODES_FILE="${CTDB_NODES_FILE}"
CTDB_BASE="$CTDB_BASE"
$(which ctdb)

##################################################
Required output (Exit status: ${required_rc}):
##################################################
$required_output
##################################################
Actual output (Exit status: ${_rc}):
##################################################
$_out
EOF
	return 1
    fi
}