blob: 8c682cf6bce149d3e64bdf1548f0da340371fb6b [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 Moolenaar249e1b92022-08-14 22:23:02 +0100600 char_u *tofree = NULL;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200601
Yegappan Lakshmanana764e732021-07-25 15:57:32 +0200602 if (check_for_string_or_number_arg(argvars, 0) == FAIL
603 || check_for_opt_string_or_list_arg(argvars, 1) == FAIL
604 || (argvars[1].v_type != VAR_UNKNOWN
605 && (argvars[2].v_type != VAR_UNKNOWN
606 && (check_for_opt_number_arg(argvars, 3) == FAIL
607 || (argvars[3].v_type != VAR_UNKNOWN
608 && check_for_opt_string_arg(argvars, 4) == FAIL)))))
609 return;
610
611 cmd = tv_get_string_chk(&argvars[0]);
612
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200613 // trylevel must be zero for a ":throw" command to be considered failed
614 trylevel = 0;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200615 suppress_errthrow = TRUE;
Bram Moolenaar28ee8922020-10-28 20:20:00 +0100616 in_assert_fails = TRUE;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200617
618 do_cmdline_cmd(cmd);
Bram Moolenaar53989552019-12-23 22:59:18 +0100619 if (called_emsg == called_emsg_before)
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200620 {
621 prepare_assert_error(&ga);
622 ga_concat(&ga, (char_u *)"command did not fail: ");
623 assert_append_cmd_or_arg(&ga, argvars, cmd);
624 assert_error(&ga);
625 ga_clear(&ga);
626 rettv->vval.v_number = 1;
627 }
628 else if (argvars[1].v_type != VAR_UNKNOWN)
629 {
630 char_u buf[NUMBUFLEN];
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200631 char_u *expected;
Bram Moolenaar631e8f92020-11-04 15:07:16 +0100632 char_u *expected_str = NULL;
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200633 int error_found = FALSE;
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200634 int error_found_index = 1;
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200635 char_u *actual = emsg_assert_fails_msg == NULL ? (char_u *)"[unknown]"
636 : emsg_assert_fails_msg;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200637
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200638 if (argvars[1].v_type == VAR_STRING)
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200639 {
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200640 expected = tv_get_string_buf_chk(&argvars[1], buf);
641 error_found = expected == NULL
642 || strstr((char *)actual, (char *)expected) == NULL;
643 }
644 else if (argvars[1].v_type == VAR_LIST)
645 {
646 list_T *list = argvars[1].vval.v_list;
647 typval_T *tv;
648
649 if (list == NULL || list->lv_len < 1 || list->lv_len > 2)
650 {
Bram Moolenaar44d66522020-09-06 22:26:57 +0200651 wrong_arg_msg = e_assert_fails_second_arg;
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200652 goto theend;
653 }
654 CHECK_LIST_MATERIALIZE(list);
655 tv = &list->lv_first->li_tv;
656 expected = tv_get_string_buf_chk(tv, buf);
657 if (!pattern_match(expected, actual, FALSE))
658 {
659 error_found = TRUE;
Bram Moolenaar631e8f92020-11-04 15:07:16 +0100660 expected_str = expected;
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200661 }
662 else if (list->lv_len == 2)
663 {
Bram Moolenaar249e1b92022-08-14 22:23:02 +0100664 // make a copy, an error in pattern_match() may free it
665 tofree = actual = vim_strsave(get_vim_var_str(VV_ERRMSG));
666 if (actual != NULL)
Bram Moolenaar631e8f92020-11-04 15:07:16 +0100667 {
Bram Moolenaar249e1b92022-08-14 22:23:02 +0100668 tv = &list->lv_u.mat.lv_last->li_tv;
669 expected = tv_get_string_buf_chk(tv, buf);
670 if (!pattern_match(expected, actual, FALSE))
671 {
672 error_found = TRUE;
673 expected_str = expected;
674 }
Bram Moolenaar631e8f92020-11-04 15:07:16 +0100675 }
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200676 }
677 }
678 else
679 {
Bram Moolenaar44d66522020-09-06 22:26:57 +0200680 wrong_arg_msg = e_assert_fails_second_arg;
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200681 goto theend;
682 }
683
Bram Moolenaar9b02d642020-08-18 23:24:13 +0200684 if (!error_found && argvars[2].v_type != VAR_UNKNOWN
Bram Moolenaar44d66522020-09-06 22:26:57 +0200685 && argvars[3].v_type != VAR_UNKNOWN)
Bram Moolenaar1d634542020-08-18 13:41:50 +0200686 {
Bram Moolenaar44d66522020-09-06 22:26:57 +0200687 if (argvars[3].v_type != VAR_NUMBER)
688 {
689 wrong_arg_msg = e_assert_fails_fourth_argument;
690 goto theend;
691 }
692 else if (argvars[3].vval.v_number >= 0
693 && argvars[3].vval.v_number != emsg_assert_fails_lnum)
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200694 {
695 error_found = TRUE;
696 error_found_index = 3;
697 }
Bram Moolenaar44d66522020-09-06 22:26:57 +0200698 if (!error_found && argvars[4].v_type != VAR_UNKNOWN)
699 {
700 if (argvars[4].v_type != VAR_STRING)
701 {
702 wrong_arg_msg = e_assert_fails_fifth_argument;
703 goto theend;
704 }
705 else if (argvars[4].vval.v_string != NULL
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200706 && !pattern_match(argvars[4].vval.v_string,
707 emsg_assert_fails_context, FALSE))
Bram Moolenaar44d66522020-09-06 22:26:57 +0200708 {
709 error_found = TRUE;
710 error_found_index = 4;
711 }
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200712 }
Bram Moolenaar1d634542020-08-18 13:41:50 +0200713 }
714
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200715 if (error_found)
716 {
717 typval_T actual_tv;
718
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200719 prepare_assert_error(&ga);
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200720 if (error_found_index == 3)
Bram Moolenaar1d634542020-08-18 13:41:50 +0200721 {
722 actual_tv.v_type = VAR_NUMBER;
723 actual_tv.vval.v_number = emsg_assert_fails_lnum;
724 }
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200725 else if (error_found_index == 4)
726 {
727 actual_tv.v_type = VAR_STRING;
728 actual_tv.vval.v_string = emsg_assert_fails_context;
729 }
Bram Moolenaar1d634542020-08-18 13:41:50 +0200730 else
731 {
732 actual_tv.v_type = VAR_STRING;
733 actual_tv.vval.v_string = actual;
734 }
Bram Moolenaar631e8f92020-11-04 15:07:16 +0100735 fill_assert_error(&ga, &argvars[2], expected_str,
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200736 &argvars[error_found_index], &actual_tv, ASSERT_OTHER);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200737 ga_concat(&ga, (char_u *)": ");
738 assert_append_cmd_or_arg(&ga, argvars, cmd);
739 assert_error(&ga);
740 ga_clear(&ga);
741 rettv->vval.v_number = 1;
742 }
743 }
744
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200745theend:
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200746 trylevel = save_trylevel;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200747 suppress_errthrow = FALSE;
Bram Moolenaar28ee8922020-10-28 20:20:00 +0100748 in_assert_fails = FALSE;
749 did_emsg = FALSE;
Bram Moolenaar8bea1712022-06-15 20:49:35 +0100750 got_int = FALSE;
Bram Moolenaar28ee8922020-10-28 20:20:00 +0100751 msg_col = 0;
752 need_wait_return = FALSE;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200753 emsg_on_display = FALSE;
Bram Moolenaar28ee8922020-10-28 20:20:00 +0100754 msg_scrolled = 0;
755 lines_left = Rows;
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200756 VIM_CLEAR(emsg_assert_fails_msg);
Bram Moolenaar249e1b92022-08-14 22:23:02 +0100757 vim_free(tofree);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200758 set_vim_var_string(VV_ERRMSG, NULL, 0);
Bram Moolenaar44d66522020-09-06 22:26:57 +0200759 if (wrong_arg_msg != NULL)
760 emsg(_(wrong_arg_msg));
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200761}
762
763/*
764 * "assert_false(actual[, msg])" function
765 */
766 void
767f_assert_false(typval_T *argvars, typval_T *rettv)
768{
769 rettv->vval.v_number = assert_bool(argvars, FALSE);
770}
771
772 static int
773assert_inrange(typval_T *argvars)
774{
775 garray_T ga;
776 int error = FALSE;
777 char_u *tofree;
778 char msg[200];
779 char_u numbuf[NUMBUFLEN];
780
781#ifdef FEAT_FLOAT
782 if (argvars[0].v_type == VAR_FLOAT
783 || argvars[1].v_type == VAR_FLOAT
784 || argvars[2].v_type == VAR_FLOAT)
785 {
786 float_T flower = tv_get_float(&argvars[0]);
787 float_T fupper = tv_get_float(&argvars[1]);
788 float_T factual = tv_get_float(&argvars[2]);
789
790 if (factual < flower || factual > fupper)
791 {
792 prepare_assert_error(&ga);
793 if (argvars[3].v_type != VAR_UNKNOWN)
794 {
795 ga_concat(&ga, tv2string(&argvars[3], &tofree, numbuf, 0));
796 vim_free(tofree);
797 }
798 else
799 {
800 vim_snprintf(msg, 200, "Expected range %g - %g, but got %g",
801 flower, fupper, factual);
802 ga_concat(&ga, (char_u *)msg);
803 }
804 assert_error(&ga);
805 ga_clear(&ga);
806 return 1;
807 }
808 }
809 else
810#endif
811 {
812 varnumber_T lower = tv_get_number_chk(&argvars[0], &error);
813 varnumber_T upper = tv_get_number_chk(&argvars[1], &error);
814 varnumber_T actual = tv_get_number_chk(&argvars[2], &error);
815
816 if (error)
817 return 0;
818 if (actual < lower || actual > upper)
819 {
820 prepare_assert_error(&ga);
821 if (argvars[3].v_type != VAR_UNKNOWN)
822 {
823 ga_concat(&ga, tv2string(&argvars[3], &tofree, numbuf, 0));
824 vim_free(tofree);
825 }
826 else
827 {
828 vim_snprintf(msg, 200, "Expected range %ld - %ld, but got %ld",
829 (long)lower, (long)upper, (long)actual);
830 ga_concat(&ga, (char_u *)msg);
831 }
832 assert_error(&ga);
833 ga_clear(&ga);
834 return 1;
835 }
836 }
837 return 0;
838}
839
840/*
841 * "assert_inrange(lower, upper[, msg])" function
842 */
843 void
844f_assert_inrange(typval_T *argvars, typval_T *rettv)
845{
Yegappan Lakshmanana764e732021-07-25 15:57:32 +0200846 if (check_for_float_or_nr_arg(argvars, 0) == FAIL
847 || check_for_float_or_nr_arg(argvars, 1) == FAIL
848 || check_for_float_or_nr_arg(argvars, 2) == FAIL
849 || check_for_opt_string_arg(argvars, 3) == FAIL)
850 return;
851
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200852 rettv->vval.v_number = assert_inrange(argvars);
853}
854
855/*
856 * "assert_match(pattern, actual[, msg])" function
857 */
858 void
859f_assert_match(typval_T *argvars, typval_T *rettv)
860{
861 rettv->vval.v_number = assert_match_common(argvars, ASSERT_MATCH);
862}
863
864/*
865 * "assert_notmatch(pattern, actual[, msg])" function
866 */
867 void
868f_assert_notmatch(typval_T *argvars, typval_T *rettv)
869{
870 rettv->vval.v_number = assert_match_common(argvars, ASSERT_NOTMATCH);
871}
872
873/*
874 * "assert_report(msg)" function
875 */
876 void
877f_assert_report(typval_T *argvars, typval_T *rettv)
878{
879 garray_T ga;
880
Yegappan Lakshmananc72bdd22021-07-11 19:44:18 +0200881 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
882 return;
883
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200884 prepare_assert_error(&ga);
885 ga_concat(&ga, tv_get_string(&argvars[0]));
886 assert_error(&ga);
887 ga_clear(&ga);
888 rettv->vval.v_number = 1;
889}
890
891/*
892 * "assert_true(actual[, msg])" function
893 */
894 void
895f_assert_true(typval_T *argvars, typval_T *rettv)
896{
897 rettv->vval.v_number = assert_bool(argvars, TRUE);
898}
899
900/*
901 * "test_alloc_fail(id, countdown, repeat)" function
902 */
903 void
904f_test_alloc_fail(typval_T *argvars, typval_T *rettv UNUSED)
905{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200906 if (in_vim9script()
907 && (check_for_number_arg(argvars, 0) == FAIL
908 || check_for_number_arg(argvars, 1) == FAIL
909 || check_for_number_arg(argvars, 2) == FAIL))
910 return;
911
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200912 if (argvars[0].v_type != VAR_NUMBER
913 || argvars[0].vval.v_number <= 0
914 || argvars[1].v_type != VAR_NUMBER
915 || argvars[1].vval.v_number < 0
916 || argvars[2].v_type != VAR_NUMBER)
Bram Moolenaar436b5ad2021-12-31 22:49:24 +0000917 emsg(_(e_invalid_argument));
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200918 else
919 {
920 alloc_fail_id = argvars[0].vval.v_number;
921 if (alloc_fail_id >= aid_last)
Bram Moolenaar436b5ad2021-12-31 22:49:24 +0000922 emsg(_(e_invalid_argument));
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200923 alloc_fail_countdown = argvars[1].vval.v_number;
924 alloc_fail_repeat = argvars[2].vval.v_number;
925 did_outofmem_msg = FALSE;
926 }
927}
928
929/*
930 * "test_autochdir()"
931 */
932 void
933f_test_autochdir(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
934{
935#if defined(FEAT_AUTOCHDIR)
936 test_autochdir = TRUE;
937#endif
938}
939
940/*
941 * "test_feedinput()"
942 */
943 void
944f_test_feedinput(typval_T *argvars, typval_T *rettv UNUSED)
945{
946#ifdef USE_INPUT_BUF
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200947 char_u *val;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200948
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200949 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
950 return;
951
952 val = tv_get_string_chk(&argvars[0]);
Bram Moolenaar272ca952020-01-28 20:49:11 +0100953# ifdef VIMDLL
954 // this doesn't work in the console
955 if (!gui.in_use)
956 return;
957# endif
958
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200959 if (val != NULL)
960 {
961 trash_input_buf();
962 add_to_input_buf_csi(val, (int)STRLEN(val));
963 }
964#endif
965}
966
967/*
968 * "test_getvalue({name})" function
969 */
970 void
971f_test_getvalue(typval_T *argvars, typval_T *rettv)
972{
Yegappan Lakshmanan8deb2b32022-09-02 15:15:27 +0100973 char_u *name;
974
975 if (check_for_string_arg(argvars, 0) == FAIL)
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200976 return;
977
Yegappan Lakshmanan8deb2b32022-09-02 15:15:27 +0100978 name = tv_get_string(&argvars[0]);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200979
Yegappan Lakshmanan8deb2b32022-09-02 15:15:27 +0100980 if (STRCMP(name, (char_u *)"need_fileinfo") == 0)
981 rettv->vval.v_number = need_fileinfo;
982 else
983 semsg(_(e_invalid_argument_str), name);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200984}
985
986/*
987 * "test_option_not_set({name})" function
988 */
989 void
990f_test_option_not_set(typval_T *argvars, typval_T *rettv UNUSED)
991{
992 char_u *name = (char_u *)"";
993
Yegappan Lakshmanan8deb2b32022-09-02 15:15:27 +0100994 if (check_for_string_arg(argvars, 0) == FAIL)
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200995 return;
996
Yegappan Lakshmanan8deb2b32022-09-02 15:15:27 +0100997 name = tv_get_string(&argvars[0]);
998 if (reset_option_was_set(name) == FAIL)
999 semsg(_(e_invalid_argument_str), name);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001000}
1001
1002/*
1003 * "test_override({name}, {val})" function
1004 */
1005 void
1006f_test_override(typval_T *argvars, typval_T *rettv UNUSED)
1007{
1008 char_u *name = (char_u *)"";
1009 int val;
1010 static int save_starting = -1;
1011
Yegappan Lakshmanan8deb2b32022-09-02 15:15:27 +01001012 if (check_for_string_arg(argvars, 0) == FAIL
1013 || check_for_number_arg(argvars, 1) == FAIL)
Yegappan Lakshmanan1a71d312021-07-15 12:49:58 +02001014 return;
1015
Yegappan Lakshmanan8deb2b32022-09-02 15:15:27 +01001016 name = tv_get_string(&argvars[0]);
1017 val = (int)tv_get_number(&argvars[1]);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001018
Yegappan Lakshmanan8deb2b32022-09-02 15:15:27 +01001019 if (STRCMP(name, (char_u *)"redraw") == 0)
1020 disable_redraw_for_testing = val;
1021 else if (STRCMP(name, (char_u *)"redraw_flag") == 0)
1022 ignore_redraw_flag_for_testing = val;
1023 else if (STRCMP(name, (char_u *)"char_avail") == 0)
1024 disable_char_avail_for_testing = val;
1025 else if (STRCMP(name, (char_u *)"starting") == 0)
1026 {
1027 if (val)
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001028 {
Yegappan Lakshmanan8deb2b32022-09-02 15:15:27 +01001029 if (save_starting < 0)
1030 save_starting = starting;
1031 starting = 0;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001032 }
1033 else
Yegappan Lakshmanan8deb2b32022-09-02 15:15:27 +01001034 {
1035 starting = save_starting;
1036 save_starting = -1;
1037 }
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001038 }
Yegappan Lakshmanan8deb2b32022-09-02 15:15:27 +01001039 else if (STRCMP(name, (char_u *)"nfa_fail") == 0)
1040 nfa_fail_for_testing = val;
1041 else if (STRCMP(name, (char_u *)"no_query_mouse") == 0)
1042 no_query_mouse_for_testing = val;
1043 else if (STRCMP(name, (char_u *)"no_wait_return") == 0)
1044 no_wait_return = val;
1045 else if (STRCMP(name, (char_u *)"ui_delay") == 0)
1046 ui_delay_for_testing = val;
1047 else if (STRCMP(name, (char_u *)"term_props") == 0)
1048 reset_term_props_on_termresponse = val;
1049 else if (STRCMP(name, (char_u *)"vterm_title") == 0)
1050 disable_vterm_title_for_testing = val;
1051 else if (STRCMP(name, (char_u *)"uptime") == 0)
1052 override_sysinfo_uptime = val;
1053 else if (STRCMP(name, (char_u *)"alloc_lines") == 0)
1054 ml_get_alloc_lines = val;
1055 else if (STRCMP(name, (char_u *)"autoload") == 0)
1056 override_autoload = val;
1057 else if (STRCMP(name, (char_u *)"ALL") == 0)
1058 {
1059 disable_char_avail_for_testing = FALSE;
1060 disable_redraw_for_testing = FALSE;
1061 ignore_redraw_flag_for_testing = FALSE;
1062 nfa_fail_for_testing = FALSE;
1063 no_query_mouse_for_testing = FALSE;
1064 ui_delay_for_testing = 0;
1065 reset_term_props_on_termresponse = FALSE;
1066 override_sysinfo_uptime = -1;
1067 // ml_get_alloc_lines is not reset by "ALL"
1068 if (save_starting >= 0)
1069 {
1070 starting = save_starting;
1071 save_starting = -1;
1072 }
1073 }
1074 else
1075 semsg(_(e_invalid_argument_str), name);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001076}
1077
1078/*
1079 * "test_refcount({expr})" function
1080 */
1081 void
1082f_test_refcount(typval_T *argvars, typval_T *rettv)
1083{
1084 int retval = -1;
1085
1086 switch (argvars[0].v_type)
1087 {
1088 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02001089 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001090 case VAR_VOID:
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001091 case VAR_NUMBER:
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01001092 case VAR_BOOL:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001093 case VAR_FLOAT:
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001094 case VAR_SPECIAL:
1095 case VAR_STRING:
Bram Moolenaarf18332f2021-05-07 17:55:55 +02001096 case VAR_INSTR:
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001097 break;
1098 case VAR_JOB:
1099#ifdef FEAT_JOB_CHANNEL
1100 if (argvars[0].vval.v_job != NULL)
1101 retval = argvars[0].vval.v_job->jv_refcount - 1;
1102#endif
1103 break;
1104 case VAR_CHANNEL:
1105#ifdef FEAT_JOB_CHANNEL
1106 if (argvars[0].vval.v_channel != NULL)
1107 retval = argvars[0].vval.v_channel->ch_refcount - 1;
1108#endif
1109 break;
1110 case VAR_FUNC:
1111 if (argvars[0].vval.v_string != NULL)
1112 {
1113 ufunc_T *fp;
1114
Bram Moolenaard9d2fd02022-01-13 21:15:21 +00001115 fp = find_func(argvars[0].vval.v_string, FALSE);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001116 if (fp != NULL)
1117 retval = fp->uf_refcount;
1118 }
1119 break;
1120 case VAR_PARTIAL:
1121 if (argvars[0].vval.v_partial != NULL)
1122 retval = argvars[0].vval.v_partial->pt_refcount - 1;
1123 break;
1124 case VAR_BLOB:
1125 if (argvars[0].vval.v_blob != NULL)
1126 retval = argvars[0].vval.v_blob->bv_refcount - 1;
1127 break;
1128 case VAR_LIST:
1129 if (argvars[0].vval.v_list != NULL)
1130 retval = argvars[0].vval.v_list->lv_refcount - 1;
1131 break;
1132 case VAR_DICT:
1133 if (argvars[0].vval.v_dict != NULL)
1134 retval = argvars[0].vval.v_dict->dv_refcount - 1;
1135 break;
1136 }
1137
1138 rettv->v_type = VAR_NUMBER;
1139 rettv->vval.v_number = retval;
1140
1141}
1142
1143/*
1144 * "test_garbagecollect_now()" function
1145 */
1146 void
1147f_test_garbagecollect_now(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
1148{
Bram Moolenaar0d6f5d92019-12-05 21:33:15 +01001149 // This is dangerous, any Lists and Dicts used internally may be freed
1150 // while still in use.
Bram Moolenaarb3d83982022-01-27 19:59:47 +00001151 if (!get_vim_var_nr(VV_TESTING))
1152 emsg(_(e_calling_test_garbagecollect_now_while_v_testing_is_not_set));
1153 else
1154 garbage_collect(TRUE);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001155}
1156
1157/*
1158 * "test_garbagecollect_soon()" function
1159 */
1160 void
1161f_test_garbagecollect_soon(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
1162{
1163 may_garbage_collect = TRUE;
1164}
1165
1166/*
1167 * "test_ignore_error()" function
1168 */
1169 void
1170f_test_ignore_error(typval_T *argvars, typval_T *rettv UNUSED)
1171{
Yegappan Lakshmanan8deb2b32022-09-02 15:15:27 +01001172 if (check_for_string_arg(argvars, 0) == FAIL)
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001173 return;
1174
Yegappan Lakshmanan8deb2b32022-09-02 15:15:27 +01001175 ignore_error_for_testing(tv_get_string(&argvars[0]));
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001176}
1177
1178 void
1179f_test_null_blob(typval_T *argvars UNUSED, typval_T *rettv)
1180{
1181 rettv->v_type = VAR_BLOB;
1182 rettv->vval.v_blob = NULL;
1183}
1184
1185#ifdef FEAT_JOB_CHANNEL
1186 void
1187f_test_null_channel(typval_T *argvars UNUSED, typval_T *rettv)
1188{
1189 rettv->v_type = VAR_CHANNEL;
1190 rettv->vval.v_channel = NULL;
1191}
1192#endif
1193
1194 void
1195f_test_null_dict(typval_T *argvars UNUSED, typval_T *rettv)
1196{
1197 rettv_dict_set(rettv, NULL);
1198}
1199
1200#ifdef FEAT_JOB_CHANNEL
1201 void
1202f_test_null_job(typval_T *argvars UNUSED, typval_T *rettv)
1203{
1204 rettv->v_type = VAR_JOB;
1205 rettv->vval.v_job = NULL;
1206}
1207#endif
1208
1209 void
1210f_test_null_list(typval_T *argvars UNUSED, typval_T *rettv)
1211{
1212 rettv_list_set(rettv, NULL);
1213}
1214
1215 void
Bram Moolenaare69f6d02020-04-01 22:11:01 +02001216f_test_null_function(typval_T *argvars UNUSED, typval_T *rettv)
1217{
1218 rettv->v_type = VAR_FUNC;
1219 rettv->vval.v_string = NULL;
1220}
1221
1222 void
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001223f_test_null_partial(typval_T *argvars UNUSED, typval_T *rettv)
1224{
1225 rettv->v_type = VAR_PARTIAL;
1226 rettv->vval.v_partial = NULL;
1227}
1228
1229 void
1230f_test_null_string(typval_T *argvars UNUSED, typval_T *rettv)
1231{
1232 rettv->v_type = VAR_STRING;
1233 rettv->vval.v_string = NULL;
1234}
1235
Bram Moolenaar8ed04582020-02-22 19:07:28 +01001236 void
1237f_test_unknown(typval_T *argvars UNUSED, typval_T *rettv)
1238{
1239 rettv->v_type = VAR_UNKNOWN;
1240}
1241
1242 void
1243f_test_void(typval_T *argvars UNUSED, typval_T *rettv)
1244{
1245 rettv->v_type = VAR_VOID;
1246}
1247
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001248 void
1249f_test_setmouse(typval_T *argvars, typval_T *rettv UNUSED)
1250{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001251 if (in_vim9script()
1252 && (check_for_number_arg(argvars, 0) == FAIL
1253 || check_for_number_arg(argvars, 1) == FAIL))
1254 return;
1255
Yegappan Lakshmanan8deb2b32022-09-02 15:15:27 +01001256 if (argvars[0].v_type != VAR_NUMBER || argvars[1].v_type != VAR_NUMBER)
Yegappan Lakshmanan7237cab2021-06-22 19:52:27 +02001257 {
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00001258 emsg(_(e_invalid_argument));
Yegappan Lakshmanan7237cab2021-06-22 19:52:27 +02001259 return;
1260 }
1261
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001262 mouse_row = (time_t)tv_get_number(&argvars[0]) - 1;
1263 mouse_col = (time_t)tv_get_number(&argvars[1]) - 1;
1264}
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001265
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001266# ifdef FEAT_GUI
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001267 static int
1268test_gui_drop_files(dict_T *args UNUSED)
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001269{
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001270# if defined(HAVE_DROP_FILE)
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001271 int row;
1272 int col;
1273 int_u mods;
1274 char_u **fnames;
1275 int count = 0;
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001276 typval_T t;
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001277 list_T *l;
1278 listitem_T *li;
1279
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +01001280 if (!dict_has_key(args, "files")
1281 || !dict_has_key(args, "row")
1282 || !dict_has_key(args, "col")
1283 || !dict_has_key(args, "modifiers"))
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001284 return FALSE;
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001285
Bram Moolenaard61efa52022-07-23 09:52:04 +01001286 (void)dict_get_tv(args, "files", &t);
1287 row = (int)dict_get_number(args, "row");
1288 col = (int)dict_get_number(args, "col");
1289 mods = (int)dict_get_number(args, "modifiers");
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001290
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001291 if (t.v_type != VAR_LIST || list_len(t.vval.v_list) == 0)
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001292 return FALSE;
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001293
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001294 l = t.vval.v_list;
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001295 fnames = ALLOC_MULT(char_u *, list_len(l));
1296 if (fnames == NULL)
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001297 return FALSE;
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001298
1299 FOR_ALL_LIST_ITEMS(l, li)
1300 {
1301 // ignore non-string items
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001302 if (li->li_tv.v_type != VAR_STRING
1303 || li->li_tv.vval.v_string == NULL)
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001304 continue;
1305
1306 fnames[count] = vim_strsave(li->li_tv.vval.v_string);
1307 if (fnames[count] == NULL)
1308 {
1309 while (--count >= 0)
1310 vim_free(fnames[count]);
1311 vim_free(fnames);
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001312 return FALSE;
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001313 }
1314 count++;
1315 }
1316
1317 if (count > 0)
1318 gui_handle_drop(TEXT_X(col - 1), TEXT_Y(row - 1), mods, fnames, count);
1319 else
1320 vim_free(fnames);
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001321# endif
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001322
1323 return TRUE;
1324}
1325
Bram Moolenaar4e3b3182022-02-01 10:16:00 +00001326#if defined(FIND_REPLACE_DIALOG)
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001327 static int
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001328test_gui_find_repl(dict_T *args)
1329{
1330 int flags;
1331 char_u *find_text;
1332 char_u *repl_text;
1333 int forward;
1334 int retval;
1335
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +01001336 if (!dict_has_key(args, "find_text")
1337 || !dict_has_key(args, "repl_text")
1338 || !dict_has_key(args, "flags")
1339 || !dict_has_key(args, "forward"))
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001340 return FALSE;
1341
Bram Moolenaard61efa52022-07-23 09:52:04 +01001342 find_text = dict_get_string(args, "find_text", TRUE);
1343 repl_text = dict_get_string(args, "repl_text", TRUE);
1344 flags = (int)dict_get_number(args, "flags");
1345 forward = (int)dict_get_number(args, "forward");
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001346
1347 retval = gui_do_findrepl(flags, find_text, repl_text, forward);
1348 vim_free(find_text);
1349 vim_free(repl_text);
1350
1351 return retval;
1352}
Bram Moolenaar4e3b3182022-02-01 10:16:00 +00001353#endif
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001354
1355 static int
1356test_gui_mouse_event(dict_T *args)
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001357{
1358 int button;
1359 int row;
1360 int col;
1361 int repeated_click;
1362 int_u mods;
Ernie Raelc4cb5442022-04-03 15:47:28 +01001363 int move;
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001364
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +01001365 if (!dict_has_key(args, "row")
1366 || !dict_has_key(args, "col"))
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001367 return FALSE;
1368
Ernie Raelc4cb5442022-04-03 15:47:28 +01001369 // Note: "move" is optional, requires fewer arguments
Bram Moolenaard61efa52022-07-23 09:52:04 +01001370 move = (int)dict_get_bool(args, "move", FALSE);
Ernie Raelc4cb5442022-04-03 15:47:28 +01001371
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +01001372 if (!move && (!dict_has_key(args, "button")
1373 || !dict_has_key(args, "multiclick")
1374 || !dict_has_key(args, "modifiers")))
Ernie Raelc4cb5442022-04-03 15:47:28 +01001375 return FALSE;
1376
Bram Moolenaard61efa52022-07-23 09:52:04 +01001377 row = (int)dict_get_number(args, "row");
1378 col = (int)dict_get_number(args, "col");
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001379
Ernie Raelc4cb5442022-04-03 15:47:28 +01001380 if (move)
Bram Moolenaar7add8d32022-05-16 15:27:46 +01001381 {
Bram Moolenaard61efa52022-07-23 09:52:04 +01001382 if (dict_get_bool(args, "cell", FALSE))
Bram Moolenaar7add8d32022-05-16 15:27:46 +01001383 {
1384 // click in the middle of the character cell
1385 row = row * gui.char_height + gui.char_height / 2;
1386 col = col * gui.char_width + gui.char_width / 2;
1387 }
Ernie Raelc4cb5442022-04-03 15:47:28 +01001388 gui_mouse_moved(col, row);
Bram Moolenaar7add8d32022-05-16 15:27:46 +01001389 }
Ernie Raelc4cb5442022-04-03 15:47:28 +01001390 else
1391 {
Bram Moolenaard61efa52022-07-23 09:52:04 +01001392 button = (int)dict_get_number(args, "button");
1393 repeated_click = (int)dict_get_number(args, "multiclick");
1394 mods = (int)dict_get_number(args, "modifiers");
Ernie Raelc4cb5442022-04-03 15:47:28 +01001395
LemonBoyc27747e2022-05-07 12:25:40 +01001396 // Reset the scroll values to known values.
1397 // XXX: Remove this when/if the scroll step is made configurable.
1398 mouse_set_hor_scroll_step(6);
1399 mouse_set_vert_scroll_step(3);
1400
Ernie Raelc4cb5442022-04-03 15:47:28 +01001401 gui_send_mouse_event(button, TEXT_X(col - 1), TEXT_Y(row - 1),
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001402 repeated_click, mods);
Ernie Raelc4cb5442022-04-03 15:47:28 +01001403 }
1404
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001405 return TRUE;
1406}
1407
1408 static int
Yegappan Lakshmanan9e0208f2022-01-31 17:40:55 +00001409test_gui_scrollbar(dict_T *args)
1410{
1411 char_u *which;
1412 long value;
1413 int dragging;
1414 scrollbar_T *sb = NULL;
1415
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +01001416 if (!dict_has_key(args, "which")
1417 || !dict_has_key(args, "value")
1418 || !dict_has_key(args, "dragging"))
Yegappan Lakshmanan9e0208f2022-01-31 17:40:55 +00001419 return FALSE;
1420
Bram Moolenaard61efa52022-07-23 09:52:04 +01001421 which = dict_get_string(args, "which", FALSE);
1422 value = (long)dict_get_number(args, "value");
1423 dragging = (int)dict_get_number(args, "dragging");
Yegappan Lakshmanan9e0208f2022-01-31 17:40:55 +00001424
1425 if (STRCMP(which, "left") == 0)
1426 sb = &curwin->w_scrollbars[SBAR_LEFT];
1427 else if (STRCMP(which, "right") == 0)
1428 sb = &curwin->w_scrollbars[SBAR_RIGHT];
1429 else if (STRCMP(which, "hor") == 0)
1430 sb = &gui.bottom_sbar;
1431 if (sb == NULL)
1432 {
1433 semsg(_(e_invalid_argument_str), which);
1434 return FALSE;
1435 }
1436 gui_drag_scrollbar(sb, value, dragging);
1437# ifndef USE_ON_FLY_SCROLL
1438 // need to loop through normal_cmd() to handle the scroll events
1439 exec_normal(FALSE, TRUE, FALSE);
1440# endif
1441
1442 return TRUE;
1443}
1444
1445 static int
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001446test_gui_tabline_event(dict_T *args UNUSED)
1447{
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001448# ifdef FEAT_GUI_TABLINE
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001449 int tabnr;
1450
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +01001451 if (!dict_has_key(args, "tabnr"))
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001452 return FALSE;
1453
Bram Moolenaard61efa52022-07-23 09:52:04 +01001454 tabnr = (int)dict_get_number(args, "tabnr");
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001455
1456 return send_tabline_event(tabnr);
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001457# else
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001458 return FALSE;
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001459# endif
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001460}
1461
1462 static int
1463test_gui_tabmenu_event(dict_T *args UNUSED)
1464{
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001465# ifdef FEAT_GUI_TABLINE
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001466 int tabnr;
1467 int item;
1468
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +01001469 if (!dict_has_key(args, "tabnr")
1470 || !dict_has_key(args, "item"))
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001471 return FALSE;
1472
Bram Moolenaard61efa52022-07-23 09:52:04 +01001473 tabnr = (int)dict_get_number(args, "tabnr");
1474 item = (int)dict_get_number(args, "item");
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001475
1476 send_tabline_menu_event(tabnr, item);
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001477# endif
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001478 return TRUE;
1479}
1480# endif
1481
1482 void
1483f_test_gui_event(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
1484{
1485# ifdef FEAT_GUI
1486 char_u *event;
1487
1488 rettv->v_type = VAR_BOOL;
1489 rettv->vval.v_number = FALSE;
1490
Yegappan Lakshmanan81a3ff92022-07-23 05:04:16 +01001491 if (sandbox != 0)
1492 {
1493 emsg(_(e_not_allowed_in_sandbox));
1494 return;
1495 }
1496
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001497 if (check_for_string_arg(argvars, 0) == FAIL
1498 || check_for_dict_arg(argvars, 1) == FAIL
1499 || argvars[1].vval.v_dict == NULL)
1500 return;
1501
1502 event = tv_get_string(&argvars[0]);
1503 if (STRCMP(event, "dropfiles") == 0)
1504 rettv->vval.v_number = test_gui_drop_files(argvars[1].vval.v_dict);
Bram Moolenaar4e3b3182022-02-01 10:16:00 +00001505# if defined(FIND_REPLACE_DIALOG)
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001506 else if (STRCMP(event, "findrepl") == 0)
1507 rettv->vval.v_number = test_gui_find_repl(argvars[1].vval.v_dict);
Bram Moolenaar4e3b3182022-02-01 10:16:00 +00001508# endif
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001509 else if (STRCMP(event, "mouse") == 0)
1510 rettv->vval.v_number = test_gui_mouse_event(argvars[1].vval.v_dict);
Yegappan Lakshmanan9e0208f2022-01-31 17:40:55 +00001511 else if (STRCMP(event, "scrollbar") == 0)
1512 rettv->vval.v_number = test_gui_scrollbar(argvars[1].vval.v_dict);
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001513 else if (STRCMP(event, "tabline") == 0)
1514 rettv->vval.v_number = test_gui_tabline_event(argvars[1].vval.v_dict);
1515 else if (STRCMP(event, "tabmenu") == 0)
1516 rettv->vval.v_number = test_gui_tabmenu_event(argvars[1].vval.v_dict);
Yegappan Lakshmanan81a3ff92022-07-23 05:04:16 +01001517# ifdef FEAT_GUI_MSWIN
1518 else if (STRCMP(event, "sendevent") == 0)
1519 rettv->vval.v_number = test_gui_w32_sendevent(argvars[1].vval.v_dict);
1520# endif
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001521 else
1522 {
1523 semsg(_(e_invalid_argument_str), event);
1524 return;
1525 }
1526# endif
1527}
1528
1529 void
1530f_test_settime(typval_T *argvars, typval_T *rettv UNUSED)
1531{
1532 if (in_vim9script() && check_for_number_arg(argvars, 0) == FAIL)
1533 return;
1534
1535 time_for_testing = (time_t)tv_get_number(&argvars[0]);
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001536}
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001537
1538#endif // defined(FEAT_EVAL)