// SPDX-License-Identifier: GPL-2.0-only

#include <kunit/test.h>

#include <linux/sort.h>
#include <linux/slab.h>
#include <linux/module.h>

/* a simple boot-time regression test */

#define TEST_LEN 1000

static int cmpint(const void *a, const void *b)
{
 return *(int *)a - *(int *)b;
}

static void test_sort(struct kunit *test)
{
 int *a, i, r = 1;

 a = kunit_kmalloc_array(test, TEST_LEN, sizeof(*a), GFP_KERNEL);
 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, a);

 for (i = 0; i < TEST_LEN; i++) {
  r = (r * 725861) % 6599;
  a[i] = r;
 }

 sort(a, TEST_LEN, sizeof(*a), cmpint, NULL);

 for (i = 0; i < TEST_LEN - 1; i++)
  KUNIT_ASSERT_LE(test, a[i], a[i + 1]);

 r = 48;

 for (i = 0; i < TEST_LEN - 1; i++) {
  r = (r * 725861) % 6599;
  a[i] = r;
 }

 sort(a, TEST_LEN - 1, sizeof(*a), cmpint, NULL);

 for (i = 0; i < TEST_LEN - 2; i++)
  KUNIT_ASSERT_LE(test, a[i], a[i + 1]);
}

static struct kunit_case sort_test_cases[] = {
 KUNIT_CASE(test_sort),
 {}
};

static struct kunit_suite sort_test_suite = {
 .name = "lib_sort",
 .test_cases = sort_test_cases,
};

kunit_test_suites(&sort_test_suite);

MODULE_DESCRIPTION("sort() KUnit test suite");
MODULE_LICENSE("GPL");