162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * KUnit test for the input core.
462306a36Sopenharmony_ci *
562306a36Sopenharmony_ci * Copyright (c) 2023 Red Hat Inc
662306a36Sopenharmony_ci */
762306a36Sopenharmony_ci
862306a36Sopenharmony_ci#include <linux/delay.h>
962306a36Sopenharmony_ci#include <linux/input.h>
1062306a36Sopenharmony_ci
1162306a36Sopenharmony_ci#include <kunit/test.h>
1262306a36Sopenharmony_ci
1362306a36Sopenharmony_ci#define POLL_INTERVAL 100
1462306a36Sopenharmony_ci
1562306a36Sopenharmony_cistatic int input_test_init(struct kunit *test)
1662306a36Sopenharmony_ci{
1762306a36Sopenharmony_ci	struct input_dev *input_dev;
1862306a36Sopenharmony_ci	int ret;
1962306a36Sopenharmony_ci
2062306a36Sopenharmony_ci	input_dev = input_allocate_device();
2162306a36Sopenharmony_ci	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, input_dev);
2262306a36Sopenharmony_ci
2362306a36Sopenharmony_ci	input_dev->name = "Test input device";
2462306a36Sopenharmony_ci	input_dev->id.bustype = BUS_VIRTUAL;
2562306a36Sopenharmony_ci	input_dev->id.vendor = 1;
2662306a36Sopenharmony_ci	input_dev->id.product = 1;
2762306a36Sopenharmony_ci	input_dev->id.version = 1;
2862306a36Sopenharmony_ci	input_set_capability(input_dev, EV_KEY, BTN_LEFT);
2962306a36Sopenharmony_ci	input_set_capability(input_dev, EV_KEY, BTN_RIGHT);
3062306a36Sopenharmony_ci
3162306a36Sopenharmony_ci	ret = input_register_device(input_dev);
3262306a36Sopenharmony_ci	if (ret) {
3362306a36Sopenharmony_ci		input_free_device(input_dev);
3462306a36Sopenharmony_ci		KUNIT_ASSERT_FAILURE(test, "Register device failed: %d", ret);
3562306a36Sopenharmony_ci	}
3662306a36Sopenharmony_ci
3762306a36Sopenharmony_ci	test->priv = input_dev;
3862306a36Sopenharmony_ci
3962306a36Sopenharmony_ci	return 0;
4062306a36Sopenharmony_ci}
4162306a36Sopenharmony_ci
4262306a36Sopenharmony_cistatic void input_test_exit(struct kunit *test)
4362306a36Sopenharmony_ci{
4462306a36Sopenharmony_ci	struct input_dev *input_dev = test->priv;
4562306a36Sopenharmony_ci
4662306a36Sopenharmony_ci	if (input_dev)
4762306a36Sopenharmony_ci		input_unregister_device(input_dev);
4862306a36Sopenharmony_ci}
4962306a36Sopenharmony_ci
5062306a36Sopenharmony_cistatic void input_test_poll(struct input_dev *input) { }
5162306a36Sopenharmony_ci
5262306a36Sopenharmony_cistatic void input_test_polling(struct kunit *test)
5362306a36Sopenharmony_ci{
5462306a36Sopenharmony_ci	struct input_dev *input_dev = test->priv;
5562306a36Sopenharmony_ci
5662306a36Sopenharmony_ci	/* Must fail because a poll handler has not been set-up yet */
5762306a36Sopenharmony_ci	KUNIT_ASSERT_EQ(test, input_get_poll_interval(input_dev), -EINVAL);
5862306a36Sopenharmony_ci
5962306a36Sopenharmony_ci	KUNIT_ASSERT_EQ(test, input_setup_polling(input_dev, input_test_poll), 0);
6062306a36Sopenharmony_ci
6162306a36Sopenharmony_ci	input_set_poll_interval(input_dev, POLL_INTERVAL);
6262306a36Sopenharmony_ci
6362306a36Sopenharmony_ci	/* Must succeed because poll handler was set-up and poll interval set */
6462306a36Sopenharmony_ci	KUNIT_ASSERT_EQ(test, input_get_poll_interval(input_dev), POLL_INTERVAL);
6562306a36Sopenharmony_ci}
6662306a36Sopenharmony_ci
6762306a36Sopenharmony_cistatic void input_test_timestamp(struct kunit *test)
6862306a36Sopenharmony_ci{
6962306a36Sopenharmony_ci	const ktime_t invalid_timestamp = ktime_set(0, 0);
7062306a36Sopenharmony_ci	struct input_dev *input_dev = test->priv;
7162306a36Sopenharmony_ci	ktime_t *timestamp, time;
7262306a36Sopenharmony_ci
7362306a36Sopenharmony_ci	timestamp = input_get_timestamp(input_dev);
7462306a36Sopenharmony_ci	time = timestamp[INPUT_CLK_MONO];
7562306a36Sopenharmony_ci
7662306a36Sopenharmony_ci	/* The returned timestamp must always be valid */
7762306a36Sopenharmony_ci	KUNIT_ASSERT_EQ(test, ktime_compare(time, invalid_timestamp), 1);
7862306a36Sopenharmony_ci
7962306a36Sopenharmony_ci	time = ktime_get();
8062306a36Sopenharmony_ci	input_set_timestamp(input_dev, time);
8162306a36Sopenharmony_ci
8262306a36Sopenharmony_ci	timestamp = input_get_timestamp(input_dev);
8362306a36Sopenharmony_ci	/* The timestamp must be the same than set before */
8462306a36Sopenharmony_ci	KUNIT_ASSERT_EQ(test, ktime_compare(timestamp[INPUT_CLK_MONO], time), 0);
8562306a36Sopenharmony_ci}
8662306a36Sopenharmony_ci
8762306a36Sopenharmony_cistatic void input_test_match_device_id(struct kunit *test)
8862306a36Sopenharmony_ci{
8962306a36Sopenharmony_ci	struct input_dev *input_dev = test->priv;
9062306a36Sopenharmony_ci	struct input_device_id id = { 0 };
9162306a36Sopenharmony_ci
9262306a36Sopenharmony_ci	/*
9362306a36Sopenharmony_ci	 * Must match when the input device bus, vendor, product, version
9462306a36Sopenharmony_ci	 * and events capable of handling are the same and fail to match
9562306a36Sopenharmony_ci	 * otherwise.
9662306a36Sopenharmony_ci	 */
9762306a36Sopenharmony_ci	id.flags = INPUT_DEVICE_ID_MATCH_BUS;
9862306a36Sopenharmony_ci	id.bustype = BUS_VIRTUAL;
9962306a36Sopenharmony_ci	KUNIT_ASSERT_TRUE(test, input_match_device_id(input_dev, &id));
10062306a36Sopenharmony_ci
10162306a36Sopenharmony_ci	id.bustype = BUS_I2C;
10262306a36Sopenharmony_ci	KUNIT_ASSERT_FALSE(test, input_match_device_id(input_dev, &id));
10362306a36Sopenharmony_ci
10462306a36Sopenharmony_ci	id.flags = INPUT_DEVICE_ID_MATCH_VENDOR;
10562306a36Sopenharmony_ci	id.vendor = 1;
10662306a36Sopenharmony_ci	KUNIT_ASSERT_TRUE(test, input_match_device_id(input_dev, &id));
10762306a36Sopenharmony_ci
10862306a36Sopenharmony_ci	id.vendor = 2;
10962306a36Sopenharmony_ci	KUNIT_ASSERT_FALSE(test, input_match_device_id(input_dev, &id));
11062306a36Sopenharmony_ci
11162306a36Sopenharmony_ci	id.flags = INPUT_DEVICE_ID_MATCH_PRODUCT;
11262306a36Sopenharmony_ci	id.product = 1;
11362306a36Sopenharmony_ci	KUNIT_ASSERT_TRUE(test, input_match_device_id(input_dev, &id));
11462306a36Sopenharmony_ci
11562306a36Sopenharmony_ci	id.product = 2;
11662306a36Sopenharmony_ci	KUNIT_ASSERT_FALSE(test, input_match_device_id(input_dev, &id));
11762306a36Sopenharmony_ci
11862306a36Sopenharmony_ci	id.flags = INPUT_DEVICE_ID_MATCH_VERSION;
11962306a36Sopenharmony_ci	id.version = 1;
12062306a36Sopenharmony_ci	KUNIT_ASSERT_TRUE(test, input_match_device_id(input_dev, &id));
12162306a36Sopenharmony_ci
12262306a36Sopenharmony_ci	id.version = 2;
12362306a36Sopenharmony_ci	KUNIT_ASSERT_FALSE(test, input_match_device_id(input_dev, &id));
12462306a36Sopenharmony_ci
12562306a36Sopenharmony_ci	id.flags = INPUT_DEVICE_ID_MATCH_EVBIT;
12662306a36Sopenharmony_ci	__set_bit(EV_KEY, id.evbit);
12762306a36Sopenharmony_ci	KUNIT_ASSERT_TRUE(test, input_match_device_id(input_dev, &id));
12862306a36Sopenharmony_ci
12962306a36Sopenharmony_ci	__set_bit(EV_ABS, id.evbit);
13062306a36Sopenharmony_ci	KUNIT_ASSERT_FALSE(test, input_match_device_id(input_dev, &id));
13162306a36Sopenharmony_ci}
13262306a36Sopenharmony_ci
13362306a36Sopenharmony_cistatic void input_test_grab(struct kunit *test)
13462306a36Sopenharmony_ci{
13562306a36Sopenharmony_ci	struct input_dev *input_dev = test->priv;
13662306a36Sopenharmony_ci	struct input_handle test_handle;
13762306a36Sopenharmony_ci	struct input_handler handler;
13862306a36Sopenharmony_ci	struct input_handle handle;
13962306a36Sopenharmony_ci	struct input_device_id id;
14062306a36Sopenharmony_ci	int res;
14162306a36Sopenharmony_ci
14262306a36Sopenharmony_ci	handler.name = "handler";
14362306a36Sopenharmony_ci	handler.id_table = &id;
14462306a36Sopenharmony_ci
14562306a36Sopenharmony_ci	handle.dev = input_get_device(input_dev);
14662306a36Sopenharmony_ci	handle.name = dev_name(&input_dev->dev);
14762306a36Sopenharmony_ci	handle.handler = &handler;
14862306a36Sopenharmony_ci	res = input_grab_device(&handle);
14962306a36Sopenharmony_ci	KUNIT_ASSERT_TRUE(test, res == 0);
15062306a36Sopenharmony_ci
15162306a36Sopenharmony_ci	test_handle.dev = input_get_device(input_dev);
15262306a36Sopenharmony_ci	test_handle.name = dev_name(&input_dev->dev);
15362306a36Sopenharmony_ci	test_handle.handler = &handler;
15462306a36Sopenharmony_ci	res = input_grab_device(&test_handle);
15562306a36Sopenharmony_ci	KUNIT_ASSERT_EQ(test, res, -EBUSY);
15662306a36Sopenharmony_ci
15762306a36Sopenharmony_ci	input_release_device(&handle);
15862306a36Sopenharmony_ci	input_put_device(input_dev);
15962306a36Sopenharmony_ci	res = input_grab_device(&test_handle);
16062306a36Sopenharmony_ci	KUNIT_ASSERT_TRUE(test, res == 0);
16162306a36Sopenharmony_ci	input_put_device(input_dev);
16262306a36Sopenharmony_ci}
16362306a36Sopenharmony_ci
16462306a36Sopenharmony_cistatic struct kunit_case input_tests[] = {
16562306a36Sopenharmony_ci	KUNIT_CASE(input_test_polling),
16662306a36Sopenharmony_ci	KUNIT_CASE(input_test_timestamp),
16762306a36Sopenharmony_ci	KUNIT_CASE(input_test_match_device_id),
16862306a36Sopenharmony_ci	KUNIT_CASE(input_test_grab),
16962306a36Sopenharmony_ci	{ /* sentinel */ }
17062306a36Sopenharmony_ci};
17162306a36Sopenharmony_ci
17262306a36Sopenharmony_cistatic struct kunit_suite input_test_suite = {
17362306a36Sopenharmony_ci	.name = "input_core",
17462306a36Sopenharmony_ci	.init = input_test_init,
17562306a36Sopenharmony_ci	.exit = input_test_exit,
17662306a36Sopenharmony_ci	.test_cases = input_tests,
17762306a36Sopenharmony_ci};
17862306a36Sopenharmony_ci
17962306a36Sopenharmony_cikunit_test_suite(input_test_suite);
18062306a36Sopenharmony_ci
18162306a36Sopenharmony_ciMODULE_AUTHOR("Javier Martinez Canillas <javierm@redhat.com>");
18262306a36Sopenharmony_ciMODULE_LICENSE("GPL");
183