blob: ca34abef8995422df9aa0eb87e1ee880c0eaebb5 [file] [log] [blame]
Marek Vasutf08dcd92022-12-20 07:26:00 +01001// SPDX-License-Identifier: GPL-2.0+
2/*
3 * Tests for exit command
4 *
5 * Copyright 2022 Marek Vasut <marex@denx.de>
6 */
7
8#include <common.h>
9#include <console.h>
10#include <mapmem.h>
11#include <asm/global_data.h>
12#include <test/suites.h>
13#include <test/ut.h>
14
15DECLARE_GLOBAL_DATA_PTR;
16
17/* Declare a new exit test */
18#define EXIT_TEST(_name, _flags) UNIT_TEST(_name, _flags, exit_test)
19
20/* Test 'exit addr' getting/setting address */
21static int cmd_exit_test(struct unit_test_state *uts)
22{
23 int i;
24
25 /*
26 * Test 'exit' with parameter -3, -2, -1, 0, 1, 2, 3 . Use all those
27 * parameters to cover also the special return value -2 that is used
28 * in HUSH to detect exit command.
29 *
30 * Always test whether 'exit' command:
31 * - exits out of the 'run' command
32 * - return value is propagated out of the 'run' command
33 * - return value can be tested on outside of 'run' command
34 * - return value can be printed outside of 'run' command
35 */
36 for (i = -3; i <= 3; i++) {
37 ut_assertok(console_record_reset_enable());
38 ut_assertok(run_commandf("setenv foo 'echo bar ; exit %d ; echo baz' ; run foo ; echo $?", i));
39 ut_assert_nextline("bar");
40 ut_assert_nextline("%d", i > 0 ? i : 0);
41 ut_assertok(ut_check_console_end(uts));
42
43 ut_assertok(console_record_reset_enable());
44 ut_assertok(run_commandf("setenv foo 'echo bar ; exit %d ; echo baz' ; run foo && echo quux ; echo $?", i));
45 ut_assert_nextline("bar");
46 if (i <= 0)
47 ut_assert_nextline("quux");
48 ut_assert_nextline("%d", i > 0 ? i : 0);
49 ut_assertok(ut_check_console_end(uts));
50
51 ut_assertok(console_record_reset_enable());
52 ut_assertok(run_commandf("setenv foo 'echo bar ; exit %d ; echo baz' ; run foo || echo quux ; echo $?", i));
53 ut_assert_nextline("bar");
54 if (i > 0)
55 ut_assert_nextline("quux");
56 /* Either 'exit' returns 0, or 'echo quux' returns 0 */
57 ut_assert_nextline("0");
58 ut_assertok(ut_check_console_end(uts));
59 }
60
61 /* Validate that 'exit' behaves the same way as 'exit 0' */
62 ut_assertok(console_record_reset_enable());
63 ut_assertok(run_commandf("setenv foo 'echo bar ; exit ; echo baz' ; run foo ; echo $?", i));
64 ut_assert_nextline("bar");
65 ut_assert_nextline("0");
66 ut_assertok(ut_check_console_end(uts));
67
68 ut_assertok(console_record_reset_enable());
69 ut_assertok(run_commandf("setenv foo 'echo bar ; exit ; echo baz' ; run foo && echo quux ; echo $?", i));
70 ut_assert_nextline("bar");
71 ut_assert_nextline("quux");
72 ut_assert_nextline("0");
73 ut_assertok(ut_check_console_end(uts));
74
75 ut_assertok(console_record_reset_enable());
76 ut_assertok(run_commandf("setenv foo 'echo bar ; exit ; echo baz' ; run foo || echo quux ; echo $?", i));
77 ut_assert_nextline("bar");
78 /* Either 'exit' returns 0, or 'echo quux' returns 0 */
79 ut_assert_nextline("0");
80 ut_assertok(ut_check_console_end(uts));
81
82 /* Validate that return value still propagates from 'run' command */
83 ut_assertok(console_record_reset_enable());
84 ut_assertok(run_commandf("setenv foo 'echo bar ; true' ; run foo ; echo $?", i));
85 ut_assert_nextline("bar");
86 ut_assert_nextline("0");
87 ut_assertok(ut_check_console_end(uts));
88
89 ut_assertok(console_record_reset_enable());
90 ut_assertok(run_commandf("setenv foo 'echo bar ; true' ; run foo && echo quux ; echo $?", i));
91 ut_assert_nextline("bar");
92 ut_assert_nextline("quux");
93 ut_assert_nextline("0");
94 ut_assertok(ut_check_console_end(uts));
95
96 ut_assertok(console_record_reset_enable());
97 ut_assertok(run_commandf("setenv foo 'echo bar ; true' ; run foo || echo quux ; echo $?", i));
98 ut_assert_nextline("bar");
99 /* The 'true' returns 0 */
100 ut_assert_nextline("0");
101 ut_assertok(ut_check_console_end(uts));
102
103 ut_assertok(console_record_reset_enable());
104 ut_assertok(run_commandf("setenv foo 'echo bar ; false' ; run foo ; echo $?", i));
105 ut_assert_nextline("bar");
106 ut_assert_nextline("1");
107 ut_assertok(ut_check_console_end(uts));
108
109 ut_assertok(console_record_reset_enable());
110 ut_assertok(run_commandf("setenv foo 'echo bar ; false' ; run foo && echo quux ; echo $?", i));
111 ut_assert_nextline("bar");
112 ut_assert_nextline("1");
113 ut_assertok(ut_check_console_end(uts));
114
115 ut_assertok(console_record_reset_enable());
116 ut_assertok(run_commandf("setenv foo 'echo bar ; false' ; run foo || echo quux ; echo $?", i));
117 ut_assert_nextline("bar");
118 ut_assert_nextline("quux");
119 /* The 'echo quux' returns 0 */
120 ut_assert_nextline("0");
121 ut_assertok(ut_check_console_end(uts));
122
123 return 0;
124}
125
126EXIT_TEST(cmd_exit_test, UT_TESTF_CONSOLE_REC);
127
128int do_ut_exit(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[])
129{
130 struct unit_test *tests = UNIT_TEST_SUITE_START(exit_test);
131 const int n_ents = UNIT_TEST_SUITE_COUNT(exit_test);
132
133 return cmd_ut_category("cmd_exit", "exit_test_", tests, n_ents,
134 argc, argv);
135}