1 | /* SPDX-License-Identifier: GPL-2.0 */ |
2 | /* |
3 | * Data Access Monitor Unit Tests |
4 | * |
5 | * Author: SeongJae Park <sj@kernel.org> |
6 | */ |
7 | |
8 | #ifdef CONFIG_DAMON_SYSFS_KUNIT_TEST |
9 | |
10 | #ifndef _DAMON_SYSFS_TEST_H |
11 | #define _DAMON_SYSFS_TEST_H |
12 | |
13 | #include <kunit/test.h> |
14 | |
15 | static unsigned int nr_damon_targets(struct damon_ctx *ctx) |
16 | { |
17 | struct damon_target *t; |
18 | unsigned int nr_targets = 0; |
19 | |
20 | damon_for_each_target(t, ctx) |
21 | nr_targets++; |
22 | |
23 | return nr_targets; |
24 | } |
25 | |
26 | static int __damon_sysfs_test_get_any_pid(int min, int max) |
27 | { |
28 | struct pid *pid; |
29 | int i; |
30 | |
31 | for (i = min; i <= max; i++) { |
32 | pid = find_get_pid(nr: i); |
33 | if (pid) { |
34 | put_pid(pid); |
35 | return i; |
36 | } |
37 | } |
38 | return -1; |
39 | } |
40 | |
41 | static void damon_sysfs_test_set_targets(struct kunit *test) |
42 | { |
43 | struct damon_sysfs_targets *sysfs_targets; |
44 | struct damon_sysfs_target *sysfs_target; |
45 | struct damon_ctx *ctx; |
46 | |
47 | sysfs_targets = damon_sysfs_targets_alloc(); |
48 | sysfs_targets->nr = 1; |
49 | sysfs_targets->targets_arr = kmalloc_array(n: 1, |
50 | size: sizeof(*sysfs_targets->targets_arr), GFP_KERNEL); |
51 | |
52 | sysfs_target = damon_sysfs_target_alloc(); |
53 | sysfs_target->pid = __damon_sysfs_test_get_any_pid(min: 12, max: 100); |
54 | sysfs_target->regions = damon_sysfs_regions_alloc(); |
55 | sysfs_targets->targets_arr[0] = sysfs_target; |
56 | |
57 | ctx = damon_new_ctx(); |
58 | |
59 | damon_sysfs_set_targets(ctx, sysfs_targets); |
60 | KUNIT_EXPECT_EQ(test, 1u, nr_damon_targets(ctx)); |
61 | |
62 | sysfs_target->pid = __damon_sysfs_test_get_any_pid( |
63 | min: sysfs_target->pid + 1, max: 200); |
64 | damon_sysfs_set_targets(ctx, sysfs_targets); |
65 | KUNIT_EXPECT_EQ(test, 1u, nr_damon_targets(ctx)); |
66 | |
67 | damon_destroy_ctx(ctx); |
68 | kfree(objp: sysfs_targets->targets_arr); |
69 | kfree(objp: sysfs_targets); |
70 | kfree(objp: sysfs_target); |
71 | } |
72 | |
73 | static struct kunit_case damon_sysfs_test_cases[] = { |
74 | KUNIT_CASE(damon_sysfs_test_set_targets), |
75 | {}, |
76 | }; |
77 | |
78 | static struct kunit_suite damon_sysfs_test_suite = { |
79 | .name = "damon-sysfs" , |
80 | .test_cases = damon_sysfs_test_cases, |
81 | }; |
82 | kunit_test_suite(damon_sysfs_test_suite); |
83 | |
84 | #endif /* _DAMON_SYSFS_TEST_H */ |
85 | |
86 | #endif /* CONFIG_DAMON_SYSFS_KUNIT_TEST */ |
87 | |