1 // SPDX-License-Identifier: GPL-2.0-only
3 * KUnit tests for cpumask.
5 * Author: Sander Vanheule <sander@svanheule.net>
8 #include <kunit/test.h>
10 #include <linux/cpumask.h>
12 #define EXPECT_FOR_EACH_CPU_EQ(test, mask) \
14 const cpumask_t *m = (mask); \
15 int mask_weight = cpumask_weight(m); \
17 for_each_cpu(cpu, m) \
19 KUNIT_EXPECT_EQ((test), mask_weight, iter); \
22 #define EXPECT_FOR_EACH_CPU_NOT_EQ(test, mask) \
24 const cpumask_t *m = (mask); \
25 int mask_weight = cpumask_weight(m); \
27 for_each_cpu_not(cpu, m) \
29 KUNIT_EXPECT_EQ((test), nr_cpu_ids - mask_weight, iter); \
32 #define EXPECT_FOR_EACH_CPU_WRAP_EQ(test, mask) \
34 const cpumask_t *m = (mask); \
35 int mask_weight = cpumask_weight(m); \
37 for_each_cpu_wrap(cpu, m, nr_cpu_ids / 2) \
39 KUNIT_EXPECT_EQ((test), mask_weight, iter); \
42 #define EXPECT_FOR_EACH_CPU_BUILTIN_EQ(test, name) \
44 int mask_weight = num_##name##_cpus(); \
46 for_each_##name##_cpu(cpu) \
48 KUNIT_EXPECT_EQ((test), mask_weight, iter); \
51 static cpumask_t mask_empty;
52 static cpumask_t mask_all;
54 static void test_cpumask_weight(struct kunit *test)
56 KUNIT_EXPECT_TRUE(test, cpumask_empty(&mask_empty));
57 KUNIT_EXPECT_TRUE(test, cpumask_full(&mask_all));
59 KUNIT_EXPECT_EQ(test, 0, cpumask_weight(&mask_empty));
60 KUNIT_EXPECT_EQ(test, nr_cpu_ids, cpumask_weight(cpu_possible_mask));
61 KUNIT_EXPECT_EQ(test, nr_cpumask_bits, cpumask_weight(&mask_all));
64 static void test_cpumask_first(struct kunit *test)
66 KUNIT_EXPECT_LE(test, nr_cpu_ids, cpumask_first(&mask_empty));
67 KUNIT_EXPECT_EQ(test, 0, cpumask_first(cpu_possible_mask));
69 KUNIT_EXPECT_EQ(test, 0, cpumask_first_zero(&mask_empty));
70 KUNIT_EXPECT_LE(test, nr_cpu_ids, cpumask_first_zero(cpu_possible_mask));
73 static void test_cpumask_last(struct kunit *test)
75 KUNIT_EXPECT_LE(test, nr_cpumask_bits, cpumask_last(&mask_empty));
76 KUNIT_EXPECT_EQ(test, nr_cpu_ids - 1, cpumask_last(cpu_possible_mask));
79 static void test_cpumask_next(struct kunit *test)
81 KUNIT_EXPECT_EQ(test, 0, cpumask_next_zero(-1, &mask_empty));
82 KUNIT_EXPECT_LE(test, nr_cpu_ids, cpumask_next_zero(-1, cpu_possible_mask));
84 KUNIT_EXPECT_LE(test, nr_cpu_ids, cpumask_next(-1, &mask_empty));
85 KUNIT_EXPECT_EQ(test, 0, cpumask_next(-1, cpu_possible_mask));
88 static void test_cpumask_iterators(struct kunit *test)
90 EXPECT_FOR_EACH_CPU_EQ(test, &mask_empty);
91 EXPECT_FOR_EACH_CPU_NOT_EQ(test, &mask_empty);
92 EXPECT_FOR_EACH_CPU_WRAP_EQ(test, &mask_empty);
94 EXPECT_FOR_EACH_CPU_EQ(test, cpu_possible_mask);
95 EXPECT_FOR_EACH_CPU_NOT_EQ(test, cpu_possible_mask);
96 EXPECT_FOR_EACH_CPU_WRAP_EQ(test, cpu_possible_mask);
99 static void test_cpumask_iterators_builtin(struct kunit *test)
101 EXPECT_FOR_EACH_CPU_BUILTIN_EQ(test, possible);
103 /* Ensure the dynamic masks are stable while running the tests */
104 cpu_hotplug_disable();
106 EXPECT_FOR_EACH_CPU_BUILTIN_EQ(test, online);
107 EXPECT_FOR_EACH_CPU_BUILTIN_EQ(test, present);
109 cpu_hotplug_enable();
112 static int test_cpumask_init(struct kunit *test)
114 cpumask_clear(&mask_empty);
115 cpumask_setall(&mask_all);
120 static struct kunit_case test_cpumask_cases[] = {
121 KUNIT_CASE(test_cpumask_weight),
122 KUNIT_CASE(test_cpumask_first),
123 KUNIT_CASE(test_cpumask_last),
124 KUNIT_CASE(test_cpumask_next),
125 KUNIT_CASE(test_cpumask_iterators),
126 KUNIT_CASE(test_cpumask_iterators_builtin),
130 static struct kunit_suite test_cpumask_suite = {
132 .init = test_cpumask_init,
133 .test_cases = test_cpumask_cases,
135 kunit_test_suite(test_cpumask_suite);
137 MODULE_LICENSE("GPL");