blob: 43fc68ef71e8be918b3e73151ec2c14433e8f485 [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 Moolenaar1983f1a2022-02-28 20:55:02 +0000553def Test_call_ufunc_failure()
554 var lines =<< trim END
555 vim9script
556 def Tryit()
557 g:Global(1, 2, 3)
558 enddef
559
560 func g:Global(a, b, c)
561 echo a:a a:b a:c
562 endfunc
563
564 defcompile
565
566 func! g:Global(a, b)
567 echo a:a a:b
568 endfunc
569 Tryit()
570 END
571 v9.CheckScriptFailure(lines, 'E118: Too many arguments for function: Global')
572 delfunc g:Global
573
574 lines =<< trim END
575 vim9script
576
577 g:Ref = function('len')
578 def Tryit()
579 g:Ref('x')
580 enddef
581
582 defcompile
583
584 g:Ref = function('add')
585 Tryit()
586 END
587 v9.CheckScriptFailure(lines, 'E119: Not enough arguments for function: add')
588 unlet g:Ref
589enddef
590
Bram Moolenaar62aec932022-01-29 21:45:34 +0000591def s:MyVarargs(arg: string, ...rest: list<string>): string
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200592 var res = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200593 for s in rest
594 res ..= ',' .. s
595 endfor
596 return res
597enddef
598
599def Test_call_varargs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200600 MyVarargs('one')->assert_equal('one')
601 MyVarargs('one', 'two')->assert_equal('one,two')
602 MyVarargs('one', 'two', 'three')->assert_equal('one,two,three')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200603enddef
604
Bram Moolenaar01dd6c32021-09-05 16:36:23 +0200605def Test_call_white_space()
Bram Moolenaar62aec932022-01-29 21:45:34 +0000606 v9.CheckDefAndScriptFailure(["call Test ('text')"], ['E476:', 'E1068:'])
Bram Moolenaar01dd6c32021-09-05 16:36:23 +0200607enddef
608
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200609def MyDefaultArgs(name = 'string'): string
610 return name
611enddef
612
Bram Moolenaar62aec932022-01-29 21:45:34 +0000613def s:MyDefaultSecond(name: string, second: bool = true): string
Bram Moolenaare30f64b2020-07-15 19:48:20 +0200614 return second ? name : 'none'
615enddef
616
Bram Moolenaar38a3bfa2021-03-29 22:14:55 +0200617
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200618def Test_call_default_args()
Bram Moolenaar62aec932022-01-29 21:45:34 +0000619 g:MyDefaultArgs()->assert_equal('string')
620 g:MyDefaultArgs(v:none)->assert_equal('string')
621 g:MyDefaultArgs('one')->assert_equal('one')
622 assert_fails('g:MyDefaultArgs("one", "two")', 'E118:', '', 4, 'Test_call_default_args')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200623
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200624 MyDefaultSecond('test')->assert_equal('test')
625 MyDefaultSecond('test', true)->assert_equal('test')
626 MyDefaultSecond('test', false)->assert_equal('none')
Bram Moolenaare30f64b2020-07-15 19:48:20 +0200627
Bram Moolenaar38a3bfa2021-03-29 22:14:55 +0200628 var lines =<< trim END
629 def MyDefaultThird(name: string, aa = 'aa', bb = 'bb'): string
630 return name .. aa .. bb
631 enddef
632
633 MyDefaultThird('->')->assert_equal('->aabb')
634 MyDefaultThird('->', v:none)->assert_equal('->aabb')
635 MyDefaultThird('->', 'xx')->assert_equal('->xxbb')
636 MyDefaultThird('->', v:none, v:none)->assert_equal('->aabb')
637 MyDefaultThird('->', 'xx', v:none)->assert_equal('->xxbb')
638 MyDefaultThird('->', v:none, 'yy')->assert_equal('->aayy')
639 MyDefaultThird('->', 'xx', 'yy')->assert_equal('->xxyy')
Bram Moolenaare28d9b32021-07-03 18:56:53 +0200640
641 def DefArg(mandatory: any, optional = mandatory): string
642 return mandatory .. optional
643 enddef
644 DefArg(1234)->assert_equal('12341234')
645 DefArg("ok")->assert_equal('okok')
Bram Moolenaar38a3bfa2021-03-29 22:14:55 +0200646 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000647 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar38a3bfa2021-03-29 22:14:55 +0200648
Bram Moolenaar62aec932022-01-29 21:45:34 +0000649 v9.CheckScriptFailure(['def Func(arg: number = asdf)', 'enddef', 'defcompile'], 'E1001:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100650 delfunc g:Func
Bram Moolenaar62aec932022-01-29 21:45:34 +0000651 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 +0100652 delfunc g:Func
Bram Moolenaar62aec932022-01-29 21:45:34 +0000653 v9.CheckDefFailure(['def Func(x: number = )', 'enddef'], 'E15:')
Bram Moolenaar12bce952021-03-11 20:04:04 +0100654
Bram Moolenaar38a3bfa2021-03-29 22:14:55 +0200655 lines =<< trim END
Bram Moolenaar12bce952021-03-11 20:04:04 +0100656 vim9script
657 def Func(a = b == 0 ? 1 : 2, b = 0)
658 enddef
659 defcompile
660 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000661 v9.CheckScriptFailure(lines, 'E1001: Variable not found: b')
Bram Moolenaar59618fe2021-12-21 12:32:17 +0000662
Bram Moolenaarfa46ead2021-12-22 13:18:39 +0000663 # using script variable requires matching type or type cast when executed
Bram Moolenaar59618fe2021-12-21 12:32:17 +0000664 lines =<< trim END
665 vim9script
666 var a: any
667 def Func(arg: string = a)
668 echo arg
669 enddef
670 defcompile
671 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000672 v9.CheckScriptSuccess(lines + ['a = "text"', 'Func()'])
673 v9.CheckScriptFailure(lines + ['a = 123', 'Func()'], 'E1013: Argument 1: type mismatch, expected string but got number')
Bram Moolenaar59618fe2021-12-21 12:32:17 +0000674
675 # using global variable does not require type cast
676 lines =<< trim END
677 vim9script
678 def Func(arg: string = g:str)
679 echo arg
680 enddef
681 g:str = 'works'
682 Func()
683 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000684 v9.CheckScriptSuccess(lines)
Bram Moolenaar04b12692020-05-04 23:24:44 +0200685enddef
686
Bram Moolenaar62aec932022-01-29 21:45:34 +0000687def s:FuncWithComment( # comment
Bram Moolenaarcef12702021-01-04 14:09:43 +0100688 a: number, #comment
689 b: bool, # comment
690 c: string) #comment
691 assert_equal(4, a)
692 assert_equal(true, b)
693 assert_equal('yes', c)
694enddef
695
696def Test_func_with_comments()
697 FuncWithComment(4, true, 'yes')
698
699 var lines =<< trim END
700 def Func(# comment
701 arg: string)
702 enddef
703 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000704 v9.CheckScriptFailure(lines, 'E125:', 1)
Bram Moolenaarcef12702021-01-04 14:09:43 +0100705
706 lines =<< trim END
707 def Func(
708 arg: string# comment
709 )
710 enddef
711 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000712 v9.CheckScriptFailure(lines, 'E475:', 2)
Bram Moolenaarcef12702021-01-04 14:09:43 +0100713
714 lines =<< trim END
715 def Func(
716 arg: string
717 )# comment
718 enddef
719 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000720 v9.CheckScriptFailure(lines, 'E488:', 3)
Bram Moolenaarcef12702021-01-04 14:09:43 +0100721enddef
722
Bram Moolenaar04b12692020-05-04 23:24:44 +0200723def Test_nested_function()
Bram Moolenaar38453522021-11-28 22:00:12 +0000724 def NestedDef(arg: string): string
Bram Moolenaar04b12692020-05-04 23:24:44 +0200725 return 'nested ' .. arg
726 enddef
Bram Moolenaar38453522021-11-28 22:00:12 +0000727 NestedDef(':def')->assert_equal('nested :def')
728
729 func NestedFunc(arg)
730 return 'nested ' .. a:arg
731 endfunc
732 NestedFunc(':func')->assert_equal('nested :func')
Bram Moolenaar04b12692020-05-04 23:24:44 +0200733
Bram Moolenaar62aec932022-01-29 21:45:34 +0000734 v9.CheckDefFailure(['def Nested()', 'enddef', 'Nested(66)'], 'E118:')
735 v9.CheckDefFailure(['def Nested(arg: string)', 'enddef', 'Nested()'], 'E119:')
Bram Moolenaar0e65d3d2020-05-05 17:53:16 +0200736
Bram Moolenaar62aec932022-01-29 21:45:34 +0000737 v9.CheckDefFailure(['def s:Nested()', 'enddef'], 'E1075:')
738 v9.CheckDefFailure(['def b:Nested()', 'enddef'], 'E1075:')
Bram Moolenaar8b848ca2020-09-10 22:28:01 +0200739
Bram Moolenaar54021752020-12-06 18:50:36 +0100740 var lines =<< trim END
741 def Outer()
742 def Inner()
743 # comment
744 enddef
745 def Inner()
746 enddef
747 enddef
748 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000749 v9.CheckDefFailure(lines, 'E1073:')
Bram Moolenaar54021752020-12-06 18:50:36 +0100750
751 lines =<< trim END
752 def Outer()
753 def Inner()
754 # comment
755 enddef
756 def! Inner()
757 enddef
758 enddef
759 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000760 v9.CheckDefFailure(lines, 'E1117:')
Bram Moolenaar54021752020-12-06 18:50:36 +0100761
Bram Moolenaardb8e5c22021-12-25 19:58:22 +0000762 lines =<< trim END
763 vim9script
764 def Outer()
765 def Inner()
766 g:result = 'ok'
767 enddef
768 Inner()
769 enddef
770 Outer()
771 Inner()
772 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000773 v9.CheckScriptFailure(lines, 'E117: Unknown function: Inner')
Bram Moolenaardb8e5c22021-12-25 19:58:22 +0000774 assert_equal('ok', g:result)
775 unlet g:result
776
Bram Moolenaarf681cfb2022-02-07 20:30:57 +0000777 lines =<< trim END
778 vim9script
779 def Outer()
780 def _Inner()
781 echo 'bad'
782 enddef
Bram Moolenaar3787f262022-02-07 21:54:01 +0000783 _Inner()
Bram Moolenaarf681cfb2022-02-07 20:30:57 +0000784 enddef
785 defcompile
786 END
Bram Moolenaar3787f262022-02-07 21:54:01 +0000787 v9.CheckScriptFailure(lines, 'E1267:')
Bram Moolenaarf681cfb2022-02-07 20:30:57 +0000788
789 lines =<< trim END
790 vim9script
791 def Outer()
792 def g:inner()
793 echo 'bad'
794 enddef
Bram Moolenaar3787f262022-02-07 21:54:01 +0000795 g:inner()
Bram Moolenaarf681cfb2022-02-07 20:30:57 +0000796 enddef
797 defcompile
798 END
Bram Moolenaar3787f262022-02-07 21:54:01 +0000799 v9.CheckScriptFailure(lines, 'E1267:')
800
801 lines =<< trim END
802 vim9script
803 def g:_Func()
804 echo 'bad'
805 enddef
806 END
807 v9.CheckScriptFailure(lines, 'E1267:')
808
809 lines =<< trim END
810 vim9script
Bram Moolenaara749a422022-02-12 19:52:25 +0000811 def _Func()
Bram Moolenaar3787f262022-02-07 21:54:01 +0000812 echo 'bad'
813 enddef
814 END
815 v9.CheckScriptFailure(lines, 'E1267:')
Bram Moolenaarf681cfb2022-02-07 20:30:57 +0000816
Bram Moolenaar54021752020-12-06 18:50:36 +0100817 # nested function inside conditional
Bram Moolenaar54021752020-12-06 18:50:36 +0100818 lines =<< trim END
819 vim9script
820 var thecount = 0
821 if true
822 def Test(): number
823 def TheFunc(): number
824 thecount += 1
825 return thecount
826 enddef
827 return TheFunc()
828 enddef
829 endif
830 defcompile
831 assert_equal(1, Test())
832 assert_equal(2, Test())
833 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000834 v9.CheckScriptSuccess(lines)
Bram Moolenaar8863bda2021-03-17 18:42:08 +0100835
836 # also works when "thecount" is inside the "if" block
837 lines =<< trim END
838 vim9script
839 if true
840 var thecount = 0
841 def Test(): number
842 def TheFunc(): number
843 thecount += 1
844 return thecount
845 enddef
846 return TheFunc()
847 enddef
848 endif
849 defcompile
850 assert_equal(1, Test())
851 assert_equal(2, Test())
852 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000853 v9.CheckScriptSuccess(lines)
Bram Moolenaar4bba16d2021-08-15 19:28:05 +0200854
855 lines =<< trim END
856 vim9script
857 def Outer()
858 def Inner()
859 echo 'hello'
860 enddef burp
861 enddef
862 defcompile
863 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000864 v9.CheckScriptFailure(lines, 'E1173: Text found after enddef: burp', 3)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200865enddef
866
Bram Moolenaaradc8e442020-12-31 18:28:18 +0100867def Test_not_nested_function()
868 echo printf('%d',
869 function('len')('xxx'))
870enddef
871
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200872func Test_call_default_args_from_func()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200873 call MyDefaultArgs()->assert_equal('string')
874 call MyDefaultArgs('one')->assert_equal('one')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200875 call assert_fails('call MyDefaultArgs("one", "two")', 'E118:', '', 3, 'Test_call_default_args_from_func')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200876endfunc
877
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200878def Test_nested_global_function()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200879 var lines =<< trim END
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200880 vim9script
881 def Outer()
882 def g:Inner(): string
883 return 'inner'
884 enddef
885 enddef
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200886 defcompile
887 Outer()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200888 g:Inner()->assert_equal('inner')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200889 delfunc g:Inner
890 Outer()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200891 g:Inner()->assert_equal('inner')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200892 delfunc g:Inner
893 Outer()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200894 g:Inner()->assert_equal('inner')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200895 delfunc g:Inner
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200896 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000897 v9.CheckScriptSuccess(lines)
Bram Moolenaar2c79e9d2020-08-01 18:57:52 +0200898
899 lines =<< trim END
900 vim9script
901 def Outer()
Bram Moolenaar38453522021-11-28 22:00:12 +0000902 func g:Inner()
903 return 'inner'
904 endfunc
905 enddef
906 defcompile
907 Outer()
908 g:Inner()->assert_equal('inner')
909 delfunc g:Inner
910 Outer()
911 g:Inner()->assert_equal('inner')
912 delfunc g:Inner
913 Outer()
914 g:Inner()->assert_equal('inner')
915 delfunc g:Inner
916 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000917 v9.CheckScriptSuccess(lines)
Bram Moolenaar38453522021-11-28 22:00:12 +0000918
919 lines =<< trim END
920 vim9script
921 def Outer()
Bram Moolenaar2c79e9d2020-08-01 18:57:52 +0200922 def g:Inner(): string
923 return 'inner'
924 enddef
925 enddef
926 defcompile
927 Outer()
928 Outer()
929 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000930 v9.CheckScriptFailure(lines, "E122:")
Bram Moolenaarcd45ed02020-12-22 17:35:54 +0100931 delfunc g:Inner
Bram Moolenaarad486a02020-08-01 23:22:18 +0200932
933 lines =<< trim END
934 vim9script
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100935 def Outer()
936 def g:Inner()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100937 echo map([1, 2, 3], (_, v) => v + 1)
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100938 enddef
939 g:Inner()
940 enddef
941 Outer()
942 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000943 v9.CheckScriptSuccess(lines)
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100944 delfunc g:Inner
945
946 lines =<< trim END
947 vim9script
Bram Moolenaarad486a02020-08-01 23:22:18 +0200948 def Func()
949 echo 'script'
950 enddef
951 def Outer()
952 def Func()
953 echo 'inner'
954 enddef
955 enddef
956 defcompile
957 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000958 v9.CheckScriptFailure(lines, "E1073:", 1)
Bram Moolenaard604d782021-11-20 21:46:20 +0000959
960 lines =<< trim END
961 vim9script
962 def Func()
963 echo 'script'
964 enddef
965 def Func()
966 echo 'script'
967 enddef
968 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000969 v9.CheckScriptFailure(lines, "E1073:", 5)
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200970enddef
971
Bram Moolenaar6abdcf82020-11-22 18:15:44 +0100972def DefListAll()
973 def
974enddef
975
976def DefListOne()
977 def DefListOne
978enddef
979
980def DefListMatches()
981 def /DefList
982enddef
983
984def Test_nested_def_list()
985 var funcs = split(execute('call DefListAll()'), "\n")
986 assert_true(len(funcs) > 10)
987 assert_true(funcs->index('def DefListAll()') >= 0)
988
989 funcs = split(execute('call DefListOne()'), "\n")
990 assert_equal([' def DefListOne()', '1 def DefListOne', ' enddef'], funcs)
991
992 funcs = split(execute('call DefListMatches()'), "\n")
993 assert_true(len(funcs) >= 3)
994 assert_true(funcs->index('def DefListAll()') >= 0)
995 assert_true(funcs->index('def DefListOne()') >= 0)
996 assert_true(funcs->index('def DefListMatches()') >= 0)
Bram Moolenaar54021752020-12-06 18:50:36 +0100997
998 var lines =<< trim END
999 vim9script
1000 def Func()
1001 def +Func+
1002 enddef
1003 defcompile
1004 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001005 v9.CheckScriptFailure(lines, 'E476:', 1)
Bram Moolenaar6abdcf82020-11-22 18:15:44 +01001006enddef
1007
Bram Moolenaare08be092022-02-17 13:08:26 +00001008def Test_global_function_not_found()
1009 var lines =<< trim END
1010 g:Ref = 123
1011 call g:Ref()
1012 END
1013 v9.CheckDefExecAndScriptFailure(lines, ['E117:', 'E1085:'], 2)
1014enddef
1015
Bram Moolenaar333894b2020-08-01 18:53:07 +02001016def Test_global_local_function()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001017 var lines =<< trim END
Bram Moolenaar333894b2020-08-01 18:53:07 +02001018 vim9script
1019 def g:Func(): string
1020 return 'global'
1021 enddef
1022 def Func(): string
1023 return 'local'
1024 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001025 g:Func()->assert_equal('global')
1026 Func()->assert_equal('local')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001027 delfunc g:Func
Bram Moolenaar333894b2020-08-01 18:53:07 +02001028 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001029 v9.CheckScriptSuccess(lines)
Bram Moolenaar035d6e92020-08-11 22:30:42 +02001030
1031 lines =<< trim END
1032 vim9script
1033 def g:Funcy()
1034 echo 'funcy'
1035 enddef
Bram Moolenaara749a422022-02-12 19:52:25 +00001036 Funcy()
Bram Moolenaar035d6e92020-08-11 22:30:42 +02001037 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001038 v9.CheckScriptFailure(lines, 'E117:')
Bram Moolenaar333894b2020-08-01 18:53:07 +02001039enddef
1040
Bram Moolenaar0f769812020-09-12 18:32:34 +02001041def Test_local_function_shadows_global()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001042 var lines =<< trim END
Bram Moolenaar0f769812020-09-12 18:32:34 +02001043 vim9script
1044 def g:Gfunc(): string
1045 return 'global'
1046 enddef
1047 def AnotherFunc(): number
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001048 var Gfunc = function('len')
Bram Moolenaar0f769812020-09-12 18:32:34 +02001049 return Gfunc('testing')
1050 enddef
1051 g:Gfunc()->assert_equal('global')
1052 AnotherFunc()->assert_equal(7)
1053 delfunc g:Gfunc
1054 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001055 v9.CheckScriptSuccess(lines)
Bram Moolenaar0f769812020-09-12 18:32:34 +02001056
1057 lines =<< trim END
1058 vim9script
1059 def g:Func(): string
1060 return 'global'
1061 enddef
1062 def AnotherFunc()
1063 g:Func = function('len')
1064 enddef
1065 AnotherFunc()
1066 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001067 v9.CheckScriptFailure(lines, 'E705:')
Bram Moolenaar0f769812020-09-12 18:32:34 +02001068 delfunc g:Func
Bram Moolenaar0865b152021-04-05 15:38:51 +02001069
Bram Moolenaar62aec932022-01-29 21:45:34 +00001070 # global function is not found with g: prefix
Bram Moolenaar0865b152021-04-05 15:38:51 +02001071 lines =<< trim END
1072 vim9script
1073 def g:Func(): string
1074 return 'global'
1075 enddef
1076 def AnotherFunc(): string
1077 return Func()
1078 enddef
1079 assert_equal('global', AnotherFunc())
Bram Moolenaar0865b152021-04-05 15:38:51 +02001080 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001081 v9.CheckScriptFailure(lines, 'E117:')
1082 delfunc g:Func
Bram Moolenaar0865b152021-04-05 15:38:51 +02001083
1084 lines =<< trim END
1085 vim9script
1086 def g:Func(): string
1087 return 'global'
1088 enddef
Bram Moolenaar848fadd2022-01-30 15:28:30 +00001089 assert_equal('global', g:Func())
Bram Moolenaar0865b152021-04-05 15:38:51 +02001090 delfunc g:Func
1091 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001092 v9.CheckScriptSuccess(lines)
Bram Moolenaar58493cf2022-01-06 12:23:30 +00001093
1094 # This does not shadow "i" which is visible only inside the for loop
1095 lines =<< trim END
1096 vim9script
1097
1098 def Foo(i: number)
1099 echo i
1100 enddef
1101
1102 for i in range(3)
1103 # Foo() is compiled here
1104 Foo(i)
1105 endfor
1106 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001107 v9.CheckScriptSuccess(lines)
Bram Moolenaar0f769812020-09-12 18:32:34 +02001108enddef
1109
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001110func TakesOneArg(arg)
1111 echo a:arg
1112endfunc
1113
1114def Test_call_wrong_args()
Bram Moolenaar62aec932022-01-29 21:45:34 +00001115 v9.CheckDefFailure(['g:TakesOneArg()'], 'E119:')
1116 v9.CheckDefFailure(['g:TakesOneArg(11, 22)'], 'E118:')
1117 v9.CheckDefFailure(['bufnr(xxx)'], 'E1001:')
1118 v9.CheckScriptFailure(['def Func(Ref: func(s: string))'], 'E475:')
Bram Moolenaaree8580e2020-08-28 17:19:07 +02001119
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001120 var lines =<< trim END
Bram Moolenaaree8580e2020-08-28 17:19:07 +02001121 vim9script
1122 def Func(s: string)
1123 echo s
1124 enddef
1125 Func([])
1126 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001127 v9.CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected string but got list<unknown>', 5)
Bram Moolenaarb185a402020-09-18 22:42:00 +02001128
Bram Moolenaar9a015112021-12-31 14:06:45 +00001129 # argument name declared earlier is found when declaring a function
Bram Moolenaarb185a402020-09-18 22:42:00 +02001130 lines =<< trim END
1131 vim9script
Bram Moolenaarb4893b82021-02-21 22:20:24 +01001132 var name = 'piet'
1133 def FuncOne(name: string)
Bram Moolenaar3a5988c2022-02-08 19:23:35 +00001134 echo name
Bram Moolenaarb4893b82021-02-21 22:20:24 +01001135 enddef
1136 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001137 v9.CheckScriptFailure(lines, 'E1168:')
Bram Moolenaarb4893b82021-02-21 22:20:24 +01001138
Bram Moolenaar3a5988c2022-02-08 19:23:35 +00001139 # same, inside the same block
1140 lines =<< trim END
1141 vim9script
1142 if true
1143 var name = 'piet'
1144 def FuncOne(name: string)
1145 echo name
1146 enddef
1147 endif
1148 END
1149 v9.CheckScriptFailure(lines, 'E1168:')
1150
1151 # variable in other block is OK
1152 lines =<< trim END
1153 vim9script
1154 if true
1155 var name = 'piet'
1156 endif
1157 def FuncOne(name: string)
1158 echo name
1159 enddef
1160 END
1161 v9.CheckScriptSuccess(lines)
1162
Bram Moolenaardce24412022-02-08 20:35:30 +00001163 # with another variable in another block
1164 lines =<< trim END
1165 vim9script
1166 if true
1167 var name = 'piet'
1168 # define a function so that the variable isn't cleared
1169 def GetItem(): string
1170 return item
1171 enddef
1172 endif
1173 if true
1174 var name = 'peter'
1175 def FuncOne(name: string)
1176 echo name
1177 enddef
1178 endif
1179 END
1180 v9.CheckScriptFailure(lines, 'E1168:')
1181
1182 # only variable in another block is OK
1183 lines =<< trim END
1184 vim9script
1185 if true
1186 var name = 'piet'
1187 # define a function so that the variable isn't cleared
1188 def GetItem(): string
1189 return item
1190 enddef
1191 endif
1192 if true
1193 def FuncOne(name: string)
1194 echo name
1195 enddef
1196 endif
1197 END
1198 v9.CheckScriptSuccess(lines)
1199
Bram Moolenaar9a015112021-12-31 14:06:45 +00001200 # argument name declared later is only found when compiling
1201 lines =<< trim END
1202 vim9script
1203 def FuncOne(name: string)
1204 echo nr
1205 enddef
1206 var name = 'piet'
1207 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001208 v9.CheckScriptSuccess(lines)
1209 v9.CheckScriptFailure(lines + ['defcompile'], 'E1168:')
Bram Moolenaar9a015112021-12-31 14:06:45 +00001210
Bram Moolenaarb4893b82021-02-21 22:20:24 +01001211 lines =<< trim END
1212 vim9script
Bram Moolenaarb185a402020-09-18 22:42:00 +02001213 def FuncOne(nr: number)
1214 echo nr
1215 enddef
1216 def FuncTwo()
1217 FuncOne()
1218 enddef
1219 defcompile
1220 END
1221 writefile(lines, 'Xscript')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001222 var didCatch = false
Bram Moolenaarb185a402020-09-18 22:42:00 +02001223 try
1224 source Xscript
1225 catch
1226 assert_match('E119: Not enough arguments for function: <SNR>\d\+_FuncOne', v:exception)
1227 assert_match('Xscript\[8\]..function <SNR>\d\+_FuncTwo, line 1', v:throwpoint)
1228 didCatch = true
1229 endtry
1230 assert_true(didCatch)
1231
1232 lines =<< trim END
1233 vim9script
1234 def FuncOne(nr: number)
1235 echo nr
1236 enddef
1237 def FuncTwo()
1238 FuncOne(1, 2)
1239 enddef
1240 defcompile
1241 END
1242 writefile(lines, 'Xscript')
1243 didCatch = false
1244 try
1245 source Xscript
1246 catch
1247 assert_match('E118: Too many arguments for function: <SNR>\d\+_FuncOne', v:exception)
1248 assert_match('Xscript\[8\]..function <SNR>\d\+_FuncTwo, line 1', v:throwpoint)
1249 didCatch = true
1250 endtry
1251 assert_true(didCatch)
1252
1253 delete('Xscript')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001254enddef
1255
Bram Moolenaar50824712020-12-20 21:10:17 +01001256def Test_call_funcref_wrong_args()
1257 var head =<< trim END
1258 vim9script
1259 def Func3(a1: string, a2: number, a3: list<number>)
1260 echo a1 .. a2 .. a3[0]
1261 enddef
1262 def Testme()
1263 var funcMap: dict<func> = {func: Func3}
1264 END
1265 var tail =<< trim END
1266 enddef
1267 Testme()
1268 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001269 v9.CheckScriptSuccess(head + ["funcMap['func']('str', 123, [1, 2, 3])"] + tail)
Bram Moolenaar50824712020-12-20 21:10:17 +01001270
Bram Moolenaar62aec932022-01-29 21:45:34 +00001271 v9.CheckScriptFailure(head + ["funcMap['func']('str', 123)"] + tail, 'E119:')
1272 v9.CheckScriptFailure(head + ["funcMap['func']('str', 123, [1], 4)"] + tail, 'E118:')
Bram Moolenaar32b3f822021-01-06 21:59:39 +01001273
1274 var lines =<< trim END
1275 vim9script
1276 var Ref: func(number): any
1277 Ref = (j) => !j
1278 echo Ref(false)
1279 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001280 v9.CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected number but got bool', 4)
Bram Moolenaar32b3f822021-01-06 21:59:39 +01001281
1282 lines =<< trim END
1283 vim9script
1284 var Ref: func(number): any
1285 Ref = (j) => !j
1286 call Ref(false)
1287 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001288 v9.CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected number but got bool', 4)
Bram Moolenaar50824712020-12-20 21:10:17 +01001289enddef
1290
Bram Moolenaarb4d16cb2020-11-05 18:45:46 +01001291def Test_call_lambda_args()
Bram Moolenaar2a389082021-04-09 20:24:31 +02001292 var lines =<< trim END
1293 var Callback = (..._) => 'anything'
1294 assert_equal('anything', Callback())
1295 assert_equal('anything', Callback(1))
1296 assert_equal('anything', Callback('a', 2))
Bram Moolenaar1088b692021-04-09 22:12:44 +02001297
1298 assert_equal('xyz', ((a: string): string => a)('xyz'))
Bram Moolenaar2a389082021-04-09 20:24:31 +02001299 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001300 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar2a389082021-04-09 20:24:31 +02001301
Bram Moolenaar62aec932022-01-29 21:45:34 +00001302 v9.CheckDefFailure(['echo ((i) => 0)()'],
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001303 'E119: Not enough arguments for function: ((i) => 0)()')
Bram Moolenaarb4d16cb2020-11-05 18:45:46 +01001304
Bram Moolenaar2a389082021-04-09 20:24:31 +02001305 lines =<< trim END
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001306 var Ref = (x: number, y: number) => x + y
Bram Moolenaarb4d16cb2020-11-05 18:45:46 +01001307 echo Ref(1, 'x')
1308 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001309 v9.CheckDefFailure(lines, 'E1013: Argument 2: type mismatch, expected number but got string')
Bram Moolenaare68b02a2021-01-03 13:09:51 +01001310
1311 lines =<< trim END
1312 var Ref: func(job, string, number)
1313 Ref = (x, y) => 0
1314 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001315 v9.CheckDefAndScriptFailure(lines, 'E1012:')
Bram Moolenaare68b02a2021-01-03 13:09:51 +01001316
1317 lines =<< trim END
1318 var Ref: func(job, string)
1319 Ref = (x, y, z) => 0
1320 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001321 v9.CheckDefAndScriptFailure(lines, 'E1012:')
Bram Moolenaar057e84a2021-02-28 16:55:11 +01001322
1323 lines =<< trim END
1324 var one = 1
1325 var l = [1, 2, 3]
1326 echo map(l, (one) => one)
1327 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001328 v9.CheckDefFailure(lines, 'E1167:')
1329 v9.CheckScriptFailure(['vim9script'] + lines, 'E1168:')
Bram Moolenaar057e84a2021-02-28 16:55:11 +01001330
1331 lines =<< trim END
Bram Moolenaar14ded112021-06-26 19:25:49 +02001332 var Ref: func(any, ?any): bool
1333 Ref = (_, y = 1) => false
1334 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001335 v9.CheckDefAndScriptFailure(lines, 'E1172:')
Bram Moolenaar14ded112021-06-26 19:25:49 +02001336
1337 lines =<< trim END
Bram Moolenaar015cf102021-06-26 21:52:02 +02001338 var a = 0
1339 var b = (a == 0 ? 1 : 2)
1340 assert_equal(1, b)
Bram Moolenaar98f9a5f2021-06-26 22:22:38 +02001341 var txt = 'a'
1342 b = (txt =~ 'x' ? 1 : 2)
1343 assert_equal(2, b)
Bram Moolenaar015cf102021-06-26 21:52:02 +02001344 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001345 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar015cf102021-06-26 21:52:02 +02001346
1347 lines =<< trim END
Bram Moolenaar057e84a2021-02-28 16:55:11 +01001348 def ShadowLocal()
1349 var one = 1
1350 var l = [1, 2, 3]
1351 echo map(l, (one) => one)
1352 enddef
1353 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001354 v9.CheckDefFailure(lines, 'E1167:')
Bram Moolenaar057e84a2021-02-28 16:55:11 +01001355
1356 lines =<< trim END
1357 def Shadowarg(one: number)
1358 var l = [1, 2, 3]
1359 echo map(l, (one) => one)
1360 enddef
1361 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001362 v9.CheckDefFailure(lines, 'E1167:')
Bram Moolenaar767034c2021-04-09 17:24:52 +02001363
1364 lines =<< trim END
1365 echo ((a) => a)('aa', 'bb')
1366 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001367 v9.CheckDefAndScriptFailure(lines, 'E118:', 1)
Bram Moolenaarc4c56422021-07-21 20:38:46 +02001368
1369 lines =<< trim END
1370 echo 'aa'->((a) => a)('bb')
1371 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001372 v9.CheckDefFailure(lines, 'E118: Too many arguments for function: ->((a) => a)(''bb'')', 1)
1373 v9.CheckScriptFailure(['vim9script'] + lines, 'E118: Too many arguments for function: <lambda>', 2)
Bram Moolenaarb4d16cb2020-11-05 18:45:46 +01001374enddef
1375
Bram Moolenaara755fdb2021-11-20 21:35:41 +00001376def Test_lambda_line_nr()
1377 var lines =<< trim END
1378 vim9script
1379 # comment
1380 # comment
1381 var id = timer_start(1'000, (_) => 0)
1382 var out = execute('verbose ' .. timer_info(id)[0].callback
1383 ->string()
1384 ->substitute("('\\|')", ' ', 'g'))
1385 assert_match('Last set from .* line 4', out)
1386 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001387 v9.CheckScriptSuccess(lines)
Bram Moolenaara755fdb2021-11-20 21:35:41 +00001388enddef
1389
Bram Moolenaar5f91e742021-03-17 21:29:29 +01001390def FilterWithCond(x: string, Cond: func(string): bool): bool
1391 return Cond(x)
1392enddef
1393
Bram Moolenaar0346b792021-01-31 22:18:29 +01001394def Test_lambda_return_type()
1395 var lines =<< trim END
1396 var Ref = (): => 123
1397 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001398 v9.CheckDefAndScriptFailure(lines, 'E1157:', 1)
Bram Moolenaar5f91e742021-03-17 21:29:29 +01001399
Yegappan Lakshmanan611728f2021-05-24 15:15:47 +02001400 # no space before the return type
1401 lines =<< trim END
1402 var Ref = (x):number => x + 1
1403 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001404 v9.CheckDefAndScriptFailure(lines, 'E1069:', 1)
Yegappan Lakshmanan611728f2021-05-24 15:15:47 +02001405
Bram Moolenaar5f91e742021-03-17 21:29:29 +01001406 # this works
1407 for x in ['foo', 'boo']
Bram Moolenaar62aec932022-01-29 21:45:34 +00001408 echo g:FilterWithCond(x, (v) => v =~ '^b')
Bram Moolenaar5f91e742021-03-17 21:29:29 +01001409 endfor
1410
1411 # this fails
1412 lines =<< trim END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001413 echo g:FilterWithCond('foo', (v) => v .. '^b')
Bram Moolenaar5f91e742021-03-17 21:29:29 +01001414 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001415 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 +02001416
1417 lines =<< trim END
1418 var Lambda1 = (x) => {
1419 return x
1420 }
1421 assert_equal('asdf', Lambda1('asdf'))
1422 var Lambda2 = (x): string => {
1423 return x
1424 }
1425 assert_equal('foo', Lambda2('foo'))
1426 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001427 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaara9931532021-06-12 15:58:16 +02001428
1429 lines =<< trim END
1430 var Lambda = (x): string => {
1431 return x
1432 }
1433 echo Lambda(['foo'])
1434 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001435 v9.CheckDefExecAndScriptFailure(lines, 'E1012:')
Bram Moolenaar0346b792021-01-31 22:18:29 +01001436enddef
1437
Bram Moolenaar709664c2020-12-12 14:33:41 +01001438def Test_lambda_uses_assigned_var()
Bram Moolenaar62aec932022-01-29 21:45:34 +00001439 v9.CheckDefSuccess([
Bram Moolenaar709664c2020-12-12 14:33:41 +01001440 'var x: any = "aaa"'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001441 'x = filter(["bbb"], (_, v) => v =~ x)'])
Bram Moolenaar709664c2020-12-12 14:33:41 +01001442enddef
1443
Bram Moolenaar18062fc2021-03-05 21:35:47 +01001444def Test_pass_legacy_lambda_to_def_func()
1445 var lines =<< trim END
1446 vim9script
1447 func Foo()
1448 eval s:Bar({x -> 0})
1449 endfunc
1450 def Bar(y: any)
1451 enddef
1452 Foo()
1453 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001454 v9.CheckScriptSuccess(lines)
Bram Moolenaar831bdf82021-06-22 19:32:17 +02001455
1456 lines =<< trim END
1457 vim9script
Bram Moolenaar7a40ff02021-07-04 15:54:08 +02001458 def g:TestFunc(f: func)
Bram Moolenaar831bdf82021-06-22 19:32:17 +02001459 enddef
1460 legacy call g:TestFunc({-> 0})
1461 delfunc g:TestFunc
1462
1463 def g:TestFunc(f: func(number))
1464 enddef
1465 legacy call g:TestFunc({nr -> 0})
1466 delfunc g:TestFunc
1467 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001468 v9.CheckScriptSuccess(lines)
Bram Moolenaar18062fc2021-03-05 21:35:47 +01001469enddef
1470
Bram Moolenaar844fb642021-10-23 13:32:30 +01001471def Test_lambda_in_reduce_line_break()
1472 # this was using freed memory
1473 var lines =<< trim END
1474 vim9script
1475 const result: dict<number> =
1476 ['Bob', 'Sam', 'Cat', 'Bob', 'Cat', 'Cat']
1477 ->reduce((acc, val) => {
1478 if has_key(acc, val)
1479 acc[val] += 1
1480 return acc
1481 else
1482 acc[val] = 1
1483 return acc
1484 endif
1485 }, {})
1486 assert_equal({Bob: 2, Sam: 1, Cat: 3}, result)
1487 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001488 v9.CheckScriptSuccess(lines)
Bram Moolenaar844fb642021-10-23 13:32:30 +01001489enddef
1490
Bram Moolenaardcb53be2021-12-09 14:23:43 +00001491def Test_set_opfunc_to_lambda()
1492 var lines =<< trim END
1493 vim9script
1494 nnoremap <expr> <F4> <SID>CountSpaces() .. '_'
1495 def CountSpaces(type = ''): string
1496 if type == ''
1497 &operatorfunc = (t) => CountSpaces(t)
1498 return 'g@'
1499 endif
1500 normal! '[V']y
1501 g:result = getreg('"')->count(' ')
1502 return ''
1503 enddef
1504 new
1505 'a b c d e'->setline(1)
1506 feedkeys("\<F4>", 'x')
1507 assert_equal(4, g:result)
1508 bwipe!
1509 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001510 v9.CheckScriptSuccess(lines)
Bram Moolenaardcb53be2021-12-09 14:23:43 +00001511enddef
1512
Bram Moolenaaref082e12021-12-12 21:02:03 +00001513def Test_set_opfunc_to_global_function()
1514 var lines =<< trim END
1515 vim9script
1516 def g:CountSpaces(type = ''): string
1517 normal! '[V']y
1518 g:result = getreg('"')->count(' ')
1519 return ''
1520 enddef
Bram Moolenaarb15cf442021-12-16 15:49:43 +00001521 # global function works at script level
Bram Moolenaaref082e12021-12-12 21:02:03 +00001522 &operatorfunc = g:CountSpaces
1523 new
1524 'a b c d e'->setline(1)
1525 feedkeys("g@_", 'x')
1526 assert_equal(4, g:result)
Bram Moolenaarb15cf442021-12-16 15:49:43 +00001527
1528 &operatorfunc = ''
1529 g:result = 0
1530 # global function works in :def function
1531 def Func()
1532 &operatorfunc = g:CountSpaces
1533 enddef
1534 Func()
1535 feedkeys("g@_", 'x')
1536 assert_equal(4, g:result)
1537
Bram Moolenaaref082e12021-12-12 21:02:03 +00001538 bwipe!
1539 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001540 v9.CheckScriptSuccess(lines)
Bram Moolenaaref082e12021-12-12 21:02:03 +00001541 &operatorfunc = ''
1542enddef
1543
Bram Moolenaar33b968d2021-12-13 11:31:04 +00001544def Test_use_script_func_name_with_prefix()
1545 var lines =<< trim END
1546 vim9script
Bram Moolenaara749a422022-02-12 19:52:25 +00001547 func g:Getit()
Bram Moolenaar33b968d2021-12-13 11:31:04 +00001548 return 'it'
1549 endfunc
Bram Moolenaara749a422022-02-12 19:52:25 +00001550 var Fn = g:Getit
Bram Moolenaar33b968d2021-12-13 11:31:04 +00001551 assert_equal('it', Fn())
1552 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001553 v9.CheckScriptSuccess(lines)
Bram Moolenaar33b968d2021-12-13 11:31:04 +00001554enddef
1555
Bram Moolenaardd297bc2021-12-10 10:37:38 +00001556def Test_lambda_type_allocated()
1557 # Check that unreferencing a partial using a lambda can use the variable type
1558 # after the lambda has been freed and does not leak memory.
1559 var lines =<< trim END
1560 vim9script
1561
1562 func MyomniFunc1(val, findstart, base)
1563 return a:findstart ? 0 : []
1564 endfunc
1565
1566 var Lambda = (a, b) => MyomniFunc1(19, a, b)
1567 &omnifunc = Lambda
1568 Lambda = (a, b) => MyomniFunc1(20, a, b)
1569 &omnifunc = string(Lambda)
1570 Lambda = (a, b) => strlen(a)
1571 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001572 v9.CheckScriptSuccess(lines)
Bram Moolenaardd297bc2021-12-10 10:37:38 +00001573enddef
1574
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001575" Default arg and varargs
1576def MyDefVarargs(one: string, two = 'foo', ...rest: list<string>): string
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001577 var res = one .. ',' .. two
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001578 for s in rest
1579 res ..= ',' .. s
1580 endfor
1581 return res
1582enddef
1583
1584def Test_call_def_varargs()
Bram Moolenaar62aec932022-01-29 21:45:34 +00001585 assert_fails('g:MyDefVarargs()', 'E119:', '', 1, 'Test_call_def_varargs')
1586 g:MyDefVarargs('one')->assert_equal('one,foo')
1587 g:MyDefVarargs('one', 'two')->assert_equal('one,two')
1588 g:MyDefVarargs('one', 'two', 'three')->assert_equal('one,two,three')
1589 v9.CheckDefFailure(['g:MyDefVarargs("one", 22)'],
Bram Moolenaar77072282020-09-16 17:55:40 +02001590 'E1013: Argument 2: type mismatch, expected string but got number')
Bram Moolenaar62aec932022-01-29 21:45:34 +00001591 v9.CheckDefFailure(['g:MyDefVarargs("one", "two", 123)'],
Bram Moolenaar77072282020-09-16 17:55:40 +02001592 'E1013: Argument 3: type mismatch, expected string but got number')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +02001593
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001594 var lines =<< trim END
Bram Moolenaar24aa48b2020-07-25 16:33:02 +02001595 vim9script
1596 def Func(...l: list<string>)
1597 echo l
1598 enddef
1599 Func('a', 'b', 'c')
1600 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001601 v9.CheckScriptSuccess(lines)
Bram Moolenaar24aa48b2020-07-25 16:33:02 +02001602
1603 lines =<< trim END
1604 vim9script
1605 def Func(...l: list<string>)
1606 echo l
1607 enddef
1608 Func()
1609 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001610 v9.CheckScriptSuccess(lines)
Bram Moolenaar24aa48b2020-07-25 16:33:02 +02001611
1612 lines =<< trim END
1613 vim9script
Bram Moolenaar2a389082021-04-09 20:24:31 +02001614 def Func(...l: list<any>)
Bram Moolenaar2f8cbc42020-09-16 17:22:59 +02001615 echo l
1616 enddef
1617 Func(0)
1618 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001619 v9.CheckScriptSuccess(lines)
Bram Moolenaar2f8cbc42020-09-16 17:22:59 +02001620
1621 lines =<< trim END
1622 vim9script
Bram Moolenaar2a389082021-04-09 20:24:31 +02001623 def Func(...l: any)
1624 echo l
1625 enddef
1626 Func(0)
1627 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001628 v9.CheckScriptFailure(lines, 'E1180:', 2)
Bram Moolenaar2a389082021-04-09 20:24:31 +02001629
1630 lines =<< trim END
1631 vim9script
Bram Moolenaar28022722020-09-21 22:02:49 +02001632 def Func(..._l: list<string>)
1633 echo _l
1634 enddef
1635 Func('a', 'b', 'c')
1636 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001637 v9.CheckScriptSuccess(lines)
Bram Moolenaar28022722020-09-21 22:02:49 +02001638
1639 lines =<< trim END
1640 vim9script
Bram Moolenaar24aa48b2020-07-25 16:33:02 +02001641 def Func(...l: list<string>)
1642 echo l
1643 enddef
1644 Func(1, 2, 3)
1645 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001646 v9.CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +02001647
1648 lines =<< trim END
1649 vim9script
1650 def Func(...l: list<string>)
1651 echo l
1652 enddef
1653 Func('a', 9)
1654 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001655 v9.CheckScriptFailure(lines, 'E1013: Argument 2: type mismatch')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +02001656
1657 lines =<< trim END
1658 vim9script
1659 def Func(...l: list<string>)
1660 echo l
1661 enddef
1662 Func(1, 'a')
1663 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001664 v9.CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch')
Bram Moolenaar4f53b792021-02-07 15:59:49 +01001665
1666 lines =<< trim END
1667 vim9script
1668 def Func( # some comment
1669 ...l = []
1670 )
1671 echo l
1672 enddef
1673 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001674 v9.CheckScriptFailure(lines, 'E1160:')
Bram Moolenaar6ce46b92021-08-07 15:35:36 +02001675
1676 lines =<< trim END
1677 vim9script
1678 def DoIt()
1679 g:Later('')
1680 enddef
1681 defcompile
1682 def g:Later(...l: list<number>)
1683 enddef
1684 DoIt()
1685 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001686 v9.CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected number but got string')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001687enddef
1688
Bram Moolenaar1378fbc2020-04-11 20:50:33 +02001689let s:value = ''
1690
1691def FuncOneDefArg(opt = 'text')
1692 s:value = opt
1693enddef
1694
1695def FuncTwoDefArg(nr = 123, opt = 'text'): string
1696 return nr .. opt
1697enddef
1698
1699def FuncVarargs(...arg: list<string>): string
1700 return join(arg, ',')
1701enddef
1702
1703def Test_func_type_varargs()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001704 var RefDefArg: func(?string)
Bram Moolenaar848fadd2022-01-30 15:28:30 +00001705 RefDefArg = g:FuncOneDefArg
Bram Moolenaar1378fbc2020-04-11 20:50:33 +02001706 RefDefArg()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001707 s:value->assert_equal('text')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +02001708 RefDefArg('some')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001709 s:value->assert_equal('some')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +02001710
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001711 var RefDef2Arg: func(?number, ?string): string
Bram Moolenaar848fadd2022-01-30 15:28:30 +00001712 RefDef2Arg = g:FuncTwoDefArg
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001713 RefDef2Arg()->assert_equal('123text')
1714 RefDef2Arg(99)->assert_equal('99text')
1715 RefDef2Arg(77, 'some')->assert_equal('77some')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +02001716
Bram Moolenaar62aec932022-01-29 21:45:34 +00001717 v9.CheckDefFailure(['var RefWrong: func(string?)'], 'E1010:')
1718 v9.CheckDefFailure(['var RefWrong: func(?string, string)'], 'E1007:')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +02001719
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001720 var RefVarargs: func(...list<string>): string
Bram Moolenaar848fadd2022-01-30 15:28:30 +00001721 RefVarargs = g:FuncVarargs
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001722 RefVarargs()->assert_equal('')
1723 RefVarargs('one')->assert_equal('one')
1724 RefVarargs('one', 'two')->assert_equal('one,two')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +02001725
Bram Moolenaar62aec932022-01-29 21:45:34 +00001726 v9.CheckDefFailure(['var RefWrong: func(...list<string>, string)'], 'E110:')
1727 v9.CheckDefFailure(['var RefWrong: func(...list<string>, ?string)'], 'E110:')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +02001728enddef
1729
Bram Moolenaar0b76b422020-04-07 22:05:08 +02001730" Only varargs
1731def MyVarargsOnly(...args: list<string>): string
1732 return join(args, ',')
1733enddef
1734
1735def Test_call_varargs_only()
Bram Moolenaar62aec932022-01-29 21:45:34 +00001736 g:MyVarargsOnly()->assert_equal('')
1737 g:MyVarargsOnly('one')->assert_equal('one')
1738 g:MyVarargsOnly('one', 'two')->assert_equal('one,two')
1739 v9.CheckDefFailure(['g:MyVarargsOnly(1)'], 'E1013: Argument 1: type mismatch, expected string but got number')
1740 v9.CheckDefFailure(['g:MyVarargsOnly("one", 2)'], 'E1013: Argument 2: type mismatch, expected string but got number')
Bram Moolenaar0b76b422020-04-07 22:05:08 +02001741enddef
1742
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001743def Test_using_var_as_arg()
Bram Moolenaard2939812021-12-30 17:09:05 +00001744 var lines =<< trim END
1745 def Func(x: number)
1746 var x = 234
1747 enddef
1748 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001749 v9.CheckDefFailure(lines, 'E1006:')
Bram Moolenaard2939812021-12-30 17:09:05 +00001750
1751 lines =<< trim END
1752 def Func(Ref: number)
1753 def Ref()
1754 enddef
1755 enddef
1756 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001757 v9.CheckDefFailure(lines, 'E1073:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001758enddef
1759
Bram Moolenaar62aec932022-01-29 21:45:34 +00001760def s:DictArg(arg: dict<string>)
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +02001761 arg['key'] = 'value'
1762enddef
1763
Bram Moolenaar62aec932022-01-29 21:45:34 +00001764def s:ListArg(arg: list<string>)
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +02001765 arg[0] = 'value'
1766enddef
1767
1768def Test_assign_to_argument()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +02001769 # works for dict and list
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001770 var d: dict<string> = {}
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +02001771 DictArg(d)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001772 d['key']->assert_equal('value')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001773 var l: list<string> = []
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +02001774 ListArg(l)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001775 l[0]->assert_equal('value')
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +02001776
Bram Moolenaar62aec932022-01-29 21:45:34 +00001777 v9.CheckScriptFailure(['def Func(arg: number)', 'arg = 3', 'enddef', 'defcompile'], 'E1090:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001778 delfunc! g:Func
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +02001779enddef
1780
Bram Moolenaarb816dae2020-09-20 22:04:00 +02001781" These argument names are reserved in legacy functions.
Bram Moolenaar62aec932022-01-29 21:45:34 +00001782def s:WithReservedNames(firstline: string, lastline: string): string
Bram Moolenaarb816dae2020-09-20 22:04:00 +02001783 return firstline .. lastline
1784enddef
1785
1786def Test_argument_names()
1787 assert_equal('OK', WithReservedNames('O', 'K'))
1788enddef
1789
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001790def Test_call_func_defined_later()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001791 g:DefinedLater('one')->assert_equal('one')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001792 assert_fails('NotDefined("one")', 'E117:', '', 2, 'Test_call_func_defined_later')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001793enddef
1794
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +02001795func DefinedLater(arg)
1796 return a:arg
1797endfunc
1798
1799def Test_call_funcref()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001800 g:SomeFunc('abc')->assert_equal(3)
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001801 assert_fails('NotAFunc()', 'E117:', '', 2, 'Test_call_funcref') # comment after call
Bram Moolenaar2ef91562021-12-11 16:14:07 +00001802 assert_fails('g:NotAFunc()', 'E1085:', '', 3, 'Test_call_funcref')
Bram Moolenaar2f1980f2020-07-22 19:30:06 +02001803
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001804 var lines =<< trim END
Bram Moolenaar2f1980f2020-07-22 19:30:06 +02001805 vim9script
1806 def RetNumber(): number
1807 return 123
1808 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001809 var Funcref: func: number = function('RetNumber')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001810 Funcref()->assert_equal(123)
Bram Moolenaar2f1980f2020-07-22 19:30:06 +02001811 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001812 v9.CheckScriptSuccess(lines)
Bram Moolenaar0f60e802020-07-22 20:16:11 +02001813
1814 lines =<< trim END
1815 vim9script
1816 def RetNumber(): number
1817 return 123
1818 enddef
1819 def Bar(F: func: number): number
1820 return F()
1821 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001822 var Funcref = function('RetNumber')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001823 Bar(Funcref)->assert_equal(123)
Bram Moolenaar0f60e802020-07-22 20:16:11 +02001824 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001825 v9.CheckScriptSuccess(lines)
Bram Moolenaarbfba8652020-07-23 20:09:10 +02001826
1827 lines =<< trim END
1828 vim9script
1829 def UseNumber(nr: number)
1830 echo nr
1831 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001832 var Funcref: func(number) = function('UseNumber')
Bram Moolenaarbfba8652020-07-23 20:09:10 +02001833 Funcref(123)
1834 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001835 v9.CheckScriptSuccess(lines)
Bram Moolenaarb8070e32020-07-23 20:56:04 +02001836
1837 lines =<< trim END
1838 vim9script
1839 def UseNumber(nr: number)
1840 echo nr
1841 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001842 var Funcref: func(string) = function('UseNumber')
Bram Moolenaarb8070e32020-07-23 20:56:04 +02001843 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001844 v9.CheckScriptFailure(lines, 'E1012: Type mismatch; expected func(string) but got func(number)')
Bram Moolenaar4fc224c2020-07-26 17:56:25 +02001845
1846 lines =<< trim END
1847 vim9script
1848 def EchoNr(nr = 34)
1849 g:echo = nr
1850 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001851 var Funcref: func(?number) = function('EchoNr')
Bram Moolenaar4fc224c2020-07-26 17:56:25 +02001852 Funcref()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001853 g:echo->assert_equal(34)
Bram Moolenaar4fc224c2020-07-26 17:56:25 +02001854 Funcref(123)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001855 g:echo->assert_equal(123)
Bram Moolenaar4fc224c2020-07-26 17:56:25 +02001856 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001857 v9.CheckScriptSuccess(lines)
Bram Moolenaarace61322020-07-26 18:16:58 +02001858
1859 lines =<< trim END
1860 vim9script
1861 def EchoList(...l: list<number>)
1862 g:echo = l
1863 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001864 var Funcref: func(...list<number>) = function('EchoList')
Bram Moolenaarace61322020-07-26 18:16:58 +02001865 Funcref()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001866 g:echo->assert_equal([])
Bram Moolenaarace61322020-07-26 18:16:58 +02001867 Funcref(1, 2, 3)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001868 g:echo->assert_equal([1, 2, 3])
Bram Moolenaarace61322020-07-26 18:16:58 +02001869 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001870 v9.CheckScriptSuccess(lines)
Bram Moolenaar01865ad2020-07-26 18:33:09 +02001871
1872 lines =<< trim END
1873 vim9script
1874 def OptAndVar(nr: number, opt = 12, ...l: list<number>): number
1875 g:optarg = opt
1876 g:listarg = l
1877 return nr
1878 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001879 var Funcref: func(number, ?number, ...list<number>): number = function('OptAndVar')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001880 Funcref(10)->assert_equal(10)
1881 g:optarg->assert_equal(12)
1882 g:listarg->assert_equal([])
Bram Moolenaar01865ad2020-07-26 18:33:09 +02001883
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001884 Funcref(11, 22)->assert_equal(11)
1885 g:optarg->assert_equal(22)
1886 g:listarg->assert_equal([])
Bram Moolenaar01865ad2020-07-26 18:33:09 +02001887
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001888 Funcref(17, 18, 1, 2, 3)->assert_equal(17)
1889 g:optarg->assert_equal(18)
1890 g:listarg->assert_equal([1, 2, 3])
Bram Moolenaar01865ad2020-07-26 18:33:09 +02001891 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001892 v9.CheckScriptSuccess(lines)
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +02001893enddef
1894
1895let SomeFunc = function('len')
1896let NotAFunc = 'text'
1897
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +02001898def CombineFuncrefTypes()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +02001899 # same arguments, different return type
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001900 var Ref1: func(bool): string
1901 var Ref2: func(bool): number
1902 var Ref3: func(bool): any
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +02001903 Ref3 = g:cond ? Ref1 : Ref2
1904
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +02001905 # different number of arguments
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001906 var Refa1: func(bool): number
1907 var Refa2: func(bool, number): number
1908 var Refa3: func: number
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +02001909 Refa3 = g:cond ? Refa1 : Refa2
1910
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +02001911 # different argument types
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001912 var Refb1: func(bool, string): number
1913 var Refb2: func(string, number): number
1914 var Refb3: func(any, any): number
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +02001915 Refb3 = g:cond ? Refb1 : Refb2
1916enddef
1917
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001918def FuncWithForwardCall()
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +02001919 return g:DefinedEvenLater("yes")
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001920enddef
1921
1922def DefinedEvenLater(arg: string): string
1923 return arg
1924enddef
1925
1926def Test_error_in_nested_function()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +02001927 # Error in called function requires unwinding the call stack.
Bram Moolenaar848fadd2022-01-30 15:28:30 +00001928 assert_fails('g:FuncWithForwardCall()', 'E1096:', '', 1, 'FuncWithForwardCall')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001929enddef
1930
Bram Moolenaar4bf10062021-12-28 17:23:12 +00001931def Test_nested_function_with_nextcmd()
Bram Moolenaar9c23f9b2021-12-26 14:23:22 +00001932 var lines =<< trim END
1933 vim9script
1934 # Define an outer function
1935 def FirstFunction()
1936 # Define an inner function
1937 def SecondFunction()
1938 # the function has a body, a double free is detected.
1939 AAAAA
1940
1941 # enddef followed by | or } followed by # one or more characters
1942 enddef|BBBB
1943 enddef
1944
1945 # Compile all functions
1946 defcompile
1947 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001948 v9.CheckScriptFailure(lines, 'E1173: Text found after enddef: BBBB')
Bram Moolenaar9c23f9b2021-12-26 14:23:22 +00001949enddef
1950
Bram Moolenaar4bf10062021-12-28 17:23:12 +00001951def Test_nested_function_with_args_split()
1952 var lines =<< trim END
1953 vim9script
1954 def FirstFunction()
1955 def SecondFunction(
1956 )
1957 # had a double free if the right parenthesis of the nested function is
1958 # on the next line
1959
1960 enddef|BBBB
1961 enddef
1962 # Compile all functions
1963 defcompile
1964 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001965 v9.CheckScriptFailure(lines, 'E1173: Text found after enddef: BBBB')
Bram Moolenaar7473a842021-12-28 17:55:26 +00001966
1967 lines =<< trim END
1968 vim9script
1969 def FirstFunction()
1970 func SecondFunction()
1971 endfunc|BBBB
1972 enddef
1973 defcompile
1974 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001975 v9.CheckScriptFailure(lines, 'E1173: Text found after endfunction: BBBB')
Bram Moolenaar4bf10062021-12-28 17:23:12 +00001976enddef
1977
Bram Moolenaar9f1a39a2022-01-08 15:39:39 +00001978def Test_error_in_function_args()
1979 var lines =<< trim END
1980 def FirstFunction()
1981 def SecondFunction(J =
1982 # Nois
1983 # one
1984
1985 enddef|BBBB
1986 enddef
1987 # Compile all functions
1988 defcompile
1989 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001990 v9.CheckScriptFailure(lines, 'E488:')
Bram Moolenaar9f1a39a2022-01-08 15:39:39 +00001991enddef
1992
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001993def Test_return_type_wrong()
Bram Moolenaar62aec932022-01-29 21:45:34 +00001994 v9.CheckScriptFailure([
Bram Moolenaar5a849da2020-08-08 16:47:30 +02001995 'def Func(): number',
1996 'return "a"',
1997 'enddef',
1998 'defcompile'], 'expected number but got string')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001999 delfunc! g:Func
Bram Moolenaar62aec932022-01-29 21:45:34 +00002000 v9.CheckScriptFailure([
Bram Moolenaar5a849da2020-08-08 16:47:30 +02002001 'def Func(): string',
2002 'return 1',
2003 'enddef',
2004 'defcompile'], 'expected string but got number')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01002005 delfunc! g:Func
Bram Moolenaar62aec932022-01-29 21:45:34 +00002006 v9.CheckScriptFailure([
Bram Moolenaar5a849da2020-08-08 16:47:30 +02002007 'def Func(): void',
2008 'return "a"',
2009 'enddef',
2010 'defcompile'],
2011 'E1096: Returning a value in a function without a return type')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01002012 delfunc! g:Func
Bram Moolenaar62aec932022-01-29 21:45:34 +00002013 v9.CheckScriptFailure([
Bram Moolenaar5a849da2020-08-08 16:47:30 +02002014 'def Func()',
2015 'return "a"',
2016 'enddef',
2017 'defcompile'],
2018 'E1096: Returning a value in a function without a return type')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01002019 delfunc! g:Func
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002020
Bram Moolenaar62aec932022-01-29 21:45:34 +00002021 v9.CheckScriptFailure([
Bram Moolenaar5a849da2020-08-08 16:47:30 +02002022 'def Func(): number',
2023 'return',
2024 'enddef',
2025 'defcompile'], 'E1003:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01002026 delfunc! g:Func
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002027
Bram Moolenaar62aec932022-01-29 21:45:34 +00002028 v9.CheckScriptFailure([
Bram Moolenaar33ea9fd2021-08-08 19:07:37 +02002029 'def Func():number',
2030 'return 123',
2031 'enddef',
2032 'defcompile'], 'E1069:')
2033 delfunc! g:Func
2034
Bram Moolenaar62aec932022-01-29 21:45:34 +00002035 v9.CheckScriptFailure([
Bram Moolenaar33ea9fd2021-08-08 19:07:37 +02002036 'def Func() :number',
2037 'return 123',
2038 'enddef',
2039 'defcompile'], 'E1059:')
2040 delfunc! g:Func
2041
Bram Moolenaar62aec932022-01-29 21:45:34 +00002042 v9.CheckScriptFailure([
Bram Moolenaar33ea9fd2021-08-08 19:07:37 +02002043 'def Func() : number',
2044 'return 123',
2045 'enddef',
2046 'defcompile'], 'E1059:')
2047 delfunc! g:Func
2048
Bram Moolenaar62aec932022-01-29 21:45:34 +00002049 v9.CheckScriptFailure(['def Func(): list', 'return []', 'enddef'], 'E1008:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01002050 delfunc! g:Func
Bram Moolenaar62aec932022-01-29 21:45:34 +00002051 v9.CheckScriptFailure(['def Func(): dict', 'return {}', 'enddef'], 'E1008:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01002052 delfunc! g:Func
Bram Moolenaar62aec932022-01-29 21:45:34 +00002053 v9.CheckScriptFailure(['def Func()', 'return 1'], 'E1057:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01002054 delfunc! g:Func
Bram Moolenaar5a849da2020-08-08 16:47:30 +02002055
Bram Moolenaar62aec932022-01-29 21:45:34 +00002056 v9.CheckScriptFailure([
Bram Moolenaar5a849da2020-08-08 16:47:30 +02002057 'vim9script',
2058 'def FuncB()',
2059 ' return 123',
2060 'enddef',
2061 'def FuncA()',
2062 ' FuncB()',
2063 'enddef',
2064 'defcompile'], 'E1096:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002065enddef
2066
2067def Test_arg_type_wrong()
Bram Moolenaar62aec932022-01-29 21:45:34 +00002068 v9.CheckScriptFailure(['def Func3(items: list)', 'echo "a"', 'enddef'], 'E1008: Missing <type>')
2069 v9.CheckScriptFailure(['def Func4(...)', 'echo "a"', 'enddef'], 'E1055: Missing name after ...')
2070 v9.CheckScriptFailure(['def Func5(items:string)', 'echo "a"'], 'E1069:')
2071 v9.CheckScriptFailure(['def Func5(items)', 'echo "a"'], 'E1077:')
2072 v9.CheckScriptFailure(['def Func6(...x:list<number>)', 'echo "a"', 'enddef'], 'E1069:')
2073 v9.CheckScriptFailure(['def Func7(...x: int)', 'echo "a"', 'enddef'], 'E1010:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002074enddef
2075
Bram Moolenaar86cdb8a2021-04-06 19:01:03 +02002076def Test_white_space_before_comma()
2077 var lines =<< trim END
2078 vim9script
2079 def Func(a: number , b: number)
2080 enddef
2081 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002082 v9.CheckScriptFailure(lines, 'E1068:')
Yegappan Lakshmanan611728f2021-05-24 15:15:47 +02002083 call assert_fails('vim9cmd echo stridx("a" .. "b" , "a")', 'E1068:')
Bram Moolenaar86cdb8a2021-04-06 19:01:03 +02002084enddef
2085
Bram Moolenaar608d78f2021-03-06 22:33:12 +01002086def Test_white_space_after_comma()
2087 var lines =<< trim END
2088 vim9script
2089 def Func(a: number,b: number)
2090 enddef
2091 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002092 v9.CheckScriptFailure(lines, 'E1069:')
Bram Moolenaar608d78f2021-03-06 22:33:12 +01002093
2094 # OK in legacy function
2095 lines =<< trim END
2096 vim9script
2097 func Func(a,b)
2098 endfunc
2099 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002100 v9.CheckScriptSuccess(lines)
Bram Moolenaar608d78f2021-03-06 22:33:12 +01002101enddef
2102
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002103def Test_vim9script_call()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002104 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002105 vim9script
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002106 var name = ''
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002107 def MyFunc(arg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002108 name = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002109 enddef
2110 MyFunc('foobar')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002111 name->assert_equal('foobar')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002112
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002113 var str = 'barfoo'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002114 str->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002115 name->assert_equal('barfoo')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002116
Bram Moolenaar67979662020-06-20 22:50:47 +02002117 g:value = 'value'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002118 g:value->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002119 name->assert_equal('value')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002120
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002121 var listvar = []
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002122 def ListFunc(arg: list<number>)
2123 listvar = arg
2124 enddef
2125 [1, 2, 3]->ListFunc()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002126 listvar->assert_equal([1, 2, 3])
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002127
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002128 var dictvar = {}
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002129 def DictFunc(arg: dict<number>)
2130 dictvar = arg
2131 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +01002132 {a: 1, b: 2}->DictFunc()
2133 dictvar->assert_equal({a: 1, b: 2})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002134 def CompiledDict()
Bram Moolenaare0de1712020-12-02 17:36:54 +01002135 {a: 3, b: 4}->DictFunc()
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002136 enddef
2137 CompiledDict()
Bram Moolenaare0de1712020-12-02 17:36:54 +01002138 dictvar->assert_equal({a: 3, b: 4})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002139
Bram Moolenaare0de1712020-12-02 17:36:54 +01002140 {a: 3, b: 4}->DictFunc()
2141 dictvar->assert_equal({a: 3, b: 4})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002142
2143 ('text')->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002144 name->assert_equal('text')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002145 ("some")->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002146 name->assert_equal('some')
Bram Moolenaare6b53242020-07-01 17:28:33 +02002147
Bram Moolenaar13e12b82020-07-24 18:47:22 +02002148 # line starting with single quote is not a mark
Bram Moolenaar10409562020-07-29 20:00:38 +02002149 # line starting with double quote can be a method call
Bram Moolenaar3d48e252020-07-15 14:15:52 +02002150 'asdfasdf'->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002151 name->assert_equal('asdfasdf')
Bram Moolenaar10409562020-07-29 20:00:38 +02002152 "xyz"->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002153 name->assert_equal('xyz')
Bram Moolenaar3d48e252020-07-15 14:15:52 +02002154
2155 def UseString()
2156 'xyork'->MyFunc()
2157 enddef
2158 UseString()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002159 name->assert_equal('xyork')
Bram Moolenaar3d48e252020-07-15 14:15:52 +02002160
Bram Moolenaar10409562020-07-29 20:00:38 +02002161 def UseString2()
2162 "knife"->MyFunc()
2163 enddef
2164 UseString2()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002165 name->assert_equal('knife')
Bram Moolenaar10409562020-07-29 20:00:38 +02002166
Bram Moolenaar13e12b82020-07-24 18:47:22 +02002167 # prepending a colon makes it a mark
2168 new
2169 setline(1, ['aaa', 'bbb', 'ccc'])
2170 normal! 3Gmt1G
2171 :'t
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002172 getcurpos()[1]->assert_equal(3)
Bram Moolenaar13e12b82020-07-24 18:47:22 +02002173 bwipe!
2174
Bram Moolenaare6b53242020-07-01 17:28:33 +02002175 MyFunc(
2176 'continued'
2177 )
2178 assert_equal('continued',
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002179 name
Bram Moolenaare6b53242020-07-01 17:28:33 +02002180 )
2181
2182 call MyFunc(
2183 'more'
2184 ..
2185 'lines'
2186 )
2187 assert_equal(
2188 'morelines',
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002189 name)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002190 END
2191 writefile(lines, 'Xcall.vim')
2192 source Xcall.vim
2193 delete('Xcall.vim')
2194enddef
2195
2196def Test_vim9script_call_fail_decl()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002197 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002198 vim9script
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002199 var name = ''
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002200 def MyFunc(arg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002201 var name = 123
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002202 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +02002203 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002204 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002205 v9.CheckScriptFailure(lines, 'E1054:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002206enddef
2207
Bram Moolenaar65b95452020-07-19 14:03:09 +02002208def Test_vim9script_call_fail_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002209 var lines =<< trim END
Bram Moolenaar65b95452020-07-19 14:03:09 +02002210 vim9script
2211 def MyFunc(arg: string)
2212 echo arg
2213 enddef
2214 MyFunc(1234)
2215 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002216 v9.CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected string but got number')
Bram Moolenaar65b95452020-07-19 14:03:09 +02002217enddef
2218
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002219def Test_vim9script_call_fail_const()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002220 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002221 vim9script
2222 const var = ''
2223 def MyFunc(arg: string)
2224 var = 'asdf'
2225 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +02002226 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002227 END
2228 writefile(lines, 'Xcall_const.vim')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02002229 assert_fails('source Xcall_const.vim', 'E46:', '', 1, 'MyFunc')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002230 delete('Xcall_const.vim')
Bram Moolenaar3bdc90b2020-12-22 20:35:40 +01002231
2232 lines =<< trim END
2233 const g:Aconst = 77
2234 def Change()
2235 # comment
2236 g:Aconst = 99
2237 enddef
2238 call Change()
2239 unlet g:Aconst
2240 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002241 v9.CheckScriptFailure(lines, 'E741: Value is locked: Aconst', 2)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002242enddef
2243
2244" Test that inside :function a Python function can be defined, :def is not
2245" recognized.
2246func Test_function_python()
2247 CheckFeature python3
Bram Moolenaar727345e2020-09-27 23:33:59 +02002248 let py = 'python3'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002249 execute py "<< EOF"
2250def do_something():
2251 return 1
2252EOF
2253endfunc
2254
2255def Test_delfunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002256 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002257 vim9script
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02002258 def g:GoneSoon()
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002259 echo 'hello'
2260 enddef
2261
2262 def CallGoneSoon()
Bram Moolenaar62aec932022-01-29 21:45:34 +00002263 g:GoneSoon()
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002264 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +02002265 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002266
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02002267 delfunc g:GoneSoon
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002268 CallGoneSoon()
2269 END
2270 writefile(lines, 'XToDelFunc')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02002271 assert_fails('so XToDelFunc', 'E933:', '', 1, 'CallGoneSoon')
2272 assert_fails('so XToDelFunc', 'E933:', '', 1, 'CallGoneSoon')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002273
2274 delete('XToDelFunc')
2275enddef
2276
Bram Moolenaar7509ad82021-12-14 18:14:37 +00002277func Test_free_dict_while_in_funcstack()
2278 " relies on the sleep command
2279 CheckUnix
2280 call Run_Test_free_dict_while_in_funcstack()
2281endfunc
2282
2283def Run_Test_free_dict_while_in_funcstack()
Bram Moolenaar7509ad82021-12-14 18:14:37 +00002284 # this was freeing the TermRun() default argument dictionary while it was
2285 # still referenced in a funcstack_T
2286 var lines =<< trim END
2287 vim9script
2288
2289 &updatetime = 400
2290 def TermRun(_ = {})
2291 def Post()
2292 enddef
2293 def Exec()
2294 term_start('sleep 1', {
2295 term_finish: 'close',
2296 exit_cb: (_, _) => Post(),
2297 })
2298 enddef
2299 Exec()
2300 enddef
2301 nnoremap <F4> <Cmd>call <SID>TermRun()<CR>
2302 timer_start(100, (_) => feedkeys("\<F4>"))
2303 timer_start(1000, (_) => feedkeys("\<F4>"))
2304 sleep 1500m
2305 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002306 v9.CheckScriptSuccess(lines)
Bram Moolenaar7509ad82021-12-14 18:14:37 +00002307 nunmap <F4>
2308 set updatetime&
2309enddef
2310
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002311def Test_redef_failure()
Bram Moolenaard2c61702020-09-06 15:58:36 +02002312 writefile(['def Func0(): string', 'return "Func0"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002313 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +02002314 writefile(['def Func1(): string', 'return "Func1"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002315 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +02002316 writefile(['def! Func0(): string', 'enddef', 'defcompile'], 'Xdef')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02002317 assert_fails('so Xdef', 'E1027:', '', 1, 'Func0')
Bram Moolenaard2c61702020-09-06 15:58:36 +02002318 writefile(['def Func2(): string', 'return "Func2"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002319 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +02002320 delete('Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002321
Bram Moolenaar701cc6c2021-04-10 13:33:48 +02002322 assert_fails('g:Func0()', 'E1091:')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002323 g:Func1()->assert_equal('Func1')
2324 g:Func2()->assert_equal('Func2')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002325
2326 delfunc! Func0
2327 delfunc! Func1
2328 delfunc! Func2
2329enddef
2330
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +02002331def Test_vim9script_func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002332 var lines =<< trim END
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +02002333 vim9script
2334 func Func(arg)
2335 echo a:arg
2336 endfunc
2337 Func('text')
2338 END
2339 writefile(lines, 'XVim9Func')
2340 so XVim9Func
2341
2342 delete('XVim9Func')
2343enddef
2344
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002345let s:funcResult = 0
2346
2347def FuncNoArgNoRet()
Bram Moolenaar53900992020-08-22 19:02:02 +02002348 s:funcResult = 11
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002349enddef
2350
2351def FuncNoArgRetNumber(): number
Bram Moolenaar53900992020-08-22 19:02:02 +02002352 s:funcResult = 22
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002353 return 1234
2354enddef
2355
Bram Moolenaarec5929d2020-04-07 20:53:39 +02002356def FuncNoArgRetString(): string
Bram Moolenaar53900992020-08-22 19:02:02 +02002357 s:funcResult = 45
Bram Moolenaarec5929d2020-04-07 20:53:39 +02002358 return 'text'
2359enddef
2360
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002361def FuncOneArgNoRet(arg: number)
Bram Moolenaar53900992020-08-22 19:02:02 +02002362 s:funcResult = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002363enddef
2364
2365def FuncOneArgRetNumber(arg: number): number
Bram Moolenaar53900992020-08-22 19:02:02 +02002366 s:funcResult = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002367 return arg
2368enddef
2369
Bram Moolenaar08938ee2020-04-11 23:17:17 +02002370def FuncTwoArgNoRet(one: bool, two: number)
Bram Moolenaar53900992020-08-22 19:02:02 +02002371 s:funcResult = two
Bram Moolenaar08938ee2020-04-11 23:17:17 +02002372enddef
2373
Bram Moolenaar62aec932022-01-29 21:45:34 +00002374def s:FuncOneArgRetString(arg: string): string
Bram Moolenaarec5929d2020-04-07 20:53:39 +02002375 return arg
2376enddef
2377
Bram Moolenaar62aec932022-01-29 21:45:34 +00002378def s:FuncOneArgRetAny(arg: any): any
Bram Moolenaar89228602020-04-05 22:14:54 +02002379 return arg
2380enddef
2381
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002382def Test_func_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002383 var Ref1: func()
Bram Moolenaar53900992020-08-22 19:02:02 +02002384 s:funcResult = 0
Bram Moolenaar62aec932022-01-29 21:45:34 +00002385 Ref1 = g:FuncNoArgNoRet
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002386 Ref1()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002387 s:funcResult->assert_equal(11)
Bram Moolenaar4c683752020-04-05 21:38:23 +02002388
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002389 var Ref2: func
Bram Moolenaar53900992020-08-22 19:02:02 +02002390 s:funcResult = 0
Bram Moolenaar62aec932022-01-29 21:45:34 +00002391 Ref2 = g:FuncNoArgNoRet
Bram Moolenaar4c683752020-04-05 21:38:23 +02002392 Ref2()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002393 s:funcResult->assert_equal(11)
Bram Moolenaar4c683752020-04-05 21:38:23 +02002394
Bram Moolenaar53900992020-08-22 19:02:02 +02002395 s:funcResult = 0
Bram Moolenaar62aec932022-01-29 21:45:34 +00002396 Ref2 = g:FuncOneArgNoRet
Bram Moolenaar4c683752020-04-05 21:38:23 +02002397 Ref2(12)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002398 s:funcResult->assert_equal(12)
Bram Moolenaar4c683752020-04-05 21:38:23 +02002399
Bram Moolenaar53900992020-08-22 19:02:02 +02002400 s:funcResult = 0
Bram Moolenaar62aec932022-01-29 21:45:34 +00002401 Ref2 = g:FuncNoArgRetNumber
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002402 Ref2()->assert_equal(1234)
2403 s:funcResult->assert_equal(22)
Bram Moolenaar4c683752020-04-05 21:38:23 +02002404
Bram Moolenaar53900992020-08-22 19:02:02 +02002405 s:funcResult = 0
Bram Moolenaar848fadd2022-01-30 15:28:30 +00002406 Ref2 = g:FuncOneArgRetNumber
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002407 Ref2(13)->assert_equal(13)
2408 s:funcResult->assert_equal(13)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002409enddef
2410
Bram Moolenaar9978d472020-07-05 16:01:56 +02002411def Test_repeat_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002412 var res = 0
Bram Moolenaar9978d472020-07-05 16:01:56 +02002413 for n in repeat([1], 3)
2414 res += n
2415 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002416 res->assert_equal(3)
Bram Moolenaarfce82b32020-07-05 16:07:21 +02002417
2418 res = 0
2419 for n in add([1, 2], 3)
2420 res += n
2421 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002422 res->assert_equal(6)
Bram Moolenaar9978d472020-07-05 16:01:56 +02002423enddef
2424
Bram Moolenaar846178a2020-07-05 17:04:13 +02002425def Test_argv_return_type()
2426 next fileone filetwo
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002427 var res = ''
Bram Moolenaar846178a2020-07-05 17:04:13 +02002428 for name in argv()
2429 res ..= name
2430 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002431 res->assert_equal('fileonefiletwo')
Bram Moolenaar846178a2020-07-05 17:04:13 +02002432enddef
2433
Bram Moolenaarec5929d2020-04-07 20:53:39 +02002434def Test_func_type_part()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002435 var RefVoid: func: void
Bram Moolenaar62aec932022-01-29 21:45:34 +00002436 RefVoid = g:FuncNoArgNoRet
2437 RefVoid = g:FuncOneArgNoRet
2438 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 +00002439 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 +02002440
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002441 var RefAny: func(): any
Bram Moolenaar62aec932022-01-29 21:45:34 +00002442 RefAny = g:FuncNoArgRetNumber
Bram Moolenaar848fadd2022-01-30 15:28:30 +00002443 RefAny = g:FuncNoArgRetString
Bram Moolenaar62aec932022-01-29 21:45:34 +00002444 v9.CheckDefFailure(['var RefAny: func(): any', 'RefAny = g:FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(): any but got func()')
2445 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 +02002446
Bram Moolenaar6abd3dc2020-10-04 14:17:32 +02002447 var RefAnyNoArgs: func: any = RefAny
2448
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002449 var RefNr: func: number
Bram Moolenaar62aec932022-01-29 21:45:34 +00002450 RefNr = g:FuncNoArgRetNumber
Bram Moolenaar848fadd2022-01-30 15:28:30 +00002451 RefNr = g:FuncOneArgRetNumber
Bram Moolenaar62aec932022-01-29 21:45:34 +00002452 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 +00002453 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 +02002454
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002455 var RefStr: func: string
Bram Moolenaar848fadd2022-01-30 15:28:30 +00002456 RefStr = g:FuncNoArgRetString
Bram Moolenaarec5929d2020-04-07 20:53:39 +02002457 RefStr = FuncOneArgRetString
Bram Moolenaar62aec932022-01-29 21:45:34 +00002458 v9.CheckDefFailure(['var RefStr: func: string', 'RefStr = g:FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(...): string but got func()')
2459 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 +02002460enddef
2461
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002462def Test_func_type_fails()
Bram Moolenaar62aec932022-01-29 21:45:34 +00002463 v9.CheckDefFailure(['var ref1: func()'], 'E704:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002464
Bram Moolenaar62aec932022-01-29 21:45:34 +00002465 v9.CheckDefFailure(['var Ref1: func()', 'Ref1 = g:FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func() but got func(): number')
2466 v9.CheckDefFailure(['var Ref1: func()', 'Ref1 = g:FuncOneArgNoRet'], 'E1012: Type mismatch; expected func() but got func(number)')
Bram Moolenaar848fadd2022-01-30 15:28:30 +00002467 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 +00002468 v9.CheckDefFailure(['var Ref1: func(bool)', 'Ref1 = g:FuncTwoArgNoRet'], 'E1012: Type mismatch; expected func(bool) but got func(bool, number)')
2469 v9.CheckDefFailure(['var Ref1: func(?bool)', 'Ref1 = g:FuncTwoArgNoRet'], 'E1012: Type mismatch; expected func(?bool) but got func(bool, number)')
2470 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 +02002471
Bram Moolenaar62aec932022-01-29 21:45:34 +00002472 v9.CheckDefFailure(['var RefWrong: func(string ,number)'], 'E1068:')
2473 v9.CheckDefFailure(['var RefWrong: func(string,number)'], 'E1069:')
2474 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:')
2475 v9.CheckDefFailure(['var RefWrong: func(bool):string'], 'E1069:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002476enddef
2477
Bram Moolenaar89228602020-04-05 22:14:54 +02002478def Test_func_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002479 var nr: number
Bram Moolenaar62aec932022-01-29 21:45:34 +00002480 nr = g:FuncNoArgRetNumber()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002481 nr->assert_equal(1234)
Bram Moolenaar89228602020-04-05 22:14:54 +02002482
2483 nr = FuncOneArgRetAny(122)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002484 nr->assert_equal(122)
Bram Moolenaar89228602020-04-05 22:14:54 +02002485
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002486 var str: string
Bram Moolenaar89228602020-04-05 22:14:54 +02002487 str = FuncOneArgRetAny('yes')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002488 str->assert_equal('yes')
Bram Moolenaar89228602020-04-05 22:14:54 +02002489
Bram Moolenaar62aec932022-01-29 21:45:34 +00002490 v9.CheckDefFailure(['var str: string', 'str = g:FuncNoArgRetNumber()'], 'E1012: Type mismatch; expected string but got number')
Bram Moolenaar89228602020-04-05 22:14:54 +02002491enddef
2492
Bram Moolenaar6abd3dc2020-10-04 14:17:32 +02002493def Test_func_common_type()
2494 def FuncOne(n: number): number
2495 return n
2496 enddef
2497 def FuncTwo(s: string): number
2498 return len(s)
2499 enddef
2500 def FuncThree(n: number, s: string): number
2501 return n + len(s)
2502 enddef
2503 var list = [FuncOne, FuncTwo, FuncThree]
2504 assert_equal(8, list[0](8))
2505 assert_equal(4, list[1]('word'))
2506 assert_equal(7, list[2](3, 'word'))
2507enddef
2508
Bram Moolenaar62aec932022-01-29 21:45:34 +00002509def s:MultiLine(
Bram Moolenaar5e774c72020-04-12 21:53:00 +02002510 arg1: string,
2511 arg2 = 1234,
2512 ...rest: list<string>
2513 ): string
2514 return arg1 .. arg2 .. join(rest, '-')
2515enddef
2516
Bram Moolenaar2c330432020-04-13 14:41:35 +02002517def MultiLineComment(
2518 arg1: string, # comment
2519 arg2 = 1234, # comment
2520 ...rest: list<string> # comment
2521 ): string # comment
2522 return arg1 .. arg2 .. join(rest, '-')
2523enddef
2524
Bram Moolenaar5e774c72020-04-12 21:53:00 +02002525def Test_multiline()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002526 MultiLine('text')->assert_equal('text1234')
2527 MultiLine('text', 777)->assert_equal('text777')
2528 MultiLine('text', 777, 'one')->assert_equal('text777one')
2529 MultiLine('text', 777, 'one', 'two')->assert_equal('text777one-two')
Bram Moolenaar5e774c72020-04-12 21:53:00 +02002530enddef
2531
Bram Moolenaar23e03252020-04-12 22:22:31 +02002532func Test_multiline_not_vim9()
Bram Moolenaar62aec932022-01-29 21:45:34 +00002533 call s:MultiLine('text')->assert_equal('text1234')
2534 call s:MultiLine('text', 777)->assert_equal('text777')
2535 call s:MultiLine('text', 777, 'one')->assert_equal('text777one')
2536 call s:MultiLine('text', 777, 'one', 'two')->assert_equal('text777one-two')
Bram Moolenaar23e03252020-04-12 22:22:31 +02002537endfunc
2538
Bram Moolenaar5e774c72020-04-12 21:53:00 +02002539
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02002540" When using CheckScriptFailure() for the below test, E1010 is generated instead
2541" of E1056.
2542func Test_E1056_1059()
2543 let caught_1056 = 0
2544 try
2545 def F():
2546 return 1
2547 enddef
2548 catch /E1056:/
2549 let caught_1056 = 1
2550 endtry
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002551 eval caught_1056->assert_equal(1)
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02002552
2553 let caught_1059 = 0
2554 try
2555 def F5(items : list)
2556 echo 'a'
2557 enddef
2558 catch /E1059:/
2559 let caught_1059 = 1
2560 endtry
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002561 eval caught_1059->assert_equal(1)
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02002562endfunc
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002563
Bram Moolenaar015f4262020-05-05 21:25:22 +02002564func DelMe()
2565 echo 'DelMe'
2566endfunc
2567
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02002568def Test_error_reporting()
2569 # comment lines at the start of the function
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002570 var lines =<< trim END
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02002571 " comment
2572 def Func()
2573 # comment
2574 # comment
2575 invalid
2576 enddef
2577 defcompile
2578 END
Bram Moolenaar08052222020-09-14 17:04:31 +02002579 writefile(lines, 'Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02002580 try
2581 source Xdef
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02002582 assert_report('should have failed')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02002583 catch /E476:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002584 v:exception->assert_match('Invalid command: invalid')
2585 v:throwpoint->assert_match(', line 3$')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02002586 endtry
Bram Moolenaar2d870f82020-12-05 13:41:01 +01002587 delfunc! g:Func
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02002588
2589 # comment lines after the start of the function
2590 lines =<< trim END
2591 " comment
2592 def Func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002593 var x = 1234
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02002594 # comment
2595 # comment
2596 invalid
2597 enddef
2598 defcompile
2599 END
Bram Moolenaar08052222020-09-14 17:04:31 +02002600 writefile(lines, 'Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02002601 try
2602 source Xdef
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02002603 assert_report('should have failed')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02002604 catch /E476:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002605 v:exception->assert_match('Invalid command: invalid')
2606 v:throwpoint->assert_match(', line 4$')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02002607 endtry
Bram Moolenaar2d870f82020-12-05 13:41:01 +01002608 delfunc! g:Func
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02002609
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02002610 lines =<< trim END
2611 vim9script
2612 def Func()
Bram Moolenaare0de1712020-12-02 17:36:54 +01002613 var db = {foo: 1, bar: 2}
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02002614 # comment
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002615 var x = db.asdf
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02002616 enddef
2617 defcompile
2618 Func()
2619 END
Bram Moolenaar08052222020-09-14 17:04:31 +02002620 writefile(lines, 'Xdef')
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02002621 try
2622 source Xdef
2623 assert_report('should have failed')
2624 catch /E716:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002625 v:throwpoint->assert_match('_Func, line 3$')
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02002626 endtry
Bram Moolenaar2d870f82020-12-05 13:41:01 +01002627 delfunc! g:Func
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02002628
Bram Moolenaar08052222020-09-14 17:04:31 +02002629 delete('Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02002630enddef
2631
Bram Moolenaar015f4262020-05-05 21:25:22 +02002632def Test_deleted_function()
Bram Moolenaar62aec932022-01-29 21:45:34 +00002633 v9.CheckDefExecFailure([
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002634 'var RefMe: func = function("g:DelMe")',
Bram Moolenaar015f4262020-05-05 21:25:22 +02002635 'delfunc g:DelMe',
2636 'echo RefMe()'], 'E117:')
2637enddef
2638
2639def Test_unknown_function()
Bram Moolenaar62aec932022-01-29 21:45:34 +00002640 v9.CheckDefExecFailure([
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002641 'var Ref: func = function("NotExist")',
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +02002642 'delfunc g:NotExist'], 'E700:')
Bram Moolenaar015f4262020-05-05 21:25:22 +02002643enddef
2644
Bram Moolenaar62aec932022-01-29 21:45:34 +00002645def s:RefFunc(Ref: func(any): any): string
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +02002646 return Ref('more')
2647enddef
2648
2649def Test_closure_simple()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002650 var local = 'some '
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002651 RefFunc((s) => local .. s)->assert_equal('some more')
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +02002652enddef
2653
Bram Moolenaar62aec932022-01-29 21:45:34 +00002654def s:MakeRef()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002655 var local = 'some '
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002656 g:Ref = (s) => local .. s
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02002657enddef
2658
2659def Test_closure_ref_after_return()
2660 MakeRef()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002661 g:Ref('thing')->assert_equal('some thing')
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02002662 unlet g:Ref
2663enddef
2664
Bram Moolenaar62aec932022-01-29 21:45:34 +00002665def s:MakeTwoRefs()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002666 var local = ['some']
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002667 g:Extend = (s) => local->add(s)
2668 g:Read = () => local
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02002669enddef
2670
2671def Test_closure_two_refs()
2672 MakeTwoRefs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002673 join(g:Read(), ' ')->assert_equal('some')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02002674 g:Extend('more')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002675 join(g:Read(), ' ')->assert_equal('some more')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02002676 g:Extend('even')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002677 join(g:Read(), ' ')->assert_equal('some more even')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02002678
2679 unlet g:Extend
2680 unlet g:Read
2681enddef
2682
Bram Moolenaar62aec932022-01-29 21:45:34 +00002683def s:ReadRef(Ref: func(): list<string>): string
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02002684 return join(Ref(), ' ')
2685enddef
2686
Bram Moolenaar62aec932022-01-29 21:45:34 +00002687def s:ExtendRef(Ref: func(string): list<string>, add: string)
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02002688 Ref(add)
2689enddef
2690
2691def Test_closure_two_indirect_refs()
Bram Moolenaarf7779c62020-05-03 15:38:16 +02002692 MakeTwoRefs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002693 ReadRef(g:Read)->assert_equal('some')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02002694 ExtendRef(g:Extend, 'more')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002695 ReadRef(g:Read)->assert_equal('some more')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02002696 ExtendRef(g:Extend, 'even')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002697 ReadRef(g:Read)->assert_equal('some more even')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02002698
2699 unlet g:Extend
2700 unlet g:Read
2701enddef
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02002702
Bram Moolenaar62aec932022-01-29 21:45:34 +00002703def s:MakeArgRefs(theArg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002704 var local = 'loc_val'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002705 g:UseArg = (s) => theArg .. '/' .. local .. '/' .. s
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02002706enddef
2707
Bram Moolenaar62aec932022-01-29 21:45:34 +00002708def s:MakeArgRefsVarargs(theArg: string, ...rest: list<string>)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002709 var local = 'the_loc'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002710 g:UseVararg = (s) => theArg .. '/' .. local .. '/' .. s .. '/' .. join(rest)
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02002711enddef
2712
2713def Test_closure_using_argument()
2714 MakeArgRefs('arg_val')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002715 g:UseArg('call_val')->assert_equal('arg_val/loc_val/call_val')
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02002716
2717 MakeArgRefsVarargs('arg_val', 'one', 'two')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002718 g:UseVararg('call_val')->assert_equal('arg_val/the_loc/call_val/one two')
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02002719
2720 unlet g:UseArg
2721 unlet g:UseVararg
Bram Moolenaar44ec21c2021-02-12 21:50:57 +01002722
2723 var lines =<< trim END
2724 vim9script
2725 def Test(Fun: func(number): number): list<number>
2726 return map([1, 2, 3], (_, i) => Fun(i))
2727 enddef
2728 def Inc(nr: number): number
2729 return nr + 2
2730 enddef
2731 assert_equal([3, 4, 5], Test(Inc))
2732 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002733 v9.CheckScriptSuccess(lines)
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02002734enddef
2735
Bram Moolenaar62aec932022-01-29 21:45:34 +00002736def s:MakeGetAndAppendRefs()
Bram Moolenaar85d5e2b2020-10-10 14:13:01 +02002737 var local = 'a'
2738
2739 def Append(arg: string)
2740 local ..= arg
2741 enddef
2742 g:Append = Append
2743
2744 def Get(): string
2745 return local
2746 enddef
2747 g:Get = Get
2748enddef
2749
2750def Test_closure_append_get()
2751 MakeGetAndAppendRefs()
2752 g:Get()->assert_equal('a')
2753 g:Append('-b')
2754 g:Get()->assert_equal('a-b')
2755 g:Append('-c')
2756 g:Get()->assert_equal('a-b-c')
2757
2758 unlet g:Append
2759 unlet g:Get
2760enddef
Bram Moolenaarb68b3462020-05-06 21:06:30 +02002761
Bram Moolenaar04b12692020-05-04 23:24:44 +02002762def Test_nested_closure()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002763 var local = 'text'
Bram Moolenaar04b12692020-05-04 23:24:44 +02002764 def Closure(arg: string): string
2765 return local .. arg
2766 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002767 Closure('!!!')->assert_equal('text!!!')
Bram Moolenaar04b12692020-05-04 23:24:44 +02002768enddef
2769
Bram Moolenaar62aec932022-01-29 21:45:34 +00002770func s:GetResult(Ref)
Bram Moolenaar6f5b6df2020-05-16 21:20:12 +02002771 return a:Ref('some')
2772endfunc
2773
2774def Test_call_closure_not_compiled()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002775 var text = 'text'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002776 g:Ref = (s) => s .. text
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002777 GetResult(g:Ref)->assert_equal('sometext')
Bram Moolenaar6f5b6df2020-05-16 21:20:12 +02002778enddef
2779
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02002780def Test_double_closure_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002781 var lines =<< trim END
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02002782 vim9script
2783 def Func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002784 var name = 0
2785 for i in range(2)
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002786 timer_start(0, () => name)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002787 endfor
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02002788 enddef
2789 Func()
2790 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002791 v9.CheckScriptSuccess(lines)
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02002792enddef
2793
Bram Moolenaar85d5e2b2020-10-10 14:13:01 +02002794def Test_nested_closure_used()
2795 var lines =<< trim END
2796 vim9script
2797 def Func()
2798 var x = 'hello'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002799 var Closure = () => x
2800 g:Myclosure = () => Closure()
Bram Moolenaar85d5e2b2020-10-10 14:13:01 +02002801 enddef
2802 Func()
2803 assert_equal('hello', g:Myclosure())
2804 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002805 v9.CheckScriptSuccess(lines)
Bram Moolenaar85d5e2b2020-10-10 14:13:01 +02002806enddef
Bram Moolenaar0876c782020-10-07 19:08:04 +02002807
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02002808def Test_nested_closure_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002809 var lines =<< trim END
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02002810 vim9script
2811 def FuncA()
2812 FuncB(0)
2813 enddef
2814 def FuncB(n: number): list<string>
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002815 return map([0], (_, v) => n)
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02002816 enddef
2817 FuncA()
2818 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002819 v9.CheckScriptFailure(lines, 'E1012:')
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02002820enddef
2821
Bram Moolenaarf112f302020-12-20 17:47:52 +01002822def Test_global_closure()
2823 var lines =<< trim END
2824 vim9script
2825 def ReverseEveryNLines(n: number, line1: number, line2: number)
2826 var mods = 'sil keepj keepp lockm '
2827 var range = ':' .. line1 .. ',' .. line2
2828 def g:Offset(): number
2829 var offset = (line('.') - line1 + 1) % n
2830 return offset != 0 ? offset : n
2831 enddef
2832 exe mods .. range .. 'g/^/exe "m .-" .. g:Offset()'
2833 enddef
2834
2835 new
2836 repeat(['aaa', 'bbb', 'ccc'], 3)->setline(1)
2837 ReverseEveryNLines(3, 1, 9)
2838 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002839 v9.CheckScriptSuccess(lines)
Bram Moolenaarf112f302020-12-20 17:47:52 +01002840 var expected = repeat(['ccc', 'bbb', 'aaa'], 3)
2841 assert_equal(expected, getline(1, 9))
2842 bwipe!
2843enddef
2844
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01002845def Test_global_closure_called_directly()
2846 var lines =<< trim END
2847 vim9script
2848 def Outer()
2849 var x = 1
2850 def g:Inner()
2851 var y = x
2852 x += 1
2853 assert_equal(1, y)
2854 enddef
2855 g:Inner()
2856 assert_equal(2, x)
2857 enddef
2858 Outer()
2859 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002860 v9.CheckScriptSuccess(lines)
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01002861 delfunc g:Inner
2862enddef
2863
Bram Moolenaar69c76172021-12-02 16:38:52 +00002864def Test_closure_called_from_legacy()
2865 var lines =<< trim END
2866 vim9script
2867 def Func()
2868 var outer = 'foo'
2869 var F = () => {
2870 outer = 'bar'
2871 }
2872 execute printf('call %s()', string(F))
2873 enddef
2874 Func()
2875 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002876 v9.CheckScriptFailure(lines, 'E1248')
Bram Moolenaar69c76172021-12-02 16:38:52 +00002877enddef
2878
Bram Moolenaar34c54eb2020-11-25 19:15:19 +01002879def Test_failure_in_called_function()
2880 # this was using the frame index as the return value
2881 var lines =<< trim END
2882 vim9script
2883 au TerminalWinOpen * eval [][0]
2884 def PopupTerm(a: any)
2885 # make sure typvals on stack are string
2886 ['a', 'b', 'c', 'd', 'e', 'f', 'g']->join()
2887 FireEvent()
2888 enddef
2889 def FireEvent()
2890 do TerminalWinOpen
2891 enddef
2892 # use try/catch to make eval fail
2893 try
2894 call PopupTerm(0)
2895 catch
2896 endtry
2897 au! TerminalWinOpen
2898 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002899 v9.CheckScriptSuccess(lines)
Bram Moolenaar34c54eb2020-11-25 19:15:19 +01002900enddef
2901
Bram Moolenaar5366e1a2020-10-01 13:01:34 +02002902def Test_nested_lambda()
2903 var lines =<< trim END
2904 vim9script
2905 def Func()
2906 var x = 4
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002907 var Lambda1 = () => 7
2908 var Lambda2 = () => [Lambda1(), x]
Bram Moolenaar5366e1a2020-10-01 13:01:34 +02002909 var res = Lambda2()
2910 assert_equal([7, 4], res)
2911 enddef
2912 Func()
2913 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002914 v9.CheckScriptSuccess(lines)
Bram Moolenaar5366e1a2020-10-01 13:01:34 +02002915enddef
2916
Bram Moolenaarc04f2a42021-06-09 19:30:03 +02002917def Test_double_nested_lambda()
2918 var lines =<< trim END
2919 vim9script
2920 def F(head: string): func(string): func(string): string
2921 return (sep: string): func(string): string => ((tail: string): string => {
2922 return head .. sep .. tail
2923 })
2924 enddef
2925 assert_equal('hello-there', F('hello')('-')('there'))
2926 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002927 v9.CheckScriptSuccess(lines)
Bram Moolenaarc04f2a42021-06-09 19:30:03 +02002928enddef
2929
Bram Moolenaar074f84c2021-05-18 11:47:44 +02002930def Test_nested_inline_lambda()
Bram Moolenaar074f84c2021-05-18 11:47:44 +02002931 var lines =<< trim END
2932 vim9script
2933 def F(text: string): func(string): func(string): string
2934 return (arg: string): func(string): string => ((sep: string): string => {
Bram Moolenaar23e2e112021-08-03 21:16:18 +02002935 return sep .. arg .. text
Bram Moolenaar074f84c2021-05-18 11:47:44 +02002936 })
2937 enddef
Bram Moolenaar23e2e112021-08-03 21:16:18 +02002938 assert_equal('--there++', F('++')('there')('--'))
Bram Moolenaar074f84c2021-05-18 11:47:44 +02002939 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002940 v9.CheckScriptSuccess(lines)
Bram Moolenaar5245beb2021-07-15 22:03:50 +02002941
2942 lines =<< trim END
2943 vim9script
2944 echo range(4)->mapnew((_, v) => {
2945 return range(v) ->mapnew((_, s) => {
2946 return string(s)
2947 })
2948 })
2949 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002950 v9.CheckScriptSuccess(lines)
Bram Moolenaarc6ba2f92021-07-18 13:42:29 +02002951
2952 lines =<< trim END
2953 vim9script
2954
Bram Moolenaara749a422022-02-12 19:52:25 +00002955 def Func()
Bram Moolenaarc6ba2f92021-07-18 13:42:29 +02002956 range(10)
2957 ->mapnew((_, _) => ({
2958 key: range(10)->mapnew((_, _) => {
2959 return ' '
2960 }),
2961 }))
2962 enddef
2963
2964 defcomp
2965 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002966 v9.CheckScriptSuccess(lines)
Bram Moolenaar074f84c2021-05-18 11:47:44 +02002967enddef
2968
Bram Moolenaar52bf81c2020-11-17 18:50:44 +01002969def Shadowed(): list<number>
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002970 var FuncList: list<func: number> = [() => 42]
Bram Moolenaar75ab91f2021-01-10 22:42:50 +01002971 return FuncList->mapnew((_, Shadowed) => Shadowed())
Bram Moolenaar52bf81c2020-11-17 18:50:44 +01002972enddef
2973
2974def Test_lambda_arg_shadows_func()
Bram Moolenaar62aec932022-01-29 21:45:34 +00002975 assert_equal([42], g:Shadowed())
Bram Moolenaar52bf81c2020-11-17 18:50:44 +01002976enddef
2977
Bram Moolenaar62aec932022-01-29 21:45:34 +00002978def s:Line_continuation_in_def(dir: string = ''): string
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002979 var path: string = empty(dir)
2980 \ ? 'empty'
2981 \ : 'full'
2982 return path
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02002983enddef
2984
2985def Test_line_continuation_in_def()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002986 Line_continuation_in_def('.')->assert_equal('full')
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02002987enddef
2988
Bram Moolenaar2ea95b62020-11-19 21:47:56 +01002989def Test_script_var_in_lambda()
2990 var lines =<< trim END
2991 vim9script
2992 var script = 'test'
Bram Moolenaarbb8a7ce2021-04-10 20:10:26 +02002993 assert_equal(['test'], map(['one'], (_, _) => script))
Bram Moolenaar2ea95b62020-11-19 21:47:56 +01002994 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002995 v9.CheckScriptSuccess(lines)
Bram Moolenaar2ea95b62020-11-19 21:47:56 +01002996enddef
2997
Bram Moolenaar62aec932022-01-29 21:45:34 +00002998def s:Line_continuation_in_lambda(): list<string>
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002999 var x = range(97, 100)
Bram Moolenaar75ab91f2021-01-10 22:42:50 +01003000 ->mapnew((_, v) => nr2char(v)
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01003001 ->toupper())
Bram Moolenaar7a4b8982020-07-08 17:36:21 +02003002 ->reverse()
3003 return x
3004enddef
3005
3006def Test_line_continuation_in_lambda()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02003007 Line_continuation_in_lambda()->assert_equal(['D', 'C', 'B', 'A'])
Bram Moolenaarf898f7c2021-01-16 18:09:52 +01003008
3009 var lines =<< trim END
3010 vim9script
3011 var res = [{n: 1, m: 2, s: 'xxx'}]
3012 ->mapnew((_, v: dict<any>): string => printf('%d:%d:%s',
3013 v.n,
3014 v.m,
3015 substitute(v.s, '.*', 'yyy', '')
3016 ))
3017 assert_equal(['1:2:yyy'], res)
3018 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003019 v9.CheckScriptSuccess(lines)
Bram Moolenaar7a4b8982020-07-08 17:36:21 +02003020enddef
3021
Bram Moolenaarb6571982021-01-08 22:24:19 +01003022def Test_list_lambda()
3023 timer_start(1000, (_) => 0)
3024 var body = execute(timer_info()[0].callback
3025 ->string()
3026 ->substitute("('", ' ', '')
3027 ->substitute("')", '', '')
3028 ->substitute('function\zs', ' ', ''))
Bram Moolenaar767034c2021-04-09 17:24:52 +02003029 assert_match('def <lambda>\d\+(_: any): number\n1 return 0\n enddef', body)
Bram Moolenaarb6571982021-01-08 22:24:19 +01003030enddef
3031
Bram Moolenaar3c77b6a2021-07-25 18:07:00 +02003032def Test_lambda_block_variable()
Bram Moolenaar88421d62021-07-24 14:14:52 +02003033 var lines =<< trim END
3034 vim9script
3035 var flist: list<func>
3036 for i in range(10)
3037 var inloop = i
3038 flist[i] = () => inloop
3039 endfor
3040 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003041 v9.CheckScriptSuccess(lines)
Bram Moolenaar88421d62021-07-24 14:14:52 +02003042
3043 lines =<< trim END
3044 vim9script
3045 if true
3046 var outloop = 5
3047 var flist: list<func>
3048 for i in range(10)
3049 flist[i] = () => outloop
3050 endfor
3051 endif
3052 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003053 v9.CheckScriptSuccess(lines)
Bram Moolenaar88421d62021-07-24 14:14:52 +02003054
3055 lines =<< trim END
3056 vim9script
3057 if true
3058 var outloop = 5
3059 endif
3060 var flist: list<func>
3061 for i in range(10)
3062 flist[i] = () => outloop
3063 endfor
3064 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003065 v9.CheckScriptFailure(lines, 'E1001: Variable not found: outloop', 1)
Bram Moolenaar3c77b6a2021-07-25 18:07:00 +02003066
3067 lines =<< trim END
3068 vim9script
3069 for i in range(10)
3070 var Ref = () => 0
3071 endfor
3072 assert_equal(0, ((i) => 0)(0))
3073 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003074 v9.CheckScriptSuccess(lines)
Bram Moolenaar88421d62021-07-24 14:14:52 +02003075enddef
3076
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02003077def Test_legacy_lambda()
3078 legacy echo {x -> 'hello ' .. x}('foo')
Bram Moolenaardc4c2302021-04-25 13:54:42 +02003079
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02003080 var lines =<< trim END
3081 echo {x -> 'hello ' .. x}('foo')
3082 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003083 v9.CheckDefAndScriptFailure(lines, 'E720:')
Bram Moolenaardc4c2302021-04-25 13:54:42 +02003084
3085 lines =<< trim END
3086 vim9script
3087 def Func()
3088 echo (() => 'no error')()
3089 enddef
3090 legacy call s:Func()
3091 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003092 v9.CheckScriptSuccess(lines)
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02003093enddef
3094
Bram Moolenaarce024c32021-06-26 13:00:49 +02003095def Test_legacy()
3096 var lines =<< trim END
3097 vim9script
3098 func g:LegacyFunction()
3099 let g:legacyvar = 1
3100 endfunc
3101 def Testit()
3102 legacy call g:LegacyFunction()
3103 enddef
3104 Testit()
3105 assert_equal(1, g:legacyvar)
3106 unlet g:legacyvar
3107 delfunc g:LegacyFunction
3108 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003109 v9.CheckScriptSuccess(lines)
Bram Moolenaarce024c32021-06-26 13:00:49 +02003110enddef
3111
Bram Moolenaarc3cb1c92021-06-02 16:47:53 +02003112def Test_legacy_errors()
3113 for cmd in ['if', 'elseif', 'else', 'endif',
3114 'for', 'endfor', 'continue', 'break',
3115 'while', 'endwhile',
3116 'try', 'catch', 'finally', 'endtry']
Bram Moolenaar62aec932022-01-29 21:45:34 +00003117 v9.CheckDefFailure(['legacy ' .. cmd .. ' expr'], 'E1189:')
Bram Moolenaarc3cb1c92021-06-02 16:47:53 +02003118 endfor
3119enddef
3120
Bram Moolenaarb1b6f4d2021-09-13 18:25:54 +02003121def Test_call_legacy_with_dict()
3122 var lines =<< trim END
3123 vim9script
3124 func Legacy() dict
3125 let g:result = self.value
3126 endfunc
3127 def TestDirect()
3128 var d = {value: 'yes', func: Legacy}
3129 d.func()
3130 enddef
3131 TestDirect()
3132 assert_equal('yes', g:result)
3133 unlet g:result
3134
3135 def TestIndirect()
3136 var d = {value: 'foo', func: Legacy}
3137 var Fi = d.func
3138 Fi()
3139 enddef
3140 TestIndirect()
3141 assert_equal('foo', g:result)
3142 unlet g:result
3143
3144 var d = {value: 'bar', func: Legacy}
3145 d.func()
3146 assert_equal('bar', g:result)
3147 unlet g:result
3148 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003149 v9.CheckScriptSuccess(lines)
Bram Moolenaarb1b6f4d2021-09-13 18:25:54 +02003150enddef
3151
Bram Moolenaar62aec932022-01-29 21:45:34 +00003152def s:DoFilterThis(a: string): list<string>
Bram Moolenaarab360522021-01-10 14:02:28 +01003153 # closure nested inside another closure using argument
3154 var Filter = (l) => filter(l, (_, v) => stridx(v, a) == 0)
3155 return ['x', 'y', 'a', 'x2', 'c']->Filter()
3156enddef
3157
3158def Test_nested_closure_using_argument()
3159 assert_equal(['x', 'x2'], DoFilterThis('x'))
3160enddef
3161
Bram Moolenaar0186e582021-01-10 18:33:11 +01003162def Test_triple_nested_closure()
3163 var what = 'x'
3164 var Match = (val: string, cmp: string): bool => stridx(val, cmp) == 0
3165 var Filter = (l) => filter(l, (_, v) => Match(v, what))
3166 assert_equal(['x', 'x2'], ['x', 'y', 'a', 'x2', 'c']->Filter())
3167enddef
3168
Bram Moolenaar8f510af2020-07-05 18:48:23 +02003169func Test_silent_echo()
Bram Moolenaar47e7d702020-07-05 18:18:42 +02003170 CheckScreendump
Bram Moolenaar3b309f12021-12-13 18:19:55 +00003171 call Run_Test_silent_echo()
3172endfunc
Bram Moolenaar47e7d702020-07-05 18:18:42 +02003173
Bram Moolenaar3b309f12021-12-13 18:19:55 +00003174def Run_Test_silent_echo()
3175 var lines =<< trim END
Bram Moolenaar47e7d702020-07-05 18:18:42 +02003176 vim9script
3177 def EchoNothing()
3178 silent echo ''
3179 enddef
3180 defcompile
3181 END
Bram Moolenaar3b309f12021-12-13 18:19:55 +00003182 writefile(lines, 'XTest_silent_echo')
Bram Moolenaar47e7d702020-07-05 18:18:42 +02003183
Bram Moolenaar3b309f12021-12-13 18:19:55 +00003184 # Check that the balloon shows up after a mouse move
Bram Moolenaar62aec932022-01-29 21:45:34 +00003185 var buf = g:RunVimInTerminal('-S XTest_silent_echo', {'rows': 6})
Bram Moolenaar3b309f12021-12-13 18:19:55 +00003186 term_sendkeys(buf, ":abc")
Bram Moolenaar62aec932022-01-29 21:45:34 +00003187 g:VerifyScreenDump(buf, 'Test_vim9_silent_echo', {})
Bram Moolenaar47e7d702020-07-05 18:18:42 +02003188
Bram Moolenaar3b309f12021-12-13 18:19:55 +00003189 # clean up
Bram Moolenaar62aec932022-01-29 21:45:34 +00003190 g:StopVimInTerminal(buf)
Bram Moolenaar3b309f12021-12-13 18:19:55 +00003191 delete('XTest_silent_echo')
3192enddef
Bram Moolenaar47e7d702020-07-05 18:18:42 +02003193
Bram Moolenaar171fb922020-10-28 16:54:47 +01003194def SilentlyError()
3195 execute('silent! invalid')
3196 g:did_it = 'yes'
3197enddef
3198
Bram Moolenaar62aec932022-01-29 21:45:34 +00003199func s:UserError()
Bram Moolenaar28ee8922020-10-28 20:20:00 +01003200 silent! invalid
3201endfunc
3202
3203def SilentlyUserError()
3204 UserError()
3205 g:did_it = 'yes'
3206enddef
Bram Moolenaar171fb922020-10-28 16:54:47 +01003207
3208" This can't be a :def function, because the assert would not be reached.
Bram Moolenaar171fb922020-10-28 16:54:47 +01003209func Test_ignore_silent_error()
3210 let g:did_it = 'no'
3211 call SilentlyError()
3212 call assert_equal('yes', g:did_it)
3213
Bram Moolenaar28ee8922020-10-28 20:20:00 +01003214 let g:did_it = 'no'
3215 call SilentlyUserError()
3216 call assert_equal('yes', g:did_it)
Bram Moolenaar171fb922020-10-28 16:54:47 +01003217
3218 unlet g:did_it
3219endfunc
3220
Bram Moolenaarcd030c42020-10-30 21:49:40 +01003221def Test_ignore_silent_error_in_filter()
3222 var lines =<< trim END
3223 vim9script
3224 def Filter(winid: number, key: string): bool
3225 if key == 'o'
3226 silent! eval [][0]
3227 return true
3228 endif
3229 return popup_filter_menu(winid, key)
3230 enddef
3231
Bram Moolenaare0de1712020-12-02 17:36:54 +01003232 popup_create('popup', {filter: Filter})
Bram Moolenaarcd030c42020-10-30 21:49:40 +01003233 feedkeys("o\r", 'xnt')
3234 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003235 v9.CheckScriptSuccess(lines)
Bram Moolenaarcd030c42020-10-30 21:49:40 +01003236enddef
3237
Bram Moolenaar62aec932022-01-29 21:45:34 +00003238def s:Fibonacci(n: number): number
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02003239 if n < 2
3240 return n
3241 else
3242 return Fibonacci(n - 1) + Fibonacci(n - 2)
3243 endif
3244enddef
3245
Bram Moolenaar985116a2020-07-12 17:31:09 +02003246def Test_recursive_call()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02003247 Fibonacci(20)->assert_equal(6765)
Bram Moolenaar985116a2020-07-12 17:31:09 +02003248enddef
3249
Bram Moolenaar62aec932022-01-29 21:45:34 +00003250def s:TreeWalk(dir: string): list<any>
Bram Moolenaar75ab91f2021-01-10 22:42:50 +01003251 return readdir(dir)->mapnew((_, val) =>
Bram Moolenaar08f7a412020-07-13 20:41:08 +02003252 fnamemodify(dir .. '/' .. val, ':p')->isdirectory()
Bram Moolenaar2bede172020-11-19 18:53:18 +01003253 ? {[val]: TreeWalk(dir .. '/' .. val)}
Bram Moolenaar08f7a412020-07-13 20:41:08 +02003254 : val
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01003255 )
Bram Moolenaar08f7a412020-07-13 20:41:08 +02003256enddef
3257
3258def Test_closure_in_map()
3259 mkdir('XclosureDir/tdir', 'p')
3260 writefile(['111'], 'XclosureDir/file1')
3261 writefile(['222'], 'XclosureDir/file2')
3262 writefile(['333'], 'XclosureDir/tdir/file3')
3263
Bram Moolenaare0de1712020-12-02 17:36:54 +01003264 TreeWalk('XclosureDir')->assert_equal(['file1', 'file2', {tdir: ['file3']}])
Bram Moolenaar08f7a412020-07-13 20:41:08 +02003265
3266 delete('XclosureDir', 'rf')
3267enddef
3268
Bram Moolenaar7b5d5442020-10-04 13:42:34 +02003269def Test_invalid_function_name()
3270 var lines =<< trim END
3271 vim9script
3272 def s: list<string>
3273 END
Bram Moolenaara749a422022-02-12 19:52:25 +00003274 v9.CheckScriptFailure(lines, 'E1268:')
Bram Moolenaar7b5d5442020-10-04 13:42:34 +02003275
3276 lines =<< trim END
3277 vim9script
3278 def g: list<string>
3279 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003280 v9.CheckScriptFailure(lines, 'E129:')
Bram Moolenaar7b5d5442020-10-04 13:42:34 +02003281
3282 lines =<< trim END
3283 vim9script
3284 def <SID>: list<string>
3285 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003286 v9.CheckScriptFailure(lines, 'E884:')
Bram Moolenaar7b5d5442020-10-04 13:42:34 +02003287
3288 lines =<< trim END
3289 vim9script
3290 def F list<string>
3291 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003292 v9.CheckScriptFailure(lines, 'E488:')
Bram Moolenaar7b5d5442020-10-04 13:42:34 +02003293enddef
3294
Bram Moolenaara90afb92020-07-15 22:38:56 +02003295def Test_partial_call()
Bram Moolenaarf78da4f2021-08-01 15:40:31 +02003296 var lines =<< trim END
3297 var Xsetlist: func
3298 Xsetlist = function('setloclist', [0])
3299 Xsetlist([], ' ', {title: 'test'})
3300 getloclist(0, {title: 1})->assert_equal({title: 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02003301
Bram Moolenaarf78da4f2021-08-01 15:40:31 +02003302 Xsetlist = function('setloclist', [0, [], ' '])
3303 Xsetlist({title: 'test'})
3304 getloclist(0, {title: 1})->assert_equal({title: 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02003305
Bram Moolenaarf78da4f2021-08-01 15:40:31 +02003306 Xsetlist = function('setqflist')
3307 Xsetlist([], ' ', {title: 'test'})
3308 getqflist({title: 1})->assert_equal({title: 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02003309
Bram Moolenaarf78da4f2021-08-01 15:40:31 +02003310 Xsetlist = function('setqflist', [[], ' '])
3311 Xsetlist({title: 'test'})
3312 getqflist({title: 1})->assert_equal({title: 'test'})
Bram Moolenaar6abd3dc2020-10-04 14:17:32 +02003313
Bram Moolenaarf78da4f2021-08-01 15:40:31 +02003314 var Len: func: number = function('len', ['word'])
3315 assert_equal(4, Len())
3316
3317 var RepeatFunc = function('repeat', ['o'])
3318 assert_equal('ooooo', RepeatFunc(5))
3319 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003320 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaarc66f6452021-08-19 21:08:30 +02003321
3322 lines =<< trim END
3323 vim9script
3324 def Foo(Parser: any)
3325 enddef
3326 var Expr: func(dict<any>): dict<any>
3327 const Call = Foo(Expr)
3328 END
Bram Moolenaar8acb9cc2022-03-08 13:18:55 +00003329 v9.CheckScriptFailure(lines, 'E1031:')
Bram Moolenaara90afb92020-07-15 22:38:56 +02003330enddef
3331
Bram Moolenaarcd1cda22022-02-16 21:48:25 +00003332def Test_partial_double_nested()
3333 var idx = 123
3334 var Get = () => idx
3335 var Ref = function(Get, [])
3336 var RefRef = function(Ref, [])
3337 assert_equal(123, RefRef())
3338enddef
3339
Bram Moolenaar673bcb12022-03-08 16:52:24 +00003340def Test_partial_null_function()
3341 var lines =<< trim END
3342 var d: dict<func> = {f: null_function}
3343 var Ref = d.f
Bram Moolenaared0c62e2022-03-08 19:43:55 +00003344 assert_equal('func(...): unknown', typename(Ref))
Bram Moolenaar673bcb12022-03-08 16:52:24 +00003345 END
3346 v9.CheckDefAndScriptSuccess(lines)
3347enddef
3348
Bram Moolenaarfe1bfc92022-02-06 13:55:03 +00003349" Using "idx" from a legacy global function does not work.
3350" This caused a crash when called from legacy context.
3351func Test_partial_call_fails()
3352 let lines =<< trim END
3353 vim9script
3354
3355 var l = ['a', 'b', 'c']
3356 def Iter(container: any): any
3357 var idx = -1
3358 var obj = {state: container}
Bram Moolenaarf681cfb2022-02-07 20:30:57 +00003359 def g:NextItem__(self: dict<any>): any
Bram Moolenaarfe1bfc92022-02-06 13:55:03 +00003360 ++idx
3361 return self.state[idx]
3362 enddef
Bram Moolenaarf681cfb2022-02-07 20:30:57 +00003363 obj.__next__ = function('g:NextItem__', [obj])
Bram Moolenaarfe1bfc92022-02-06 13:55:03 +00003364 return obj
3365 enddef
3366
3367 var it = Iter(l)
3368 echo it.__next__()
3369 END
3370 call writefile(lines, 'XpartialCall')
3371 try
3372 source XpartialCall
3373 catch /E1248:/
3374 endtry
3375 call delete('XpartialCall')
3376endfunc
3377
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02003378def Test_cmd_modifier()
3379 tab echo '0'
Bram Moolenaar62aec932022-01-29 21:45:34 +00003380 v9.CheckDefFailure(['5tab echo 3'], 'E16:')
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02003381enddef
3382
3383def Test_restore_modifiers()
3384 # check that when compiling a :def function command modifiers are not messed
3385 # up.
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02003386 var lines =<< trim END
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02003387 vim9script
3388 set eventignore=
3389 autocmd QuickFixCmdPost * copen
3390 def AutocmdsDisabled()
Bram Moolenaarc3235272021-07-10 19:42:03 +02003391 eval 1 + 2
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02003392 enddef
3393 func Func()
3394 noautocmd call s:AutocmdsDisabled()
3395 let g:ei_after = &eventignore
3396 endfunc
3397 Func()
3398 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003399 v9.CheckScriptSuccess(lines)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02003400 g:ei_after->assert_equal('')
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02003401enddef
3402
Bram Moolenaardfa3d552020-09-10 22:05:08 +02003403def StackTop()
Bram Moolenaarc3235272021-07-10 19:42:03 +02003404 eval 1 + 2
3405 eval 2 + 3
Bram Moolenaardfa3d552020-09-10 22:05:08 +02003406 # call not on fourth line
Bram Moolenaar62aec932022-01-29 21:45:34 +00003407 g:StackBot()
Bram Moolenaardfa3d552020-09-10 22:05:08 +02003408enddef
3409
3410def StackBot()
3411 # throw an error
3412 eval [][0]
3413enddef
3414
3415def Test_callstack_def()
3416 try
Bram Moolenaar62aec932022-01-29 21:45:34 +00003417 g:StackTop()
Bram Moolenaardfa3d552020-09-10 22:05:08 +02003418 catch
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02003419 v:throwpoint->assert_match('Test_callstack_def\[2\]..StackTop\[4\]..StackBot, line 2')
Bram Moolenaardfa3d552020-09-10 22:05:08 +02003420 endtry
3421enddef
3422
Bram Moolenaare8211a32020-10-09 22:04:29 +02003423" Re-using spot for variable used in block
3424def Test_block_scoped_var()
3425 var lines =<< trim END
3426 vim9script
3427 def Func()
3428 var x = ['a', 'b', 'c']
3429 if 1
3430 var y = 'x'
Bram Moolenaarbb8a7ce2021-04-10 20:10:26 +02003431 map(x, (_, _) => y)
Bram Moolenaare8211a32020-10-09 22:04:29 +02003432 endif
3433 var z = x
3434 assert_equal(['x', 'x', 'x'], z)
3435 enddef
3436 Func()
3437 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003438 v9.CheckScriptSuccess(lines)
Bram Moolenaare8211a32020-10-09 22:04:29 +02003439enddef
3440
Bram Moolenaareeece9e2020-11-20 19:26:48 +01003441def Test_reset_did_emsg()
3442 var lines =<< trim END
3443 @s = 'blah'
3444 au BufWinLeave * #
3445 def Func()
3446 var winid = popup_create('popup', {})
3447 exe '*s'
3448 popup_close(winid)
3449 enddef
3450 Func()
3451 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003452 v9.CheckScriptFailure(lines, 'E492:', 8)
Bram Moolenaar2d870f82020-12-05 13:41:01 +01003453 delfunc! g:Func
Bram Moolenaareeece9e2020-11-20 19:26:48 +01003454enddef
3455
Bram Moolenaar57f799e2020-12-12 20:42:19 +01003456def Test_did_emsg_reset()
3457 # executing an autocommand resets did_emsg, this should not result in a
3458 # builtin function considered failing
3459 var lines =<< trim END
3460 vim9script
3461 au BufWinLeave * #
3462 def Func()
Bram Moolenaar767034c2021-04-09 17:24:52 +02003463 popup_menu('', {callback: (a, b) => popup_create('', {})->popup_close()})
Bram Moolenaar57f799e2020-12-12 20:42:19 +01003464 eval [][0]
3465 enddef
3466 nno <F3> <cmd>call <sid>Func()<cr>
3467 feedkeys("\<F3>\e", 'xt')
3468 END
3469 writefile(lines, 'XemsgReset')
3470 assert_fails('so XemsgReset', ['E684:', 'E684:'], lines, 2)
3471 delete('XemsgReset')
3472 nunmap <F3>
3473 au! BufWinLeave
3474enddef
3475
Bram Moolenaar56602ba2020-12-05 21:22:08 +01003476def Test_abort_with_silent_call()
3477 var lines =<< trim END
3478 vim9script
3479 g:result = 'none'
3480 def Func()
3481 g:result += 3
3482 g:result = 'yes'
3483 enddef
3484 # error is silenced, but function aborts on error
3485 silent! Func()
3486 assert_equal('none', g:result)
3487 unlet g:result
3488 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003489 v9.CheckScriptSuccess(lines)
Bram Moolenaar56602ba2020-12-05 21:22:08 +01003490enddef
3491
Bram Moolenaarf665e972020-12-05 19:17:16 +01003492def Test_continues_with_silent_error()
3493 var lines =<< trim END
3494 vim9script
3495 g:result = 'none'
3496 def Func()
3497 silent! g:result += 3
3498 g:result = 'yes'
3499 enddef
3500 # error is silenced, function does not abort
3501 Func()
3502 assert_equal('yes', g:result)
3503 unlet g:result
3504 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003505 v9.CheckScriptSuccess(lines)
Bram Moolenaarf665e972020-12-05 19:17:16 +01003506enddef
3507
Bram Moolenaaraf0df472020-12-02 20:51:22 +01003508def Test_abort_even_with_silent()
3509 var lines =<< trim END
3510 vim9script
3511 g:result = 'none'
3512 def Func()
3513 eval {-> ''}() .. '' .. {}['X']
3514 g:result = 'yes'
3515 enddef
Bram Moolenaarf665e972020-12-05 19:17:16 +01003516 silent! Func()
Bram Moolenaaraf0df472020-12-02 20:51:22 +01003517 assert_equal('none', g:result)
Bram Moolenaar4029cab2020-12-05 18:13:27 +01003518 unlet g:result
3519 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003520 v9.CheckScriptSuccess(lines)
Bram Moolenaar4029cab2020-12-05 18:13:27 +01003521enddef
3522
Bram Moolenaarf665e972020-12-05 19:17:16 +01003523def Test_cmdmod_silent_restored()
3524 var lines =<< trim END
3525 vim9script
3526 def Func()
3527 g:result = 'none'
3528 silent! g:result += 3
3529 g:result = 'none'
3530 g:result += 3
3531 enddef
3532 Func()
3533 END
3534 # can't use CheckScriptFailure, it ignores the :silent!
3535 var fname = 'Xdefsilent'
3536 writefile(lines, fname)
3537 var caught = 'no'
3538 try
3539 exe 'source ' .. fname
3540 catch /E1030:/
3541 caught = 'yes'
3542 assert_match('Func, line 4', v:throwpoint)
3543 endtry
3544 assert_equal('yes', caught)
3545 delete(fname)
3546enddef
3547
Bram Moolenaar2fecb532021-03-24 22:00:56 +01003548def Test_cmdmod_silent_nested()
3549 var lines =<< trim END
3550 vim9script
3551 var result = ''
3552
3553 def Error()
3554 result ..= 'Eb'
3555 eval [][0]
3556 result ..= 'Ea'
3557 enddef
3558
3559 def Crash()
3560 result ..= 'Cb'
3561 sil! Error()
3562 result ..= 'Ca'
3563 enddef
3564
3565 Crash()
3566 assert_equal('CbEbEaCa', result)
3567 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003568 v9.CheckScriptSuccess(lines)
Bram Moolenaar2fecb532021-03-24 22:00:56 +01003569enddef
3570
Bram Moolenaar4029cab2020-12-05 18:13:27 +01003571def Test_dict_member_with_silent()
3572 var lines =<< trim END
3573 vim9script
3574 g:result = 'none'
3575 var d: dict<any>
3576 def Func()
3577 try
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01003578 g:result = map([], (_, v) => ({}[v]))->join() .. d['']
Bram Moolenaar4029cab2020-12-05 18:13:27 +01003579 catch
3580 endtry
3581 enddef
3582 silent! Func()
3583 assert_equal('0', g:result)
3584 unlet g:result
Bram Moolenaaraf0df472020-12-02 20:51:22 +01003585 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003586 v9.CheckScriptSuccess(lines)
Bram Moolenaaraf0df472020-12-02 20:51:22 +01003587enddef
3588
Bram Moolenaarf9041332021-01-21 19:41:16 +01003589def Test_skip_cmds_with_silent()
3590 var lines =<< trim END
3591 vim9script
3592
3593 def Func(b: bool)
3594 Crash()
3595 enddef
3596
3597 def Crash()
3598 sil! :/not found/d _
3599 sil! :/not found/put _
3600 enddef
3601
3602 Func(true)
3603 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003604 v9.CheckScriptSuccess(lines)
Bram Moolenaarf9041332021-01-21 19:41:16 +01003605enddef
3606
Bram Moolenaar5b3d1bb2020-12-22 12:20:08 +01003607def Test_opfunc()
Bram Moolenaar848fadd2022-01-30 15:28:30 +00003608 nnoremap <F3> <cmd>set opfunc=g:Opfunc<cr>g@
Bram Moolenaar5b3d1bb2020-12-22 12:20:08 +01003609 def g:Opfunc(_: any): string
3610 setline(1, 'ASDF')
3611 return ''
3612 enddef
3613 new
3614 setline(1, 'asdf')
3615 feedkeys("\<F3>$", 'x')
3616 assert_equal('ASDF', getline(1))
3617
3618 bwipe!
3619 nunmap <F3>
3620enddef
3621
Bram Moolenaar3b309f12021-12-13 18:19:55 +00003622func Test_opfunc_error()
3623 CheckScreendump
3624 call Run_Test_opfunc_error()
3625endfunc
3626
3627def Run_Test_opfunc_error()
3628 # test that the error from Opfunc() is displayed right away
3629 var lines =<< trim END
3630 vim9script
3631
3632 def Opfunc(type: string)
3633 try
3634 eval [][0]
3635 catch /nothing/ # error not caught
3636 endtry
3637 enddef
3638 &operatorfunc = Opfunc
3639 nnoremap <expr> l <SID>L()
3640 def L(): string
3641 return 'l'
3642 enddef
3643 'x'->repeat(10)->setline(1)
3644 feedkeys('g@l', 'n')
3645 feedkeys('llll')
3646 END
3647 call writefile(lines, 'XTest_opfunc_error')
3648
Bram Moolenaar62aec932022-01-29 21:45:34 +00003649 var buf = g:RunVimInTerminal('-S XTest_opfunc_error', {rows: 6, wait_for_ruler: 0})
3650 g:WaitForAssert(() => assert_match('Press ENTER', term_getline(buf, 6)))
3651 g:WaitForAssert(() => assert_match('E684: list index out of range: 0', term_getline(buf, 5)))
Bram Moolenaar3b309f12021-12-13 18:19:55 +00003652
3653 # clean up
Bram Moolenaar62aec932022-01-29 21:45:34 +00003654 g:StopVimInTerminal(buf)
Bram Moolenaar3b309f12021-12-13 18:19:55 +00003655 delete('XTest_opfunc_error')
3656enddef
3657
Bram Moolenaar077a4232020-12-22 18:33:27 +01003658" this was crashing on exit
3659def Test_nested_lambda_in_closure()
3660 var lines =<< trim END
3661 vim9script
Bram Moolenaar227c58a2021-04-28 20:40:44 +02003662 command WriteDone writefile(['Done'], 'XnestedDone')
Bram Moolenaar077a4232020-12-22 18:33:27 +01003663 def Outer()
3664 def g:Inner()
3665 echo map([1, 2, 3], {_, v -> v + 1})
3666 enddef
3667 g:Inner()
3668 enddef
3669 defcompile
Bram Moolenaar227c58a2021-04-28 20:40:44 +02003670 # not reached
Bram Moolenaar077a4232020-12-22 18:33:27 +01003671 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003672 if !g:RunVim([], lines, '--clean -c WriteDone -c quit')
Bram Moolenaar077a4232020-12-22 18:33:27 +01003673 return
3674 endif
3675 assert_equal(['Done'], readfile('XnestedDone'))
3676 delete('XnestedDone')
3677enddef
3678
Bram Moolenaar92368aa2022-02-07 17:50:39 +00003679def Test_nested_closure_funcref()
3680 var lines =<< trim END
3681 vim9script
3682 def Func()
3683 var n: number
3684 def Nested()
3685 ++n
3686 enddef
3687 Nested()
3688 g:result_one = n
3689 var Ref = function(Nested)
3690 Ref()
3691 g:result_two = n
3692 enddef
3693 Func()
3694 END
3695 v9.CheckScriptSuccess(lines)
3696 assert_equal(1, g:result_one)
3697 assert_equal(2, g:result_two)
3698 unlet g:result_one g:result_two
3699enddef
3700
Bram Moolenaar7aca5ca2022-02-07 19:56:43 +00003701def Test_nested_closure_in_dict()
3702 var lines =<< trim END
3703 vim9script
3704 def Func(): dict<any>
3705 var n: number
3706 def Inc(): number
3707 ++n
3708 return n
3709 enddef
3710 return {inc: function(Inc)}
3711 enddef
3712 disas Func
3713 var d = Func()
3714 assert_equal(1, d.inc())
3715 assert_equal(2, d.inc())
3716 END
3717 v9.CheckScriptSuccess(lines)
3718enddef
3719
Bram Moolenaarfb43cfc2022-03-11 18:54:17 +00003720def Test_script_local_other_script()
3721 var lines =<< trim END
3722 function LegacyJob()
3723 let FuncRef = function('s:close_cb')
3724 endfunction
3725 function s:close_cb(...)
3726 endfunction
3727 END
3728 lines->writefile('Xlegacy.vim')
3729 source Xlegacy.vim
3730 g:LegacyJob()
3731 g:LegacyJob()
3732 g:LegacyJob()
3733
3734 delfunc g:LegacyJob
3735 delete('Xlegacy.vim')
3736enddef
3737
Bram Moolenaar04947cc2021-03-06 19:26:46 +01003738def Test_check_func_arg_types()
3739 var lines =<< trim END
3740 vim9script
3741 def F1(x: string): string
3742 return x
3743 enddef
3744
3745 def F2(x: number): number
3746 return x + 1
3747 enddef
3748
3749 def G(g: func): dict<func>
3750 return {f: g}
3751 enddef
3752
3753 def H(d: dict<func>): string
3754 return d.f('a')
3755 enddef
3756 END
3757
Bram Moolenaar62aec932022-01-29 21:45:34 +00003758 v9.CheckScriptSuccess(lines + ['echo H(G(F1))'])
3759 v9.CheckScriptFailure(lines + ['echo H(G(F2))'], 'E1013:')
Bram Moolenaar04947cc2021-03-06 19:26:46 +01003760enddef
3761
Bram Moolenaar6e48b842021-08-10 22:52:02 +02003762def Test_list_any_type_checked()
3763 var lines =<< trim END
3764 vim9script
3765 def Foo()
3766 --decl--
3767 Bar(l)
3768 enddef
3769 def Bar(ll: list<dict<any>>)
3770 enddef
3771 Foo()
3772 END
Bram Moolenaar2d3ac2e2022-02-03 12:34:05 +00003773 # "any" could be "dict<any>", thus OK
Bram Moolenaar6e48b842021-08-10 22:52:02 +02003774 lines[2] = 'var l: list<any>'
Bram Moolenaar2d3ac2e2022-02-03 12:34:05 +00003775 v9.CheckScriptSuccess(lines)
Bram Moolenaar6e48b842021-08-10 22:52:02 +02003776 lines[2] = 'var l: list<any> = []'
Bram Moolenaar2d3ac2e2022-02-03 12:34:05 +00003777 v9.CheckScriptSuccess(lines)
Bram Moolenaar6e48b842021-08-10 22:52:02 +02003778
3779 lines[2] = 'var l: list<any> = [11]'
Bram Moolenaar62aec932022-01-29 21:45:34 +00003780 v9.CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected list<dict<any>> but got list<number>', 2)
Bram Moolenaar6e48b842021-08-10 22:52:02 +02003781enddef
3782
Bram Moolenaar701cc6c2021-04-10 13:33:48 +02003783def Test_compile_error()
3784 var lines =<< trim END
3785 def g:Broken()
3786 echo 'a' + {}
3787 enddef
3788 call g:Broken()
3789 END
3790 # First call: compilation error
Bram Moolenaar62aec932022-01-29 21:45:34 +00003791 v9.CheckScriptFailure(lines, 'E1051: Wrong argument type for +')
Bram Moolenaar701cc6c2021-04-10 13:33:48 +02003792
3793 # Second call won't try compiling again
3794 assert_fails('call g:Broken()', 'E1091: Function is not compiled: Broken')
Bram Moolenaar599410c2021-04-10 14:03:43 +02003795 delfunc g:Broken
3796
3797 # No error when compiling with :silent!
3798 lines =<< trim END
3799 def g:Broken()
3800 echo 'a' + []
3801 enddef
3802 silent! defcompile
3803 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003804 v9.CheckScriptSuccess(lines)
Bram Moolenaar599410c2021-04-10 14:03:43 +02003805
3806 # Calling the function won't try compiling again
3807 assert_fails('call g:Broken()', 'E1091: Function is not compiled: Broken')
3808 delfunc g:Broken
Bram Moolenaar701cc6c2021-04-10 13:33:48 +02003809enddef
3810
Bram Moolenaar962c43b2021-04-10 17:18:09 +02003811def Test_ignored_argument()
3812 var lines =<< trim END
3813 vim9script
3814 def Ignore(_, _): string
3815 return 'yes'
3816 enddef
3817 assert_equal('yes', Ignore(1, 2))
3818
3819 func Ok(_)
3820 return a:_
3821 endfunc
3822 assert_equal('ok', Ok('ok'))
3823
3824 func Oktoo()
3825 let _ = 'too'
3826 return _
3827 endfunc
3828 assert_equal('too', Oktoo())
Bram Moolenaarda479c72021-04-10 21:01:38 +02003829
3830 assert_equal([[1], [2], [3]], range(3)->mapnew((_, v) => [v]->map((_, w) => w + 1)))
Bram Moolenaar962c43b2021-04-10 17:18:09 +02003831 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003832 v9.CheckScriptSuccess(lines)
Bram Moolenaar962c43b2021-04-10 17:18:09 +02003833
3834 lines =<< trim END
3835 def Ignore(_: string): string
3836 return _
3837 enddef
3838 defcompile
3839 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003840 v9.CheckScriptFailure(lines, 'E1181:', 1)
Bram Moolenaar962c43b2021-04-10 17:18:09 +02003841
3842 lines =<< trim END
3843 var _ = 1
3844 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003845 v9.CheckDefAndScriptFailure(lines, 'E1181:', 1)
Yegappan Lakshmanan34fcb692021-05-25 20:14:00 +02003846
3847 lines =<< trim END
3848 var x = _
3849 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003850 v9.CheckDefAndScriptFailure(lines, 'E1181:', 1)
Bram Moolenaar962c43b2021-04-10 17:18:09 +02003851enddef
3852
Bram Moolenaarbb8a7ce2021-04-10 20:10:26 +02003853def Test_too_many_arguments()
3854 var lines =<< trim END
3855 echo [0, 1, 2]->map(() => 123)
3856 END
Bram Moolenaareddd4fc2022-02-20 15:52:28 +00003857 v9.CheckDefAndScriptFailure(lines, ['E176:', 'E1106: 2 arguments too many'], 1)
Bram Moolenaarbb8a7ce2021-04-10 20:10:26 +02003858
3859 lines =<< trim END
3860 echo [0, 1, 2]->map((_) => 123)
3861 END
Bram Moolenaareddd4fc2022-02-20 15:52:28 +00003862 v9.CheckDefAndScriptFailure(lines, ['E176', 'E1106: One argument too many'], 1)
Bram Moolenaarbb8a7ce2021-04-10 20:10:26 +02003863enddef
Bram Moolenaar077a4232020-12-22 18:33:27 +01003864
Bram Moolenaara6aa1642021-04-23 19:32:23 +02003865def Test_closing_brace_at_start_of_line()
3866 var lines =<< trim END
3867 def Func()
3868 enddef
3869 Func(
3870 )
3871 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003872 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaara6aa1642021-04-23 19:32:23 +02003873enddef
3874
Bram Moolenaar62aec932022-01-29 21:45:34 +00003875func s:CreateMydict()
Bram Moolenaarb033ee22021-08-15 16:08:36 +02003876 let g:mydict = {}
3877 func g:mydict.afunc()
3878 let g:result = self.key
3879 endfunc
3880endfunc
3881
3882def Test_numbered_function_reference()
3883 CreateMydict()
3884 var output = execute('legacy func g:mydict.afunc')
3885 var funcName = 'g:' .. substitute(output, '.*function \(\d\+\).*', '\1', '')
3886 execute 'function(' .. funcName .. ', [], {key: 42})()'
3887 # check that the function still exists
3888 assert_equal(output, execute('legacy func g:mydict.afunc'))
3889 unlet g:mydict
3890enddef
3891
Bram Moolenaard3a11782022-01-05 16:50:40 +00003892def Test_go_beyond_end_of_cmd()
3893 # this was reading the byte after the end of the line
3894 var lines =<< trim END
3895 def F()
3896 cal
3897 enddef
3898 defcompile
3899 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00003900 v9.CheckScriptFailure(lines, 'E476:')
Bram Moolenaard3a11782022-01-05 16:50:40 +00003901enddef
3902
Bram Moolenaar8b716f52022-02-15 21:17:56 +00003903" The following messes up syntax highlight, keep near the end.
Bram Moolenaar20677332021-06-06 17:02:53 +02003904if has('python3')
Bram Moolenaar8b716f52022-02-15 21:17:56 +00003905 def Test_python3_command()
3906 py3 import vim
Bram Moolenaarf5288c52022-02-15 21:33:29 +00003907 py3 vim.command("g:done = 'yes'")
Bram Moolenaar8b716f52022-02-15 21:17:56 +00003908 assert_equal('yes', g:done)
3909 unlet g:done
3910 enddef
3911
Bram Moolenaar20677332021-06-06 17:02:53 +02003912 def Test_python3_heredoc()
3913 py3 << trim EOF
3914 import vim
3915 vim.vars['didit'] = 'yes'
3916 EOF
3917 assert_equal('yes', g:didit)
3918
3919 python3 << trim EOF
3920 import vim
3921 vim.vars['didit'] = 'again'
3922 EOF
3923 assert_equal('again', g:didit)
3924 enddef
3925endif
3926
Bram Moolenaar20677332021-06-06 17:02:53 +02003927if has('lua')
3928 def Test_lua_heredoc()
3929 g:d = {}
3930 lua << trim EOF
3931 x = vim.eval('g:d')
3932 x['key'] = 'val'
3933 EOF
3934 assert_equal('val', g:d.key)
3935 enddef
3936endif
3937
Bram Moolenaarf7779c62020-05-03 15:38:16 +02003938
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02003939" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker