Loading...
1// SPDX-License-Identifier: GPL-2.0
2/*
3 * KUnit test for the FPGA Bridge
4 *
5 * Copyright (C) 2023 Red Hat, Inc.
6 *
7 * Author: Marco Pagani <marpagan@redhat.com>
8 */
9
10#include <kunit/device.h>
11#include <kunit/test.h>
12#include <linux/fpga/fpga-bridge.h>
13#include <linux/module.h>
14#include <linux/types.h>
15
16struct bridge_stats {
17 bool enable;
18};
19
20struct bridge_ctx {
21 struct fpga_bridge *bridge;
22 struct device *dev;
23 struct bridge_stats stats;
24};
25
26/*
27 * Wrapper to avoid a cast warning when passing the action function directly
28 * to kunit_add_action().
29 */
30KUNIT_DEFINE_ACTION_WRAPPER(fpga_bridge_unregister_wrapper, fpga_bridge_unregister,
31 struct fpga_bridge *);
32
33static int op_enable_set(struct fpga_bridge *bridge, bool enable)
34{
35 struct bridge_stats *stats = bridge->priv;
36
37 stats->enable = enable;
38
39 return 0;
40}
41
42/*
43 * Fake FPGA bridge that implements only the enable_set op to track
44 * the state.
45 */
46static const struct fpga_bridge_ops fake_bridge_ops = {
47 .enable_set = op_enable_set,
48};
49
50/**
51 * register_test_bridge() - Register a fake FPGA bridge for testing.
52 * @test: KUnit test context object.
53 * @dev_name: name of the kunit device to be registered
54 *
55 * Return: Context of the newly registered FPGA bridge.
56 */
57static struct bridge_ctx *register_test_bridge(struct kunit *test, const char *dev_name)
58{
59 struct bridge_ctx *ctx;
60 int ret;
61
62 ctx = kunit_kzalloc(test, sizeof(*ctx), GFP_KERNEL);
63 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, ctx);
64
65 ctx->dev = kunit_device_register(test, dev_name);
66 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, ctx->dev);
67
68 ctx->bridge = fpga_bridge_register(ctx->dev, "Fake FPGA bridge", &fake_bridge_ops,
69 &ctx->stats);
70 KUNIT_ASSERT_FALSE(test, IS_ERR_OR_NULL(ctx->bridge));
71
72 ret = kunit_add_action_or_reset(test, fpga_bridge_unregister_wrapper, ctx->bridge);
73 KUNIT_ASSERT_EQ(test, ret, 0);
74
75 return ctx;
76}
77
78static void fpga_bridge_test_get(struct kunit *test)
79{
80 struct bridge_ctx *ctx = test->priv;
81 struct fpga_bridge *bridge;
82
83 bridge = fpga_bridge_get(ctx->dev, NULL);
84 KUNIT_EXPECT_PTR_EQ(test, bridge, ctx->bridge);
85
86 bridge = fpga_bridge_get(ctx->dev, NULL);
87 KUNIT_EXPECT_EQ(test, PTR_ERR(bridge), -EBUSY);
88
89 fpga_bridge_put(ctx->bridge);
90}
91
92static void fpga_bridge_test_toggle(struct kunit *test)
93{
94 struct bridge_ctx *ctx = test->priv;
95 int ret;
96
97 ret = fpga_bridge_disable(ctx->bridge);
98 KUNIT_EXPECT_EQ(test, ret, 0);
99 KUNIT_EXPECT_FALSE(test, ctx->stats.enable);
100
101 ret = fpga_bridge_enable(ctx->bridge);
102 KUNIT_EXPECT_EQ(test, ret, 0);
103 KUNIT_EXPECT_TRUE(test, ctx->stats.enable);
104}
105
106/* Test the functions for getting and controlling a list of bridges */
107static void fpga_bridge_test_get_put_list(struct kunit *test)
108{
109 struct list_head bridge_list;
110 struct bridge_ctx *ctx_0, *ctx_1;
111 int ret;
112
113 ctx_0 = test->priv;
114 ctx_1 = register_test_bridge(test, "fpga-bridge-test-dev-1");
115
116 INIT_LIST_HEAD(&bridge_list);
117
118 /* Get bridge 0 and add it to the list */
119 ret = fpga_bridge_get_to_list(ctx_0->dev, NULL, &bridge_list);
120 KUNIT_EXPECT_EQ(test, ret, 0);
121
122 KUNIT_EXPECT_PTR_EQ(test, ctx_0->bridge,
123 list_first_entry_or_null(&bridge_list, struct fpga_bridge, node));
124
125 /* Get bridge 1 and add it to the list */
126 ret = fpga_bridge_get_to_list(ctx_1->dev, NULL, &bridge_list);
127 KUNIT_EXPECT_EQ(test, ret, 0);
128
129 KUNIT_EXPECT_PTR_EQ(test, ctx_1->bridge,
130 list_first_entry_or_null(&bridge_list, struct fpga_bridge, node));
131
132 /* Disable an then enable both bridges from the list */
133 ret = fpga_bridges_disable(&bridge_list);
134 KUNIT_EXPECT_EQ(test, ret, 0);
135
136 KUNIT_EXPECT_FALSE(test, ctx_0->stats.enable);
137 KUNIT_EXPECT_FALSE(test, ctx_1->stats.enable);
138
139 ret = fpga_bridges_enable(&bridge_list);
140 KUNIT_EXPECT_EQ(test, ret, 0);
141
142 KUNIT_EXPECT_TRUE(test, ctx_0->stats.enable);
143 KUNIT_EXPECT_TRUE(test, ctx_1->stats.enable);
144
145 /* Put and remove both bridges from the list */
146 fpga_bridges_put(&bridge_list);
147
148 KUNIT_EXPECT_TRUE(test, list_empty(&bridge_list));
149}
150
151static int fpga_bridge_test_init(struct kunit *test)
152{
153 test->priv = register_test_bridge(test, "fpga-bridge-test-dev-0");
154
155 return 0;
156}
157
158static struct kunit_case fpga_bridge_test_cases[] = {
159 KUNIT_CASE(fpga_bridge_test_get),
160 KUNIT_CASE(fpga_bridge_test_toggle),
161 KUNIT_CASE(fpga_bridge_test_get_put_list),
162 {}
163};
164
165static struct kunit_suite fpga_bridge_suite = {
166 .name = "fpga_bridge",
167 .init = fpga_bridge_test_init,
168 .test_cases = fpga_bridge_test_cases,
169};
170
171kunit_test_suite(fpga_bridge_suite);
172
173MODULE_LICENSE("GPL");
1// SPDX-License-Identifier: GPL-2.0
2/*
3 * KUnit test for the FPGA Bridge
4 *
5 * Copyright (C) 2023 Red Hat, Inc.
6 *
7 * Author: Marco Pagani <marpagan@redhat.com>
8 */
9
10#include <kunit/test.h>
11#include <linux/device.h>
12#include <linux/fpga/fpga-bridge.h>
13#include <linux/module.h>
14#include <linux/types.h>
15
16struct bridge_stats {
17 bool enable;
18};
19
20struct bridge_ctx {
21 struct fpga_bridge *bridge;
22 struct platform_device *pdev;
23 struct bridge_stats stats;
24};
25
26static int op_enable_set(struct fpga_bridge *bridge, bool enable)
27{
28 struct bridge_stats *stats = bridge->priv;
29
30 stats->enable = enable;
31
32 return 0;
33}
34
35/*
36 * Fake FPGA bridge that implements only the enable_set op to track
37 * the state.
38 */
39static const struct fpga_bridge_ops fake_bridge_ops = {
40 .enable_set = op_enable_set,
41};
42
43/**
44 * register_test_bridge() - Register a fake FPGA bridge for testing.
45 * @test: KUnit test context object.
46 *
47 * Return: Context of the newly registered FPGA bridge.
48 */
49static struct bridge_ctx *register_test_bridge(struct kunit *test)
50{
51 struct bridge_ctx *ctx;
52
53 ctx = kunit_kzalloc(test, sizeof(*ctx), GFP_KERNEL);
54 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, ctx);
55
56 ctx->pdev = platform_device_register_simple("bridge_pdev", PLATFORM_DEVID_AUTO, NULL, 0);
57 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, ctx->pdev);
58
59 ctx->bridge = fpga_bridge_register(&ctx->pdev->dev, "Fake FPGA bridge", &fake_bridge_ops,
60 &ctx->stats);
61 KUNIT_ASSERT_FALSE(test, IS_ERR_OR_NULL(ctx->bridge));
62
63 return ctx;
64}
65
66static void unregister_test_bridge(struct bridge_ctx *ctx)
67{
68 fpga_bridge_unregister(ctx->bridge);
69 platform_device_unregister(ctx->pdev);
70}
71
72static void fpga_bridge_test_get(struct kunit *test)
73{
74 struct bridge_ctx *ctx = test->priv;
75 struct fpga_bridge *bridge;
76
77 bridge = fpga_bridge_get(&ctx->pdev->dev, NULL);
78 KUNIT_EXPECT_PTR_EQ(test, bridge, ctx->bridge);
79
80 bridge = fpga_bridge_get(&ctx->pdev->dev, NULL);
81 KUNIT_EXPECT_EQ(test, PTR_ERR(bridge), -EBUSY);
82
83 fpga_bridge_put(ctx->bridge);
84}
85
86static void fpga_bridge_test_toggle(struct kunit *test)
87{
88 struct bridge_ctx *ctx = test->priv;
89 int ret;
90
91 ret = fpga_bridge_disable(ctx->bridge);
92 KUNIT_EXPECT_EQ(test, ret, 0);
93 KUNIT_EXPECT_FALSE(test, ctx->stats.enable);
94
95 ret = fpga_bridge_enable(ctx->bridge);
96 KUNIT_EXPECT_EQ(test, ret, 0);
97 KUNIT_EXPECT_TRUE(test, ctx->stats.enable);
98}
99
100/* Test the functions for getting and controlling a list of bridges */
101static void fpga_bridge_test_get_put_list(struct kunit *test)
102{
103 struct list_head bridge_list;
104 struct bridge_ctx *ctx_0, *ctx_1;
105 int ret;
106
107 ctx_0 = test->priv;
108 ctx_1 = register_test_bridge(test);
109
110 INIT_LIST_HEAD(&bridge_list);
111
112 /* Get bridge 0 and add it to the list */
113 ret = fpga_bridge_get_to_list(&ctx_0->pdev->dev, NULL, &bridge_list);
114 KUNIT_EXPECT_EQ(test, ret, 0);
115
116 KUNIT_EXPECT_PTR_EQ(test, ctx_0->bridge,
117 list_first_entry_or_null(&bridge_list, struct fpga_bridge, node));
118
119 /* Get bridge 1 and add it to the list */
120 ret = fpga_bridge_get_to_list(&ctx_1->pdev->dev, NULL, &bridge_list);
121 KUNIT_EXPECT_EQ(test, ret, 0);
122
123 KUNIT_EXPECT_PTR_EQ(test, ctx_1->bridge,
124 list_first_entry_or_null(&bridge_list, struct fpga_bridge, node));
125
126 /* Disable an then enable both bridges from the list */
127 ret = fpga_bridges_disable(&bridge_list);
128 KUNIT_EXPECT_EQ(test, ret, 0);
129
130 KUNIT_EXPECT_FALSE(test, ctx_0->stats.enable);
131 KUNIT_EXPECT_FALSE(test, ctx_1->stats.enable);
132
133 ret = fpga_bridges_enable(&bridge_list);
134 KUNIT_EXPECT_EQ(test, ret, 0);
135
136 KUNIT_EXPECT_TRUE(test, ctx_0->stats.enable);
137 KUNIT_EXPECT_TRUE(test, ctx_1->stats.enable);
138
139 /* Put and remove both bridges from the list */
140 fpga_bridges_put(&bridge_list);
141
142 KUNIT_EXPECT_TRUE(test, list_empty(&bridge_list));
143
144 unregister_test_bridge(ctx_1);
145}
146
147static int fpga_bridge_test_init(struct kunit *test)
148{
149 test->priv = register_test_bridge(test);
150
151 return 0;
152}
153
154static void fpga_bridge_test_exit(struct kunit *test)
155{
156 unregister_test_bridge(test->priv);
157}
158
159static struct kunit_case fpga_bridge_test_cases[] = {
160 KUNIT_CASE(fpga_bridge_test_get),
161 KUNIT_CASE(fpga_bridge_test_toggle),
162 KUNIT_CASE(fpga_bridge_test_get_put_list),
163 {}
164};
165
166static struct kunit_suite fpga_bridge_suite = {
167 .name = "fpga_bridge",
168 .init = fpga_bridge_test_init,
169 .exit = fpga_bridge_test_exit,
170 .test_cases = fpga_bridge_test_cases,
171};
172
173kunit_test_suite(fpga_bridge_suite);
174
175MODULE_LICENSE("GPL");