]>
Commit | Line | Data |
---|---|---|
83d290c5 | 1 | // SPDX-License-Identifier: GPL-2.0+ |
2e7d35d2 | 2 | /* |
e721b882 | 3 | * Simple unit test library |
2e7d35d2 SG |
4 | * |
5 | * Copyright (c) 2013 Google, Inc | |
2e7d35d2 SG |
6 | */ |
7 | ||
8 | #include <common.h> | |
400175b0 | 9 | #include <console.h> |
8109863f | 10 | #include <malloc.h> |
ef7e2649 SG |
11 | #ifdef CONFIG_SANDBOX |
12 | #include <asm/state.h> | |
13 | #endif | |
401d1c4f | 14 | #include <asm/global_data.h> |
e721b882 JH |
15 | #include <test/test.h> |
16 | #include <test/ut.h> | |
2e7d35d2 | 17 | |
9ce8b402 SG |
18 | DECLARE_GLOBAL_DATA_PTR; |
19 | ||
e721b882 | 20 | void ut_fail(struct unit_test_state *uts, const char *fname, int line, |
2e7d35d2 SG |
21 | const char *func, const char *cond) |
22 | { | |
9ce8b402 | 23 | gd->flags &= ~(GD_FLG_SILENT | GD_FLG_RECORD); |
2e7d35d2 | 24 | printf("%s:%d, %s(): %s\n", fname, line, func, cond); |
e721b882 | 25 | uts->fail_count++; |
2e7d35d2 SG |
26 | } |
27 | ||
e721b882 | 28 | void ut_failf(struct unit_test_state *uts, const char *fname, int line, |
2e7d35d2 SG |
29 | const char *func, const char *cond, const char *fmt, ...) |
30 | { | |
31 | va_list args; | |
32 | ||
9ce8b402 | 33 | gd->flags &= ~(GD_FLG_SILENT | GD_FLG_RECORD); |
2e7d35d2 SG |
34 | printf("%s:%d, %s(): %s: ", fname, line, func, cond); |
35 | va_start(args, fmt); | |
36 | vprintf(fmt, args); | |
37 | va_end(args); | |
38 | putc('\n'); | |
e721b882 | 39 | uts->fail_count++; |
2e7d35d2 | 40 | } |
8109863f SG |
41 | |
42 | ulong ut_check_free(void) | |
43 | { | |
44 | struct mallinfo info = mallinfo(); | |
45 | ||
46 | return info.uordblks; | |
47 | } | |
48 | ||
49 | long ut_check_delta(ulong last) | |
50 | { | |
51 | return ut_check_free() - last; | |
52 | } | |
53 | ||
c1a2bb4f SG |
54 | static int readline_check(struct unit_test_state *uts) |
55 | { | |
56 | int ret; | |
57 | ||
58 | ret = console_record_readline(uts->actual_str, sizeof(uts->actual_str)); | |
59 | if (ret == -ENOSPC) { | |
60 | ut_fail(uts, __FILE__, __LINE__, __func__, | |
61 | "Console record buffer too small - increase CONFIG_CONSOLE_RECORD_OUT_SIZE"); | |
62 | return ret; | |
63 | } | |
64 | ||
65 | return 0; | |
66 | } | |
67 | ||
400175b0 SG |
68 | int ut_check_console_line(struct unit_test_state *uts, const char *fmt, ...) |
69 | { | |
70 | va_list args; | |
090d664e | 71 | int len; |
c1a2bb4f | 72 | int ret; |
400175b0 SG |
73 | |
74 | va_start(args, fmt); | |
090d664e | 75 | len = vsnprintf(uts->expect_str, sizeof(uts->expect_str), fmt, args); |
400175b0 | 76 | va_end(args); |
090d664e SG |
77 | if (len >= sizeof(uts->expect_str)) { |
78 | ut_fail(uts, __FILE__, __LINE__, __func__, | |
79 | "unit_test_state->expect_str too small"); | |
80 | return -EOVERFLOW; | |
81 | } | |
c1a2bb4f SG |
82 | ret = readline_check(uts); |
83 | if (ret < 0) | |
84 | return ret; | |
400175b0 SG |
85 | |
86 | return strcmp(uts->expect_str, uts->actual_str); | |
87 | } | |
88 | ||
33d7edfd SG |
89 | int ut_check_console_linen(struct unit_test_state *uts, const char *fmt, ...) |
90 | { | |
91 | va_list args; | |
090d664e | 92 | int len; |
c1a2bb4f | 93 | int ret; |
33d7edfd SG |
94 | |
95 | va_start(args, fmt); | |
090d664e | 96 | len = vsnprintf(uts->expect_str, sizeof(uts->expect_str), fmt, args); |
33d7edfd | 97 | va_end(args); |
090d664e SG |
98 | if (len >= sizeof(uts->expect_str)) { |
99 | ut_fail(uts, __FILE__, __LINE__, __func__, | |
100 | "unit_test_state->expect_str too small"); | |
101 | return -EOVERFLOW; | |
102 | } | |
c1a2bb4f SG |
103 | ret = readline_check(uts); |
104 | if (ret < 0) | |
105 | return ret; | |
33d7edfd SG |
106 | |
107 | return strncmp(uts->expect_str, uts->actual_str, | |
108 | strlen(uts->expect_str)); | |
109 | } | |
110 | ||
111 | int ut_check_skipline(struct unit_test_state *uts) | |
112 | { | |
c1a2bb4f SG |
113 | int ret; |
114 | ||
33d7edfd SG |
115 | if (!console_record_avail()) |
116 | return -ENFILE; | |
c1a2bb4f SG |
117 | ret = readline_check(uts); |
118 | if (ret < 0) | |
119 | return ret; | |
33d7edfd SG |
120 | |
121 | return 0; | |
122 | } | |
123 | ||
400175b0 SG |
124 | int ut_check_console_end(struct unit_test_state *uts) |
125 | { | |
c1a2bb4f SG |
126 | int ret; |
127 | ||
400175b0 SG |
128 | if (!console_record_avail()) |
129 | return 0; | |
c1a2bb4f SG |
130 | ret = readline_check(uts); |
131 | if (ret < 0) | |
132 | return ret; | |
400175b0 SG |
133 | |
134 | return 1; | |
135 | } | |
136 | ||
137 | int ut_check_console_dump(struct unit_test_state *uts, int total_bytes) | |
138 | { | |
139 | char *str = uts->actual_str; | |
140 | int upto; | |
141 | ||
142 | /* Handle empty dump */ | |
143 | if (!total_bytes) | |
144 | return 0; | |
145 | ||
146 | for (upto = 0; upto < total_bytes;) { | |
147 | int len; | |
148 | int bytes; | |
149 | ||
150 | len = console_record_readline(str, sizeof(uts->actual_str)); | |
151 | if (str[8] != ':' || str[9] != ' ') | |
152 | return 1; | |
153 | ||
c7b16d83 | 154 | bytes = len - 8 - 2 - 3 * 16 - 2; |
400175b0 SG |
155 | upto += bytes; |
156 | } | |
157 | ||
158 | return upto == total_bytes ? 0 : 1; | |
159 | } | |
ef7e2649 SG |
160 | |
161 | void ut_silence_console(struct unit_test_state *uts) | |
162 | { | |
163 | #ifdef CONFIG_SANDBOX | |
164 | struct sandbox_state *state = state_get_current(); | |
165 | ||
166 | if (!state->show_test_output) | |
167 | gd->flags |= GD_FLG_SILENT; | |
168 | #endif | |
169 | } | |
170 | ||
171 | void ut_unsilence_console(struct unit_test_state *uts) | |
172 | { | |
173 | gd->flags &= ~(GD_FLG_SILENT | GD_FLG_RECORD); | |
174 | } | |
47ec3ede SG |
175 | |
176 | void ut_set_skip_delays(struct unit_test_state *uts, bool skip_delays) | |
177 | { | |
178 | #ifdef CONFIG_SANDBOX | |
179 | state_set_skip_delays(skip_delays); | |
180 | #endif | |
181 | } |