freebsd-dev/bin/pwait/tests/pwait_test.sh
Pawel Jakub Dawidek dee496fc37 Implement tests for the newly added -o flag.
Sponsored by:	Fudo Security
2020-01-26 11:03:45 +00:00

323 lines
5.3 KiB
Bash

# $FreeBSD$
atf_test_case basic
basic_head()
{
atf_set "descr" "Basic tests on pwait(1) utility"
}
basic_body()
{
sleep 1 &
p1=$!
sleep 5 &
p5=$!
sleep 10 &
p10=$!
atf_check \
-o empty \
-e empty \
-s exit:0 \
timeout --preserve-status 15 pwait $p1 $p5 $p10
atf_check \
-o empty \
-e inline:"kill: $p1: No such process\n" \
-s exit:1 \
kill -0 $p1
atf_check \
-o empty \
-e inline:"kill: $p5: No such process\n" \
-s exit:1 \
kill -0 $p5
atf_check \
-o empty \
-e inline:"kill: $p10: No such process\n" \
-s exit:1 \
kill -0 $p10
}
basic_cleanup()
{
kill $p1 $p5 $p10 >/dev/null 2>&1
wait $p1 $p5 $p10 >/dev/null 2>&1
}
atf_test_case time_unit
time_unit_head()
{
atf_set "descr" "Test parsing the timeout unit and value"
}
time_unit_body()
{
init=1
atf_check \
-o empty \
-e inline:"pwait: timeout unit\n" \
-s exit:65 \
timeout --preserve-status 2 pwait -t 1d $init
atf_check \
-o empty \
-e inline:"pwait: timeout unit\n" \
-s exit:65 \
timeout --preserve-status 2 pwait -t 1d $init
atf_check \
-o empty \
-e inline:"pwait: timeout value\n" \
-s exit:65 \
timeout --preserve-status 2 pwait -t -1 $init
atf_check \
-o empty \
-e inline:"pwait: timeout value\n" \
-s exit:65 \
timeout --preserve-status 2 pwait -t 100000001 $init
# These long duration cases are expected to timeout from the
# timeout utility rather than pwait -t.
atf_check \
-o empty \
-e empty \
-s exit:143 \
timeout --preserve-status 2 pwait -t 100000000 $init
atf_check \
-o empty \
-e empty \
-s exit:143 \
timeout --preserve-status 2 pwait -t 1h $init
atf_check \
-o empty \
-e empty \
-s exit:143 \
timeout --preserve-status 2 pwait -t 1.5h $init
atf_check \
-o empty \
-e empty \
-s exit:143 \
timeout --preserve-status 2 pwait -t 1m $init
atf_check \
-o empty \
-e empty \
-s exit:143 \
timeout --preserve-status 2 pwait -t 1.5m $init
atf_check \
-o empty \
-e empty \
-s exit:143 \
timeout --preserve-status 2 pwait -t 0 $init
# The rest are fast enough that pwait -t is expected to trigger
# the timeout.
atf_check \
-o empty \
-e empty \
-s exit:124 \
timeout --preserve-status 2 pwait -t 1s $init
atf_check \
-o empty \
-e empty \
-s exit:124 \
timeout --preserve-status 2 pwait -t 1.5s $init
atf_check \
-o empty \
-e empty \
-s exit:124 \
timeout --preserve-status 2 pwait -t 1 $init
atf_check \
-o empty \
-e empty \
-s exit:124 \
timeout --preserve-status 2 pwait -t 1.5 $init
atf_check \
-o empty \
-e empty \
-s exit:124 \
timeout --preserve-status 2 pwait -t 0.5 $init
}
atf_test_case timeout_trigger_timeout
timeout_trigger_timeout_head()
{
atf_set "descr" "Test that exceeding the timeout is detected"
}
timeout_trigger_timeout_body()
{
sleep 10 &
p10=$!
atf_check \
-o empty \
-e empty \
-s exit:124 \
timeout --preserve-status 6.5 pwait -t 5 $p10
}
timeout_trigger_timeout_cleanup()
{
kill $p10 >/dev/null 2>&1
wait $p10 >/dev/null 2>&1
}
atf_test_case timeout_no_timeout
timeout_no_timeout_head()
{
atf_set "descr" "Test that not exceeding the timeout continues to wait"
}
timeout_no_timeout_body()
{
sleep 10 &
p10=$!
atf_check \
-o empty \
-e empty \
-s exit:0 \
timeout --preserve-status 11.5 pwait -t 12 $p10
}
timeout_no_timeout_cleanup()
{
kill $p10 >/dev/null 2>&1
wait $p10 >/dev/null 2>&1
}
atf_test_case timeout_many
timeout_many_head()
{
atf_set "descr" "Test timeout on many processes"
}
timeout_many_body()
{
sleep 1 &
p1=$!
sleep 5 &
p5=$!
sleep 10 &
p10=$!
atf_check \
-o empty \
-e empty \
-s exit:124 \
timeout --preserve-status 7.5 pwait -t 6 $p1 $p5 $p10
}
timeout_many_cleanup()
{
kill $p1 $p5 $p10 >/dev/null 2>&1
wait $p1 $p5 $p10 >/dev/null 2>&1
}
atf_test_case or_flag
or_flag_head()
{
atf_set "descr" "Test OR flag"
}
or_flag_body()
{
sleep 2 &
p2=$!
sleep 4 &
p4=$!
sleep 6 &
p6=$!
atf_check \
-o inline:"$p2: exited with status 0.\n" \
-e empty \
-s exit:0 \
timeout --preserve-status 15 pwait -o -v $p2 $p4 $p6
atf_check \
-o empty \
-e inline:"pwait: $p2: No such process\n" \
-s exit:0 \
timeout --preserve-status 15 pwait -o $p2 $p4 $p6
atf_check \
-o empty \
-e empty \
-s exit:0 \
timeout --preserve-status 15 pwait -o $p4 $p6
atf_check \
-o empty \
-e inline:"pwait: $p4: No such process\n" \
-s exit:0 \
timeout --preserve-status 15 pwait -o $p4 $p6
atf_check \
-o inline:"$p6: exited with status 0.\n" \
-e empty \
-s exit:0 \
timeout --preserve-status 15 pwait -o -v $p6
atf_check \
-o empty \
-e inline:"pwait: $p6: No such process\n" \
-s exit:0 \
timeout --preserve-status 15 pwait -o $p6
atf_check \
-o empty \
-e inline:"kill: $p2: No such process\n" \
-s exit:1 \
kill -0 $p2
atf_check \
-o empty \
-e inline:"kill: $p4: No such process\n" \
-s exit:1 \
kill -0 $p4
atf_check \
-o empty \
-e inline:"kill: $p6: No such process\n" \
-s exit:1 \
kill -0 $p6
}
or_flag_cleanup()
{
kill $p2 $p4 $p6 >/dev/null 2>&1
wait $p2 $p4 $p6 >/dev/null 2>&1
}
atf_init_test_cases()
{
atf_add_test_case basic
atf_add_test_case time_unit
atf_add_test_case timeout_trigger_timeout
atf_add_test_case timeout_no_timeout
atf_add_test_case timeout_many
atf_add_test_case or_flag
}