blob: e1d48534118755fcbb58a3b1c16830b9dc3183f4 [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
Bram Moolenaarcef12702021-01-04 14:09:43 +0100244def FuncWithComment( # comment
245 a: number, #comment
246 b: bool, # comment
247 c: string) #comment
248 assert_equal(4, a)
249 assert_equal(true, b)
250 assert_equal('yes', c)
251enddef
252
253def Test_func_with_comments()
254 FuncWithComment(4, true, 'yes')
255
256 var lines =<< trim END
257 def Func(# comment
258 arg: string)
259 enddef
260 END
261 CheckScriptFailure(lines, 'E125:', 1)
262
263 lines =<< trim END
264 def Func(
265 arg: string# comment
266 )
267 enddef
268 END
269 CheckScriptFailure(lines, 'E475:', 2)
270
271 lines =<< trim END
272 def Func(
273 arg: string
274 )# comment
275 enddef
276 END
277 CheckScriptFailure(lines, 'E488:', 3)
278enddef
279
Bram Moolenaar04b12692020-05-04 23:24:44 +0200280def Test_nested_function()
281 def Nested(arg: string): string
282 return 'nested ' .. arg
283 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200284 Nested('function')->assert_equal('nested function')
Bram Moolenaar04b12692020-05-04 23:24:44 +0200285
Bram Moolenaar0e65d3d2020-05-05 17:53:16 +0200286 CheckDefFailure(['def Nested()', 'enddef', 'Nested(66)'], 'E118:')
287 CheckDefFailure(['def Nested(arg: string)', 'enddef', 'Nested()'], 'E119:')
288
Bram Moolenaar04b12692020-05-04 23:24:44 +0200289 CheckDefFailure(['func Nested()', 'endfunc'], 'E1086:')
Bram Moolenaarbcbf4132020-08-01 22:35:13 +0200290 CheckDefFailure(['def s:Nested()', 'enddef'], 'E1075:')
291 CheckDefFailure(['def b:Nested()', 'enddef'], 'E1075:')
Bram Moolenaar8b848ca2020-09-10 22:28:01 +0200292
Bram Moolenaar54021752020-12-06 18:50:36 +0100293 var lines =<< trim END
294 def Outer()
295 def Inner()
296 # comment
297 enddef
298 def Inner()
299 enddef
300 enddef
301 END
302 CheckDefFailure(lines, 'E1073:')
303
304 lines =<< trim END
305 def Outer()
306 def Inner()
307 # comment
308 enddef
309 def! Inner()
310 enddef
311 enddef
312 END
313 CheckDefFailure(lines, 'E1117:')
314
315 # nested function inside conditional
316 # TODO: should it work when "thecount" is inside the "if"?
317 lines =<< trim END
318 vim9script
319 var thecount = 0
320 if true
321 def Test(): number
322 def TheFunc(): number
323 thecount += 1
324 return thecount
325 enddef
326 return TheFunc()
327 enddef
328 endif
329 defcompile
330 assert_equal(1, Test())
331 assert_equal(2, Test())
332 END
333 CheckScriptSuccess(lines)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200334enddef
335
Bram Moolenaaradc8e442020-12-31 18:28:18 +0100336def Test_not_nested_function()
337 echo printf('%d',
338 function('len')('xxx'))
339enddef
340
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200341func Test_call_default_args_from_func()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200342 call MyDefaultArgs()->assert_equal('string')
343 call MyDefaultArgs('one')->assert_equal('one')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200344 call assert_fails('call MyDefaultArgs("one", "two")', 'E118:', '', 3, 'Test_call_default_args_from_func')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200345endfunc
346
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200347def Test_nested_global_function()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200348 var lines =<< trim END
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200349 vim9script
350 def Outer()
351 def g:Inner(): string
352 return 'inner'
353 enddef
354 enddef
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200355 defcompile
356 Outer()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200357 g:Inner()->assert_equal('inner')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200358 delfunc g:Inner
359 Outer()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200360 g:Inner()->assert_equal('inner')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200361 delfunc g:Inner
362 Outer()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200363 g:Inner()->assert_equal('inner')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200364 delfunc g:Inner
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200365 END
366 CheckScriptSuccess(lines)
Bram Moolenaar2c79e9d2020-08-01 18:57:52 +0200367
368 lines =<< trim END
369 vim9script
370 def Outer()
371 def g:Inner(): string
372 return 'inner'
373 enddef
374 enddef
375 defcompile
376 Outer()
377 Outer()
378 END
379 CheckScriptFailure(lines, "E122:")
Bram Moolenaarcd45ed02020-12-22 17:35:54 +0100380 delfunc g:Inner
Bram Moolenaarad486a02020-08-01 23:22:18 +0200381
382 lines =<< trim END
383 vim9script
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100384 def Outer()
385 def g:Inner()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100386 echo map([1, 2, 3], (_, v) => v + 1)
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100387 enddef
388 g:Inner()
389 enddef
390 Outer()
391 END
392 CheckScriptSuccess(lines)
393 delfunc g:Inner
394
395 lines =<< trim END
396 vim9script
Bram Moolenaarad486a02020-08-01 23:22:18 +0200397 def Func()
398 echo 'script'
399 enddef
400 def Outer()
401 def Func()
402 echo 'inner'
403 enddef
404 enddef
405 defcompile
406 END
407 CheckScriptFailure(lines, "E1073:")
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200408enddef
409
Bram Moolenaar6abdcf82020-11-22 18:15:44 +0100410def DefListAll()
411 def
412enddef
413
414def DefListOne()
415 def DefListOne
416enddef
417
418def DefListMatches()
419 def /DefList
420enddef
421
422def Test_nested_def_list()
423 var funcs = split(execute('call DefListAll()'), "\n")
424 assert_true(len(funcs) > 10)
425 assert_true(funcs->index('def DefListAll()') >= 0)
426
427 funcs = split(execute('call DefListOne()'), "\n")
428 assert_equal([' def DefListOne()', '1 def DefListOne', ' enddef'], funcs)
429
430 funcs = split(execute('call DefListMatches()'), "\n")
431 assert_true(len(funcs) >= 3)
432 assert_true(funcs->index('def DefListAll()') >= 0)
433 assert_true(funcs->index('def DefListOne()') >= 0)
434 assert_true(funcs->index('def DefListMatches()') >= 0)
Bram Moolenaar54021752020-12-06 18:50:36 +0100435
436 var lines =<< trim END
437 vim9script
438 def Func()
439 def +Func+
440 enddef
441 defcompile
442 END
443 CheckScriptFailure(lines, 'E476:', 1)
Bram Moolenaar6abdcf82020-11-22 18:15:44 +0100444enddef
445
Bram Moolenaar333894b2020-08-01 18:53:07 +0200446def Test_global_local_function()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200447 var lines =<< trim END
Bram Moolenaar333894b2020-08-01 18:53:07 +0200448 vim9script
449 def g:Func(): string
450 return 'global'
451 enddef
452 def Func(): string
453 return 'local'
454 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200455 g:Func()->assert_equal('global')
456 Func()->assert_equal('local')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100457 delfunc g:Func
Bram Moolenaar333894b2020-08-01 18:53:07 +0200458 END
459 CheckScriptSuccess(lines)
Bram Moolenaar035d6e92020-08-11 22:30:42 +0200460
461 lines =<< trim END
462 vim9script
463 def g:Funcy()
464 echo 'funcy'
465 enddef
466 s:Funcy()
467 END
468 CheckScriptFailure(lines, 'E117:')
Bram Moolenaar333894b2020-08-01 18:53:07 +0200469enddef
470
Bram Moolenaar0f769812020-09-12 18:32:34 +0200471def Test_local_function_shadows_global()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200472 var lines =<< trim END
Bram Moolenaar0f769812020-09-12 18:32:34 +0200473 vim9script
474 def g:Gfunc(): string
475 return 'global'
476 enddef
477 def AnotherFunc(): number
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200478 var Gfunc = function('len')
Bram Moolenaar0f769812020-09-12 18:32:34 +0200479 return Gfunc('testing')
480 enddef
481 g:Gfunc()->assert_equal('global')
482 AnotherFunc()->assert_equal(7)
483 delfunc g:Gfunc
484 END
485 CheckScriptSuccess(lines)
486
487 lines =<< trim END
488 vim9script
489 def g:Func(): string
490 return 'global'
491 enddef
492 def AnotherFunc()
493 g:Func = function('len')
494 enddef
495 AnotherFunc()
496 END
497 CheckScriptFailure(lines, 'E705:')
498 delfunc g:Func
499enddef
500
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200501func TakesOneArg(arg)
502 echo a:arg
503endfunc
504
505def Test_call_wrong_args()
Bram Moolenaard2c61702020-09-06 15:58:36 +0200506 CheckDefFailure(['TakesOneArg()'], 'E119:')
507 CheckDefFailure(['TakesOneArg(11, 22)'], 'E118:')
508 CheckDefFailure(['bufnr(xxx)'], 'E1001:')
509 CheckScriptFailure(['def Func(Ref: func(s: string))'], 'E475:')
Bram Moolenaaree8580e2020-08-28 17:19:07 +0200510
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200511 var lines =<< trim END
Bram Moolenaaree8580e2020-08-28 17:19:07 +0200512 vim9script
513 def Func(s: string)
514 echo s
515 enddef
516 Func([])
517 END
Bram Moolenaar77072282020-09-16 17:55:40 +0200518 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected string but got list<unknown>', 5)
Bram Moolenaarb185a402020-09-18 22:42:00 +0200519
520 lines =<< trim END
521 vim9script
522 def FuncOne(nr: number)
523 echo nr
524 enddef
525 def FuncTwo()
526 FuncOne()
527 enddef
528 defcompile
529 END
530 writefile(lines, 'Xscript')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200531 var didCatch = false
Bram Moolenaarb185a402020-09-18 22:42:00 +0200532 try
533 source Xscript
534 catch
535 assert_match('E119: Not enough arguments for function: <SNR>\d\+_FuncOne', v:exception)
536 assert_match('Xscript\[8\]..function <SNR>\d\+_FuncTwo, line 1', v:throwpoint)
537 didCatch = true
538 endtry
539 assert_true(didCatch)
540
541 lines =<< trim END
542 vim9script
543 def FuncOne(nr: number)
544 echo nr
545 enddef
546 def FuncTwo()
547 FuncOne(1, 2)
548 enddef
549 defcompile
550 END
551 writefile(lines, 'Xscript')
552 didCatch = false
553 try
554 source Xscript
555 catch
556 assert_match('E118: Too many arguments for function: <SNR>\d\+_FuncOne', v:exception)
557 assert_match('Xscript\[8\]..function <SNR>\d\+_FuncTwo, line 1', v:throwpoint)
558 didCatch = true
559 endtry
560 assert_true(didCatch)
561
562 delete('Xscript')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200563enddef
564
Bram Moolenaar50824712020-12-20 21:10:17 +0100565def Test_call_funcref_wrong_args()
566 var head =<< trim END
567 vim9script
568 def Func3(a1: string, a2: number, a3: list<number>)
569 echo a1 .. a2 .. a3[0]
570 enddef
571 def Testme()
572 var funcMap: dict<func> = {func: Func3}
573 END
574 var tail =<< trim END
575 enddef
576 Testme()
577 END
578 CheckScriptSuccess(head + ["funcMap['func']('str', 123, [1, 2, 3])"] + tail)
579
580 CheckScriptFailure(head + ["funcMap['func']('str', 123)"] + tail, 'E119:')
581 CheckScriptFailure(head + ["funcMap['func']('str', 123, [1], 4)"] + tail, 'E118:')
582enddef
583
Bram Moolenaarb4d16cb2020-11-05 18:45:46 +0100584def Test_call_lambda_args()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100585 CheckDefFailure(['echo ((i) => 0)()'],
586 'E119: Not enough arguments for function: ((i) => 0)()')
Bram Moolenaarb4d16cb2020-11-05 18:45:46 +0100587
588 var lines =<< trim END
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100589 var Ref = (x: number, y: number) => x + y
Bram Moolenaarb4d16cb2020-11-05 18:45:46 +0100590 echo Ref(1, 'x')
591 END
592 CheckDefFailure(lines, 'E1013: Argument 2: type mismatch, expected number but got string')
Bram Moolenaare68b02a2021-01-03 13:09:51 +0100593
594 lines =<< trim END
595 var Ref: func(job, string, number)
596 Ref = (x, y) => 0
597 END
598 CheckDefAndScriptFailure(lines, 'E1012:')
599
600 lines =<< trim END
601 var Ref: func(job, string)
602 Ref = (x, y, z) => 0
603 END
604 CheckDefAndScriptFailure(lines, 'E1012:')
Bram Moolenaarb4d16cb2020-11-05 18:45:46 +0100605enddef
606
Bram Moolenaar709664c2020-12-12 14:33:41 +0100607def Test_lambda_uses_assigned_var()
608 CheckDefSuccess([
609 'var x: any = "aaa"'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100610 'x = filter(["bbb"], (_, v) => v =~ x)'])
Bram Moolenaar709664c2020-12-12 14:33:41 +0100611enddef
612
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200613" Default arg and varargs
614def MyDefVarargs(one: string, two = 'foo', ...rest: list<string>): string
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200615 var res = one .. ',' .. two
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200616 for s in rest
617 res ..= ',' .. s
618 endfor
619 return res
620enddef
621
622def Test_call_def_varargs()
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200623 assert_fails('MyDefVarargs()', 'E119:', '', 1, 'Test_call_def_varargs')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200624 MyDefVarargs('one')->assert_equal('one,foo')
625 MyDefVarargs('one', 'two')->assert_equal('one,two')
626 MyDefVarargs('one', 'two', 'three')->assert_equal('one,two,three')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200627 CheckDefFailure(['MyDefVarargs("one", 22)'],
Bram Moolenaar77072282020-09-16 17:55:40 +0200628 'E1013: Argument 2: type mismatch, expected string but got number')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200629 CheckDefFailure(['MyDefVarargs("one", "two", 123)'],
Bram Moolenaar77072282020-09-16 17:55:40 +0200630 'E1013: Argument 3: type mismatch, expected string but got number')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200631
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200632 var lines =<< trim END
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200633 vim9script
634 def Func(...l: list<string>)
635 echo l
636 enddef
637 Func('a', 'b', 'c')
638 END
639 CheckScriptSuccess(lines)
640
641 lines =<< trim END
642 vim9script
643 def Func(...l: list<string>)
644 echo l
645 enddef
646 Func()
647 END
648 CheckScriptSuccess(lines)
649
650 lines =<< trim END
651 vim9script
Bram Moolenaar2f8cbc42020-09-16 17:22:59 +0200652 def Func(...l: any)
653 echo l
654 enddef
655 Func(0)
656 END
657 CheckScriptSuccess(lines)
658
659 lines =<< trim END
660 vim9script
Bram Moolenaar28022722020-09-21 22:02:49 +0200661 def Func(..._l: list<string>)
662 echo _l
663 enddef
664 Func('a', 'b', 'c')
665 END
666 CheckScriptSuccess(lines)
667
668 lines =<< trim END
669 vim9script
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200670 def Func(...l: list<string>)
671 echo l
672 enddef
673 Func(1, 2, 3)
674 END
Bram Moolenaar77072282020-09-16 17:55:40 +0200675 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200676
677 lines =<< trim END
678 vim9script
679 def Func(...l: list<string>)
680 echo l
681 enddef
682 Func('a', 9)
683 END
Bram Moolenaar77072282020-09-16 17:55:40 +0200684 CheckScriptFailure(lines, 'E1013: Argument 2: type mismatch')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200685
686 lines =<< trim END
687 vim9script
688 def Func(...l: list<string>)
689 echo l
690 enddef
691 Func(1, 'a')
692 END
Bram Moolenaar77072282020-09-16 17:55:40 +0200693 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200694enddef
695
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200696let s:value = ''
697
698def FuncOneDefArg(opt = 'text')
699 s:value = opt
700enddef
701
702def FuncTwoDefArg(nr = 123, opt = 'text'): string
703 return nr .. opt
704enddef
705
706def FuncVarargs(...arg: list<string>): string
707 return join(arg, ',')
708enddef
709
710def Test_func_type_varargs()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200711 var RefDefArg: func(?string)
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200712 RefDefArg = FuncOneDefArg
713 RefDefArg()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200714 s:value->assert_equal('text')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200715 RefDefArg('some')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200716 s:value->assert_equal('some')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200717
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200718 var RefDef2Arg: func(?number, ?string): string
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200719 RefDef2Arg = FuncTwoDefArg
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200720 RefDef2Arg()->assert_equal('123text')
721 RefDef2Arg(99)->assert_equal('99text')
722 RefDef2Arg(77, 'some')->assert_equal('77some')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200723
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200724 CheckDefFailure(['var RefWrong: func(string?)'], 'E1010:')
725 CheckDefFailure(['var RefWrong: func(?string, string)'], 'E1007:')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200726
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200727 var RefVarargs: func(...list<string>): string
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200728 RefVarargs = FuncVarargs
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200729 RefVarargs()->assert_equal('')
730 RefVarargs('one')->assert_equal('one')
731 RefVarargs('one', 'two')->assert_equal('one,two')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200732
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200733 CheckDefFailure(['var RefWrong: func(...list<string>, string)'], 'E110:')
734 CheckDefFailure(['var RefWrong: func(...list<string>, ?string)'], 'E110:')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200735enddef
736
Bram Moolenaar0b76b422020-04-07 22:05:08 +0200737" Only varargs
738def MyVarargsOnly(...args: list<string>): string
739 return join(args, ',')
740enddef
741
742def Test_call_varargs_only()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200743 MyVarargsOnly()->assert_equal('')
744 MyVarargsOnly('one')->assert_equal('one')
745 MyVarargsOnly('one', 'two')->assert_equal('one,two')
Bram Moolenaar77072282020-09-16 17:55:40 +0200746 CheckDefFailure(['MyVarargsOnly(1)'], 'E1013: Argument 1: type mismatch, expected string but got number')
747 CheckDefFailure(['MyVarargsOnly("one", 2)'], 'E1013: Argument 2: type mismatch, expected string but got number')
Bram Moolenaar0b76b422020-04-07 22:05:08 +0200748enddef
749
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200750def Test_using_var_as_arg()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200751 writefile(['def Func(x: number)', 'var x = 234', 'enddef', 'defcompile'], 'Xdef')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200752 assert_fails('so Xdef', 'E1006:', '', 1, 'Func')
Bram Moolenaard2c61702020-09-06 15:58:36 +0200753 delete('Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200754enddef
755
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200756def DictArg(arg: dict<string>)
757 arg['key'] = 'value'
758enddef
759
760def ListArg(arg: list<string>)
761 arg[0] = 'value'
762enddef
763
764def Test_assign_to_argument()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200765 # works for dict and list
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200766 var d: dict<string> = {}
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200767 DictArg(d)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200768 d['key']->assert_equal('value')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200769 var l: list<string> = []
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200770 ListArg(l)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200771 l[0]->assert_equal('value')
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200772
Bram Moolenaard2c61702020-09-06 15:58:36 +0200773 CheckScriptFailure(['def Func(arg: number)', 'arg = 3', 'enddef', 'defcompile'], 'E1090:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100774 delfunc! g:Func
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200775enddef
776
Bram Moolenaarb816dae2020-09-20 22:04:00 +0200777" These argument names are reserved in legacy functions.
778def WithReservedNames(firstline: string, lastline: string): string
779 return firstline .. lastline
780enddef
781
782def Test_argument_names()
783 assert_equal('OK', WithReservedNames('O', 'K'))
784enddef
785
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200786def Test_call_func_defined_later()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200787 g:DefinedLater('one')->assert_equal('one')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200788 assert_fails('NotDefined("one")', 'E117:', '', 2, 'Test_call_func_defined_later')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200789enddef
790
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200791func DefinedLater(arg)
792 return a:arg
793endfunc
794
795def Test_call_funcref()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200796 g:SomeFunc('abc')->assert_equal(3)
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200797 assert_fails('NotAFunc()', 'E117:', '', 2, 'Test_call_funcref') # comment after call
798 assert_fails('g:NotAFunc()', 'E117:', '', 3, 'Test_call_funcref')
Bram Moolenaar2f1980f2020-07-22 19:30:06 +0200799
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200800 var lines =<< trim END
Bram Moolenaar2f1980f2020-07-22 19:30:06 +0200801 vim9script
802 def RetNumber(): number
803 return 123
804 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200805 var Funcref: func: number = function('RetNumber')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200806 Funcref()->assert_equal(123)
Bram Moolenaar2f1980f2020-07-22 19:30:06 +0200807 END
808 CheckScriptSuccess(lines)
Bram Moolenaar0f60e802020-07-22 20:16:11 +0200809
810 lines =<< trim END
811 vim9script
812 def RetNumber(): number
813 return 123
814 enddef
815 def Bar(F: func: number): number
816 return F()
817 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200818 var Funcref = function('RetNumber')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200819 Bar(Funcref)->assert_equal(123)
Bram Moolenaar0f60e802020-07-22 20:16:11 +0200820 END
821 CheckScriptSuccess(lines)
Bram Moolenaarbfba8652020-07-23 20:09:10 +0200822
823 lines =<< trim END
824 vim9script
825 def UseNumber(nr: number)
826 echo nr
827 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200828 var Funcref: func(number) = function('UseNumber')
Bram Moolenaarbfba8652020-07-23 20:09:10 +0200829 Funcref(123)
830 END
831 CheckScriptSuccess(lines)
Bram Moolenaarb8070e32020-07-23 20:56:04 +0200832
833 lines =<< trim END
834 vim9script
835 def UseNumber(nr: number)
836 echo nr
837 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200838 var Funcref: func(string) = function('UseNumber')
Bram Moolenaarb8070e32020-07-23 20:56:04 +0200839 END
Bram Moolenaar5e654232020-09-16 15:22:00 +0200840 CheckScriptFailure(lines, 'E1012: Type mismatch; expected func(string) but got func(number)')
Bram Moolenaar4fc224c2020-07-26 17:56:25 +0200841
842 lines =<< trim END
843 vim9script
844 def EchoNr(nr = 34)
845 g:echo = nr
846 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200847 var Funcref: func(?number) = function('EchoNr')
Bram Moolenaar4fc224c2020-07-26 17:56:25 +0200848 Funcref()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200849 g:echo->assert_equal(34)
Bram Moolenaar4fc224c2020-07-26 17:56:25 +0200850 Funcref(123)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200851 g:echo->assert_equal(123)
Bram Moolenaar4fc224c2020-07-26 17:56:25 +0200852 END
853 CheckScriptSuccess(lines)
Bram Moolenaarace61322020-07-26 18:16:58 +0200854
855 lines =<< trim END
856 vim9script
857 def EchoList(...l: list<number>)
858 g:echo = l
859 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200860 var Funcref: func(...list<number>) = function('EchoList')
Bram Moolenaarace61322020-07-26 18:16:58 +0200861 Funcref()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200862 g:echo->assert_equal([])
Bram Moolenaarace61322020-07-26 18:16:58 +0200863 Funcref(1, 2, 3)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200864 g:echo->assert_equal([1, 2, 3])
Bram Moolenaarace61322020-07-26 18:16:58 +0200865 END
866 CheckScriptSuccess(lines)
Bram Moolenaar01865ad2020-07-26 18:33:09 +0200867
868 lines =<< trim END
869 vim9script
870 def OptAndVar(nr: number, opt = 12, ...l: list<number>): number
871 g:optarg = opt
872 g:listarg = l
873 return nr
874 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200875 var Funcref: func(number, ?number, ...list<number>): number = function('OptAndVar')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200876 Funcref(10)->assert_equal(10)
877 g:optarg->assert_equal(12)
878 g:listarg->assert_equal([])
Bram Moolenaar01865ad2020-07-26 18:33:09 +0200879
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200880 Funcref(11, 22)->assert_equal(11)
881 g:optarg->assert_equal(22)
882 g:listarg->assert_equal([])
Bram Moolenaar01865ad2020-07-26 18:33:09 +0200883
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200884 Funcref(17, 18, 1, 2, 3)->assert_equal(17)
885 g:optarg->assert_equal(18)
886 g:listarg->assert_equal([1, 2, 3])
Bram Moolenaar01865ad2020-07-26 18:33:09 +0200887 END
888 CheckScriptSuccess(lines)
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200889enddef
890
891let SomeFunc = function('len')
892let NotAFunc = 'text'
893
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200894def CombineFuncrefTypes()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200895 # same arguments, different return type
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200896 var Ref1: func(bool): string
897 var Ref2: func(bool): number
898 var Ref3: func(bool): any
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200899 Ref3 = g:cond ? Ref1 : Ref2
900
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200901 # different number of arguments
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200902 var Refa1: func(bool): number
903 var Refa2: func(bool, number): number
904 var Refa3: func: number
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200905 Refa3 = g:cond ? Refa1 : Refa2
906
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200907 # different argument types
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200908 var Refb1: func(bool, string): number
909 var Refb2: func(string, number): number
910 var Refb3: func(any, any): number
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200911 Refb3 = g:cond ? Refb1 : Refb2
912enddef
913
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200914def FuncWithForwardCall()
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200915 return g:DefinedEvenLater("yes")
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200916enddef
917
918def DefinedEvenLater(arg: string): string
919 return arg
920enddef
921
922def Test_error_in_nested_function()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200923 # Error in called function requires unwinding the call stack.
Bram Moolenaar44d66522020-09-06 22:26:57 +0200924 assert_fails('FuncWithForwardCall()', 'E1096:', '', 1, 'FuncWithForwardCall')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200925enddef
926
927def Test_return_type_wrong()
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200928 CheckScriptFailure([
929 'def Func(): number',
930 'return "a"',
931 'enddef',
932 'defcompile'], 'expected number but got string')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100933 delfunc! g:Func
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200934 CheckScriptFailure([
935 'def Func(): string',
936 'return 1',
937 'enddef',
938 'defcompile'], 'expected string but got number')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100939 delfunc! g:Func
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200940 CheckScriptFailure([
941 'def Func(): void',
942 'return "a"',
943 'enddef',
944 'defcompile'],
945 'E1096: Returning a value in a function without a return type')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100946 delfunc! g:Func
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200947 CheckScriptFailure([
948 'def Func()',
949 'return "a"',
950 'enddef',
951 'defcompile'],
952 'E1096: Returning a value in a function without a return type')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100953 delfunc! g:Func
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200954
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200955 CheckScriptFailure([
956 'def Func(): number',
957 'return',
958 'enddef',
959 'defcompile'], 'E1003:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100960 delfunc! g:Func
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200961
962 CheckScriptFailure(['def Func(): list', 'return []', 'enddef'], 'E1008:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100963 delfunc! g:Func
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200964 CheckScriptFailure(['def Func(): dict', 'return {}', 'enddef'], 'E1008:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100965 delfunc! g:Func
Bram Moolenaaree4e0c12020-04-06 21:35:05 +0200966 CheckScriptFailure(['def Func()', 'return 1'], 'E1057:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100967 delfunc! g:Func
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200968
969 CheckScriptFailure([
970 'vim9script',
971 'def FuncB()',
972 ' return 123',
973 'enddef',
974 'def FuncA()',
975 ' FuncB()',
976 'enddef',
977 'defcompile'], 'E1096:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200978enddef
979
980def Test_arg_type_wrong()
981 CheckScriptFailure(['def Func3(items: list)', 'echo "a"', 'enddef'], 'E1008: Missing <type>')
Bram Moolenaaree4e0c12020-04-06 21:35:05 +0200982 CheckScriptFailure(['def Func4(...)', 'echo "a"', 'enddef'], 'E1055: Missing name after ...')
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +0200983 CheckScriptFailure(['def Func5(items:string)', 'echo "a"'], 'E1069:')
Bram Moolenaar6e949782020-04-13 17:21:00 +0200984 CheckScriptFailure(['def Func5(items)', 'echo "a"'], 'E1077:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200985enddef
986
987def Test_vim9script_call()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200988 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200989 vim9script
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200990 var name = ''
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200991 def MyFunc(arg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200992 name = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200993 enddef
994 MyFunc('foobar')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200995 name->assert_equal('foobar')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200996
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200997 var str = 'barfoo'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200998 str->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200999 name->assert_equal('barfoo')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001000
Bram Moolenaar67979662020-06-20 22:50:47 +02001001 g:value = 'value'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001002 g:value->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001003 name->assert_equal('value')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001004
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001005 var listvar = []
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001006 def ListFunc(arg: list<number>)
1007 listvar = arg
1008 enddef
1009 [1, 2, 3]->ListFunc()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001010 listvar->assert_equal([1, 2, 3])
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001011
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001012 var dictvar = {}
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001013 def DictFunc(arg: dict<number>)
1014 dictvar = arg
1015 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +01001016 {a: 1, b: 2}->DictFunc()
1017 dictvar->assert_equal({a: 1, b: 2})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001018 def CompiledDict()
Bram Moolenaare0de1712020-12-02 17:36:54 +01001019 {a: 3, b: 4}->DictFunc()
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001020 enddef
1021 CompiledDict()
Bram Moolenaare0de1712020-12-02 17:36:54 +01001022 dictvar->assert_equal({a: 3, b: 4})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001023
Bram Moolenaare0de1712020-12-02 17:36:54 +01001024 {a: 3, b: 4}->DictFunc()
1025 dictvar->assert_equal({a: 3, b: 4})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001026
1027 ('text')->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001028 name->assert_equal('text')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001029 ("some")->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001030 name->assert_equal('some')
Bram Moolenaare6b53242020-07-01 17:28:33 +02001031
Bram Moolenaar13e12b82020-07-24 18:47:22 +02001032 # line starting with single quote is not a mark
Bram Moolenaar10409562020-07-29 20:00:38 +02001033 # line starting with double quote can be a method call
Bram Moolenaar3d48e252020-07-15 14:15:52 +02001034 'asdfasdf'->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001035 name->assert_equal('asdfasdf')
Bram Moolenaar10409562020-07-29 20:00:38 +02001036 "xyz"->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001037 name->assert_equal('xyz')
Bram Moolenaar3d48e252020-07-15 14:15:52 +02001038
1039 def UseString()
1040 'xyork'->MyFunc()
1041 enddef
1042 UseString()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001043 name->assert_equal('xyork')
Bram Moolenaar3d48e252020-07-15 14:15:52 +02001044
Bram Moolenaar10409562020-07-29 20:00:38 +02001045 def UseString2()
1046 "knife"->MyFunc()
1047 enddef
1048 UseString2()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001049 name->assert_equal('knife')
Bram Moolenaar10409562020-07-29 20:00:38 +02001050
Bram Moolenaar13e12b82020-07-24 18:47:22 +02001051 # prepending a colon makes it a mark
1052 new
1053 setline(1, ['aaa', 'bbb', 'ccc'])
1054 normal! 3Gmt1G
1055 :'t
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001056 getcurpos()[1]->assert_equal(3)
Bram Moolenaar13e12b82020-07-24 18:47:22 +02001057 bwipe!
1058
Bram Moolenaare6b53242020-07-01 17:28:33 +02001059 MyFunc(
1060 'continued'
1061 )
1062 assert_equal('continued',
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001063 name
Bram Moolenaare6b53242020-07-01 17:28:33 +02001064 )
1065
1066 call MyFunc(
1067 'more'
1068 ..
1069 'lines'
1070 )
1071 assert_equal(
1072 'morelines',
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001073 name)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001074 END
1075 writefile(lines, 'Xcall.vim')
1076 source Xcall.vim
1077 delete('Xcall.vim')
1078enddef
1079
1080def Test_vim9script_call_fail_decl()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001081 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001082 vim9script
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001083 var name = ''
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001084 def MyFunc(arg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001085 var name = 123
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001086 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +02001087 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001088 END
Bram Moolenaar6c4bfe42020-07-23 18:26:30 +02001089 CheckScriptFailure(lines, 'E1054:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001090enddef
1091
Bram Moolenaar65b95452020-07-19 14:03:09 +02001092def Test_vim9script_call_fail_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001093 var lines =<< trim END
Bram Moolenaar65b95452020-07-19 14:03:09 +02001094 vim9script
1095 def MyFunc(arg: string)
1096 echo arg
1097 enddef
1098 MyFunc(1234)
1099 END
Bram Moolenaar77072282020-09-16 17:55:40 +02001100 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected string but got number')
Bram Moolenaar65b95452020-07-19 14:03:09 +02001101enddef
1102
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001103def Test_vim9script_call_fail_const()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001104 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001105 vim9script
1106 const var = ''
1107 def MyFunc(arg: string)
1108 var = 'asdf'
1109 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +02001110 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001111 END
1112 writefile(lines, 'Xcall_const.vim')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001113 assert_fails('source Xcall_const.vim', 'E46:', '', 1, 'MyFunc')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001114 delete('Xcall_const.vim')
Bram Moolenaar3bdc90b2020-12-22 20:35:40 +01001115
1116 lines =<< trim END
1117 const g:Aconst = 77
1118 def Change()
1119 # comment
1120 g:Aconst = 99
1121 enddef
1122 call Change()
1123 unlet g:Aconst
1124 END
Bram Moolenaar1dcf55d2020-12-22 22:07:30 +01001125 CheckScriptFailure(lines, 'E741: Value is locked: Aconst', 2)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001126enddef
1127
1128" Test that inside :function a Python function can be defined, :def is not
1129" recognized.
1130func Test_function_python()
1131 CheckFeature python3
Bram Moolenaar727345e2020-09-27 23:33:59 +02001132 let py = 'python3'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001133 execute py "<< EOF"
1134def do_something():
1135 return 1
1136EOF
1137endfunc
1138
1139def Test_delfunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001140 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001141 vim9script
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02001142 def g:GoneSoon()
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001143 echo 'hello'
1144 enddef
1145
1146 def CallGoneSoon()
1147 GoneSoon()
1148 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +02001149 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001150
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02001151 delfunc g:GoneSoon
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001152 CallGoneSoon()
1153 END
1154 writefile(lines, 'XToDelFunc')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001155 assert_fails('so XToDelFunc', 'E933:', '', 1, 'CallGoneSoon')
1156 assert_fails('so XToDelFunc', 'E933:', '', 1, 'CallGoneSoon')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001157
1158 delete('XToDelFunc')
1159enddef
1160
1161def Test_redef_failure()
Bram Moolenaard2c61702020-09-06 15:58:36 +02001162 writefile(['def Func0(): string', 'return "Func0"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001163 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +02001164 writefile(['def Func1(): string', 'return "Func1"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001165 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +02001166 writefile(['def! Func0(): string', 'enddef', 'defcompile'], 'Xdef')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001167 assert_fails('so Xdef', 'E1027:', '', 1, 'Func0')
Bram Moolenaard2c61702020-09-06 15:58:36 +02001168 writefile(['def Func2(): string', 'return "Func2"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001169 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +02001170 delete('Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001171
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001172 g:Func0()->assert_equal(0)
1173 g:Func1()->assert_equal('Func1')
1174 g:Func2()->assert_equal('Func2')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001175
1176 delfunc! Func0
1177 delfunc! Func1
1178 delfunc! Func2
1179enddef
1180
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +02001181def Test_vim9script_func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001182 var lines =<< trim END
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +02001183 vim9script
1184 func Func(arg)
1185 echo a:arg
1186 endfunc
1187 Func('text')
1188 END
1189 writefile(lines, 'XVim9Func')
1190 so XVim9Func
1191
1192 delete('XVim9Func')
1193enddef
1194
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001195let s:funcResult = 0
1196
1197def FuncNoArgNoRet()
Bram Moolenaar53900992020-08-22 19:02:02 +02001198 s:funcResult = 11
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001199enddef
1200
1201def FuncNoArgRetNumber(): number
Bram Moolenaar53900992020-08-22 19:02:02 +02001202 s:funcResult = 22
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001203 return 1234
1204enddef
1205
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001206def FuncNoArgRetString(): string
Bram Moolenaar53900992020-08-22 19:02:02 +02001207 s:funcResult = 45
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001208 return 'text'
1209enddef
1210
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001211def FuncOneArgNoRet(arg: number)
Bram Moolenaar53900992020-08-22 19:02:02 +02001212 s:funcResult = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001213enddef
1214
1215def FuncOneArgRetNumber(arg: number): number
Bram Moolenaar53900992020-08-22 19:02:02 +02001216 s:funcResult = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001217 return arg
1218enddef
1219
Bram Moolenaar08938ee2020-04-11 23:17:17 +02001220def FuncTwoArgNoRet(one: bool, two: number)
Bram Moolenaar53900992020-08-22 19:02:02 +02001221 s:funcResult = two
Bram Moolenaar08938ee2020-04-11 23:17:17 +02001222enddef
1223
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001224def FuncOneArgRetString(arg: string): string
1225 return arg
1226enddef
1227
Bram Moolenaar89228602020-04-05 22:14:54 +02001228def FuncOneArgRetAny(arg: any): any
1229 return arg
1230enddef
1231
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001232def Test_func_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001233 var Ref1: func()
Bram Moolenaar53900992020-08-22 19:02:02 +02001234 s:funcResult = 0
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001235 Ref1 = FuncNoArgNoRet
1236 Ref1()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001237 s:funcResult->assert_equal(11)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001238
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001239 var Ref2: func
Bram Moolenaar53900992020-08-22 19:02:02 +02001240 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001241 Ref2 = FuncNoArgNoRet
1242 Ref2()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001243 s:funcResult->assert_equal(11)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001244
Bram Moolenaar53900992020-08-22 19:02:02 +02001245 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001246 Ref2 = FuncOneArgNoRet
1247 Ref2(12)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001248 s:funcResult->assert_equal(12)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001249
Bram Moolenaar53900992020-08-22 19:02:02 +02001250 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001251 Ref2 = FuncNoArgRetNumber
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001252 Ref2()->assert_equal(1234)
1253 s:funcResult->assert_equal(22)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001254
Bram Moolenaar53900992020-08-22 19:02:02 +02001255 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001256 Ref2 = FuncOneArgRetNumber
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001257 Ref2(13)->assert_equal(13)
1258 s:funcResult->assert_equal(13)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001259enddef
1260
Bram Moolenaar9978d472020-07-05 16:01:56 +02001261def Test_repeat_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001262 var res = 0
Bram Moolenaar9978d472020-07-05 16:01:56 +02001263 for n in repeat([1], 3)
1264 res += n
1265 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001266 res->assert_equal(3)
Bram Moolenaarfce82b32020-07-05 16:07:21 +02001267
1268 res = 0
1269 for n in add([1, 2], 3)
1270 res += n
1271 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001272 res->assert_equal(6)
Bram Moolenaar9978d472020-07-05 16:01:56 +02001273enddef
1274
Bram Moolenaar846178a2020-07-05 17:04:13 +02001275def Test_argv_return_type()
1276 next fileone filetwo
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001277 var res = ''
Bram Moolenaar846178a2020-07-05 17:04:13 +02001278 for name in argv()
1279 res ..= name
1280 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001281 res->assert_equal('fileonefiletwo')
Bram Moolenaar846178a2020-07-05 17:04:13 +02001282enddef
1283
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001284def Test_func_type_part()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001285 var RefVoid: func: void
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001286 RefVoid = FuncNoArgNoRet
1287 RefVoid = FuncOneArgNoRet
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001288 CheckDefFailure(['var RefVoid: func: void', 'RefVoid = FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func(...) but got func(): number')
1289 CheckDefFailure(['var RefVoid: func: void', 'RefVoid = FuncNoArgRetString'], 'E1012: Type mismatch; expected func(...) but got func(): string')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001290
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001291 var RefAny: func(): any
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001292 RefAny = FuncNoArgRetNumber
1293 RefAny = FuncNoArgRetString
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001294 CheckDefFailure(['var RefAny: func(): any', 'RefAny = FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(): any but got func()')
1295 CheckDefFailure(['var RefAny: func(): any', 'RefAny = FuncOneArgNoRet'], 'E1012: Type mismatch; expected func(): any but got func(number)')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001296
Bram Moolenaar6abd3dc2020-10-04 14:17:32 +02001297 var RefAnyNoArgs: func: any = RefAny
1298
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001299 var RefNr: func: number
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001300 RefNr = FuncNoArgRetNumber
1301 RefNr = FuncOneArgRetNumber
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001302 CheckDefFailure(['var RefNr: func: number', 'RefNr = FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(...): number but got func()')
1303 CheckDefFailure(['var RefNr: func: number', 'RefNr = FuncNoArgRetString'], 'E1012: Type mismatch; expected func(...): number but got func(): string')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001304
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001305 var RefStr: func: string
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001306 RefStr = FuncNoArgRetString
1307 RefStr = FuncOneArgRetString
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001308 CheckDefFailure(['var RefStr: func: string', 'RefStr = FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(...): string but got func()')
1309 CheckDefFailure(['var RefStr: func: string', 'RefStr = FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func(...): string but got func(): number')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001310enddef
1311
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001312def Test_func_type_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001313 CheckDefFailure(['var ref1: func()'], 'E704:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001314
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001315 CheckDefFailure(['var Ref1: func()', 'Ref1 = FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func() but got func(): number')
1316 CheckDefFailure(['var Ref1: func()', 'Ref1 = FuncOneArgNoRet'], 'E1012: Type mismatch; expected func() but got func(number)')
1317 CheckDefFailure(['var Ref1: func()', 'Ref1 = FuncOneArgRetNumber'], 'E1012: Type mismatch; expected func() but got func(number): number')
1318 CheckDefFailure(['var Ref1: func(bool)', 'Ref1 = FuncTwoArgNoRet'], 'E1012: Type mismatch; expected func(bool) but got func(bool, number)')
1319 CheckDefFailure(['var Ref1: func(?bool)', 'Ref1 = FuncTwoArgNoRet'], 'E1012: Type mismatch; expected func(?bool) but got func(bool, number)')
1320 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 +02001321
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001322 CheckDefFailure(['var RefWrong: func(string ,number)'], 'E1068:')
1323 CheckDefFailure(['var RefWrong: func(string,number)'], 'E1069:')
1324 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:')
1325 CheckDefFailure(['var RefWrong: func(bool):string'], 'E1069:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001326enddef
1327
Bram Moolenaar89228602020-04-05 22:14:54 +02001328def Test_func_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001329 var nr: number
Bram Moolenaar89228602020-04-05 22:14:54 +02001330 nr = FuncNoArgRetNumber()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001331 nr->assert_equal(1234)
Bram Moolenaar89228602020-04-05 22:14:54 +02001332
1333 nr = FuncOneArgRetAny(122)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001334 nr->assert_equal(122)
Bram Moolenaar89228602020-04-05 22:14:54 +02001335
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001336 var str: string
Bram Moolenaar89228602020-04-05 22:14:54 +02001337 str = FuncOneArgRetAny('yes')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001338 str->assert_equal('yes')
Bram Moolenaar89228602020-04-05 22:14:54 +02001339
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001340 CheckDefFailure(['var str: string', 'str = FuncNoArgRetNumber()'], 'E1012: Type mismatch; expected string but got number')
Bram Moolenaar89228602020-04-05 22:14:54 +02001341enddef
1342
Bram Moolenaar6abd3dc2020-10-04 14:17:32 +02001343def Test_func_common_type()
1344 def FuncOne(n: number): number
1345 return n
1346 enddef
1347 def FuncTwo(s: string): number
1348 return len(s)
1349 enddef
1350 def FuncThree(n: number, s: string): number
1351 return n + len(s)
1352 enddef
1353 var list = [FuncOne, FuncTwo, FuncThree]
1354 assert_equal(8, list[0](8))
1355 assert_equal(4, list[1]('word'))
1356 assert_equal(7, list[2](3, 'word'))
1357enddef
1358
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001359def MultiLine(
1360 arg1: string,
1361 arg2 = 1234,
1362 ...rest: list<string>
1363 ): string
1364 return arg1 .. arg2 .. join(rest, '-')
1365enddef
1366
Bram Moolenaar2c330432020-04-13 14:41:35 +02001367def MultiLineComment(
1368 arg1: string, # comment
1369 arg2 = 1234, # comment
1370 ...rest: list<string> # comment
1371 ): string # comment
1372 return arg1 .. arg2 .. join(rest, '-')
1373enddef
1374
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001375def Test_multiline()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001376 MultiLine('text')->assert_equal('text1234')
1377 MultiLine('text', 777)->assert_equal('text777')
1378 MultiLine('text', 777, 'one')->assert_equal('text777one')
1379 MultiLine('text', 777, 'one', 'two')->assert_equal('text777one-two')
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001380enddef
1381
Bram Moolenaar23e03252020-04-12 22:22:31 +02001382func Test_multiline_not_vim9()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001383 call MultiLine('text')->assert_equal('text1234')
1384 call MultiLine('text', 777)->assert_equal('text777')
1385 call MultiLine('text', 777, 'one')->assert_equal('text777one')
1386 call MultiLine('text', 777, 'one', 'two')->assert_equal('text777one-two')
Bram Moolenaar23e03252020-04-12 22:22:31 +02001387endfunc
1388
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001389
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02001390" When using CheckScriptFailure() for the below test, E1010 is generated instead
1391" of E1056.
1392func Test_E1056_1059()
1393 let caught_1056 = 0
1394 try
1395 def F():
1396 return 1
1397 enddef
1398 catch /E1056:/
1399 let caught_1056 = 1
1400 endtry
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001401 eval caught_1056->assert_equal(1)
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02001402
1403 let caught_1059 = 0
1404 try
1405 def F5(items : list)
1406 echo 'a'
1407 enddef
1408 catch /E1059:/
1409 let caught_1059 = 1
1410 endtry
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001411 eval caught_1059->assert_equal(1)
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02001412endfunc
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001413
Bram Moolenaar015f4262020-05-05 21:25:22 +02001414func DelMe()
1415 echo 'DelMe'
1416endfunc
1417
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001418def Test_error_reporting()
1419 # comment lines at the start of the function
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001420 var lines =<< trim END
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001421 " comment
1422 def Func()
1423 # comment
1424 # comment
1425 invalid
1426 enddef
1427 defcompile
1428 END
Bram Moolenaar08052222020-09-14 17:04:31 +02001429 writefile(lines, 'Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001430 try
1431 source Xdef
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001432 assert_report('should have failed')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001433 catch /E476:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001434 v:exception->assert_match('Invalid command: invalid')
1435 v:throwpoint->assert_match(', line 3$')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001436 endtry
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001437 delfunc! g:Func
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001438
1439 # comment lines after the start of the function
1440 lines =<< trim END
1441 " comment
1442 def Func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001443 var x = 1234
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001444 # comment
1445 # comment
1446 invalid
1447 enddef
1448 defcompile
1449 END
Bram Moolenaar08052222020-09-14 17:04:31 +02001450 writefile(lines, 'Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001451 try
1452 source Xdef
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001453 assert_report('should have failed')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001454 catch /E476:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001455 v:exception->assert_match('Invalid command: invalid')
1456 v:throwpoint->assert_match(', line 4$')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001457 endtry
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001458 delfunc! g:Func
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001459
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001460 lines =<< trim END
1461 vim9script
1462 def Func()
Bram Moolenaare0de1712020-12-02 17:36:54 +01001463 var db = {foo: 1, bar: 2}
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001464 # comment
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001465 var x = db.asdf
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001466 enddef
1467 defcompile
1468 Func()
1469 END
Bram Moolenaar08052222020-09-14 17:04:31 +02001470 writefile(lines, 'Xdef')
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001471 try
1472 source Xdef
1473 assert_report('should have failed')
1474 catch /E716:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001475 v:throwpoint->assert_match('_Func, line 3$')
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001476 endtry
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001477 delfunc! g:Func
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001478
Bram Moolenaar08052222020-09-14 17:04:31 +02001479 delete('Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001480enddef
1481
Bram Moolenaar015f4262020-05-05 21:25:22 +02001482def Test_deleted_function()
1483 CheckDefExecFailure([
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001484 'var RefMe: func = function("g:DelMe")',
Bram Moolenaar015f4262020-05-05 21:25:22 +02001485 'delfunc g:DelMe',
1486 'echo RefMe()'], 'E117:')
1487enddef
1488
1489def Test_unknown_function()
1490 CheckDefExecFailure([
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001491 'var Ref: func = function("NotExist")',
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +02001492 'delfunc g:NotExist'], 'E700:')
Bram Moolenaar015f4262020-05-05 21:25:22 +02001493enddef
1494
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +02001495def RefFunc(Ref: func(string): string): string
1496 return Ref('more')
1497enddef
1498
1499def Test_closure_simple()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001500 var local = 'some '
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001501 RefFunc((s) => local .. s)->assert_equal('some more')
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +02001502enddef
1503
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001504def MakeRef()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001505 var local = 'some '
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001506 g:Ref = (s) => local .. s
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001507enddef
1508
1509def Test_closure_ref_after_return()
1510 MakeRef()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001511 g:Ref('thing')->assert_equal('some thing')
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001512 unlet g:Ref
1513enddef
1514
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001515def MakeTwoRefs()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001516 var local = ['some']
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001517 g:Extend = (s) => local->add(s)
1518 g:Read = () => local
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001519enddef
1520
1521def Test_closure_two_refs()
1522 MakeTwoRefs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001523 join(g:Read(), ' ')->assert_equal('some')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001524 g:Extend('more')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001525 join(g:Read(), ' ')->assert_equal('some more')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001526 g:Extend('even')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001527 join(g:Read(), ' ')->assert_equal('some more even')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001528
1529 unlet g:Extend
1530 unlet g:Read
1531enddef
1532
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001533def ReadRef(Ref: func(): list<string>): string
1534 return join(Ref(), ' ')
1535enddef
1536
Bram Moolenaar5e654232020-09-16 15:22:00 +02001537def ExtendRef(Ref: func(string): list<string>, add: string)
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001538 Ref(add)
1539enddef
1540
1541def Test_closure_two_indirect_refs()
Bram Moolenaarf7779c62020-05-03 15:38:16 +02001542 MakeTwoRefs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001543 ReadRef(g:Read)->assert_equal('some')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001544 ExtendRef(g:Extend, 'more')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001545 ReadRef(g:Read)->assert_equal('some more')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001546 ExtendRef(g:Extend, 'even')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001547 ReadRef(g:Read)->assert_equal('some more even')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001548
1549 unlet g:Extend
1550 unlet g:Read
1551enddef
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001552
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001553def MakeArgRefs(theArg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001554 var local = 'loc_val'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001555 g:UseArg = (s) => theArg .. '/' .. local .. '/' .. s
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001556enddef
1557
1558def MakeArgRefsVarargs(theArg: string, ...rest: list<string>)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001559 var local = 'the_loc'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001560 g:UseVararg = (s) => theArg .. '/' .. local .. '/' .. s .. '/' .. join(rest)
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001561enddef
1562
1563def Test_closure_using_argument()
1564 MakeArgRefs('arg_val')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001565 g:UseArg('call_val')->assert_equal('arg_val/loc_val/call_val')
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001566
1567 MakeArgRefsVarargs('arg_val', 'one', 'two')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001568 g:UseVararg('call_val')->assert_equal('arg_val/the_loc/call_val/one two')
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001569
1570 unlet g:UseArg
1571 unlet g:UseVararg
1572enddef
1573
Bram Moolenaar85d5e2b2020-10-10 14:13:01 +02001574def MakeGetAndAppendRefs()
1575 var local = 'a'
1576
1577 def Append(arg: string)
1578 local ..= arg
1579 enddef
1580 g:Append = Append
1581
1582 def Get(): string
1583 return local
1584 enddef
1585 g:Get = Get
1586enddef
1587
1588def Test_closure_append_get()
1589 MakeGetAndAppendRefs()
1590 g:Get()->assert_equal('a')
1591 g:Append('-b')
1592 g:Get()->assert_equal('a-b')
1593 g:Append('-c')
1594 g:Get()->assert_equal('a-b-c')
1595
1596 unlet g:Append
1597 unlet g:Get
1598enddef
Bram Moolenaarb68b3462020-05-06 21:06:30 +02001599
Bram Moolenaar04b12692020-05-04 23:24:44 +02001600def Test_nested_closure()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001601 var local = 'text'
Bram Moolenaar04b12692020-05-04 23:24:44 +02001602 def Closure(arg: string): string
1603 return local .. arg
1604 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001605 Closure('!!!')->assert_equal('text!!!')
Bram Moolenaar04b12692020-05-04 23:24:44 +02001606enddef
1607
Bram Moolenaar6f5b6df2020-05-16 21:20:12 +02001608func GetResult(Ref)
1609 return a:Ref('some')
1610endfunc
1611
1612def Test_call_closure_not_compiled()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001613 var text = 'text'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001614 g:Ref = (s) => s .. text
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001615 GetResult(g:Ref)->assert_equal('sometext')
Bram Moolenaar6f5b6df2020-05-16 21:20:12 +02001616enddef
1617
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001618def Test_double_closure_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001619 var lines =<< trim END
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001620 vim9script
1621 def Func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001622 var name = 0
1623 for i in range(2)
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001624 timer_start(0, () => name)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001625 endfor
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001626 enddef
1627 Func()
1628 END
Bram Moolenaar148ce7a2020-09-23 21:57:23 +02001629 CheckScriptSuccess(lines)
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001630enddef
1631
Bram Moolenaar85d5e2b2020-10-10 14:13:01 +02001632def Test_nested_closure_used()
1633 var lines =<< trim END
1634 vim9script
1635 def Func()
1636 var x = 'hello'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001637 var Closure = () => x
1638 g:Myclosure = () => Closure()
Bram Moolenaar85d5e2b2020-10-10 14:13:01 +02001639 enddef
1640 Func()
1641 assert_equal('hello', g:Myclosure())
1642 END
1643 CheckScriptSuccess(lines)
1644enddef
Bram Moolenaar0876c782020-10-07 19:08:04 +02001645
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02001646def Test_nested_closure_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001647 var lines =<< trim END
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02001648 vim9script
1649 def FuncA()
1650 FuncB(0)
1651 enddef
1652 def FuncB(n: number): list<string>
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001653 return map([0], (_, v) => n)
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02001654 enddef
1655 FuncA()
1656 END
1657 CheckScriptFailure(lines, 'E1012:')
1658enddef
1659
Bram Moolenaarf112f302020-12-20 17:47:52 +01001660def Test_global_closure()
1661 var lines =<< trim END
1662 vim9script
1663 def ReverseEveryNLines(n: number, line1: number, line2: number)
1664 var mods = 'sil keepj keepp lockm '
1665 var range = ':' .. line1 .. ',' .. line2
1666 def g:Offset(): number
1667 var offset = (line('.') - line1 + 1) % n
1668 return offset != 0 ? offset : n
1669 enddef
1670 exe mods .. range .. 'g/^/exe "m .-" .. g:Offset()'
1671 enddef
1672
1673 new
1674 repeat(['aaa', 'bbb', 'ccc'], 3)->setline(1)
1675 ReverseEveryNLines(3, 1, 9)
1676 END
1677 CheckScriptSuccess(lines)
1678 var expected = repeat(['ccc', 'bbb', 'aaa'], 3)
1679 assert_equal(expected, getline(1, 9))
1680 bwipe!
1681enddef
1682
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01001683def Test_global_closure_called_directly()
1684 var lines =<< trim END
1685 vim9script
1686 def Outer()
1687 var x = 1
1688 def g:Inner()
1689 var y = x
1690 x += 1
1691 assert_equal(1, y)
1692 enddef
1693 g:Inner()
1694 assert_equal(2, x)
1695 enddef
1696 Outer()
1697 END
1698 CheckScriptSuccess(lines)
1699 delfunc g:Inner
1700enddef
1701
Bram Moolenaar34c54eb2020-11-25 19:15:19 +01001702def Test_failure_in_called_function()
1703 # this was using the frame index as the return value
1704 var lines =<< trim END
1705 vim9script
1706 au TerminalWinOpen * eval [][0]
1707 def PopupTerm(a: any)
1708 # make sure typvals on stack are string
1709 ['a', 'b', 'c', 'd', 'e', 'f', 'g']->join()
1710 FireEvent()
1711 enddef
1712 def FireEvent()
1713 do TerminalWinOpen
1714 enddef
1715 # use try/catch to make eval fail
1716 try
1717 call PopupTerm(0)
1718 catch
1719 endtry
1720 au! TerminalWinOpen
1721 END
1722 CheckScriptSuccess(lines)
1723enddef
1724
Bram Moolenaar5366e1a2020-10-01 13:01:34 +02001725def Test_nested_lambda()
1726 var lines =<< trim END
1727 vim9script
1728 def Func()
1729 var x = 4
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001730 var Lambda1 = () => 7
1731 var Lambda2 = () => [Lambda1(), x]
Bram Moolenaar5366e1a2020-10-01 13:01:34 +02001732 var res = Lambda2()
1733 assert_equal([7, 4], res)
1734 enddef
1735 Func()
1736 END
1737 CheckScriptSuccess(lines)
1738enddef
1739
Bram Moolenaar52bf81c2020-11-17 18:50:44 +01001740def Shadowed(): list<number>
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001741 var FuncList: list<func: number> = [() => 42]
1742 return FuncList->map((_, Shadowed) => Shadowed())
Bram Moolenaar52bf81c2020-11-17 18:50:44 +01001743enddef
1744
1745def Test_lambda_arg_shadows_func()
1746 assert_equal([42], Shadowed())
1747enddef
1748
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02001749def Line_continuation_in_def(dir: string = ''): string
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001750 var path: string = empty(dir)
1751 \ ? 'empty'
1752 \ : 'full'
1753 return path
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02001754enddef
1755
1756def Test_line_continuation_in_def()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001757 Line_continuation_in_def('.')->assert_equal('full')
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02001758enddef
1759
Bram Moolenaar2ea95b62020-11-19 21:47:56 +01001760def Test_script_var_in_lambda()
1761 var lines =<< trim END
1762 vim9script
1763 var script = 'test'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001764 assert_equal(['test'], map(['one'], () => script))
Bram Moolenaar2ea95b62020-11-19 21:47:56 +01001765 END
1766 CheckScriptSuccess(lines)
1767enddef
1768
Bram Moolenaar5e654232020-09-16 15:22:00 +02001769def Line_continuation_in_lambda(): list<string>
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001770 var x = range(97, 100)
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001771 ->map((_, v) => nr2char(v)
1772 ->toupper())
Bram Moolenaar7a4b8982020-07-08 17:36:21 +02001773 ->reverse()
1774 return x
1775enddef
1776
1777def Test_line_continuation_in_lambda()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001778 Line_continuation_in_lambda()->assert_equal(['D', 'C', 'B', 'A'])
Bram Moolenaar7a4b8982020-07-08 17:36:21 +02001779enddef
1780
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001781func Test_silent_echo()
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001782 CheckScreendump
1783
1784 let lines =<< trim END
1785 vim9script
1786 def EchoNothing()
1787 silent echo ''
1788 enddef
1789 defcompile
1790 END
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001791 call writefile(lines, 'XTest_silent_echo')
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001792
1793 " Check that the balloon shows up after a mouse move
1794 let buf = RunVimInTerminal('-S XTest_silent_echo', {'rows': 6})
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001795 call term_sendkeys(buf, ":abc")
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001796 call VerifyScreenDump(buf, 'Test_vim9_silent_echo', {})
1797
1798 " clean up
1799 call StopVimInTerminal(buf)
1800 call delete('XTest_silent_echo')
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001801endfunc
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001802
Bram Moolenaar171fb922020-10-28 16:54:47 +01001803def SilentlyError()
1804 execute('silent! invalid')
1805 g:did_it = 'yes'
1806enddef
1807
Bram Moolenaar28ee8922020-10-28 20:20:00 +01001808func UserError()
1809 silent! invalid
1810endfunc
1811
1812def SilentlyUserError()
1813 UserError()
1814 g:did_it = 'yes'
1815enddef
Bram Moolenaar171fb922020-10-28 16:54:47 +01001816
1817" This can't be a :def function, because the assert would not be reached.
Bram Moolenaar171fb922020-10-28 16:54:47 +01001818func Test_ignore_silent_error()
1819 let g:did_it = 'no'
1820 call SilentlyError()
1821 call assert_equal('yes', g:did_it)
1822
Bram Moolenaar28ee8922020-10-28 20:20:00 +01001823 let g:did_it = 'no'
1824 call SilentlyUserError()
1825 call assert_equal('yes', g:did_it)
Bram Moolenaar171fb922020-10-28 16:54:47 +01001826
1827 unlet g:did_it
1828endfunc
1829
Bram Moolenaarcd030c42020-10-30 21:49:40 +01001830def Test_ignore_silent_error_in_filter()
1831 var lines =<< trim END
1832 vim9script
1833 def Filter(winid: number, key: string): bool
1834 if key == 'o'
1835 silent! eval [][0]
1836 return true
1837 endif
1838 return popup_filter_menu(winid, key)
1839 enddef
1840
Bram Moolenaare0de1712020-12-02 17:36:54 +01001841 popup_create('popup', {filter: Filter})
Bram Moolenaarcd030c42020-10-30 21:49:40 +01001842 feedkeys("o\r", 'xnt')
1843 END
1844 CheckScriptSuccess(lines)
1845enddef
1846
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001847def Fibonacci(n: number): number
1848 if n < 2
1849 return n
1850 else
1851 return Fibonacci(n - 1) + Fibonacci(n - 2)
1852 endif
1853enddef
1854
Bram Moolenaar985116a2020-07-12 17:31:09 +02001855def Test_recursive_call()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001856 Fibonacci(20)->assert_equal(6765)
Bram Moolenaar985116a2020-07-12 17:31:09 +02001857enddef
1858
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001859def TreeWalk(dir: string): list<any>
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001860 return readdir(dir)->map((_, val) =>
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001861 fnamemodify(dir .. '/' .. val, ':p')->isdirectory()
Bram Moolenaar2bede172020-11-19 18:53:18 +01001862 ? {[val]: TreeWalk(dir .. '/' .. val)}
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001863 : val
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001864 )
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001865enddef
1866
1867def Test_closure_in_map()
1868 mkdir('XclosureDir/tdir', 'p')
1869 writefile(['111'], 'XclosureDir/file1')
1870 writefile(['222'], 'XclosureDir/file2')
1871 writefile(['333'], 'XclosureDir/tdir/file3')
1872
Bram Moolenaare0de1712020-12-02 17:36:54 +01001873 TreeWalk('XclosureDir')->assert_equal(['file1', 'file2', {tdir: ['file3']}])
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001874
1875 delete('XclosureDir', 'rf')
1876enddef
1877
Bram Moolenaar7b5d5442020-10-04 13:42:34 +02001878def Test_invalid_function_name()
1879 var lines =<< trim END
1880 vim9script
1881 def s: list<string>
1882 END
1883 CheckScriptFailure(lines, 'E129:')
1884
1885 lines =<< trim END
1886 vim9script
1887 def g: list<string>
1888 END
1889 CheckScriptFailure(lines, 'E129:')
1890
1891 lines =<< trim END
1892 vim9script
1893 def <SID>: list<string>
1894 END
1895 CheckScriptFailure(lines, 'E884:')
1896
1897 lines =<< trim END
1898 vim9script
1899 def F list<string>
1900 END
1901 CheckScriptFailure(lines, 'E488:')
1902enddef
1903
Bram Moolenaara90afb92020-07-15 22:38:56 +02001904def Test_partial_call()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001905 var Xsetlist = function('setloclist', [0])
Bram Moolenaare0de1712020-12-02 17:36:54 +01001906 Xsetlist([], ' ', {title: 'test'})
1907 getloclist(0, {title: 1})->assert_equal({title: 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02001908
1909 Xsetlist = function('setloclist', [0, [], ' '])
Bram Moolenaare0de1712020-12-02 17:36:54 +01001910 Xsetlist({title: 'test'})
1911 getloclist(0, {title: 1})->assert_equal({title: 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02001912
1913 Xsetlist = function('setqflist')
Bram Moolenaare0de1712020-12-02 17:36:54 +01001914 Xsetlist([], ' ', {title: 'test'})
1915 getqflist({title: 1})->assert_equal({title: 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02001916
1917 Xsetlist = function('setqflist', [[], ' '])
Bram Moolenaare0de1712020-12-02 17:36:54 +01001918 Xsetlist({title: 'test'})
1919 getqflist({title: 1})->assert_equal({title: 'test'})
Bram Moolenaar6abd3dc2020-10-04 14:17:32 +02001920
1921 var Len: func: number = function('len', ['word'])
1922 assert_equal(4, Len())
Bram Moolenaara90afb92020-07-15 22:38:56 +02001923enddef
1924
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001925def Test_cmd_modifier()
1926 tab echo '0'
Bram Moolenaard2c61702020-09-06 15:58:36 +02001927 CheckDefFailure(['5tab echo 3'], 'E16:')
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001928enddef
1929
1930def Test_restore_modifiers()
1931 # check that when compiling a :def function command modifiers are not messed
1932 # up.
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001933 var lines =<< trim END
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001934 vim9script
1935 set eventignore=
1936 autocmd QuickFixCmdPost * copen
1937 def AutocmdsDisabled()
Bram Moolenaar6cf7e3b2020-10-28 14:31:16 +01001938 eval 0
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001939 enddef
1940 func Func()
1941 noautocmd call s:AutocmdsDisabled()
1942 let g:ei_after = &eventignore
1943 endfunc
1944 Func()
1945 END
1946 CheckScriptSuccess(lines)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001947 g:ei_after->assert_equal('')
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001948enddef
1949
Bram Moolenaardfa3d552020-09-10 22:05:08 +02001950def StackTop()
1951 eval 1
1952 eval 2
1953 # call not on fourth line
1954 StackBot()
1955enddef
1956
1957def StackBot()
1958 # throw an error
1959 eval [][0]
1960enddef
1961
1962def Test_callstack_def()
1963 try
1964 StackTop()
1965 catch
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001966 v:throwpoint->assert_match('Test_callstack_def\[2\]..StackTop\[4\]..StackBot, line 2')
Bram Moolenaardfa3d552020-09-10 22:05:08 +02001967 endtry
1968enddef
1969
Bram Moolenaare8211a32020-10-09 22:04:29 +02001970" Re-using spot for variable used in block
1971def Test_block_scoped_var()
1972 var lines =<< trim END
1973 vim9script
1974 def Func()
1975 var x = ['a', 'b', 'c']
1976 if 1
1977 var y = 'x'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001978 map(x, () => y)
Bram Moolenaare8211a32020-10-09 22:04:29 +02001979 endif
1980 var z = x
1981 assert_equal(['x', 'x', 'x'], z)
1982 enddef
1983 Func()
1984 END
1985 CheckScriptSuccess(lines)
1986enddef
1987
Bram Moolenaareeece9e2020-11-20 19:26:48 +01001988def Test_reset_did_emsg()
1989 var lines =<< trim END
1990 @s = 'blah'
1991 au BufWinLeave * #
1992 def Func()
1993 var winid = popup_create('popup', {})
1994 exe '*s'
1995 popup_close(winid)
1996 enddef
1997 Func()
1998 END
1999 CheckScriptFailure(lines, 'E492:', 8)
Bram Moolenaar2d870f82020-12-05 13:41:01 +01002000 delfunc! g:Func
Bram Moolenaareeece9e2020-11-20 19:26:48 +01002001enddef
2002
Bram Moolenaar57f799e2020-12-12 20:42:19 +01002003def Test_did_emsg_reset()
2004 # executing an autocommand resets did_emsg, this should not result in a
2005 # builtin function considered failing
2006 var lines =<< trim END
2007 vim9script
2008 au BufWinLeave * #
2009 def Func()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002010 popup_menu('', {callback: () => popup_create('', {})->popup_close()})
Bram Moolenaar57f799e2020-12-12 20:42:19 +01002011 eval [][0]
2012 enddef
2013 nno <F3> <cmd>call <sid>Func()<cr>
2014 feedkeys("\<F3>\e", 'xt')
2015 END
2016 writefile(lines, 'XemsgReset')
2017 assert_fails('so XemsgReset', ['E684:', 'E684:'], lines, 2)
2018 delete('XemsgReset')
2019 nunmap <F3>
2020 au! BufWinLeave
2021enddef
2022
Bram Moolenaar56602ba2020-12-05 21:22:08 +01002023def Test_abort_with_silent_call()
2024 var lines =<< trim END
2025 vim9script
2026 g:result = 'none'
2027 def Func()
2028 g:result += 3
2029 g:result = 'yes'
2030 enddef
2031 # error is silenced, but function aborts on error
2032 silent! Func()
2033 assert_equal('none', g:result)
2034 unlet g:result
2035 END
2036 CheckScriptSuccess(lines)
2037enddef
2038
Bram Moolenaarf665e972020-12-05 19:17:16 +01002039def Test_continues_with_silent_error()
2040 var lines =<< trim END
2041 vim9script
2042 g:result = 'none'
2043 def Func()
2044 silent! g:result += 3
2045 g:result = 'yes'
2046 enddef
2047 # error is silenced, function does not abort
2048 Func()
2049 assert_equal('yes', g:result)
2050 unlet g:result
2051 END
2052 CheckScriptSuccess(lines)
2053enddef
2054
Bram Moolenaaraf0df472020-12-02 20:51:22 +01002055def Test_abort_even_with_silent()
2056 var lines =<< trim END
2057 vim9script
2058 g:result = 'none'
2059 def Func()
2060 eval {-> ''}() .. '' .. {}['X']
2061 g:result = 'yes'
2062 enddef
Bram Moolenaarf665e972020-12-05 19:17:16 +01002063 silent! Func()
Bram Moolenaaraf0df472020-12-02 20:51:22 +01002064 assert_equal('none', g:result)
Bram Moolenaar4029cab2020-12-05 18:13:27 +01002065 unlet g:result
2066 END
2067 CheckScriptSuccess(lines)
2068enddef
2069
Bram Moolenaarf665e972020-12-05 19:17:16 +01002070def Test_cmdmod_silent_restored()
2071 var lines =<< trim END
2072 vim9script
2073 def Func()
2074 g:result = 'none'
2075 silent! g:result += 3
2076 g:result = 'none'
2077 g:result += 3
2078 enddef
2079 Func()
2080 END
2081 # can't use CheckScriptFailure, it ignores the :silent!
2082 var fname = 'Xdefsilent'
2083 writefile(lines, fname)
2084 var caught = 'no'
2085 try
2086 exe 'source ' .. fname
2087 catch /E1030:/
2088 caught = 'yes'
2089 assert_match('Func, line 4', v:throwpoint)
2090 endtry
2091 assert_equal('yes', caught)
2092 delete(fname)
2093enddef
2094
Bram Moolenaar4029cab2020-12-05 18:13:27 +01002095def Test_dict_member_with_silent()
2096 var lines =<< trim END
2097 vim9script
2098 g:result = 'none'
2099 var d: dict<any>
2100 def Func()
2101 try
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002102 g:result = map([], (_, v) => ({}[v]))->join() .. d['']
Bram Moolenaar4029cab2020-12-05 18:13:27 +01002103 catch
2104 endtry
2105 enddef
2106 silent! Func()
2107 assert_equal('0', g:result)
2108 unlet g:result
Bram Moolenaaraf0df472020-12-02 20:51:22 +01002109 END
2110 CheckScriptSuccess(lines)
2111enddef
2112
Bram Moolenaar5b3d1bb2020-12-22 12:20:08 +01002113def Test_opfunc()
2114 nnoremap <F3> <cmd>set opfunc=Opfunc<cr>g@
2115 def g:Opfunc(_: any): string
2116 setline(1, 'ASDF')
2117 return ''
2118 enddef
2119 new
2120 setline(1, 'asdf')
2121 feedkeys("\<F3>$", 'x')
2122 assert_equal('ASDF', getline(1))
2123
2124 bwipe!
2125 nunmap <F3>
2126enddef
2127
Bram Moolenaar077a4232020-12-22 18:33:27 +01002128" this was crashing on exit
2129def Test_nested_lambda_in_closure()
2130 var lines =<< trim END
2131 vim9script
2132 def Outer()
2133 def g:Inner()
2134 echo map([1, 2, 3], {_, v -> v + 1})
2135 enddef
2136 g:Inner()
2137 enddef
2138 defcompile
2139 writefile(['Done'], 'XnestedDone')
2140 quit
2141 END
2142 if !RunVim([], lines, '--clean')
2143 return
2144 endif
2145 assert_equal(['Done'], readfile('XnestedDone'))
2146 delete('XnestedDone')
2147enddef
2148
2149
Bram Moolenaarf7779c62020-05-03 15:38:16 +02002150
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002151" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker