bcachefs: add eytzinger0_find_gt self test

Add an eytzinger0_find_gt() self test similar to eytzinger0_find_le().

Signed-off-by: Andreas Gruenbacher <agruenba@redhat.com>
Signed-off-by: Kent Overstreet <kent.overstreet@linux.dev>

diff --git a/fs/bcachefs/util.c b/fs/bcachefs/util.c
index d2f7ffcc4fd6ead96500b1c8b39ed3a003f29fdc..9c6e5d7122b4a85ed1d74f4ff164804fba589315 100644 (file)
--- a/fs/bcachefs/util.c
+++ b/fs/bcachefs/util.c
@@ -825,9 +825,47 @@ static void eytzinger0_find_test_le(u16 *test_array, unsigned nr, u16 search)
        }
 }
 
+static void eytzinger0_find_test_gt(u16 *test_array, unsigned nr, u16 search)
+{
+       int r, s;
+       bool bad;
+
+       r = eytzinger0_find_gt(test_array, nr,
+                              sizeof(test_array[0]),
+                              cmp_u16, &search);
+       if (r >= 0) {
+               if (test_array[r] <= search) {
+                       bad = true;
+               } else {
+                       s = eytzinger0_prev(r, nr);
+                       bad = s >= 0 && test_array[s] > search;
+               }
+       } else {
+               s = eytzinger0_first(nr);
+               bad = s >= 0 && test_array[s] > search;
+       }
+
+       if (bad) {
+               s = -1;
+               eytzinger0_for_each(j, nr) {
+                       if (test_array[j] > search) {
+                               s = j;
+                               break;
+                       }
+               }
+
+               eytzinger0_for_each(j, nr)
+                       pr_info("[%3u] = %12u\n", j, test_array[j]);
+               pr_info("find_gt(%12u) = %3i should be %3i\n",
+                       search, r, s);
+               BUG();
+       }
+}
+
 static void eytzinger0_find_test_val(u16 *test_array, unsigned nr, u16 search)
 {
        eytzinger0_find_test_le(test_array, nr, search);
+       eytzinger0_find_test_gt(test_array, nr, search);
 }
 
 void eytzinger0_find_test(void)