1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * Tests for echo command
4  *
5  * Copyright 2020, Heinrich Schuchadt <xypron.glpk@gmx.de>
6  */
7 
8 #include <common.h>
9 #include <command.h>
10 #include <asm/global_data.h>
11 #include <display_options.h>
12 #include <test/lib.h>
13 #include <test/test.h>
14 #include <test/ut.h>
15 
16 DECLARE_GLOBAL_DATA_PTR;
17 
18 struct test_data {
19 	char *cmd;
20 	char *expected;
21 };
22 
23 static struct test_data echo_data[] = {
24 	{"echo 1 2 3",
25 	 "1 2 3"},
26 	/* Test new line handling */
27 	{"echo -n 1 2 3; echo a b c",
28 	 "1 2 3a b c"},
29 	/*
30 	 * Test handling of environment variables.
31 	 *
32 	 * j, q, x are among the least frequent letters in English.
33 	 * Hence no collision for the variable name jQx is expected.
34 	 */
35 	{"setenv jQx X; echo \"a)\" ${jQx} 'b)' '${jQx}' c) ${jQx}; setenv jQx",
36 	 "a) X b) ${jQx} c) X"},
37 	/* Test handling of shell variables. */
38 	{"setenv jQx; for jQx in 1 2 3; do echo -n \"${jQx}, \"; done; echo;",
39 	 "1, 2, 3, "},
40 };
41 
lib_test_hush_echo(struct unit_test_state * uts)42 static int lib_test_hush_echo(struct unit_test_state *uts)
43 {
44 	int i;
45 
46 	for (i = 0; i < ARRAY_SIZE(echo_data); ++i) {
47 		console_record_reset_enable();
48 		ut_assertok(run_command(echo_data[i].cmd, 0));
49 		gd->flags &= ~GD_FLG_RECORD;
50 		console_record_readline(uts->actual_str,
51 					sizeof(uts->actual_str));
52 		ut_asserteq_str(echo_data[i].expected, uts->actual_str);
53 		ut_assertok(ut_check_console_end(uts));
54 	}
55 	return 0;
56 }
57 
58 LIB_TEST(lib_test_hush_echo, 0);
59