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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
// SPDX-License-Identifier: GPL-2.0+
/*
* Copyright 2023 Google LLC
*/
#include <common.h>
#include <cli.h>
#include <test/common.h>
#include <test/test.h>
#include <test/ut.h>
static int cli_ch_test(struct unit_test_state *uts)
{
struct cli_ch_state s_cch, *cch = &s_cch;
cli_ch_init(cch);
/* should be nothing to return at first */
ut_asserteq(0, cli_ch_process(cch, 0));
/* check normal entry */
ut_asserteq('a', cli_ch_process(cch, 'a'));
ut_asserteq('b', cli_ch_process(cch, 'b'));
ut_asserteq('c', cli_ch_process(cch, 'c'));
ut_asserteq(0, cli_ch_process(cch, 0));
/* send an invalid escape sequence */
ut_asserteq(0, cli_ch_process(cch, '\e'));
ut_asserteq(0, cli_ch_process(cch, '['));
/*
* with the next char it sees that the sequence is invalid, so starts
* emitting it
*/
ut_asserteq('\e', cli_ch_process(cch, 'X'));
/* now we set 0 bytes to empty the buffer */
ut_asserteq('[', cli_ch_process(cch, 0));
ut_asserteq('X', cli_ch_process(cch, 0));
ut_asserteq(0, cli_ch_process(cch, 0));
/* things are normal again */
ut_asserteq('a', cli_ch_process(cch, 'a'));
ut_asserteq(0, cli_ch_process(cch, 0));
/* unexpected 'Esc' */
ut_asserteq('a', cli_ch_process(cch, 'a'));
ut_asserteq(0, cli_ch_process(cch, '\e'));
ut_asserteq('b', cli_ch_process(cch, 'b'));
ut_asserteq(0, cli_ch_process(cch, 0));
return 0;
}
COMMON_TEST(cli_ch_test, 0);
static int cread_test(struct unit_test_state *uts)
{
int duration;
ulong start;
char buf[10];
/*
* useful for debugging
*
* gd->flags &= ~GD_FLG_RECORD;
* print_buffer(0, buf, 1, 7, 0);
*/
console_record_reset_enable();
/* simple input */
*buf = '\0';
ut_asserteq(4, console_in_puts("abc\n"));
ut_asserteq(3, cli_readline_into_buffer("-> ", buf, 1));
ut_asserteq_str("abc", buf);
/* try an escape sequence (cursor left after the 'c') */
*buf = '\0';
ut_asserteq(8, console_in_puts("abc\e[Dx\n"));
ut_asserteq(4, cli_readline_into_buffer("-> ", buf, 1));
ut_asserteq_str("abxc", buf);
/* invalid escape sequence */
*buf = '\0';
ut_asserteq(8, console_in_puts("abc\e[Xx\n"));
ut_asserteq(7, cli_readline_into_buffer("-> ", buf, 1));
ut_asserteq_str("abc\e[Xx", buf);
/* unexpected 'Esc' */
*buf = '\0';
ut_asserteq(7, console_in_puts("abc\eXx\n"));
ut_asserteq(5, cli_readline_into_buffer("-> ", buf, 1));
ut_asserteq_str("abcXx", buf);
/* check timeout, should be between 1000 and 1050ms */
start = get_timer(0);
*buf = '\0';
ut_asserteq(-2, cli_readline_into_buffer("-> ", buf, 1));
duration = get_timer(start) - 1000;
ut_assert(duration >= 0);
ut_assert(duration < 50);
return 0;
}
COMMON_TEST(cread_test, 0);
|
