Linux Audio

Check our new training course

Open Menu / tests / subsys / shell / shell_device_filter / src / main.c
Loading...
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
/*
 * Copyright 2024 Google LLC
 *
 * SPDX-License-Identifier: Apache-2.0
 */

#include <string.h>
#include <zephyr/device.h>
#include <zephyr/kernel.h>
#include <zephyr/shell/shell.h>
#include <zephyr/ztest.h>

DEVICE_DEFINE(device_0, "device@0", NULL, NULL,
	      NULL, NULL,
	      POST_KERNEL, 0, NULL);

DEVICE_DEFINE(device_1, "device@1", NULL, NULL,
	      NULL, NULL,
	      POST_KERNEL, 1, NULL);

DEVICE_DEFINE(device_2, "xx_device@2", NULL, NULL,
	      NULL, NULL,
	      POST_KERNEL, 2, NULL);

static const struct device *d0 = &DEVICE_NAME_GET(device_0);
static const struct device *d1 = &DEVICE_NAME_GET(device_1);
static const struct device *d2 = &DEVICE_NAME_GET(device_2);

ZTEST(shell_device_filter, test_unfiltered)
{
	zassert_equal_ptr(d0, shell_device_filter(0, NULL));
	zassert_equal_ptr(d1, shell_device_filter(1, NULL));
	zassert_equal_ptr(d2, shell_device_filter(2, NULL));
	zassert_equal_ptr(NULL, shell_device_filter(3, NULL));

	zassert_equal_ptr(d0, shell_device_lookup(0, NULL));
	zassert_equal_ptr(d1, shell_device_lookup(1, NULL));
	zassert_equal_ptr(d2, shell_device_lookup(2, NULL));
	zassert_equal_ptr(NULL, shell_device_lookup(3, NULL));
}

ZTEST(shell_device_filter, test_prefix)
{
	zassert_equal_ptr(d2, shell_device_lookup(0, "xx_"));
	zassert_equal_ptr(NULL, shell_device_lookup(1, "xx_"));
}

static bool pm_device_test_filter(const struct device *dev)
{
	return strstr(dev->name, "@1") != NULL;
}

ZTEST(shell_device_filter, test_filter)
{
	zassert_equal_ptr(d1, shell_device_filter(0, pm_device_test_filter));
	zassert_equal_ptr(NULL, shell_device_filter(1, pm_device_test_filter));
}

ZTEST_SUITE(shell_device_filter, NULL, NULL, NULL, NULL, NULL);