summaryrefslogtreecommitdiff
path: root/tests/modules/t_threadpool.sh
blob: ab84e4f316e29dd7a9b51c00c8e130514148d7ef (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
# $NetBSD: t_threadpool.sh,v 1.1 2019/01/25 18:34:45 christos Exp $
#
# Copyright (c) 2018 The NetBSD Foundation, Inc.
# All rights reserved.
#
# This code is derived from software contributed to The NetBSD Foundation
# by Jason R. Thorpe.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
# 1. Redistributions of source code must retain the above copyright
#    notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
#    notice, this list of conditions and the following disclaimer in the
#    documentation and/or other materials provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
# ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
# TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
# PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
# BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
#

# Pick an arbitrary priority that is not likely to be used.
tp_pri=5

# The kernel test jig includes a 1 second delay in the job.  We need to
# wait longer for it to complete.
job_delay=2

read_sysctl() {
	echo "${1} = ${2}" >expout
	atf_check -s eq:0 -o file:expout -e empty sysctl ${1}
}

write_sysctl() {
	atf_check -s eq:0 -o ignore -e empty sysctl -w "${1}=${2}"
}

write_sysctl_fail() {
	echo "${3}" >experr
	atf_check -s eq:1 -o ignore -e file:experr sysctl -w "${1}=${2}"
}

atf_test_case unbound cleanup
unbound_head() {
	atf_set "descr" "Test unbound thread pools"
	atf_set "require.user" "root"
}
unbound_body() {
	modload $(atf_get_srcdir)/threadpool_tester/threadpool_tester.kmod
	if [ $? -ne 0 ]; then
		atf_skip "cannot load threadpool_tester.kmod"
	fi

	# Ensure that the state is clean.
	read_sysctl kern.threadpool_tester.test_value 0

	# Create an unbound pool.
	write_sysctl kern.threadpool_tester.get_unbound $tp_pri

	# Do it again.  We expect this to fail, but the test jig will
	# do some additional threadpool object lifecycle validation.
	# (It will not hold the additional reference.)
	write_sysctl_fail kern.threadpool_tester.get_unbound $tp_pri \
	    "sysctl: kern.threadpool_tester.get_unbound: File exists"

	# Schedule the test jig job on the pool.
	# Wait for a short period of time and then check that the job
	# successfully ran.
	write_sysctl kern.threadpool_tester.run_unbound $tp_pri
	sleep $job_delay
	read_sysctl kern.threadpool_tester.test_value 1

	# ...and again.
	write_sysctl kern.threadpool_tester.run_unbound $tp_pri
	sleep $job_delay
	read_sysctl kern.threadpool_tester.test_value 2

	# Now destroy the threadpool.
	write_sysctl kern.threadpool_tester.put_unbound $tp_pri
}
unbound_cleanup() {
	modunload threadpool_tester >/dev/null 2>&1
}

atf_test_case percpu cleanup
percpu_head() {
	atf_set "descr" "Test percpu thread pools"
	atf_set "require.user" "root"
}
percpu_body() {
	modload $(atf_get_srcdir)/threadpool_tester/threadpool_tester.kmod
	if [ $? -ne 0 ]; then
		atf_skip "cannot load threadpool_tester.kmod"
	fi

	# Ensure that the state is clean.
	read_sysctl kern.threadpool_tester.test_value 0

	# Create an percpu pool.
	write_sysctl kern.threadpool_tester.get_percpu $tp_pri

	# Do it again.  We expect this to fail, but the test jig will
	# do some additional threadpool object lifecycle validation.
	# (It will not hold the additional reference.)
	write_sysctl_fail kern.threadpool_tester.get_percpu $tp_pri \
	    "sysctl: kern.threadpool_tester.get_percpu: File exists"

	# Schedule the test jig job on the pool.
	# Wait for a short period of time and then check that the job
	# successfully ran.
	write_sysctl kern.threadpool_tester.run_percpu $tp_pri
	sleep $job_delay
	read_sysctl kern.threadpool_tester.test_value 1

	# ...and again.
	write_sysctl kern.threadpool_tester.run_percpu $tp_pri
	sleep $job_delay
	read_sysctl kern.threadpool_tester.test_value 2

	# Now destroy the threadpool.
	write_sysctl kern.threadpool_tester.put_percpu $tp_pri
}
percpu_cleanup() {
	modunload threadpool_tester >/dev/null 2>&1
}

atf_init_test_cases()
{
	atf_add_test_case unbound
	atf_add_test_case percpu
}