Open Menu / test / common / cread.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
 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
// SPDX-License-Identifier: GPL-2.0+
/*
 * Copyright 2023 Google LLC
 */

#include <cli.h>
#include <time.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);
	 */

	/* 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, UTF_CONSOLE);