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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
/* Copyright 2022 The ChromiumOS Authors
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#include "console.h"
#include "test/drivers/test_state.h"
#include "uart.h"
#include <zephyr/device.h>
#include <zephyr/drivers/uart.h>
#include <zephyr/shell/shell.h>
#include <zephyr/ztest.h>
void uart_callback(const struct device *dev, void *user_data);
void bypass_cb(const struct shell *shell, uint8_t *data, size_t len);
static const struct device *uart_shell_dev =
DEVICE_DT_GET(DT_CHOSEN(zephyr_shell_uart));
static void shell_before(void *f)
{
ARG_UNUSED(f);
uart_shell_start();
k_msleep(500);
uart_clear_input();
}
ZTEST_SUITE(shell, drivers_predicate_post_main, NULL, shell_before, NULL, NULL);
ZTEST(shell, test_shell_stop_read_raw_data)
{
const char uart_data[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZ";
uart_shell_stop();
k_msleep(500);
zassert_true(uart_irq_tx_ready(uart_shell_dev) != 0);
uart_clear_input();
/* Run the callback once, should be empty */
uart_callback(uart_shell_dev, NULL);
zassert_equal(-1, uart_getc());
uart_clear_input();
/* Manually fill the buffer */
for (size_t i = 0; i < CONFIG_UART_RX_BUF_SIZE; ++i) {
bypass_cb(get_ec_shell(),
(uint8_t *)&uart_data[i % ARRAY_SIZE(uart_data)], 1);
}
/* Push 1 extra character that should be dropped */
bypass_cb(get_ec_shell(), (uint8_t *)uart_data, 1);
/* Run the callback again to make sure we didn't lose any data */
uart_callback(uart_shell_dev, NULL);
for (size_t i = 0; i < CONFIG_UART_RX_BUF_SIZE; ++i) {
int c = uart_getc();
zassert_equal(
(int)uart_data[i % ARRAY_SIZE(uart_data)], c,
"Expected %uth character to be %c, but uart_getc() returned %c",
i, uart_data[i % ARRAY_SIZE(uart_data)], (char)c);
}
zassert_equal(-1, uart_getc());
}
ZTEST(shell, test_help_command)
{
/* Verify that the `help` subcommand works for a random command */
zassert_ok(shell_execute_cmd(get_ec_shell(), "accelinfo help"));
}
ZTEST(shell, test_rx_bypass)
{
const char uart_data = 'T';
bypass_cb(get_ec_shell(), (uint8_t *)&uart_data, 1);
/* Check that with the shell running and rx bypass disabled (default),
* we cannot pull values from the uart buffer directly.
*/
zassert_equal(-1, uart_getc());
}
|
