blob: 74883975bb3b179eca33d8e98e61379296b0960c [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 Moolenaar62aec932022-01-29 21:45:34 +00006import './vim9.vim' as v9
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()
Bram Moolenaare8c46602021-04-05 22:27:37 +020021 call TestCompilingErrorInTry()
Bram Moolenaar2b9b17e2020-10-13 18:38:11 +020022endfunc
23
24def TestCompilingError()
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +020025 var lines =<< trim END
26 vim9script
27 def Fails()
28 echo nothing
29 enddef
30 defcompile
31 END
Bram Moolenaare8c46602021-04-05 22:27:37 +020032 writefile(lines, 'XTest_compile_error')
Bram Moolenaar62aec932022-01-29 21:45:34 +000033 var buf = g:RunVimInTerminal('-S XTest_compile_error',
Bram Moolenaare0de1712020-12-02 17:36:54 +010034 {rows: 10, wait_for_ruler: 0})
Bram Moolenaar62aec932022-01-29 21:45:34 +000035 g:WaitForAssert(() => assert_match('Error detected while compiling command line.*Fails.*Variable not found: nothing',
36 g:Term_getlines(buf, range(1, 9))))
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +020037
38 # clean up
Bram Moolenaar62aec932022-01-29 21:45:34 +000039 g:StopVimInTerminal(buf)
Bram Moolenaare8c46602021-04-05 22:27:37 +020040 delete('XTest_compile_error')
41enddef
42
43def TestCompilingErrorInTry()
44 var dir = 'Xdir/autoload'
45 mkdir(dir, 'p')
46
47 var lines =<< trim END
48 vim9script
Bram Moolenaard8fe6d32022-01-30 18:40:44 +000049 export def OnlyCompiled()
Bram Moolenaare8c46602021-04-05 22:27:37 +020050 g:runtime = 'yes'
51 invalid
52 enddef
53 END
54 writefile(lines, dir .. '/script.vim')
55
56 lines =<< trim END
57 vim9script
58 todo
59 try
60 script#OnlyCompiled()
61 catch /nothing/
62 endtry
63 END
64 lines[1] = 'set rtp=' .. getcwd() .. '/Xdir'
65 writefile(lines, 'XTest_compile_error')
66
Bram Moolenaar62aec932022-01-29 21:45:34 +000067 var buf = g:RunVimInTerminal('-S XTest_compile_error', {rows: 10, wait_for_ruler: 0})
68 g:WaitForAssert(() => assert_match('Error detected while compiling command line.*function script#OnlyCompiled.*Invalid command: invalid',
69 g:Term_getlines(buf, range(1, 9))))
Bram Moolenaare8c46602021-04-05 22:27:37 +020070
71 # clean up
Bram Moolenaar62aec932022-01-29 21:45:34 +000072 g:StopVimInTerminal(buf)
Bram Moolenaare8c46602021-04-05 22:27:37 +020073 delete('XTest_compile_error')
74 delete('Xdir', 'rf')
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +020075enddef
76
Bram Moolenaarb55d6182021-06-08 22:01:53 +020077def Test_compile_error_in_called_function()
78 var lines =<< trim END
79 vim9script
80 var n: number
81 def Foo()
82 &hls = n
83 enddef
84 def Bar()
85 Foo()
86 enddef
87 silent! Foo()
88 Bar()
89 END
Bram Moolenaar62aec932022-01-29 21:45:34 +000090 v9.CheckScriptFailureList(lines, ['E1012:', 'E1191:'])
Bram Moolenaarb55d6182021-06-08 22:01:53 +020091enddef
92
Bram Moolenaar22f17a22021-06-21 20:48:58 +020093def Test_wrong_function_name()
94 var lines =<< trim END
95 vim9script
96 func _Foo()
97 echo 'foo'
98 endfunc
99 END
Bram Moolenaar3787f262022-02-07 21:54:01 +0000100 v9.CheckScriptFailure(lines, 'E1267:')
Bram Moolenaar22f17a22021-06-21 20:48:58 +0200101
102 lines =<< trim END
103 vim9script
104 def _Foo()
105 echo 'foo'
106 enddef
107 END
Bram Moolenaar3787f262022-02-07 21:54:01 +0000108 v9.CheckScriptFailure(lines, 'E1267:')
Bram Moolenaardea5ab02022-02-23 22:12:02 +0000109
110 lines =<< trim END
111 vim9script
112 var Object = {}
113 function Object.Method()
114 endfunction
115 END
116 v9.CheckScriptFailure(lines, 'E1182:')
117
118 lines =<< trim END
119 vim9script
120 var Object = {}
121 def Object.Method()
122 enddef
123 END
124 v9.CheckScriptFailure(lines, 'E1182:')
125
126 lines =<< trim END
127 vim9script
128 g:Object = {}
129 function g:Object.Method()
130 endfunction
131 END
132 v9.CheckScriptFailure(lines, 'E1182:')
133
134 lines =<< trim END
135 let s:Object = {}
136 def Define()
137 function s:Object.Method()
138 endfunction
139 enddef
140 defcompile
141 END
142 v9.CheckScriptFailure(lines, 'E1182:')
143 delfunc g:Define
144
145 lines =<< trim END
146 let s:Object = {}
147 def Define()
148 def Object.Method()
149 enddef
150 enddef
151 defcompile
152 END
153 v9.CheckScriptFailure(lines, 'E1182:')
154 delfunc g:Define
155
156 lines =<< trim END
157 let g:Object = {}
158 def Define()
159 function g:Object.Method()
160 endfunction
161 enddef
162 defcompile
163 END
164 v9.CheckScriptFailure(lines, 'E1182:')
165 delfunc g:Define
Bram Moolenaar22f17a22021-06-21 20:48:58 +0200166enddef
167
Bram Moolenaarf48b2fa2021-04-12 22:02:36 +0200168def Test_autoload_name_mismatch()
169 var dir = 'Xdir/autoload'
170 mkdir(dir, 'p')
171
172 var lines =<< trim END
173 vim9script
Bram Moolenaard8fe6d32022-01-30 18:40:44 +0000174 export def NoFunction()
Bram Moolenaarf48b2fa2021-04-12 22:02:36 +0200175 # comment
176 g:runtime = 'yes'
177 enddef
178 END
179 writefile(lines, dir .. '/script.vim')
180
181 var save_rtp = &rtp
182 exe 'set rtp=' .. getcwd() .. '/Xdir'
183 lines =<< trim END
184 call script#Function()
185 END
Bram Moolenaard8fe6d32022-01-30 18:40:44 +0000186 v9.CheckScriptFailure(lines, 'E117:', 1)
Bram Moolenaarf48b2fa2021-04-12 22:02:36 +0200187
188 &rtp = save_rtp
189 delete(dir, 'rf')
190enddef
191
Bram Moolenaarf0a40692021-06-11 22:05:47 +0200192def Test_autoload_names()
193 var dir = 'Xdir/autoload'
194 mkdir(dir, 'p')
195
196 var lines =<< trim END
197 func foobar#function()
198 return 'yes'
199 endfunc
200 let foobar#var = 'no'
201 END
202 writefile(lines, dir .. '/foobar.vim')
203
204 var save_rtp = &rtp
205 exe 'set rtp=' .. getcwd() .. '/Xdir'
206
207 lines =<< trim END
208 assert_equal('yes', foobar#function())
209 var Function = foobar#function
210 assert_equal('yes', Function())
211
212 assert_equal('no', foobar#var)
213 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000214 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaarf0a40692021-06-11 22:05:47 +0200215
216 &rtp = save_rtp
217 delete(dir, 'rf')
218enddef
219
Bram Moolenaar88c89c72021-08-14 14:01:05 +0200220def Test_autoload_error_in_script()
221 var dir = 'Xdir/autoload'
222 mkdir(dir, 'p')
223
224 var lines =<< trim END
225 func scripterror#function()
226 let g:called_function = 'yes'
227 endfunc
228 let 0 = 1
229 END
230 writefile(lines, dir .. '/scripterror.vim')
231
232 var save_rtp = &rtp
233 exe 'set rtp=' .. getcwd() .. '/Xdir'
234
235 g:called_function = 'no'
236 # The error in the autoload script cannot be checked with assert_fails(), use
237 # CheckDefSuccess() instead of CheckDefFailure()
238 try
Bram Moolenaar62aec932022-01-29 21:45:34 +0000239 v9.CheckDefSuccess(['scripterror#function()'])
Bram Moolenaar88c89c72021-08-14 14:01:05 +0200240 catch
241 assert_match('E121: Undefined variable: 0', v:exception)
242 endtry
243 assert_equal('no', g:called_function)
244
245 lines =<< trim END
246 func scriptcaught#function()
247 let g:called_function = 'yes'
248 endfunc
249 try
250 let 0 = 1
251 catch
252 let g:caught = v:exception
253 endtry
254 END
255 writefile(lines, dir .. '/scriptcaught.vim')
256
257 g:called_function = 'no'
Bram Moolenaar62aec932022-01-29 21:45:34 +0000258 v9.CheckDefSuccess(['scriptcaught#function()'])
Bram Moolenaar88c89c72021-08-14 14:01:05 +0200259 assert_match('E121: Undefined variable: 0', g:caught)
260 assert_equal('yes', g:called_function)
261
262 &rtp = save_rtp
263 delete(dir, 'rf')
264enddef
265
Bram Moolenaar62aec932022-01-29 21:45:34 +0000266def s:CallRecursive(n: number): number
Bram Moolenaar0ba48e82020-11-17 18:23:19 +0100267 return CallRecursive(n + 1)
268enddef
269
Bram Moolenaar62aec932022-01-29 21:45:34 +0000270def s:CallMapRecursive(l: list<number>): number
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100271 return map(l, (_, v) => CallMapRecursive([v]))[0]
Bram Moolenaar0ba48e82020-11-17 18:23:19 +0100272enddef
273
274def Test_funcdepth_error()
275 set maxfuncdepth=10
276
277 var caught = false
278 try
279 CallRecursive(1)
280 catch /E132:/
281 caught = true
282 endtry
283 assert_true(caught)
284
285 caught = false
286 try
287 CallMapRecursive([1])
288 catch /E132:/
289 caught = true
290 endtry
291 assert_true(caught)
292
293 set maxfuncdepth&
294enddef
295
Bram Moolenaar5178b1b2021-01-01 18:43:51 +0100296def Test_endfunc_enddef()
297 var lines =<< trim END
298 def Test()
299 echo 'test'
300 endfunc
301 enddef
302 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000303 v9.CheckScriptFailure(lines, 'E1151:', 3)
Bram Moolenaar5178b1b2021-01-01 18:43:51 +0100304
305 lines =<< trim END
306 def Test()
307 func Nested()
308 echo 'test'
309 enddef
310 enddef
311 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000312 v9.CheckScriptFailure(lines, 'E1152:', 4)
Bram Moolenaar49f1e9e2021-03-22 20:49:02 +0100313
314 lines =<< trim END
315 def Ok()
316 echo 'hello'
317 enddef | echo 'there'
318 def Bad()
319 echo 'hello'
320 enddef there
321 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000322 v9.CheckScriptFailure(lines, 'E1173: Text found after enddef: there', 6)
Bram Moolenaar5178b1b2021-01-01 18:43:51 +0100323enddef
324
Bram Moolenaarb8ba9b92021-01-01 18:54:34 +0100325def Test_missing_endfunc_enddef()
326 var lines =<< trim END
327 vim9script
328 def Test()
329 echo 'test'
330 endef
331 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000332 v9.CheckScriptFailure(lines, 'E1057:', 2)
Bram Moolenaarb8ba9b92021-01-01 18:54:34 +0100333
334 lines =<< trim END
335 vim9script
336 func Some()
337 echo 'test'
338 enfffunc
339 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000340 v9.CheckScriptFailure(lines, 'E126:', 2)
Bram Moolenaarb8ba9b92021-01-01 18:54:34 +0100341enddef
342
Bram Moolenaar4efd9942021-01-24 21:14:20 +0100343def Test_white_space_before_paren()
344 var lines =<< trim END
345 vim9script
346 def Test ()
347 echo 'test'
348 enddef
349 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000350 v9.CheckScriptFailure(lines, 'E1068:', 2)
Bram Moolenaar4efd9942021-01-24 21:14:20 +0100351
352 lines =<< trim END
353 vim9script
354 func Test ()
355 echo 'test'
356 endfunc
357 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000358 v9.CheckScriptFailure(lines, 'E1068:', 2)
Bram Moolenaar4efd9942021-01-24 21:14:20 +0100359
360 lines =<< trim END
361 def Test ()
362 echo 'test'
363 enddef
364 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000365 v9.CheckScriptFailure(lines, 'E1068:', 1)
Bram Moolenaar4efd9942021-01-24 21:14:20 +0100366
367 lines =<< trim END
368 func Test ()
369 echo 'test'
370 endfunc
371 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000372 v9.CheckScriptSuccess(lines)
Bram Moolenaar4efd9942021-01-24 21:14:20 +0100373enddef
374
Bram Moolenaar832ea892021-01-08 21:55:26 +0100375def Test_enddef_dict_key()
376 var d = {
377 enddef: 'x',
378 endfunc: 'y',
379 }
380 assert_equal({enddef: 'x', endfunc: 'y'}, d)
381enddef
382
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200383def ReturnString(): string
384 return 'string'
385enddef
386
387def ReturnNumber(): number
388 return 123
389enddef
390
391let g:notNumber = 'string'
392
393def ReturnGlobal(): number
394 return g:notNumber
395enddef
396
397def Test_return_something()
Bram Moolenaar62aec932022-01-29 21:45:34 +0000398 g:ReturnString()->assert_equal('string')
399 g:ReturnNumber()->assert_equal(123)
Bram Moolenaar848fadd2022-01-30 15:28:30 +0000400 assert_fails('g:ReturnGlobal()', 'E1012: Type mismatch; expected number but got string', '', 1, 'ReturnGlobal')
Bram Moolenaaref7aadb2022-01-18 18:46:07 +0000401
402 var lines =<< trim END
403 vim9script
404
405 def Msg()
406 echomsg 'in Msg()...'
407 enddef
408
409 def Func()
410 return Msg()
411 enddef
412 defcompile
413 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000414 v9.CheckScriptFailure(lines, 'E1096:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200415enddef
416
Bram Moolenaare32e5162021-01-21 20:21:29 +0100417def Test_check_argument_type()
418 var lines =<< trim END
419 vim9script
420 def Val(a: number, b: number): number
421 return 0
422 enddef
423 def Func()
424 var x: any = true
425 Val(0, x)
426 enddef
427 disass Func
428 Func()
429 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000430 v9.CheckScriptFailure(lines, 'E1013: Argument 2: type mismatch, expected number but got bool', 2)
Bram Moolenaare32e5162021-01-21 20:21:29 +0100431enddef
432
Bram Moolenaarefd88552020-06-18 20:50:10 +0200433def Test_missing_return()
Bram Moolenaar62aec932022-01-29 21:45:34 +0000434 v9.CheckDefFailure(['def Missing(): number',
Bram Moolenaarefd88552020-06-18 20:50:10 +0200435 ' if g:cond',
436 ' echo "no return"',
437 ' else',
438 ' return 0',
439 ' endif'
440 'enddef'], 'E1027:')
Bram Moolenaar62aec932022-01-29 21:45:34 +0000441 v9.CheckDefFailure(['def Missing(): number',
Bram Moolenaarefd88552020-06-18 20:50:10 +0200442 ' if g:cond',
443 ' return 1',
444 ' else',
445 ' echo "no return"',
446 ' endif'
447 'enddef'], 'E1027:')
Bram Moolenaar62aec932022-01-29 21:45:34 +0000448 v9.CheckDefFailure(['def Missing(): number',
Bram Moolenaarefd88552020-06-18 20:50:10 +0200449 ' if g:cond',
450 ' return 1',
451 ' else',
452 ' return 2',
453 ' endif'
454 ' return 3'
455 'enddef'], 'E1095:')
456enddef
457
Bram Moolenaar403dc312020-10-17 19:29:51 +0200458def Test_return_bool()
459 var lines =<< trim END
460 vim9script
461 def MenuFilter(id: number, key: string): bool
462 return popup_filter_menu(id, key)
463 enddef
464 def YesnoFilter(id: number, key: string): bool
465 return popup_filter_yesno(id, key)
466 enddef
467 defcompile
468 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000469 v9.CheckScriptSuccess(lines)
Bram Moolenaar403dc312020-10-17 19:29:51 +0200470enddef
471
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200472let s:nothing = 0
473def ReturnNothing()
474 s:nothing = 1
475 if true
476 return
477 endif
478 s:nothing = 2
479enddef
480
481def Test_return_nothing()
Bram Moolenaar62aec932022-01-29 21:45:34 +0000482 g:ReturnNothing()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200483 s:nothing->assert_equal(1)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200484enddef
485
Bram Moolenaar648ea762021-01-15 19:04:32 +0100486def Test_return_invalid()
487 var lines =<< trim END
488 vim9script
489 def Func(): invalid
490 return xxx
491 enddef
492 defcompile
493 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000494 v9.CheckScriptFailure(lines, 'E1010:', 2)
Bram Moolenaar31842cd2021-02-12 22:10:21 +0100495
496 lines =<< trim END
497 vim9script
498 def Test(Fun: func(number): number): list<number>
499 return map([1, 2, 3], (_, i) => Fun(i))
500 enddef
501 defcompile
502 def Inc(nr: number): nr
503 return nr + 2
504 enddef
505 echo Test(Inc)
506 END
507 # doing this twice was leaking memory
Bram Moolenaar62aec932022-01-29 21:45:34 +0000508 v9.CheckScriptFailure(lines, 'E1010:')
509 v9.CheckScriptFailure(lines, 'E1010:')
Bram Moolenaar648ea762021-01-15 19:04:32 +0100510enddef
511
Bram Moolenaarefc084e2021-09-09 22:30:52 +0200512def Test_return_list_any()
Bram Moolenaar114dbda2022-01-03 12:28:03 +0000513 # This used to fail but now the actual list type is checked, and since it has
514 # an item of type string it can be used as list<string>.
Bram Moolenaarefc084e2021-09-09 22:30:52 +0200515 var lines =<< trim END
516 vim9script
517 def Func(): list<string>
518 var l: list<any>
519 l->add('string')
520 return l
521 enddef
522 echo Func()
523 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000524 v9.CheckScriptSuccess(lines)
Bram Moolenaar114dbda2022-01-03 12:28:03 +0000525
Bram Moolenaarefc084e2021-09-09 22:30:52 +0200526 lines =<< trim END
527 vim9script
528 def Func(): list<string>
529 var l: list<any>
530 l += ['string']
531 return l
532 enddef
533 echo Func()
534 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000535 v9.CheckScriptSuccess(lines)
Bram Moolenaarefc084e2021-09-09 22:30:52 +0200536enddef
537
Bram Moolenaar62aec932022-01-29 21:45:34 +0000538func s:Increment()
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200539 let g:counter += 1
540endfunc
541
542def Test_call_ufunc_count()
543 g:counter = 1
544 Increment()
545 Increment()
546 Increment()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200547 # works with and without :call
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200548 g:counter->assert_equal(4)
549 eval g:counter->assert_equal(4)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200550 unlet g:counter
551enddef
552
Bram Moolenaar62aec932022-01-29 21:45:34 +0000553def s:MyVarargs(arg: string, ...rest: list<string>): string
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200554 var res = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200555 for s in rest
556 res ..= ',' .. s
557 endfor
558 return res
559enddef
560
561def Test_call_varargs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200562 MyVarargs('one')->assert_equal('one')
563 MyVarargs('one', 'two')->assert_equal('one,two')
564 MyVarargs('one', 'two', 'three')->assert_equal('one,two,three')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200565enddef
566
Bram Moolenaar01dd6c32021-09-05 16:36:23 +0200567def Test_call_white_space()
Bram Moolenaar62aec932022-01-29 21:45:34 +0000568 v9.CheckDefAndScriptFailure(["call Test ('text')"], ['E476:', 'E1068:'])
Bram Moolenaar01dd6c32021-09-05 16:36:23 +0200569enddef
570
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200571def MyDefaultArgs(name = 'string'): string
572 return name
573enddef
574
Bram Moolenaar62aec932022-01-29 21:45:34 +0000575def s:MyDefaultSecond(name: string, second: bool = true): string
Bram Moolenaare30f64b2020-07-15 19:48:20 +0200576 return second ? name : 'none'
577enddef
578
Bram Moolenaar38a3bfa2021-03-29 22:14:55 +0200579
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200580def Test_call_default_args()
Bram Moolenaar62aec932022-01-29 21:45:34 +0000581 g:MyDefaultArgs()->assert_equal('string')
582 g:MyDefaultArgs(v:none)->assert_equal('string')
583 g:MyDefaultArgs('one')->assert_equal('one')
584 assert_fails('g:MyDefaultArgs("one", "two")', 'E118:', '', 4, 'Test_call_default_args')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200585
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200586 MyDefaultSecond('test')->assert_equal('test')
587 MyDefaultSecond('test', true)->assert_equal('test')
588 MyDefaultSecond('test', false)->assert_equal('none')
Bram Moolenaare30f64b2020-07-15 19:48:20 +0200589
Bram Moolenaar38a3bfa2021-03-29 22:14:55 +0200590 var lines =<< trim END
591 def MyDefaultThird(name: string, aa = 'aa', bb = 'bb'): string
592 return name .. aa .. bb
593 enddef
594
595 MyDefaultThird('->')->assert_equal('->aabb')
596 MyDefaultThird('->', v:none)->assert_equal('->aabb')
597 MyDefaultThird('->', 'xx')->assert_equal('->xxbb')
598 MyDefaultThird('->', v:none, v:none)->assert_equal('->aabb')
599 MyDefaultThird('->', 'xx', v:none)->assert_equal('->xxbb')
600 MyDefaultThird('->', v:none, 'yy')->assert_equal('->aayy')
601 MyDefaultThird('->', 'xx', 'yy')->assert_equal('->xxyy')
Bram Moolenaare28d9b32021-07-03 18:56:53 +0200602
603 def DefArg(mandatory: any, optional = mandatory): string
604 return mandatory .. optional
605 enddef
606 DefArg(1234)->assert_equal('12341234')
607 DefArg("ok")->assert_equal('okok')
Bram Moolenaar38a3bfa2021-03-29 22:14:55 +0200608 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000609 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar38a3bfa2021-03-29 22:14:55 +0200610
Bram Moolenaar62aec932022-01-29 21:45:34 +0000611 v9.CheckScriptFailure(['def Func(arg: number = asdf)', 'enddef', 'defcompile'], 'E1001:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100612 delfunc g:Func
Bram Moolenaar62aec932022-01-29 21:45:34 +0000613 v9.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 +0100614 delfunc g:Func
Bram Moolenaar62aec932022-01-29 21:45:34 +0000615 v9.CheckDefFailure(['def Func(x: number = )', 'enddef'], 'E15:')
Bram Moolenaar12bce952021-03-11 20:04:04 +0100616
Bram Moolenaar38a3bfa2021-03-29 22:14:55 +0200617 lines =<< trim END
Bram Moolenaar12bce952021-03-11 20:04:04 +0100618 vim9script
619 def Func(a = b == 0 ? 1 : 2, b = 0)
620 enddef
621 defcompile
622 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000623 v9.CheckScriptFailure(lines, 'E1001: Variable not found: b')
Bram Moolenaar59618fe2021-12-21 12:32:17 +0000624
Bram Moolenaarfa46ead2021-12-22 13:18:39 +0000625 # using script variable requires matching type or type cast when executed
Bram Moolenaar59618fe2021-12-21 12:32:17 +0000626 lines =<< trim END
627 vim9script
628 var a: any
629 def Func(arg: string = a)
630 echo arg
631 enddef
632 defcompile
633 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000634 v9.CheckScriptSuccess(lines + ['a = "text"', 'Func()'])
635 v9.CheckScriptFailure(lines + ['a = 123', 'Func()'], 'E1013: Argument 1: type mismatch, expected string but got number')
Bram Moolenaar59618fe2021-12-21 12:32:17 +0000636
637 # using global variable does not require type cast
638 lines =<< trim END
639 vim9script
640 def Func(arg: string = g:str)
641 echo arg
642 enddef
643 g:str = 'works'
644 Func()
645 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000646 v9.CheckScriptSuccess(lines)
Bram Moolenaar04b12692020-05-04 23:24:44 +0200647enddef
648
Bram Moolenaar62aec932022-01-29 21:45:34 +0000649def s:FuncWithComment( # comment
Bram Moolenaarcef12702021-01-04 14:09:43 +0100650 a: number, #comment
651 b: bool, # comment
652 c: string) #comment
653 assert_equal(4, a)
654 assert_equal(true, b)
655 assert_equal('yes', c)
656enddef
657
658def Test_func_with_comments()
659 FuncWithComment(4, true, 'yes')
660
661 var lines =<< trim END
662 def Func(# comment
663 arg: string)
664 enddef
665 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000666 v9.CheckScriptFailure(lines, 'E125:', 1)
Bram Moolenaarcef12702021-01-04 14:09:43 +0100667
668 lines =<< trim END
669 def Func(
670 arg: string# comment
671 )
672 enddef
673 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000674 v9.CheckScriptFailure(lines, 'E475:', 2)
Bram Moolenaarcef12702021-01-04 14:09:43 +0100675
676 lines =<< trim END
677 def Func(
678 arg: string
679 )# comment
680 enddef
681 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000682 v9.CheckScriptFailure(lines, 'E488:', 3)
Bram Moolenaarcef12702021-01-04 14:09:43 +0100683enddef
684
Bram Moolenaar04b12692020-05-04 23:24:44 +0200685def Test_nested_function()
Bram Moolenaar38453522021-11-28 22:00:12 +0000686 def NestedDef(arg: string): string
Bram Moolenaar04b12692020-05-04 23:24:44 +0200687 return 'nested ' .. arg
688 enddef
Bram Moolenaar38453522021-11-28 22:00:12 +0000689 NestedDef(':def')->assert_equal('nested :def')
690
691 func NestedFunc(arg)
692 return 'nested ' .. a:arg
693 endfunc
694 NestedFunc(':func')->assert_equal('nested :func')
Bram Moolenaar04b12692020-05-04 23:24:44 +0200695
Bram Moolenaar62aec932022-01-29 21:45:34 +0000696 v9.CheckDefFailure(['def Nested()', 'enddef', 'Nested(66)'], 'E118:')
697 v9.CheckDefFailure(['def Nested(arg: string)', 'enddef', 'Nested()'], 'E119:')
Bram Moolenaar0e65d3d2020-05-05 17:53:16 +0200698
Bram Moolenaar62aec932022-01-29 21:45:34 +0000699 v9.CheckDefFailure(['def s:Nested()', 'enddef'], 'E1075:')
700 v9.CheckDefFailure(['def b:Nested()', 'enddef'], 'E1075:')
Bram Moolenaar8b848ca2020-09-10 22:28:01 +0200701
Bram Moolenaar54021752020-12-06 18:50:36 +0100702 var lines =<< trim END
703 def Outer()
704 def Inner()
705 # comment
706 enddef
707 def Inner()
708 enddef
709 enddef
710 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000711 v9.CheckDefFailure(lines, 'E1073:')
Bram Moolenaar54021752020-12-06 18:50:36 +0100712
713 lines =<< trim END
714 def Outer()
715 def Inner()
716 # comment
717 enddef
718 def! Inner()
719 enddef
720 enddef
721 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000722 v9.CheckDefFailure(lines, 'E1117:')
Bram Moolenaar54021752020-12-06 18:50:36 +0100723
Bram Moolenaardb8e5c22021-12-25 19:58:22 +0000724 lines =<< trim END
725 vim9script
726 def Outer()
727 def Inner()
728 g:result = 'ok'
729 enddef
730 Inner()
731 enddef
732 Outer()
733 Inner()
734 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000735 v9.CheckScriptFailure(lines, 'E117: Unknown function: Inner')
Bram Moolenaardb8e5c22021-12-25 19:58:22 +0000736 assert_equal('ok', g:result)
737 unlet g:result
738
Bram Moolenaarf681cfb2022-02-07 20:30:57 +0000739 lines =<< trim END
740 vim9script
741 def Outer()
742 def _Inner()
743 echo 'bad'
744 enddef
Bram Moolenaar3787f262022-02-07 21:54:01 +0000745 _Inner()
Bram Moolenaarf681cfb2022-02-07 20:30:57 +0000746 enddef
747 defcompile
748 END
Bram Moolenaar3787f262022-02-07 21:54:01 +0000749 v9.CheckScriptFailure(lines, 'E1267:')
Bram Moolenaarf681cfb2022-02-07 20:30:57 +0000750
751 lines =<< trim END
752 vim9script
753 def Outer()
754 def g:inner()
755 echo 'bad'
756 enddef
Bram Moolenaar3787f262022-02-07 21:54:01 +0000757 g:inner()
Bram Moolenaarf681cfb2022-02-07 20:30:57 +0000758 enddef
759 defcompile
760 END
Bram Moolenaar3787f262022-02-07 21:54:01 +0000761 v9.CheckScriptFailure(lines, 'E1267:')
762
763 lines =<< trim END
764 vim9script
765 def g:_Func()
766 echo 'bad'
767 enddef
768 END
769 v9.CheckScriptFailure(lines, 'E1267:')
770
771 lines =<< trim END
772 vim9script
Bram Moolenaara749a422022-02-12 19:52:25 +0000773 def _Func()
Bram Moolenaar3787f262022-02-07 21:54:01 +0000774 echo 'bad'
775 enddef
776 END
777 v9.CheckScriptFailure(lines, 'E1267:')
Bram Moolenaarf681cfb2022-02-07 20:30:57 +0000778
Bram Moolenaar54021752020-12-06 18:50:36 +0100779 # nested function inside conditional
Bram Moolenaar54021752020-12-06 18:50:36 +0100780 lines =<< trim END
781 vim9script
782 var thecount = 0
783 if true
784 def Test(): number
785 def TheFunc(): number
786 thecount += 1
787 return thecount
788 enddef
789 return TheFunc()
790 enddef
791 endif
792 defcompile
793 assert_equal(1, Test())
794 assert_equal(2, Test())
795 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000796 v9.CheckScriptSuccess(lines)
Bram Moolenaar8863bda2021-03-17 18:42:08 +0100797
798 # also works when "thecount" is inside the "if" block
799 lines =<< trim END
800 vim9script
801 if true
802 var thecount = 0
803 def Test(): number
804 def TheFunc(): number
805 thecount += 1
806 return thecount
807 enddef
808 return TheFunc()
809 enddef
810 endif
811 defcompile
812 assert_equal(1, Test())
813 assert_equal(2, Test())
814 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000815 v9.CheckScriptSuccess(lines)
Bram Moolenaar4bba16d2021-08-15 19:28:05 +0200816
817 lines =<< trim END
818 vim9script
819 def Outer()
820 def Inner()
821 echo 'hello'
822 enddef burp
823 enddef
824 defcompile
825 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000826 v9.CheckScriptFailure(lines, 'E1173: Text found after enddef: burp', 3)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200827enddef
828
Bram Moolenaaradc8e442020-12-31 18:28:18 +0100829def Test_not_nested_function()
830 echo printf('%d',
831 function('len')('xxx'))
832enddef
833
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200834func Test_call_default_args_from_func()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200835 call MyDefaultArgs()->assert_equal('string')
836 call MyDefaultArgs('one')->assert_equal('one')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200837 call assert_fails('call MyDefaultArgs("one", "two")', 'E118:', '', 3, 'Test_call_default_args_from_func')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200838endfunc
839
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200840def Test_nested_global_function()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200841 var lines =<< trim END
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200842 vim9script
843 def Outer()
844 def g:Inner(): string
845 return 'inner'
846 enddef
847 enddef
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200848 defcompile
849 Outer()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200850 g:Inner()->assert_equal('inner')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200851 delfunc g:Inner
852 Outer()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200853 g:Inner()->assert_equal('inner')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200854 delfunc g:Inner
855 Outer()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200856 g:Inner()->assert_equal('inner')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200857 delfunc g:Inner
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200858 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000859 v9.CheckScriptSuccess(lines)
Bram Moolenaar2c79e9d2020-08-01 18:57:52 +0200860
861 lines =<< trim END
862 vim9script
863 def Outer()
Bram Moolenaar38453522021-11-28 22:00:12 +0000864 func g:Inner()
865 return 'inner'
866 endfunc
867 enddef
868 defcompile
869 Outer()
870 g:Inner()->assert_equal('inner')
871 delfunc g:Inner
872 Outer()
873 g:Inner()->assert_equal('inner')
874 delfunc g:Inner
875 Outer()
876 g:Inner()->assert_equal('inner')
877 delfunc g:Inner
878 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000879 v9.CheckScriptSuccess(lines)
Bram Moolenaar38453522021-11-28 22:00:12 +0000880
881 lines =<< trim END
882 vim9script
883 def Outer()
Bram Moolenaar2c79e9d2020-08-01 18:57:52 +0200884 def g:Inner(): string
885 return 'inner'
886 enddef
887 enddef
888 defcompile
889 Outer()
890 Outer()
891 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000892 v9.CheckScriptFailure(lines, "E122:")
Bram Moolenaarcd45ed02020-12-22 17:35:54 +0100893 delfunc g:Inner
Bram Moolenaarad486a02020-08-01 23:22:18 +0200894
895 lines =<< trim END
896 vim9script
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100897 def Outer()
898 def g:Inner()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100899 echo map([1, 2, 3], (_, v) => v + 1)
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100900 enddef
901 g:Inner()
902 enddef
903 Outer()
904 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000905 v9.CheckScriptSuccess(lines)
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100906 delfunc g:Inner
907
908 lines =<< trim END
909 vim9script
Bram Moolenaarad486a02020-08-01 23:22:18 +0200910 def Func()
911 echo 'script'
912 enddef
913 def Outer()
914 def Func()
915 echo 'inner'
916 enddef
917 enddef
918 defcompile
919 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000920 v9.CheckScriptFailure(lines, "E1073:", 1)
Bram Moolenaard604d782021-11-20 21:46:20 +0000921
922 lines =<< trim END
923 vim9script
924 def Func()
925 echo 'script'
926 enddef
927 def Func()
928 echo 'script'
929 enddef
930 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000931 v9.CheckScriptFailure(lines, "E1073:", 5)
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200932enddef
933
Bram Moolenaar6abdcf82020-11-22 18:15:44 +0100934def DefListAll()
935 def
936enddef
937
938def DefListOne()
939 def DefListOne
940enddef
941
942def DefListMatches()
943 def /DefList
944enddef
945
946def Test_nested_def_list()
947 var funcs = split(execute('call DefListAll()'), "\n")
948 assert_true(len(funcs) > 10)
949 assert_true(funcs->index('def DefListAll()') >= 0)
950
951 funcs = split(execute('call DefListOne()'), "\n")
952 assert_equal([' def DefListOne()', '1 def DefListOne', ' enddef'], funcs)
953
954 funcs = split(execute('call DefListMatches()'), "\n")
955 assert_true(len(funcs) >= 3)
956 assert_true(funcs->index('def DefListAll()') >= 0)
957 assert_true(funcs->index('def DefListOne()') >= 0)
958 assert_true(funcs->index('def DefListMatches()') >= 0)
Bram Moolenaar54021752020-12-06 18:50:36 +0100959
960 var lines =<< trim END
961 vim9script
962 def Func()
963 def +Func+
964 enddef
965 defcompile
966 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000967 v9.CheckScriptFailure(lines, 'E476:', 1)
Bram Moolenaar6abdcf82020-11-22 18:15:44 +0100968enddef
969
Bram Moolenaare08be092022-02-17 13:08:26 +0000970def Test_global_function_not_found()
971 var lines =<< trim END
972 g:Ref = 123
973 call g:Ref()
974 END
975 v9.CheckDefExecAndScriptFailure(lines, ['E117:', 'E1085:'], 2)
976enddef
977
Bram Moolenaar333894b2020-08-01 18:53:07 +0200978def Test_global_local_function()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200979 var lines =<< trim END
Bram Moolenaar333894b2020-08-01 18:53:07 +0200980 vim9script
981 def g:Func(): string
982 return 'global'
983 enddef
984 def Func(): string
985 return 'local'
986 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200987 g:Func()->assert_equal('global')
988 Func()->assert_equal('local')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100989 delfunc g:Func
Bram Moolenaar333894b2020-08-01 18:53:07 +0200990 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000991 v9.CheckScriptSuccess(lines)
Bram Moolenaar035d6e92020-08-11 22:30:42 +0200992
993 lines =<< trim END
994 vim9script
995 def g:Funcy()
996 echo 'funcy'
997 enddef
Bram Moolenaara749a422022-02-12 19:52:25 +0000998 Funcy()
Bram Moolenaar035d6e92020-08-11 22:30:42 +0200999 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001000 v9.CheckScriptFailure(lines, 'E117:')
Bram Moolenaar333894b2020-08-01 18:53:07 +02001001enddef
1002
Bram Moolenaar0f769812020-09-12 18:32:34 +02001003def Test_local_function_shadows_global()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001004 var lines =<< trim END
Bram Moolenaar0f769812020-09-12 18:32:34 +02001005 vim9script
1006 def g:Gfunc(): string
1007 return 'global'
1008 enddef
1009 def AnotherFunc(): number
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001010 var Gfunc = function('len')
Bram Moolenaar0f769812020-09-12 18:32:34 +02001011 return Gfunc('testing')
1012 enddef
1013 g:Gfunc()->assert_equal('global')
1014 AnotherFunc()->assert_equal(7)
1015 delfunc g:Gfunc
1016 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001017 v9.CheckScriptSuccess(lines)
Bram Moolenaar0f769812020-09-12 18:32:34 +02001018
1019 lines =<< trim END
1020 vim9script
1021 def g:Func(): string
1022 return 'global'
1023 enddef
1024 def AnotherFunc()
1025 g:Func = function('len')
1026 enddef
1027 AnotherFunc()
1028 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001029 v9.CheckScriptFailure(lines, 'E705:')
Bram Moolenaar0f769812020-09-12 18:32:34 +02001030 delfunc g:Func
Bram Moolenaar0865b152021-04-05 15:38:51 +02001031
Bram Moolenaar62aec932022-01-29 21:45:34 +00001032 # global function is not found with g: prefix
Bram Moolenaar0865b152021-04-05 15:38:51 +02001033 lines =<< trim END
1034 vim9script
1035 def g:Func(): string
1036 return 'global'
1037 enddef
1038 def AnotherFunc(): string
1039 return Func()
1040 enddef
1041 assert_equal('global', AnotherFunc())
Bram Moolenaar0865b152021-04-05 15:38:51 +02001042 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001043 v9.CheckScriptFailure(lines, 'E117:')
1044 delfunc g:Func
Bram Moolenaar0865b152021-04-05 15:38:51 +02001045
1046 lines =<< trim END
1047 vim9script
1048 def g:Func(): string
1049 return 'global'
1050 enddef
Bram Moolenaar848fadd2022-01-30 15:28:30 +00001051 assert_equal('global', g:Func())
Bram Moolenaar0865b152021-04-05 15:38:51 +02001052 delfunc g:Func
1053 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001054 v9.CheckScriptSuccess(lines)
Bram Moolenaar58493cf2022-01-06 12:23:30 +00001055
1056 # This does not shadow "i" which is visible only inside the for loop
1057 lines =<< trim END
1058 vim9script
1059
1060 def Foo(i: number)
1061 echo i
1062 enddef
1063
1064 for i in range(3)
1065 # Foo() is compiled here
1066 Foo(i)
1067 endfor
1068 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001069 v9.CheckScriptSuccess(lines)
Bram Moolenaar0f769812020-09-12 18:32:34 +02001070enddef
1071
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001072func TakesOneArg(arg)
1073 echo a:arg
1074endfunc
1075
1076def Test_call_wrong_args()
Bram Moolenaar62aec932022-01-29 21:45:34 +00001077 v9.CheckDefFailure(['g:TakesOneArg()'], 'E119:')
1078 v9.CheckDefFailure(['g:TakesOneArg(11, 22)'], 'E118:')
1079 v9.CheckDefFailure(['bufnr(xxx)'], 'E1001:')
1080 v9.CheckScriptFailure(['def Func(Ref: func(s: string))'], 'E475:')
Bram Moolenaaree8580e2020-08-28 17:19:07 +02001081
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001082 var lines =<< trim END
Bram Moolenaaree8580e2020-08-28 17:19:07 +02001083 vim9script
1084 def Func(s: string)
1085 echo s
1086 enddef
1087 Func([])
1088 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001089 v9.CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected string but got list<unknown>', 5)
Bram Moolenaarb185a402020-09-18 22:42:00 +02001090
Bram Moolenaar9a015112021-12-31 14:06:45 +00001091 # argument name declared earlier is found when declaring a function
Bram Moolenaarb185a402020-09-18 22:42:00 +02001092 lines =<< trim END
1093 vim9script
Bram Moolenaarb4893b82021-02-21 22:20:24 +01001094 var name = 'piet'
1095 def FuncOne(name: string)
Bram Moolenaar3a5988c2022-02-08 19:23:35 +00001096 echo name
Bram Moolenaarb4893b82021-02-21 22:20:24 +01001097 enddef
1098 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001099 v9.CheckScriptFailure(lines, 'E1168:')
Bram Moolenaarb4893b82021-02-21 22:20:24 +01001100
Bram Moolenaar3a5988c2022-02-08 19:23:35 +00001101 # same, inside the same block
1102 lines =<< trim END
1103 vim9script
1104 if true
1105 var name = 'piet'
1106 def FuncOne(name: string)
1107 echo name
1108 enddef
1109 endif
1110 END
1111 v9.CheckScriptFailure(lines, 'E1168:')
1112
1113 # variable in other block is OK
1114 lines =<< trim END
1115 vim9script
1116 if true
1117 var name = 'piet'
1118 endif
1119 def FuncOne(name: string)
1120 echo name
1121 enddef
1122 END
1123 v9.CheckScriptSuccess(lines)
1124
Bram Moolenaardce24412022-02-08 20:35:30 +00001125 # with another variable in another block
1126 lines =<< trim END
1127 vim9script
1128 if true
1129 var name = 'piet'
1130 # define a function so that the variable isn't cleared
1131 def GetItem(): string
1132 return item
1133 enddef
1134 endif
1135 if true
1136 var name = 'peter'
1137 def FuncOne(name: string)
1138 echo name
1139 enddef
1140 endif
1141 END
1142 v9.CheckScriptFailure(lines, 'E1168:')
1143
1144 # only variable in another block is OK
1145 lines =<< trim END
1146 vim9script
1147 if true
1148 var name = 'piet'
1149 # define a function so that the variable isn't cleared
1150 def GetItem(): string
1151 return item
1152 enddef
1153 endif
1154 if true
1155 def FuncOne(name: string)
1156 echo name
1157 enddef
1158 endif
1159 END
1160 v9.CheckScriptSuccess(lines)
1161
Bram Moolenaar9a015112021-12-31 14:06:45 +00001162 # argument name declared later is only found when compiling
1163 lines =<< trim END
1164 vim9script
1165 def FuncOne(name: string)
1166 echo nr
1167 enddef
1168 var name = 'piet'
1169 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001170 v9.CheckScriptSuccess(lines)
1171 v9.CheckScriptFailure(lines + ['defcompile'], 'E1168:')
Bram Moolenaar9a015112021-12-31 14:06:45 +00001172
Bram Moolenaarb4893b82021-02-21 22:20:24 +01001173 lines =<< trim END
1174 vim9script
Bram Moolenaarb185a402020-09-18 22:42:00 +02001175 def FuncOne(nr: number)
1176 echo nr
1177 enddef
1178 def FuncTwo()
1179 FuncOne()
1180 enddef
1181 defcompile
1182 END
1183 writefile(lines, 'Xscript')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001184 var didCatch = false
Bram Moolenaarb185a402020-09-18 22:42:00 +02001185 try
1186 source Xscript
1187 catch
1188 assert_match('E119: Not enough arguments for function: <SNR>\d\+_FuncOne', v:exception)
1189 assert_match('Xscript\[8\]..function <SNR>\d\+_FuncTwo, line 1', v:throwpoint)
1190 didCatch = true
1191 endtry
1192 assert_true(didCatch)
1193
1194 lines =<< trim END
1195 vim9script
1196 def FuncOne(nr: number)
1197 echo nr
1198 enddef
1199 def FuncTwo()
1200 FuncOne(1, 2)
1201 enddef
1202 defcompile
1203 END
1204 writefile(lines, 'Xscript')
1205 didCatch = false
1206 try
1207 source Xscript
1208 catch
1209 assert_match('E118: Too many arguments for function: <SNR>\d\+_FuncOne', v:exception)
1210 assert_match('Xscript\[8\]..function <SNR>\d\+_FuncTwo, line 1', v:throwpoint)
1211 didCatch = true
1212 endtry
1213 assert_true(didCatch)
1214
1215 delete('Xscript')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001216enddef
1217
Bram Moolenaar50824712020-12-20 21:10:17 +01001218def Test_call_funcref_wrong_args()
1219 var head =<< trim END
1220 vim9script
1221 def Func3(a1: string, a2: number, a3: list<number>)
1222 echo a1 .. a2 .. a3[0]
1223 enddef
1224 def Testme()
1225 var funcMap: dict<func> = {func: Func3}
1226 END
1227 var tail =<< trim END
1228 enddef
1229 Testme()
1230 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001231 v9.CheckScriptSuccess(head + ["funcMap['func']('str', 123, [1, 2, 3])"] + tail)
Bram Moolenaar50824712020-12-20 21:10:17 +01001232
Bram Moolenaar62aec932022-01-29 21:45:34 +00001233 v9.CheckScriptFailure(head + ["funcMap['func']('str', 123)"] + tail, 'E119:')
1234 v9.CheckScriptFailure(head + ["funcMap['func']('str', 123, [1], 4)"] + tail, 'E118:')
Bram Moolenaar32b3f822021-01-06 21:59:39 +01001235
1236 var lines =<< trim END
1237 vim9script
1238 var Ref: func(number): any
1239 Ref = (j) => !j
1240 echo Ref(false)
1241 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001242 v9.CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected number but got bool', 4)
Bram Moolenaar32b3f822021-01-06 21:59:39 +01001243
1244 lines =<< trim END
1245 vim9script
1246 var Ref: func(number): any
1247 Ref = (j) => !j
1248 call Ref(false)
1249 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001250 v9.CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected number but got bool', 4)
Bram Moolenaar50824712020-12-20 21:10:17 +01001251enddef
1252
Bram Moolenaarb4d16cb2020-11-05 18:45:46 +01001253def Test_call_lambda_args()
Bram Moolenaar2a389082021-04-09 20:24:31 +02001254 var lines =<< trim END
1255 var Callback = (..._) => 'anything'
1256 assert_equal('anything', Callback())
1257 assert_equal('anything', Callback(1))
1258 assert_equal('anything', Callback('a', 2))
Bram Moolenaar1088b692021-04-09 22:12:44 +02001259
1260 assert_equal('xyz', ((a: string): string => a)('xyz'))
Bram Moolenaar2a389082021-04-09 20:24:31 +02001261 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001262 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar2a389082021-04-09 20:24:31 +02001263
Bram Moolenaar62aec932022-01-29 21:45:34 +00001264 v9.CheckDefFailure(['echo ((i) => 0)()'],
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001265 'E119: Not enough arguments for function: ((i) => 0)()')
Bram Moolenaarb4d16cb2020-11-05 18:45:46 +01001266
Bram Moolenaar2a389082021-04-09 20:24:31 +02001267 lines =<< trim END
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001268 var Ref = (x: number, y: number) => x + y
Bram Moolenaarb4d16cb2020-11-05 18:45:46 +01001269 echo Ref(1, 'x')
1270 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001271 v9.CheckDefFailure(lines, 'E1013: Argument 2: type mismatch, expected number but got string')
Bram Moolenaare68b02a2021-01-03 13:09:51 +01001272
1273 lines =<< trim END
1274 var Ref: func(job, string, number)
1275 Ref = (x, y) => 0
1276 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001277 v9.CheckDefAndScriptFailure(lines, 'E1012:')
Bram Moolenaare68b02a2021-01-03 13:09:51 +01001278
1279 lines =<< trim END
1280 var Ref: func(job, string)
1281 Ref = (x, y, z) => 0
1282 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001283 v9.CheckDefAndScriptFailure(lines, 'E1012:')
Bram Moolenaar057e84a2021-02-28 16:55:11 +01001284
1285 lines =<< trim END
1286 var one = 1
1287 var l = [1, 2, 3]
1288 echo map(l, (one) => one)
1289 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001290 v9.CheckDefFailure(lines, 'E1167:')
1291 v9.CheckScriptFailure(['vim9script'] + lines, 'E1168:')
Bram Moolenaar057e84a2021-02-28 16:55:11 +01001292
1293 lines =<< trim END
Bram Moolenaar14ded112021-06-26 19:25:49 +02001294 var Ref: func(any, ?any): bool
1295 Ref = (_, y = 1) => false
1296 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001297 v9.CheckDefAndScriptFailure(lines, 'E1172:')
Bram Moolenaar14ded112021-06-26 19:25:49 +02001298
1299 lines =<< trim END
Bram Moolenaar015cf102021-06-26 21:52:02 +02001300 var a = 0
1301 var b = (a == 0 ? 1 : 2)
1302 assert_equal(1, b)
Bram Moolenaar98f9a5f2021-06-26 22:22:38 +02001303 var txt = 'a'
1304 b = (txt =~ 'x' ? 1 : 2)
1305 assert_equal(2, b)
Bram Moolenaar015cf102021-06-26 21:52:02 +02001306 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001307 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar015cf102021-06-26 21:52:02 +02001308
1309 lines =<< trim END
Bram Moolenaar057e84a2021-02-28 16:55:11 +01001310 def ShadowLocal()
1311 var one = 1
1312 var l = [1, 2, 3]
1313 echo map(l, (one) => one)
1314 enddef
1315 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001316 v9.CheckDefFailure(lines, 'E1167:')
Bram Moolenaar057e84a2021-02-28 16:55:11 +01001317
1318 lines =<< trim END
1319 def Shadowarg(one: number)
1320 var l = [1, 2, 3]
1321 echo map(l, (one) => one)
1322 enddef
1323 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001324 v9.CheckDefFailure(lines, 'E1167:')
Bram Moolenaar767034c2021-04-09 17:24:52 +02001325
1326 lines =<< trim END
1327 echo ((a) => a)('aa', 'bb')
1328 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001329 v9.CheckDefAndScriptFailure(lines, 'E118:', 1)
Bram Moolenaarc4c56422021-07-21 20:38:46 +02001330
1331 lines =<< trim END
1332 echo 'aa'->((a) => a)('bb')
1333 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001334 v9.CheckDefFailure(lines, 'E118: Too many arguments for function: ->((a) => a)(''bb'')', 1)
1335 v9.CheckScriptFailure(['vim9script'] + lines, 'E118: Too many arguments for function: <lambda>', 2)
Bram Moolenaarb4d16cb2020-11-05 18:45:46 +01001336enddef
1337
Bram Moolenaara755fdb2021-11-20 21:35:41 +00001338def Test_lambda_line_nr()
1339 var lines =<< trim END
1340 vim9script
1341 # comment
1342 # comment
1343 var id = timer_start(1'000, (_) => 0)
1344 var out = execute('verbose ' .. timer_info(id)[0].callback
1345 ->string()
1346 ->substitute("('\\|')", ' ', 'g'))
1347 assert_match('Last set from .* line 4', out)
1348 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001349 v9.CheckScriptSuccess(lines)
Bram Moolenaara755fdb2021-11-20 21:35:41 +00001350enddef
1351
Bram Moolenaar5f91e742021-03-17 21:29:29 +01001352def FilterWithCond(x: string, Cond: func(string): bool): bool
1353 return Cond(x)
1354enddef
1355
Bram Moolenaar0346b792021-01-31 22:18:29 +01001356def Test_lambda_return_type()
1357 var lines =<< trim END
1358 var Ref = (): => 123
1359 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001360 v9.CheckDefAndScriptFailure(lines, 'E1157:', 1)
Bram Moolenaar5f91e742021-03-17 21:29:29 +01001361
Yegappan Lakshmanan611728f2021-05-24 15:15:47 +02001362 # no space before the return type
1363 lines =<< trim END
1364 var Ref = (x):number => x + 1
1365 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001366 v9.CheckDefAndScriptFailure(lines, 'E1069:', 1)
Yegappan Lakshmanan611728f2021-05-24 15:15:47 +02001367
Bram Moolenaar5f91e742021-03-17 21:29:29 +01001368 # this works
1369 for x in ['foo', 'boo']
Bram Moolenaar62aec932022-01-29 21:45:34 +00001370 echo g:FilterWithCond(x, (v) => v =~ '^b')
Bram Moolenaar5f91e742021-03-17 21:29:29 +01001371 endfor
1372
1373 # this fails
1374 lines =<< trim END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001375 echo g:FilterWithCond('foo', (v) => v .. '^b')
Bram Moolenaar5f91e742021-03-17 21:29:29 +01001376 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001377 v9.CheckDefAndScriptFailure(lines, 'E1013: Argument 2: type mismatch, expected func(string): bool but got func(any): string', 1)
Bram Moolenaara9931532021-06-12 15:58:16 +02001378
1379 lines =<< trim END
1380 var Lambda1 = (x) => {
1381 return x
1382 }
1383 assert_equal('asdf', Lambda1('asdf'))
1384 var Lambda2 = (x): string => {
1385 return x
1386 }
1387 assert_equal('foo', Lambda2('foo'))
1388 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001389 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaara9931532021-06-12 15:58:16 +02001390
1391 lines =<< trim END
1392 var Lambda = (x): string => {
1393 return x
1394 }
1395 echo Lambda(['foo'])
1396 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001397 v9.CheckDefExecAndScriptFailure(lines, 'E1012:')
Bram Moolenaar0346b792021-01-31 22:18:29 +01001398enddef
1399
Bram Moolenaar709664c2020-12-12 14:33:41 +01001400def Test_lambda_uses_assigned_var()
Bram Moolenaar62aec932022-01-29 21:45:34 +00001401 v9.CheckDefSuccess([
Bram Moolenaar709664c2020-12-12 14:33:41 +01001402 'var x: any = "aaa"'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001403 'x = filter(["bbb"], (_, v) => v =~ x)'])
Bram Moolenaar709664c2020-12-12 14:33:41 +01001404enddef
1405
Bram Moolenaar18062fc2021-03-05 21:35:47 +01001406def Test_pass_legacy_lambda_to_def_func()
1407 var lines =<< trim END
1408 vim9script
1409 func Foo()
1410 eval s:Bar({x -> 0})
1411 endfunc
1412 def Bar(y: any)
1413 enddef
1414 Foo()
1415 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001416 v9.CheckScriptSuccess(lines)
Bram Moolenaar831bdf82021-06-22 19:32:17 +02001417
1418 lines =<< trim END
1419 vim9script
Bram Moolenaar7a40ff02021-07-04 15:54:08 +02001420 def g:TestFunc(f: func)
Bram Moolenaar831bdf82021-06-22 19:32:17 +02001421 enddef
1422 legacy call g:TestFunc({-> 0})
1423 delfunc g:TestFunc
1424
1425 def g:TestFunc(f: func(number))
1426 enddef
1427 legacy call g:TestFunc({nr -> 0})
1428 delfunc g:TestFunc
1429 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001430 v9.CheckScriptSuccess(lines)
Bram Moolenaar18062fc2021-03-05 21:35:47 +01001431enddef
1432
Bram Moolenaar844fb642021-10-23 13:32:30 +01001433def Test_lambda_in_reduce_line_break()
1434 # this was using freed memory
1435 var lines =<< trim END
1436 vim9script
1437 const result: dict<number> =
1438 ['Bob', 'Sam', 'Cat', 'Bob', 'Cat', 'Cat']
1439 ->reduce((acc, val) => {
1440 if has_key(acc, val)
1441 acc[val] += 1
1442 return acc
1443 else
1444 acc[val] = 1
1445 return acc
1446 endif
1447 }, {})
1448 assert_equal({Bob: 2, Sam: 1, Cat: 3}, result)
1449 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001450 v9.CheckScriptSuccess(lines)
Bram Moolenaar844fb642021-10-23 13:32:30 +01001451enddef
1452
Bram Moolenaardcb53be2021-12-09 14:23:43 +00001453def Test_set_opfunc_to_lambda()
1454 var lines =<< trim END
1455 vim9script
1456 nnoremap <expr> <F4> <SID>CountSpaces() .. '_'
1457 def CountSpaces(type = ''): string
1458 if type == ''
1459 &operatorfunc = (t) => CountSpaces(t)
1460 return 'g@'
1461 endif
1462 normal! '[V']y
1463 g:result = getreg('"')->count(' ')
1464 return ''
1465 enddef
1466 new
1467 'a b c d e'->setline(1)
1468 feedkeys("\<F4>", 'x')
1469 assert_equal(4, g:result)
1470 bwipe!
1471 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001472 v9.CheckScriptSuccess(lines)
Bram Moolenaardcb53be2021-12-09 14:23:43 +00001473enddef
1474
Bram Moolenaaref082e12021-12-12 21:02:03 +00001475def Test_set_opfunc_to_global_function()
1476 var lines =<< trim END
1477 vim9script
1478 def g:CountSpaces(type = ''): string
1479 normal! '[V']y
1480 g:result = getreg('"')->count(' ')
1481 return ''
1482 enddef
Bram Moolenaarb15cf442021-12-16 15:49:43 +00001483 # global function works at script level
Bram Moolenaaref082e12021-12-12 21:02:03 +00001484 &operatorfunc = g:CountSpaces
1485 new
1486 'a b c d e'->setline(1)
1487 feedkeys("g@_", 'x')
1488 assert_equal(4, g:result)
Bram Moolenaarb15cf442021-12-16 15:49:43 +00001489
1490 &operatorfunc = ''
1491 g:result = 0
1492 # global function works in :def function
1493 def Func()
1494 &operatorfunc = g:CountSpaces
1495 enddef
1496 Func()
1497 feedkeys("g@_", 'x')
1498 assert_equal(4, g:result)
1499
Bram Moolenaaref082e12021-12-12 21:02:03 +00001500 bwipe!
1501 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001502 v9.CheckScriptSuccess(lines)
Bram Moolenaaref082e12021-12-12 21:02:03 +00001503 &operatorfunc = ''
1504enddef
1505
Bram Moolenaar33b968d2021-12-13 11:31:04 +00001506def Test_use_script_func_name_with_prefix()
1507 var lines =<< trim END
1508 vim9script
Bram Moolenaara749a422022-02-12 19:52:25 +00001509 func g:Getit()
Bram Moolenaar33b968d2021-12-13 11:31:04 +00001510 return 'it'
1511 endfunc
Bram Moolenaara749a422022-02-12 19:52:25 +00001512 var Fn = g:Getit
Bram Moolenaar33b968d2021-12-13 11:31:04 +00001513 assert_equal('it', Fn())
1514 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001515 v9.CheckScriptSuccess(lines)
Bram Moolenaar33b968d2021-12-13 11:31:04 +00001516enddef
1517
Bram Moolenaardd297bc2021-12-10 10:37:38 +00001518def Test_lambda_type_allocated()
1519 # Check that unreferencing a partial using a lambda can use the variable type
1520 # after the lambda has been freed and does not leak memory.
1521 var lines =<< trim END
1522 vim9script
1523
1524 func MyomniFunc1(val, findstart, base)
1525 return a:findstart ? 0 : []
1526 endfunc
1527
1528 var Lambda = (a, b) => MyomniFunc1(19, a, b)
1529 &omnifunc = Lambda
1530 Lambda = (a, b) => MyomniFunc1(20, a, b)
1531 &omnifunc = string(Lambda)
1532 Lambda = (a, b) => strlen(a)
1533 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001534 v9.CheckScriptSuccess(lines)
Bram Moolenaardd297bc2021-12-10 10:37:38 +00001535enddef
1536
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001537" Default arg and varargs
1538def MyDefVarargs(one: string, two = 'foo', ...rest: list<string>): string
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001539 var res = one .. ',' .. two
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001540 for s in rest
1541 res ..= ',' .. s
1542 endfor
1543 return res
1544enddef
1545
1546def Test_call_def_varargs()
Bram Moolenaar62aec932022-01-29 21:45:34 +00001547 assert_fails('g:MyDefVarargs()', 'E119:', '', 1, 'Test_call_def_varargs')
1548 g:MyDefVarargs('one')->assert_equal('one,foo')
1549 g:MyDefVarargs('one', 'two')->assert_equal('one,two')
1550 g:MyDefVarargs('one', 'two', 'three')->assert_equal('one,two,three')
1551 v9.CheckDefFailure(['g:MyDefVarargs("one", 22)'],
Bram Moolenaar77072282020-09-16 17:55:40 +02001552 'E1013: Argument 2: type mismatch, expected string but got number')
Bram Moolenaar62aec932022-01-29 21:45:34 +00001553 v9.CheckDefFailure(['g:MyDefVarargs("one", "two", 123)'],
Bram Moolenaar77072282020-09-16 17:55:40 +02001554 'E1013: Argument 3: type mismatch, expected string but got number')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +02001555
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001556 var lines =<< trim END
Bram Moolenaar24aa48b2020-07-25 16:33:02 +02001557 vim9script
1558 def Func(...l: list<string>)
1559 echo l
1560 enddef
1561 Func('a', 'b', 'c')
1562 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001563 v9.CheckScriptSuccess(lines)
Bram Moolenaar24aa48b2020-07-25 16:33:02 +02001564
1565 lines =<< trim END
1566 vim9script
1567 def Func(...l: list<string>)
1568 echo l
1569 enddef
1570 Func()
1571 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001572 v9.CheckScriptSuccess(lines)
Bram Moolenaar24aa48b2020-07-25 16:33:02 +02001573
1574 lines =<< trim END
1575 vim9script
Bram Moolenaar2a389082021-04-09 20:24:31 +02001576 def Func(...l: list<any>)
Bram Moolenaar2f8cbc42020-09-16 17:22:59 +02001577 echo l
1578 enddef
1579 Func(0)
1580 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001581 v9.CheckScriptSuccess(lines)
Bram Moolenaar2f8cbc42020-09-16 17:22:59 +02001582
1583 lines =<< trim END
1584 vim9script
Bram Moolenaar2a389082021-04-09 20:24:31 +02001585 def Func(...l: any)
1586 echo l
1587 enddef
1588 Func(0)
1589 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001590 v9.CheckScriptFailure(lines, 'E1180:', 2)
Bram Moolenaar2a389082021-04-09 20:24:31 +02001591
1592 lines =<< trim END
1593 vim9script
Bram Moolenaar28022722020-09-21 22:02:49 +02001594 def Func(..._l: list<string>)
1595 echo _l
1596 enddef
1597 Func('a', 'b', 'c')
1598 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001599 v9.CheckScriptSuccess(lines)
Bram Moolenaar28022722020-09-21 22:02:49 +02001600
1601 lines =<< trim END
1602 vim9script
Bram Moolenaar24aa48b2020-07-25 16:33:02 +02001603 def Func(...l: list<string>)
1604 echo l
1605 enddef
1606 Func(1, 2, 3)
1607 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001608 v9.CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +02001609
1610 lines =<< trim END
1611 vim9script
1612 def Func(...l: list<string>)
1613 echo l
1614 enddef
1615 Func('a', 9)
1616 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001617 v9.CheckScriptFailure(lines, 'E1013: Argument 2: type mismatch')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +02001618
1619 lines =<< trim END
1620 vim9script
1621 def Func(...l: list<string>)
1622 echo l
1623 enddef
1624 Func(1, 'a')
1625 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001626 v9.CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch')
Bram Moolenaar4f53b792021-02-07 15:59:49 +01001627
1628 lines =<< trim END
1629 vim9script
1630 def Func( # some comment
1631 ...l = []
1632 )
1633 echo l
1634 enddef
1635 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001636 v9.CheckScriptFailure(lines, 'E1160:')
Bram Moolenaar6ce46b92021-08-07 15:35:36 +02001637
1638 lines =<< trim END
1639 vim9script
1640 def DoIt()
1641 g:Later('')
1642 enddef
1643 defcompile
1644 def g:Later(...l: list<number>)
1645 enddef
1646 DoIt()
1647 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001648 v9.CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected number but got string')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001649enddef
1650
Bram Moolenaar1378fbc2020-04-11 20:50:33 +02001651let s:value = ''
1652
1653def FuncOneDefArg(opt = 'text')
1654 s:value = opt
1655enddef
1656
1657def FuncTwoDefArg(nr = 123, opt = 'text'): string
1658 return nr .. opt
1659enddef
1660
1661def FuncVarargs(...arg: list<string>): string
1662 return join(arg, ',')
1663enddef
1664
1665def Test_func_type_varargs()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001666 var RefDefArg: func(?string)
Bram Moolenaar848fadd2022-01-30 15:28:30 +00001667 RefDefArg = g:FuncOneDefArg
Bram Moolenaar1378fbc2020-04-11 20:50:33 +02001668 RefDefArg()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001669 s:value->assert_equal('text')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +02001670 RefDefArg('some')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001671 s:value->assert_equal('some')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +02001672
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001673 var RefDef2Arg: func(?number, ?string): string
Bram Moolenaar848fadd2022-01-30 15:28:30 +00001674 RefDef2Arg = g:FuncTwoDefArg
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001675 RefDef2Arg()->assert_equal('123text')
1676 RefDef2Arg(99)->assert_equal('99text')
1677 RefDef2Arg(77, 'some')->assert_equal('77some')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +02001678
Bram Moolenaar62aec932022-01-29 21:45:34 +00001679 v9.CheckDefFailure(['var RefWrong: func(string?)'], 'E1010:')
1680 v9.CheckDefFailure(['var RefWrong: func(?string, string)'], 'E1007:')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +02001681
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001682 var RefVarargs: func(...list<string>): string
Bram Moolenaar848fadd2022-01-30 15:28:30 +00001683 RefVarargs = g:FuncVarargs
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001684 RefVarargs()->assert_equal('')
1685 RefVarargs('one')->assert_equal('one')
1686 RefVarargs('one', 'two')->assert_equal('one,two')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +02001687
Bram Moolenaar62aec932022-01-29 21:45:34 +00001688 v9.CheckDefFailure(['var RefWrong: func(...list<string>, string)'], 'E110:')
1689 v9.CheckDefFailure(['var RefWrong: func(...list<string>, ?string)'], 'E110:')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +02001690enddef
1691
Bram Moolenaar0b76b422020-04-07 22:05:08 +02001692" Only varargs
1693def MyVarargsOnly(...args: list<string>): string
1694 return join(args, ',')
1695enddef
1696
1697def Test_call_varargs_only()
Bram Moolenaar62aec932022-01-29 21:45:34 +00001698 g:MyVarargsOnly()->assert_equal('')
1699 g:MyVarargsOnly('one')->assert_equal('one')
1700 g:MyVarargsOnly('one', 'two')->assert_equal('one,two')
1701 v9.CheckDefFailure(['g:MyVarargsOnly(1)'], 'E1013: Argument 1: type mismatch, expected string but got number')
1702 v9.CheckDefFailure(['g:MyVarargsOnly("one", 2)'], 'E1013: Argument 2: type mismatch, expected string but got number')
Bram Moolenaar0b76b422020-04-07 22:05:08 +02001703enddef
1704
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001705def Test_using_var_as_arg()
Bram Moolenaard2939812021-12-30 17:09:05 +00001706 var lines =<< trim END
1707 def Func(x: number)
1708 var x = 234
1709 enddef
1710 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001711 v9.CheckDefFailure(lines, 'E1006:')
Bram Moolenaard2939812021-12-30 17:09:05 +00001712
1713 lines =<< trim END
1714 def Func(Ref: number)
1715 def Ref()
1716 enddef
1717 enddef
1718 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001719 v9.CheckDefFailure(lines, 'E1073:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001720enddef
1721
Bram Moolenaar62aec932022-01-29 21:45:34 +00001722def s:DictArg(arg: dict<string>)
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +02001723 arg['key'] = 'value'
1724enddef
1725
Bram Moolenaar62aec932022-01-29 21:45:34 +00001726def s:ListArg(arg: list<string>)
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +02001727 arg[0] = 'value'
1728enddef
1729
1730def Test_assign_to_argument()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +02001731 # works for dict and list
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001732 var d: dict<string> = {}
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +02001733 DictArg(d)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001734 d['key']->assert_equal('value')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001735 var l: list<string> = []
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +02001736 ListArg(l)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001737 l[0]->assert_equal('value')
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +02001738
Bram Moolenaar62aec932022-01-29 21:45:34 +00001739 v9.CheckScriptFailure(['def Func(arg: number)', 'arg = 3', 'enddef', 'defcompile'], 'E1090:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001740 delfunc! g:Func
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +02001741enddef
1742
Bram Moolenaarb816dae2020-09-20 22:04:00 +02001743" These argument names are reserved in legacy functions.
Bram Moolenaar62aec932022-01-29 21:45:34 +00001744def s:WithReservedNames(firstline: string, lastline: string): string
Bram Moolenaarb816dae2020-09-20 22:04:00 +02001745 return firstline .. lastline
1746enddef
1747
1748def Test_argument_names()
1749 assert_equal('OK', WithReservedNames('O', 'K'))
1750enddef
1751
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001752def Test_call_func_defined_later()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001753 g:DefinedLater('one')->assert_equal('one')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001754 assert_fails('NotDefined("one")', 'E117:', '', 2, 'Test_call_func_defined_later')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001755enddef
1756
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +02001757func DefinedLater(arg)
1758 return a:arg
1759endfunc
1760
1761def Test_call_funcref()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001762 g:SomeFunc('abc')->assert_equal(3)
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001763 assert_fails('NotAFunc()', 'E117:', '', 2, 'Test_call_funcref') # comment after call
Bram Moolenaar2ef91562021-12-11 16:14:07 +00001764 assert_fails('g:NotAFunc()', 'E1085:', '', 3, 'Test_call_funcref')
Bram Moolenaar2f1980f2020-07-22 19:30:06 +02001765
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001766 var lines =<< trim END
Bram Moolenaar2f1980f2020-07-22 19:30:06 +02001767 vim9script
1768 def RetNumber(): number
1769 return 123
1770 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001771 var Funcref: func: number = function('RetNumber')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001772 Funcref()->assert_equal(123)
Bram Moolenaar2f1980f2020-07-22 19:30:06 +02001773 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001774 v9.CheckScriptSuccess(lines)
Bram Moolenaar0f60e802020-07-22 20:16:11 +02001775
1776 lines =<< trim END
1777 vim9script
1778 def RetNumber(): number
1779 return 123
1780 enddef
1781 def Bar(F: func: number): number
1782 return F()
1783 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001784 var Funcref = function('RetNumber')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001785 Bar(Funcref)->assert_equal(123)
Bram Moolenaar0f60e802020-07-22 20:16:11 +02001786 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001787 v9.CheckScriptSuccess(lines)
Bram Moolenaarbfba8652020-07-23 20:09:10 +02001788
1789 lines =<< trim END
1790 vim9script
1791 def UseNumber(nr: number)
1792 echo nr
1793 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001794 var Funcref: func(number) = function('UseNumber')
Bram Moolenaarbfba8652020-07-23 20:09:10 +02001795 Funcref(123)
1796 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001797 v9.CheckScriptSuccess(lines)
Bram Moolenaarb8070e32020-07-23 20:56:04 +02001798
1799 lines =<< trim END
1800 vim9script
1801 def UseNumber(nr: number)
1802 echo nr
1803 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001804 var Funcref: func(string) = function('UseNumber')
Bram Moolenaarb8070e32020-07-23 20:56:04 +02001805 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001806 v9.CheckScriptFailure(lines, 'E1012: Type mismatch; expected func(string) but got func(number)')
Bram Moolenaar4fc224c2020-07-26 17:56:25 +02001807
1808 lines =<< trim END
1809 vim9script
1810 def EchoNr(nr = 34)
1811 g:echo = nr
1812 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001813 var Funcref: func(?number) = function('EchoNr')
Bram Moolenaar4fc224c2020-07-26 17:56:25 +02001814 Funcref()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001815 g:echo->assert_equal(34)
Bram Moolenaar4fc224c2020-07-26 17:56:25 +02001816 Funcref(123)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001817 g:echo->assert_equal(123)
Bram Moolenaar4fc224c2020-07-26 17:56:25 +02001818 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001819 v9.CheckScriptSuccess(lines)
Bram Moolenaarace61322020-07-26 18:16:58 +02001820
1821 lines =<< trim END
1822 vim9script
1823 def EchoList(...l: list<number>)
1824 g:echo = l
1825 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001826 var Funcref: func(...list<number>) = function('EchoList')
Bram Moolenaarace61322020-07-26 18:16:58 +02001827 Funcref()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001828 g:echo->assert_equal([])
Bram Moolenaarace61322020-07-26 18:16:58 +02001829 Funcref(1, 2, 3)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001830 g:echo->assert_equal([1, 2, 3])
Bram Moolenaarace61322020-07-26 18:16:58 +02001831 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001832 v9.CheckScriptSuccess(lines)
Bram Moolenaar01865ad2020-07-26 18:33:09 +02001833
1834 lines =<< trim END
1835 vim9script
1836 def OptAndVar(nr: number, opt = 12, ...l: list<number>): number
1837 g:optarg = opt
1838 g:listarg = l
1839 return nr
1840 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001841 var Funcref: func(number, ?number, ...list<number>): number = function('OptAndVar')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001842 Funcref(10)->assert_equal(10)
1843 g:optarg->assert_equal(12)
1844 g:listarg->assert_equal([])
Bram Moolenaar01865ad2020-07-26 18:33:09 +02001845
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001846 Funcref(11, 22)->assert_equal(11)
1847 g:optarg->assert_equal(22)
1848 g:listarg->assert_equal([])
Bram Moolenaar01865ad2020-07-26 18:33:09 +02001849
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001850 Funcref(17, 18, 1, 2, 3)->assert_equal(17)
1851 g:optarg->assert_equal(18)
1852 g:listarg->assert_equal([1, 2, 3])
Bram Moolenaar01865ad2020-07-26 18:33:09 +02001853 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001854 v9.CheckScriptSuccess(lines)
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +02001855enddef
1856
1857let SomeFunc = function('len')
1858let NotAFunc = 'text'
1859
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +02001860def CombineFuncrefTypes()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +02001861 # same arguments, different return type
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001862 var Ref1: func(bool): string
1863 var Ref2: func(bool): number
1864 var Ref3: func(bool): any
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +02001865 Ref3 = g:cond ? Ref1 : Ref2
1866
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +02001867 # different number of arguments
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001868 var Refa1: func(bool): number
1869 var Refa2: func(bool, number): number
1870 var Refa3: func: number
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +02001871 Refa3 = g:cond ? Refa1 : Refa2
1872
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +02001873 # different argument types
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001874 var Refb1: func(bool, string): number
1875 var Refb2: func(string, number): number
1876 var Refb3: func(any, any): number
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +02001877 Refb3 = g:cond ? Refb1 : Refb2
1878enddef
1879
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001880def FuncWithForwardCall()
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +02001881 return g:DefinedEvenLater("yes")
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001882enddef
1883
1884def DefinedEvenLater(arg: string): string
1885 return arg
1886enddef
1887
1888def Test_error_in_nested_function()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +02001889 # Error in called function requires unwinding the call stack.
Bram Moolenaar848fadd2022-01-30 15:28:30 +00001890 assert_fails('g:FuncWithForwardCall()', 'E1096:', '', 1, 'FuncWithForwardCall')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001891enddef
1892
Bram Moolenaar4bf10062021-12-28 17:23:12 +00001893def Test_nested_function_with_nextcmd()
Bram Moolenaar9c23f9b2021-12-26 14:23:22 +00001894 var lines =<< trim END
1895 vim9script
1896 # Define an outer function
1897 def FirstFunction()
1898 # Define an inner function
1899 def SecondFunction()
1900 # the function has a body, a double free is detected.
1901 AAAAA
1902
1903 # enddef followed by | or } followed by # one or more characters
1904 enddef|BBBB
1905 enddef
1906
1907 # Compile all functions
1908 defcompile
1909 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001910 v9.CheckScriptFailure(lines, 'E1173: Text found after enddef: BBBB')
Bram Moolenaar9c23f9b2021-12-26 14:23:22 +00001911enddef
1912
Bram Moolenaar4bf10062021-12-28 17:23:12 +00001913def Test_nested_function_with_args_split()
1914 var lines =<< trim END
1915 vim9script
1916 def FirstFunction()
1917 def SecondFunction(
1918 )
1919 # had a double free if the right parenthesis of the nested function is
1920 # on the next line
1921
1922 enddef|BBBB
1923 enddef
1924 # Compile all functions
1925 defcompile
1926 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001927 v9.CheckScriptFailure(lines, 'E1173: Text found after enddef: BBBB')
Bram Moolenaar7473a842021-12-28 17:55:26 +00001928
1929 lines =<< trim END
1930 vim9script
1931 def FirstFunction()
1932 func SecondFunction()
1933 endfunc|BBBB
1934 enddef
1935 defcompile
1936 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001937 v9.CheckScriptFailure(lines, 'E1173: Text found after endfunction: BBBB')
Bram Moolenaar4bf10062021-12-28 17:23:12 +00001938enddef
1939
Bram Moolenaar9f1a39a2022-01-08 15:39:39 +00001940def Test_error_in_function_args()
1941 var lines =<< trim END
1942 def FirstFunction()
1943 def SecondFunction(J =
1944 # Nois
1945 # one
1946
1947 enddef|BBBB
1948 enddef
1949 # Compile all functions
1950 defcompile
1951 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001952 v9.CheckScriptFailure(lines, 'E488:')
Bram Moolenaar9f1a39a2022-01-08 15:39:39 +00001953enddef
1954
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001955def Test_return_type_wrong()
Bram Moolenaar62aec932022-01-29 21:45:34 +00001956 v9.CheckScriptFailure([
Bram Moolenaar5a849da2020-08-08 16:47:30 +02001957 'def Func(): number',
1958 'return "a"',
1959 'enddef',
1960 'defcompile'], 'expected number but got string')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001961 delfunc! g:Func
Bram Moolenaar62aec932022-01-29 21:45:34 +00001962 v9.CheckScriptFailure([
Bram Moolenaar5a849da2020-08-08 16:47:30 +02001963 'def Func(): string',
1964 'return 1',
1965 'enddef',
1966 'defcompile'], 'expected string but got number')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001967 delfunc! g:Func
Bram Moolenaar62aec932022-01-29 21:45:34 +00001968 v9.CheckScriptFailure([
Bram Moolenaar5a849da2020-08-08 16:47:30 +02001969 'def Func(): void',
1970 'return "a"',
1971 'enddef',
1972 'defcompile'],
1973 'E1096: Returning a value in a function without a return type')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001974 delfunc! g:Func
Bram Moolenaar62aec932022-01-29 21:45:34 +00001975 v9.CheckScriptFailure([
Bram Moolenaar5a849da2020-08-08 16:47:30 +02001976 'def Func()',
1977 'return "a"',
1978 'enddef',
1979 'defcompile'],
1980 'E1096: Returning a value in a function without a return type')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001981 delfunc! g:Func
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001982
Bram Moolenaar62aec932022-01-29 21:45:34 +00001983 v9.CheckScriptFailure([
Bram Moolenaar5a849da2020-08-08 16:47:30 +02001984 'def Func(): number',
1985 'return',
1986 'enddef',
1987 'defcompile'], 'E1003:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001988 delfunc! g:Func
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001989
Bram Moolenaar62aec932022-01-29 21:45:34 +00001990 v9.CheckScriptFailure([
Bram Moolenaar33ea9fd2021-08-08 19:07:37 +02001991 'def Func():number',
1992 'return 123',
1993 'enddef',
1994 'defcompile'], 'E1069:')
1995 delfunc! g:Func
1996
Bram Moolenaar62aec932022-01-29 21:45:34 +00001997 v9.CheckScriptFailure([
Bram Moolenaar33ea9fd2021-08-08 19:07:37 +02001998 'def Func() :number',
1999 'return 123',
2000 'enddef',
2001 'defcompile'], 'E1059:')
2002 delfunc! g:Func
2003
Bram Moolenaar62aec932022-01-29 21:45:34 +00002004 v9.CheckScriptFailure([
Bram Moolenaar33ea9fd2021-08-08 19:07:37 +02002005 'def Func() : number',
2006 'return 123',
2007 'enddef',
2008 'defcompile'], 'E1059:')
2009 delfunc! g:Func
2010
Bram Moolenaar62aec932022-01-29 21:45:34 +00002011 v9.CheckScriptFailure(['def Func(): list', 'return []', 'enddef'], 'E1008:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01002012 delfunc! g:Func
Bram Moolenaar62aec932022-01-29 21:45:34 +00002013 v9.CheckScriptFailure(['def Func(): dict', 'return {}', 'enddef'], 'E1008:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01002014 delfunc! g:Func
Bram Moolenaar62aec932022-01-29 21:45:34 +00002015 v9.CheckScriptFailure(['def Func()', 'return 1'], 'E1057:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01002016 delfunc! g:Func
Bram Moolenaar5a849da2020-08-08 16:47:30 +02002017
Bram Moolenaar62aec932022-01-29 21:45:34 +00002018 v9.CheckScriptFailure([
Bram Moolenaar5a849da2020-08-08 16:47:30 +02002019 'vim9script',
2020 'def FuncB()',
2021 ' return 123',
2022 'enddef',
2023 'def FuncA()',
2024 ' FuncB()',
2025 'enddef',
2026 'defcompile'], 'E1096:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002027enddef
2028
2029def Test_arg_type_wrong()
Bram Moolenaar62aec932022-01-29 21:45:34 +00002030 v9.CheckScriptFailure(['def Func3(items: list)', 'echo "a"', 'enddef'], 'E1008: Missing <type>')
2031 v9.CheckScriptFailure(['def Func4(...)', 'echo "a"', 'enddef'], 'E1055: Missing name after ...')
2032 v9.CheckScriptFailure(['def Func5(items:string)', 'echo "a"'], 'E1069:')
2033 v9.CheckScriptFailure(['def Func5(items)', 'echo "a"'], 'E1077:')
2034 v9.CheckScriptFailure(['def Func6(...x:list<number>)', 'echo "a"', 'enddef'], 'E1069:')
2035 v9.CheckScriptFailure(['def Func7(...x: int)', 'echo "a"', 'enddef'], 'E1010:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002036enddef
2037
Bram Moolenaar86cdb8a2021-04-06 19:01:03 +02002038def Test_white_space_before_comma()
2039 var lines =<< trim END
2040 vim9script
2041 def Func(a: number , b: number)
2042 enddef
2043 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002044 v9.CheckScriptFailure(lines, 'E1068:')
Yegappan Lakshmanan611728f2021-05-24 15:15:47 +02002045 call assert_fails('vim9cmd echo stridx("a" .. "b" , "a")', 'E1068:')
Bram Moolenaar86cdb8a2021-04-06 19:01:03 +02002046enddef
2047
Bram Moolenaar608d78f2021-03-06 22:33:12 +01002048def Test_white_space_after_comma()
2049 var lines =<< trim END
2050 vim9script
2051 def Func(a: number,b: number)
2052 enddef
2053 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002054 v9.CheckScriptFailure(lines, 'E1069:')
Bram Moolenaar608d78f2021-03-06 22:33:12 +01002055
2056 # OK in legacy function
2057 lines =<< trim END
2058 vim9script
2059 func Func(a,b)
2060 endfunc
2061 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002062 v9.CheckScriptSuccess(lines)
Bram Moolenaar608d78f2021-03-06 22:33:12 +01002063enddef
2064
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002065def Test_vim9script_call()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002066 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002067 vim9script
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002068 var name = ''
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002069 def MyFunc(arg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002070 name = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002071 enddef
2072 MyFunc('foobar')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002073 name->assert_equal('foobar')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002074
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002075 var str = 'barfoo'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002076 str->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002077 name->assert_equal('barfoo')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002078
Bram Moolenaar67979662020-06-20 22:50:47 +02002079 g:value = 'value'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002080 g:value->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002081 name->assert_equal('value')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002082
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002083 var listvar = []
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002084 def ListFunc(arg: list<number>)
2085 listvar = arg
2086 enddef
2087 [1, 2, 3]->ListFunc()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002088 listvar->assert_equal([1, 2, 3])
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002089
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002090 var dictvar = {}
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002091 def DictFunc(arg: dict<number>)
2092 dictvar = arg
2093 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +01002094 {a: 1, b: 2}->DictFunc()
2095 dictvar->assert_equal({a: 1, b: 2})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002096 def CompiledDict()
Bram Moolenaare0de1712020-12-02 17:36:54 +01002097 {a: 3, b: 4}->DictFunc()
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002098 enddef
2099 CompiledDict()
Bram Moolenaare0de1712020-12-02 17:36:54 +01002100 dictvar->assert_equal({a: 3, b: 4})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002101
Bram Moolenaare0de1712020-12-02 17:36:54 +01002102 {a: 3, b: 4}->DictFunc()
2103 dictvar->assert_equal({a: 3, b: 4})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002104
2105 ('text')->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002106 name->assert_equal('text')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002107 ("some")->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002108 name->assert_equal('some')
Bram Moolenaare6b53242020-07-01 17:28:33 +02002109
Bram Moolenaar13e12b82020-07-24 18:47:22 +02002110 # line starting with single quote is not a mark
Bram Moolenaar10409562020-07-29 20:00:38 +02002111 # line starting with double quote can be a method call
Bram Moolenaar3d48e252020-07-15 14:15:52 +02002112 'asdfasdf'->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002113 name->assert_equal('asdfasdf')
Bram Moolenaar10409562020-07-29 20:00:38 +02002114 "xyz"->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002115 name->assert_equal('xyz')
Bram Moolenaar3d48e252020-07-15 14:15:52 +02002116
2117 def UseString()
2118 'xyork'->MyFunc()
2119 enddef
2120 UseString()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002121 name->assert_equal('xyork')
Bram Moolenaar3d48e252020-07-15 14:15:52 +02002122
Bram Moolenaar10409562020-07-29 20:00:38 +02002123 def UseString2()
2124 "knife"->MyFunc()
2125 enddef
2126 UseString2()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002127 name->assert_equal('knife')
Bram Moolenaar10409562020-07-29 20:00:38 +02002128
Bram Moolenaar13e12b82020-07-24 18:47:22 +02002129 # prepending a colon makes it a mark
2130 new
2131 setline(1, ['aaa', 'bbb', 'ccc'])
2132 normal! 3Gmt1G
2133 :'t
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002134 getcurpos()[1]->assert_equal(3)
Bram Moolenaar13e12b82020-07-24 18:47:22 +02002135 bwipe!
2136
Bram Moolenaare6b53242020-07-01 17:28:33 +02002137 MyFunc(
2138 'continued'
2139 )
2140 assert_equal('continued',
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002141 name
Bram Moolenaare6b53242020-07-01 17:28:33 +02002142 )
2143
2144 call MyFunc(
2145 'more'
2146 ..
2147 'lines'
2148 )
2149 assert_equal(
2150 'morelines',
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002151 name)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002152 END
2153 writefile(lines, 'Xcall.vim')
2154 source Xcall.vim
2155 delete('Xcall.vim')
2156enddef
2157
2158def Test_vim9script_call_fail_decl()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002159 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002160 vim9script
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002161 var name = ''
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002162 def MyFunc(arg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002163 var name = 123
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002164 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +02002165 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002166 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002167 v9.CheckScriptFailure(lines, 'E1054:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002168enddef
2169
Bram Moolenaar65b95452020-07-19 14:03:09 +02002170def Test_vim9script_call_fail_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002171 var lines =<< trim END
Bram Moolenaar65b95452020-07-19 14:03:09 +02002172 vim9script
2173 def MyFunc(arg: string)
2174 echo arg
2175 enddef
2176 MyFunc(1234)
2177 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002178 v9.CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected string but got number')
Bram Moolenaar65b95452020-07-19 14:03:09 +02002179enddef
2180
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002181def Test_vim9script_call_fail_const()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002182 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002183 vim9script
2184 const var = ''
2185 def MyFunc(arg: string)
2186 var = 'asdf'
2187 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +02002188 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002189 END
2190 writefile(lines, 'Xcall_const.vim')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02002191 assert_fails('source Xcall_const.vim', 'E46:', '', 1, 'MyFunc')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002192 delete('Xcall_const.vim')
Bram Moolenaar3bdc90b2020-12-22 20:35:40 +01002193
2194 lines =<< trim END
2195 const g:Aconst = 77
2196 def Change()
2197 # comment
2198 g:Aconst = 99
2199 enddef
2200 call Change()
2201 unlet g:Aconst
2202 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002203 v9.CheckScriptFailure(lines, 'E741: Value is locked: Aconst', 2)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002204enddef
2205
2206" Test that inside :function a Python function can be defined, :def is not
2207" recognized.
2208func Test_function_python()
2209 CheckFeature python3
Bram Moolenaar727345e2020-09-27 23:33:59 +02002210 let py = 'python3'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002211 execute py "<< EOF"
2212def do_something():
2213 return 1
2214EOF
2215endfunc
2216
2217def Test_delfunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002218 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002219 vim9script
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02002220 def g:GoneSoon()
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002221 echo 'hello'
2222 enddef
2223
2224 def CallGoneSoon()
Bram Moolenaar62aec932022-01-29 21:45:34 +00002225 g:GoneSoon()
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002226 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +02002227 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002228
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02002229 delfunc g:GoneSoon
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002230 CallGoneSoon()
2231 END
2232 writefile(lines, 'XToDelFunc')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02002233 assert_fails('so XToDelFunc', 'E933:', '', 1, 'CallGoneSoon')
2234 assert_fails('so XToDelFunc', 'E933:', '', 1, 'CallGoneSoon')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002235
2236 delete('XToDelFunc')
2237enddef
2238
Bram Moolenaar7509ad82021-12-14 18:14:37 +00002239func Test_free_dict_while_in_funcstack()
2240 " relies on the sleep command
2241 CheckUnix
2242 call Run_Test_free_dict_while_in_funcstack()
2243endfunc
2244
2245def Run_Test_free_dict_while_in_funcstack()
Bram Moolenaar7509ad82021-12-14 18:14:37 +00002246 # this was freeing the TermRun() default argument dictionary while it was
2247 # still referenced in a funcstack_T
2248 var lines =<< trim END
2249 vim9script
2250
2251 &updatetime = 400
2252 def TermRun(_ = {})
2253 def Post()
2254 enddef
2255 def Exec()
2256 term_start('sleep 1', {
2257 term_finish: 'close',
2258 exit_cb: (_, _) => Post(),
2259 })
2260 enddef
2261 Exec()
2262 enddef
2263 nnoremap <F4> <Cmd>call <SID>TermRun()<CR>
2264 timer_start(100, (_) => feedkeys("\<F4>"))
2265 timer_start(1000, (_) => feedkeys("\<F4>"))
2266 sleep 1500m
2267 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002268 v9.CheckScriptSuccess(lines)
Bram Moolenaar7509ad82021-12-14 18:14:37 +00002269 nunmap <F4>
2270 set updatetime&
2271enddef
2272
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002273def Test_redef_failure()
Bram Moolenaard2c61702020-09-06 15:58:36 +02002274 writefile(['def Func0(): string', 'return "Func0"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002275 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +02002276 writefile(['def Func1(): string', 'return "Func1"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002277 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +02002278 writefile(['def! Func0(): string', 'enddef', 'defcompile'], 'Xdef')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02002279 assert_fails('so Xdef', 'E1027:', '', 1, 'Func0')
Bram Moolenaard2c61702020-09-06 15:58:36 +02002280 writefile(['def Func2(): string', 'return "Func2"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002281 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +02002282 delete('Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002283
Bram Moolenaar701cc6c2021-04-10 13:33:48 +02002284 assert_fails('g:Func0()', 'E1091:')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002285 g:Func1()->assert_equal('Func1')
2286 g:Func2()->assert_equal('Func2')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002287
2288 delfunc! Func0
2289 delfunc! Func1
2290 delfunc! Func2
2291enddef
2292
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +02002293def Test_vim9script_func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002294 var lines =<< trim END
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +02002295 vim9script
2296 func Func(arg)
2297 echo a:arg
2298 endfunc
2299 Func('text')
2300 END
2301 writefile(lines, 'XVim9Func')
2302 so XVim9Func
2303
2304 delete('XVim9Func')
2305enddef
2306
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002307let s:funcResult = 0
2308
2309def FuncNoArgNoRet()
Bram Moolenaar53900992020-08-22 19:02:02 +02002310 s:funcResult = 11
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002311enddef
2312
2313def FuncNoArgRetNumber(): number
Bram Moolenaar53900992020-08-22 19:02:02 +02002314 s:funcResult = 22
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002315 return 1234
2316enddef
2317
Bram Moolenaarec5929d2020-04-07 20:53:39 +02002318def FuncNoArgRetString(): string
Bram Moolenaar53900992020-08-22 19:02:02 +02002319 s:funcResult = 45
Bram Moolenaarec5929d2020-04-07 20:53:39 +02002320 return 'text'
2321enddef
2322
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002323def FuncOneArgNoRet(arg: number)
Bram Moolenaar53900992020-08-22 19:02:02 +02002324 s:funcResult = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002325enddef
2326
2327def FuncOneArgRetNumber(arg: number): number
Bram Moolenaar53900992020-08-22 19:02:02 +02002328 s:funcResult = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002329 return arg
2330enddef
2331
Bram Moolenaar08938ee2020-04-11 23:17:17 +02002332def FuncTwoArgNoRet(one: bool, two: number)
Bram Moolenaar53900992020-08-22 19:02:02 +02002333 s:funcResult = two
Bram Moolenaar08938ee2020-04-11 23:17:17 +02002334enddef
2335
Bram Moolenaar62aec932022-01-29 21:45:34 +00002336def s:FuncOneArgRetString(arg: string): string
Bram Moolenaarec5929d2020-04-07 20:53:39 +02002337 return arg
2338enddef
2339
Bram Moolenaar62aec932022-01-29 21:45:34 +00002340def s:FuncOneArgRetAny(arg: any): any
Bram Moolenaar89228602020-04-05 22:14:54 +02002341 return arg
2342enddef
2343
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002344def Test_func_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002345 var Ref1: func()
Bram Moolenaar53900992020-08-22 19:02:02 +02002346 s:funcResult = 0
Bram Moolenaar62aec932022-01-29 21:45:34 +00002347 Ref1 = g:FuncNoArgNoRet
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002348 Ref1()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002349 s:funcResult->assert_equal(11)
Bram Moolenaar4c683752020-04-05 21:38:23 +02002350
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002351 var Ref2: func
Bram Moolenaar53900992020-08-22 19:02:02 +02002352 s:funcResult = 0
Bram Moolenaar62aec932022-01-29 21:45:34 +00002353 Ref2 = g:FuncNoArgNoRet
Bram Moolenaar4c683752020-04-05 21:38:23 +02002354 Ref2()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002355 s:funcResult->assert_equal(11)
Bram Moolenaar4c683752020-04-05 21:38:23 +02002356
Bram Moolenaar53900992020-08-22 19:02:02 +02002357 s:funcResult = 0
Bram Moolenaar62aec932022-01-29 21:45:34 +00002358 Ref2 = g:FuncOneArgNoRet
Bram Moolenaar4c683752020-04-05 21:38:23 +02002359 Ref2(12)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002360 s:funcResult->assert_equal(12)
Bram Moolenaar4c683752020-04-05 21:38:23 +02002361
Bram Moolenaar53900992020-08-22 19:02:02 +02002362 s:funcResult = 0
Bram Moolenaar62aec932022-01-29 21:45:34 +00002363 Ref2 = g:FuncNoArgRetNumber
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002364 Ref2()->assert_equal(1234)
2365 s:funcResult->assert_equal(22)
Bram Moolenaar4c683752020-04-05 21:38:23 +02002366
Bram Moolenaar53900992020-08-22 19:02:02 +02002367 s:funcResult = 0
Bram Moolenaar848fadd2022-01-30 15:28:30 +00002368 Ref2 = g:FuncOneArgRetNumber
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002369 Ref2(13)->assert_equal(13)
2370 s:funcResult->assert_equal(13)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002371enddef
2372
Bram Moolenaar9978d472020-07-05 16:01:56 +02002373def Test_repeat_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002374 var res = 0
Bram Moolenaar9978d472020-07-05 16:01:56 +02002375 for n in repeat([1], 3)
2376 res += n
2377 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002378 res->assert_equal(3)
Bram Moolenaarfce82b32020-07-05 16:07:21 +02002379
2380 res = 0
2381 for n in add([1, 2], 3)
2382 res += n
2383 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002384 res->assert_equal(6)
Bram Moolenaar9978d472020-07-05 16:01:56 +02002385enddef
2386
Bram Moolenaar846178a2020-07-05 17:04:13 +02002387def Test_argv_return_type()
2388 next fileone filetwo
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002389 var res = ''
Bram Moolenaar846178a2020-07-05 17:04:13 +02002390 for name in argv()
2391 res ..= name
2392 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002393 res->assert_equal('fileonefiletwo')
Bram Moolenaar846178a2020-07-05 17:04:13 +02002394enddef
2395
Bram Moolenaarec5929d2020-04-07 20:53:39 +02002396def Test_func_type_part()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002397 var RefVoid: func: void
Bram Moolenaar62aec932022-01-29 21:45:34 +00002398 RefVoid = g:FuncNoArgNoRet
2399 RefVoid = g:FuncOneArgNoRet
2400 v9.CheckDefFailure(['var RefVoid: func: void', 'RefVoid = g:FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func(...) but got func(): number')
Bram Moolenaar848fadd2022-01-30 15:28:30 +00002401 v9.CheckDefFailure(['var RefVoid: func: void', 'RefVoid = g:FuncNoArgRetString'], 'E1012: Type mismatch; expected func(...) but got func(): string')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02002402
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002403 var RefAny: func(): any
Bram Moolenaar62aec932022-01-29 21:45:34 +00002404 RefAny = g:FuncNoArgRetNumber
Bram Moolenaar848fadd2022-01-30 15:28:30 +00002405 RefAny = g:FuncNoArgRetString
Bram Moolenaar62aec932022-01-29 21:45:34 +00002406 v9.CheckDefFailure(['var RefAny: func(): any', 'RefAny = g:FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(): any but got func()')
2407 v9.CheckDefFailure(['var RefAny: func(): any', 'RefAny = g:FuncOneArgNoRet'], 'E1012: Type mismatch; expected func(): any but got func(number)')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02002408
Bram Moolenaar6abd3dc2020-10-04 14:17:32 +02002409 var RefAnyNoArgs: func: any = RefAny
2410
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002411 var RefNr: func: number
Bram Moolenaar62aec932022-01-29 21:45:34 +00002412 RefNr = g:FuncNoArgRetNumber
Bram Moolenaar848fadd2022-01-30 15:28:30 +00002413 RefNr = g:FuncOneArgRetNumber
Bram Moolenaar62aec932022-01-29 21:45:34 +00002414 v9.CheckDefFailure(['var RefNr: func: number', 'RefNr = g:FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(...): number but got func()')
Bram Moolenaar848fadd2022-01-30 15:28:30 +00002415 v9.CheckDefFailure(['var RefNr: func: number', 'RefNr = g:FuncNoArgRetString'], 'E1012: Type mismatch; expected func(...): number but got func(): string')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02002416
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002417 var RefStr: func: string
Bram Moolenaar848fadd2022-01-30 15:28:30 +00002418 RefStr = g:FuncNoArgRetString
Bram Moolenaarec5929d2020-04-07 20:53:39 +02002419 RefStr = FuncOneArgRetString
Bram Moolenaar62aec932022-01-29 21:45:34 +00002420 v9.CheckDefFailure(['var RefStr: func: string', 'RefStr = g:FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(...): string but got func()')
2421 v9.CheckDefFailure(['var RefStr: func: string', 'RefStr = g:FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func(...): string but got func(): number')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02002422enddef
2423
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002424def Test_func_type_fails()
Bram Moolenaar62aec932022-01-29 21:45:34 +00002425 v9.CheckDefFailure(['var ref1: func()'], 'E704:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002426
Bram Moolenaar62aec932022-01-29 21:45:34 +00002427 v9.CheckDefFailure(['var Ref1: func()', 'Ref1 = g:FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func() but got func(): number')
2428 v9.CheckDefFailure(['var Ref1: func()', 'Ref1 = g:FuncOneArgNoRet'], 'E1012: Type mismatch; expected func() but got func(number)')
Bram Moolenaar848fadd2022-01-30 15:28:30 +00002429 v9.CheckDefFailure(['var Ref1: func()', 'Ref1 = g:FuncOneArgRetNumber'], 'E1012: Type mismatch; expected func() but got func(number): number')
Bram Moolenaar62aec932022-01-29 21:45:34 +00002430 v9.CheckDefFailure(['var Ref1: func(bool)', 'Ref1 = g:FuncTwoArgNoRet'], 'E1012: Type mismatch; expected func(bool) but got func(bool, number)')
2431 v9.CheckDefFailure(['var Ref1: func(?bool)', 'Ref1 = g:FuncTwoArgNoRet'], 'E1012: Type mismatch; expected func(?bool) but got func(bool, number)')
2432 v9.CheckDefFailure(['var Ref1: func(...bool)', 'Ref1 = g:FuncTwoArgNoRet'], 'E1012: Type mismatch; expected func(...bool) but got func(bool, number)')
Bram Moolenaar08938ee2020-04-11 23:17:17 +02002433
Bram Moolenaar62aec932022-01-29 21:45:34 +00002434 v9.CheckDefFailure(['var RefWrong: func(string ,number)'], 'E1068:')
2435 v9.CheckDefFailure(['var RefWrong: func(string,number)'], 'E1069:')
2436 v9.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:')
2437 v9.CheckDefFailure(['var RefWrong: func(bool):string'], 'E1069:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002438enddef
2439
Bram Moolenaar89228602020-04-05 22:14:54 +02002440def Test_func_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002441 var nr: number
Bram Moolenaar62aec932022-01-29 21:45:34 +00002442 nr = g:FuncNoArgRetNumber()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002443 nr->assert_equal(1234)
Bram Moolenaar89228602020-04-05 22:14:54 +02002444
2445 nr = FuncOneArgRetAny(122)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002446 nr->assert_equal(122)
Bram Moolenaar89228602020-04-05 22:14:54 +02002447
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002448 var str: string
Bram Moolenaar89228602020-04-05 22:14:54 +02002449 str = FuncOneArgRetAny('yes')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002450 str->assert_equal('yes')
Bram Moolenaar89228602020-04-05 22:14:54 +02002451
Bram Moolenaar62aec932022-01-29 21:45:34 +00002452 v9.CheckDefFailure(['var str: string', 'str = g:FuncNoArgRetNumber()'], 'E1012: Type mismatch; expected string but got number')
Bram Moolenaar89228602020-04-05 22:14:54 +02002453enddef
2454
Bram Moolenaar6abd3dc2020-10-04 14:17:32 +02002455def Test_func_common_type()
2456 def FuncOne(n: number): number
2457 return n
2458 enddef
2459 def FuncTwo(s: string): number
2460 return len(s)
2461 enddef
2462 def FuncThree(n: number, s: string): number
2463 return n + len(s)
2464 enddef
2465 var list = [FuncOne, FuncTwo, FuncThree]
2466 assert_equal(8, list[0](8))
2467 assert_equal(4, list[1]('word'))
2468 assert_equal(7, list[2](3, 'word'))
2469enddef
2470
Bram Moolenaar62aec932022-01-29 21:45:34 +00002471def s:MultiLine(
Bram Moolenaar5e774c72020-04-12 21:53:00 +02002472 arg1: string,
2473 arg2 = 1234,
2474 ...rest: list<string>
2475 ): string
2476 return arg1 .. arg2 .. join(rest, '-')
2477enddef
2478
Bram Moolenaar2c330432020-04-13 14:41:35 +02002479def MultiLineComment(
2480 arg1: string, # comment
2481 arg2 = 1234, # comment
2482 ...rest: list<string> # comment
2483 ): string # comment
2484 return arg1 .. arg2 .. join(rest, '-')
2485enddef
2486
Bram Moolenaar5e774c72020-04-12 21:53:00 +02002487def Test_multiline()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002488 MultiLine('text')->assert_equal('text1234')
2489 MultiLine('text', 777)->assert_equal('text777')
2490 MultiLine('text', 777, 'one')->assert_equal('text777one')
2491 MultiLine('text', 777, 'one', 'two')->assert_equal('text777one-two')
Bram Moolenaar5e774c72020-04-12 21:53:00 +02002492enddef
2493
Bram Moolenaar23e03252020-04-12 22:22:31 +02002494func Test_multiline_not_vim9()
Bram Moolenaar62aec932022-01-29 21:45:34 +00002495 call s:MultiLine('text')->assert_equal('text1234')
2496 call s:MultiLine('text', 777)->assert_equal('text777')
2497 call s:MultiLine('text', 777, 'one')->assert_equal('text777one')
2498 call s:MultiLine('text', 777, 'one', 'two')->assert_equal('text777one-two')
Bram Moolenaar23e03252020-04-12 22:22:31 +02002499endfunc
2500
Bram Moolenaar5e774c72020-04-12 21:53:00 +02002501
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02002502" When using CheckScriptFailure() for the below test, E1010 is generated instead
2503" of E1056.
2504func Test_E1056_1059()
2505 let caught_1056 = 0
2506 try
2507 def F():
2508 return 1
2509 enddef
2510 catch /E1056:/
2511 let caught_1056 = 1
2512 endtry
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002513 eval caught_1056->assert_equal(1)
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02002514
2515 let caught_1059 = 0
2516 try
2517 def F5(items : list)
2518 echo 'a'
2519 enddef
2520 catch /E1059:/
2521 let caught_1059 = 1
2522 endtry
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002523 eval caught_1059->assert_equal(1)
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02002524endfunc
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002525
Bram Moolenaar015f4262020-05-05 21:25:22 +02002526func DelMe()
2527 echo 'DelMe'
2528endfunc
2529
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02002530def Test_error_reporting()
2531 # comment lines at the start of the function
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002532 var lines =<< trim END
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02002533 " comment
2534 def Func()
2535 # comment
2536 # comment
2537 invalid
2538 enddef
2539 defcompile
2540 END
Bram Moolenaar08052222020-09-14 17:04:31 +02002541 writefile(lines, 'Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02002542 try
2543 source Xdef
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02002544 assert_report('should have failed')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02002545 catch /E476:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002546 v:exception->assert_match('Invalid command: invalid')
2547 v:throwpoint->assert_match(', line 3$')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02002548 endtry
Bram Moolenaar2d870f82020-12-05 13:41:01 +01002549 delfunc! g:Func
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02002550
2551 # comment lines after the start of the function
2552 lines =<< trim END
2553 " comment
2554 def Func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002555 var x = 1234
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02002556 # comment
2557 # comment
2558 invalid
2559 enddef
2560 defcompile
2561 END
Bram Moolenaar08052222020-09-14 17:04:31 +02002562 writefile(lines, 'Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02002563 try
2564 source Xdef
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02002565 assert_report('should have failed')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02002566 catch /E476:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002567 v:exception->assert_match('Invalid command: invalid')
2568 v:throwpoint->assert_match(', line 4$')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02002569 endtry
Bram Moolenaar2d870f82020-12-05 13:41:01 +01002570 delfunc! g:Func
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02002571
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02002572 lines =<< trim END
2573 vim9script
2574 def Func()
Bram Moolenaare0de1712020-12-02 17:36:54 +01002575 var db = {foo: 1, bar: 2}
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02002576 # comment
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002577 var x = db.asdf
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02002578 enddef
2579 defcompile
2580 Func()
2581 END
Bram Moolenaar08052222020-09-14 17:04:31 +02002582 writefile(lines, 'Xdef')
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02002583 try
2584 source Xdef
2585 assert_report('should have failed')
2586 catch /E716:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002587 v:throwpoint->assert_match('_Func, line 3$')
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02002588 endtry
Bram Moolenaar2d870f82020-12-05 13:41:01 +01002589 delfunc! g:Func
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02002590
Bram Moolenaar08052222020-09-14 17:04:31 +02002591 delete('Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02002592enddef
2593
Bram Moolenaar015f4262020-05-05 21:25:22 +02002594def Test_deleted_function()
Bram Moolenaar62aec932022-01-29 21:45:34 +00002595 v9.CheckDefExecFailure([
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002596 'var RefMe: func = function("g:DelMe")',
Bram Moolenaar015f4262020-05-05 21:25:22 +02002597 'delfunc g:DelMe',
2598 'echo RefMe()'], 'E117:')
2599enddef
2600
2601def Test_unknown_function()
Bram Moolenaar62aec932022-01-29 21:45:34 +00002602 v9.CheckDefExecFailure([
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002603 'var Ref: func = function("NotExist")',
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +02002604 'delfunc g:NotExist'], 'E700:')
Bram Moolenaar015f4262020-05-05 21:25:22 +02002605enddef
2606
Bram Moolenaar62aec932022-01-29 21:45:34 +00002607def s:RefFunc(Ref: func(any): any): string
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +02002608 return Ref('more')
2609enddef
2610
2611def Test_closure_simple()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002612 var local = 'some '
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002613 RefFunc((s) => local .. s)->assert_equal('some more')
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +02002614enddef
2615
Bram Moolenaar62aec932022-01-29 21:45:34 +00002616def s:MakeRef()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002617 var local = 'some '
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002618 g:Ref = (s) => local .. s
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02002619enddef
2620
2621def Test_closure_ref_after_return()
2622 MakeRef()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002623 g:Ref('thing')->assert_equal('some thing')
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02002624 unlet g:Ref
2625enddef
2626
Bram Moolenaar62aec932022-01-29 21:45:34 +00002627def s:MakeTwoRefs()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002628 var local = ['some']
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002629 g:Extend = (s) => local->add(s)
2630 g:Read = () => local
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02002631enddef
2632
2633def Test_closure_two_refs()
2634 MakeTwoRefs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002635 join(g:Read(), ' ')->assert_equal('some')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02002636 g:Extend('more')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002637 join(g:Read(), ' ')->assert_equal('some more')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02002638 g:Extend('even')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002639 join(g:Read(), ' ')->assert_equal('some more even')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02002640
2641 unlet g:Extend
2642 unlet g:Read
2643enddef
2644
Bram Moolenaar62aec932022-01-29 21:45:34 +00002645def s:ReadRef(Ref: func(): list<string>): string
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02002646 return join(Ref(), ' ')
2647enddef
2648
Bram Moolenaar62aec932022-01-29 21:45:34 +00002649def s:ExtendRef(Ref: func(string): list<string>, add: string)
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02002650 Ref(add)
2651enddef
2652
2653def Test_closure_two_indirect_refs()
Bram Moolenaarf7779c62020-05-03 15:38:16 +02002654 MakeTwoRefs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002655 ReadRef(g:Read)->assert_equal('some')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02002656 ExtendRef(g:Extend, 'more')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002657 ReadRef(g:Read)->assert_equal('some more')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02002658 ExtendRef(g:Extend, 'even')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002659 ReadRef(g:Read)->assert_equal('some more even')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02002660
2661 unlet g:Extend
2662 unlet g:Read
2663enddef
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02002664
Bram Moolenaar62aec932022-01-29 21:45:34 +00002665def s:MakeArgRefs(theArg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002666 var local = 'loc_val'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002667 g:UseArg = (s) => theArg .. '/' .. local .. '/' .. s
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02002668enddef
2669
Bram Moolenaar62aec932022-01-29 21:45:34 +00002670def s:MakeArgRefsVarargs(theArg: string, ...rest: list<string>)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002671 var local = 'the_loc'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002672 g:UseVararg = (s) => theArg .. '/' .. local .. '/' .. s .. '/' .. join(rest)
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02002673enddef
2674
2675def Test_closure_using_argument()
2676 MakeArgRefs('arg_val')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002677 g:UseArg('call_val')->assert_equal('arg_val/loc_val/call_val')
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02002678
2679 MakeArgRefsVarargs('arg_val', 'one', 'two')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002680 g:UseVararg('call_val')->assert_equal('arg_val/the_loc/call_val/one two')
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02002681
2682 unlet g:UseArg
2683 unlet g:UseVararg
Bram Moolenaar44ec21c2021-02-12 21:50:57 +01002684
2685 var lines =<< trim END
2686 vim9script
2687 def Test(Fun: func(number): number): list<number>
2688 return map([1, 2, 3], (_, i) => Fun(i))
2689 enddef
2690 def Inc(nr: number): number
2691 return nr + 2
2692 enddef
2693 assert_equal([3, 4, 5], Test(Inc))
2694 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002695 v9.CheckScriptSuccess(lines)
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02002696enddef
2697
Bram Moolenaar62aec932022-01-29 21:45:34 +00002698def s:MakeGetAndAppendRefs()
Bram Moolenaar85d5e2b2020-10-10 14:13:01 +02002699 var local = 'a'
2700
2701 def Append(arg: string)
2702 local ..= arg
2703 enddef
2704 g:Append = Append
2705
2706 def Get(): string
2707 return local
2708 enddef
2709 g:Get = Get
2710enddef
2711
2712def Test_closure_append_get()
2713 MakeGetAndAppendRefs()
2714 g:Get()->assert_equal('a')
2715 g:Append('-b')
2716 g:Get()->assert_equal('a-b')
2717 g:Append('-c')
2718 g:Get()->assert_equal('a-b-c')
2719
2720 unlet g:Append
2721 unlet g:Get
2722enddef
Bram Moolenaarb68b3462020-05-06 21:06:30 +02002723
Bram Moolenaar04b12692020-05-04 23:24:44 +02002724def Test_nested_closure()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002725 var local = 'text'
Bram Moolenaar04b12692020-05-04 23:24:44 +02002726 def Closure(arg: string): string
2727 return local .. arg
2728 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002729 Closure('!!!')->assert_equal('text!!!')
Bram Moolenaar04b12692020-05-04 23:24:44 +02002730enddef
2731
Bram Moolenaar62aec932022-01-29 21:45:34 +00002732func s:GetResult(Ref)
Bram Moolenaar6f5b6df2020-05-16 21:20:12 +02002733 return a:Ref('some')
2734endfunc
2735
2736def Test_call_closure_not_compiled()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002737 var text = 'text'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002738 g:Ref = (s) => s .. text
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002739 GetResult(g:Ref)->assert_equal('sometext')
Bram Moolenaar6f5b6df2020-05-16 21:20:12 +02002740enddef
2741
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02002742def Test_double_closure_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002743 var lines =<< trim END
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02002744 vim9script
2745 def Func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002746 var name = 0
2747 for i in range(2)
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002748 timer_start(0, () => name)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002749 endfor
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02002750 enddef
2751 Func()
2752 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002753 v9.CheckScriptSuccess(lines)
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02002754enddef
2755
Bram Moolenaar85d5e2b2020-10-10 14:13:01 +02002756def Test_nested_closure_used()
2757 var lines =<< trim END
2758 vim9script
2759 def Func()
2760 var x = 'hello'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002761 var Closure = () => x
2762 g:Myclosure = () => Closure()
Bram Moolenaar85d5e2b2020-10-10 14:13:01 +02002763 enddef
2764 Func()
2765 assert_equal('hello', g:Myclosure())
2766 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002767 v9.CheckScriptSuccess(lines)
Bram Moolenaar85d5e2b2020-10-10 14:13:01 +02002768enddef
Bram Moolenaar0876c782020-10-07 19:08:04 +02002769
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02002770def Test_nested_closure_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002771 var lines =<< trim END
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02002772 vim9script
2773 def FuncA()
2774 FuncB(0)
2775 enddef
2776 def FuncB(n: number): list<string>
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002777 return map([0], (_, v) => n)
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02002778 enddef
2779 FuncA()
2780 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002781 v9.CheckScriptFailure(lines, 'E1012:')
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02002782enddef
2783
Bram Moolenaarf112f302020-12-20 17:47:52 +01002784def Test_global_closure()
2785 var lines =<< trim END
2786 vim9script
2787 def ReverseEveryNLines(n: number, line1: number, line2: number)
2788 var mods = 'sil keepj keepp lockm '
2789 var range = ':' .. line1 .. ',' .. line2
2790 def g:Offset(): number
2791 var offset = (line('.') - line1 + 1) % n
2792 return offset != 0 ? offset : n
2793 enddef
2794 exe mods .. range .. 'g/^/exe "m .-" .. g:Offset()'
2795 enddef
2796
2797 new
2798 repeat(['aaa', 'bbb', 'ccc'], 3)->setline(1)
2799 ReverseEveryNLines(3, 1, 9)
2800 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002801 v9.CheckScriptSuccess(lines)
Bram Moolenaarf112f302020-12-20 17:47:52 +01002802 var expected = repeat(['ccc', 'bbb', 'aaa'], 3)
2803 assert_equal(expected, getline(1, 9))
2804 bwipe!
2805enddef
2806
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01002807def Test_global_closure_called_directly()
2808 var lines =<< trim END
2809 vim9script
2810 def Outer()
2811 var x = 1
2812 def g:Inner()
2813 var y = x
2814 x += 1
2815 assert_equal(1, y)
2816 enddef
2817 g:Inner()
2818 assert_equal(2, x)
2819 enddef
2820 Outer()
2821 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002822 v9.CheckScriptSuccess(lines)
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01002823 delfunc g:Inner
2824enddef
2825
Bram Moolenaar69c76172021-12-02 16:38:52 +00002826def Test_closure_called_from_legacy()
2827 var lines =<< trim END
2828 vim9script
2829 def Func()
2830 var outer = 'foo'
2831 var F = () => {
2832 outer = 'bar'
2833 }
2834 execute printf('call %s()', string(F))
2835 enddef
2836 Func()
2837 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002838 v9.CheckScriptFailure(lines, 'E1248')
Bram Moolenaar69c76172021-12-02 16:38:52 +00002839enddef
2840
Bram Moolenaar34c54eb2020-11-25 19:15:19 +01002841def Test_failure_in_called_function()
2842 # this was using the frame index as the return value
2843 var lines =<< trim END
2844 vim9script
2845 au TerminalWinOpen * eval [][0]
2846 def PopupTerm(a: any)
2847 # make sure typvals on stack are string
2848 ['a', 'b', 'c', 'd', 'e', 'f', 'g']->join()
2849 FireEvent()
2850 enddef
2851 def FireEvent()
2852 do TerminalWinOpen
2853 enddef
2854 # use try/catch to make eval fail
2855 try
2856 call PopupTerm(0)
2857 catch
2858 endtry
2859 au! TerminalWinOpen
2860 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002861 v9.CheckScriptSuccess(lines)
Bram Moolenaar34c54eb2020-11-25 19:15:19 +01002862enddef
2863
Bram Moolenaar5366e1a2020-10-01 13:01:34 +02002864def Test_nested_lambda()
2865 var lines =<< trim END
2866 vim9script
2867 def Func()
2868 var x = 4
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002869 var Lambda1 = () => 7
2870 var Lambda2 = () => [Lambda1(), x]
Bram Moolenaar5366e1a2020-10-01 13:01:34 +02002871 var res = Lambda2()
2872 assert_equal([7, 4], res)
2873 enddef
2874 Func()
2875 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002876 v9.CheckScriptSuccess(lines)
Bram Moolenaar5366e1a2020-10-01 13:01:34 +02002877enddef
2878
Bram Moolenaarc04f2a42021-06-09 19:30:03 +02002879def Test_double_nested_lambda()
2880 var lines =<< trim END
2881 vim9script
2882 def F(head: string): func(string): func(string): string
2883 return (sep: string): func(string): string => ((tail: string): string => {
2884 return head .. sep .. tail
2885 })
2886 enddef
2887 assert_equal('hello-there', F('hello')('-')('there'))
2888 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002889 v9.CheckScriptSuccess(lines)
Bram Moolenaarc04f2a42021-06-09 19:30:03 +02002890enddef
2891
Bram Moolenaar074f84c2021-05-18 11:47:44 +02002892def Test_nested_inline_lambda()
Bram Moolenaar074f84c2021-05-18 11:47:44 +02002893 var lines =<< trim END
2894 vim9script
2895 def F(text: string): func(string): func(string): string
2896 return (arg: string): func(string): string => ((sep: string): string => {
Bram Moolenaar23e2e112021-08-03 21:16:18 +02002897 return sep .. arg .. text
Bram Moolenaar074f84c2021-05-18 11:47:44 +02002898 })
2899 enddef
Bram Moolenaar23e2e112021-08-03 21:16:18 +02002900 assert_equal('--there++', F('++')('there')('--'))
Bram Moolenaar074f84c2021-05-18 11:47:44 +02002901 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002902 v9.CheckScriptSuccess(lines)
Bram Moolenaar5245beb2021-07-15 22:03:50 +02002903
2904 lines =<< trim END
2905 vim9script
2906 echo range(4)->mapnew((_, v) => {
2907 return range(v) ->mapnew((_, s) => {
2908 return string(s)
2909 })
2910 })
2911 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002912 v9.CheckScriptSuccess(lines)
Bram Moolenaarc6ba2f92021-07-18 13:42:29 +02002913
2914 lines =<< trim END
2915 vim9script
2916
Bram Moolenaara749a422022-02-12 19:52:25 +00002917 def Func()
Bram Moolenaarc6ba2f92021-07-18 13:42:29 +02002918 range(10)
2919 ->mapnew((_, _) => ({
2920 key: range(10)->mapnew((_, _) => {
2921 return ' '
2922 }),
2923 }))
2924 enddef
2925
2926 defcomp
2927 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002928 v9.CheckScriptSuccess(lines)
Bram Moolenaar074f84c2021-05-18 11:47:44 +02002929enddef
2930
Bram Moolenaar52bf81c2020-11-17 18:50:44 +01002931def Shadowed(): list<number>
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002932 var FuncList: list<func: number> = [() => 42]
Bram Moolenaar75ab91f2021-01-10 22:42:50 +01002933 return FuncList->mapnew((_, Shadowed) => Shadowed())
Bram Moolenaar52bf81c2020-11-17 18:50:44 +01002934enddef
2935
2936def Test_lambda_arg_shadows_func()
Bram Moolenaar62aec932022-01-29 21:45:34 +00002937 assert_equal([42], g:Shadowed())
Bram Moolenaar52bf81c2020-11-17 18:50:44 +01002938enddef
2939
Bram Moolenaar62aec932022-01-29 21:45:34 +00002940def s:Line_continuation_in_def(dir: string = ''): string
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002941 var path: string = empty(dir)
2942 \ ? 'empty'
2943 \ : 'full'
2944 return path
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02002945enddef
2946
2947def Test_line_continuation_in_def()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002948 Line_continuation_in_def('.')->assert_equal('full')
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02002949enddef
2950
Bram Moolenaar2ea95b62020-11-19 21:47:56 +01002951def Test_script_var_in_lambda()
2952 var lines =<< trim END
2953 vim9script
2954 var script = 'test'
Bram Moolenaarbb8a7ce2021-04-10 20:10:26 +02002955 assert_equal(['test'], map(['one'], (_, _) => script))
Bram Moolenaar2ea95b62020-11-19 21:47:56 +01002956 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002957 v9.CheckScriptSuccess(lines)
Bram Moolenaar2ea95b62020-11-19 21:47:56 +01002958enddef
2959
Bram Moolenaar62aec932022-01-29 21:45:34 +00002960def s:Line_continuation_in_lambda(): list<string>
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002961 var x = range(97, 100)
Bram Moolenaar75ab91f2021-01-10 22:42:50 +01002962 ->mapnew((_, v) => nr2char(v)
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002963 ->toupper())
Bram Moolenaar7a4b8982020-07-08 17:36:21 +02002964 ->reverse()
2965 return x
2966enddef
2967
2968def Test_line_continuation_in_lambda()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002969 Line_continuation_in_lambda()->assert_equal(['D', 'C', 'B', 'A'])
Bram Moolenaarf898f7c2021-01-16 18:09:52 +01002970
2971 var lines =<< trim END
2972 vim9script
2973 var res = [{n: 1, m: 2, s: 'xxx'}]
2974 ->mapnew((_, v: dict<any>): string => printf('%d:%d:%s',
2975 v.n,
2976 v.m,
2977 substitute(v.s, '.*', 'yyy', '')
2978 ))
2979 assert_equal(['1:2:yyy'], res)
2980 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002981 v9.CheckScriptSuccess(lines)
Bram Moolenaar7a4b8982020-07-08 17:36:21 +02002982enddef
2983
Bram Moolenaarb6571982021-01-08 22:24:19 +01002984def Test_list_lambda()
2985 timer_start(1000, (_) => 0)
2986 var body = execute(timer_info()[0].callback
2987 ->string()
2988 ->substitute("('", ' ', '')
2989 ->substitute("')", '', '')
2990 ->substitute('function\zs', ' ', ''))
Bram Moolenaar767034c2021-04-09 17:24:52 +02002991 assert_match('def <lambda>\d\+(_: any): number\n1 return 0\n enddef', body)
Bram Moolenaarb6571982021-01-08 22:24:19 +01002992enddef
2993
Bram Moolenaar3c77b6a2021-07-25 18:07:00 +02002994def Test_lambda_block_variable()
Bram Moolenaar88421d62021-07-24 14:14:52 +02002995 var lines =<< trim END
2996 vim9script
2997 var flist: list<func>
2998 for i in range(10)
2999 var inloop = i
3000 flist[i] = () => inloop
3001 endfor
3002 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003003 v9.CheckScriptSuccess(lines)
Bram Moolenaar88421d62021-07-24 14:14:52 +02003004
3005 lines =<< trim END
3006 vim9script
3007 if true
3008 var outloop = 5
3009 var flist: list<func>
3010 for i in range(10)
3011 flist[i] = () => outloop
3012 endfor
3013 endif
3014 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003015 v9.CheckScriptSuccess(lines)
Bram Moolenaar88421d62021-07-24 14:14:52 +02003016
3017 lines =<< trim END
3018 vim9script
3019 if true
3020 var outloop = 5
3021 endif
3022 var flist: list<func>
3023 for i in range(10)
3024 flist[i] = () => outloop
3025 endfor
3026 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003027 v9.CheckScriptFailure(lines, 'E1001: Variable not found: outloop', 1)
Bram Moolenaar3c77b6a2021-07-25 18:07:00 +02003028
3029 lines =<< trim END
3030 vim9script
3031 for i in range(10)
3032 var Ref = () => 0
3033 endfor
3034 assert_equal(0, ((i) => 0)(0))
3035 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003036 v9.CheckScriptSuccess(lines)
Bram Moolenaar88421d62021-07-24 14:14:52 +02003037enddef
3038
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02003039def Test_legacy_lambda()
3040 legacy echo {x -> 'hello ' .. x}('foo')
Bram Moolenaardc4c2302021-04-25 13:54:42 +02003041
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02003042 var lines =<< trim END
3043 echo {x -> 'hello ' .. x}('foo')
3044 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003045 v9.CheckDefAndScriptFailure(lines, 'E720:')
Bram Moolenaardc4c2302021-04-25 13:54:42 +02003046
3047 lines =<< trim END
3048 vim9script
3049 def Func()
3050 echo (() => 'no error')()
3051 enddef
3052 legacy call s:Func()
3053 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003054 v9.CheckScriptSuccess(lines)
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02003055enddef
3056
Bram Moolenaarce024c32021-06-26 13:00:49 +02003057def Test_legacy()
3058 var lines =<< trim END
3059 vim9script
3060 func g:LegacyFunction()
3061 let g:legacyvar = 1
3062 endfunc
3063 def Testit()
3064 legacy call g:LegacyFunction()
3065 enddef
3066 Testit()
3067 assert_equal(1, g:legacyvar)
3068 unlet g:legacyvar
3069 delfunc g:LegacyFunction
3070 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003071 v9.CheckScriptSuccess(lines)
Bram Moolenaarce024c32021-06-26 13:00:49 +02003072enddef
3073
Bram Moolenaarc3cb1c92021-06-02 16:47:53 +02003074def Test_legacy_errors()
3075 for cmd in ['if', 'elseif', 'else', 'endif',
3076 'for', 'endfor', 'continue', 'break',
3077 'while', 'endwhile',
3078 'try', 'catch', 'finally', 'endtry']
Bram Moolenaar62aec932022-01-29 21:45:34 +00003079 v9.CheckDefFailure(['legacy ' .. cmd .. ' expr'], 'E1189:')
Bram Moolenaarc3cb1c92021-06-02 16:47:53 +02003080 endfor
3081enddef
3082
Bram Moolenaarb1b6f4d2021-09-13 18:25:54 +02003083def Test_call_legacy_with_dict()
3084 var lines =<< trim END
3085 vim9script
3086 func Legacy() dict
3087 let g:result = self.value
3088 endfunc
3089 def TestDirect()
3090 var d = {value: 'yes', func: Legacy}
3091 d.func()
3092 enddef
3093 TestDirect()
3094 assert_equal('yes', g:result)
3095 unlet g:result
3096
3097 def TestIndirect()
3098 var d = {value: 'foo', func: Legacy}
3099 var Fi = d.func
3100 Fi()
3101 enddef
3102 TestIndirect()
3103 assert_equal('foo', g:result)
3104 unlet g:result
3105
3106 var d = {value: 'bar', func: Legacy}
3107 d.func()
3108 assert_equal('bar', g:result)
3109 unlet g:result
3110 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003111 v9.CheckScriptSuccess(lines)
Bram Moolenaarb1b6f4d2021-09-13 18:25:54 +02003112enddef
3113
Bram Moolenaar62aec932022-01-29 21:45:34 +00003114def s:DoFilterThis(a: string): list<string>
Bram Moolenaarab360522021-01-10 14:02:28 +01003115 # closure nested inside another closure using argument
3116 var Filter = (l) => filter(l, (_, v) => stridx(v, a) == 0)
3117 return ['x', 'y', 'a', 'x2', 'c']->Filter()
3118enddef
3119
3120def Test_nested_closure_using_argument()
3121 assert_equal(['x', 'x2'], DoFilterThis('x'))
3122enddef
3123
Bram Moolenaar0186e582021-01-10 18:33:11 +01003124def Test_triple_nested_closure()
3125 var what = 'x'
3126 var Match = (val: string, cmp: string): bool => stridx(val, cmp) == 0
3127 var Filter = (l) => filter(l, (_, v) => Match(v, what))
3128 assert_equal(['x', 'x2'], ['x', 'y', 'a', 'x2', 'c']->Filter())
3129enddef
3130
Bram Moolenaar8f510af2020-07-05 18:48:23 +02003131func Test_silent_echo()
Bram Moolenaar47e7d702020-07-05 18:18:42 +02003132 CheckScreendump
Bram Moolenaar3b309f12021-12-13 18:19:55 +00003133 call Run_Test_silent_echo()
3134endfunc
Bram Moolenaar47e7d702020-07-05 18:18:42 +02003135
Bram Moolenaar3b309f12021-12-13 18:19:55 +00003136def Run_Test_silent_echo()
3137 var lines =<< trim END
Bram Moolenaar47e7d702020-07-05 18:18:42 +02003138 vim9script
3139 def EchoNothing()
3140 silent echo ''
3141 enddef
3142 defcompile
3143 END
Bram Moolenaar3b309f12021-12-13 18:19:55 +00003144 writefile(lines, 'XTest_silent_echo')
Bram Moolenaar47e7d702020-07-05 18:18:42 +02003145
Bram Moolenaar3b309f12021-12-13 18:19:55 +00003146 # Check that the balloon shows up after a mouse move
Bram Moolenaar62aec932022-01-29 21:45:34 +00003147 var buf = g:RunVimInTerminal('-S XTest_silent_echo', {'rows': 6})
Bram Moolenaar3b309f12021-12-13 18:19:55 +00003148 term_sendkeys(buf, ":abc")
Bram Moolenaar62aec932022-01-29 21:45:34 +00003149 g:VerifyScreenDump(buf, 'Test_vim9_silent_echo', {})
Bram Moolenaar47e7d702020-07-05 18:18:42 +02003150
Bram Moolenaar3b309f12021-12-13 18:19:55 +00003151 # clean up
Bram Moolenaar62aec932022-01-29 21:45:34 +00003152 g:StopVimInTerminal(buf)
Bram Moolenaar3b309f12021-12-13 18:19:55 +00003153 delete('XTest_silent_echo')
3154enddef
Bram Moolenaar47e7d702020-07-05 18:18:42 +02003155
Bram Moolenaar171fb922020-10-28 16:54:47 +01003156def SilentlyError()
3157 execute('silent! invalid')
3158 g:did_it = 'yes'
3159enddef
3160
Bram Moolenaar62aec932022-01-29 21:45:34 +00003161func s:UserError()
Bram Moolenaar28ee8922020-10-28 20:20:00 +01003162 silent! invalid
3163endfunc
3164
3165def SilentlyUserError()
3166 UserError()
3167 g:did_it = 'yes'
3168enddef
Bram Moolenaar171fb922020-10-28 16:54:47 +01003169
3170" This can't be a :def function, because the assert would not be reached.
Bram Moolenaar171fb922020-10-28 16:54:47 +01003171func Test_ignore_silent_error()
3172 let g:did_it = 'no'
3173 call SilentlyError()
3174 call assert_equal('yes', g:did_it)
3175
Bram Moolenaar28ee8922020-10-28 20:20:00 +01003176 let g:did_it = 'no'
3177 call SilentlyUserError()
3178 call assert_equal('yes', g:did_it)
Bram Moolenaar171fb922020-10-28 16:54:47 +01003179
3180 unlet g:did_it
3181endfunc
3182
Bram Moolenaarcd030c42020-10-30 21:49:40 +01003183def Test_ignore_silent_error_in_filter()
3184 var lines =<< trim END
3185 vim9script
3186 def Filter(winid: number, key: string): bool
3187 if key == 'o'
3188 silent! eval [][0]
3189 return true
3190 endif
3191 return popup_filter_menu(winid, key)
3192 enddef
3193
Bram Moolenaare0de1712020-12-02 17:36:54 +01003194 popup_create('popup', {filter: Filter})
Bram Moolenaarcd030c42020-10-30 21:49:40 +01003195 feedkeys("o\r", 'xnt')
3196 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003197 v9.CheckScriptSuccess(lines)
Bram Moolenaarcd030c42020-10-30 21:49:40 +01003198enddef
3199
Bram Moolenaar62aec932022-01-29 21:45:34 +00003200def s:Fibonacci(n: number): number
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02003201 if n < 2
3202 return n
3203 else
3204 return Fibonacci(n - 1) + Fibonacci(n - 2)
3205 endif
3206enddef
3207
Bram Moolenaar985116a2020-07-12 17:31:09 +02003208def Test_recursive_call()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02003209 Fibonacci(20)->assert_equal(6765)
Bram Moolenaar985116a2020-07-12 17:31:09 +02003210enddef
3211
Bram Moolenaar62aec932022-01-29 21:45:34 +00003212def s:TreeWalk(dir: string): list<any>
Bram Moolenaar75ab91f2021-01-10 22:42:50 +01003213 return readdir(dir)->mapnew((_, val) =>
Bram Moolenaar08f7a412020-07-13 20:41:08 +02003214 fnamemodify(dir .. '/' .. val, ':p')->isdirectory()
Bram Moolenaar2bede172020-11-19 18:53:18 +01003215 ? {[val]: TreeWalk(dir .. '/' .. val)}
Bram Moolenaar08f7a412020-07-13 20:41:08 +02003216 : val
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01003217 )
Bram Moolenaar08f7a412020-07-13 20:41:08 +02003218enddef
3219
3220def Test_closure_in_map()
3221 mkdir('XclosureDir/tdir', 'p')
3222 writefile(['111'], 'XclosureDir/file1')
3223 writefile(['222'], 'XclosureDir/file2')
3224 writefile(['333'], 'XclosureDir/tdir/file3')
3225
Bram Moolenaare0de1712020-12-02 17:36:54 +01003226 TreeWalk('XclosureDir')->assert_equal(['file1', 'file2', {tdir: ['file3']}])
Bram Moolenaar08f7a412020-07-13 20:41:08 +02003227
3228 delete('XclosureDir', 'rf')
3229enddef
3230
Bram Moolenaar7b5d5442020-10-04 13:42:34 +02003231def Test_invalid_function_name()
3232 var lines =<< trim END
3233 vim9script
3234 def s: list<string>
3235 END
Bram Moolenaara749a422022-02-12 19:52:25 +00003236 v9.CheckScriptFailure(lines, 'E1268:')
Bram Moolenaar7b5d5442020-10-04 13:42:34 +02003237
3238 lines =<< trim END
3239 vim9script
3240 def g: list<string>
3241 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003242 v9.CheckScriptFailure(lines, 'E129:')
Bram Moolenaar7b5d5442020-10-04 13:42:34 +02003243
3244 lines =<< trim END
3245 vim9script
3246 def <SID>: list<string>
3247 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003248 v9.CheckScriptFailure(lines, 'E884:')
Bram Moolenaar7b5d5442020-10-04 13:42:34 +02003249
3250 lines =<< trim END
3251 vim9script
3252 def F list<string>
3253 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003254 v9.CheckScriptFailure(lines, 'E488:')
Bram Moolenaar7b5d5442020-10-04 13:42:34 +02003255enddef
3256
Bram Moolenaara90afb92020-07-15 22:38:56 +02003257def Test_partial_call()
Bram Moolenaarf78da4f2021-08-01 15:40:31 +02003258 var lines =<< trim END
3259 var Xsetlist: func
3260 Xsetlist = function('setloclist', [0])
3261 Xsetlist([], ' ', {title: 'test'})
3262 getloclist(0, {title: 1})->assert_equal({title: 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02003263
Bram Moolenaarf78da4f2021-08-01 15:40:31 +02003264 Xsetlist = function('setloclist', [0, [], ' '])
3265 Xsetlist({title: 'test'})
3266 getloclist(0, {title: 1})->assert_equal({title: 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02003267
Bram Moolenaarf78da4f2021-08-01 15:40:31 +02003268 Xsetlist = function('setqflist')
3269 Xsetlist([], ' ', {title: 'test'})
3270 getqflist({title: 1})->assert_equal({title: 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02003271
Bram Moolenaarf78da4f2021-08-01 15:40:31 +02003272 Xsetlist = function('setqflist', [[], ' '])
3273 Xsetlist({title: 'test'})
3274 getqflist({title: 1})->assert_equal({title: 'test'})
Bram Moolenaar6abd3dc2020-10-04 14:17:32 +02003275
Bram Moolenaarf78da4f2021-08-01 15:40:31 +02003276 var Len: func: number = function('len', ['word'])
3277 assert_equal(4, Len())
3278
3279 var RepeatFunc = function('repeat', ['o'])
3280 assert_equal('ooooo', RepeatFunc(5))
3281 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003282 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaarc66f6452021-08-19 21:08:30 +02003283
3284 lines =<< trim END
3285 vim9script
3286 def Foo(Parser: any)
3287 enddef
3288 var Expr: func(dict<any>): dict<any>
3289 const Call = Foo(Expr)
3290 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003291 v9.CheckScriptFailure(lines, 'E1235:')
Bram Moolenaara90afb92020-07-15 22:38:56 +02003292enddef
3293
Bram Moolenaarcd1cda22022-02-16 21:48:25 +00003294def Test_partial_double_nested()
3295 var idx = 123
3296 var Get = () => idx
3297 var Ref = function(Get, [])
3298 var RefRef = function(Ref, [])
3299 assert_equal(123, RefRef())
3300enddef
3301
Bram Moolenaarfe1bfc92022-02-06 13:55:03 +00003302" Using "idx" from a legacy global function does not work.
3303" This caused a crash when called from legacy context.
3304func Test_partial_call_fails()
3305 let lines =<< trim END
3306 vim9script
3307
3308 var l = ['a', 'b', 'c']
3309 def Iter(container: any): any
3310 var idx = -1
3311 var obj = {state: container}
Bram Moolenaarf681cfb2022-02-07 20:30:57 +00003312 def g:NextItem__(self: dict<any>): any
Bram Moolenaarfe1bfc92022-02-06 13:55:03 +00003313 ++idx
3314 return self.state[idx]
3315 enddef
Bram Moolenaarf681cfb2022-02-07 20:30:57 +00003316 obj.__next__ = function('g:NextItem__', [obj])
Bram Moolenaarfe1bfc92022-02-06 13:55:03 +00003317 return obj
3318 enddef
3319
3320 var it = Iter(l)
3321 echo it.__next__()
3322 END
3323 call writefile(lines, 'XpartialCall')
3324 try
3325 source XpartialCall
3326 catch /E1248:/
3327 endtry
3328 call delete('XpartialCall')
3329endfunc
3330
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02003331def Test_cmd_modifier()
3332 tab echo '0'
Bram Moolenaar62aec932022-01-29 21:45:34 +00003333 v9.CheckDefFailure(['5tab echo 3'], 'E16:')
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02003334enddef
3335
3336def Test_restore_modifiers()
3337 # check that when compiling a :def function command modifiers are not messed
3338 # up.
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02003339 var lines =<< trim END
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02003340 vim9script
3341 set eventignore=
3342 autocmd QuickFixCmdPost * copen
3343 def AutocmdsDisabled()
Bram Moolenaarc3235272021-07-10 19:42:03 +02003344 eval 1 + 2
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02003345 enddef
3346 func Func()
3347 noautocmd call s:AutocmdsDisabled()
3348 let g:ei_after = &eventignore
3349 endfunc
3350 Func()
3351 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003352 v9.CheckScriptSuccess(lines)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02003353 g:ei_after->assert_equal('')
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02003354enddef
3355
Bram Moolenaardfa3d552020-09-10 22:05:08 +02003356def StackTop()
Bram Moolenaarc3235272021-07-10 19:42:03 +02003357 eval 1 + 2
3358 eval 2 + 3
Bram Moolenaardfa3d552020-09-10 22:05:08 +02003359 # call not on fourth line
Bram Moolenaar62aec932022-01-29 21:45:34 +00003360 g:StackBot()
Bram Moolenaardfa3d552020-09-10 22:05:08 +02003361enddef
3362
3363def StackBot()
3364 # throw an error
3365 eval [][0]
3366enddef
3367
3368def Test_callstack_def()
3369 try
Bram Moolenaar62aec932022-01-29 21:45:34 +00003370 g:StackTop()
Bram Moolenaardfa3d552020-09-10 22:05:08 +02003371 catch
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02003372 v:throwpoint->assert_match('Test_callstack_def\[2\]..StackTop\[4\]..StackBot, line 2')
Bram Moolenaardfa3d552020-09-10 22:05:08 +02003373 endtry
3374enddef
3375
Bram Moolenaare8211a32020-10-09 22:04:29 +02003376" Re-using spot for variable used in block
3377def Test_block_scoped_var()
3378 var lines =<< trim END
3379 vim9script
3380 def Func()
3381 var x = ['a', 'b', 'c']
3382 if 1
3383 var y = 'x'
Bram Moolenaarbb8a7ce2021-04-10 20:10:26 +02003384 map(x, (_, _) => y)
Bram Moolenaare8211a32020-10-09 22:04:29 +02003385 endif
3386 var z = x
3387 assert_equal(['x', 'x', 'x'], z)
3388 enddef
3389 Func()
3390 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003391 v9.CheckScriptSuccess(lines)
Bram Moolenaare8211a32020-10-09 22:04:29 +02003392enddef
3393
Bram Moolenaareeece9e2020-11-20 19:26:48 +01003394def Test_reset_did_emsg()
3395 var lines =<< trim END
3396 @s = 'blah'
3397 au BufWinLeave * #
3398 def Func()
3399 var winid = popup_create('popup', {})
3400 exe '*s'
3401 popup_close(winid)
3402 enddef
3403 Func()
3404 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003405 v9.CheckScriptFailure(lines, 'E492:', 8)
Bram Moolenaar2d870f82020-12-05 13:41:01 +01003406 delfunc! g:Func
Bram Moolenaareeece9e2020-11-20 19:26:48 +01003407enddef
3408
Bram Moolenaar57f799e2020-12-12 20:42:19 +01003409def Test_did_emsg_reset()
3410 # executing an autocommand resets did_emsg, this should not result in a
3411 # builtin function considered failing
3412 var lines =<< trim END
3413 vim9script
3414 au BufWinLeave * #
3415 def Func()
Bram Moolenaar767034c2021-04-09 17:24:52 +02003416 popup_menu('', {callback: (a, b) => popup_create('', {})->popup_close()})
Bram Moolenaar57f799e2020-12-12 20:42:19 +01003417 eval [][0]
3418 enddef
3419 nno <F3> <cmd>call <sid>Func()<cr>
3420 feedkeys("\<F3>\e", 'xt')
3421 END
3422 writefile(lines, 'XemsgReset')
3423 assert_fails('so XemsgReset', ['E684:', 'E684:'], lines, 2)
3424 delete('XemsgReset')
3425 nunmap <F3>
3426 au! BufWinLeave
3427enddef
3428
Bram Moolenaar56602ba2020-12-05 21:22:08 +01003429def Test_abort_with_silent_call()
3430 var lines =<< trim END
3431 vim9script
3432 g:result = 'none'
3433 def Func()
3434 g:result += 3
3435 g:result = 'yes'
3436 enddef
3437 # error is silenced, but function aborts on error
3438 silent! Func()
3439 assert_equal('none', g:result)
3440 unlet g:result
3441 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003442 v9.CheckScriptSuccess(lines)
Bram Moolenaar56602ba2020-12-05 21:22:08 +01003443enddef
3444
Bram Moolenaarf665e972020-12-05 19:17:16 +01003445def Test_continues_with_silent_error()
3446 var lines =<< trim END
3447 vim9script
3448 g:result = 'none'
3449 def Func()
3450 silent! g:result += 3
3451 g:result = 'yes'
3452 enddef
3453 # error is silenced, function does not abort
3454 Func()
3455 assert_equal('yes', g:result)
3456 unlet g:result
3457 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003458 v9.CheckScriptSuccess(lines)
Bram Moolenaarf665e972020-12-05 19:17:16 +01003459enddef
3460
Bram Moolenaaraf0df472020-12-02 20:51:22 +01003461def Test_abort_even_with_silent()
3462 var lines =<< trim END
3463 vim9script
3464 g:result = 'none'
3465 def Func()
3466 eval {-> ''}() .. '' .. {}['X']
3467 g:result = 'yes'
3468 enddef
Bram Moolenaarf665e972020-12-05 19:17:16 +01003469 silent! Func()
Bram Moolenaaraf0df472020-12-02 20:51:22 +01003470 assert_equal('none', g:result)
Bram Moolenaar4029cab2020-12-05 18:13:27 +01003471 unlet g:result
3472 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003473 v9.CheckScriptSuccess(lines)
Bram Moolenaar4029cab2020-12-05 18:13:27 +01003474enddef
3475
Bram Moolenaarf665e972020-12-05 19:17:16 +01003476def Test_cmdmod_silent_restored()
3477 var lines =<< trim END
3478 vim9script
3479 def Func()
3480 g:result = 'none'
3481 silent! g:result += 3
3482 g:result = 'none'
3483 g:result += 3
3484 enddef
3485 Func()
3486 END
3487 # can't use CheckScriptFailure, it ignores the :silent!
3488 var fname = 'Xdefsilent'
3489 writefile(lines, fname)
3490 var caught = 'no'
3491 try
3492 exe 'source ' .. fname
3493 catch /E1030:/
3494 caught = 'yes'
3495 assert_match('Func, line 4', v:throwpoint)
3496 endtry
3497 assert_equal('yes', caught)
3498 delete(fname)
3499enddef
3500
Bram Moolenaar2fecb532021-03-24 22:00:56 +01003501def Test_cmdmod_silent_nested()
3502 var lines =<< trim END
3503 vim9script
3504 var result = ''
3505
3506 def Error()
3507 result ..= 'Eb'
3508 eval [][0]
3509 result ..= 'Ea'
3510 enddef
3511
3512 def Crash()
3513 result ..= 'Cb'
3514 sil! Error()
3515 result ..= 'Ca'
3516 enddef
3517
3518 Crash()
3519 assert_equal('CbEbEaCa', result)
3520 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003521 v9.CheckScriptSuccess(lines)
Bram Moolenaar2fecb532021-03-24 22:00:56 +01003522enddef
3523
Bram Moolenaar4029cab2020-12-05 18:13:27 +01003524def Test_dict_member_with_silent()
3525 var lines =<< trim END
3526 vim9script
3527 g:result = 'none'
3528 var d: dict<any>
3529 def Func()
3530 try
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01003531 g:result = map([], (_, v) => ({}[v]))->join() .. d['']
Bram Moolenaar4029cab2020-12-05 18:13:27 +01003532 catch
3533 endtry
3534 enddef
3535 silent! Func()
3536 assert_equal('0', g:result)
3537 unlet g:result
Bram Moolenaaraf0df472020-12-02 20:51:22 +01003538 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003539 v9.CheckScriptSuccess(lines)
Bram Moolenaaraf0df472020-12-02 20:51:22 +01003540enddef
3541
Bram Moolenaarf9041332021-01-21 19:41:16 +01003542def Test_skip_cmds_with_silent()
3543 var lines =<< trim END
3544 vim9script
3545
3546 def Func(b: bool)
3547 Crash()
3548 enddef
3549
3550 def Crash()
3551 sil! :/not found/d _
3552 sil! :/not found/put _
3553 enddef
3554
3555 Func(true)
3556 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003557 v9.CheckScriptSuccess(lines)
Bram Moolenaarf9041332021-01-21 19:41:16 +01003558enddef
3559
Bram Moolenaar5b3d1bb2020-12-22 12:20:08 +01003560def Test_opfunc()
Bram Moolenaar848fadd2022-01-30 15:28:30 +00003561 nnoremap <F3> <cmd>set opfunc=g:Opfunc<cr>g@
Bram Moolenaar5b3d1bb2020-12-22 12:20:08 +01003562 def g:Opfunc(_: any): string
3563 setline(1, 'ASDF')
3564 return ''
3565 enddef
3566 new
3567 setline(1, 'asdf')
3568 feedkeys("\<F3>$", 'x')
3569 assert_equal('ASDF', getline(1))
3570
3571 bwipe!
3572 nunmap <F3>
3573enddef
3574
Bram Moolenaar3b309f12021-12-13 18:19:55 +00003575func Test_opfunc_error()
3576 CheckScreendump
3577 call Run_Test_opfunc_error()
3578endfunc
3579
3580def Run_Test_opfunc_error()
3581 # test that the error from Opfunc() is displayed right away
3582 var lines =<< trim END
3583 vim9script
3584
3585 def Opfunc(type: string)
3586 try
3587 eval [][0]
3588 catch /nothing/ # error not caught
3589 endtry
3590 enddef
3591 &operatorfunc = Opfunc
3592 nnoremap <expr> l <SID>L()
3593 def L(): string
3594 return 'l'
3595 enddef
3596 'x'->repeat(10)->setline(1)
3597 feedkeys('g@l', 'n')
3598 feedkeys('llll')
3599 END
3600 call writefile(lines, 'XTest_opfunc_error')
3601
Bram Moolenaar62aec932022-01-29 21:45:34 +00003602 var buf = g:RunVimInTerminal('-S XTest_opfunc_error', {rows: 6, wait_for_ruler: 0})
3603 g:WaitForAssert(() => assert_match('Press ENTER', term_getline(buf, 6)))
3604 g:WaitForAssert(() => assert_match('E684: list index out of range: 0', term_getline(buf, 5)))
Bram Moolenaar3b309f12021-12-13 18:19:55 +00003605
3606 # clean up
Bram Moolenaar62aec932022-01-29 21:45:34 +00003607 g:StopVimInTerminal(buf)
Bram Moolenaar3b309f12021-12-13 18:19:55 +00003608 delete('XTest_opfunc_error')
3609enddef
3610
Bram Moolenaar077a4232020-12-22 18:33:27 +01003611" this was crashing on exit
3612def Test_nested_lambda_in_closure()
3613 var lines =<< trim END
3614 vim9script
Bram Moolenaar227c58a2021-04-28 20:40:44 +02003615 command WriteDone writefile(['Done'], 'XnestedDone')
Bram Moolenaar077a4232020-12-22 18:33:27 +01003616 def Outer()
3617 def g:Inner()
3618 echo map([1, 2, 3], {_, v -> v + 1})
3619 enddef
3620 g:Inner()
3621 enddef
3622 defcompile
Bram Moolenaar227c58a2021-04-28 20:40:44 +02003623 # not reached
Bram Moolenaar077a4232020-12-22 18:33:27 +01003624 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003625 if !g:RunVim([], lines, '--clean -c WriteDone -c quit')
Bram Moolenaar077a4232020-12-22 18:33:27 +01003626 return
3627 endif
3628 assert_equal(['Done'], readfile('XnestedDone'))
3629 delete('XnestedDone')
3630enddef
3631
Bram Moolenaar92368aa2022-02-07 17:50:39 +00003632def Test_nested_closure_funcref()
3633 var lines =<< trim END
3634 vim9script
3635 def Func()
3636 var n: number
3637 def Nested()
3638 ++n
3639 enddef
3640 Nested()
3641 g:result_one = n
3642 var Ref = function(Nested)
3643 Ref()
3644 g:result_two = n
3645 enddef
3646 Func()
3647 END
3648 v9.CheckScriptSuccess(lines)
3649 assert_equal(1, g:result_one)
3650 assert_equal(2, g:result_two)
3651 unlet g:result_one g:result_two
3652enddef
3653
Bram Moolenaar7aca5ca2022-02-07 19:56:43 +00003654def Test_nested_closure_in_dict()
3655 var lines =<< trim END
3656 vim9script
3657 def Func(): dict<any>
3658 var n: number
3659 def Inc(): number
3660 ++n
3661 return n
3662 enddef
3663 return {inc: function(Inc)}
3664 enddef
3665 disas Func
3666 var d = Func()
3667 assert_equal(1, d.inc())
3668 assert_equal(2, d.inc())
3669 END
3670 v9.CheckScriptSuccess(lines)
3671enddef
3672
Bram Moolenaar04947cc2021-03-06 19:26:46 +01003673def Test_check_func_arg_types()
3674 var lines =<< trim END
3675 vim9script
3676 def F1(x: string): string
3677 return x
3678 enddef
3679
3680 def F2(x: number): number
3681 return x + 1
3682 enddef
3683
3684 def G(g: func): dict<func>
3685 return {f: g}
3686 enddef
3687
3688 def H(d: dict<func>): string
3689 return d.f('a')
3690 enddef
3691 END
3692
Bram Moolenaar62aec932022-01-29 21:45:34 +00003693 v9.CheckScriptSuccess(lines + ['echo H(G(F1))'])
3694 v9.CheckScriptFailure(lines + ['echo H(G(F2))'], 'E1013:')
Bram Moolenaar04947cc2021-03-06 19:26:46 +01003695enddef
3696
Bram Moolenaar6e48b842021-08-10 22:52:02 +02003697def Test_list_any_type_checked()
3698 var lines =<< trim END
3699 vim9script
3700 def Foo()
3701 --decl--
3702 Bar(l)
3703 enddef
3704 def Bar(ll: list<dict<any>>)
3705 enddef
3706 Foo()
3707 END
Bram Moolenaar2d3ac2e2022-02-03 12:34:05 +00003708 # "any" could be "dict<any>", thus OK
Bram Moolenaar6e48b842021-08-10 22:52:02 +02003709 lines[2] = 'var l: list<any>'
Bram Moolenaar2d3ac2e2022-02-03 12:34:05 +00003710 v9.CheckScriptSuccess(lines)
Bram Moolenaar6e48b842021-08-10 22:52:02 +02003711 lines[2] = 'var l: list<any> = []'
Bram Moolenaar2d3ac2e2022-02-03 12:34:05 +00003712 v9.CheckScriptSuccess(lines)
Bram Moolenaar6e48b842021-08-10 22:52:02 +02003713
3714 lines[2] = 'var l: list<any> = [11]'
Bram Moolenaar62aec932022-01-29 21:45:34 +00003715 v9.CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected list<dict<any>> but got list<number>', 2)
Bram Moolenaar6e48b842021-08-10 22:52:02 +02003716enddef
3717
Bram Moolenaar701cc6c2021-04-10 13:33:48 +02003718def Test_compile_error()
3719 var lines =<< trim END
3720 def g:Broken()
3721 echo 'a' + {}
3722 enddef
3723 call g:Broken()
3724 END
3725 # First call: compilation error
Bram Moolenaar62aec932022-01-29 21:45:34 +00003726 v9.CheckScriptFailure(lines, 'E1051: Wrong argument type for +')
Bram Moolenaar701cc6c2021-04-10 13:33:48 +02003727
3728 # Second call won't try compiling again
3729 assert_fails('call g:Broken()', 'E1091: Function is not compiled: Broken')
Bram Moolenaar599410c2021-04-10 14:03:43 +02003730 delfunc g:Broken
3731
3732 # No error when compiling with :silent!
3733 lines =<< trim END
3734 def g:Broken()
3735 echo 'a' + []
3736 enddef
3737 silent! defcompile
3738 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003739 v9.CheckScriptSuccess(lines)
Bram Moolenaar599410c2021-04-10 14:03:43 +02003740
3741 # Calling the function won't try compiling again
3742 assert_fails('call g:Broken()', 'E1091: Function is not compiled: Broken')
3743 delfunc g:Broken
Bram Moolenaar701cc6c2021-04-10 13:33:48 +02003744enddef
3745
Bram Moolenaar962c43b2021-04-10 17:18:09 +02003746def Test_ignored_argument()
3747 var lines =<< trim END
3748 vim9script
3749 def Ignore(_, _): string
3750 return 'yes'
3751 enddef
3752 assert_equal('yes', Ignore(1, 2))
3753
3754 func Ok(_)
3755 return a:_
3756 endfunc
3757 assert_equal('ok', Ok('ok'))
3758
3759 func Oktoo()
3760 let _ = 'too'
3761 return _
3762 endfunc
3763 assert_equal('too', Oktoo())
Bram Moolenaarda479c72021-04-10 21:01:38 +02003764
3765 assert_equal([[1], [2], [3]], range(3)->mapnew((_, v) => [v]->map((_, w) => w + 1)))
Bram Moolenaar962c43b2021-04-10 17:18:09 +02003766 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003767 v9.CheckScriptSuccess(lines)
Bram Moolenaar962c43b2021-04-10 17:18:09 +02003768
3769 lines =<< trim END
3770 def Ignore(_: string): string
3771 return _
3772 enddef
3773 defcompile
3774 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003775 v9.CheckScriptFailure(lines, 'E1181:', 1)
Bram Moolenaar962c43b2021-04-10 17:18:09 +02003776
3777 lines =<< trim END
3778 var _ = 1
3779 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003780 v9.CheckDefAndScriptFailure(lines, 'E1181:', 1)
Yegappan Lakshmanan34fcb692021-05-25 20:14:00 +02003781
3782 lines =<< trim END
3783 var x = _
3784 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003785 v9.CheckDefAndScriptFailure(lines, 'E1181:', 1)
Bram Moolenaar962c43b2021-04-10 17:18:09 +02003786enddef
3787
Bram Moolenaarbb8a7ce2021-04-10 20:10:26 +02003788def Test_too_many_arguments()
3789 var lines =<< trim END
3790 echo [0, 1, 2]->map(() => 123)
3791 END
Bram Moolenaareddd4fc2022-02-20 15:52:28 +00003792 v9.CheckDefAndScriptFailure(lines, ['E176:', 'E1106: 2 arguments too many'], 1)
Bram Moolenaarbb8a7ce2021-04-10 20:10:26 +02003793
3794 lines =<< trim END
3795 echo [0, 1, 2]->map((_) => 123)
3796 END
Bram Moolenaareddd4fc2022-02-20 15:52:28 +00003797 v9.CheckDefAndScriptFailure(lines, ['E176', 'E1106: One argument too many'], 1)
Bram Moolenaarbb8a7ce2021-04-10 20:10:26 +02003798enddef
Bram Moolenaar077a4232020-12-22 18:33:27 +01003799
Bram Moolenaara6aa1642021-04-23 19:32:23 +02003800def Test_closing_brace_at_start_of_line()
3801 var lines =<< trim END
3802 def Func()
3803 enddef
3804 Func(
3805 )
3806 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003807 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaara6aa1642021-04-23 19:32:23 +02003808enddef
3809
Bram Moolenaar62aec932022-01-29 21:45:34 +00003810func s:CreateMydict()
Bram Moolenaarb033ee22021-08-15 16:08:36 +02003811 let g:mydict = {}
3812 func g:mydict.afunc()
3813 let g:result = self.key
3814 endfunc
3815endfunc
3816
3817def Test_numbered_function_reference()
3818 CreateMydict()
3819 var output = execute('legacy func g:mydict.afunc')
3820 var funcName = 'g:' .. substitute(output, '.*function \(\d\+\).*', '\1', '')
3821 execute 'function(' .. funcName .. ', [], {key: 42})()'
3822 # check that the function still exists
3823 assert_equal(output, execute('legacy func g:mydict.afunc'))
3824 unlet g:mydict
3825enddef
3826
Bram Moolenaard3a11782022-01-05 16:50:40 +00003827def Test_go_beyond_end_of_cmd()
3828 # this was reading the byte after the end of the line
3829 var lines =<< trim END
3830 def F()
3831 cal
3832 enddef
3833 defcompile
3834 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003835 v9.CheckScriptFailure(lines, 'E476:')
Bram Moolenaard3a11782022-01-05 16:50:40 +00003836enddef
3837
Bram Moolenaar8b716f52022-02-15 21:17:56 +00003838" The following messes up syntax highlight, keep near the end.
Bram Moolenaar20677332021-06-06 17:02:53 +02003839if has('python3')
Bram Moolenaar8b716f52022-02-15 21:17:56 +00003840 def Test_python3_command()
3841 py3 import vim
Bram Moolenaarf5288c52022-02-15 21:33:29 +00003842 py3 vim.command("g:done = 'yes'")
Bram Moolenaar8b716f52022-02-15 21:17:56 +00003843 assert_equal('yes', g:done)
3844 unlet g:done
3845 enddef
3846
Bram Moolenaar20677332021-06-06 17:02:53 +02003847 def Test_python3_heredoc()
3848 py3 << trim EOF
3849 import vim
3850 vim.vars['didit'] = 'yes'
3851 EOF
3852 assert_equal('yes', g:didit)
3853
3854 python3 << trim EOF
3855 import vim
3856 vim.vars['didit'] = 'again'
3857 EOF
3858 assert_equal('again', g:didit)
3859 enddef
3860endif
3861
Bram Moolenaar20677332021-06-06 17:02:53 +02003862if has('lua')
3863 def Test_lua_heredoc()
3864 g:d = {}
3865 lua << trim EOF
3866 x = vim.eval('g:d')
3867 x['key'] = 'val'
3868 EOF
3869 assert_equal('val', g:d.key)
3870 enddef
3871endif
3872
Bram Moolenaarf7779c62020-05-03 15:38:16 +02003873
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02003874" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker