blob: 23ab98f5c1b04de7da5a3287b74038f45b76f116 [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;
745 msg_col = 0;
746 need_wait_return = FALSE;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200747 emsg_on_display = FALSE;
Bram Moolenaar28ee8922020-10-28 20:20:00 +0100748 msg_scrolled = 0;
749 lines_left = Rows;
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200750 VIM_CLEAR(emsg_assert_fails_msg);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200751 set_vim_var_string(VV_ERRMSG, NULL, 0);
Bram Moolenaar44d66522020-09-06 22:26:57 +0200752 if (wrong_arg_msg != NULL)
753 emsg(_(wrong_arg_msg));
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200754}
755
756/*
757 * "assert_false(actual[, msg])" function
758 */
759 void
760f_assert_false(typval_T *argvars, typval_T *rettv)
761{
762 rettv->vval.v_number = assert_bool(argvars, FALSE);
763}
764
765 static int
766assert_inrange(typval_T *argvars)
767{
768 garray_T ga;
769 int error = FALSE;
770 char_u *tofree;
771 char msg[200];
772 char_u numbuf[NUMBUFLEN];
773
774#ifdef FEAT_FLOAT
775 if (argvars[0].v_type == VAR_FLOAT
776 || argvars[1].v_type == VAR_FLOAT
777 || argvars[2].v_type == VAR_FLOAT)
778 {
779 float_T flower = tv_get_float(&argvars[0]);
780 float_T fupper = tv_get_float(&argvars[1]);
781 float_T factual = tv_get_float(&argvars[2]);
782
783 if (factual < flower || factual > fupper)
784 {
785 prepare_assert_error(&ga);
786 if (argvars[3].v_type != VAR_UNKNOWN)
787 {
788 ga_concat(&ga, tv2string(&argvars[3], &tofree, numbuf, 0));
789 vim_free(tofree);
790 }
791 else
792 {
793 vim_snprintf(msg, 200, "Expected range %g - %g, but got %g",
794 flower, fupper, factual);
795 ga_concat(&ga, (char_u *)msg);
796 }
797 assert_error(&ga);
798 ga_clear(&ga);
799 return 1;
800 }
801 }
802 else
803#endif
804 {
805 varnumber_T lower = tv_get_number_chk(&argvars[0], &error);
806 varnumber_T upper = tv_get_number_chk(&argvars[1], &error);
807 varnumber_T actual = tv_get_number_chk(&argvars[2], &error);
808
809 if (error)
810 return 0;
811 if (actual < lower || actual > upper)
812 {
813 prepare_assert_error(&ga);
814 if (argvars[3].v_type != VAR_UNKNOWN)
815 {
816 ga_concat(&ga, tv2string(&argvars[3], &tofree, numbuf, 0));
817 vim_free(tofree);
818 }
819 else
820 {
821 vim_snprintf(msg, 200, "Expected range %ld - %ld, but got %ld",
822 (long)lower, (long)upper, (long)actual);
823 ga_concat(&ga, (char_u *)msg);
824 }
825 assert_error(&ga);
826 ga_clear(&ga);
827 return 1;
828 }
829 }
830 return 0;
831}
832
833/*
834 * "assert_inrange(lower, upper[, msg])" function
835 */
836 void
837f_assert_inrange(typval_T *argvars, typval_T *rettv)
838{
Yegappan Lakshmanana764e732021-07-25 15:57:32 +0200839 if (check_for_float_or_nr_arg(argvars, 0) == FAIL
840 || check_for_float_or_nr_arg(argvars, 1) == FAIL
841 || check_for_float_or_nr_arg(argvars, 2) == FAIL
842 || check_for_opt_string_arg(argvars, 3) == FAIL)
843 return;
844
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200845 rettv->vval.v_number = assert_inrange(argvars);
846}
847
848/*
849 * "assert_match(pattern, actual[, msg])" function
850 */
851 void
852f_assert_match(typval_T *argvars, typval_T *rettv)
853{
854 rettv->vval.v_number = assert_match_common(argvars, ASSERT_MATCH);
855}
856
857/*
858 * "assert_notmatch(pattern, actual[, msg])" function
859 */
860 void
861f_assert_notmatch(typval_T *argvars, typval_T *rettv)
862{
863 rettv->vval.v_number = assert_match_common(argvars, ASSERT_NOTMATCH);
864}
865
866/*
867 * "assert_report(msg)" function
868 */
869 void
870f_assert_report(typval_T *argvars, typval_T *rettv)
871{
872 garray_T ga;
873
Yegappan Lakshmananc72bdd22021-07-11 19:44:18 +0200874 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
875 return;
876
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200877 prepare_assert_error(&ga);
878 ga_concat(&ga, tv_get_string(&argvars[0]));
879 assert_error(&ga);
880 ga_clear(&ga);
881 rettv->vval.v_number = 1;
882}
883
884/*
885 * "assert_true(actual[, msg])" function
886 */
887 void
888f_assert_true(typval_T *argvars, typval_T *rettv)
889{
890 rettv->vval.v_number = assert_bool(argvars, TRUE);
891}
892
893/*
894 * "test_alloc_fail(id, countdown, repeat)" function
895 */
896 void
897f_test_alloc_fail(typval_T *argvars, typval_T *rettv UNUSED)
898{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200899 if (in_vim9script()
900 && (check_for_number_arg(argvars, 0) == FAIL
901 || check_for_number_arg(argvars, 1) == FAIL
902 || check_for_number_arg(argvars, 2) == FAIL))
903 return;
904
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200905 if (argvars[0].v_type != VAR_NUMBER
906 || argvars[0].vval.v_number <= 0
907 || argvars[1].v_type != VAR_NUMBER
908 || argvars[1].vval.v_number < 0
909 || argvars[2].v_type != VAR_NUMBER)
Bram Moolenaar436b5ad2021-12-31 22:49:24 +0000910 emsg(_(e_invalid_argument));
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200911 else
912 {
913 alloc_fail_id = argvars[0].vval.v_number;
914 if (alloc_fail_id >= aid_last)
Bram Moolenaar436b5ad2021-12-31 22:49:24 +0000915 emsg(_(e_invalid_argument));
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200916 alloc_fail_countdown = argvars[1].vval.v_number;
917 alloc_fail_repeat = argvars[2].vval.v_number;
918 did_outofmem_msg = FALSE;
919 }
920}
921
922/*
923 * "test_autochdir()"
924 */
925 void
926f_test_autochdir(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
927{
928#if defined(FEAT_AUTOCHDIR)
929 test_autochdir = TRUE;
930#endif
931}
932
933/*
934 * "test_feedinput()"
935 */
936 void
937f_test_feedinput(typval_T *argvars, typval_T *rettv UNUSED)
938{
939#ifdef USE_INPUT_BUF
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200940 char_u *val;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200941
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200942 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
943 return;
944
945 val = tv_get_string_chk(&argvars[0]);
Bram Moolenaar272ca952020-01-28 20:49:11 +0100946# ifdef VIMDLL
947 // this doesn't work in the console
948 if (!gui.in_use)
949 return;
950# endif
951
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200952 if (val != NULL)
953 {
954 trash_input_buf();
955 add_to_input_buf_csi(val, (int)STRLEN(val));
956 }
957#endif
958}
959
960/*
961 * "test_getvalue({name})" function
962 */
963 void
964f_test_getvalue(typval_T *argvars, typval_T *rettv)
965{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200966 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
967 return;
968
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200969 if (argvars[0].v_type != VAR_STRING)
Bram Moolenaar436b5ad2021-12-31 22:49:24 +0000970 emsg(_(e_invalid_argument));
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200971 else
972 {
973 char_u *name = tv_get_string(&argvars[0]);
974
975 if (STRCMP(name, (char_u *)"need_fileinfo") == 0)
976 rettv->vval.v_number = need_fileinfo;
977 else
Bram Moolenaar436b5ad2021-12-31 22:49:24 +0000978 semsg(_(e_invalid_argument_str), name);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200979 }
980}
981
982/*
983 * "test_option_not_set({name})" function
984 */
985 void
986f_test_option_not_set(typval_T *argvars, typval_T *rettv UNUSED)
987{
988 char_u *name = (char_u *)"";
989
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +0200990 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
991 return;
992
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200993 if (argvars[0].v_type != VAR_STRING)
Bram Moolenaar436b5ad2021-12-31 22:49:24 +0000994 emsg(_(e_invalid_argument));
Bram Moolenaarecaa70e2019-07-14 14:55:39 +0200995 else
996 {
997 name = tv_get_string(&argvars[0]);
998 if (reset_option_was_set(name) == FAIL)
Bram Moolenaar436b5ad2021-12-31 22:49:24 +0000999 semsg(_(e_invalid_argument_str), name);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001000 }
1001}
1002
1003/*
1004 * "test_override({name}, {val})" function
1005 */
1006 void
1007f_test_override(typval_T *argvars, typval_T *rettv UNUSED)
1008{
1009 char_u *name = (char_u *)"";
1010 int val;
1011 static int save_starting = -1;
1012
Yegappan Lakshmanan1a71d312021-07-15 12:49:58 +02001013 if (in_vim9script()
1014 && (check_for_string_arg(argvars, 0) == FAIL
1015 || check_for_number_arg(argvars, 1) == FAIL))
1016 return;
1017
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001018 if (argvars[0].v_type != VAR_STRING
1019 || (argvars[1].v_type) != VAR_NUMBER)
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00001020 emsg(_(e_invalid_argument));
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001021 else
1022 {
1023 name = tv_get_string(&argvars[0]);
1024 val = (int)tv_get_number(&argvars[1]);
1025
1026 if (STRCMP(name, (char_u *)"redraw") == 0)
1027 disable_redraw_for_testing = val;
1028 else if (STRCMP(name, (char_u *)"redraw_flag") == 0)
1029 ignore_redraw_flag_for_testing = val;
1030 else if (STRCMP(name, (char_u *)"char_avail") == 0)
1031 disable_char_avail_for_testing = val;
1032 else if (STRCMP(name, (char_u *)"starting") == 0)
1033 {
1034 if (val)
1035 {
1036 if (save_starting < 0)
1037 save_starting = starting;
1038 starting = 0;
1039 }
1040 else
1041 {
1042 starting = save_starting;
1043 save_starting = -1;
1044 }
1045 }
1046 else if (STRCMP(name, (char_u *)"nfa_fail") == 0)
1047 nfa_fail_for_testing = val;
1048 else if (STRCMP(name, (char_u *)"no_query_mouse") == 0)
1049 no_query_mouse_for_testing = val;
1050 else if (STRCMP(name, (char_u *)"no_wait_return") == 0)
1051 no_wait_return = val;
Bram Moolenaarb340bae2020-06-15 19:51:56 +02001052 else if (STRCMP(name, (char_u *)"ui_delay") == 0)
1053 ui_delay_for_testing = val;
Bram Moolenaar0c0eddd2020-06-13 15:47:25 +02001054 else if (STRCMP(name, (char_u *)"term_props") == 0)
1055 reset_term_props_on_termresponse = val;
ichizokae1bd872022-01-20 14:57:29 +00001056 else if (STRCMP(name, (char_u *)"vterm_title") == 0)
1057 disable_vterm_title_for_testing = val;
Bram Moolenaarf52f0602021-03-10 21:26:37 +01001058 else if (STRCMP(name, (char_u *)"uptime") == 0)
1059 override_sysinfo_uptime = val;
Bram Moolenaar3e4fa3d2022-01-13 22:05:09 +00001060 else if (STRCMP(name, (char_u *)"autoload") == 0)
1061 override_autoload = val;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001062 else if (STRCMP(name, (char_u *)"ALL") == 0)
1063 {
1064 disable_char_avail_for_testing = FALSE;
1065 disable_redraw_for_testing = FALSE;
1066 ignore_redraw_flag_for_testing = FALSE;
1067 nfa_fail_for_testing = FALSE;
1068 no_query_mouse_for_testing = FALSE;
Bram Moolenaarb340bae2020-06-15 19:51:56 +02001069 ui_delay_for_testing = 0;
Bram Moolenaar0c0eddd2020-06-13 15:47:25 +02001070 reset_term_props_on_termresponse = FALSE;
Bram Moolenaarf52f0602021-03-10 21:26:37 +01001071 override_sysinfo_uptime = -1;
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001072 if (save_starting >= 0)
1073 {
1074 starting = save_starting;
1075 save_starting = -1;
1076 }
1077 }
1078 else
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00001079 semsg(_(e_invalid_argument_str), name);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001080 }
1081}
1082
1083/*
1084 * "test_refcount({expr})" function
1085 */
1086 void
1087f_test_refcount(typval_T *argvars, typval_T *rettv)
1088{
1089 int retval = -1;
1090
1091 switch (argvars[0].v_type)
1092 {
1093 case VAR_UNKNOWN:
Bram Moolenaar4c683752020-04-05 21:38:23 +02001094 case VAR_ANY:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001095 case VAR_VOID:
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001096 case VAR_NUMBER:
Bram Moolenaar9b4a15d2020-01-11 16:05:23 +01001097 case VAR_BOOL:
Bram Moolenaar8a7d6542020-01-26 15:56:19 +01001098 case VAR_FLOAT:
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001099 case VAR_SPECIAL:
1100 case VAR_STRING:
Bram Moolenaarf18332f2021-05-07 17:55:55 +02001101 case VAR_INSTR:
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001102 break;
1103 case VAR_JOB:
1104#ifdef FEAT_JOB_CHANNEL
1105 if (argvars[0].vval.v_job != NULL)
1106 retval = argvars[0].vval.v_job->jv_refcount - 1;
1107#endif
1108 break;
1109 case VAR_CHANNEL:
1110#ifdef FEAT_JOB_CHANNEL
1111 if (argvars[0].vval.v_channel != NULL)
1112 retval = argvars[0].vval.v_channel->ch_refcount - 1;
1113#endif
1114 break;
1115 case VAR_FUNC:
1116 if (argvars[0].vval.v_string != NULL)
1117 {
1118 ufunc_T *fp;
1119
Bram Moolenaard9d2fd02022-01-13 21:15:21 +00001120 fp = find_func(argvars[0].vval.v_string, FALSE);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001121 if (fp != NULL)
1122 retval = fp->uf_refcount;
1123 }
1124 break;
1125 case VAR_PARTIAL:
1126 if (argvars[0].vval.v_partial != NULL)
1127 retval = argvars[0].vval.v_partial->pt_refcount - 1;
1128 break;
1129 case VAR_BLOB:
1130 if (argvars[0].vval.v_blob != NULL)
1131 retval = argvars[0].vval.v_blob->bv_refcount - 1;
1132 break;
1133 case VAR_LIST:
1134 if (argvars[0].vval.v_list != NULL)
1135 retval = argvars[0].vval.v_list->lv_refcount - 1;
1136 break;
1137 case VAR_DICT:
1138 if (argvars[0].vval.v_dict != NULL)
1139 retval = argvars[0].vval.v_dict->dv_refcount - 1;
1140 break;
1141 }
1142
1143 rettv->v_type = VAR_NUMBER;
1144 rettv->vval.v_number = retval;
1145
1146}
1147
1148/*
1149 * "test_garbagecollect_now()" function
1150 */
1151 void
1152f_test_garbagecollect_now(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
1153{
Bram Moolenaar0d6f5d92019-12-05 21:33:15 +01001154 // This is dangerous, any Lists and Dicts used internally may be freed
1155 // while still in use.
Bram Moolenaarb3d83982022-01-27 19:59:47 +00001156 if (!get_vim_var_nr(VV_TESTING))
1157 emsg(_(e_calling_test_garbagecollect_now_while_v_testing_is_not_set));
1158 else
1159 garbage_collect(TRUE);
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001160}
1161
1162/*
1163 * "test_garbagecollect_soon()" function
1164 */
1165 void
1166f_test_garbagecollect_soon(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
1167{
1168 may_garbage_collect = TRUE;
1169}
1170
1171/*
1172 * "test_ignore_error()" function
1173 */
1174 void
1175f_test_ignore_error(typval_T *argvars, typval_T *rettv UNUSED)
1176{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001177 if (in_vim9script() && check_for_string_arg(argvars, 0) == FAIL)
1178 return;
1179
Yegappan Lakshmanan5b739922021-07-10 13:15:41 +02001180 if (argvars[0].v_type != VAR_STRING)
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00001181 emsg(_(e_invalid_argument));
Yegappan Lakshmanan5b739922021-07-10 13:15:41 +02001182 else
1183 ignore_error_for_testing(tv_get_string(&argvars[0]));
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001184}
1185
1186 void
1187f_test_null_blob(typval_T *argvars UNUSED, typval_T *rettv)
1188{
1189 rettv->v_type = VAR_BLOB;
1190 rettv->vval.v_blob = NULL;
1191}
1192
1193#ifdef FEAT_JOB_CHANNEL
1194 void
1195f_test_null_channel(typval_T *argvars UNUSED, typval_T *rettv)
1196{
1197 rettv->v_type = VAR_CHANNEL;
1198 rettv->vval.v_channel = NULL;
1199}
1200#endif
1201
1202 void
1203f_test_null_dict(typval_T *argvars UNUSED, typval_T *rettv)
1204{
1205 rettv_dict_set(rettv, NULL);
1206}
1207
1208#ifdef FEAT_JOB_CHANNEL
1209 void
1210f_test_null_job(typval_T *argvars UNUSED, typval_T *rettv)
1211{
1212 rettv->v_type = VAR_JOB;
1213 rettv->vval.v_job = NULL;
1214}
1215#endif
1216
1217 void
1218f_test_null_list(typval_T *argvars UNUSED, typval_T *rettv)
1219{
1220 rettv_list_set(rettv, NULL);
1221}
1222
1223 void
Bram Moolenaare69f6d02020-04-01 22:11:01 +02001224f_test_null_function(typval_T *argvars UNUSED, typval_T *rettv)
1225{
1226 rettv->v_type = VAR_FUNC;
1227 rettv->vval.v_string = NULL;
1228}
1229
1230 void
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001231f_test_null_partial(typval_T *argvars UNUSED, typval_T *rettv)
1232{
1233 rettv->v_type = VAR_PARTIAL;
1234 rettv->vval.v_partial = NULL;
1235}
1236
1237 void
1238f_test_null_string(typval_T *argvars UNUSED, typval_T *rettv)
1239{
1240 rettv->v_type = VAR_STRING;
1241 rettv->vval.v_string = NULL;
1242}
1243
Bram Moolenaar8ed04582020-02-22 19:07:28 +01001244 void
1245f_test_unknown(typval_T *argvars UNUSED, typval_T *rettv)
1246{
1247 rettv->v_type = VAR_UNKNOWN;
1248}
1249
1250 void
1251f_test_void(typval_T *argvars UNUSED, typval_T *rettv)
1252{
1253 rettv->v_type = VAR_VOID;
1254}
1255
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001256 void
1257f_test_setmouse(typval_T *argvars, typval_T *rettv UNUSED)
1258{
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02001259 if (in_vim9script()
1260 && (check_for_number_arg(argvars, 0) == FAIL
1261 || check_for_number_arg(argvars, 1) == FAIL))
1262 return;
1263
Yegappan Lakshmanan7237cab2021-06-22 19:52:27 +02001264 if (argvars[0].v_type != VAR_NUMBER || (argvars[1].v_type) != VAR_NUMBER)
1265 {
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00001266 emsg(_(e_invalid_argument));
Yegappan Lakshmanan7237cab2021-06-22 19:52:27 +02001267 return;
1268 }
1269
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001270 mouse_row = (time_t)tv_get_number(&argvars[0]) - 1;
1271 mouse_col = (time_t)tv_get_number(&argvars[1]) - 1;
1272}
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001273
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001274# ifdef FEAT_GUI
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001275 static int
1276test_gui_drop_files(dict_T *args UNUSED)
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001277{
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001278# if defined(HAVE_DROP_FILE)
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001279 int row;
1280 int col;
1281 int_u mods;
1282 char_u **fnames;
1283 int count = 0;
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001284 typval_T t;
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001285 list_T *l;
1286 listitem_T *li;
1287
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +01001288 if (!dict_has_key(args, "files")
1289 || !dict_has_key(args, "row")
1290 || !dict_has_key(args, "col")
1291 || !dict_has_key(args, "modifiers"))
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 (void)dict_get_tv(args, (char_u *)"files", &t);
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001295 row = (int)dict_get_number(args, (char_u *)"row");
1296 col = (int)dict_get_number(args, (char_u *)"col");
1297 mods = (int)dict_get_number(args, (char_u *)"modifiers");
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001298
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001299 if (t.v_type != VAR_LIST || list_len(t.vval.v_list) == 0)
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001300 return FALSE;
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001301
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001302 l = t.vval.v_list;
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001303 fnames = ALLOC_MULT(char_u *, list_len(l));
1304 if (fnames == NULL)
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001305 return FALSE;
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001306
1307 FOR_ALL_LIST_ITEMS(l, li)
1308 {
1309 // ignore non-string items
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001310 if (li->li_tv.v_type != VAR_STRING
1311 || li->li_tv.vval.v_string == NULL)
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001312 continue;
1313
1314 fnames[count] = vim_strsave(li->li_tv.vval.v_string);
1315 if (fnames[count] == NULL)
1316 {
1317 while (--count >= 0)
1318 vim_free(fnames[count]);
1319 vim_free(fnames);
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001320 return FALSE;
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001321 }
1322 count++;
1323 }
1324
1325 if (count > 0)
1326 gui_handle_drop(TEXT_X(col - 1), TEXT_Y(row - 1), mods, fnames, count);
1327 else
1328 vim_free(fnames);
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001329# endif
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001330
1331 return TRUE;
1332}
1333
Bram Moolenaar4e3b3182022-02-01 10:16:00 +00001334#if defined(FIND_REPLACE_DIALOG)
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001335 static int
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001336test_gui_find_repl(dict_T *args)
1337{
1338 int flags;
1339 char_u *find_text;
1340 char_u *repl_text;
1341 int forward;
1342 int retval;
1343
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +01001344 if (!dict_has_key(args, "find_text")
1345 || !dict_has_key(args, "repl_text")
1346 || !dict_has_key(args, "flags")
1347 || !dict_has_key(args, "forward"))
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001348 return FALSE;
1349
1350 find_text = dict_get_string(args, (char_u *)"find_text", TRUE);
1351 repl_text = dict_get_string(args, (char_u *)"repl_text", TRUE);
1352 flags = (int)dict_get_number(args, (char_u *)"flags");
1353 forward = (int)dict_get_number(args, (char_u *)"forward");
1354
1355 retval = gui_do_findrepl(flags, find_text, repl_text, forward);
1356 vim_free(find_text);
1357 vim_free(repl_text);
1358
1359 return retval;
1360}
Bram Moolenaar4e3b3182022-02-01 10:16:00 +00001361#endif
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001362
1363 static int
1364test_gui_mouse_event(dict_T *args)
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001365{
1366 int button;
1367 int row;
1368 int col;
1369 int repeated_click;
1370 int_u mods;
Ernie Raelc4cb5442022-04-03 15:47:28 +01001371 int move;
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001372
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +01001373 if (!dict_has_key(args, "row")
1374 || !dict_has_key(args, "col"))
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001375 return FALSE;
1376
Ernie Raelc4cb5442022-04-03 15:47:28 +01001377 // Note: "move" is optional, requires fewer arguments
1378 move = (int)dict_get_bool(args, (char_u *)"move", FALSE);
1379
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +01001380 if (!move && (!dict_has_key(args, "button")
1381 || !dict_has_key(args, "multiclick")
1382 || !dict_has_key(args, "modifiers")))
Ernie Raelc4cb5442022-04-03 15:47:28 +01001383 return FALSE;
1384
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001385 row = (int)dict_get_number(args, (char_u *)"row");
1386 col = (int)dict_get_number(args, (char_u *)"col");
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001387
Ernie Raelc4cb5442022-04-03 15:47:28 +01001388 if (move)
Bram Moolenaar7add8d32022-05-16 15:27:46 +01001389 {
1390 if (dict_get_bool(args, (char_u *)"cell", FALSE))
1391 {
1392 // click in the middle of the character cell
1393 row = row * gui.char_height + gui.char_height / 2;
1394 col = col * gui.char_width + gui.char_width / 2;
1395 }
Ernie Raelc4cb5442022-04-03 15:47:28 +01001396 gui_mouse_moved(col, row);
Bram Moolenaar7add8d32022-05-16 15:27:46 +01001397 }
Ernie Raelc4cb5442022-04-03 15:47:28 +01001398 else
1399 {
1400 button = (int)dict_get_number(args, (char_u *)"button");
1401 repeated_click = (int)dict_get_number(args, (char_u *)"multiclick");
1402 mods = (int)dict_get_number(args, (char_u *)"modifiers");
1403
LemonBoyc27747e2022-05-07 12:25:40 +01001404 // Reset the scroll values to known values.
1405 // XXX: Remove this when/if the scroll step is made configurable.
1406 mouse_set_hor_scroll_step(6);
1407 mouse_set_vert_scroll_step(3);
1408
Ernie Raelc4cb5442022-04-03 15:47:28 +01001409 gui_send_mouse_event(button, TEXT_X(col - 1), TEXT_Y(row - 1),
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001410 repeated_click, mods);
Ernie Raelc4cb5442022-04-03 15:47:28 +01001411 }
1412
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001413 return TRUE;
1414}
1415
1416 static int
Yegappan Lakshmanan9e0208f2022-01-31 17:40:55 +00001417test_gui_scrollbar(dict_T *args)
1418{
1419 char_u *which;
1420 long value;
1421 int dragging;
1422 scrollbar_T *sb = NULL;
1423
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +01001424 if (!dict_has_key(args, "which")
1425 || !dict_has_key(args, "value")
1426 || !dict_has_key(args, "dragging"))
Yegappan Lakshmanan9e0208f2022-01-31 17:40:55 +00001427 return FALSE;
1428
1429 which = dict_get_string(args, (char_u *)"which", FALSE);
1430 value = (long)dict_get_number(args, (char_u *)"value");
1431 dragging = (int)dict_get_number(args, (char_u *)"dragging");
1432
1433 if (STRCMP(which, "left") == 0)
1434 sb = &curwin->w_scrollbars[SBAR_LEFT];
1435 else if (STRCMP(which, "right") == 0)
1436 sb = &curwin->w_scrollbars[SBAR_RIGHT];
1437 else if (STRCMP(which, "hor") == 0)
1438 sb = &gui.bottom_sbar;
1439 if (sb == NULL)
1440 {
1441 semsg(_(e_invalid_argument_str), which);
1442 return FALSE;
1443 }
1444 gui_drag_scrollbar(sb, value, dragging);
1445# ifndef USE_ON_FLY_SCROLL
1446 // need to loop through normal_cmd() to handle the scroll events
1447 exec_normal(FALSE, TRUE, FALSE);
1448# endif
1449
1450 return TRUE;
1451}
1452
1453 static int
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001454test_gui_tabline_event(dict_T *args UNUSED)
1455{
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001456# ifdef FEAT_GUI_TABLINE
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001457 int tabnr;
1458
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +01001459 if (!dict_has_key(args, "tabnr"))
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001460 return FALSE;
1461
1462 tabnr = (int)dict_get_number(args, (char_u *)"tabnr");
1463
1464 return send_tabline_event(tabnr);
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001465# else
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001466 return FALSE;
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001467# endif
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001468}
1469
1470 static int
1471test_gui_tabmenu_event(dict_T *args UNUSED)
1472{
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001473# ifdef FEAT_GUI_TABLINE
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001474 int tabnr;
1475 int item;
1476
Yegappan Lakshmanan4829c1c2022-04-04 15:16:54 +01001477 if (!dict_has_key(args, "tabnr")
1478 || !dict_has_key(args, "item"))
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001479 return FALSE;
1480
1481 tabnr = (int)dict_get_number(args, (char_u *)"tabnr");
1482 item = (int)dict_get_number(args, (char_u *)"item");
1483
1484 send_tabline_menu_event(tabnr, item);
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001485# endif
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001486 return TRUE;
1487}
1488# endif
1489
1490 void
1491f_test_gui_event(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
1492{
1493# ifdef FEAT_GUI
1494 char_u *event;
1495
1496 rettv->v_type = VAR_BOOL;
1497 rettv->vval.v_number = FALSE;
1498
1499 if (check_for_string_arg(argvars, 0) == FAIL
1500 || check_for_dict_arg(argvars, 1) == FAIL
1501 || argvars[1].vval.v_dict == NULL)
1502 return;
1503
1504 event = tv_get_string(&argvars[0]);
1505 if (STRCMP(event, "dropfiles") == 0)
1506 rettv->vval.v_number = test_gui_drop_files(argvars[1].vval.v_dict);
Bram Moolenaar4e3b3182022-02-01 10:16:00 +00001507# if defined(FIND_REPLACE_DIALOG)
Yegappan Lakshmananec3637c2022-01-30 18:01:24 +00001508 else if (STRCMP(event, "findrepl") == 0)
1509 rettv->vval.v_number = test_gui_find_repl(argvars[1].vval.v_dict);
Bram Moolenaar4e3b3182022-02-01 10:16:00 +00001510# endif
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001511 else if (STRCMP(event, "mouse") == 0)
1512 rettv->vval.v_number = test_gui_mouse_event(argvars[1].vval.v_dict);
Yegappan Lakshmanan9e0208f2022-01-31 17:40:55 +00001513 else if (STRCMP(event, "scrollbar") == 0)
1514 rettv->vval.v_number = test_gui_scrollbar(argvars[1].vval.v_dict);
Yegappan Lakshmanan06011e12022-01-30 12:37:29 +00001515 else if (STRCMP(event, "tabline") == 0)
1516 rettv->vval.v_number = test_gui_tabline_event(argvars[1].vval.v_dict);
1517 else if (STRCMP(event, "tabmenu") == 0)
1518 rettv->vval.v_number = test_gui_tabmenu_event(argvars[1].vval.v_dict);
1519 else
1520 {
1521 semsg(_(e_invalid_argument_str), event);
1522 return;
1523 }
1524# endif
1525}
1526
1527 void
1528f_test_settime(typval_T *argvars, typval_T *rettv UNUSED)
1529{
1530 if (in_vim9script() && check_for_number_arg(argvars, 0) == FAIL)
1531 return;
1532
1533 time_for_testing = (time_t)tv_get_number(&argvars[0]);
Yegappan Lakshmanan18d46582021-06-23 20:46:52 +02001534}
Bram Moolenaarecaa70e2019-07-14 14:55:39 +02001535
1536#endif // defined(FEAT_EVAL)