blob: c1bdfe9a4bff62f38cfceb7349cc1e5b49c20870 [file] [log] [blame]
// SPDX-License-Identifier: GPL-2.0
/*
* KUnit test for core test infrastructure.
*
* Copyright (C) 2018, Google LLC.
* Author: Brendan Higgins <brendanhiggins@google.com>
*/
#include <kunit/test.h>
typedef void (*test_resource_free_t)(struct KUNIT_RESOURCE_T *);
/*
* Context for testing test managed resources
* is_resource_initialized is used to test arbitrary resources
*/
struct test_test_context {
struct KUNIT_T test;
bool is_resource_initialized;
};
static int fake_resource_init(struct KUNIT_RESOURCE_T *res, void *context)
{
struct test_test_context *ctx = context;
res->allocation = &ctx->is_resource_initialized;
ctx->is_resource_initialized = true;
return 0;
}
static void fake_resource_free(struct KUNIT_RESOURCE_T *res)
{
bool *is_resource_initialized = res->allocation;
*is_resource_initialized = false;
}
static void test_test_init_resources(struct KUNIT_T *test)
{
struct test_test_context *ctx = test->priv;
test_init_test(&ctx->test, "testing_test_init_test");
EXPECT_TRUE(test, list_empty(&ctx->test.resources));
}
static void test_test_alloc_resource(struct KUNIT_T *test)
{
struct test_test_context *ctx = test->priv;
struct KUNIT_RESOURCE_T *res;
res = test_alloc_resource(&ctx->test,
fake_resource_init,
fake_resource_free,
ctx);
ASSERT_NOT_ERR_OR_NULL(test, res);
EXPECT_EQ(test, &ctx->is_resource_initialized, res->allocation);
EXPECT_TRUE(test, list_is_last(&res->node, &ctx->test.resources));
EXPECT_EQ(test,
(test_resource_free_t)fake_resource_free,
(test_resource_free_t)res->free);
}
static void test_test_free_resource(struct KUNIT_T *test)
{
struct test_test_context *ctx = test->priv;
struct KUNIT_RESOURCE_T *res = test_alloc_resource(&ctx->test,
fake_resource_init,
fake_resource_free,
ctx);
test_free_resource(&ctx->test, res);
EXPECT_EQ(test, false, ctx->is_resource_initialized);
EXPECT_TRUE(test, list_empty(&ctx->test.resources));
}
static void test_test_cleanup_resources(struct KUNIT_T *test)
{
int i;
const int num_res = 5;
struct test_test_context *ctx = test->priv;
struct KUNIT_RESOURCE_T *resources[num_res];
for (i = 0; i < num_res; i++) {
resources[i] = test_alloc_resource(&ctx->test,
fake_resource_init,
fake_resource_free,
ctx);
}
test_cleanup(&ctx->test);
EXPECT_TRUE(test, list_empty(&ctx->test.resources));
}
static int test_test_init(struct KUNIT_T *test)
{
struct test_test_context *ctx = kzalloc(sizeof(*ctx), GFP_KERNEL);
if (!ctx)
return -ENOMEM;
test->priv = ctx;
test_init_test(&ctx->test, "test_test_context");
return 0;
}
static void test_test_exit(struct KUNIT_T *test)
{
struct test_test_context *ctx = test->priv;
test_cleanup(&ctx->test);
kfree(ctx);
}
static struct KUNIT_CASE_T test_test_cases[] = {
TEST_CASE(test_test_init_resources),
TEST_CASE(test_test_alloc_resource),
TEST_CASE(test_test_free_resource),
TEST_CASE(test_test_cleanup_resources),
{},
};
static struct KUNIT_SUITE_T test_test_module = {
.name = "test-test",
.init = test_test_init,
.exit = test_test_exit,
.test_cases = test_test_cases,
};
kunit_test_suites(&test_test_module);