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