blob: 251790bc5710e20aa2465c8fccb9595a84380a47 [file] [log] [blame]
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001" Test various aspects of the Vim9 script language.
2
3source check.vim
Bram Moolenaarad304702020-09-06 18:22:53 +02004source term_util.vim
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02005source view_util.vim
Bram Moolenaar04b12692020-05-04 23:24:44 +02006source vim9.vim
Bram Moolenaar47e7d702020-07-05 18:18:42 +02007source screendump.vim
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02008
9func Test_def_basic()
10 def SomeFunc(): string
11 return 'yes'
12 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +020013 call SomeFunc()->assert_equal('yes')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +020014endfunc
15
Bram Moolenaar2b9b17e2020-10-13 18:38:11 +020016func Test_compiling_error()
17 " use a terminal to see the whole error message
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +020018 CheckRunVimInTerminal
19
Bram Moolenaar2b9b17e2020-10-13 18:38:11 +020020 call TestCompilingError()
21endfunc
22
23def TestCompilingError()
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +020024 var lines =<< trim END
25 vim9script
26 def Fails()
27 echo nothing
28 enddef
29 defcompile
30 END
31 call writefile(lines, 'XTest_compile_error')
32 var buf = RunVimInTerminal('-S XTest_compile_error',
Bram Moolenaare0de1712020-12-02 17:36:54 +010033 {rows: 10, wait_for_ruler: 0})
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +020034 var text = ''
35 for loop in range(100)
36 text = ''
37 for i in range(1, 9)
38 text ..= term_getline(buf, i)
39 endfor
Bram Moolenaar18dc3552020-11-22 14:24:00 +010040 if text =~ 'Variable not found: nothing'
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +020041 break
42 endif
43 sleep 20m
44 endfor
45 assert_match('Error detected while compiling command line.*Fails.*Variable not found: nothing', text)
46
47 # clean up
48 call StopVimInTerminal(buf)
49 call delete('XTest_compile_error')
50enddef
51
Bram Moolenaar0ba48e82020-11-17 18:23:19 +010052def CallRecursive(n: number): number
53 return CallRecursive(n + 1)
54enddef
55
56def CallMapRecursive(l: list<number>): number
Bram Moolenaar2949cfd2020-12-31 21:28:47 +010057 return map(l, (_, v) => CallMapRecursive([v]))[0]
Bram Moolenaar0ba48e82020-11-17 18:23:19 +010058enddef
59
60def Test_funcdepth_error()
61 set maxfuncdepth=10
62
63 var caught = false
64 try
65 CallRecursive(1)
66 catch /E132:/
67 caught = true
68 endtry
69 assert_true(caught)
70
71 caught = false
72 try
73 CallMapRecursive([1])
74 catch /E132:/
75 caught = true
76 endtry
77 assert_true(caught)
78
79 set maxfuncdepth&
80enddef
81
Bram Moolenaar5178b1b2021-01-01 18:43:51 +010082def Test_endfunc_enddef()
83 var lines =<< trim END
84 def Test()
85 echo 'test'
86 endfunc
87 enddef
88 END
89 CheckScriptFailure(lines, 'E1151:', 3)
90
91 lines =<< trim END
92 def Test()
93 func Nested()
94 echo 'test'
95 enddef
96 enddef
97 END
98 CheckScriptFailure(lines, 'E1152:', 4)
99enddef
100
Bram Moolenaarb8ba9b92021-01-01 18:54:34 +0100101def Test_missing_endfunc_enddef()
102 var lines =<< trim END
103 vim9script
104 def Test()
105 echo 'test'
106 endef
107 END
108 CheckScriptFailure(lines, 'E1057:', 2)
109
110 lines =<< trim END
111 vim9script
112 func Some()
113 echo 'test'
114 enfffunc
115 END
116 CheckScriptFailure(lines, 'E126:', 2)
117enddef
118
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200119def ReturnString(): string
120 return 'string'
121enddef
122
123def ReturnNumber(): number
124 return 123
125enddef
126
127let g:notNumber = 'string'
128
129def ReturnGlobal(): number
130 return g:notNumber
131enddef
132
133def Test_return_something()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200134 ReturnString()->assert_equal('string')
135 ReturnNumber()->assert_equal(123)
Bram Moolenaar5e654232020-09-16 15:22:00 +0200136 assert_fails('ReturnGlobal()', 'E1012: Type mismatch; expected number but got string', '', 1, 'ReturnGlobal')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200137enddef
138
Bram Moolenaarefd88552020-06-18 20:50:10 +0200139def Test_missing_return()
140 CheckDefFailure(['def Missing(): number',
141 ' if g:cond',
142 ' echo "no return"',
143 ' else',
144 ' return 0',
145 ' endif'
146 'enddef'], 'E1027:')
147 CheckDefFailure(['def Missing(): number',
148 ' if g:cond',
149 ' return 1',
150 ' else',
151 ' echo "no return"',
152 ' endif'
153 'enddef'], 'E1027:')
154 CheckDefFailure(['def Missing(): number',
155 ' if g:cond',
156 ' return 1',
157 ' else',
158 ' return 2',
159 ' endif'
160 ' return 3'
161 'enddef'], 'E1095:')
162enddef
163
Bram Moolenaar403dc312020-10-17 19:29:51 +0200164def Test_return_bool()
165 var lines =<< trim END
166 vim9script
167 def MenuFilter(id: number, key: string): bool
168 return popup_filter_menu(id, key)
169 enddef
170 def YesnoFilter(id: number, key: string): bool
171 return popup_filter_yesno(id, key)
172 enddef
173 defcompile
174 END
175 CheckScriptSuccess(lines)
176enddef
177
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200178let s:nothing = 0
179def ReturnNothing()
180 s:nothing = 1
181 if true
182 return
183 endif
184 s:nothing = 2
185enddef
186
187def Test_return_nothing()
188 ReturnNothing()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200189 s:nothing->assert_equal(1)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200190enddef
191
192func Increment()
193 let g:counter += 1
194endfunc
195
196def Test_call_ufunc_count()
197 g:counter = 1
198 Increment()
199 Increment()
200 Increment()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200201 # works with and without :call
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200202 g:counter->assert_equal(4)
203 eval g:counter->assert_equal(4)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200204 unlet g:counter
205enddef
206
207def MyVarargs(arg: string, ...rest: list<string>): string
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200208 var res = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200209 for s in rest
210 res ..= ',' .. s
211 endfor
212 return res
213enddef
214
215def Test_call_varargs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200216 MyVarargs('one')->assert_equal('one')
217 MyVarargs('one', 'two')->assert_equal('one,two')
218 MyVarargs('one', 'two', 'three')->assert_equal('one,two,three')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200219enddef
220
221def MyDefaultArgs(name = 'string'): string
222 return name
223enddef
224
Bram Moolenaare30f64b2020-07-15 19:48:20 +0200225def MyDefaultSecond(name: string, second: bool = true): string
226 return second ? name : 'none'
227enddef
228
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200229def Test_call_default_args()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200230 MyDefaultArgs()->assert_equal('string')
231 MyDefaultArgs('one')->assert_equal('one')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200232 assert_fails('MyDefaultArgs("one", "two")', 'E118:', '', 3, 'Test_call_default_args')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200233
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200234 MyDefaultSecond('test')->assert_equal('test')
235 MyDefaultSecond('test', true)->assert_equal('test')
236 MyDefaultSecond('test', false)->assert_equal('none')
Bram Moolenaare30f64b2020-07-15 19:48:20 +0200237
Bram Moolenaar822ba242020-05-24 23:00:18 +0200238 CheckScriptFailure(['def Func(arg: number = asdf)', 'enddef', 'defcompile'], 'E1001:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100239 delfunc g:Func
Bram Moolenaar77072282020-09-16 17:55:40 +0200240 CheckScriptFailure(['def Func(arg: number = "text")', 'enddef', 'defcompile'], 'E1013: Argument 1: type mismatch, expected number but got string')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100241 delfunc g:Func
Bram Moolenaar04b12692020-05-04 23:24:44 +0200242enddef
243
244def Test_nested_function()
245 def Nested(arg: string): string
246 return 'nested ' .. arg
247 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200248 Nested('function')->assert_equal('nested function')
Bram Moolenaar04b12692020-05-04 23:24:44 +0200249
Bram Moolenaar0e65d3d2020-05-05 17:53:16 +0200250 CheckDefFailure(['def Nested()', 'enddef', 'Nested(66)'], 'E118:')
251 CheckDefFailure(['def Nested(arg: string)', 'enddef', 'Nested()'], 'E119:')
252
Bram Moolenaar04b12692020-05-04 23:24:44 +0200253 CheckDefFailure(['func Nested()', 'endfunc'], 'E1086:')
Bram Moolenaarbcbf4132020-08-01 22:35:13 +0200254 CheckDefFailure(['def s:Nested()', 'enddef'], 'E1075:')
255 CheckDefFailure(['def b:Nested()', 'enddef'], 'E1075:')
Bram Moolenaar8b848ca2020-09-10 22:28:01 +0200256
Bram Moolenaar54021752020-12-06 18:50:36 +0100257 var lines =<< trim END
258 def Outer()
259 def Inner()
260 # comment
261 enddef
262 def Inner()
263 enddef
264 enddef
265 END
266 CheckDefFailure(lines, 'E1073:')
267
268 lines =<< trim END
269 def Outer()
270 def Inner()
271 # comment
272 enddef
273 def! Inner()
274 enddef
275 enddef
276 END
277 CheckDefFailure(lines, 'E1117:')
278
279 # nested function inside conditional
280 # TODO: should it work when "thecount" is inside the "if"?
281 lines =<< trim END
282 vim9script
283 var thecount = 0
284 if true
285 def Test(): number
286 def TheFunc(): number
287 thecount += 1
288 return thecount
289 enddef
290 return TheFunc()
291 enddef
292 endif
293 defcompile
294 assert_equal(1, Test())
295 assert_equal(2, Test())
296 END
297 CheckScriptSuccess(lines)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200298enddef
299
Bram Moolenaaradc8e442020-12-31 18:28:18 +0100300def Test_not_nested_function()
301 echo printf('%d',
302 function('len')('xxx'))
303enddef
304
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200305func Test_call_default_args_from_func()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200306 call MyDefaultArgs()->assert_equal('string')
307 call MyDefaultArgs('one')->assert_equal('one')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200308 call assert_fails('call MyDefaultArgs("one", "two")', 'E118:', '', 3, 'Test_call_default_args_from_func')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200309endfunc
310
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200311def Test_nested_global_function()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200312 var lines =<< trim END
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200313 vim9script
314 def Outer()
315 def g:Inner(): string
316 return 'inner'
317 enddef
318 enddef
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200319 defcompile
320 Outer()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200321 g:Inner()->assert_equal('inner')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200322 delfunc g:Inner
323 Outer()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200324 g:Inner()->assert_equal('inner')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200325 delfunc g:Inner
326 Outer()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200327 g:Inner()->assert_equal('inner')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200328 delfunc g:Inner
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200329 END
330 CheckScriptSuccess(lines)
Bram Moolenaar2c79e9d2020-08-01 18:57:52 +0200331
332 lines =<< trim END
333 vim9script
334 def Outer()
335 def g:Inner(): string
336 return 'inner'
337 enddef
338 enddef
339 defcompile
340 Outer()
341 Outer()
342 END
343 CheckScriptFailure(lines, "E122:")
Bram Moolenaarcd45ed02020-12-22 17:35:54 +0100344 delfunc g:Inner
Bram Moolenaarad486a02020-08-01 23:22:18 +0200345
346 lines =<< trim END
347 vim9script
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100348 def Outer()
349 def g:Inner()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100350 echo map([1, 2, 3], (_, v) => v + 1)
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100351 enddef
352 g:Inner()
353 enddef
354 Outer()
355 END
356 CheckScriptSuccess(lines)
357 delfunc g:Inner
358
359 lines =<< trim END
360 vim9script
Bram Moolenaarad486a02020-08-01 23:22:18 +0200361 def Func()
362 echo 'script'
363 enddef
364 def Outer()
365 def Func()
366 echo 'inner'
367 enddef
368 enddef
369 defcompile
370 END
371 CheckScriptFailure(lines, "E1073:")
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200372enddef
373
Bram Moolenaar6abdcf82020-11-22 18:15:44 +0100374def DefListAll()
375 def
376enddef
377
378def DefListOne()
379 def DefListOne
380enddef
381
382def DefListMatches()
383 def /DefList
384enddef
385
386def Test_nested_def_list()
387 var funcs = split(execute('call DefListAll()'), "\n")
388 assert_true(len(funcs) > 10)
389 assert_true(funcs->index('def DefListAll()') >= 0)
390
391 funcs = split(execute('call DefListOne()'), "\n")
392 assert_equal([' def DefListOne()', '1 def DefListOne', ' enddef'], funcs)
393
394 funcs = split(execute('call DefListMatches()'), "\n")
395 assert_true(len(funcs) >= 3)
396 assert_true(funcs->index('def DefListAll()') >= 0)
397 assert_true(funcs->index('def DefListOne()') >= 0)
398 assert_true(funcs->index('def DefListMatches()') >= 0)
Bram Moolenaar54021752020-12-06 18:50:36 +0100399
400 var lines =<< trim END
401 vim9script
402 def Func()
403 def +Func+
404 enddef
405 defcompile
406 END
407 CheckScriptFailure(lines, 'E476:', 1)
Bram Moolenaar6abdcf82020-11-22 18:15:44 +0100408enddef
409
Bram Moolenaar333894b2020-08-01 18:53:07 +0200410def Test_global_local_function()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200411 var lines =<< trim END
Bram Moolenaar333894b2020-08-01 18:53:07 +0200412 vim9script
413 def g:Func(): string
414 return 'global'
415 enddef
416 def Func(): string
417 return 'local'
418 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200419 g:Func()->assert_equal('global')
420 Func()->assert_equal('local')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100421 delfunc g:Func
Bram Moolenaar333894b2020-08-01 18:53:07 +0200422 END
423 CheckScriptSuccess(lines)
Bram Moolenaar035d6e92020-08-11 22:30:42 +0200424
425 lines =<< trim END
426 vim9script
427 def g:Funcy()
428 echo 'funcy'
429 enddef
430 s:Funcy()
431 END
432 CheckScriptFailure(lines, 'E117:')
Bram Moolenaar333894b2020-08-01 18:53:07 +0200433enddef
434
Bram Moolenaar0f769812020-09-12 18:32:34 +0200435def Test_local_function_shadows_global()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200436 var lines =<< trim END
Bram Moolenaar0f769812020-09-12 18:32:34 +0200437 vim9script
438 def g:Gfunc(): string
439 return 'global'
440 enddef
441 def AnotherFunc(): number
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200442 var Gfunc = function('len')
Bram Moolenaar0f769812020-09-12 18:32:34 +0200443 return Gfunc('testing')
444 enddef
445 g:Gfunc()->assert_equal('global')
446 AnotherFunc()->assert_equal(7)
447 delfunc g:Gfunc
448 END
449 CheckScriptSuccess(lines)
450
451 lines =<< trim END
452 vim9script
453 def g:Func(): string
454 return 'global'
455 enddef
456 def AnotherFunc()
457 g:Func = function('len')
458 enddef
459 AnotherFunc()
460 END
461 CheckScriptFailure(lines, 'E705:')
462 delfunc g:Func
463enddef
464
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200465func TakesOneArg(arg)
466 echo a:arg
467endfunc
468
469def Test_call_wrong_args()
Bram Moolenaard2c61702020-09-06 15:58:36 +0200470 CheckDefFailure(['TakesOneArg()'], 'E119:')
471 CheckDefFailure(['TakesOneArg(11, 22)'], 'E118:')
472 CheckDefFailure(['bufnr(xxx)'], 'E1001:')
473 CheckScriptFailure(['def Func(Ref: func(s: string))'], 'E475:')
Bram Moolenaaree8580e2020-08-28 17:19:07 +0200474
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200475 var lines =<< trim END
Bram Moolenaaree8580e2020-08-28 17:19:07 +0200476 vim9script
477 def Func(s: string)
478 echo s
479 enddef
480 Func([])
481 END
Bram Moolenaar77072282020-09-16 17:55:40 +0200482 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected string but got list<unknown>', 5)
Bram Moolenaarb185a402020-09-18 22:42:00 +0200483
484 lines =<< trim END
485 vim9script
486 def FuncOne(nr: number)
487 echo nr
488 enddef
489 def FuncTwo()
490 FuncOne()
491 enddef
492 defcompile
493 END
494 writefile(lines, 'Xscript')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200495 var didCatch = false
Bram Moolenaarb185a402020-09-18 22:42:00 +0200496 try
497 source Xscript
498 catch
499 assert_match('E119: Not enough arguments for function: <SNR>\d\+_FuncOne', v:exception)
500 assert_match('Xscript\[8\]..function <SNR>\d\+_FuncTwo, line 1', v:throwpoint)
501 didCatch = true
502 endtry
503 assert_true(didCatch)
504
505 lines =<< trim END
506 vim9script
507 def FuncOne(nr: number)
508 echo nr
509 enddef
510 def FuncTwo()
511 FuncOne(1, 2)
512 enddef
513 defcompile
514 END
515 writefile(lines, 'Xscript')
516 didCatch = false
517 try
518 source Xscript
519 catch
520 assert_match('E118: Too many arguments for function: <SNR>\d\+_FuncOne', v:exception)
521 assert_match('Xscript\[8\]..function <SNR>\d\+_FuncTwo, line 1', v:throwpoint)
522 didCatch = true
523 endtry
524 assert_true(didCatch)
525
526 delete('Xscript')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200527enddef
528
Bram Moolenaar50824712020-12-20 21:10:17 +0100529def Test_call_funcref_wrong_args()
530 var head =<< trim END
531 vim9script
532 def Func3(a1: string, a2: number, a3: list<number>)
533 echo a1 .. a2 .. a3[0]
534 enddef
535 def Testme()
536 var funcMap: dict<func> = {func: Func3}
537 END
538 var tail =<< trim END
539 enddef
540 Testme()
541 END
542 CheckScriptSuccess(head + ["funcMap['func']('str', 123, [1, 2, 3])"] + tail)
543
544 CheckScriptFailure(head + ["funcMap['func']('str', 123)"] + tail, 'E119:')
545 CheckScriptFailure(head + ["funcMap['func']('str', 123, [1], 4)"] + tail, 'E118:')
546enddef
547
Bram Moolenaarb4d16cb2020-11-05 18:45:46 +0100548def Test_call_lambda_args()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100549 CheckDefFailure(['echo ((i) => 0)()'],
550 'E119: Not enough arguments for function: ((i) => 0)()')
Bram Moolenaarb4d16cb2020-11-05 18:45:46 +0100551
552 var lines =<< trim END
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100553 var Ref = (x: number, y: number) => x + y
Bram Moolenaarb4d16cb2020-11-05 18:45:46 +0100554 echo Ref(1, 'x')
555 END
556 CheckDefFailure(lines, 'E1013: Argument 2: type mismatch, expected number but got string')
Bram Moolenaare68b02a2021-01-03 13:09:51 +0100557
558 lines =<< trim END
559 var Ref: func(job, string, number)
560 Ref = (x, y) => 0
561 END
562 CheckDefAndScriptFailure(lines, 'E1012:')
563
564 lines =<< trim END
565 var Ref: func(job, string)
566 Ref = (x, y, z) => 0
567 END
568 CheckDefAndScriptFailure(lines, 'E1012:')
Bram Moolenaarb4d16cb2020-11-05 18:45:46 +0100569enddef
570
Bram Moolenaar709664c2020-12-12 14:33:41 +0100571def Test_lambda_uses_assigned_var()
572 CheckDefSuccess([
573 'var x: any = "aaa"'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100574 'x = filter(["bbb"], (_, v) => v =~ x)'])
Bram Moolenaar709664c2020-12-12 14:33:41 +0100575enddef
576
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200577" Default arg and varargs
578def MyDefVarargs(one: string, two = 'foo', ...rest: list<string>): string
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200579 var res = one .. ',' .. two
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200580 for s in rest
581 res ..= ',' .. s
582 endfor
583 return res
584enddef
585
586def Test_call_def_varargs()
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200587 assert_fails('MyDefVarargs()', 'E119:', '', 1, 'Test_call_def_varargs')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200588 MyDefVarargs('one')->assert_equal('one,foo')
589 MyDefVarargs('one', 'two')->assert_equal('one,two')
590 MyDefVarargs('one', 'two', 'three')->assert_equal('one,two,three')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200591 CheckDefFailure(['MyDefVarargs("one", 22)'],
Bram Moolenaar77072282020-09-16 17:55:40 +0200592 'E1013: Argument 2: type mismatch, expected string but got number')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200593 CheckDefFailure(['MyDefVarargs("one", "two", 123)'],
Bram Moolenaar77072282020-09-16 17:55:40 +0200594 'E1013: Argument 3: type mismatch, expected string but got number')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200595
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200596 var lines =<< trim END
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200597 vim9script
598 def Func(...l: list<string>)
599 echo l
600 enddef
601 Func('a', 'b', 'c')
602 END
603 CheckScriptSuccess(lines)
604
605 lines =<< trim END
606 vim9script
607 def Func(...l: list<string>)
608 echo l
609 enddef
610 Func()
611 END
612 CheckScriptSuccess(lines)
613
614 lines =<< trim END
615 vim9script
Bram Moolenaar2f8cbc42020-09-16 17:22:59 +0200616 def Func(...l: any)
617 echo l
618 enddef
619 Func(0)
620 END
621 CheckScriptSuccess(lines)
622
623 lines =<< trim END
624 vim9script
Bram Moolenaar28022722020-09-21 22:02:49 +0200625 def Func(..._l: list<string>)
626 echo _l
627 enddef
628 Func('a', 'b', 'c')
629 END
630 CheckScriptSuccess(lines)
631
632 lines =<< trim END
633 vim9script
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200634 def Func(...l: list<string>)
635 echo l
636 enddef
637 Func(1, 2, 3)
638 END
Bram Moolenaar77072282020-09-16 17:55:40 +0200639 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200640
641 lines =<< trim END
642 vim9script
643 def Func(...l: list<string>)
644 echo l
645 enddef
646 Func('a', 9)
647 END
Bram Moolenaar77072282020-09-16 17:55:40 +0200648 CheckScriptFailure(lines, 'E1013: Argument 2: type mismatch')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200649
650 lines =<< trim END
651 vim9script
652 def Func(...l: list<string>)
653 echo l
654 enddef
655 Func(1, 'a')
656 END
Bram Moolenaar77072282020-09-16 17:55:40 +0200657 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200658enddef
659
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200660let s:value = ''
661
662def FuncOneDefArg(opt = 'text')
663 s:value = opt
664enddef
665
666def FuncTwoDefArg(nr = 123, opt = 'text'): string
667 return nr .. opt
668enddef
669
670def FuncVarargs(...arg: list<string>): string
671 return join(arg, ',')
672enddef
673
674def Test_func_type_varargs()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200675 var RefDefArg: func(?string)
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200676 RefDefArg = FuncOneDefArg
677 RefDefArg()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200678 s:value->assert_equal('text')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200679 RefDefArg('some')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200680 s:value->assert_equal('some')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200681
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200682 var RefDef2Arg: func(?number, ?string): string
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200683 RefDef2Arg = FuncTwoDefArg
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200684 RefDef2Arg()->assert_equal('123text')
685 RefDef2Arg(99)->assert_equal('99text')
686 RefDef2Arg(77, 'some')->assert_equal('77some')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200687
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200688 CheckDefFailure(['var RefWrong: func(string?)'], 'E1010:')
689 CheckDefFailure(['var RefWrong: func(?string, string)'], 'E1007:')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200690
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200691 var RefVarargs: func(...list<string>): string
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200692 RefVarargs = FuncVarargs
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200693 RefVarargs()->assert_equal('')
694 RefVarargs('one')->assert_equal('one')
695 RefVarargs('one', 'two')->assert_equal('one,two')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200696
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200697 CheckDefFailure(['var RefWrong: func(...list<string>, string)'], 'E110:')
698 CheckDefFailure(['var RefWrong: func(...list<string>, ?string)'], 'E110:')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200699enddef
700
Bram Moolenaar0b76b422020-04-07 22:05:08 +0200701" Only varargs
702def MyVarargsOnly(...args: list<string>): string
703 return join(args, ',')
704enddef
705
706def Test_call_varargs_only()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200707 MyVarargsOnly()->assert_equal('')
708 MyVarargsOnly('one')->assert_equal('one')
709 MyVarargsOnly('one', 'two')->assert_equal('one,two')
Bram Moolenaar77072282020-09-16 17:55:40 +0200710 CheckDefFailure(['MyVarargsOnly(1)'], 'E1013: Argument 1: type mismatch, expected string but got number')
711 CheckDefFailure(['MyVarargsOnly("one", 2)'], 'E1013: Argument 2: type mismatch, expected string but got number')
Bram Moolenaar0b76b422020-04-07 22:05:08 +0200712enddef
713
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200714def Test_using_var_as_arg()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200715 writefile(['def Func(x: number)', 'var x = 234', 'enddef', 'defcompile'], 'Xdef')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200716 assert_fails('so Xdef', 'E1006:', '', 1, 'Func')
Bram Moolenaard2c61702020-09-06 15:58:36 +0200717 delete('Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200718enddef
719
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200720def DictArg(arg: dict<string>)
721 arg['key'] = 'value'
722enddef
723
724def ListArg(arg: list<string>)
725 arg[0] = 'value'
726enddef
727
728def Test_assign_to_argument()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200729 # works for dict and list
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200730 var d: dict<string> = {}
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200731 DictArg(d)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200732 d['key']->assert_equal('value')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200733 var l: list<string> = []
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200734 ListArg(l)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200735 l[0]->assert_equal('value')
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200736
Bram Moolenaard2c61702020-09-06 15:58:36 +0200737 CheckScriptFailure(['def Func(arg: number)', 'arg = 3', 'enddef', 'defcompile'], 'E1090:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100738 delfunc! g:Func
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200739enddef
740
Bram Moolenaarb816dae2020-09-20 22:04:00 +0200741" These argument names are reserved in legacy functions.
742def WithReservedNames(firstline: string, lastline: string): string
743 return firstline .. lastline
744enddef
745
746def Test_argument_names()
747 assert_equal('OK', WithReservedNames('O', 'K'))
748enddef
749
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200750def Test_call_func_defined_later()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200751 g:DefinedLater('one')->assert_equal('one')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200752 assert_fails('NotDefined("one")', 'E117:', '', 2, 'Test_call_func_defined_later')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200753enddef
754
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200755func DefinedLater(arg)
756 return a:arg
757endfunc
758
759def Test_call_funcref()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200760 g:SomeFunc('abc')->assert_equal(3)
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200761 assert_fails('NotAFunc()', 'E117:', '', 2, 'Test_call_funcref') # comment after call
762 assert_fails('g:NotAFunc()', 'E117:', '', 3, 'Test_call_funcref')
Bram Moolenaar2f1980f2020-07-22 19:30:06 +0200763
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200764 var lines =<< trim END
Bram Moolenaar2f1980f2020-07-22 19:30:06 +0200765 vim9script
766 def RetNumber(): number
767 return 123
768 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200769 var Funcref: func: number = function('RetNumber')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200770 Funcref()->assert_equal(123)
Bram Moolenaar2f1980f2020-07-22 19:30:06 +0200771 END
772 CheckScriptSuccess(lines)
Bram Moolenaar0f60e802020-07-22 20:16:11 +0200773
774 lines =<< trim END
775 vim9script
776 def RetNumber(): number
777 return 123
778 enddef
779 def Bar(F: func: number): number
780 return F()
781 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200782 var Funcref = function('RetNumber')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200783 Bar(Funcref)->assert_equal(123)
Bram Moolenaar0f60e802020-07-22 20:16:11 +0200784 END
785 CheckScriptSuccess(lines)
Bram Moolenaarbfba8652020-07-23 20:09:10 +0200786
787 lines =<< trim END
788 vim9script
789 def UseNumber(nr: number)
790 echo nr
791 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200792 var Funcref: func(number) = function('UseNumber')
Bram Moolenaarbfba8652020-07-23 20:09:10 +0200793 Funcref(123)
794 END
795 CheckScriptSuccess(lines)
Bram Moolenaarb8070e32020-07-23 20:56:04 +0200796
797 lines =<< trim END
798 vim9script
799 def UseNumber(nr: number)
800 echo nr
801 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200802 var Funcref: func(string) = function('UseNumber')
Bram Moolenaarb8070e32020-07-23 20:56:04 +0200803 END
Bram Moolenaar5e654232020-09-16 15:22:00 +0200804 CheckScriptFailure(lines, 'E1012: Type mismatch; expected func(string) but got func(number)')
Bram Moolenaar4fc224c2020-07-26 17:56:25 +0200805
806 lines =<< trim END
807 vim9script
808 def EchoNr(nr = 34)
809 g:echo = nr
810 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200811 var Funcref: func(?number) = function('EchoNr')
Bram Moolenaar4fc224c2020-07-26 17:56:25 +0200812 Funcref()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200813 g:echo->assert_equal(34)
Bram Moolenaar4fc224c2020-07-26 17:56:25 +0200814 Funcref(123)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200815 g:echo->assert_equal(123)
Bram Moolenaar4fc224c2020-07-26 17:56:25 +0200816 END
817 CheckScriptSuccess(lines)
Bram Moolenaarace61322020-07-26 18:16:58 +0200818
819 lines =<< trim END
820 vim9script
821 def EchoList(...l: list<number>)
822 g:echo = l
823 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200824 var Funcref: func(...list<number>) = function('EchoList')
Bram Moolenaarace61322020-07-26 18:16:58 +0200825 Funcref()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200826 g:echo->assert_equal([])
Bram Moolenaarace61322020-07-26 18:16:58 +0200827 Funcref(1, 2, 3)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200828 g:echo->assert_equal([1, 2, 3])
Bram Moolenaarace61322020-07-26 18:16:58 +0200829 END
830 CheckScriptSuccess(lines)
Bram Moolenaar01865ad2020-07-26 18:33:09 +0200831
832 lines =<< trim END
833 vim9script
834 def OptAndVar(nr: number, opt = 12, ...l: list<number>): number
835 g:optarg = opt
836 g:listarg = l
837 return nr
838 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200839 var Funcref: func(number, ?number, ...list<number>): number = function('OptAndVar')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200840 Funcref(10)->assert_equal(10)
841 g:optarg->assert_equal(12)
842 g:listarg->assert_equal([])
Bram Moolenaar01865ad2020-07-26 18:33:09 +0200843
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200844 Funcref(11, 22)->assert_equal(11)
845 g:optarg->assert_equal(22)
846 g:listarg->assert_equal([])
Bram Moolenaar01865ad2020-07-26 18:33:09 +0200847
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200848 Funcref(17, 18, 1, 2, 3)->assert_equal(17)
849 g:optarg->assert_equal(18)
850 g:listarg->assert_equal([1, 2, 3])
Bram Moolenaar01865ad2020-07-26 18:33:09 +0200851 END
852 CheckScriptSuccess(lines)
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200853enddef
854
855let SomeFunc = function('len')
856let NotAFunc = 'text'
857
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200858def CombineFuncrefTypes()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200859 # same arguments, different return type
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200860 var Ref1: func(bool): string
861 var Ref2: func(bool): number
862 var Ref3: func(bool): any
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200863 Ref3 = g:cond ? Ref1 : Ref2
864
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200865 # different number of arguments
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200866 var Refa1: func(bool): number
867 var Refa2: func(bool, number): number
868 var Refa3: func: number
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200869 Refa3 = g:cond ? Refa1 : Refa2
870
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200871 # different argument types
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200872 var Refb1: func(bool, string): number
873 var Refb2: func(string, number): number
874 var Refb3: func(any, any): number
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200875 Refb3 = g:cond ? Refb1 : Refb2
876enddef
877
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200878def FuncWithForwardCall()
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200879 return g:DefinedEvenLater("yes")
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200880enddef
881
882def DefinedEvenLater(arg: string): string
883 return arg
884enddef
885
886def Test_error_in_nested_function()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200887 # Error in called function requires unwinding the call stack.
Bram Moolenaar44d66522020-09-06 22:26:57 +0200888 assert_fails('FuncWithForwardCall()', 'E1096:', '', 1, 'FuncWithForwardCall')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200889enddef
890
891def Test_return_type_wrong()
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200892 CheckScriptFailure([
893 'def Func(): number',
894 'return "a"',
895 'enddef',
896 'defcompile'], 'expected number but got string')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100897 delfunc! g:Func
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200898 CheckScriptFailure([
899 'def Func(): string',
900 'return 1',
901 'enddef',
902 'defcompile'], 'expected string but got number')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100903 delfunc! g:Func
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200904 CheckScriptFailure([
905 'def Func(): void',
906 'return "a"',
907 'enddef',
908 'defcompile'],
909 'E1096: Returning a value in a function without a return type')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100910 delfunc! g:Func
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200911 CheckScriptFailure([
912 'def Func()',
913 'return "a"',
914 'enddef',
915 'defcompile'],
916 'E1096: Returning a value in a function without a return type')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100917 delfunc! g:Func
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200918
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200919 CheckScriptFailure([
920 'def Func(): number',
921 'return',
922 'enddef',
923 'defcompile'], 'E1003:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100924 delfunc! g:Func
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200925
926 CheckScriptFailure(['def Func(): list', 'return []', 'enddef'], 'E1008:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100927 delfunc! g:Func
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200928 CheckScriptFailure(['def Func(): dict', 'return {}', 'enddef'], 'E1008:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100929 delfunc! g:Func
Bram Moolenaaree4e0c12020-04-06 21:35:05 +0200930 CheckScriptFailure(['def Func()', 'return 1'], 'E1057:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100931 delfunc! g:Func
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200932
933 CheckScriptFailure([
934 'vim9script',
935 'def FuncB()',
936 ' return 123',
937 'enddef',
938 'def FuncA()',
939 ' FuncB()',
940 'enddef',
941 'defcompile'], 'E1096:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200942enddef
943
944def Test_arg_type_wrong()
945 CheckScriptFailure(['def Func3(items: list)', 'echo "a"', 'enddef'], 'E1008: Missing <type>')
Bram Moolenaaree4e0c12020-04-06 21:35:05 +0200946 CheckScriptFailure(['def Func4(...)', 'echo "a"', 'enddef'], 'E1055: Missing name after ...')
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +0200947 CheckScriptFailure(['def Func5(items:string)', 'echo "a"'], 'E1069:')
Bram Moolenaar6e949782020-04-13 17:21:00 +0200948 CheckScriptFailure(['def Func5(items)', 'echo "a"'], 'E1077:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200949enddef
950
951def Test_vim9script_call()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200952 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200953 vim9script
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200954 var name = ''
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200955 def MyFunc(arg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200956 name = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200957 enddef
958 MyFunc('foobar')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200959 name->assert_equal('foobar')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200960
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200961 var str = 'barfoo'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200962 str->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200963 name->assert_equal('barfoo')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200964
Bram Moolenaar67979662020-06-20 22:50:47 +0200965 g:value = 'value'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200966 g:value->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200967 name->assert_equal('value')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200968
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200969 var listvar = []
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200970 def ListFunc(arg: list<number>)
971 listvar = arg
972 enddef
973 [1, 2, 3]->ListFunc()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200974 listvar->assert_equal([1, 2, 3])
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200975
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200976 var dictvar = {}
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200977 def DictFunc(arg: dict<number>)
978 dictvar = arg
979 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +0100980 {a: 1, b: 2}->DictFunc()
981 dictvar->assert_equal({a: 1, b: 2})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200982 def CompiledDict()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100983 {a: 3, b: 4}->DictFunc()
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200984 enddef
985 CompiledDict()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100986 dictvar->assert_equal({a: 3, b: 4})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200987
Bram Moolenaare0de1712020-12-02 17:36:54 +0100988 {a: 3, b: 4}->DictFunc()
989 dictvar->assert_equal({a: 3, b: 4})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200990
991 ('text')->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200992 name->assert_equal('text')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200993 ("some")->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200994 name->assert_equal('some')
Bram Moolenaare6b53242020-07-01 17:28:33 +0200995
Bram Moolenaar13e12b82020-07-24 18:47:22 +0200996 # line starting with single quote is not a mark
Bram Moolenaar10409562020-07-29 20:00:38 +0200997 # line starting with double quote can be a method call
Bram Moolenaar3d48e252020-07-15 14:15:52 +0200998 'asdfasdf'->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200999 name->assert_equal('asdfasdf')
Bram Moolenaar10409562020-07-29 20:00:38 +02001000 "xyz"->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001001 name->assert_equal('xyz')
Bram Moolenaar3d48e252020-07-15 14:15:52 +02001002
1003 def UseString()
1004 'xyork'->MyFunc()
1005 enddef
1006 UseString()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001007 name->assert_equal('xyork')
Bram Moolenaar3d48e252020-07-15 14:15:52 +02001008
Bram Moolenaar10409562020-07-29 20:00:38 +02001009 def UseString2()
1010 "knife"->MyFunc()
1011 enddef
1012 UseString2()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001013 name->assert_equal('knife')
Bram Moolenaar10409562020-07-29 20:00:38 +02001014
Bram Moolenaar13e12b82020-07-24 18:47:22 +02001015 # prepending a colon makes it a mark
1016 new
1017 setline(1, ['aaa', 'bbb', 'ccc'])
1018 normal! 3Gmt1G
1019 :'t
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001020 getcurpos()[1]->assert_equal(3)
Bram Moolenaar13e12b82020-07-24 18:47:22 +02001021 bwipe!
1022
Bram Moolenaare6b53242020-07-01 17:28:33 +02001023 MyFunc(
1024 'continued'
1025 )
1026 assert_equal('continued',
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001027 name
Bram Moolenaare6b53242020-07-01 17:28:33 +02001028 )
1029
1030 call MyFunc(
1031 'more'
1032 ..
1033 'lines'
1034 )
1035 assert_equal(
1036 'morelines',
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001037 name)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001038 END
1039 writefile(lines, 'Xcall.vim')
1040 source Xcall.vim
1041 delete('Xcall.vim')
1042enddef
1043
1044def Test_vim9script_call_fail_decl()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001045 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001046 vim9script
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001047 var name = ''
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001048 def MyFunc(arg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001049 var name = 123
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001050 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +02001051 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001052 END
Bram Moolenaar6c4bfe42020-07-23 18:26:30 +02001053 CheckScriptFailure(lines, 'E1054:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001054enddef
1055
Bram Moolenaar65b95452020-07-19 14:03:09 +02001056def Test_vim9script_call_fail_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001057 var lines =<< trim END
Bram Moolenaar65b95452020-07-19 14:03:09 +02001058 vim9script
1059 def MyFunc(arg: string)
1060 echo arg
1061 enddef
1062 MyFunc(1234)
1063 END
Bram Moolenaar77072282020-09-16 17:55:40 +02001064 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected string but got number')
Bram Moolenaar65b95452020-07-19 14:03:09 +02001065enddef
1066
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001067def Test_vim9script_call_fail_const()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001068 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001069 vim9script
1070 const var = ''
1071 def MyFunc(arg: string)
1072 var = 'asdf'
1073 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +02001074 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001075 END
1076 writefile(lines, 'Xcall_const.vim')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001077 assert_fails('source Xcall_const.vim', 'E46:', '', 1, 'MyFunc')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001078 delete('Xcall_const.vim')
Bram Moolenaar3bdc90b2020-12-22 20:35:40 +01001079
1080 lines =<< trim END
1081 const g:Aconst = 77
1082 def Change()
1083 # comment
1084 g:Aconst = 99
1085 enddef
1086 call Change()
1087 unlet g:Aconst
1088 END
Bram Moolenaar1dcf55d2020-12-22 22:07:30 +01001089 CheckScriptFailure(lines, 'E741: Value is locked: Aconst', 2)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001090enddef
1091
1092" Test that inside :function a Python function can be defined, :def is not
1093" recognized.
1094func Test_function_python()
1095 CheckFeature python3
Bram Moolenaar727345e2020-09-27 23:33:59 +02001096 let py = 'python3'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001097 execute py "<< EOF"
1098def do_something():
1099 return 1
1100EOF
1101endfunc
1102
1103def Test_delfunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001104 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001105 vim9script
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02001106 def g:GoneSoon()
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001107 echo 'hello'
1108 enddef
1109
1110 def CallGoneSoon()
1111 GoneSoon()
1112 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +02001113 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001114
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02001115 delfunc g:GoneSoon
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001116 CallGoneSoon()
1117 END
1118 writefile(lines, 'XToDelFunc')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001119 assert_fails('so XToDelFunc', 'E933:', '', 1, 'CallGoneSoon')
1120 assert_fails('so XToDelFunc', 'E933:', '', 1, 'CallGoneSoon')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001121
1122 delete('XToDelFunc')
1123enddef
1124
1125def Test_redef_failure()
Bram Moolenaard2c61702020-09-06 15:58:36 +02001126 writefile(['def Func0(): string', 'return "Func0"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001127 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +02001128 writefile(['def Func1(): string', 'return "Func1"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001129 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +02001130 writefile(['def! Func0(): string', 'enddef', 'defcompile'], 'Xdef')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001131 assert_fails('so Xdef', 'E1027:', '', 1, 'Func0')
Bram Moolenaard2c61702020-09-06 15:58:36 +02001132 writefile(['def Func2(): string', 'return "Func2"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001133 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +02001134 delete('Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001135
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001136 g:Func0()->assert_equal(0)
1137 g:Func1()->assert_equal('Func1')
1138 g:Func2()->assert_equal('Func2')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001139
1140 delfunc! Func0
1141 delfunc! Func1
1142 delfunc! Func2
1143enddef
1144
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +02001145def Test_vim9script_func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001146 var lines =<< trim END
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +02001147 vim9script
1148 func Func(arg)
1149 echo a:arg
1150 endfunc
1151 Func('text')
1152 END
1153 writefile(lines, 'XVim9Func')
1154 so XVim9Func
1155
1156 delete('XVim9Func')
1157enddef
1158
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001159let s:funcResult = 0
1160
1161def FuncNoArgNoRet()
Bram Moolenaar53900992020-08-22 19:02:02 +02001162 s:funcResult = 11
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001163enddef
1164
1165def FuncNoArgRetNumber(): number
Bram Moolenaar53900992020-08-22 19:02:02 +02001166 s:funcResult = 22
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001167 return 1234
1168enddef
1169
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001170def FuncNoArgRetString(): string
Bram Moolenaar53900992020-08-22 19:02:02 +02001171 s:funcResult = 45
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001172 return 'text'
1173enddef
1174
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001175def FuncOneArgNoRet(arg: number)
Bram Moolenaar53900992020-08-22 19:02:02 +02001176 s:funcResult = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001177enddef
1178
1179def FuncOneArgRetNumber(arg: number): number
Bram Moolenaar53900992020-08-22 19:02:02 +02001180 s:funcResult = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001181 return arg
1182enddef
1183
Bram Moolenaar08938ee2020-04-11 23:17:17 +02001184def FuncTwoArgNoRet(one: bool, two: number)
Bram Moolenaar53900992020-08-22 19:02:02 +02001185 s:funcResult = two
Bram Moolenaar08938ee2020-04-11 23:17:17 +02001186enddef
1187
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001188def FuncOneArgRetString(arg: string): string
1189 return arg
1190enddef
1191
Bram Moolenaar89228602020-04-05 22:14:54 +02001192def FuncOneArgRetAny(arg: any): any
1193 return arg
1194enddef
1195
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001196def Test_func_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001197 var Ref1: func()
Bram Moolenaar53900992020-08-22 19:02:02 +02001198 s:funcResult = 0
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001199 Ref1 = FuncNoArgNoRet
1200 Ref1()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001201 s:funcResult->assert_equal(11)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001202
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001203 var Ref2: func
Bram Moolenaar53900992020-08-22 19:02:02 +02001204 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001205 Ref2 = FuncNoArgNoRet
1206 Ref2()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001207 s:funcResult->assert_equal(11)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001208
Bram Moolenaar53900992020-08-22 19:02:02 +02001209 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001210 Ref2 = FuncOneArgNoRet
1211 Ref2(12)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001212 s:funcResult->assert_equal(12)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001213
Bram Moolenaar53900992020-08-22 19:02:02 +02001214 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001215 Ref2 = FuncNoArgRetNumber
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001216 Ref2()->assert_equal(1234)
1217 s:funcResult->assert_equal(22)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001218
Bram Moolenaar53900992020-08-22 19:02:02 +02001219 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001220 Ref2 = FuncOneArgRetNumber
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001221 Ref2(13)->assert_equal(13)
1222 s:funcResult->assert_equal(13)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001223enddef
1224
Bram Moolenaar9978d472020-07-05 16:01:56 +02001225def Test_repeat_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001226 var res = 0
Bram Moolenaar9978d472020-07-05 16:01:56 +02001227 for n in repeat([1], 3)
1228 res += n
1229 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001230 res->assert_equal(3)
Bram Moolenaarfce82b32020-07-05 16:07:21 +02001231
1232 res = 0
1233 for n in add([1, 2], 3)
1234 res += n
1235 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001236 res->assert_equal(6)
Bram Moolenaar9978d472020-07-05 16:01:56 +02001237enddef
1238
Bram Moolenaar846178a2020-07-05 17:04:13 +02001239def Test_argv_return_type()
1240 next fileone filetwo
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001241 var res = ''
Bram Moolenaar846178a2020-07-05 17:04:13 +02001242 for name in argv()
1243 res ..= name
1244 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001245 res->assert_equal('fileonefiletwo')
Bram Moolenaar846178a2020-07-05 17:04:13 +02001246enddef
1247
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001248def Test_func_type_part()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001249 var RefVoid: func: void
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001250 RefVoid = FuncNoArgNoRet
1251 RefVoid = FuncOneArgNoRet
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001252 CheckDefFailure(['var RefVoid: func: void', 'RefVoid = FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func(...) but got func(): number')
1253 CheckDefFailure(['var RefVoid: func: void', 'RefVoid = FuncNoArgRetString'], 'E1012: Type mismatch; expected func(...) but got func(): string')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001254
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001255 var RefAny: func(): any
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001256 RefAny = FuncNoArgRetNumber
1257 RefAny = FuncNoArgRetString
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001258 CheckDefFailure(['var RefAny: func(): any', 'RefAny = FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(): any but got func()')
1259 CheckDefFailure(['var RefAny: func(): any', 'RefAny = FuncOneArgNoRet'], 'E1012: Type mismatch; expected func(): any but got func(number)')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001260
Bram Moolenaar6abd3dc2020-10-04 14:17:32 +02001261 var RefAnyNoArgs: func: any = RefAny
1262
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001263 var RefNr: func: number
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001264 RefNr = FuncNoArgRetNumber
1265 RefNr = FuncOneArgRetNumber
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001266 CheckDefFailure(['var RefNr: func: number', 'RefNr = FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(...): number but got func()')
1267 CheckDefFailure(['var RefNr: func: number', 'RefNr = FuncNoArgRetString'], 'E1012: Type mismatch; expected func(...): number but got func(): string')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001268
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001269 var RefStr: func: string
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001270 RefStr = FuncNoArgRetString
1271 RefStr = FuncOneArgRetString
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001272 CheckDefFailure(['var RefStr: func: string', 'RefStr = FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(...): string but got func()')
1273 CheckDefFailure(['var RefStr: func: string', 'RefStr = FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func(...): string but got func(): number')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001274enddef
1275
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001276def Test_func_type_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001277 CheckDefFailure(['var ref1: func()'], 'E704:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001278
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001279 CheckDefFailure(['var Ref1: func()', 'Ref1 = FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func() but got func(): number')
1280 CheckDefFailure(['var Ref1: func()', 'Ref1 = FuncOneArgNoRet'], 'E1012: Type mismatch; expected func() but got func(number)')
1281 CheckDefFailure(['var Ref1: func()', 'Ref1 = FuncOneArgRetNumber'], 'E1012: Type mismatch; expected func() but got func(number): number')
1282 CheckDefFailure(['var Ref1: func(bool)', 'Ref1 = FuncTwoArgNoRet'], 'E1012: Type mismatch; expected func(bool) but got func(bool, number)')
1283 CheckDefFailure(['var Ref1: func(?bool)', 'Ref1 = FuncTwoArgNoRet'], 'E1012: Type mismatch; expected func(?bool) but got func(bool, number)')
1284 CheckDefFailure(['var Ref1: func(...bool)', 'Ref1 = FuncTwoArgNoRet'], 'E1012: Type mismatch; expected func(...bool) but got func(bool, number)')
Bram Moolenaar08938ee2020-04-11 23:17:17 +02001285
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001286 CheckDefFailure(['var RefWrong: func(string ,number)'], 'E1068:')
1287 CheckDefFailure(['var RefWrong: func(string,number)'], 'E1069:')
1288 CheckDefFailure(['var RefWrong: func(bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool)'], 'E1005:')
1289 CheckDefFailure(['var RefWrong: func(bool):string'], 'E1069:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001290enddef
1291
Bram Moolenaar89228602020-04-05 22:14:54 +02001292def Test_func_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001293 var nr: number
Bram Moolenaar89228602020-04-05 22:14:54 +02001294 nr = FuncNoArgRetNumber()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001295 nr->assert_equal(1234)
Bram Moolenaar89228602020-04-05 22:14:54 +02001296
1297 nr = FuncOneArgRetAny(122)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001298 nr->assert_equal(122)
Bram Moolenaar89228602020-04-05 22:14:54 +02001299
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001300 var str: string
Bram Moolenaar89228602020-04-05 22:14:54 +02001301 str = FuncOneArgRetAny('yes')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001302 str->assert_equal('yes')
Bram Moolenaar89228602020-04-05 22:14:54 +02001303
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001304 CheckDefFailure(['var str: string', 'str = FuncNoArgRetNumber()'], 'E1012: Type mismatch; expected string but got number')
Bram Moolenaar89228602020-04-05 22:14:54 +02001305enddef
1306
Bram Moolenaar6abd3dc2020-10-04 14:17:32 +02001307def Test_func_common_type()
1308 def FuncOne(n: number): number
1309 return n
1310 enddef
1311 def FuncTwo(s: string): number
1312 return len(s)
1313 enddef
1314 def FuncThree(n: number, s: string): number
1315 return n + len(s)
1316 enddef
1317 var list = [FuncOne, FuncTwo, FuncThree]
1318 assert_equal(8, list[0](8))
1319 assert_equal(4, list[1]('word'))
1320 assert_equal(7, list[2](3, 'word'))
1321enddef
1322
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001323def MultiLine(
1324 arg1: string,
1325 arg2 = 1234,
1326 ...rest: list<string>
1327 ): string
1328 return arg1 .. arg2 .. join(rest, '-')
1329enddef
1330
Bram Moolenaar2c330432020-04-13 14:41:35 +02001331def MultiLineComment(
1332 arg1: string, # comment
1333 arg2 = 1234, # comment
1334 ...rest: list<string> # comment
1335 ): string # comment
1336 return arg1 .. arg2 .. join(rest, '-')
1337enddef
1338
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001339def Test_multiline()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001340 MultiLine('text')->assert_equal('text1234')
1341 MultiLine('text', 777)->assert_equal('text777')
1342 MultiLine('text', 777, 'one')->assert_equal('text777one')
1343 MultiLine('text', 777, 'one', 'two')->assert_equal('text777one-two')
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001344enddef
1345
Bram Moolenaar23e03252020-04-12 22:22:31 +02001346func Test_multiline_not_vim9()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001347 call MultiLine('text')->assert_equal('text1234')
1348 call MultiLine('text', 777)->assert_equal('text777')
1349 call MultiLine('text', 777, 'one')->assert_equal('text777one')
1350 call MultiLine('text', 777, 'one', 'two')->assert_equal('text777one-two')
Bram Moolenaar23e03252020-04-12 22:22:31 +02001351endfunc
1352
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001353
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02001354" When using CheckScriptFailure() for the below test, E1010 is generated instead
1355" of E1056.
1356func Test_E1056_1059()
1357 let caught_1056 = 0
1358 try
1359 def F():
1360 return 1
1361 enddef
1362 catch /E1056:/
1363 let caught_1056 = 1
1364 endtry
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001365 eval caught_1056->assert_equal(1)
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02001366
1367 let caught_1059 = 0
1368 try
1369 def F5(items : list)
1370 echo 'a'
1371 enddef
1372 catch /E1059:/
1373 let caught_1059 = 1
1374 endtry
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001375 eval caught_1059->assert_equal(1)
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02001376endfunc
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001377
Bram Moolenaar015f4262020-05-05 21:25:22 +02001378func DelMe()
1379 echo 'DelMe'
1380endfunc
1381
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001382def Test_error_reporting()
1383 # comment lines at the start of the function
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001384 var lines =<< trim END
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001385 " comment
1386 def Func()
1387 # comment
1388 # comment
1389 invalid
1390 enddef
1391 defcompile
1392 END
Bram Moolenaar08052222020-09-14 17:04:31 +02001393 writefile(lines, 'Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001394 try
1395 source Xdef
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001396 assert_report('should have failed')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001397 catch /E476:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001398 v:exception->assert_match('Invalid command: invalid')
1399 v:throwpoint->assert_match(', line 3$')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001400 endtry
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001401 delfunc! g:Func
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001402
1403 # comment lines after the start of the function
1404 lines =<< trim END
1405 " comment
1406 def Func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001407 var x = 1234
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001408 # comment
1409 # comment
1410 invalid
1411 enddef
1412 defcompile
1413 END
Bram Moolenaar08052222020-09-14 17:04:31 +02001414 writefile(lines, 'Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001415 try
1416 source Xdef
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001417 assert_report('should have failed')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001418 catch /E476:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001419 v:exception->assert_match('Invalid command: invalid')
1420 v:throwpoint->assert_match(', line 4$')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001421 endtry
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001422 delfunc! g:Func
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001423
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001424 lines =<< trim END
1425 vim9script
1426 def Func()
Bram Moolenaare0de1712020-12-02 17:36:54 +01001427 var db = {foo: 1, bar: 2}
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001428 # comment
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001429 var x = db.asdf
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001430 enddef
1431 defcompile
1432 Func()
1433 END
Bram Moolenaar08052222020-09-14 17:04:31 +02001434 writefile(lines, 'Xdef')
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001435 try
1436 source Xdef
1437 assert_report('should have failed')
1438 catch /E716:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001439 v:throwpoint->assert_match('_Func, line 3$')
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001440 endtry
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001441 delfunc! g:Func
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001442
Bram Moolenaar08052222020-09-14 17:04:31 +02001443 delete('Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001444enddef
1445
Bram Moolenaar015f4262020-05-05 21:25:22 +02001446def Test_deleted_function()
1447 CheckDefExecFailure([
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001448 'var RefMe: func = function("g:DelMe")',
Bram Moolenaar015f4262020-05-05 21:25:22 +02001449 'delfunc g:DelMe',
1450 'echo RefMe()'], 'E117:')
1451enddef
1452
1453def Test_unknown_function()
1454 CheckDefExecFailure([
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001455 'var Ref: func = function("NotExist")',
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +02001456 'delfunc g:NotExist'], 'E700:')
Bram Moolenaar015f4262020-05-05 21:25:22 +02001457enddef
1458
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +02001459def RefFunc(Ref: func(string): string): string
1460 return Ref('more')
1461enddef
1462
1463def Test_closure_simple()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001464 var local = 'some '
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001465 RefFunc((s) => local .. s)->assert_equal('some more')
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +02001466enddef
1467
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001468def MakeRef()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001469 var local = 'some '
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001470 g:Ref = (s) => local .. s
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001471enddef
1472
1473def Test_closure_ref_after_return()
1474 MakeRef()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001475 g:Ref('thing')->assert_equal('some thing')
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001476 unlet g:Ref
1477enddef
1478
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001479def MakeTwoRefs()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001480 var local = ['some']
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001481 g:Extend = (s) => local->add(s)
1482 g:Read = () => local
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001483enddef
1484
1485def Test_closure_two_refs()
1486 MakeTwoRefs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001487 join(g:Read(), ' ')->assert_equal('some')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001488 g:Extend('more')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001489 join(g:Read(), ' ')->assert_equal('some more')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001490 g:Extend('even')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001491 join(g:Read(), ' ')->assert_equal('some more even')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001492
1493 unlet g:Extend
1494 unlet g:Read
1495enddef
1496
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001497def ReadRef(Ref: func(): list<string>): string
1498 return join(Ref(), ' ')
1499enddef
1500
Bram Moolenaar5e654232020-09-16 15:22:00 +02001501def ExtendRef(Ref: func(string): list<string>, add: string)
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001502 Ref(add)
1503enddef
1504
1505def Test_closure_two_indirect_refs()
Bram Moolenaarf7779c62020-05-03 15:38:16 +02001506 MakeTwoRefs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001507 ReadRef(g:Read)->assert_equal('some')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001508 ExtendRef(g:Extend, 'more')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001509 ReadRef(g:Read)->assert_equal('some more')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001510 ExtendRef(g:Extend, 'even')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001511 ReadRef(g:Read)->assert_equal('some more even')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001512
1513 unlet g:Extend
1514 unlet g:Read
1515enddef
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001516
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001517def MakeArgRefs(theArg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001518 var local = 'loc_val'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001519 g:UseArg = (s) => theArg .. '/' .. local .. '/' .. s
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001520enddef
1521
1522def MakeArgRefsVarargs(theArg: string, ...rest: list<string>)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001523 var local = 'the_loc'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001524 g:UseVararg = (s) => theArg .. '/' .. local .. '/' .. s .. '/' .. join(rest)
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001525enddef
1526
1527def Test_closure_using_argument()
1528 MakeArgRefs('arg_val')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001529 g:UseArg('call_val')->assert_equal('arg_val/loc_val/call_val')
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001530
1531 MakeArgRefsVarargs('arg_val', 'one', 'two')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001532 g:UseVararg('call_val')->assert_equal('arg_val/the_loc/call_val/one two')
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001533
1534 unlet g:UseArg
1535 unlet g:UseVararg
1536enddef
1537
Bram Moolenaar85d5e2b2020-10-10 14:13:01 +02001538def MakeGetAndAppendRefs()
1539 var local = 'a'
1540
1541 def Append(arg: string)
1542 local ..= arg
1543 enddef
1544 g:Append = Append
1545
1546 def Get(): string
1547 return local
1548 enddef
1549 g:Get = Get
1550enddef
1551
1552def Test_closure_append_get()
1553 MakeGetAndAppendRefs()
1554 g:Get()->assert_equal('a')
1555 g:Append('-b')
1556 g:Get()->assert_equal('a-b')
1557 g:Append('-c')
1558 g:Get()->assert_equal('a-b-c')
1559
1560 unlet g:Append
1561 unlet g:Get
1562enddef
Bram Moolenaarb68b3462020-05-06 21:06:30 +02001563
Bram Moolenaar04b12692020-05-04 23:24:44 +02001564def Test_nested_closure()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001565 var local = 'text'
Bram Moolenaar04b12692020-05-04 23:24:44 +02001566 def Closure(arg: string): string
1567 return local .. arg
1568 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001569 Closure('!!!')->assert_equal('text!!!')
Bram Moolenaar04b12692020-05-04 23:24:44 +02001570enddef
1571
Bram Moolenaar6f5b6df2020-05-16 21:20:12 +02001572func GetResult(Ref)
1573 return a:Ref('some')
1574endfunc
1575
1576def Test_call_closure_not_compiled()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001577 var text = 'text'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001578 g:Ref = (s) => s .. text
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001579 GetResult(g:Ref)->assert_equal('sometext')
Bram Moolenaar6f5b6df2020-05-16 21:20:12 +02001580enddef
1581
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001582def Test_double_closure_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001583 var lines =<< trim END
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001584 vim9script
1585 def Func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001586 var name = 0
1587 for i in range(2)
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001588 timer_start(0, () => name)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001589 endfor
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001590 enddef
1591 Func()
1592 END
Bram Moolenaar148ce7a2020-09-23 21:57:23 +02001593 CheckScriptSuccess(lines)
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001594enddef
1595
Bram Moolenaar85d5e2b2020-10-10 14:13:01 +02001596def Test_nested_closure_used()
1597 var lines =<< trim END
1598 vim9script
1599 def Func()
1600 var x = 'hello'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001601 var Closure = () => x
1602 g:Myclosure = () => Closure()
Bram Moolenaar85d5e2b2020-10-10 14:13:01 +02001603 enddef
1604 Func()
1605 assert_equal('hello', g:Myclosure())
1606 END
1607 CheckScriptSuccess(lines)
1608enddef
Bram Moolenaar0876c782020-10-07 19:08:04 +02001609
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02001610def Test_nested_closure_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001611 var lines =<< trim END
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02001612 vim9script
1613 def FuncA()
1614 FuncB(0)
1615 enddef
1616 def FuncB(n: number): list<string>
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001617 return map([0], (_, v) => n)
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02001618 enddef
1619 FuncA()
1620 END
1621 CheckScriptFailure(lines, 'E1012:')
1622enddef
1623
Bram Moolenaarf112f302020-12-20 17:47:52 +01001624def Test_global_closure()
1625 var lines =<< trim END
1626 vim9script
1627 def ReverseEveryNLines(n: number, line1: number, line2: number)
1628 var mods = 'sil keepj keepp lockm '
1629 var range = ':' .. line1 .. ',' .. line2
1630 def g:Offset(): number
1631 var offset = (line('.') - line1 + 1) % n
1632 return offset != 0 ? offset : n
1633 enddef
1634 exe mods .. range .. 'g/^/exe "m .-" .. g:Offset()'
1635 enddef
1636
1637 new
1638 repeat(['aaa', 'bbb', 'ccc'], 3)->setline(1)
1639 ReverseEveryNLines(3, 1, 9)
1640 END
1641 CheckScriptSuccess(lines)
1642 var expected = repeat(['ccc', 'bbb', 'aaa'], 3)
1643 assert_equal(expected, getline(1, 9))
1644 bwipe!
1645enddef
1646
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01001647def Test_global_closure_called_directly()
1648 var lines =<< trim END
1649 vim9script
1650 def Outer()
1651 var x = 1
1652 def g:Inner()
1653 var y = x
1654 x += 1
1655 assert_equal(1, y)
1656 enddef
1657 g:Inner()
1658 assert_equal(2, x)
1659 enddef
1660 Outer()
1661 END
1662 CheckScriptSuccess(lines)
1663 delfunc g:Inner
1664enddef
1665
Bram Moolenaar34c54eb2020-11-25 19:15:19 +01001666def Test_failure_in_called_function()
1667 # this was using the frame index as the return value
1668 var lines =<< trim END
1669 vim9script
1670 au TerminalWinOpen * eval [][0]
1671 def PopupTerm(a: any)
1672 # make sure typvals on stack are string
1673 ['a', 'b', 'c', 'd', 'e', 'f', 'g']->join()
1674 FireEvent()
1675 enddef
1676 def FireEvent()
1677 do TerminalWinOpen
1678 enddef
1679 # use try/catch to make eval fail
1680 try
1681 call PopupTerm(0)
1682 catch
1683 endtry
1684 au! TerminalWinOpen
1685 END
1686 CheckScriptSuccess(lines)
1687enddef
1688
Bram Moolenaar5366e1a2020-10-01 13:01:34 +02001689def Test_nested_lambda()
1690 var lines =<< trim END
1691 vim9script
1692 def Func()
1693 var x = 4
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001694 var Lambda1 = () => 7
1695 var Lambda2 = () => [Lambda1(), x]
Bram Moolenaar5366e1a2020-10-01 13:01:34 +02001696 var res = Lambda2()
1697 assert_equal([7, 4], res)
1698 enddef
1699 Func()
1700 END
1701 CheckScriptSuccess(lines)
1702enddef
1703
Bram Moolenaar52bf81c2020-11-17 18:50:44 +01001704def Shadowed(): list<number>
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001705 var FuncList: list<func: number> = [() => 42]
1706 return FuncList->map((_, Shadowed) => Shadowed())
Bram Moolenaar52bf81c2020-11-17 18:50:44 +01001707enddef
1708
1709def Test_lambda_arg_shadows_func()
1710 assert_equal([42], Shadowed())
1711enddef
1712
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02001713def Line_continuation_in_def(dir: string = ''): string
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001714 var path: string = empty(dir)
1715 \ ? 'empty'
1716 \ : 'full'
1717 return path
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02001718enddef
1719
1720def Test_line_continuation_in_def()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001721 Line_continuation_in_def('.')->assert_equal('full')
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02001722enddef
1723
Bram Moolenaar2ea95b62020-11-19 21:47:56 +01001724def Test_script_var_in_lambda()
1725 var lines =<< trim END
1726 vim9script
1727 var script = 'test'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001728 assert_equal(['test'], map(['one'], () => script))
Bram Moolenaar2ea95b62020-11-19 21:47:56 +01001729 END
1730 CheckScriptSuccess(lines)
1731enddef
1732
Bram Moolenaar5e654232020-09-16 15:22:00 +02001733def Line_continuation_in_lambda(): list<string>
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001734 var x = range(97, 100)
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001735 ->map((_, v) => nr2char(v)
1736 ->toupper())
Bram Moolenaar7a4b8982020-07-08 17:36:21 +02001737 ->reverse()
1738 return x
1739enddef
1740
1741def Test_line_continuation_in_lambda()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001742 Line_continuation_in_lambda()->assert_equal(['D', 'C', 'B', 'A'])
Bram Moolenaar7a4b8982020-07-08 17:36:21 +02001743enddef
1744
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001745func Test_silent_echo()
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001746 CheckScreendump
1747
1748 let lines =<< trim END
1749 vim9script
1750 def EchoNothing()
1751 silent echo ''
1752 enddef
1753 defcompile
1754 END
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001755 call writefile(lines, 'XTest_silent_echo')
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001756
1757 " Check that the balloon shows up after a mouse move
1758 let buf = RunVimInTerminal('-S XTest_silent_echo', {'rows': 6})
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001759 call term_sendkeys(buf, ":abc")
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001760 call VerifyScreenDump(buf, 'Test_vim9_silent_echo', {})
1761
1762 " clean up
1763 call StopVimInTerminal(buf)
1764 call delete('XTest_silent_echo')
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001765endfunc
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001766
Bram Moolenaar171fb922020-10-28 16:54:47 +01001767def SilentlyError()
1768 execute('silent! invalid')
1769 g:did_it = 'yes'
1770enddef
1771
Bram Moolenaar28ee8922020-10-28 20:20:00 +01001772func UserError()
1773 silent! invalid
1774endfunc
1775
1776def SilentlyUserError()
1777 UserError()
1778 g:did_it = 'yes'
1779enddef
Bram Moolenaar171fb922020-10-28 16:54:47 +01001780
1781" This can't be a :def function, because the assert would not be reached.
Bram Moolenaar171fb922020-10-28 16:54:47 +01001782func Test_ignore_silent_error()
1783 let g:did_it = 'no'
1784 call SilentlyError()
1785 call assert_equal('yes', g:did_it)
1786
Bram Moolenaar28ee8922020-10-28 20:20:00 +01001787 let g:did_it = 'no'
1788 call SilentlyUserError()
1789 call assert_equal('yes', g:did_it)
Bram Moolenaar171fb922020-10-28 16:54:47 +01001790
1791 unlet g:did_it
1792endfunc
1793
Bram Moolenaarcd030c42020-10-30 21:49:40 +01001794def Test_ignore_silent_error_in_filter()
1795 var lines =<< trim END
1796 vim9script
1797 def Filter(winid: number, key: string): bool
1798 if key == 'o'
1799 silent! eval [][0]
1800 return true
1801 endif
1802 return popup_filter_menu(winid, key)
1803 enddef
1804
Bram Moolenaare0de1712020-12-02 17:36:54 +01001805 popup_create('popup', {filter: Filter})
Bram Moolenaarcd030c42020-10-30 21:49:40 +01001806 feedkeys("o\r", 'xnt')
1807 END
1808 CheckScriptSuccess(lines)
1809enddef
1810
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001811def Fibonacci(n: number): number
1812 if n < 2
1813 return n
1814 else
1815 return Fibonacci(n - 1) + Fibonacci(n - 2)
1816 endif
1817enddef
1818
Bram Moolenaar985116a2020-07-12 17:31:09 +02001819def Test_recursive_call()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001820 Fibonacci(20)->assert_equal(6765)
Bram Moolenaar985116a2020-07-12 17:31:09 +02001821enddef
1822
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001823def TreeWalk(dir: string): list<any>
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001824 return readdir(dir)->map((_, val) =>
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001825 fnamemodify(dir .. '/' .. val, ':p')->isdirectory()
Bram Moolenaar2bede172020-11-19 18:53:18 +01001826 ? {[val]: TreeWalk(dir .. '/' .. val)}
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001827 : val
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001828 )
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001829enddef
1830
1831def Test_closure_in_map()
1832 mkdir('XclosureDir/tdir', 'p')
1833 writefile(['111'], 'XclosureDir/file1')
1834 writefile(['222'], 'XclosureDir/file2')
1835 writefile(['333'], 'XclosureDir/tdir/file3')
1836
Bram Moolenaare0de1712020-12-02 17:36:54 +01001837 TreeWalk('XclosureDir')->assert_equal(['file1', 'file2', {tdir: ['file3']}])
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001838
1839 delete('XclosureDir', 'rf')
1840enddef
1841
Bram Moolenaar7b5d5442020-10-04 13:42:34 +02001842def Test_invalid_function_name()
1843 var lines =<< trim END
1844 vim9script
1845 def s: list<string>
1846 END
1847 CheckScriptFailure(lines, 'E129:')
1848
1849 lines =<< trim END
1850 vim9script
1851 def g: list<string>
1852 END
1853 CheckScriptFailure(lines, 'E129:')
1854
1855 lines =<< trim END
1856 vim9script
1857 def <SID>: list<string>
1858 END
1859 CheckScriptFailure(lines, 'E884:')
1860
1861 lines =<< trim END
1862 vim9script
1863 def F list<string>
1864 END
1865 CheckScriptFailure(lines, 'E488:')
1866enddef
1867
Bram Moolenaara90afb92020-07-15 22:38:56 +02001868def Test_partial_call()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001869 var Xsetlist = function('setloclist', [0])
Bram Moolenaare0de1712020-12-02 17:36:54 +01001870 Xsetlist([], ' ', {title: 'test'})
1871 getloclist(0, {title: 1})->assert_equal({title: 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02001872
1873 Xsetlist = function('setloclist', [0, [], ' '])
Bram Moolenaare0de1712020-12-02 17:36:54 +01001874 Xsetlist({title: 'test'})
1875 getloclist(0, {title: 1})->assert_equal({title: 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02001876
1877 Xsetlist = function('setqflist')
Bram Moolenaare0de1712020-12-02 17:36:54 +01001878 Xsetlist([], ' ', {title: 'test'})
1879 getqflist({title: 1})->assert_equal({title: 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02001880
1881 Xsetlist = function('setqflist', [[], ' '])
Bram Moolenaare0de1712020-12-02 17:36:54 +01001882 Xsetlist({title: 'test'})
1883 getqflist({title: 1})->assert_equal({title: 'test'})
Bram Moolenaar6abd3dc2020-10-04 14:17:32 +02001884
1885 var Len: func: number = function('len', ['word'])
1886 assert_equal(4, Len())
Bram Moolenaara90afb92020-07-15 22:38:56 +02001887enddef
1888
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001889def Test_cmd_modifier()
1890 tab echo '0'
Bram Moolenaard2c61702020-09-06 15:58:36 +02001891 CheckDefFailure(['5tab echo 3'], 'E16:')
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001892enddef
1893
1894def Test_restore_modifiers()
1895 # check that when compiling a :def function command modifiers are not messed
1896 # up.
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001897 var lines =<< trim END
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001898 vim9script
1899 set eventignore=
1900 autocmd QuickFixCmdPost * copen
1901 def AutocmdsDisabled()
Bram Moolenaar6cf7e3b2020-10-28 14:31:16 +01001902 eval 0
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001903 enddef
1904 func Func()
1905 noautocmd call s:AutocmdsDisabled()
1906 let g:ei_after = &eventignore
1907 endfunc
1908 Func()
1909 END
1910 CheckScriptSuccess(lines)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001911 g:ei_after->assert_equal('')
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001912enddef
1913
Bram Moolenaardfa3d552020-09-10 22:05:08 +02001914def StackTop()
1915 eval 1
1916 eval 2
1917 # call not on fourth line
1918 StackBot()
1919enddef
1920
1921def StackBot()
1922 # throw an error
1923 eval [][0]
1924enddef
1925
1926def Test_callstack_def()
1927 try
1928 StackTop()
1929 catch
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001930 v:throwpoint->assert_match('Test_callstack_def\[2\]..StackTop\[4\]..StackBot, line 2')
Bram Moolenaardfa3d552020-09-10 22:05:08 +02001931 endtry
1932enddef
1933
Bram Moolenaare8211a32020-10-09 22:04:29 +02001934" Re-using spot for variable used in block
1935def Test_block_scoped_var()
1936 var lines =<< trim END
1937 vim9script
1938 def Func()
1939 var x = ['a', 'b', 'c']
1940 if 1
1941 var y = 'x'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001942 map(x, () => y)
Bram Moolenaare8211a32020-10-09 22:04:29 +02001943 endif
1944 var z = x
1945 assert_equal(['x', 'x', 'x'], z)
1946 enddef
1947 Func()
1948 END
1949 CheckScriptSuccess(lines)
1950enddef
1951
Bram Moolenaareeece9e2020-11-20 19:26:48 +01001952def Test_reset_did_emsg()
1953 var lines =<< trim END
1954 @s = 'blah'
1955 au BufWinLeave * #
1956 def Func()
1957 var winid = popup_create('popup', {})
1958 exe '*s'
1959 popup_close(winid)
1960 enddef
1961 Func()
1962 END
1963 CheckScriptFailure(lines, 'E492:', 8)
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001964 delfunc! g:Func
Bram Moolenaareeece9e2020-11-20 19:26:48 +01001965enddef
1966
Bram Moolenaar57f799e2020-12-12 20:42:19 +01001967def Test_did_emsg_reset()
1968 # executing an autocommand resets did_emsg, this should not result in a
1969 # builtin function considered failing
1970 var lines =<< trim END
1971 vim9script
1972 au BufWinLeave * #
1973 def Func()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001974 popup_menu('', {callback: () => popup_create('', {})->popup_close()})
Bram Moolenaar57f799e2020-12-12 20:42:19 +01001975 eval [][0]
1976 enddef
1977 nno <F3> <cmd>call <sid>Func()<cr>
1978 feedkeys("\<F3>\e", 'xt')
1979 END
1980 writefile(lines, 'XemsgReset')
1981 assert_fails('so XemsgReset', ['E684:', 'E684:'], lines, 2)
1982 delete('XemsgReset')
1983 nunmap <F3>
1984 au! BufWinLeave
1985enddef
1986
Bram Moolenaar56602ba2020-12-05 21:22:08 +01001987def Test_abort_with_silent_call()
1988 var lines =<< trim END
1989 vim9script
1990 g:result = 'none'
1991 def Func()
1992 g:result += 3
1993 g:result = 'yes'
1994 enddef
1995 # error is silenced, but function aborts on error
1996 silent! Func()
1997 assert_equal('none', g:result)
1998 unlet g:result
1999 END
2000 CheckScriptSuccess(lines)
2001enddef
2002
Bram Moolenaarf665e972020-12-05 19:17:16 +01002003def Test_continues_with_silent_error()
2004 var lines =<< trim END
2005 vim9script
2006 g:result = 'none'
2007 def Func()
2008 silent! g:result += 3
2009 g:result = 'yes'
2010 enddef
2011 # error is silenced, function does not abort
2012 Func()
2013 assert_equal('yes', g:result)
2014 unlet g:result
2015 END
2016 CheckScriptSuccess(lines)
2017enddef
2018
Bram Moolenaaraf0df472020-12-02 20:51:22 +01002019def Test_abort_even_with_silent()
2020 var lines =<< trim END
2021 vim9script
2022 g:result = 'none'
2023 def Func()
2024 eval {-> ''}() .. '' .. {}['X']
2025 g:result = 'yes'
2026 enddef
Bram Moolenaarf665e972020-12-05 19:17:16 +01002027 silent! Func()
Bram Moolenaaraf0df472020-12-02 20:51:22 +01002028 assert_equal('none', g:result)
Bram Moolenaar4029cab2020-12-05 18:13:27 +01002029 unlet g:result
2030 END
2031 CheckScriptSuccess(lines)
2032enddef
2033
Bram Moolenaarf665e972020-12-05 19:17:16 +01002034def Test_cmdmod_silent_restored()
2035 var lines =<< trim END
2036 vim9script
2037 def Func()
2038 g:result = 'none'
2039 silent! g:result += 3
2040 g:result = 'none'
2041 g:result += 3
2042 enddef
2043 Func()
2044 END
2045 # can't use CheckScriptFailure, it ignores the :silent!
2046 var fname = 'Xdefsilent'
2047 writefile(lines, fname)
2048 var caught = 'no'
2049 try
2050 exe 'source ' .. fname
2051 catch /E1030:/
2052 caught = 'yes'
2053 assert_match('Func, line 4', v:throwpoint)
2054 endtry
2055 assert_equal('yes', caught)
2056 delete(fname)
2057enddef
2058
Bram Moolenaar4029cab2020-12-05 18:13:27 +01002059def Test_dict_member_with_silent()
2060 var lines =<< trim END
2061 vim9script
2062 g:result = 'none'
2063 var d: dict<any>
2064 def Func()
2065 try
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002066 g:result = map([], (_, v) => ({}[v]))->join() .. d['']
Bram Moolenaar4029cab2020-12-05 18:13:27 +01002067 catch
2068 endtry
2069 enddef
2070 silent! Func()
2071 assert_equal('0', g:result)
2072 unlet g:result
Bram Moolenaaraf0df472020-12-02 20:51:22 +01002073 END
2074 CheckScriptSuccess(lines)
2075enddef
2076
Bram Moolenaar5b3d1bb2020-12-22 12:20:08 +01002077def Test_opfunc()
2078 nnoremap <F3> <cmd>set opfunc=Opfunc<cr>g@
2079 def g:Opfunc(_: any): string
2080 setline(1, 'ASDF')
2081 return ''
2082 enddef
2083 new
2084 setline(1, 'asdf')
2085 feedkeys("\<F3>$", 'x')
2086 assert_equal('ASDF', getline(1))
2087
2088 bwipe!
2089 nunmap <F3>
2090enddef
2091
Bram Moolenaar077a4232020-12-22 18:33:27 +01002092" this was crashing on exit
2093def Test_nested_lambda_in_closure()
2094 var lines =<< trim END
2095 vim9script
2096 def Outer()
2097 def g:Inner()
2098 echo map([1, 2, 3], {_, v -> v + 1})
2099 enddef
2100 g:Inner()
2101 enddef
2102 defcompile
2103 writefile(['Done'], 'XnestedDone')
2104 quit
2105 END
2106 if !RunVim([], lines, '--clean')
2107 return
2108 endif
2109 assert_equal(['Done'], readfile('XnestedDone'))
2110 delete('XnestedDone')
2111enddef
2112
2113
Bram Moolenaarf7779c62020-05-03 15:38:16 +02002114
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002115" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker