mm/damon/tests/sysfs-kunit.h

   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(i);
  33                 if (pid) {
  34                         put_pid(pid);
  35                         return i;
  36                 }
  37         }
  38         return -1;
  39 }
  40
  41 static void damon_sysfs_test_add_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(1,
  50                         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(12, 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_add_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                         sysfs_target->pid + 1, 200);
  64         damon_sysfs_add_targets(ctx, sysfs_targets);
  65         KUNIT_EXPECT_EQ(test, 2u, nr_damon_targets(ctx));
  66
  67         damon_destroy_ctx(ctx);
  68         kfree(sysfs_targets->targets_arr);
  69         kfree(sysfs_targets);
  70         kfree(sysfs_target->regions);
  71         kfree(sysfs_target);
  72 }
  73
  74 static struct kunit_case damon_sysfs_test_cases[] = {
  75         KUNIT_CASE(damon_sysfs_test_add_targets),
  76         {},
  77 };
  78
  79 static struct kunit_suite damon_sysfs_test_suite = {
  80         .name = "damon-sysfs",
  81         .test_cases = damon_sysfs_test_cases,
  82 };
  83 kunit_test_suite(damon_sysfs_test_suite);
  84
  85 #endif /* _DAMON_SYSFS_TEST_H */
  86
  87 #endif /* CONFIG_DAMON_SYSFS_KUNIT_TEST */