summaryrefslogtreecommitdiff
path: root/tests/test_lib.sh
blob: d395ce2802e24124bf48d7d113697629f6f7dbb4 (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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
OK=" OK"
FAIL=" FAIL"

TESTDIR=`mktemp -d`
IMGFILE=$TESTDIR/tbdiff.img
ORIGIN=$TESTDIR/orig
TARGET=$TESTDIR/target

nums=( zero one two three four five six seven )
files=( null a.txt b.txt dirdir dirdir/fifo dirdir/symlinkb dirdir/chardev \
        dirdir/blockdev )

TOPDIR=`pwd`

insertfiles() {
    if [ $1 -ge 1 ]; then
        echo 1 > a.txt
    fi  
    if [ $1 -ge 2 ]; then 
        cat $TOPDIR/lipsum.txt > b.txt
    fi  
    if [ $1 -ge 3 ]; then 
        mkdir -p dirdir
    fi  
    if [ $1 -ge 4 ]; then 
        mkfifo dirdir/fifo
    fi  
    if [ $1 -ge 5 ]; then 
        ln -s ../b.txt dirdir/symlinkb
    fi  
    if [ $1 -ge 6 ]; then 
        mknod dirdir/chardev c `stat /dev/null -c '%t %T'`
    fi  
    if [ $1 -ge 7 ]; then 
        mknod dirdir/blockdev b `stat /dev/null -c '%t %T'`
    fi  
}

# check_same_mtime FILE_A FILE_B
check_same_mtime () {
	test $(stat -c %Y $1) = $(stat -c %Y $2)
}

# check_same_uidgid FILE_A FILE_B
check_same_uidgid () {
	test $(stat -c "%u.%g" $1) = $(stat -c "%u.%g" $2)
}

# check_same_mode FILE_A FILE_B
check_same_mode () {
	test $(stat -c "%f" $1) = $(stat -c "%f" $2)
}

# check_content FILE EXPECTED_OCTAL_PERMISSIONS
check_perm () {
	test $(stat -c %a $1) = $2
}

# check_content FILE EXPECTED_OCTAL_PERMISSIONS
check_symlink () {
	test $(readlink $1) = $2
}

# check_content FILE EXPECTED_CONTENT
check_content () {
	test $(cat $1) = $2
}

# check_group FILE EXPECTED_GROUP_NAME
check_group () {
	test $(stat -c %G $1) = $2
}

check_xattrs () {
	test "`getfattr -d $1 2>/dev/null | tail -n +2`" = \
	     "`getfattr -d $2 2>/dev/null | tail -n +2`"
}

# tests whether a command exists
is_command () {
	type $1 >/dev/null 2>/dev/null
}

#check_command COMMAND_STRING TEST_COMMAND COMMAND_DESCRIPTION
check_command () {
	COMMAND_STRING="$1"
	COMMAND_DESCRIPTION="$2"
	TEST_COMMAND="$3"
	eval $COMMAND_STRING
	RETVAL=$?
	if is_command "$TEST_COMMAND"; then #test explicitly checks return
		if $TEST_COMMAND $RETVAL; then
			if [ "$RETVAL" != "0" ]; then
				echo $COMMAND_STRING expected failure in \
				     $COMMAND_DESCRIPTION >&2
				echo $OK
				exit 0
			fi
		else
			if [ "$RETVAL" = "0" ]; then
				echo $COMMAND_STRING Unexpected success in \
				     $COMMAND_DESCRIPTION >&2
				echo $FAIL
				cleanup_and_exit
			else
				echo $COMMAND_STRING Unexpected failure in \
				     $COMMAND_DESCRIPTION >&2
				echo $FAIL
				cleanup_and_exit
			fi
		fi
	elif [ "$RETVAL" != "0" ]; then #return value expected to be 0
		echo $COMMAND_STRING Unexpected failure $COMMAND_DESCRIPTION >&2
		echo $FAIL
		cleanup_and_exit
	fi	
}

start () {
	if [ $# -ne 2 ]
	then
		echo "ERROR: Not enough arguments."
		cleanup_and_exit
	fi

	if [ ! -f "$1" ]
	then
		echo "ERROR: $1 is an invalid tbdiff-create path"  1>&2
		cleanup_and_exit
	fi

	if [ ! -f "$2" ]
	then
		echo "ERROR: $1 is an invalid tbdiff-deploy path" 1>&2
		cleanup_and_exit
	fi
}

cleanup_and_exit () {
	rm -rf $TESTDIR
	exit 1
}
command_succeeded () {
	test "$1" = "0"
}
main () {
	start "$@"
	echo -n "$TEST_ID Setting up $TEST_NAME test: "
	if [ ! -d $TESTDIR ]
	then
		echo $FAIL
		echo "Couldn't create temporary directory for test. " \
		     "Please check mktemp accepts -d and permissions." >&2
		cleanup_and_exit
	fi
	mkdir -p $ORIGIN &&
	check_command 'setup_origin' "$TEST_ID-$TEST_NAME: creating origin" \
	              'command_succeeded' &&
	mkdir -p $TARGET &&
	check_command 'setup_target' "$TEST_ID-$TEST_NAME: creating target" \
	              'command_succeeded' &&

	echo $OK

	echo "$TEST_ID Performing $TEST_NAME image creation and deployment: "
	sleep 2s &&
	CWD=$(pwd) &&
	check_command "$CREATE $IMGFILE $ORIGIN $TARGET" \
	              "$TEST_ID-$TEST_NAME: creating image" \
	              'create_test_return'

	cd $ORIGIN && 
	check_command "$DEPLOY $IMGFILE"  \
	              "$TEST_ID-$TEST_NAME: deploying image" \
	              'deploy_test_return'

	cd $CWD
	echo -n "$TEST_ID Checking $TEST_NAME results: "
	check_results
	if test "x$?" != "x0"
	then
		echo $FAIL
		echo "Applying image did not produce the expected results" 1>&2
		cleanup_and_exit
	fi
	echo $OK
}