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(cpu_possible_mask));
58 KUNIT_EXPECT_TRUE(test, cpumask_full(&mask_all));
60 KUNIT_EXPECT_EQ(test, 0, cpumask_weight(&mask_empty));
61 KUNIT_EXPECT_EQ(test, nr_cpu_ids, cpumask_weight(cpu_possible_mask));
62 KUNIT_EXPECT_EQ(test, nr_cpumask_bits, cpumask_weight(&mask_all));
65 static void test_cpumask_first(struct kunit *test)
67 KUNIT_EXPECT_LE(test, nr_cpu_ids, cpumask_first(&mask_empty));
68 KUNIT_EXPECT_EQ(test, 0, cpumask_first(cpu_possible_mask));
70 KUNIT_EXPECT_EQ(test, 0, cpumask_first_zero(&mask_empty));
71 KUNIT_EXPECT_LE(test, nr_cpu_ids, cpumask_first_zero(cpu_possible_mask));
74 static void test_cpumask_last(struct kunit *test)
76 KUNIT_EXPECT_LE(test, nr_cpumask_bits, cpumask_last(&mask_empty));
77 KUNIT_EXPECT_EQ(test, nr_cpumask_bits - 1, cpumask_last(cpu_possible_mask));
80 static void test_cpumask_next(struct kunit *test)
82 KUNIT_EXPECT_EQ(test, 0, cpumask_next_zero(-1, &mask_empty));
83 KUNIT_EXPECT_LE(test, nr_cpu_ids, cpumask_next_zero(-1, cpu_possible_mask));
85 KUNIT_EXPECT_LE(test, nr_cpu_ids, cpumask_next(-1, &mask_empty));
86 KUNIT_EXPECT_EQ(test, 0, cpumask_next(-1, cpu_possible_mask));
89 static void test_cpumask_iterators(struct kunit *test)
91 EXPECT_FOR_EACH_CPU_EQ(test, &mask_empty);
92 EXPECT_FOR_EACH_CPU_NOT_EQ(test, &mask_empty);
93 EXPECT_FOR_EACH_CPU_WRAP_EQ(test, &mask_empty);
95 EXPECT_FOR_EACH_CPU_EQ(test, cpu_possible_mask);
96 EXPECT_FOR_EACH_CPU_NOT_EQ(test, cpu_possible_mask);
97 EXPECT_FOR_EACH_CPU_WRAP_EQ(test, cpu_possible_mask);
100 static void test_cpumask_iterators_builtin(struct kunit *test)
102 EXPECT_FOR_EACH_CPU_BUILTIN_EQ(test, possible);
104 /* Ensure the dynamic masks are stable while running the tests */
105 cpu_hotplug_disable();
107 EXPECT_FOR_EACH_CPU_BUILTIN_EQ(test, online);
108 EXPECT_FOR_EACH_CPU_BUILTIN_EQ(test, present);
110 cpu_hotplug_enable();
113 static int test_cpumask_init(struct kunit *test)
115 cpumask_clear(&mask_empty);
116 cpumask_setall(&mask_all);
121 static struct kunit_case test_cpumask_cases[] = {
122 KUNIT_CASE(test_cpumask_weight),
123 KUNIT_CASE(test_cpumask_first),
124 KUNIT_CASE(test_cpumask_last),
125 KUNIT_CASE(test_cpumask_next),
126 KUNIT_CASE(test_cpumask_iterators),
127 KUNIT_CASE(test_cpumask_iterators_builtin),
131 static struct kunit_suite test_cpumask_suite = {
133 .init = test_cpumask_init,
134 .test_cases = test_cpumask_cases,
136 kunit_test_suite(test_cpumask_suite);
138 MODULE_LICENSE("GPL");