damon_test_set_regions_for() is designed to test only success-expected
damon_set_regions() calls.  Extend it to cover error-expected calls,
too.

Signed-off-by: SJ Park <[email protected]>
---
 mm/damon/tests/core-kunit.h | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

diff --git a/mm/damon/tests/core-kunit.h b/mm/damon/tests/core-kunit.h
index 4a536d41cdb2d..befc1196d6d60 100644
--- a/mm/damon/tests/core-kunit.h
+++ b/mm/damon/tests/core-kunit.h
@@ -463,11 +463,12 @@ static void damon_test_set_regions_for(struct kunit *test,
                struct damon_addr_range *old_ranges, int sz_old_ranges,
                struct damon_addr_range *new_ranges, int sz_new_ranges,
                unsigned long min_region_sz,
-               struct damon_addr_range *expect_ranges, int sz_expect_ranges)
+               struct damon_addr_range *expect_ranges, int sz_expect_ranges,
+               int expect_err)
 {
        struct damon_target *t;
        struct damon_region *r;
-       int i;
+       int i, err;
 
        t = damon_new_target();
        if (!t)
@@ -481,7 +482,8 @@ static void damon_test_set_regions_for(struct kunit *test,
                damon_add_region(r, t);
        }
 
-       damon_set_regions(t, new_ranges, sz_new_ranges, min_region_sz);
+       err = damon_set_regions(t, new_ranges, sz_new_ranges, min_region_sz);
+       KUNIT_EXPECT_EQ(test, err, expect_err);
 
        KUNIT_EXPECT_EQ(test, damon_nr_regions(t), sz_expect_ranges);
        if (damon_nr_regions(t) != sz_expect_ranges) {
@@ -510,7 +512,7 @@ static void damon_test_set_regions(struct kunit *test)
                        (struct damon_addr_range[]){
                        {.start = 5, .end = 15},
                        {.start = 15, .end = 25},
-                       }, 2);
+                       }, 2, 0);
        /* Un-intersecting regions should be removed. */
        damon_test_set_regions_for(test,
                        (struct damon_addr_range[]){
@@ -523,7 +525,7 @@ static void damon_test_set_regions(struct kunit *test)
                        1,
                        (struct damon_addr_range[]){
                        {.start = 18, .end = 23},
-                       }, 1);
+                       }, 1, 0);
        /*
         * Holes should be filled up with new regions.
         *
@@ -544,7 +546,7 @@ static void damon_test_set_regions(struct kunit *test)
                        {.start = 8, .end = 16},
                        {.start = 16, .end = 24},
                        {.start = 24, .end = 28},
-                       }, 3);
+                       }, 3, 0);
        /*
         * New regions should be able to be appended.
         *
@@ -566,7 +568,7 @@ static void damon_test_set_regions(struct kunit *test)
                        {.start = 0, .end = 4},
                        {.start = 4, .end = 15},
                        {.start = 25, .end = 40},
-                       }, 3);
+                       }, 3, 0);
        /*
         * New regions should be able to be inserted.
         *
@@ -589,7 +591,7 @@ static void damon_test_set_regions(struct kunit *test)
                        {.start = 0, .end = 15},
                        {.start = 25, .end = 40},
                        {.start = 44, .end = 50},
-                       }, 3);
+                       }, 3, 0);
 }
 
 static void damon_test_update_monitoring_result(struct kunit *test)
-- 
2.47.3

Reply via email to