blob: c49df4b494d8bd34527a4b0409652e4859f51f27 [file] [log] [blame]
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001/* vi:set ts=8 sts=4 sw=4 noet:
2 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9
10/*
11 * testing.c: Support for tests.
12 */
13
14#include "vim.h"
15
16#if defined(FEAT_EVAL) || defined(PROTO)
17
18/*
19 * Prepare "gap" for an assert error and add the sourcing position.
20 */
21 static void
22prepare_assert_error(garray_T *gap)
23{
Bram Moolenaar1a47ae32019-12-29 23:04:25 +010024 char buf[NUMBUFLEN];
Bram Moolenaar4f25b1a2020-09-10 19:25:05 +020025 char_u *sname = estack_sfile(ESTACK_NONE);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +020026
27 ga_init2(gap, 1, 100);
Bram Moolenaar1a47ae32019-12-29 23:04:25 +010028 if (sname != NULL)
Bram Moolenaarecaa70e2019-07-14 14:55:39 +020029 {
Bram Moolenaar1a47ae32019-12-29 23:04:25 +010030 ga_concat(gap, sname);
31 if (SOURCING_LNUM > 0)
Bram Moolenaarecaa70e2019-07-14 14:55:39 +020032 ga_concat(gap, (char_u *)" ");
33 }
Bram Moolenaar1a47ae32019-12-29 23:04:25 +010034 if (SOURCING_LNUM > 0)
Bram Moolenaarecaa70e2019-07-14 14:55:39 +020035 {
Bram Moolenaar1a47ae32019-12-29 23:04:25 +010036 sprintf(buf, "line %ld", (long)SOURCING_LNUM);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +020037 ga_concat(gap, (char_u *)buf);
38 }
Bram Moolenaar1a47ae32019-12-29 23:04:25 +010039 if (sname != NULL || SOURCING_LNUM > 0)
Bram Moolenaarecaa70e2019-07-14 14:55:39 +020040 ga_concat(gap, (char_u *)": ");
Bram Moolenaar1a47ae32019-12-29 23:04:25 +010041 vim_free(sname);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +020042}
43
44/*
45 * Append "p[clen]" to "gap", escaping unprintable characters.
46 * Changes NL to \n, CR to \r, etc.
47 */
48 static void
49ga_concat_esc(garray_T *gap, char_u *p, int clen)
50{
51 char_u buf[NUMBUFLEN];
52
53 if (clen > 1)
54 {
55 mch_memmove(buf, p, clen);
56 buf[clen] = NUL;
57 ga_concat(gap, buf);
58 }
59 else switch (*p)
60 {
61 case BS: ga_concat(gap, (char_u *)"\\b"); break;
62 case ESC: ga_concat(gap, (char_u *)"\\e"); break;
63 case FF: ga_concat(gap, (char_u *)"\\f"); break;
64 case NL: ga_concat(gap, (char_u *)"\\n"); break;
65 case TAB: ga_concat(gap, (char_u *)"\\t"); break;
66 case CAR: ga_concat(gap, (char_u *)"\\r"); break;
67 case '\\': ga_concat(gap, (char_u *)"\\\\"); break;
68 default:
Bram Moolenaar7177da92020-07-12 23:09:20 +020069 if (*p < ' ' || *p == 0x7f)
Bram Moolenaarecaa70e2019-07-14 14:55:39 +020070 {
71 vim_snprintf((char *)buf, NUMBUFLEN, "\\x%02x", *p);
72 ga_concat(gap, buf);
73 }
74 else
75 ga_append(gap, *p);
76 break;
77 }
78}
79
80/*
81 * Append "str" to "gap", escaping unprintable characters.
82 * Changes NL to \n, CR to \r, etc.
83 */
84 static void
85ga_concat_shorten_esc(garray_T *gap, char_u *str)
86{
87 char_u *p;
88 char_u *s;
89 int c;
90 int clen;
91 char_u buf[NUMBUFLEN];
92 int same_len;
93
94 if (str == NULL)
95 {
96 ga_concat(gap, (char_u *)"NULL");
97 return;
98 }
99
100 for (p = str; *p != NUL; ++p)
101 {
102 same_len = 1;
103 s = p;
Bram Moolenaar34f81172022-02-16 12:16:19 +0000104 c = mb_cptr2char_adv(&s);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200105 clen = s - p;
106 while (*s != NUL && c == mb_ptr2char(s))
107 {
108 ++same_len;
109 s += clen;
110 }
111 if (same_len > 20)
112 {
113 ga_concat(gap, (char_u *)"\\[");
114 ga_concat_esc(gap, p, clen);
115 ga_concat(gap, (char_u *)" occurs ");
116 vim_snprintf((char *)buf, NUMBUFLEN, "%d", same_len);
117 ga_concat(gap, buf);
118 ga_concat(gap, (char_u *)" times]");
119 p = s - 1;
120 }
121 else
122 ga_concat_esc(gap, p, clen);
123 }
124}
125
126/*
127 * Fill "gap" with information about an assert error.
128 */
129 static void
130fill_assert_error(
131 garray_T *gap,
132 typval_T *opt_msg_tv,
133 char_u *exp_str,
Bram Moolenaar4a021df2020-06-13 15:13:38 +0200134 typval_T *exp_tv_arg,
135 typval_T *got_tv_arg,
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200136 assert_type_T atype)
137{
138 char_u numbuf[NUMBUFLEN];
139 char_u *tofree;
Bram Moolenaar4a021df2020-06-13 15:13:38 +0200140 typval_T *exp_tv = exp_tv_arg;
141 typval_T *got_tv = got_tv_arg;
142 int did_copy = FALSE;
143 int omitted = 0;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200144
Bram Moolenaar1d634542020-08-18 13:41:50 +0200145 if (opt_msg_tv->v_type != VAR_UNKNOWN
146 && !(opt_msg_tv->v_type == VAR_STRING
147 && (opt_msg_tv->vval.v_string == NULL
148 || *opt_msg_tv->vval.v_string == NUL)))
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200149 {
150 ga_concat(gap, echo_string(opt_msg_tv, &tofree, numbuf, 0));
151 vim_free(tofree);
152 ga_concat(gap, (char_u *)": ");
153 }
154
155 if (atype == ASSERT_MATCH || atype == ASSERT_NOTMATCH)
156 ga_concat(gap, (char_u *)"Pattern ");
157 else if (atype == ASSERT_NOTEQUAL)
158 ga_concat(gap, (char_u *)"Expected not equal to ");
159 else
160 ga_concat(gap, (char_u *)"Expected ");
161 if (exp_str == NULL)
162 {
Bram Moolenaar4a021df2020-06-13 15:13:38 +0200163 // When comparing dictionaries, drop the items that are equal, so that
164 // it's a lot easier to see what differs.
165 if (atype != ASSERT_NOTEQUAL
166 && exp_tv->v_type == VAR_DICT && got_tv->v_type == VAR_DICT
167 && exp_tv->vval.v_dict != NULL && got_tv->vval.v_dict != NULL)
168 {
169 dict_T *exp_d = exp_tv->vval.v_dict;
170 dict_T *got_d = got_tv->vval.v_dict;
171 hashitem_T *hi;
172 dictitem_T *item2;
173 int todo;
174
175 did_copy = TRUE;
176 exp_tv->vval.v_dict = dict_alloc();
177 got_tv->vval.v_dict = dict_alloc();
178 if (exp_tv->vval.v_dict == NULL || got_tv->vval.v_dict == NULL)
179 return;
180
181 todo = (int)exp_d->dv_hashtab.ht_used;
182 for (hi = exp_d->dv_hashtab.ht_array; todo > 0; ++hi)
183 {
184 if (!HASHITEM_EMPTY(hi))
185 {
186 item2 = dict_find(got_d, hi->hi_key, -1);
187 if (item2 == NULL || !tv_equal(&HI2DI(hi)->di_tv,
188 &item2->di_tv, FALSE, FALSE))
189 {
190 // item of exp_d not present in got_d or values differ.
191 dict_add_tv(exp_tv->vval.v_dict,
192 (char *)hi->hi_key, &HI2DI(hi)->di_tv);
193 if (item2 != NULL)
194 dict_add_tv(got_tv->vval.v_dict,
195 (char *)hi->hi_key, &item2->di_tv);
196 }
197 else
198 ++omitted;
199 --todo;
200 }
201 }
202
203 // Add items only present in got_d.
204 todo = (int)got_d->dv_hashtab.ht_used;
205 for (hi = got_d->dv_hashtab.ht_array; todo > 0; ++hi)
206 {
207 if (!HASHITEM_EMPTY(hi))
208 {
209 item2 = dict_find(exp_d, hi->hi_key, -1);
210 if (item2 == NULL)
211 // item of got_d not present in exp_d
212 dict_add_tv(got_tv->vval.v_dict,
213 (char *)hi->hi_key, &HI2DI(hi)->di_tv);
214 --todo;
215 }
216 }
217 }
218
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200219 ga_concat_shorten_esc(gap, tv2string(exp_tv, &tofree, numbuf, 0));
220 vim_free(tofree);
221 }
222 else
Bram Moolenaar631e8f92020-11-04 15:07:16 +0100223 {
224 ga_concat(gap, (char_u *)"'");
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200225 ga_concat_shorten_esc(gap, exp_str);
Bram Moolenaar631e8f92020-11-04 15:07:16 +0100226 ga_concat(gap, (char_u *)"'");
227 }
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200228 if (atype != ASSERT_NOTEQUAL)
229 {
230 if (atype == ASSERT_MATCH)
231 ga_concat(gap, (char_u *)" does not match ");
232 else if (atype == ASSERT_NOTMATCH)
233 ga_concat(gap, (char_u *)" does match ");
234 else
235 ga_concat(gap, (char_u *)" but got ");
236 ga_concat_shorten_esc(gap, tv2string(got_tv, &tofree, numbuf, 0));
237 vim_free(tofree);
Bram Moolenaar4a021df2020-06-13 15:13:38 +0200238
239 if (omitted != 0)
240 {
241 char buf[100];
242
243 vim_snprintf(buf, 100, " - %d equal item%s omitted",
244 omitted, omitted == 1 ? "" : "s");
245 ga_concat(gap, (char_u *)buf);
246 }
247 }
248
249 if (did_copy)
250 {
251 clear_tv(exp_tv);
252 clear_tv(got_tv);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200253 }
254}
255
256 static int
257assert_equal_common(typval_T *argvars, assert_type_T atype)
258{
259 garray_T ga;
260
261 if (tv_equal(&argvars[0], &argvars[1], FALSE, FALSE)
262 != (atype == ASSERT_EQUAL))
263 {
264 prepare_assert_error(&ga);
265 fill_assert_error(&ga, &argvars[2], NULL, &argvars[0], &argvars[1],
266 atype);
267 assert_error(&ga);
268 ga_clear(&ga);
269 return 1;
270 }
271 return 0;
272}
273
274 static int
275assert_match_common(typval_T *argvars, assert_type_T atype)
276{
277 garray_T ga;
278 char_u buf1[NUMBUFLEN];
279 char_u buf2[NUMBUFLEN];
Bram Moolenaar7177da92020-07-12 23:09:20 +0200280 int called_emsg_before = called_emsg;
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200281 char_u *pat;
282 char_u *text;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200283
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200284 if (in_vim9script()
285 && (check_for_string_arg(argvars, 0) == FAIL
286 || check_for_string_arg(argvars, 1) == FAIL
287 || check_for_opt_string_arg(argvars, 2) == FAIL))
288 return 1;
289
290 pat = tv_get_string_buf_chk(&argvars[0], buf1);
291 text = tv_get_string_buf_chk(&argvars[1], buf2);
Bram Moolenaar7177da92020-07-12 23:09:20 +0200292 if (called_emsg == called_emsg_before
293 && pattern_match(pat, text, FALSE) != (atype == ASSERT_MATCH))
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200294 {
295 prepare_assert_error(&ga);
296 fill_assert_error(&ga, &argvars[2], NULL, &argvars[0], &argvars[1],
297 atype);
298 assert_error(&ga);
299 ga_clear(&ga);
300 return 1;
301 }
302 return 0;
303}
304
305/*
306 * Common for assert_true() and assert_false().
307 * Return non-zero for failure.
308 */
309 static int
310assert_bool(typval_T *argvars, int isTrue)
311{
312 int error = FALSE;
313 garray_T ga;
314
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +0100315 if (argvars[0].v_type == VAR_BOOL
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200316 && argvars[0].vval.v_number == (isTrue ? VVAL_TRUE : VVAL_FALSE))
317 return 0;
318 if (argvars[0].v_type != VAR_NUMBER
319 || (tv_get_number_chk(&argvars[0], &error) == 0) == isTrue
320 || error)
321 {
322 prepare_assert_error(&ga);
323 fill_assert_error(&ga, &argvars[1],
324 (char_u *)(isTrue ? "True" : "False"),
325 NULL, &argvars[0], ASSERT_OTHER);
326 assert_error(&ga);
327 ga_clear(&ga);
328 return 1;
329 }
330 return 0;
331}
332
333 static void
334assert_append_cmd_or_arg(garray_T *gap, typval_T *argvars, char_u *cmd)
335{
336 char_u *tofree;
337 char_u numbuf[NUMBUFLEN];
338
339 if (argvars[1].v_type != VAR_UNKNOWN && argvars[2].v_type != VAR_UNKNOWN)
340 {
341 ga_concat(gap, echo_string(&argvars[2], &tofree, numbuf, 0));
342 vim_free(tofree);
343 }
344 else
345 ga_concat(gap, cmd);
346}
347
348 static int
Bram Moolenaar5b8cabf2021-04-02 18:55:57 +0200349assert_beeps(typval_T *argvars, int no_beep)
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200350{
Yegappan Lakshmanan1a71d312021-07-15 12:49:58 +0200351 char_u *cmd;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200352 garray_T ga;
353 int ret = 0;
354
Yegappan Lakshmanan1a71d312021-07-15 12:49:58 +0200355 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
356 return 0;
357
358 cmd = tv_get_string_chk(&argvars[0]);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200359 called_vim_beep = FALSE;
360 suppress_errthrow = TRUE;
361 emsg_silent = FALSE;
362 do_cmdline_cmd(cmd);
Bram Moolenaar5b8cabf2021-04-02 18:55:57 +0200363 if (no_beep ? called_vim_beep : !called_vim_beep)
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200364 {
365 prepare_assert_error(&ga);
Bram Moolenaar5b8cabf2021-04-02 18:55:57 +0200366 if (no_beep)
367 ga_concat(&ga, (char_u *)"command did beep: ");
368 else
369 ga_concat(&ga, (char_u *)"command did not beep: ");
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200370 ga_concat(&ga, cmd);
371 assert_error(&ga);
372 ga_clear(&ga);
373 ret = 1;
374 }
375
376 suppress_errthrow = FALSE;
377 emsg_on_display = FALSE;
378 return ret;
379}
380
381/*
Yegappan Lakshmanan1a71d312021-07-15 12:49:58 +0200382 * "assert_beeps(cmd)" function
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200383 */
384 void
385f_assert_beeps(typval_T *argvars, typval_T *rettv)
386{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200387 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
388 return;
389
Bram Moolenaar5b8cabf2021-04-02 18:55:57 +0200390 rettv->vval.v_number = assert_beeps(argvars, FALSE);
391}
392
393/*
Yegappan Lakshmanan1a71d312021-07-15 12:49:58 +0200394 * "assert_nobeep(cmd)" function
Bram Moolenaar5b8cabf2021-04-02 18:55:57 +0200395 */
396 void
397f_assert_nobeep(typval_T *argvars, typval_T *rettv)
398{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200399 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
400 return;
401
Bram Moolenaar5b8cabf2021-04-02 18:55:57 +0200402 rettv->vval.v_number = assert_beeps(argvars, TRUE);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200403}
404
405/*
406 * "assert_equal(expected, actual[, msg])" function
407 */
408 void
409f_assert_equal(typval_T *argvars, typval_T *rettv)
410{
411 rettv->vval.v_number = assert_equal_common(argvars, ASSERT_EQUAL);
412}
413
414 static int
415assert_equalfile(typval_T *argvars)
416{
417 char_u buf1[NUMBUFLEN];
418 char_u buf2[NUMBUFLEN];
Bram Moolenaar7177da92020-07-12 23:09:20 +0200419 int called_emsg_before = called_emsg;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200420 char_u *fname1 = tv_get_string_buf_chk(&argvars[0], buf1);
421 char_u *fname2 = tv_get_string_buf_chk(&argvars[1], buf2);
422 garray_T ga;
423 FILE *fd1;
424 FILE *fd2;
Bram Moolenaar30cc44a2020-06-04 16:52:40 +0200425 char line1[200];
426 char line2[200];
427 int lineidx = 0;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200428
Bram Moolenaar7177da92020-07-12 23:09:20 +0200429 if (called_emsg > called_emsg_before)
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200430 return 0;
431
432 IObuff[0] = NUL;
433 fd1 = mch_fopen((char *)fname1, READBIN);
434 if (fd1 == NULL)
435 {
Bram Moolenaar460ae5d2022-01-01 14:19:49 +0000436 vim_snprintf((char *)IObuff, IOSIZE, (char *)e_cant_read_file_str, fname1);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200437 }
438 else
439 {
440 fd2 = mch_fopen((char *)fname2, READBIN);
441 if (fd2 == NULL)
442 {
443 fclose(fd1);
Bram Moolenaar460ae5d2022-01-01 14:19:49 +0000444 vim_snprintf((char *)IObuff, IOSIZE, (char *)e_cant_read_file_str, fname2);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200445 }
446 else
447 {
Bram Moolenaar30cc44a2020-06-04 16:52:40 +0200448 int c1, c2;
449 long count = 0;
450 long linecount = 1;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200451
452 for (;;)
453 {
454 c1 = fgetc(fd1);
455 c2 = fgetc(fd2);
456 if (c1 == EOF)
457 {
458 if (c2 != EOF)
459 STRCPY(IObuff, "first file is shorter");
460 break;
461 }
462 else if (c2 == EOF)
463 {
464 STRCPY(IObuff, "second file is shorter");
465 break;
466 }
Bram Moolenaar30cc44a2020-06-04 16:52:40 +0200467 else
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200468 {
Bram Moolenaar30cc44a2020-06-04 16:52:40 +0200469 line1[lineidx] = c1;
470 line2[lineidx] = c2;
471 ++lineidx;
472 if (c1 != c2)
473 {
474 vim_snprintf((char *)IObuff, IOSIZE,
475 "difference at byte %ld, line %ld",
476 count, linecount);
477 break;
478 }
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200479 }
480 ++count;
Bram Moolenaar30cc44a2020-06-04 16:52:40 +0200481 if (c1 == NL)
482 {
483 ++linecount;
484 lineidx = 0;
485 }
486 else if (lineidx + 2 == (int)sizeof(line1))
487 {
488 mch_memmove(line1, line1 + 100, lineidx - 100);
489 mch_memmove(line2, line2 + 100, lineidx - 100);
490 lineidx -= 100;
491 }
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200492 }
493 fclose(fd1);
494 fclose(fd2);
495 }
496 }
497 if (IObuff[0] != NUL)
498 {
499 prepare_assert_error(&ga);
Bram Moolenaarfb517ba2020-06-03 19:55:35 +0200500 if (argvars[2].v_type != VAR_UNKNOWN)
501 {
502 char_u numbuf[NUMBUFLEN];
503 char_u *tofree;
504
505 ga_concat(&ga, echo_string(&argvars[2], &tofree, numbuf, 0));
506 vim_free(tofree);
507 ga_concat(&ga, (char_u *)": ");
508 }
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200509 ga_concat(&ga, IObuff);
Bram Moolenaar30cc44a2020-06-04 16:52:40 +0200510 if (lineidx > 0)
511 {
512 line1[lineidx] = NUL;
513 line2[lineidx] = NUL;
514 ga_concat(&ga, (char_u *)" after \"");
515 ga_concat(&ga, (char_u *)line1);
516 if (STRCMP(line1, line2) != 0)
517 {
518 ga_concat(&ga, (char_u *)"\" vs \"");
519 ga_concat(&ga, (char_u *)line2);
520 }
521 ga_concat(&ga, (char_u *)"\"");
522 }
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200523 assert_error(&ga);
524 ga_clear(&ga);
525 return 1;
526 }
527 return 0;
528}
529
530/*
Bram Moolenaarfb517ba2020-06-03 19:55:35 +0200531 * "assert_equalfile(fname-one, fname-two[, msg])" function
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200532 */
533 void
534f_assert_equalfile(typval_T *argvars, typval_T *rettv)
535{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200536 if (in_vim9script()
537 && (check_for_string_arg(argvars, 0) == FAIL
538 || check_for_string_arg(argvars, 1) == FAIL
539 || check_for_opt_string_arg(argvars, 2) == FAIL))
540 return;
541
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200542 rettv->vval.v_number = assert_equalfile(argvars);
543}
544
545/*
546 * "assert_notequal(expected, actual[, msg])" function
547 */
548 void
549f_assert_notequal(typval_T *argvars, typval_T *rettv)
550{
551 rettv->vval.v_number = assert_equal_common(argvars, ASSERT_NOTEQUAL);
552}
553
554/*
555 * "assert_exception(string[, msg])" function
556 */
557 void
558f_assert_exception(typval_T *argvars, typval_T *rettv)
559{
560 garray_T ga;
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200561 char_u *error;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200562
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200563 if (in_vim9script()
564 && (check_for_string_arg(argvars, 0) == FAIL
565 || check_for_opt_string_arg(argvars, 1) == FAIL))
566 return;
567
568 error = tv_get_string_chk(&argvars[0]);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200569 if (*get_vim_var_str(VV_EXCEPTION) == NUL)
570 {
571 prepare_assert_error(&ga);
572 ga_concat(&ga, (char_u *)"v:exception is not set");
573 assert_error(&ga);
574 ga_clear(&ga);
575 rettv->vval.v_number = 1;
576 }
577 else if (error != NULL
578 && strstr((char *)get_vim_var_str(VV_EXCEPTION), (char *)error) == NULL)
579 {
580 prepare_assert_error(&ga);
581 fill_assert_error(&ga, &argvars[1], NULL, &argvars[0],
582 get_vim_var_tv(VV_EXCEPTION), ASSERT_OTHER);
583 assert_error(&ga);
584 ga_clear(&ga);
585 rettv->vval.v_number = 1;
586 }
587}
588
589/*
590 * "assert_fails(cmd [, error[, msg]])" function
591 */
592 void
593f_assert_fails(typval_T *argvars, typval_T *rettv)
594{
Yegappan Lakshmanana764e732021-07-25 15:57:32 +0200595 char_u *cmd;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200596 garray_T ga;
597 int save_trylevel = trylevel;
Bram Moolenaar53989552019-12-23 22:59:18 +0100598 int called_emsg_before = called_emsg;
Bram Moolenaar44d66522020-09-06 22:26:57 +0200599 char *wrong_arg_msg = NULL;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200600
Yegappan Lakshmanana764e732021-07-25 15:57:32 +0200601 if (check_for_string_or_number_arg(argvars, 0) == FAIL
602 || check_for_opt_string_or_list_arg(argvars, 1) == FAIL
603 || (argvars[1].v_type != VAR_UNKNOWN
604 && (argvars[2].v_type != VAR_UNKNOWN
605 && (check_for_opt_number_arg(argvars, 3) == FAIL
606 || (argvars[3].v_type != VAR_UNKNOWN
607 && check_for_opt_string_arg(argvars, 4) == FAIL)))))
608 return;
609
610 cmd = tv_get_string_chk(&argvars[0]);
611
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200612 // trylevel must be zero for a ":throw" command to be considered failed
613 trylevel = 0;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200614 suppress_errthrow = TRUE;
Bram Moolenaar28ee8922020-10-28 20:20:00 +0100615 in_assert_fails = TRUE;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200616
617 do_cmdline_cmd(cmd);
Bram Moolenaar53989552019-12-23 22:59:18 +0100618 if (called_emsg == called_emsg_before)
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200619 {
620 prepare_assert_error(&ga);
621 ga_concat(&ga, (char_u *)"command did not fail: ");
622 assert_append_cmd_or_arg(&ga, argvars, cmd);
623 assert_error(&ga);
624 ga_clear(&ga);
625 rettv->vval.v_number = 1;
626 }
627 else if (argvars[1].v_type != VAR_UNKNOWN)
628 {
629 char_u buf[NUMBUFLEN];
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200630 char_u *expected;
Bram Moolenaar631e8f92020-11-04 15:07:16 +0100631 char_u *expected_str = NULL;
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200632 int error_found = FALSE;
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200633 int error_found_index = 1;
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200634 char_u *actual = emsg_assert_fails_msg == NULL ? (char_u *)"[unknown]"
635 : emsg_assert_fails_msg;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200636
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200637 if (argvars[1].v_type == VAR_STRING)
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200638 {
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200639 expected = tv_get_string_buf_chk(&argvars[1], buf);
640 error_found = expected == NULL
641 || strstr((char *)actual, (char *)expected) == NULL;
642 }
643 else if (argvars[1].v_type == VAR_LIST)
644 {
645 list_T *list = argvars[1].vval.v_list;
646 typval_T *tv;
647
648 if (list == NULL || list->lv_len < 1 || list->lv_len > 2)
649 {
Bram Moolenaar44d66522020-09-06 22:26:57 +0200650 wrong_arg_msg = e_assert_fails_second_arg;
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200651 goto theend;
652 }
653 CHECK_LIST_MATERIALIZE(list);
654 tv = &list->lv_first->li_tv;
655 expected = tv_get_string_buf_chk(tv, buf);
656 if (!pattern_match(expected, actual, FALSE))
657 {
658 error_found = TRUE;
Bram Moolenaar631e8f92020-11-04 15:07:16 +0100659 expected_str = expected;
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200660 }
661 else if (list->lv_len == 2)
662 {
663 tv = &list->lv_u.mat.lv_last->li_tv;
664 actual = get_vim_var_str(VV_ERRMSG);
665 expected = tv_get_string_buf_chk(tv, buf);
666 if (!pattern_match(expected, actual, FALSE))
Bram Moolenaar631e8f92020-11-04 15:07:16 +0100667 {
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200668 error_found = TRUE;
Bram Moolenaar631e8f92020-11-04 15:07:16 +0100669 expected_str = expected;
670 }
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200671 }
672 }
673 else
674 {
Bram Moolenaar44d66522020-09-06 22:26:57 +0200675 wrong_arg_msg = e_assert_fails_second_arg;
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200676 goto theend;
677 }
678
Bram Moolenaar9b02d642020-08-18 23:24:13 +0200679 if (!error_found && argvars[2].v_type != VAR_UNKNOWN
Bram Moolenaar44d66522020-09-06 22:26:57 +0200680 && argvars[3].v_type != VAR_UNKNOWN)
Bram Moolenaar1d634542020-08-18 13:41:50 +0200681 {
Bram Moolenaar44d66522020-09-06 22:26:57 +0200682 if (argvars[3].v_type != VAR_NUMBER)
683 {
684 wrong_arg_msg = e_assert_fails_fourth_argument;
685 goto theend;
686 }
687 else if (argvars[3].vval.v_number >= 0
688 && argvars[3].vval.v_number != emsg_assert_fails_lnum)
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200689 {
690 error_found = TRUE;
691 error_found_index = 3;
692 }
Bram Moolenaar44d66522020-09-06 22:26:57 +0200693 if (!error_found && argvars[4].v_type != VAR_UNKNOWN)
694 {
695 if (argvars[4].v_type != VAR_STRING)
696 {
697 wrong_arg_msg = e_assert_fails_fifth_argument;
698 goto theend;
699 }
700 else if (argvars[4].vval.v_string != NULL
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200701 && !pattern_match(argvars[4].vval.v_string,
702 emsg_assert_fails_context, FALSE))
Bram Moolenaar44d66522020-09-06 22:26:57 +0200703 {
704 error_found = TRUE;
705 error_found_index = 4;
706 }
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200707 }
Bram Moolenaar1d634542020-08-18 13:41:50 +0200708 }
709
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200710 if (error_found)
711 {
712 typval_T actual_tv;
713
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200714 prepare_assert_error(&ga);
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200715 if (error_found_index == 3)
Bram Moolenaar1d634542020-08-18 13:41:50 +0200716 {
717 actual_tv.v_type = VAR_NUMBER;
718 actual_tv.vval.v_number = emsg_assert_fails_lnum;
719 }
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200720 else if (error_found_index == 4)
721 {
722 actual_tv.v_type = VAR_STRING;
723 actual_tv.vval.v_string = emsg_assert_fails_context;
724 }
Bram Moolenaar1d634542020-08-18 13:41:50 +0200725 else
726 {
727 actual_tv.v_type = VAR_STRING;
728 actual_tv.vval.v_string = actual;
729 }
Bram Moolenaar631e8f92020-11-04 15:07:16 +0100730 fill_assert_error(&ga, &argvars[2], expected_str,
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200731 &argvars[error_found_index], &actual_tv, ASSERT_OTHER);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200732 ga_concat(&ga, (char_u *)": ");
733 assert_append_cmd_or_arg(&ga, argvars, cmd);
734 assert_error(&ga);
735 ga_clear(&ga);
736 rettv->vval.v_number = 1;
737 }
738 }
739
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200740theend:
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200741 trylevel = save_trylevel;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200742 suppress_errthrow = FALSE;
Bram Moolenaar28ee8922020-10-28 20:20:00 +0100743 in_assert_fails = FALSE;
744 did_emsg = FALSE;
Bram Moolenaar8bea1712022-06-15 20:49:35 +0100745 got_int = FALSE;
Bram Moolenaar28ee8922020-10-28 20:20:00 +0100746 msg_col = 0;
747 need_wait_return = FALSE;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200748 emsg_on_display = FALSE;
Bram Moolenaar28ee8922020-10-28 20:20:00 +0100749 msg_scrolled = 0;
750 lines_left = Rows;
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200751 VIM_CLEAR(emsg_assert_fails_msg);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200752 set_vim_var_string(VV_ERRMSG, NULL, 0);
Bram Moolenaar44d66522020-09-06 22:26:57 +0200753 if (wrong_arg_msg != NULL)
754 emsg(_(wrong_arg_msg));
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200755}
756
757/*
758 * "assert_false(actual[, msg])" function
759 */
760 void
761f_assert_false(typval_T *argvars, typval_T *rettv)
762{
763 rettv->vval.v_number = assert_bool(argvars, FALSE);
764}
765
766 static int
767assert_inrange(typval_T *argvars)
768{
769 garray_T ga;
770 int error = FALSE;
771 char_u *tofree;
772 char msg[200];
773 char_u numbuf[NUMBUFLEN];
774
775#ifdef FEAT_FLOAT
776 if (argvars[0].v_type == VAR_FLOAT
777 || argvars[1].v_type == VAR_FLOAT
778 || argvars[2].v_type == VAR_FLOAT)
779 {
780 float_T flower = tv_get_float(&argvars[0]);
781 float_T fupper = tv_get_float(&argvars[1]);
782 float_T factual = tv_get_float(&argvars[2]);
783
784 if (factual < flower || factual > fupper)
785 {
786 prepare_assert_error(&ga);
787 if (argvars[3].v_type != VAR_UNKNOWN)
788 {
789 ga_concat(&ga, tv2string(&argvars[3], &tofree, numbuf, 0));
790 vim_free(tofree);
791 }
792 else
793 {
794 vim_snprintf(msg, 200, "Expected range %g - %g, but got %g",
795 flower, fupper, factual);
796 ga_concat(&ga, (char_u *)msg);
797 }
798 assert_error(&ga);
799 ga_clear(&ga);
800 return 1;
801 }
802 }
803 else
804#endif
805 {
806 varnumber_T lower = tv_get_number_chk(&argvars[0], &error);
807 varnumber_T upper = tv_get_number_chk(&argvars[1], &error);
808 varnumber_T actual = tv_get_number_chk(&argvars[2], &error);
809
810 if (error)
811 return 0;
812 if (actual < lower || actual > upper)
813 {
814 prepare_assert_error(&ga);
815 if (argvars[3].v_type != VAR_UNKNOWN)
816 {
817 ga_concat(&ga, tv2string(&argvars[3], &tofree, numbuf, 0));
818 vim_free(tofree);
819 }
820 else
821 {
822 vim_snprintf(msg, 200, "Expected range %ld - %ld, but got %ld",
823 (long)lower, (long)upper, (long)actual);
824 ga_concat(&ga, (char_u *)msg);
825 }
826 assert_error(&ga);
827 ga_clear(&ga);
828 return 1;
829 }
830 }
831 return 0;
832}
833
834/*
835 * "assert_inrange(lower, upper[, msg])" function
836 */
837 void
838f_assert_inrange(typval_T *argvars, typval_T *rettv)
839{
Yegappan Lakshmanana764e732021-07-25 15:57:32 +0200840 if (check_for_float_or_nr_arg(argvars, 0) == FAIL
841 || check_for_float_or_nr_arg(argvars, 1) == FAIL
842 || check_for_float_or_nr_arg(argvars, 2) == FAIL
843 || check_for_opt_string_arg(argvars, 3) == FAIL)
844 return;
845
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200846 rettv->vval.v_number = assert_inrange(argvars);
847}
848
849/*
850 * "assert_match(pattern, actual[, msg])" function
851 */
852 void
853f_assert_match(typval_T *argvars, typval_T *rettv)
854{
855 rettv->vval.v_number = assert_match_common(argvars, ASSERT_MATCH);
856}
857
858/*
859 * "assert_notmatch(pattern, actual[, msg])" function
860 */
861 void
862f_assert_notmatch(typval_T *argvars, typval_T *rettv)
863{
864 rettv->vval.v_number = assert_match_common(argvars, ASSERT_NOTMATCH);
865}
866
867/*
868 * "assert_report(msg)" function
869 */
870 void
871f_assert_report(typval_T *argvars, typval_T *rettv)
872{
873 garray_T ga;
874
Yegappan Lakshmananc72bdd22021-07-11 19:44:18 +0200875 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
876 return;
877
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200878 prepare_assert_error(&ga);
879 ga_concat(&ga, tv_get_string(&argvars[0]));
880 assert_error(&ga);
881 ga_clear(&ga);
882 rettv->vval.v_number = 1;
883}
884
885/*
886 * "assert_true(actual[, msg])" function
887 */
888 void
889f_assert_true(typval_T *argvars, typval_T *rettv)
890{
891 rettv->vval.v_number = assert_bool(argvars, TRUE);
892}
893
894/*
895 * "test_alloc_fail(id, countdown, repeat)" function
896 */
897 void
898f_test_alloc_fail(typval_T *argvars, typval_T *rettv UNUSED)
899{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200900 if (in_vim9script()
901 && (check_for_number_arg(argvars, 0) == FAIL
902 || check_for_number_arg(argvars, 1) == FAIL
903 || check_for_number_arg(argvars, 2) == FAIL))
904 return;
905
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200906 if (argvars[0].v_type != VAR_NUMBER
907 || argvars[0].vval.v_number <= 0
908 || argvars[1].v_type != VAR_NUMBER
909 || argvars[1].vval.v_number < 0
910 || argvars[2].v_type != VAR_NUMBER)
Bram Moolenaar436b5ad2021-12-31 22:49:24 +0000911 emsg(_(e_invalid_argument));
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200912 else
913 {
914 alloc_fail_id = argvars[0].vval.v_number;
915 if (alloc_fail_id >= aid_last)
Bram Moolenaar436b5ad2021-12-31 22:49:24 +0000916 emsg(_(e_invalid_argument));
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200917 alloc_fail_countdown = argvars[1].vval.v_number;
918 alloc_fail_repeat = argvars[2].vval.v_number;
919 did_outofmem_msg = FALSE;
920 }
921}
922
923/*
924 * "test_autochdir()"
925 */
926 void
927f_test_autochdir(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
928{
929#if defined(FEAT_AUTOCHDIR)
930 test_autochdir = TRUE;
931#endif
932}
933
934/*
935 * "test_feedinput()"
936 */
937 void
938f_test_feedinput(typval_T *argvars, typval_T *rettv UNUSED)
939{
940#ifdef USE_INPUT_BUF
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200941 char_u *val;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200942
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200943 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
944 return;
945
946 val = tv_get_string_chk(&argvars[0]);
Bram Moolenaar272ca952020-01-28 20:49:11 +0100947# ifdef VIMDLL
948 // this doesn't work in the console
949 if (!gui.in_use)
950 return;
951# endif
952
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200953 if (val != NULL)
954 {
955 trash_input_buf();
956 add_to_input_buf_csi(val, (int)STRLEN(val));
957 }
958#endif
959}
960
961/*
962 * "test_getvalue({name})" function
963 */
964 void
965f_test_getvalue(typval_T *argvars, typval_T *rettv)
966{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200967 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
968 return;
969
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200970 if (argvars[0].v_type != VAR_STRING)
Bram Moolenaar436b5ad2021-12-31 22:49:24 +0000971 emsg(_(e_invalid_argument));
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200972 else
973 {
974 char_u *name = tv_get_string(&argvars[0]);
975
976 if (STRCMP(name, (char_u *)"need_fileinfo") == 0)
977 rettv->vval.v_number = need_fileinfo;
978 else
Bram Moolenaar436b5ad2021-12-31 22:49:24 +0000979 semsg(_(e_invalid_argument_str), name);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200980 }
981}
982
983/*
984 * "test_option_not_set({name})" function
985 */
986 void
987f_test_option_not_set(typval_T *argvars, typval_T *rettv UNUSED)
988{
989 char_u *name = (char_u *)"";
990
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200991 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
992 return;
993
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200994 if (argvars[0].v_type != VAR_STRING)
Bram Moolenaar436b5ad2021-12-31 22:49:24 +0000995 emsg(_(e_invalid_argument));
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200996 else
997 {
998 name = tv_get_string(&argvars[0]);
999 if (reset_option_was_set(name) == FAIL)
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00001000 semsg(_(e_invalid_argument_str), name);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001001 }
1002}
1003
1004/*
1005 * "test_override({name}, {val})" function
1006 */
1007 void
1008f_test_override(typval_T *argvars, typval_T *rettv UNUSED)
1009{
1010 char_u *name = (char_u *)"";
1011 int val;
1012 static int save_starting = -1;
1013
Yegappan Lakshmanan1a71d312021-07-15 12:49:58 +02001014 if (in_vim9script()
1015 && (check_for_string_arg(argvars, 0) == FAIL
1016 || check_for_number_arg(argvars, 1) == FAIL))
1017 return;
1018
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001019 if (argvars[0].v_type != VAR_STRING
1020 || (argvars[1].v_type) != VAR_NUMBER)
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00001021 emsg(_(e_invalid_argument));
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001022 else
1023 {
1024 name = tv_get_string(&argvars[0]);
1025 val = (int)tv_get_number(&argvars[1]);
1026
1027 if (STRCMP(name, (char_u *)"redraw") == 0)
1028 disable_redraw_for_testing = val;
1029 else if (STRCMP(name, (char_u *)"redraw_flag") == 0)
1030 ignore_redraw_flag_for_testing = val;
1031 else if (STRCMP(name, (char_u *)"char_avail") == 0)
1032 disable_char_avail_for_testing = val;
1033 else if (STRCMP(name, (char_u *)"starting") == 0)
1034 {
1035 if (val)
1036 {
1037 if (save_starting < 0)
1038 save_starting = starting;
1039 starting = 0;
1040 }
1041 else
1042 {
1043 starting = save_starting;
1044 save_starting = -1;
1045 }
1046 }
1047 else if (STRCMP(name, (char_u *)"nfa_fail") == 0)
1048 nfa_fail_for_testing = val;
1049 else if (STRCMP(name, (char_u *)"no_query_mouse") == 0)
1050 no_query_mouse_for_testing = val;
1051 else if (STRCMP(name, (char_u *)"no_wait_return") == 0)
1052 no_wait_return = val;
Bram Moolenaarb340bae2020-06-15 19:51:56 +02001053 else if (STRCMP(name, (char_u *)"ui_delay") == 0)
1054 ui_delay_for_testing = val;
Bram Moolenaar0c0eddd2020-06-13 15:47:25 +02001055 else if (STRCMP(name, (char_u *)"term_props") == 0)
1056 reset_term_props_on_termresponse = val;
ichizokae1bd872022-01-20 14:57:29 +00001057 else if (STRCMP(name, (char_u *)"vterm_title") == 0)
1058 disable_vterm_title_for_testing = val;
Bram Moolenaarf52f0602021-03-10 21:26:37 +01001059 else if (STRCMP(name, (char_u *)"uptime") == 0)
1060 override_sysinfo_uptime = val;
Bram Moolenaar3e4fa3d2022-01-13 22:05:09 +00001061 else if (STRCMP(name, (char_u *)"autoload") == 0)
1062 override_autoload = val;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001063 else if (STRCMP(name, (char_u *)"ALL") == 0)
1064 {
1065 disable_char_avail_for_testing = FALSE;
1066 disable_redraw_for_testing = FALSE;
1067 ignore_redraw_flag_for_testing = FALSE;
1068 nfa_fail_for_testing = FALSE;
1069 no_query_mouse_for_testing = FALSE;
Bram Moolenaarb340bae2020-06-15 19:51:56 +02001070 ui_delay_for_testing = 0;
Bram Moolenaar0c0eddd2020-06-13 15:47:25 +02001071 reset_term_props_on_termresponse = FALSE;
Bram Moolenaarf52f0602021-03-10 21:26:37 +01001072 override_sysinfo_uptime = -1;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001073 if (save_starting >= 0)
1074 {
1075 starting = save_starting;
1076 save_starting = -1;
1077 }
1078 }
1079 else
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00001080 semsg(_(e_invalid_argument_str), name);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001081 }
1082}
1083
1084/*
1085 * "test_refcount({expr})" function
1086 */
1087 void
1088f_test_refcount(typval_T *argvars, typval_T *rettv)
1089{
1090 int retval = -1;
1091
1092 switch (argvars[0].v_type)
1093 {
1094 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02001095 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001096 case VAR_VOID:
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001097 case VAR_NUMBER:
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01001098 case VAR_BOOL:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001099 case VAR_FLOAT:
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001100 case VAR_SPECIAL:
1101 case VAR_STRING:
Bram Moolenaarf18332f2021-05-07 17:55:55 +02001102 case VAR_INSTR:
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001103 break;
1104 case VAR_JOB:
1105#ifdef FEAT_JOB_CHANNEL
1106 if (argvars[0].vval.v_job != NULL)
1107 retval = argvars[0].vval.v_job->jv_refcount - 1;
1108#endif
1109 break;
1110 case VAR_CHANNEL:
1111#ifdef FEAT_JOB_CHANNEL
1112 if (argvars[0].vval.v_channel != NULL)
1113 retval = argvars[0].vval.v_channel->ch_refcount - 1;
1114#endif
1115 break;
1116 case VAR_FUNC:
1117 if (argvars[0].vval.v_string != NULL)
1118 {
1119 ufunc_T *fp;
1120
Bram Moolenaard9d2fd02022-01-13 21:15:21 +00001121 fp = find_func(argvars[0].vval.v_string, FALSE);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001122 if (fp != NULL)
1123 retval = fp->uf_refcount;
1124 }
1125 break;
1126 case VAR_PARTIAL:
1127 if (argvars[0].vval.v_partial != NULL)
1128 retval = argvars[0].vval.v_partial->pt_refcount - 1;
1129 break;
1130 case VAR_BLOB:
1131 if (argvars[0].vval.v_blob != NULL)
1132 retval = argvars[0].vval.v_blob->bv_refcount - 1;
1133 break;
1134 case VAR_LIST:
1135 if (argvars[0].vval.v_list != NULL)
1136 retval = argvars[0].vval.v_list->lv_refcount - 1;
1137 break;
1138 case VAR_DICT:
1139 if (argvars[0].vval.v_dict != NULL)
1140 retval = argvars[0].vval.v_dict->dv_refcount - 1;
1141 break;
1142 }
1143
1144 rettv->v_type = VAR_NUMBER;
1145 rettv->vval.v_number = retval;
1146
1147}
1148
1149/*
1150 * "test_garbagecollect_now()" function
1151 */
1152 void
1153f_test_garbagecollect_now(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
1154{
Bram Moolenaar0d6f5d92019-12-05 21:33:15 +01001155 // This is dangerous, any Lists and Dicts used internally may be freed
1156 // while still in use.
Bram Moolenaarb3d83982022-01-27 19:59:47 +00001157 if (!get_vim_var_nr(VV_TESTING))
1158 emsg(_(e_calling_test_garbagecollect_now_while_v_testing_is_not_set));
1159 else
1160 garbage_collect(TRUE);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001161}
1162
1163/*
1164 * "test_garbagecollect_soon()" function
1165 */
1166 void
1167f_test_garbagecollect_soon(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
1168{
1169 may_garbage_collect = TRUE;
1170}
1171
1172/*
1173 * "test_ignore_error()" function
1174 */
1175 void
1176f_test_ignore_error(typval_T *argvars, typval_T *rettv UNUSED)
1177{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001178 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
1179 return;
1180
Yegappan Lakshmanan5b739922021-07-10 13:15:41 +02001181 if (argvars[0].v_type != VAR_STRING)
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00001182 emsg(_(e_invalid_argument));
Yegappan Lakshmanan5b739922021-07-10 13:15:41 +02001183 else
1184 ignore_error_for_testing(tv_get_string(&argvars[0]));
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001185}
1186
1187 void
1188f_test_null_blob(typval_T *argvars UNUSED, typval_T *rettv)
1189{
1190 rettv->v_type = VAR_BLOB;
1191 rettv->vval.v_blob = NULL;
1192}
1193
1194#ifdef FEAT_JOB_CHANNEL
1195 void
1196f_test_null_channel(typval_T *argvars UNUSED, typval_T *rettv)
1197{
1198 rettv->v_type = VAR_CHANNEL;
1199 rettv->vval.v_channel = NULL;
1200}
1201#endif
1202
1203 void
1204f_test_null_dict(typval_T *argvars UNUSED, typval_T *rettv)
1205{
1206 rettv_dict_set(rettv, NULL);
1207}
1208
1209#ifdef FEAT_JOB_CHANNEL
1210 void
1211f_test_null_job(typval_T *argvars UNUSED, typval_T *rettv)
1212{
1213 rettv->v_type = VAR_JOB;
1214 rettv->vval.v_job = NULL;
1215}
1216#endif
1217
1218 void
1219f_test_null_list(typval_T *argvars UNUSED, typval_T *rettv)
1220{
1221 rettv_list_set(rettv, NULL);
1222}
1223
1224 void
Bram Moolenaare69f6d02020-04-01 22:11:01 +02001225f_test_null_function(typval_T *argvars UNUSED, typval_T *rettv)
1226{
1227 rettv->v_type = VAR_FUNC;
1228 rettv->vval.v_string = NULL;
1229}
1230
1231 void
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001232f_test_null_partial(typval_T *argvars UNUSED, typval_T *rettv)
1233{
1234 rettv->v_type = VAR_PARTIAL;
1235 rettv->vval.v_partial = NULL;
1236}
1237
1238 void
1239f_test_null_string(typval_T *argvars UNUSED, typval_T *rettv)
1240{
1241 rettv->v_type = VAR_STRING;
1242 rettv->vval.v_string = NULL;
1243}
1244
Bram Moolenaar8ed04582020-02-22 19:07:28 +01001245 void
1246f_test_unknown(typval_T *argvars UNUSED, typval_T *rettv)
1247{
1248 rettv->v_type = VAR_UNKNOWN;
1249}
1250
1251 void
1252f_test_void(typval_T *argvars UNUSED, typval_T *rettv)
1253{
1254 rettv->v_type = VAR_VOID;
1255}
1256
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001257 void
1258f_test_setmouse(typval_T *argvars, typval_T *rettv UNUSED)
1259{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001260 if (in_vim9script()
1261 && (check_for_number_arg(argvars, 0) == FAIL
1262 || check_for_number_arg(argvars, 1) == FAIL))
1263 return;
1264
Yegappan Lakshmanan7237cab2021-06-22 19:52:27 +02001265 if (argvars[0].v_type != VAR_NUMBER || (argvars[1].v_type) != VAR_NUMBER)
1266 {
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00001267 emsg(_(e_invalid_argument));
Yegappan Lakshmanan7237cab2021-06-22 19:52:27 +02001268 return;
1269 }
1270
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001271 mouse_row = (time_t)tv_get_number(&argvars[0]) - 1;
1272 mouse_col = (time_t)tv_get_number(&argvars[1]) - 1;
1273}
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001274
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001275# ifdef FEAT_GUI
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001276 static int
1277test_gui_drop_files(dict_T *args UNUSED)
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001278{
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001279# if defined(HAVE_DROP_FILE)
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001280 int row;
1281 int col;
1282 int_u mods;
1283 char_u **fnames;
1284 int count = 0;
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001285 typval_T t;
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001286 list_T *l;
1287 listitem_T *li;
1288
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +01001289 if (!dict_has_key(args, "files")
1290 || !dict_has_key(args, "row")
1291 || !dict_has_key(args, "col")
1292 || !dict_has_key(args, "modifiers"))
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001293 return FALSE;
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001294
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001295 (void)dict_get_tv(args, (char_u *)"files", &t);
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001296 row = (int)dict_get_number(args, (char_u *)"row");
1297 col = (int)dict_get_number(args, (char_u *)"col");
1298 mods = (int)dict_get_number(args, (char_u *)"modifiers");
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001299
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001300 if (t.v_type != VAR_LIST || list_len(t.vval.v_list) == 0)
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001301 return FALSE;
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001302
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001303 l = t.vval.v_list;
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001304 fnames = ALLOC_MULT(char_u *, list_len(l));
1305 if (fnames == NULL)
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001306 return FALSE;
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001307
1308 FOR_ALL_LIST_ITEMS(l, li)
1309 {
1310 // ignore non-string items
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001311 if (li->li_tv.v_type != VAR_STRING
1312 || li->li_tv.vval.v_string == NULL)
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001313 continue;
1314
1315 fnames[count] = vim_strsave(li->li_tv.vval.v_string);
1316 if (fnames[count] == NULL)
1317 {
1318 while (--count >= 0)
1319 vim_free(fnames[count]);
1320 vim_free(fnames);
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001321 return FALSE;
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001322 }
1323 count++;
1324 }
1325
1326 if (count > 0)
1327 gui_handle_drop(TEXT_X(col - 1), TEXT_Y(row - 1), mods, fnames, count);
1328 else
1329 vim_free(fnames);
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001330# endif
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001331
1332 return TRUE;
1333}
1334
Bram Moolenaar4e3b3182022-02-01 10:16:00 +00001335#if defined(FIND_REPLACE_DIALOG)
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001336 static int
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001337test_gui_find_repl(dict_T *args)
1338{
1339 int flags;
1340 char_u *find_text;
1341 char_u *repl_text;
1342 int forward;
1343 int retval;
1344
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +01001345 if (!dict_has_key(args, "find_text")
1346 || !dict_has_key(args, "repl_text")
1347 || !dict_has_key(args, "flags")
1348 || !dict_has_key(args, "forward"))
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001349 return FALSE;
1350
1351 find_text = dict_get_string(args, (char_u *)"find_text", TRUE);
1352 repl_text = dict_get_string(args, (char_u *)"repl_text", TRUE);
1353 flags = (int)dict_get_number(args, (char_u *)"flags");
1354 forward = (int)dict_get_number(args, (char_u *)"forward");
1355
1356 retval = gui_do_findrepl(flags, find_text, repl_text, forward);
1357 vim_free(find_text);
1358 vim_free(repl_text);
1359
1360 return retval;
1361}
Bram Moolenaar4e3b3182022-02-01 10:16:00 +00001362#endif
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001363
1364 static int
1365test_gui_mouse_event(dict_T *args)
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001366{
1367 int button;
1368 int row;
1369 int col;
1370 int repeated_click;
1371 int_u mods;
Ernie Raelc4cb5442022-04-03 15:47:28 +01001372 int move;
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001373
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +01001374 if (!dict_has_key(args, "row")
1375 || !dict_has_key(args, "col"))
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001376 return FALSE;
1377
Ernie Raelc4cb5442022-04-03 15:47:28 +01001378 // Note: "move" is optional, requires fewer arguments
1379 move = (int)dict_get_bool(args, (char_u *)"move", FALSE);
1380
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +01001381 if (!move && (!dict_has_key(args, "button")
1382 || !dict_has_key(args, "multiclick")
1383 || !dict_has_key(args, "modifiers")))
Ernie Raelc4cb5442022-04-03 15:47:28 +01001384 return FALSE;
1385
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001386 row = (int)dict_get_number(args, (char_u *)"row");
1387 col = (int)dict_get_number(args, (char_u *)"col");
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001388
Ernie Raelc4cb5442022-04-03 15:47:28 +01001389 if (move)
Bram Moolenaar7add8d32022-05-16 15:27:46 +01001390 {
1391 if (dict_get_bool(args, (char_u *)"cell", FALSE))
1392 {
1393 // click in the middle of the character cell
1394 row = row * gui.char_height + gui.char_height / 2;
1395 col = col * gui.char_width + gui.char_width / 2;
1396 }
Ernie Raelc4cb5442022-04-03 15:47:28 +01001397 gui_mouse_moved(col, row);
Bram Moolenaar7add8d32022-05-16 15:27:46 +01001398 }
Ernie Raelc4cb5442022-04-03 15:47:28 +01001399 else
1400 {
1401 button = (int)dict_get_number(args, (char_u *)"button");
1402 repeated_click = (int)dict_get_number(args, (char_u *)"multiclick");
1403 mods = (int)dict_get_number(args, (char_u *)"modifiers");
1404
LemonBoyc27747e2022-05-07 12:25:40 +01001405 // Reset the scroll values to known values.
1406 // XXX: Remove this when/if the scroll step is made configurable.
1407 mouse_set_hor_scroll_step(6);
1408 mouse_set_vert_scroll_step(3);
1409
Ernie Raelc4cb5442022-04-03 15:47:28 +01001410 gui_send_mouse_event(button, TEXT_X(col - 1), TEXT_Y(row - 1),
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001411 repeated_click, mods);
Ernie Raelc4cb5442022-04-03 15:47:28 +01001412 }
1413
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001414 return TRUE;
1415}
1416
1417 static int
Yegappan Lakshmanan9e0208f2022-01-31 17:40:55 +00001418test_gui_scrollbar(dict_T *args)
1419{
1420 char_u *which;
1421 long value;
1422 int dragging;
1423 scrollbar_T *sb = NULL;
1424
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +01001425 if (!dict_has_key(args, "which")
1426 || !dict_has_key(args, "value")
1427 || !dict_has_key(args, "dragging"))
Yegappan Lakshmanan9e0208f2022-01-31 17:40:55 +00001428 return FALSE;
1429
1430 which = dict_get_string(args, (char_u *)"which", FALSE);
1431 value = (long)dict_get_number(args, (char_u *)"value");
1432 dragging = (int)dict_get_number(args, (char_u *)"dragging");
1433
1434 if (STRCMP(which, "left") == 0)
1435 sb = &curwin->w_scrollbars[SBAR_LEFT];
1436 else if (STRCMP(which, "right") == 0)
1437 sb = &curwin->w_scrollbars[SBAR_RIGHT];
1438 else if (STRCMP(which, "hor") == 0)
1439 sb = &gui.bottom_sbar;
1440 if (sb == NULL)
1441 {
1442 semsg(_(e_invalid_argument_str), which);
1443 return FALSE;
1444 }
1445 gui_drag_scrollbar(sb, value, dragging);
1446# ifndef USE_ON_FLY_SCROLL
1447 // need to loop through normal_cmd() to handle the scroll events
1448 exec_normal(FALSE, TRUE, FALSE);
1449# endif
1450
1451 return TRUE;
1452}
1453
1454 static int
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001455test_gui_tabline_event(dict_T *args UNUSED)
1456{
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001457# ifdef FEAT_GUI_TABLINE
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001458 int tabnr;
1459
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +01001460 if (!dict_has_key(args, "tabnr"))
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001461 return FALSE;
1462
1463 tabnr = (int)dict_get_number(args, (char_u *)"tabnr");
1464
1465 return send_tabline_event(tabnr);
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001466# else
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001467 return FALSE;
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001468# endif
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001469}
1470
1471 static int
1472test_gui_tabmenu_event(dict_T *args UNUSED)
1473{
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001474# ifdef FEAT_GUI_TABLINE
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001475 int tabnr;
1476 int item;
1477
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +01001478 if (!dict_has_key(args, "tabnr")
1479 || !dict_has_key(args, "item"))
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001480 return FALSE;
1481
1482 tabnr = (int)dict_get_number(args, (char_u *)"tabnr");
1483 item = (int)dict_get_number(args, (char_u *)"item");
1484
1485 send_tabline_menu_event(tabnr, item);
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001486# endif
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001487 return TRUE;
1488}
1489# endif
1490
1491 void
1492f_test_gui_event(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
1493{
1494# ifdef FEAT_GUI
1495 char_u *event;
1496
1497 rettv->v_type = VAR_BOOL;
1498 rettv->vval.v_number = FALSE;
1499
1500 if (check_for_string_arg(argvars, 0) == FAIL
1501 || check_for_dict_arg(argvars, 1) == FAIL
1502 || argvars[1].vval.v_dict == NULL)
1503 return;
1504
1505 event = tv_get_string(&argvars[0]);
1506 if (STRCMP(event, "dropfiles") == 0)
1507 rettv->vval.v_number = test_gui_drop_files(argvars[1].vval.v_dict);
Bram Moolenaar4e3b3182022-02-01 10:16:00 +00001508# if defined(FIND_REPLACE_DIALOG)
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001509 else if (STRCMP(event, "findrepl") == 0)
1510 rettv->vval.v_number = test_gui_find_repl(argvars[1].vval.v_dict);
Bram Moolenaar4e3b3182022-02-01 10:16:00 +00001511# endif
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001512 else if (STRCMP(event, "mouse") == 0)
1513 rettv->vval.v_number = test_gui_mouse_event(argvars[1].vval.v_dict);
Yegappan Lakshmanan9e0208f2022-01-31 17:40:55 +00001514 else if (STRCMP(event, "scrollbar") == 0)
1515 rettv->vval.v_number = test_gui_scrollbar(argvars[1].vval.v_dict);
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001516 else if (STRCMP(event, "tabline") == 0)
1517 rettv->vval.v_number = test_gui_tabline_event(argvars[1].vval.v_dict);
1518 else if (STRCMP(event, "tabmenu") == 0)
1519 rettv->vval.v_number = test_gui_tabmenu_event(argvars[1].vval.v_dict);
1520 else
1521 {
1522 semsg(_(e_invalid_argument_str), event);
1523 return;
1524 }
1525# endif
1526}
1527
1528 void
1529f_test_settime(typval_T *argvars, typval_T *rettv UNUSED)
1530{
1531 if (in_vim9script() && check_for_number_arg(argvars, 0) == FAIL)
1532 return;
1533
1534 time_for_testing = (time_t)tv_get_number(&argvars[0]);
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001535}
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001536
1537#endif // defined(FEAT_EVAL)