| Line | Branch | Exec | Source |
|---|---|---|---|
| 1 | #include "test.h" | ||
| 2 | #include "command/alias.h" | ||
| 3 | #include "command/args.h" | ||
| 4 | #include "command/cache.h" | ||
| 5 | #include "command/parse.h" | ||
| 6 | #include "command/run.h" | ||
| 7 | #include "command/serialize.h" | ||
| 8 | #include "commands.h" | ||
| 9 | #include "editor.h" | ||
| 10 | #include "replace.h" | ||
| 11 | #include "util/ascii.h" | ||
| 12 | #include "util/path.h" | ||
| 13 | #include "util/str-util.h" | ||
| 14 | #include "version.h" | ||
| 15 | |||
| 16 | 1 | static void test_parse_command_arg(TestContext *ctx) | |
| 17 | { | ||
| 18 | 1 | EditorState *e = ctx->userdata; | |
| 19 | 1 | CommandRunner runner = normal_mode_cmdrunner(e); | |
| 20 | 1 | runner.flags &= ~CMDRUNNER_EXPAND_TILDE_SLASH; | |
| 21 | 1 | EXPECT_UINT_EQ(runner.flags, 0); | |
| 22 | 1 | ASSERT_PTREQ(runner.e, e); | |
| 23 | 1 | ASSERT_PTREQ(runner.cmds, &normal_commands); | |
| 24 | 1 | EXPECT_PTREQ(runner.lookup_alias, find_normal_alias); | |
| 25 | 1 | EXPECT_PTREQ(runner.home_dir, &e->home_dir); | |
| 26 | 1 | EXPECT_EQ(runner.recursion_count, 0); | |
| 27 | |||
| 28 | // Single, unquoted argument | ||
| 29 | 1 | char *arg = parse_command_arg(&runner, STRN("arg")); | |
| 30 | 1 | EXPECT_STREQ(arg, "arg"); | |
| 31 | 1 | free(arg); | |
| 32 | |||
| 33 | // Two unquoted, space-separated arguments | ||
| 34 | 1 | arg = parse_command_arg(&runner, STRN("hello world")); | |
| 35 | 1 | EXPECT_STREQ(arg, "hello"); | |
| 36 | 1 | free(arg); | |
| 37 | |||
| 38 | // Two unquoted, tab-separated arguments | ||
| 39 | 1 | arg = parse_command_arg(&runner, STRN("hello\tworld")); | |
| 40 | 1 | EXPECT_STREQ(arg, "hello"); | |
| 41 | 1 | free(arg); | |
| 42 | |||
| 43 | // Unquoted argument preceded by whitespace | ||
| 44 | 1 | arg = parse_command_arg(&runner, STRN(" x")); | |
| 45 | 1 | EXPECT_STREQ(arg, ""); | |
| 46 | 1 | free(arg); | |
| 47 | |||
| 48 | // Single-quoted argument, including whitespace | ||
| 49 | 1 | arg = parse_command_arg(&runner, STRN("' foo ' ..")); | |
| 50 | 1 | EXPECT_STREQ(arg, " foo "); | |
| 51 | 1 | free(arg); | |
| 52 | |||
| 53 | // Several adjacent, quoted strings forming a single argument | ||
| 54 | 1 | arg = parse_command_arg(&runner, STRN("\"foo\"'bar'' baz '\"etc\".")); | |
| 55 | 1 | EXPECT_STREQ(arg, "foobar baz etc."); | |
| 56 | 1 | free(arg); | |
| 57 | |||
| 58 | // Control character escapes in a double-quoted string | ||
| 59 | 1 | arg = parse_command_arg(&runner, STRN("\"\\a\\b\\t\\n\\v\\f\\r\"")); | |
| 60 | 1 | EXPECT_STREQ(arg, "\a\b\t\n\v\f\r"); | |
| 61 | 1 | free(arg); | |
| 62 | |||
| 63 | // Backslash escape sequence in a double-quoted string | ||
| 64 | 1 | arg = parse_command_arg(&runner, STRN("\"\\\\\"")); | |
| 65 | 1 | EXPECT_STREQ(arg, "\\"); | |
| 66 | 1 | free(arg); | |
| 67 | |||
| 68 | // Double-quote escape sequence in a double-quoted string | ||
| 69 | 1 | arg = parse_command_arg(&runner, STRN("\"\\\"\"")); | |
| 70 | 1 | EXPECT_STREQ(arg, "\""); | |
| 71 | 1 | free(arg); | |
| 72 | |||
| 73 | // Escape character escape sequence in a double-quoted string | ||
| 74 | 1 | arg = parse_command_arg(&runner, STRN("\"\\e[0m\"")); | |
| 75 | 1 | EXPECT_STREQ(arg, "\033[0m"); | |
| 76 | 1 | free(arg); | |
| 77 | |||
| 78 | // Unrecognized escape sequence in a double-quoted string | ||
| 79 | 1 | arg = parse_command_arg(&runner, STRN("\"\\z\"")); | |
| 80 | 1 | EXPECT_STREQ(arg, "\\z"); | |
| 81 | 1 | free(arg); | |
| 82 | |||
| 83 | // Hexadecimal escape sequences in a double-quoted string | ||
| 84 | 1 | arg = parse_command_arg(&runner, STRN("\"\\x1B[31m\\x7E\\x2f\\x1b[0m\"")); | |
| 85 | 1 | EXPECT_STREQ(arg, "\x1B[31m~/\x1B[0m"); | |
| 86 | 1 | free(arg); | |
| 87 | |||
| 88 | // Invalid hexadecimal escape sequences | ||
| 89 | 1 | arg = parse_command_arg(&runner, STRN("\"\\x\\x1\\xFG\\xz1\"")); | |
| 90 | 1 | EXPECT_STREQ(arg, "Gz1"); | |
| 91 | 1 | free(arg); | |
| 92 | |||
| 93 | // Incomplete hexadecimal escape sequence | ||
| 94 | 1 | arg = parse_command_arg(&runner, STRN("\"\\x")); | |
| 95 | 1 | EXPECT_STREQ(arg, ""); | |
| 96 | 1 | free(arg); | |
| 97 | |||
| 98 | // 4-digit Unicode escape sequence | ||
| 99 | 1 | arg = parse_command_arg(&runner, STRN("\"\\u148A\"")); | |
| 100 | 1 | EXPECT_STREQ(arg, "\xE1\x92\x8A"); | |
| 101 | 1 | free(arg); | |
| 102 | |||
| 103 | // 8-digit Unicode escape sequence | ||
| 104 | 1 | arg = parse_command_arg(&runner, STRN("\"\\U0001F4A4\"")); | |
| 105 | 1 | EXPECT_STREQ(arg, "\xF0\x9F\x92\xA4"); | |
| 106 | 1 | free(arg); | |
| 107 | |||
| 108 | // "\U" escape sequence terminated by non-hexadecimal character | ||
| 109 | 1 | arg = parse_command_arg(&runner, STRN("\"\\U1F4A4...\"")); | |
| 110 | 1 | EXPECT_STREQ(arg, "\xF0\x9F\x92\xA4..."); | |
| 111 | 1 | free(arg); | |
| 112 | |||
| 113 | // Incomplete Unicode escape sequence | ||
| 114 | 1 | arg = parse_command_arg(&runner, STRN("\"\\u")); | |
| 115 | 1 | EXPECT_STREQ(arg, ""); | |
| 116 | 1 | free(arg); | |
| 117 | |||
| 118 | // Invalid Unicode escape sequence | ||
| 119 | 1 | arg = parse_command_arg(&runner, STRN("\"\\ugef\"")); | |
| 120 | 1 | EXPECT_STREQ(arg, "gef"); | |
| 121 | 1 | free(arg); | |
| 122 | |||
| 123 | // Unsupported, escape-like sequences in a single-quoted string | ||
| 124 | 1 | arg = parse_command_arg(&runner, STRN("'\\t\\n'")); | |
| 125 | 1 | EXPECT_STREQ(arg, "\\t\\n"); | |
| 126 | 1 | free(arg); | |
| 127 | |||
| 128 | // Trailing backslash | ||
| 129 | // Note: `s` is unterminated, to allow ASan to catch OOB reads | ||
| 130 | 1 | static const char s[4] = "123\\"; | |
| 131 | 1 | arg = parse_command_arg(&runner, s, sizeof s); | |
| 132 | 1 | EXPECT_STREQ(arg, "123"); | |
| 133 | 1 | free(arg); | |
| 134 | |||
| 135 | // Single-quoted, empty string | ||
| 136 | 1 | arg = parse_command_arg(&runner, STRN("''")); | |
| 137 | 1 | EXPECT_STREQ(arg, ""); | |
| 138 | 1 | free(arg); | |
| 139 | |||
| 140 | // Double-quoted, empty string | ||
| 141 | 1 | arg = parse_command_arg(&runner, STRN("\"\"")); | |
| 142 | 1 | EXPECT_STREQ(arg, ""); | |
| 143 | 1 | free(arg); | |
| 144 | |||
| 145 | // NULL input with zero length | ||
| 146 | 1 | arg = parse_command_arg(&runner, NULL, 0); | |
| 147 | 1 | EXPECT_STREQ(arg, ""); | |
| 148 | 1 | free(arg); | |
| 149 | |||
| 150 | // Empty input | ||
| 151 | 1 | arg = parse_command_arg(&runner, "", 1); | |
| 152 | 1 | EXPECT_STREQ(arg, ""); | |
| 153 | 1 | free(arg); | |
| 154 | |||
| 155 | // Built-in vars (expand to nothing; buffer isn't initialized yet) | ||
| 156 | 1 | static const char vars[] = | |
| 157 | "' '$FILE" | ||
| 158 | "' '$FILEDIR" | ||
| 159 | "' '$RFILE" | ||
| 160 | "' '$RFILEDIR" | ||
| 161 | "' '$FILETYPE" | ||
| 162 | "' '$LINENO" | ||
| 163 | "' '$COLNO" | ||
| 164 | "' '$WORD" | ||
| 165 | ; | ||
| 166 | 1 | arg = parse_command_arg(&runner, vars, sizeof(vars) - 1); | |
| 167 | 1 | EXPECT_STREQ(arg, " "); | |
| 168 | 1 | free(arg); | |
| 169 | |||
| 170 | // Built-in $DTE_HOME var (expands to user config dir) | ||
| 171 | 1 | arg = parse_command_arg(&runner, STRN("$DTE_HOME")); | |
| 172 | 1 | EXPECT_TRUE(path_is_absolute(arg)); | |
| 173 | 1 | EXPECT_TRUE(str_has_suffix(arg, "/test/DTE_HOME")); | |
| 174 | 1 | free(arg); | |
| 175 | |||
| 176 | // Built-in ${DTE_HOME} var in ${bracketed} form | ||
| 177 | 1 | arg = parse_command_arg(&runner, STRN("${DTE_HOME}")); | |
| 178 | 1 | EXPECT_TRUE(path_is_absolute(arg)); | |
| 179 | 1 | EXPECT_TRUE(str_has_suffix(arg, "/test/DTE_HOME")); | |
| 180 | 1 | free(arg); | |
| 181 | |||
| 182 | // Bracketed var with missing end delimiter | ||
| 183 | 1 | arg = parse_command_arg(&runner, STRN("${DTE_HOME")); | |
| 184 | 1 | EXPECT_STREQ(arg, ""); | |
| 185 | 1 | free(arg); | |
| 186 | |||
| 187 | // Empty bracketed var | ||
| 188 | 1 | arg = parse_command_arg(&runner, STRN("aa${}zz")); | |
| 189 | 1 | EXPECT_STREQ(arg, "aazz"); | |
| 190 | 1 | free(arg); | |
| 191 | |||
| 192 | // Built-in $MSGPOS var (expands to "1" by default) | ||
| 193 | 1 | arg = parse_command_arg(&runner, STRN("$MSGPOS\\ $MSGPOS_A\\ $MSGPOS_B\\ $MSGPOS_C")); | |
| 194 | 1 | EXPECT_STREQ(arg, "1 1 1 1"); | |
| 195 | 1 | free(arg); | |
| 196 | |||
| 197 | // Environment var (via getenv(3)) | ||
| 198 | 1 | arg = parse_command_arg(&runner, STRN("$DTE_VERSION")); | |
| 199 | 1 | EXPECT_STREQ(arg, VERSION); | |
| 200 | 1 | free(arg); | |
| 201 | |||
| 202 | // Tilde expansion | ||
| 203 | 1 | runner.flags |= CMDRUNNER_EXPAND_TILDE_SLASH; | |
| 204 | 1 | arg = parse_command_arg(&runner, STRN("~/filename")); | |
| 205 | 1 | EXPECT_TRUE(str_has_suffix(arg, "/test/HOME/filename")); | |
| 206 | 1 | free(arg); | |
| 207 | |||
| 208 | 1 | runner.flags &= ~CMDRUNNER_EXPAND_TILDE_SLASH; | |
| 209 | 1 | arg = parse_command_arg(&runner, STRN("'xyz")); | |
| 210 | 1 | EXPECT_STREQ(arg, "xyz"); | |
| 211 | 1 | free(arg); | |
| 212 | |||
| 213 | 1 | arg = parse_command_arg(&runner, STRN("\"\\u148A\"xyz'foo'\"\\x5A\"\\;\t.")); | |
| 214 | 1 | EXPECT_STREQ(arg, "\xE1\x92\x8AxyzfooZ;"); | |
| 215 | 1 | free(arg); | |
| 216 | 1 | } | |
| 217 | |||
| 218 | 3 | static CommandRunner mock_cmdrunner(const CommandSet *cmds) | |
| 219 | { | ||
| 220 | 3 | static ErrorBuffer ebuf = {.print_to_stderr = false}; | |
| 221 | 3 | return (CommandRunner){.cmds = cmds, .ebuf = &ebuf}; | |
| 222 | } | ||
| 223 | |||
| 224 | 1 | static void test_parse_commands(TestContext *ctx) | |
| 225 | { | ||
| 226 | 1 | CommandRunner runner = mock_cmdrunner(&normal_commands); | |
| 227 | 1 | PointerArray array = PTR_ARRAY_INIT; | |
| 228 | 1 | EXPECT_EQ(parse_commands(&runner, &array, " left -c;;"), CMDERR_NONE); | |
| 229 | 1 | ASSERT_EQ(array.count, 5); | |
| 230 | 1 | EXPECT_STREQ(array.ptrs[0], "left"); | |
| 231 | 1 | EXPECT_STREQ(array.ptrs[1], "-c"); | |
| 232 | 1 | EXPECT_NULL(array.ptrs[2]); | |
| 233 | 1 | EXPECT_NULL(array.ptrs[3]); | |
| 234 | 1 | EXPECT_NULL(array.ptrs[4]); | |
| 235 | 1 | ptr_array_free(&array); | |
| 236 | |||
| 237 | 1 | EXPECT_EQ(parse_commands(&runner, &array, "save -e UTF-8 file.c; close -q"), CMDERR_NONE); | |
| 238 | 1 | ASSERT_EQ(array.count, 8); | |
| 239 | 1 | EXPECT_STREQ(array.ptrs[0], "save"); | |
| 240 | 1 | EXPECT_STREQ(array.ptrs[1], "-e"); | |
| 241 | 1 | EXPECT_STREQ(array.ptrs[2], "UTF-8"); | |
| 242 | 1 | EXPECT_STREQ(array.ptrs[3], "file.c"); | |
| 243 | 1 | EXPECT_NULL(array.ptrs[4]); | |
| 244 | 1 | EXPECT_STREQ(array.ptrs[5], "close"); | |
| 245 | 1 | EXPECT_STREQ(array.ptrs[6], "-q"); | |
| 246 | 1 | EXPECT_NULL(array.ptrs[7]); | |
| 247 | 1 | ptr_array_free(&array); | |
| 248 | |||
| 249 | 1 | EXPECT_EQ(parse_commands(&runner, &array, "\n ; ; \t\n "), CMDERR_NONE); | |
| 250 | 1 | ASSERT_EQ(array.count, 3); | |
| 251 | 1 | EXPECT_NULL(array.ptrs[0]); | |
| 252 | 1 | EXPECT_NULL(array.ptrs[1]); | |
| 253 | 1 | EXPECT_NULL(array.ptrs[2]); | |
| 254 | 1 | ptr_array_free(&array); | |
| 255 | |||
| 256 | 1 | EXPECT_EQ(parse_commands(&runner, &array, ""), CMDERR_NONE); | |
| 257 | 1 | ASSERT_EQ(array.count, 1); | |
| 258 | 1 | EXPECT_NULL(array.ptrs[0]); | |
| 259 | 1 | ptr_array_free(&array); | |
| 260 | |||
| 261 | 1 | EXPECT_EQ(parse_commands(&runner, &array, "insert '... "), CMDERR_UNCLOSED_SQUOTE); | |
| 262 | 1 | ptr_array_free(&array); | |
| 263 | |||
| 264 | 1 | EXPECT_EQ(parse_commands(&runner, &array, "insert \" "), CMDERR_UNCLOSED_DQUOTE); | |
| 265 | 1 | ptr_array_free(&array); | |
| 266 | |||
| 267 | 1 | EXPECT_EQ(parse_commands(&runner, &array, "insert \"\\\" "), CMDERR_UNCLOSED_DQUOTE); | |
| 268 | 1 | ptr_array_free(&array); | |
| 269 | |||
| 270 | 1 | EXPECT_EQ(parse_commands(&runner, &array, "insert \\"), CMDERR_UNEXPECTED_EOF); | |
| 271 | 1 | ptr_array_free(&array); | |
| 272 | |||
| 273 | 1 | EXPECT_EQ(parse_commands(&runner, &array, "insert \"\\"), CMDERR_UNEXPECTED_EOF); | |
| 274 | 1 | ptr_array_free(&array); | |
| 275 | 1 | } | |
| 276 | |||
| 277 | 1 | static void test_find_normal_command(TestContext *ctx) | |
| 278 | { | ||
| 279 | 1 | const Command *cmd = find_normal_command("alias"); | |
| 280 | 1 | ASSERT_NONNULL(cmd); | |
| 281 | 1 | EXPECT_STREQ(cmd->name, "alias"); | |
| 282 | |||
| 283 | 1 | cmd = find_normal_command("bind"); | |
| 284 | 1 | ASSERT_NONNULL(cmd); | |
| 285 | 1 | EXPECT_STREQ(cmd->name, "bind"); | |
| 286 | |||
| 287 | 1 | cmd = find_normal_command("wswap"); | |
| 288 | 1 | ASSERT_NONNULL(cmd); | |
| 289 | 1 | EXPECT_STREQ(cmd->name, "wswap"); | |
| 290 | |||
| 291 | 1 | EXPECT_NULL(find_normal_command("alia")); | |
| 292 | 1 | EXPECT_NULL(find_normal_command("aliass")); | |
| 293 | 1 | EXPECT_NULL(find_normal_command("Alias")); | |
| 294 | 1 | EXPECT_NULL(find_normal_command("bin ")); | |
| 295 | 1 | EXPECT_NULL(find_normal_command("bind ")); | |
| 296 | 1 | EXPECT_NULL(find_normal_command(" bind")); | |
| 297 | 1 | EXPECT_NULL(find_normal_command("bind!")); | |
| 298 | 1 | EXPECT_NULL(find_normal_command("bind\n")); | |
| 299 | 1 | } | |
| 300 | |||
| 301 | 1 | static void test_parse_args(TestContext *ctx) | |
| 302 | { | ||
| 303 | 1 | const CommandSet *cmds = &normal_commands; | |
| 304 | 1 | const CommandRunner runner = mock_cmdrunner(cmds); | |
| 305 | 1 | const char *cmd_str = "open -g file.c file.h *.mk -e UTF-8"; | |
| 306 | 1 | PointerArray array = PTR_ARRAY_INIT; | |
| 307 | 1 | ASSERT_NONNULL(cmds); | |
| 308 | 1 | ASSERT_EQ(parse_commands(&runner, &array, cmd_str), CMDERR_NONE); | |
| 309 | 1 | ASSERT_EQ(array.count, 8); | |
| 310 | |||
| 311 | 1 | const Command *cmd = cmds->lookup(array.ptrs[0]); | |
| 312 | 1 | ASSERT_NONNULL(cmd); | |
| 313 | 1 | EXPECT_STREQ(cmd->name, "open"); | |
| 314 | |||
| 315 | 1 | CommandArgs a = cmdargs_new((char**)array.ptrs + 1); | |
| 316 | 1 | ASSERT_EQ(do_parse_args(cmd, &a), ARGERR_NONE); | |
| 317 | 1 | EXPECT_EQ(a.nr_flags, 2); | |
| 318 | 1 | EXPECT_EQ(a.flags[0], 'g'); | |
| 319 | 1 | EXPECT_EQ(a.flags[1], 'e'); | |
| 320 | 1 | EXPECT_EQ(a.flags[2], '\0'); | |
| 321 | 1 | EXPECT_TRUE(cmdargs_has_flag(&a, 'g')); | |
| 322 | 1 | EXPECT_TRUE(cmdargs_has_flag(&a, 'e')); | |
| 323 | 1 | EXPECT_FALSE(cmdargs_has_flag(&a, 'f')); | |
| 324 | 1 | EXPECT_FALSE(cmdargs_has_flag(&a, 'E')); | |
| 325 | 1 | EXPECT_FALSE(cmdargs_has_flag(&a, '0')); | |
| 326 | 1 | EXPECT_EQ(a.nr_flag_args, 1); | |
| 327 | 1 | EXPECT_STREQ(a.args[0], "UTF-8"); | |
| 328 | 1 | EXPECT_EQ(a.nr_args, 3); | |
| 329 | 1 | EXPECT_STREQ(a.args[1], "file.c"); | |
| 330 | 1 | EXPECT_STREQ(a.args[2], "file.h"); | |
| 331 | 1 | EXPECT_STREQ(a.args[3], "*.mk"); | |
| 332 | 1 | EXPECT_NULL(a.args[4]); | |
| 333 | |||
| 334 | 1 | ptr_array_free(&array); | |
| 335 | 1 | EXPECT_NULL(array.ptrs); | |
| 336 | 1 | EXPECT_EQ(array.alloc, 0); | |
| 337 | 1 | EXPECT_EQ(array.count, 0); | |
| 338 | |||
| 339 | 1 | cmd_str = "bind 1 2 3 4 5 -6"; | |
| 340 | 1 | ASSERT_EQ(parse_commands(&runner, &array, cmd_str), CMDERR_NONE); | |
| 341 | 1 | ASSERT_EQ(array.count, 8); | |
| 342 | 1 | EXPECT_STREQ(array.ptrs[0], "bind"); | |
| 343 | 1 | EXPECT_STREQ(array.ptrs[1], "1"); | |
| 344 | 1 | EXPECT_STREQ(array.ptrs[6], "-6"); | |
| 345 | 1 | EXPECT_NULL(array.ptrs[7]); | |
| 346 | |||
| 347 | 1 | cmd = cmds->lookup(array.ptrs[0]); | |
| 348 | 1 | ASSERT_NONNULL(cmd); | |
| 349 | 1 | EXPECT_STREQ(cmd->name, "bind"); | |
| 350 | 1 | EXPECT_EQ(cmd->max_args, 2); | |
| 351 | 1 | EXPECT_EQ(cmd->flags[0], 'c'); | |
| 352 | |||
| 353 | 1 | a = cmdargs_new((char**)array.ptrs + 1); | |
| 354 | 1 | a.flags[0] = 'X'; | |
| 355 | 1 | ASSERT_EQ(do_parse_args(cmd, &a), ARGERR_TOO_MANY_ARGUMENTS); | |
| 356 | 1 | EXPECT_EQ(a.nr_args, 6); | |
| 357 | 1 | EXPECT_EQ(a.nr_flags, 0); | |
| 358 | 1 | EXPECT_EQ(a.nr_flag_args, 0); | |
| 359 | 1 | EXPECT_EQ(a.flags[0], '\0'); | |
| 360 | 1 | EXPECT_EQ(a.flag_set, 0); | |
| 361 | 1 | ptr_array_free(&array); | |
| 362 | |||
| 363 | 1 | cmd_str = "open \"-\\xff\""; | |
| 364 | 1 | ASSERT_EQ(parse_commands(&runner, &array, cmd_str), CMDERR_NONE); | |
| 365 | 1 | ASSERT_EQ(array.count, 3); | |
| 366 | 1 | EXPECT_STREQ(array.ptrs[0], "open"); | |
| 367 | 1 | EXPECT_STREQ(array.ptrs[1], "-\xff"); | |
| 368 | 1 | EXPECT_NULL(array.ptrs[2]); | |
| 369 | 1 | cmd = cmds->lookup(array.ptrs[0]); | |
| 370 | 1 | ASSERT_NONNULL(cmd); | |
| 371 | 1 | a = cmdargs_new((char**)array.ptrs + 1); | |
| 372 | 1 | EXPECT_EQ(do_parse_args(cmd, &a), ARGERR_INVALID_OPTION); | |
| 373 | 1 | EXPECT_EQ(a.flags[0], '\xff'); | |
| 374 | 1 | ptr_array_free(&array); | |
| 375 | |||
| 376 | 1 | ASSERT_EQ(parse_commands(&runner, &array, "save -e"), CMDERR_NONE); | |
| 377 | 1 | ASSERT_EQ(array.count, 3); | |
| 378 | 1 | cmd = cmds->lookup(array.ptrs[0]); | |
| 379 | 1 | ASSERT_NONNULL(cmd); | |
| 380 | 1 | a = cmdargs_new((char**)array.ptrs + 1); | |
| 381 | 1 | EXPECT_EQ(do_parse_args(cmd, &a), ARGERR_OPTION_ARGUMENT_MISSING); | |
| 382 | 1 | EXPECT_EQ(a.flags[0], 'e'); | |
| 383 | 1 | ptr_array_free(&array); | |
| 384 | |||
| 385 | 1 | ASSERT_EQ(parse_commands(&runner, &array, "save -eUTF-8"), CMDERR_NONE); | |
| 386 | 1 | ASSERT_EQ(array.count, 3); | |
| 387 | 1 | cmd = cmds->lookup(array.ptrs[0]); | |
| 388 | 1 | ASSERT_NONNULL(cmd); | |
| 389 | 1 | a = cmdargs_new((char**)array.ptrs + 1); | |
| 390 | 1 | EXPECT_EQ(do_parse_args(cmd, &a), ARGERR_OPTION_ARGUMENT_NOT_SEPARATE); | |
| 391 | 1 | EXPECT_EQ(a.flags[0], 'e'); | |
| 392 | 1 | ptr_array_free(&array); | |
| 393 | 1 | } | |
| 394 | |||
| 395 | 1 | static void test_cached_command_new(TestContext *ctx) | |
| 396 | { | ||
| 397 | 1 | static const char cmd_str[] = "open -t -e UTF-8 file.c inc.h"; | |
| 398 | 1 | const CommandRunner runner = mock_cmdrunner(&normal_commands); | |
| 399 | 1 | CachedCommand *cc = cached_command_new(&runner, cmd_str); | |
| 400 | 1 | ASSERT_NONNULL(cc); | |
| 401 | 1 | ASSERT_NONNULL(cc->cmd); | |
| 402 | 1 | EXPECT_PTREQ(cc->cmd, runner.cmds->lookup("open")); | |
| 403 | 1 | EXPECT_STREQ(cc->cmd_str, cmd_str); | |
| 404 | 1 | ASSERT_EQ(cc->a.nr_args, 2); | |
| 405 | 1 | ASSERT_EQ(cc->a.nr_flag_args, 1); | |
| 406 | 1 | ASSERT_EQ(cc->a.nr_flags, 2); | |
| 407 | 1 | EXPECT_STREQ(cc->a.args[0], "UTF-8"); | |
| 408 | 1 | EXPECT_STREQ(cc->a.args[1], "file.c"); | |
| 409 | 1 | EXPECT_STREQ(cc->a.args[2], "inc.h"); | |
| 410 | 1 | EXPECT_EQ(cc->a.flags[0], 't'); | |
| 411 | 1 | EXPECT_EQ(cc->a.flags[1], 'e'); | |
| 412 | 1 | EXPECT_TRUE(cmdargs_has_flag(&cc->a, 't')); | |
| 413 | 1 | EXPECT_TRUE(cmdargs_has_flag(&cc->a, 'e')); | |
| 414 | 1 | cached_command_free(cc); | |
| 415 | 1 | cached_command_free(NULL); | |
| 416 | |||
| 417 | 1 | static const char *const uncacheable[] = { | |
| 418 | "", // No command | ||
| 419 | "zxcvbnm", // Invalid command | ||
| 420 | "left; right", // Multiple commands | ||
| 421 | "insert $DTE_HOME", // Variable expansion | ||
| 422 | "insert -xyz321", // Invalid flags | ||
| 423 | "alias 1 2 3", // Too many arguments | ||
| 424 | "alias", // Too few arguments | ||
| 425 | "insert 'xyz.", // CMDERR_UNCLOSED_SQUOTE | ||
| 426 | "insert \"xyz", // CMDERR_UNCLOSED_DQUOTE | ||
| 427 | "insert xyz\\", // CMDERR_UNEXPECTED_EOF | ||
| 428 | }; | ||
| 429 | |||
| 430 |
2/2✓ Branch 0 (28→23) taken 10 times.
✓ Branch 1 (28→29) taken 1 times.
|
11 | for (size_t i = 0; i < ARRAYLEN(uncacheable); i++) { |
| 431 | 10 | cc = cached_command_new(&runner, uncacheable[i]); | |
| 432 | 10 | ASSERT_NONNULL(cc); | |
| 433 | 10 | EXPECT_NULL(cc->cmd); | |
| 434 | 10 | cached_command_free(cc); | |
| 435 | } | ||
| 436 | 1 | } | |
| 437 | |||
| 438 | 19 | static const char *escape_command_arg(String *buf, const char *arg, bool escape_tilde) | |
| 439 | { | ||
| 440 | 19 | string_clear(buf); | |
| 441 | 19 | string_append_escaped_arg(buf, arg, escape_tilde); | |
| 442 | 19 | return string_borrow_cstring(buf); | |
| 443 | } | ||
| 444 | |||
| 445 | 1 | static void test_string_append_escaped_arg(TestContext *ctx) | |
| 446 | { | ||
| 447 | 1 | String buf = string_new(64); | |
| 448 | 1 | const char *str = escape_command_arg(&buf, "arg", false); | |
| 449 | 1 | EXPECT_STREQ(str, "arg"); | |
| 450 | |||
| 451 | 1 | str = escape_command_arg(&buf, "arg-x.y:z", false); | |
| 452 | 1 | EXPECT_STREQ(str, "arg-x.y:z"); | |
| 453 | |||
| 454 | 1 | str = escape_command_arg(&buf, "", false); | |
| 455 | 1 | EXPECT_STREQ(str, "''"); | |
| 456 | |||
| 457 | 1 | str = escape_command_arg(&buf, " ", false); | |
| 458 | 1 | EXPECT_STREQ(str, "' '"); | |
| 459 | |||
| 460 | 1 | str = escape_command_arg(&buf, "hello world", false); | |
| 461 | 1 | EXPECT_STREQ(str, "'hello world'"); | |
| 462 | |||
| 463 | 1 | str = escape_command_arg(&buf, "line1\nline2\n", false); | |
| 464 | 1 | EXPECT_STREQ(str, "\"line1\\nline2\\n\""); | |
| 465 | |||
| 466 | 1 | str = escape_command_arg(&buf, " \t\r\n\x1F\x7F", false); | |
| 467 | 1 | EXPECT_STREQ(str, "\" \\t\\r\\n\\x1F\\x7F\""); | |
| 468 | |||
| 469 | 1 | str = escape_command_arg(&buf, "x ' y", false); | |
| 470 | 1 | EXPECT_STREQ(str, "\"x ' y\""); | |
| 471 | |||
| 472 | 1 | str = escape_command_arg(&buf, "\033[P", false); | |
| 473 | 1 | EXPECT_STREQ(str, "\"\\e[P\""); | |
| 474 | |||
| 475 | 1 | str = escape_command_arg(&buf, "\"''\"", false); | |
| 476 | 1 | EXPECT_STREQ(str, "\"\\\"''\\\"\""); | |
| 477 | |||
| 478 | 1 | str = escape_command_arg(&buf, "\t\\", false); | |
| 479 | 1 | EXPECT_STREQ(str, "\"\\t\\\\\""); | |
| 480 | |||
| 481 | 1 | str = escape_command_arg(&buf, "~/file with spaces", false); | |
| 482 | 1 | EXPECT_STREQ(str, "~/'file with spaces'"); | |
| 483 | 1 | str = escape_command_arg(&buf, "~/file with spaces", true); | |
| 484 | 1 | EXPECT_STREQ(str, "'~/file with spaces'"); | |
| 485 | |||
| 486 | 1 | str = escape_command_arg(&buf, "~/need \t\ndquotes", false); | |
| 487 | 1 | EXPECT_STREQ(str, "~/\"need \\t\\ndquotes\""); | |
| 488 | 1 | str = escape_command_arg(&buf, "~/need \t\ndquotes", true); | |
| 489 | 1 | EXPECT_STREQ(str, "\"~/need \\t\\ndquotes\""); | |
| 490 | |||
| 491 | 1 | str = escape_command_arg(&buf, "~/file-with-no-spaces", false); | |
| 492 | 1 | EXPECT_STREQ(str, "~/file-with-no-spaces"); | |
| 493 | 1 | str = escape_command_arg(&buf, "~/file-with-no-spaces", true); | |
| 494 | 1 | EXPECT_STREQ(str, "\\~/file-with-no-spaces"); | |
| 495 | |||
| 496 | 1 | str = escape_command_arg(&buf, "~/", false); | |
| 497 | 1 | EXPECT_STREQ(str, "~/"); | |
| 498 | 1 | str = escape_command_arg(&buf, "~/", true); | |
| 499 | 1 | EXPECT_STREQ(str, "\\~/"); | |
| 500 | |||
| 501 | 1 | string_free(&buf); | |
| 502 | 1 | } | |
| 503 | |||
| 504 | 1 | static void test_command_struct_layout(TestContext *ctx) | |
| 505 | { | ||
| 506 | 1 | const Command *cmd = find_normal_command("open"); | |
| 507 | 1 | EXPECT_STREQ(cmd->name, "open"); | |
| 508 | 1 | EXPECT_STREQ(cmd->flags, "e=gt"); | |
| 509 | 1 | EXPECT_FALSE(cmd->cmdopts & CMDOPT_ALLOW_IN_RC); | |
| 510 | 1 | EXPECT_UINT_EQ(cmd->min_args, 0); | |
| 511 | 1 | EXPECT_UINT_EQ(cmd->max_args, 0xFF); | |
| 512 | |||
| 513 | IGNORE_WARNING("-Wpedantic") | ||
| 514 | 1 | EXPECT_NONNULL(cmd->cmd); | |
| 515 | UNIGNORE_WARNINGS | ||
| 516 | 1 | } | |
| 517 | |||
| 518 | 1 | static void test_cmdargs_flagset_idx(TestContext *ctx) | |
| 519 | { | ||
| 520 | 1 | EXPECT_EQ(cmdargs_flagset_idx('A'), 1); | |
| 521 | 1 | EXPECT_EQ(cmdargs_flagset_idx('Z'), 26); | |
| 522 | 1 | EXPECT_EQ(cmdargs_flagset_idx('a'), 27); | |
| 523 | 1 | EXPECT_EQ(cmdargs_flagset_idx('z'), 52); | |
| 524 | 1 | EXPECT_EQ(cmdargs_flagset_idx('0'), 53); | |
| 525 | 1 | EXPECT_EQ(cmdargs_flagset_idx('1'), 54); | |
| 526 | 1 | EXPECT_EQ(cmdargs_flagset_idx('9'), 62); | |
| 527 | |||
| 528 |
2/2✓ Branch 0 (22→17) taken 75 times.
✓ Branch 1 (22→23) taken 1 times.
|
76 | for (unsigned char c = '0', z = 'z'; c <= z; c++) { |
| 529 |
2/2✓ Branch 0 (17→18) taken 62 times.
✓ Branch 1 (17→21) taken 13 times.
|
75 | if (ascii_isalnum(c)) { |
| 530 | 62 | const unsigned int idx = cmdargs_flagset_idx(c); | |
| 531 | 62 | EXPECT_TRUE(idx < 63); | |
| 532 | 62 | EXPECT_EQ(idx, base64_decode(c) + 1); | |
| 533 | } | ||
| 534 | } | ||
| 535 | 1 | } | |
| 536 | |||
| 537 | 1 | static void test_cmdargs_flagset_from_str(TestContext *ctx) | |
| 538 | { | ||
| 539 | 1 | CommandFlagSet r = 0; | |
| 540 | 1 | r |= cmdargs_flagset_bit('c'); | |
| 541 | 1 | r |= cmdargs_flagset_bit('n'); | |
| 542 | 1 | r |= cmdargs_flagset_bit('s'); | |
| 543 | 1 | r |= cmdargs_flagset_bit('T'); | |
| 544 | 1 | EXPECT_EQ(u64_popcount(r), 4); | |
| 545 | 1 | EXPECT_UINT_EQ(r, cmdargs_flagset_from_str("cnsT")); | |
| 546 | |||
| 547 | 1 | r = 0; | |
| 548 | 1 | r |= cmdargs_flagset_bit('a'); | |
| 549 | 1 | r |= cmdargs_flagset_bit('d'); | |
| 550 | 1 | r |= cmdargs_flagset_bit('0'); | |
| 551 | 1 | r |= cmdargs_flagset_bit('1'); | |
| 552 | 1 | r |= cmdargs_flagset_bit('8'); | |
| 553 | 1 | r |= cmdargs_flagset_bit('Z'); | |
| 554 | 1 | EXPECT_EQ(u64_popcount(r), 6); | |
| 555 | 1 | EXPECT_UINT_EQ(r, cmdargs_flagset_from_str("ad018Z")); | |
| 556 | 1 | } | |
| 557 | |||
| 558 | 1 | static void test_cmdargs_convert_flags_1(TestContext *ctx) | |
| 559 | { | ||
| 560 | 1 | static const FlagMapping map[] = { | |
| 561 | {'b', REPLACE_BASIC}, | ||
| 562 | {'c', REPLACE_CONFIRM}, | ||
| 563 | {'g', REPLACE_GLOBAL}, | ||
| 564 | {'i', REPLACE_IGNORE_CASE}, | ||
| 565 | }; | ||
| 566 | |||
| 567 | 1 | const CommandArgs a = {.flag_set = cmdargs_flagset_from_str("cg")}; | |
| 568 | 1 | ReplaceFlags flags = cmdargs_convert_flags(&a, map, ARRAYLEN(map)); | |
| 569 | 1 | EXPECT_EQ(flags, REPLACE_CONFIRM | REPLACE_GLOBAL); | |
| 570 | 1 | } | |
| 571 | |||
| 572 | 1 | static void test_cmdargs_convert_flags_2(TestContext *ctx) | |
| 573 | { | ||
| 574 | 1 | static const FlagMapping map[] = { | |
| 575 | {'C', EFLAG_CMD_HIST}, | ||
| 576 | {'S', EFLAG_SEARCH_HIST}, | ||
| 577 | {'F', EFLAG_FILE_HIST}, | ||
| 578 | {'H', EFLAG_ALL_HIST}, | ||
| 579 | }; | ||
| 580 | |||
| 581 | 1 | CommandArgs a = {.flag_set = cmdargs_flagset_bit('C')}; | |
| 582 | 1 | EditorFlags flags = cmdargs_convert_flags(&a, map, ARRAYLEN(map)); | |
| 583 | 1 | EXPECT_EQ(flags, EFLAG_CMD_HIST); | |
| 584 | |||
| 585 | 1 | a.flag_set |= cmdargs_flagset_bit('F'); | |
| 586 | 1 | flags = cmdargs_convert_flags(&a, map, ARRAYLEN(map)); | |
| 587 | 1 | EXPECT_EQ(flags, EFLAG_CMD_HIST | EFLAG_FILE_HIST); | |
| 588 | |||
| 589 | 1 | a.flag_set |= cmdargs_flagset_bit('S'); | |
| 590 | 1 | flags = cmdargs_convert_flags(&a, map, ARRAYLEN(map)); | |
| 591 | 1 | EXPECT_EQ(flags, EFLAG_ALL_HIST); | |
| 592 | |||
| 593 | 1 | a.flag_set = cmdargs_flagset_bit('H'); | |
| 594 | 1 | EXPECT_EQ(flags, cmdargs_convert_flags(&a, map, ARRAYLEN(map))); | |
| 595 | 1 | } | |
| 596 | |||
| 597 | 1 | static void test_add_alias(TestContext *ctx) | |
| 598 | { | ||
| 599 | 1 | const char name[] = "insert-foo"; | |
| 600 | 1 | const char cmd[] = "insert -m foo"; | |
| 601 | 1 | HashMap m = HASHMAP_INIT(HMAP_NO_FLAGS); | |
| 602 | 1 | add_alias(&m, name, cmd); | |
| 603 | 1 | EXPECT_STREQ(find_alias(&m, name), cmd); | |
| 604 | 1 | EXPECT_EQ(m.count, 1); | |
| 605 | |||
| 606 | 1 | remove_alias(&m, "insert-foo"); | |
| 607 | 1 | EXPECT_NULL(find_alias(&m, name)); | |
| 608 | 1 | EXPECT_EQ(m.count, 0); | |
| 609 | 1 | hashmap_free(&m, free); | |
| 610 | 1 | } | |
| 611 | |||
| 612 | static const TestEntry tests[] = { | ||
| 613 | TEST(test_parse_command_arg), | ||
| 614 | TEST(test_parse_commands), | ||
| 615 | TEST(test_find_normal_command), | ||
| 616 | TEST(test_parse_args), | ||
| 617 | TEST(test_cached_command_new), | ||
| 618 | TEST(test_string_append_escaped_arg), | ||
| 619 | TEST(test_command_struct_layout), | ||
| 620 | TEST(test_cmdargs_flagset_idx), | ||
| 621 | TEST(test_cmdargs_flagset_from_str), | ||
| 622 | TEST(test_cmdargs_convert_flags_1), | ||
| 623 | TEST(test_cmdargs_convert_flags_2), | ||
| 624 | TEST(test_add_alias), | ||
| 625 | }; | ||
| 626 | |||
| 627 | const TestGroup command_tests = TEST_GROUP(tests); | ||
| 628 |