blob: a2d1ed3e2915fd9fddd6d32f99f81c9ee2d8c925 [file] [log] [blame]
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001" Test various aspects of the Vim9 script language.
2
3source check.vim
Bram Moolenaarad304702020-09-06 18:22:53 +02004source term_util.vim
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02005source view_util.vim
Bram Moolenaar04b12692020-05-04 23:24:44 +02006source vim9.vim
Bram Moolenaar47e7d702020-07-05 18:18:42 +02007source screendump.vim
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02008
9func Test_def_basic()
10 def SomeFunc(): string
11 return 'yes'
12 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +020013 call SomeFunc()->assert_equal('yes')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +020014endfunc
15
Bram Moolenaar2b9b17e2020-10-13 18:38:11 +020016func Test_compiling_error()
17 " use a terminal to see the whole error message
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +020018 CheckRunVimInTerminal
19
Bram Moolenaar2b9b17e2020-10-13 18:38:11 +020020 call TestCompilingError()
21endfunc
22
23def TestCompilingError()
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +020024 var lines =<< trim END
25 vim9script
26 def Fails()
27 echo nothing
28 enddef
29 defcompile
30 END
31 call writefile(lines, 'XTest_compile_error')
32 var buf = RunVimInTerminal('-S XTest_compile_error',
Bram Moolenaare0de1712020-12-02 17:36:54 +010033 {rows: 10, wait_for_ruler: 0})
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +020034 var text = ''
35 for loop in range(100)
36 text = ''
37 for i in range(1, 9)
38 text ..= term_getline(buf, i)
39 endfor
Bram Moolenaar18dc3552020-11-22 14:24:00 +010040 if text =~ 'Variable not found: nothing'
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +020041 break
42 endif
43 sleep 20m
44 endfor
45 assert_match('Error detected while compiling command line.*Fails.*Variable not found: nothing', text)
46
47 # clean up
48 call StopVimInTerminal(buf)
49 call delete('XTest_compile_error')
50enddef
51
Bram Moolenaar0ba48e82020-11-17 18:23:19 +010052def CallRecursive(n: number): number
53 return CallRecursive(n + 1)
54enddef
55
56def CallMapRecursive(l: list<number>): number
Bram Moolenaar2949cfd2020-12-31 21:28:47 +010057 return map(l, (_, v) => CallMapRecursive([v]))[0]
Bram Moolenaar0ba48e82020-11-17 18:23:19 +010058enddef
59
60def Test_funcdepth_error()
61 set maxfuncdepth=10
62
63 var caught = false
64 try
65 CallRecursive(1)
66 catch /E132:/
67 caught = true
68 endtry
69 assert_true(caught)
70
71 caught = false
72 try
73 CallMapRecursive([1])
74 catch /E132:/
75 caught = true
76 endtry
77 assert_true(caught)
78
79 set maxfuncdepth&
80enddef
81
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +020082def ReturnString(): string
83 return 'string'
84enddef
85
86def ReturnNumber(): number
87 return 123
88enddef
89
90let g:notNumber = 'string'
91
92def ReturnGlobal(): number
93 return g:notNumber
94enddef
95
96def Test_return_something()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +020097 ReturnString()->assert_equal('string')
98 ReturnNumber()->assert_equal(123)
Bram Moolenaar5e654232020-09-16 15:22:00 +020099 assert_fails('ReturnGlobal()', 'E1012: Type mismatch; expected number but got string', '', 1, 'ReturnGlobal')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200100enddef
101
Bram Moolenaarefd88552020-06-18 20:50:10 +0200102def Test_missing_return()
103 CheckDefFailure(['def Missing(): number',
104 ' if g:cond',
105 ' echo "no return"',
106 ' else',
107 ' return 0',
108 ' endif'
109 'enddef'], 'E1027:')
110 CheckDefFailure(['def Missing(): number',
111 ' if g:cond',
112 ' return 1',
113 ' else',
114 ' echo "no return"',
115 ' endif'
116 'enddef'], 'E1027:')
117 CheckDefFailure(['def Missing(): number',
118 ' if g:cond',
119 ' return 1',
120 ' else',
121 ' return 2',
122 ' endif'
123 ' return 3'
124 'enddef'], 'E1095:')
125enddef
126
Bram Moolenaar403dc312020-10-17 19:29:51 +0200127def Test_return_bool()
128 var lines =<< trim END
129 vim9script
130 def MenuFilter(id: number, key: string): bool
131 return popup_filter_menu(id, key)
132 enddef
133 def YesnoFilter(id: number, key: string): bool
134 return popup_filter_yesno(id, key)
135 enddef
136 defcompile
137 END
138 CheckScriptSuccess(lines)
139enddef
140
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200141let s:nothing = 0
142def ReturnNothing()
143 s:nothing = 1
144 if true
145 return
146 endif
147 s:nothing = 2
148enddef
149
150def Test_return_nothing()
151 ReturnNothing()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200152 s:nothing->assert_equal(1)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200153enddef
154
155func Increment()
156 let g:counter += 1
157endfunc
158
159def Test_call_ufunc_count()
160 g:counter = 1
161 Increment()
162 Increment()
163 Increment()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200164 # works with and without :call
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200165 g:counter->assert_equal(4)
166 eval g:counter->assert_equal(4)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200167 unlet g:counter
168enddef
169
170def MyVarargs(arg: string, ...rest: list<string>): string
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200171 var res = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200172 for s in rest
173 res ..= ',' .. s
174 endfor
175 return res
176enddef
177
178def Test_call_varargs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200179 MyVarargs('one')->assert_equal('one')
180 MyVarargs('one', 'two')->assert_equal('one,two')
181 MyVarargs('one', 'two', 'three')->assert_equal('one,two,three')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200182enddef
183
184def MyDefaultArgs(name = 'string'): string
185 return name
186enddef
187
Bram Moolenaare30f64b2020-07-15 19:48:20 +0200188def MyDefaultSecond(name: string, second: bool = true): string
189 return second ? name : 'none'
190enddef
191
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200192def Test_call_default_args()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200193 MyDefaultArgs()->assert_equal('string')
194 MyDefaultArgs('one')->assert_equal('one')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200195 assert_fails('MyDefaultArgs("one", "two")', 'E118:', '', 3, 'Test_call_default_args')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200196
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200197 MyDefaultSecond('test')->assert_equal('test')
198 MyDefaultSecond('test', true)->assert_equal('test')
199 MyDefaultSecond('test', false)->assert_equal('none')
Bram Moolenaare30f64b2020-07-15 19:48:20 +0200200
Bram Moolenaar822ba242020-05-24 23:00:18 +0200201 CheckScriptFailure(['def Func(arg: number = asdf)', 'enddef', 'defcompile'], 'E1001:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100202 delfunc g:Func
Bram Moolenaar77072282020-09-16 17:55:40 +0200203 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 +0100204 delfunc g:Func
Bram Moolenaar04b12692020-05-04 23:24:44 +0200205enddef
206
207def Test_nested_function()
208 def Nested(arg: string): string
209 return 'nested ' .. arg
210 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200211 Nested('function')->assert_equal('nested function')
Bram Moolenaar04b12692020-05-04 23:24:44 +0200212
Bram Moolenaar0e65d3d2020-05-05 17:53:16 +0200213 CheckDefFailure(['def Nested()', 'enddef', 'Nested(66)'], 'E118:')
214 CheckDefFailure(['def Nested(arg: string)', 'enddef', 'Nested()'], 'E119:')
215
Bram Moolenaar04b12692020-05-04 23:24:44 +0200216 CheckDefFailure(['func Nested()', 'endfunc'], 'E1086:')
Bram Moolenaarbcbf4132020-08-01 22:35:13 +0200217 CheckDefFailure(['def s:Nested()', 'enddef'], 'E1075:')
218 CheckDefFailure(['def b:Nested()', 'enddef'], 'E1075:')
Bram Moolenaar8b848ca2020-09-10 22:28:01 +0200219
Bram Moolenaar54021752020-12-06 18:50:36 +0100220 var lines =<< trim END
221 def Outer()
222 def Inner()
223 # comment
224 enddef
225 def Inner()
226 enddef
227 enddef
228 END
229 CheckDefFailure(lines, 'E1073:')
230
231 lines =<< trim END
232 def Outer()
233 def Inner()
234 # comment
235 enddef
236 def! Inner()
237 enddef
238 enddef
239 END
240 CheckDefFailure(lines, 'E1117:')
241
242 # nested function inside conditional
243 # TODO: should it work when "thecount" is inside the "if"?
244 lines =<< trim END
245 vim9script
246 var thecount = 0
247 if true
248 def Test(): number
249 def TheFunc(): number
250 thecount += 1
251 return thecount
252 enddef
253 return TheFunc()
254 enddef
255 endif
256 defcompile
257 assert_equal(1, Test())
258 assert_equal(2, Test())
259 END
260 CheckScriptSuccess(lines)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200261enddef
262
Bram Moolenaaradc8e442020-12-31 18:28:18 +0100263def Test_not_nested_function()
264 echo printf('%d',
265 function('len')('xxx'))
266enddef
267
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200268func Test_call_default_args_from_func()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200269 call MyDefaultArgs()->assert_equal('string')
270 call MyDefaultArgs('one')->assert_equal('one')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200271 call assert_fails('call MyDefaultArgs("one", "two")', 'E118:', '', 3, 'Test_call_default_args_from_func')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200272endfunc
273
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200274def Test_nested_global_function()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200275 var lines =<< trim END
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200276 vim9script
277 def Outer()
278 def g:Inner(): string
279 return 'inner'
280 enddef
281 enddef
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200282 defcompile
283 Outer()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200284 g:Inner()->assert_equal('inner')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200285 delfunc g:Inner
286 Outer()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200287 g:Inner()->assert_equal('inner')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200288 delfunc g:Inner
289 Outer()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200290 g:Inner()->assert_equal('inner')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200291 delfunc g:Inner
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200292 END
293 CheckScriptSuccess(lines)
Bram Moolenaar2c79e9d2020-08-01 18:57:52 +0200294
295 lines =<< trim END
296 vim9script
297 def Outer()
298 def g:Inner(): string
299 return 'inner'
300 enddef
301 enddef
302 defcompile
303 Outer()
304 Outer()
305 END
306 CheckScriptFailure(lines, "E122:")
Bram Moolenaarcd45ed02020-12-22 17:35:54 +0100307 delfunc g:Inner
Bram Moolenaarad486a02020-08-01 23:22:18 +0200308
309 lines =<< trim END
310 vim9script
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100311 def Outer()
312 def g:Inner()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100313 echo map([1, 2, 3], (_, v) => v + 1)
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100314 enddef
315 g:Inner()
316 enddef
317 Outer()
318 END
319 CheckScriptSuccess(lines)
320 delfunc g:Inner
321
322 lines =<< trim END
323 vim9script
Bram Moolenaarad486a02020-08-01 23:22:18 +0200324 def Func()
325 echo 'script'
326 enddef
327 def Outer()
328 def Func()
329 echo 'inner'
330 enddef
331 enddef
332 defcompile
333 END
334 CheckScriptFailure(lines, "E1073:")
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200335enddef
336
Bram Moolenaar6abdcf82020-11-22 18:15:44 +0100337def DefListAll()
338 def
339enddef
340
341def DefListOne()
342 def DefListOne
343enddef
344
345def DefListMatches()
346 def /DefList
347enddef
348
349def Test_nested_def_list()
350 var funcs = split(execute('call DefListAll()'), "\n")
351 assert_true(len(funcs) > 10)
352 assert_true(funcs->index('def DefListAll()') >= 0)
353
354 funcs = split(execute('call DefListOne()'), "\n")
355 assert_equal([' def DefListOne()', '1 def DefListOne', ' enddef'], funcs)
356
357 funcs = split(execute('call DefListMatches()'), "\n")
358 assert_true(len(funcs) >= 3)
359 assert_true(funcs->index('def DefListAll()') >= 0)
360 assert_true(funcs->index('def DefListOne()') >= 0)
361 assert_true(funcs->index('def DefListMatches()') >= 0)
Bram Moolenaar54021752020-12-06 18:50:36 +0100362
363 var lines =<< trim END
364 vim9script
365 def Func()
366 def +Func+
367 enddef
368 defcompile
369 END
370 CheckScriptFailure(lines, 'E476:', 1)
Bram Moolenaar6abdcf82020-11-22 18:15:44 +0100371enddef
372
Bram Moolenaar333894b2020-08-01 18:53:07 +0200373def Test_global_local_function()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200374 var lines =<< trim END
Bram Moolenaar333894b2020-08-01 18:53:07 +0200375 vim9script
376 def g:Func(): string
377 return 'global'
378 enddef
379 def Func(): string
380 return 'local'
381 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200382 g:Func()->assert_equal('global')
383 Func()->assert_equal('local')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100384 delfunc g:Func
Bram Moolenaar333894b2020-08-01 18:53:07 +0200385 END
386 CheckScriptSuccess(lines)
Bram Moolenaar035d6e92020-08-11 22:30:42 +0200387
388 lines =<< trim END
389 vim9script
390 def g:Funcy()
391 echo 'funcy'
392 enddef
393 s:Funcy()
394 END
395 CheckScriptFailure(lines, 'E117:')
Bram Moolenaar333894b2020-08-01 18:53:07 +0200396enddef
397
Bram Moolenaar0f769812020-09-12 18:32:34 +0200398def Test_local_function_shadows_global()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200399 var lines =<< trim END
Bram Moolenaar0f769812020-09-12 18:32:34 +0200400 vim9script
401 def g:Gfunc(): string
402 return 'global'
403 enddef
404 def AnotherFunc(): number
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200405 var Gfunc = function('len')
Bram Moolenaar0f769812020-09-12 18:32:34 +0200406 return Gfunc('testing')
407 enddef
408 g:Gfunc()->assert_equal('global')
409 AnotherFunc()->assert_equal(7)
410 delfunc g:Gfunc
411 END
412 CheckScriptSuccess(lines)
413
414 lines =<< trim END
415 vim9script
416 def g:Func(): string
417 return 'global'
418 enddef
419 def AnotherFunc()
420 g:Func = function('len')
421 enddef
422 AnotherFunc()
423 END
424 CheckScriptFailure(lines, 'E705:')
425 delfunc g:Func
426enddef
427
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200428func TakesOneArg(arg)
429 echo a:arg
430endfunc
431
432def Test_call_wrong_args()
Bram Moolenaard2c61702020-09-06 15:58:36 +0200433 CheckDefFailure(['TakesOneArg()'], 'E119:')
434 CheckDefFailure(['TakesOneArg(11, 22)'], 'E118:')
435 CheckDefFailure(['bufnr(xxx)'], 'E1001:')
436 CheckScriptFailure(['def Func(Ref: func(s: string))'], 'E475:')
Bram Moolenaaree8580e2020-08-28 17:19:07 +0200437
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200438 var lines =<< trim END
Bram Moolenaaree8580e2020-08-28 17:19:07 +0200439 vim9script
440 def Func(s: string)
441 echo s
442 enddef
443 Func([])
444 END
Bram Moolenaar77072282020-09-16 17:55:40 +0200445 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected string but got list<unknown>', 5)
Bram Moolenaarb185a402020-09-18 22:42:00 +0200446
447 lines =<< trim END
448 vim9script
449 def FuncOne(nr: number)
450 echo nr
451 enddef
452 def FuncTwo()
453 FuncOne()
454 enddef
455 defcompile
456 END
457 writefile(lines, 'Xscript')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200458 var didCatch = false
Bram Moolenaarb185a402020-09-18 22:42:00 +0200459 try
460 source Xscript
461 catch
462 assert_match('E119: Not enough arguments for function: <SNR>\d\+_FuncOne', v:exception)
463 assert_match('Xscript\[8\]..function <SNR>\d\+_FuncTwo, line 1', v:throwpoint)
464 didCatch = true
465 endtry
466 assert_true(didCatch)
467
468 lines =<< trim END
469 vim9script
470 def FuncOne(nr: number)
471 echo nr
472 enddef
473 def FuncTwo()
474 FuncOne(1, 2)
475 enddef
476 defcompile
477 END
478 writefile(lines, 'Xscript')
479 didCatch = false
480 try
481 source Xscript
482 catch
483 assert_match('E118: Too many arguments for function: <SNR>\d\+_FuncOne', v:exception)
484 assert_match('Xscript\[8\]..function <SNR>\d\+_FuncTwo, line 1', v:throwpoint)
485 didCatch = true
486 endtry
487 assert_true(didCatch)
488
489 delete('Xscript')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200490enddef
491
Bram Moolenaar50824712020-12-20 21:10:17 +0100492def Test_call_funcref_wrong_args()
493 var head =<< trim END
494 vim9script
495 def Func3(a1: string, a2: number, a3: list<number>)
496 echo a1 .. a2 .. a3[0]
497 enddef
498 def Testme()
499 var funcMap: dict<func> = {func: Func3}
500 END
501 var tail =<< trim END
502 enddef
503 Testme()
504 END
505 CheckScriptSuccess(head + ["funcMap['func']('str', 123, [1, 2, 3])"] + tail)
506
507 CheckScriptFailure(head + ["funcMap['func']('str', 123)"] + tail, 'E119:')
508 CheckScriptFailure(head + ["funcMap['func']('str', 123, [1], 4)"] + tail, 'E118:')
509enddef
510
Bram Moolenaarb4d16cb2020-11-05 18:45:46 +0100511def Test_call_lambda_args()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100512 CheckDefFailure(['echo ((i) => 0)()'],
513 'E119: Not enough arguments for function: ((i) => 0)()')
Bram Moolenaarb4d16cb2020-11-05 18:45:46 +0100514
515 var lines =<< trim END
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100516 var Ref = (x: number, y: number) => x + y
Bram Moolenaarb4d16cb2020-11-05 18:45:46 +0100517 echo Ref(1, 'x')
518 END
519 CheckDefFailure(lines, 'E1013: Argument 2: type mismatch, expected number but got string')
520enddef
521
Bram Moolenaar709664c2020-12-12 14:33:41 +0100522def Test_lambda_uses_assigned_var()
523 CheckDefSuccess([
524 'var x: any = "aaa"'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100525 'x = filter(["bbb"], (_, v) => v =~ x)'])
Bram Moolenaar709664c2020-12-12 14:33:41 +0100526enddef
527
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200528" Default arg and varargs
529def MyDefVarargs(one: string, two = 'foo', ...rest: list<string>): string
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200530 var res = one .. ',' .. two
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200531 for s in rest
532 res ..= ',' .. s
533 endfor
534 return res
535enddef
536
537def Test_call_def_varargs()
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200538 assert_fails('MyDefVarargs()', 'E119:', '', 1, 'Test_call_def_varargs')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200539 MyDefVarargs('one')->assert_equal('one,foo')
540 MyDefVarargs('one', 'two')->assert_equal('one,two')
541 MyDefVarargs('one', 'two', 'three')->assert_equal('one,two,three')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200542 CheckDefFailure(['MyDefVarargs("one", 22)'],
Bram Moolenaar77072282020-09-16 17:55:40 +0200543 'E1013: Argument 2: type mismatch, expected string but got number')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200544 CheckDefFailure(['MyDefVarargs("one", "two", 123)'],
Bram Moolenaar77072282020-09-16 17:55:40 +0200545 'E1013: Argument 3: type mismatch, expected string but got number')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200546
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200547 var lines =<< trim END
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200548 vim9script
549 def Func(...l: list<string>)
550 echo l
551 enddef
552 Func('a', 'b', 'c')
553 END
554 CheckScriptSuccess(lines)
555
556 lines =<< trim END
557 vim9script
558 def Func(...l: list<string>)
559 echo l
560 enddef
561 Func()
562 END
563 CheckScriptSuccess(lines)
564
565 lines =<< trim END
566 vim9script
Bram Moolenaar2f8cbc42020-09-16 17:22:59 +0200567 def Func(...l: any)
568 echo l
569 enddef
570 Func(0)
571 END
572 CheckScriptSuccess(lines)
573
574 lines =<< trim END
575 vim9script
Bram Moolenaar28022722020-09-21 22:02:49 +0200576 def Func(..._l: list<string>)
577 echo _l
578 enddef
579 Func('a', 'b', 'c')
580 END
581 CheckScriptSuccess(lines)
582
583 lines =<< trim END
584 vim9script
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200585 def Func(...l: list<string>)
586 echo l
587 enddef
588 Func(1, 2, 3)
589 END
Bram Moolenaar77072282020-09-16 17:55:40 +0200590 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200591
592 lines =<< trim END
593 vim9script
594 def Func(...l: list<string>)
595 echo l
596 enddef
597 Func('a', 9)
598 END
Bram Moolenaar77072282020-09-16 17:55:40 +0200599 CheckScriptFailure(lines, 'E1013: Argument 2: type mismatch')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200600
601 lines =<< trim END
602 vim9script
603 def Func(...l: list<string>)
604 echo l
605 enddef
606 Func(1, 'a')
607 END
Bram Moolenaar77072282020-09-16 17:55:40 +0200608 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200609enddef
610
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200611let s:value = ''
612
613def FuncOneDefArg(opt = 'text')
614 s:value = opt
615enddef
616
617def FuncTwoDefArg(nr = 123, opt = 'text'): string
618 return nr .. opt
619enddef
620
621def FuncVarargs(...arg: list<string>): string
622 return join(arg, ',')
623enddef
624
625def Test_func_type_varargs()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200626 var RefDefArg: func(?string)
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200627 RefDefArg = FuncOneDefArg
628 RefDefArg()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200629 s:value->assert_equal('text')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200630 RefDefArg('some')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200631 s:value->assert_equal('some')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200632
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200633 var RefDef2Arg: func(?number, ?string): string
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200634 RefDef2Arg = FuncTwoDefArg
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200635 RefDef2Arg()->assert_equal('123text')
636 RefDef2Arg(99)->assert_equal('99text')
637 RefDef2Arg(77, 'some')->assert_equal('77some')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200638
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200639 CheckDefFailure(['var RefWrong: func(string?)'], 'E1010:')
640 CheckDefFailure(['var RefWrong: func(?string, string)'], 'E1007:')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200641
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200642 var RefVarargs: func(...list<string>): string
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200643 RefVarargs = FuncVarargs
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200644 RefVarargs()->assert_equal('')
645 RefVarargs('one')->assert_equal('one')
646 RefVarargs('one', 'two')->assert_equal('one,two')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200647
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200648 CheckDefFailure(['var RefWrong: func(...list<string>, string)'], 'E110:')
649 CheckDefFailure(['var RefWrong: func(...list<string>, ?string)'], 'E110:')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200650enddef
651
Bram Moolenaar0b76b422020-04-07 22:05:08 +0200652" Only varargs
653def MyVarargsOnly(...args: list<string>): string
654 return join(args, ',')
655enddef
656
657def Test_call_varargs_only()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200658 MyVarargsOnly()->assert_equal('')
659 MyVarargsOnly('one')->assert_equal('one')
660 MyVarargsOnly('one', 'two')->assert_equal('one,two')
Bram Moolenaar77072282020-09-16 17:55:40 +0200661 CheckDefFailure(['MyVarargsOnly(1)'], 'E1013: Argument 1: type mismatch, expected string but got number')
662 CheckDefFailure(['MyVarargsOnly("one", 2)'], 'E1013: Argument 2: type mismatch, expected string but got number')
Bram Moolenaar0b76b422020-04-07 22:05:08 +0200663enddef
664
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200665def Test_using_var_as_arg()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200666 writefile(['def Func(x: number)', 'var x = 234', 'enddef', 'defcompile'], 'Xdef')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200667 assert_fails('so Xdef', 'E1006:', '', 1, 'Func')
Bram Moolenaard2c61702020-09-06 15:58:36 +0200668 delete('Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200669enddef
670
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200671def DictArg(arg: dict<string>)
672 arg['key'] = 'value'
673enddef
674
675def ListArg(arg: list<string>)
676 arg[0] = 'value'
677enddef
678
679def Test_assign_to_argument()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200680 # works for dict and list
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200681 var d: dict<string> = {}
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200682 DictArg(d)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200683 d['key']->assert_equal('value')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200684 var l: list<string> = []
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200685 ListArg(l)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200686 l[0]->assert_equal('value')
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200687
Bram Moolenaard2c61702020-09-06 15:58:36 +0200688 CheckScriptFailure(['def Func(arg: number)', 'arg = 3', 'enddef', 'defcompile'], 'E1090:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100689 delfunc! g:Func
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200690enddef
691
Bram Moolenaarb816dae2020-09-20 22:04:00 +0200692" These argument names are reserved in legacy functions.
693def WithReservedNames(firstline: string, lastline: string): string
694 return firstline .. lastline
695enddef
696
697def Test_argument_names()
698 assert_equal('OK', WithReservedNames('O', 'K'))
699enddef
700
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200701def Test_call_func_defined_later()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200702 g:DefinedLater('one')->assert_equal('one')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200703 assert_fails('NotDefined("one")', 'E117:', '', 2, 'Test_call_func_defined_later')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200704enddef
705
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200706func DefinedLater(arg)
707 return a:arg
708endfunc
709
710def Test_call_funcref()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200711 g:SomeFunc('abc')->assert_equal(3)
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200712 assert_fails('NotAFunc()', 'E117:', '', 2, 'Test_call_funcref') # comment after call
713 assert_fails('g:NotAFunc()', 'E117:', '', 3, 'Test_call_funcref')
Bram Moolenaar2f1980f2020-07-22 19:30:06 +0200714
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200715 var lines =<< trim END
Bram Moolenaar2f1980f2020-07-22 19:30:06 +0200716 vim9script
717 def RetNumber(): number
718 return 123
719 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200720 var Funcref: func: number = function('RetNumber')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200721 Funcref()->assert_equal(123)
Bram Moolenaar2f1980f2020-07-22 19:30:06 +0200722 END
723 CheckScriptSuccess(lines)
Bram Moolenaar0f60e802020-07-22 20:16:11 +0200724
725 lines =<< trim END
726 vim9script
727 def RetNumber(): number
728 return 123
729 enddef
730 def Bar(F: func: number): number
731 return F()
732 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200733 var Funcref = function('RetNumber')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200734 Bar(Funcref)->assert_equal(123)
Bram Moolenaar0f60e802020-07-22 20:16:11 +0200735 END
736 CheckScriptSuccess(lines)
Bram Moolenaarbfba8652020-07-23 20:09:10 +0200737
738 lines =<< trim END
739 vim9script
740 def UseNumber(nr: number)
741 echo nr
742 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200743 var Funcref: func(number) = function('UseNumber')
Bram Moolenaarbfba8652020-07-23 20:09:10 +0200744 Funcref(123)
745 END
746 CheckScriptSuccess(lines)
Bram Moolenaarb8070e32020-07-23 20:56:04 +0200747
748 lines =<< trim END
749 vim9script
750 def UseNumber(nr: number)
751 echo nr
752 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200753 var Funcref: func(string) = function('UseNumber')
Bram Moolenaarb8070e32020-07-23 20:56:04 +0200754 END
Bram Moolenaar5e654232020-09-16 15:22:00 +0200755 CheckScriptFailure(lines, 'E1012: Type mismatch; expected func(string) but got func(number)')
Bram Moolenaar4fc224c2020-07-26 17:56:25 +0200756
757 lines =<< trim END
758 vim9script
759 def EchoNr(nr = 34)
760 g:echo = nr
761 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200762 var Funcref: func(?number) = function('EchoNr')
Bram Moolenaar4fc224c2020-07-26 17:56:25 +0200763 Funcref()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200764 g:echo->assert_equal(34)
Bram Moolenaar4fc224c2020-07-26 17:56:25 +0200765 Funcref(123)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200766 g:echo->assert_equal(123)
Bram Moolenaar4fc224c2020-07-26 17:56:25 +0200767 END
768 CheckScriptSuccess(lines)
Bram Moolenaarace61322020-07-26 18:16:58 +0200769
770 lines =<< trim END
771 vim9script
772 def EchoList(...l: list<number>)
773 g:echo = l
774 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200775 var Funcref: func(...list<number>) = function('EchoList')
Bram Moolenaarace61322020-07-26 18:16:58 +0200776 Funcref()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200777 g:echo->assert_equal([])
Bram Moolenaarace61322020-07-26 18:16:58 +0200778 Funcref(1, 2, 3)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200779 g:echo->assert_equal([1, 2, 3])
Bram Moolenaarace61322020-07-26 18:16:58 +0200780 END
781 CheckScriptSuccess(lines)
Bram Moolenaar01865ad2020-07-26 18:33:09 +0200782
783 lines =<< trim END
784 vim9script
785 def OptAndVar(nr: number, opt = 12, ...l: list<number>): number
786 g:optarg = opt
787 g:listarg = l
788 return nr
789 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200790 var Funcref: func(number, ?number, ...list<number>): number = function('OptAndVar')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200791 Funcref(10)->assert_equal(10)
792 g:optarg->assert_equal(12)
793 g:listarg->assert_equal([])
Bram Moolenaar01865ad2020-07-26 18:33:09 +0200794
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200795 Funcref(11, 22)->assert_equal(11)
796 g:optarg->assert_equal(22)
797 g:listarg->assert_equal([])
Bram Moolenaar01865ad2020-07-26 18:33:09 +0200798
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200799 Funcref(17, 18, 1, 2, 3)->assert_equal(17)
800 g:optarg->assert_equal(18)
801 g:listarg->assert_equal([1, 2, 3])
Bram Moolenaar01865ad2020-07-26 18:33:09 +0200802 END
803 CheckScriptSuccess(lines)
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200804enddef
805
806let SomeFunc = function('len')
807let NotAFunc = 'text'
808
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200809def CombineFuncrefTypes()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200810 # same arguments, different return type
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200811 var Ref1: func(bool): string
812 var Ref2: func(bool): number
813 var Ref3: func(bool): any
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200814 Ref3 = g:cond ? Ref1 : Ref2
815
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200816 # different number of arguments
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200817 var Refa1: func(bool): number
818 var Refa2: func(bool, number): number
819 var Refa3: func: number
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200820 Refa3 = g:cond ? Refa1 : Refa2
821
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200822 # different argument types
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200823 var Refb1: func(bool, string): number
824 var Refb2: func(string, number): number
825 var Refb3: func(any, any): number
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200826 Refb3 = g:cond ? Refb1 : Refb2
827enddef
828
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200829def FuncWithForwardCall()
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200830 return g:DefinedEvenLater("yes")
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200831enddef
832
833def DefinedEvenLater(arg: string): string
834 return arg
835enddef
836
837def Test_error_in_nested_function()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200838 # Error in called function requires unwinding the call stack.
Bram Moolenaar44d66522020-09-06 22:26:57 +0200839 assert_fails('FuncWithForwardCall()', 'E1096:', '', 1, 'FuncWithForwardCall')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200840enddef
841
842def Test_return_type_wrong()
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200843 CheckScriptFailure([
844 'def Func(): number',
845 'return "a"',
846 'enddef',
847 'defcompile'], 'expected number but got string')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100848 delfunc! g:Func
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200849 CheckScriptFailure([
850 'def Func(): string',
851 'return 1',
852 'enddef',
853 'defcompile'], 'expected string but got number')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100854 delfunc! g:Func
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200855 CheckScriptFailure([
856 'def Func(): void',
857 'return "a"',
858 'enddef',
859 'defcompile'],
860 'E1096: Returning a value in a function without a return type')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100861 delfunc! g:Func
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200862 CheckScriptFailure([
863 'def Func()',
864 'return "a"',
865 'enddef',
866 'defcompile'],
867 'E1096: Returning a value in a function without a return type')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100868 delfunc! g:Func
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200869
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200870 CheckScriptFailure([
871 'def Func(): number',
872 'return',
873 'enddef',
874 'defcompile'], 'E1003:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100875 delfunc! g:Func
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200876
877 CheckScriptFailure(['def Func(): list', 'return []', 'enddef'], 'E1008:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100878 delfunc! g:Func
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200879 CheckScriptFailure(['def Func(): dict', 'return {}', 'enddef'], 'E1008:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100880 delfunc! g:Func
Bram Moolenaaree4e0c12020-04-06 21:35:05 +0200881 CheckScriptFailure(['def Func()', 'return 1'], 'E1057:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100882 delfunc! g:Func
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200883
884 CheckScriptFailure([
885 'vim9script',
886 'def FuncB()',
887 ' return 123',
888 'enddef',
889 'def FuncA()',
890 ' FuncB()',
891 'enddef',
892 'defcompile'], 'E1096:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200893enddef
894
895def Test_arg_type_wrong()
896 CheckScriptFailure(['def Func3(items: list)', 'echo "a"', 'enddef'], 'E1008: Missing <type>')
Bram Moolenaaree4e0c12020-04-06 21:35:05 +0200897 CheckScriptFailure(['def Func4(...)', 'echo "a"', 'enddef'], 'E1055: Missing name after ...')
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +0200898 CheckScriptFailure(['def Func5(items:string)', 'echo "a"'], 'E1069:')
Bram Moolenaar6e949782020-04-13 17:21:00 +0200899 CheckScriptFailure(['def Func5(items)', 'echo "a"'], 'E1077:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200900enddef
901
902def Test_vim9script_call()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200903 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200904 vim9script
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200905 var name = ''
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200906 def MyFunc(arg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200907 name = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200908 enddef
909 MyFunc('foobar')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200910 name->assert_equal('foobar')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200911
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200912 var str = 'barfoo'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200913 str->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200914 name->assert_equal('barfoo')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200915
Bram Moolenaar67979662020-06-20 22:50:47 +0200916 g:value = 'value'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200917 g:value->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200918 name->assert_equal('value')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200919
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200920 var listvar = []
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200921 def ListFunc(arg: list<number>)
922 listvar = arg
923 enddef
924 [1, 2, 3]->ListFunc()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200925 listvar->assert_equal([1, 2, 3])
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200926
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200927 var dictvar = {}
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200928 def DictFunc(arg: dict<number>)
929 dictvar = arg
930 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +0100931 {a: 1, b: 2}->DictFunc()
932 dictvar->assert_equal({a: 1, b: 2})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200933 def CompiledDict()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100934 {a: 3, b: 4}->DictFunc()
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200935 enddef
936 CompiledDict()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100937 dictvar->assert_equal({a: 3, b: 4})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200938
Bram Moolenaare0de1712020-12-02 17:36:54 +0100939 {a: 3, b: 4}->DictFunc()
940 dictvar->assert_equal({a: 3, b: 4})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200941
942 ('text')->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200943 name->assert_equal('text')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200944 ("some")->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200945 name->assert_equal('some')
Bram Moolenaare6b53242020-07-01 17:28:33 +0200946
Bram Moolenaar13e12b82020-07-24 18:47:22 +0200947 # line starting with single quote is not a mark
Bram Moolenaar10409562020-07-29 20:00:38 +0200948 # line starting with double quote can be a method call
Bram Moolenaar3d48e252020-07-15 14:15:52 +0200949 'asdfasdf'->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200950 name->assert_equal('asdfasdf')
Bram Moolenaar10409562020-07-29 20:00:38 +0200951 "xyz"->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200952 name->assert_equal('xyz')
Bram Moolenaar3d48e252020-07-15 14:15:52 +0200953
954 def UseString()
955 'xyork'->MyFunc()
956 enddef
957 UseString()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200958 name->assert_equal('xyork')
Bram Moolenaar3d48e252020-07-15 14:15:52 +0200959
Bram Moolenaar10409562020-07-29 20:00:38 +0200960 def UseString2()
961 "knife"->MyFunc()
962 enddef
963 UseString2()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200964 name->assert_equal('knife')
Bram Moolenaar10409562020-07-29 20:00:38 +0200965
Bram Moolenaar13e12b82020-07-24 18:47:22 +0200966 # prepending a colon makes it a mark
967 new
968 setline(1, ['aaa', 'bbb', 'ccc'])
969 normal! 3Gmt1G
970 :'t
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200971 getcurpos()[1]->assert_equal(3)
Bram Moolenaar13e12b82020-07-24 18:47:22 +0200972 bwipe!
973
Bram Moolenaare6b53242020-07-01 17:28:33 +0200974 MyFunc(
975 'continued'
976 )
977 assert_equal('continued',
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200978 name
Bram Moolenaare6b53242020-07-01 17:28:33 +0200979 )
980
981 call MyFunc(
982 'more'
983 ..
984 'lines'
985 )
986 assert_equal(
987 'morelines',
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200988 name)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200989 END
990 writefile(lines, 'Xcall.vim')
991 source Xcall.vim
992 delete('Xcall.vim')
993enddef
994
995def Test_vim9script_call_fail_decl()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200996 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200997 vim9script
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200998 var name = ''
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200999 def MyFunc(arg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001000 var name = 123
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001001 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +02001002 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001003 END
Bram Moolenaar6c4bfe42020-07-23 18:26:30 +02001004 CheckScriptFailure(lines, 'E1054:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001005enddef
1006
Bram Moolenaar65b95452020-07-19 14:03:09 +02001007def Test_vim9script_call_fail_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001008 var lines =<< trim END
Bram Moolenaar65b95452020-07-19 14:03:09 +02001009 vim9script
1010 def MyFunc(arg: string)
1011 echo arg
1012 enddef
1013 MyFunc(1234)
1014 END
Bram Moolenaar77072282020-09-16 17:55:40 +02001015 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected string but got number')
Bram Moolenaar65b95452020-07-19 14:03:09 +02001016enddef
1017
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001018def Test_vim9script_call_fail_const()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001019 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001020 vim9script
1021 const var = ''
1022 def MyFunc(arg: string)
1023 var = 'asdf'
1024 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +02001025 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001026 END
1027 writefile(lines, 'Xcall_const.vim')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001028 assert_fails('source Xcall_const.vim', 'E46:', '', 1, 'MyFunc')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001029 delete('Xcall_const.vim')
Bram Moolenaar3bdc90b2020-12-22 20:35:40 +01001030
1031 lines =<< trim END
1032 const g:Aconst = 77
1033 def Change()
1034 # comment
1035 g:Aconst = 99
1036 enddef
1037 call Change()
1038 unlet g:Aconst
1039 END
Bram Moolenaar1dcf55d2020-12-22 22:07:30 +01001040 CheckScriptFailure(lines, 'E741: Value is locked: Aconst', 2)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001041enddef
1042
1043" Test that inside :function a Python function can be defined, :def is not
1044" recognized.
1045func Test_function_python()
1046 CheckFeature python3
Bram Moolenaar727345e2020-09-27 23:33:59 +02001047 let py = 'python3'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001048 execute py "<< EOF"
1049def do_something():
1050 return 1
1051EOF
1052endfunc
1053
1054def Test_delfunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001055 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001056 vim9script
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02001057 def g:GoneSoon()
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001058 echo 'hello'
1059 enddef
1060
1061 def CallGoneSoon()
1062 GoneSoon()
1063 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +02001064 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001065
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02001066 delfunc g:GoneSoon
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001067 CallGoneSoon()
1068 END
1069 writefile(lines, 'XToDelFunc')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001070 assert_fails('so XToDelFunc', 'E933:', '', 1, 'CallGoneSoon')
1071 assert_fails('so XToDelFunc', 'E933:', '', 1, 'CallGoneSoon')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001072
1073 delete('XToDelFunc')
1074enddef
1075
1076def Test_redef_failure()
Bram Moolenaard2c61702020-09-06 15:58:36 +02001077 writefile(['def Func0(): string', 'return "Func0"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001078 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +02001079 writefile(['def Func1(): string', 'return "Func1"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001080 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +02001081 writefile(['def! Func0(): string', 'enddef', 'defcompile'], 'Xdef')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001082 assert_fails('so Xdef', 'E1027:', '', 1, 'Func0')
Bram Moolenaard2c61702020-09-06 15:58:36 +02001083 writefile(['def Func2(): string', 'return "Func2"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001084 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +02001085 delete('Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001086
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001087 g:Func0()->assert_equal(0)
1088 g:Func1()->assert_equal('Func1')
1089 g:Func2()->assert_equal('Func2')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001090
1091 delfunc! Func0
1092 delfunc! Func1
1093 delfunc! Func2
1094enddef
1095
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +02001096def Test_vim9script_func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001097 var lines =<< trim END
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +02001098 vim9script
1099 func Func(arg)
1100 echo a:arg
1101 endfunc
1102 Func('text')
1103 END
1104 writefile(lines, 'XVim9Func')
1105 so XVim9Func
1106
1107 delete('XVim9Func')
1108enddef
1109
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001110let s:funcResult = 0
1111
1112def FuncNoArgNoRet()
Bram Moolenaar53900992020-08-22 19:02:02 +02001113 s:funcResult = 11
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001114enddef
1115
1116def FuncNoArgRetNumber(): number
Bram Moolenaar53900992020-08-22 19:02:02 +02001117 s:funcResult = 22
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001118 return 1234
1119enddef
1120
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001121def FuncNoArgRetString(): string
Bram Moolenaar53900992020-08-22 19:02:02 +02001122 s:funcResult = 45
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001123 return 'text'
1124enddef
1125
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001126def FuncOneArgNoRet(arg: number)
Bram Moolenaar53900992020-08-22 19:02:02 +02001127 s:funcResult = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001128enddef
1129
1130def FuncOneArgRetNumber(arg: number): number
Bram Moolenaar53900992020-08-22 19:02:02 +02001131 s:funcResult = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001132 return arg
1133enddef
1134
Bram Moolenaar08938ee2020-04-11 23:17:17 +02001135def FuncTwoArgNoRet(one: bool, two: number)
Bram Moolenaar53900992020-08-22 19:02:02 +02001136 s:funcResult = two
Bram Moolenaar08938ee2020-04-11 23:17:17 +02001137enddef
1138
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001139def FuncOneArgRetString(arg: string): string
1140 return arg
1141enddef
1142
Bram Moolenaar89228602020-04-05 22:14:54 +02001143def FuncOneArgRetAny(arg: any): any
1144 return arg
1145enddef
1146
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001147def Test_func_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001148 var Ref1: func()
Bram Moolenaar53900992020-08-22 19:02:02 +02001149 s:funcResult = 0
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001150 Ref1 = FuncNoArgNoRet
1151 Ref1()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001152 s:funcResult->assert_equal(11)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001153
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001154 var Ref2: func
Bram Moolenaar53900992020-08-22 19:02:02 +02001155 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001156 Ref2 = FuncNoArgNoRet
1157 Ref2()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001158 s:funcResult->assert_equal(11)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001159
Bram Moolenaar53900992020-08-22 19:02:02 +02001160 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001161 Ref2 = FuncOneArgNoRet
1162 Ref2(12)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001163 s:funcResult->assert_equal(12)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001164
Bram Moolenaar53900992020-08-22 19:02:02 +02001165 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001166 Ref2 = FuncNoArgRetNumber
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001167 Ref2()->assert_equal(1234)
1168 s:funcResult->assert_equal(22)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001169
Bram Moolenaar53900992020-08-22 19:02:02 +02001170 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001171 Ref2 = FuncOneArgRetNumber
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001172 Ref2(13)->assert_equal(13)
1173 s:funcResult->assert_equal(13)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001174enddef
1175
Bram Moolenaar9978d472020-07-05 16:01:56 +02001176def Test_repeat_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001177 var res = 0
Bram Moolenaar9978d472020-07-05 16:01:56 +02001178 for n in repeat([1], 3)
1179 res += n
1180 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001181 res->assert_equal(3)
Bram Moolenaarfce82b32020-07-05 16:07:21 +02001182
1183 res = 0
1184 for n in add([1, 2], 3)
1185 res += n
1186 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001187 res->assert_equal(6)
Bram Moolenaar9978d472020-07-05 16:01:56 +02001188enddef
1189
Bram Moolenaar846178a2020-07-05 17:04:13 +02001190def Test_argv_return_type()
1191 next fileone filetwo
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001192 var res = ''
Bram Moolenaar846178a2020-07-05 17:04:13 +02001193 for name in argv()
1194 res ..= name
1195 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001196 res->assert_equal('fileonefiletwo')
Bram Moolenaar846178a2020-07-05 17:04:13 +02001197enddef
1198
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001199def Test_func_type_part()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001200 var RefVoid: func: void
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001201 RefVoid = FuncNoArgNoRet
1202 RefVoid = FuncOneArgNoRet
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001203 CheckDefFailure(['var RefVoid: func: void', 'RefVoid = FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func(...) but got func(): number')
1204 CheckDefFailure(['var RefVoid: func: void', 'RefVoid = FuncNoArgRetString'], 'E1012: Type mismatch; expected func(...) but got func(): string')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001205
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001206 var RefAny: func(): any
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001207 RefAny = FuncNoArgRetNumber
1208 RefAny = FuncNoArgRetString
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001209 CheckDefFailure(['var RefAny: func(): any', 'RefAny = FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(): any but got func()')
1210 CheckDefFailure(['var RefAny: func(): any', 'RefAny = FuncOneArgNoRet'], 'E1012: Type mismatch; expected func(): any but got func(number)')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001211
Bram Moolenaar6abd3dc2020-10-04 14:17:32 +02001212 var RefAnyNoArgs: func: any = RefAny
1213
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001214 var RefNr: func: number
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001215 RefNr = FuncNoArgRetNumber
1216 RefNr = FuncOneArgRetNumber
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001217 CheckDefFailure(['var RefNr: func: number', 'RefNr = FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(...): number but got func()')
1218 CheckDefFailure(['var RefNr: func: number', 'RefNr = FuncNoArgRetString'], 'E1012: Type mismatch; expected func(...): number but got func(): string')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001219
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001220 var RefStr: func: string
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001221 RefStr = FuncNoArgRetString
1222 RefStr = FuncOneArgRetString
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001223 CheckDefFailure(['var RefStr: func: string', 'RefStr = FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(...): string but got func()')
1224 CheckDefFailure(['var RefStr: func: string', 'RefStr = FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func(...): string but got func(): number')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001225enddef
1226
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001227def Test_func_type_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001228 CheckDefFailure(['var ref1: func()'], 'E704:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001229
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001230 CheckDefFailure(['var Ref1: func()', 'Ref1 = FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func() but got func(): number')
1231 CheckDefFailure(['var Ref1: func()', 'Ref1 = FuncOneArgNoRet'], 'E1012: Type mismatch; expected func() but got func(number)')
1232 CheckDefFailure(['var Ref1: func()', 'Ref1 = FuncOneArgRetNumber'], 'E1012: Type mismatch; expected func() but got func(number): number')
1233 CheckDefFailure(['var Ref1: func(bool)', 'Ref1 = FuncTwoArgNoRet'], 'E1012: Type mismatch; expected func(bool) but got func(bool, number)')
1234 CheckDefFailure(['var Ref1: func(?bool)', 'Ref1 = FuncTwoArgNoRet'], 'E1012: Type mismatch; expected func(?bool) but got func(bool, number)')
1235 CheckDefFailure(['var Ref1: func(...bool)', 'Ref1 = FuncTwoArgNoRet'], 'E1012: Type mismatch; expected func(...bool) but got func(bool, number)')
Bram Moolenaar08938ee2020-04-11 23:17:17 +02001236
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001237 CheckDefFailure(['var RefWrong: func(string ,number)'], 'E1068:')
1238 CheckDefFailure(['var RefWrong: func(string,number)'], 'E1069:')
1239 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:')
1240 CheckDefFailure(['var RefWrong: func(bool):string'], 'E1069:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001241enddef
1242
Bram Moolenaar89228602020-04-05 22:14:54 +02001243def Test_func_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001244 var nr: number
Bram Moolenaar89228602020-04-05 22:14:54 +02001245 nr = FuncNoArgRetNumber()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001246 nr->assert_equal(1234)
Bram Moolenaar89228602020-04-05 22:14:54 +02001247
1248 nr = FuncOneArgRetAny(122)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001249 nr->assert_equal(122)
Bram Moolenaar89228602020-04-05 22:14:54 +02001250
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001251 var str: string
Bram Moolenaar89228602020-04-05 22:14:54 +02001252 str = FuncOneArgRetAny('yes')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001253 str->assert_equal('yes')
Bram Moolenaar89228602020-04-05 22:14:54 +02001254
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001255 CheckDefFailure(['var str: string', 'str = FuncNoArgRetNumber()'], 'E1012: Type mismatch; expected string but got number')
Bram Moolenaar89228602020-04-05 22:14:54 +02001256enddef
1257
Bram Moolenaar6abd3dc2020-10-04 14:17:32 +02001258def Test_func_common_type()
1259 def FuncOne(n: number): number
1260 return n
1261 enddef
1262 def FuncTwo(s: string): number
1263 return len(s)
1264 enddef
1265 def FuncThree(n: number, s: string): number
1266 return n + len(s)
1267 enddef
1268 var list = [FuncOne, FuncTwo, FuncThree]
1269 assert_equal(8, list[0](8))
1270 assert_equal(4, list[1]('word'))
1271 assert_equal(7, list[2](3, 'word'))
1272enddef
1273
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001274def MultiLine(
1275 arg1: string,
1276 arg2 = 1234,
1277 ...rest: list<string>
1278 ): string
1279 return arg1 .. arg2 .. join(rest, '-')
1280enddef
1281
Bram Moolenaar2c330432020-04-13 14:41:35 +02001282def MultiLineComment(
1283 arg1: string, # comment
1284 arg2 = 1234, # comment
1285 ...rest: list<string> # comment
1286 ): string # comment
1287 return arg1 .. arg2 .. join(rest, '-')
1288enddef
1289
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001290def Test_multiline()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001291 MultiLine('text')->assert_equal('text1234')
1292 MultiLine('text', 777)->assert_equal('text777')
1293 MultiLine('text', 777, 'one')->assert_equal('text777one')
1294 MultiLine('text', 777, 'one', 'two')->assert_equal('text777one-two')
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001295enddef
1296
Bram Moolenaar23e03252020-04-12 22:22:31 +02001297func Test_multiline_not_vim9()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001298 call MultiLine('text')->assert_equal('text1234')
1299 call MultiLine('text', 777)->assert_equal('text777')
1300 call MultiLine('text', 777, 'one')->assert_equal('text777one')
1301 call MultiLine('text', 777, 'one', 'two')->assert_equal('text777one-two')
Bram Moolenaar23e03252020-04-12 22:22:31 +02001302endfunc
1303
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001304
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02001305" When using CheckScriptFailure() for the below test, E1010 is generated instead
1306" of E1056.
1307func Test_E1056_1059()
1308 let caught_1056 = 0
1309 try
1310 def F():
1311 return 1
1312 enddef
1313 catch /E1056:/
1314 let caught_1056 = 1
1315 endtry
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001316 eval caught_1056->assert_equal(1)
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02001317
1318 let caught_1059 = 0
1319 try
1320 def F5(items : list)
1321 echo 'a'
1322 enddef
1323 catch /E1059:/
1324 let caught_1059 = 1
1325 endtry
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001326 eval caught_1059->assert_equal(1)
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02001327endfunc
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001328
Bram Moolenaar015f4262020-05-05 21:25:22 +02001329func DelMe()
1330 echo 'DelMe'
1331endfunc
1332
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001333def Test_error_reporting()
1334 # comment lines at the start of the function
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001335 var lines =<< trim END
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001336 " comment
1337 def Func()
1338 # comment
1339 # comment
1340 invalid
1341 enddef
1342 defcompile
1343 END
Bram Moolenaar08052222020-09-14 17:04:31 +02001344 writefile(lines, 'Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001345 try
1346 source Xdef
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001347 assert_report('should have failed')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001348 catch /E476:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001349 v:exception->assert_match('Invalid command: invalid')
1350 v:throwpoint->assert_match(', line 3$')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001351 endtry
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001352 delfunc! g:Func
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001353
1354 # comment lines after the start of the function
1355 lines =<< trim END
1356 " comment
1357 def Func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001358 var x = 1234
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001359 # comment
1360 # comment
1361 invalid
1362 enddef
1363 defcompile
1364 END
Bram Moolenaar08052222020-09-14 17:04:31 +02001365 writefile(lines, 'Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001366 try
1367 source Xdef
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001368 assert_report('should have failed')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001369 catch /E476:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001370 v:exception->assert_match('Invalid command: invalid')
1371 v:throwpoint->assert_match(', line 4$')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001372 endtry
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001373 delfunc! g:Func
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001374
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001375 lines =<< trim END
1376 vim9script
1377 def Func()
Bram Moolenaare0de1712020-12-02 17:36:54 +01001378 var db = {foo: 1, bar: 2}
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001379 # comment
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001380 var x = db.asdf
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001381 enddef
1382 defcompile
1383 Func()
1384 END
Bram Moolenaar08052222020-09-14 17:04:31 +02001385 writefile(lines, 'Xdef')
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001386 try
1387 source Xdef
1388 assert_report('should have failed')
1389 catch /E716:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001390 v:throwpoint->assert_match('_Func, line 3$')
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001391 endtry
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001392 delfunc! g:Func
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001393
Bram Moolenaar08052222020-09-14 17:04:31 +02001394 delete('Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001395enddef
1396
Bram Moolenaar015f4262020-05-05 21:25:22 +02001397def Test_deleted_function()
1398 CheckDefExecFailure([
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001399 'var RefMe: func = function("g:DelMe")',
Bram Moolenaar015f4262020-05-05 21:25:22 +02001400 'delfunc g:DelMe',
1401 'echo RefMe()'], 'E117:')
1402enddef
1403
1404def Test_unknown_function()
1405 CheckDefExecFailure([
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001406 'var Ref: func = function("NotExist")',
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +02001407 'delfunc g:NotExist'], 'E700:')
Bram Moolenaar015f4262020-05-05 21:25:22 +02001408enddef
1409
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +02001410def RefFunc(Ref: func(string): string): string
1411 return Ref('more')
1412enddef
1413
1414def Test_closure_simple()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001415 var local = 'some '
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001416 RefFunc((s) => local .. s)->assert_equal('some more')
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +02001417enddef
1418
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001419def MakeRef()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001420 var local = 'some '
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001421 g:Ref = (s) => local .. s
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001422enddef
1423
1424def Test_closure_ref_after_return()
1425 MakeRef()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001426 g:Ref('thing')->assert_equal('some thing')
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001427 unlet g:Ref
1428enddef
1429
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001430def MakeTwoRefs()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001431 var local = ['some']
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001432 g:Extend = (s) => local->add(s)
1433 g:Read = () => local
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001434enddef
1435
1436def Test_closure_two_refs()
1437 MakeTwoRefs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001438 join(g:Read(), ' ')->assert_equal('some')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001439 g:Extend('more')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001440 join(g:Read(), ' ')->assert_equal('some more')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001441 g:Extend('even')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001442 join(g:Read(), ' ')->assert_equal('some more even')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001443
1444 unlet g:Extend
1445 unlet g:Read
1446enddef
1447
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001448def ReadRef(Ref: func(): list<string>): string
1449 return join(Ref(), ' ')
1450enddef
1451
Bram Moolenaar5e654232020-09-16 15:22:00 +02001452def ExtendRef(Ref: func(string): list<string>, add: string)
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001453 Ref(add)
1454enddef
1455
1456def Test_closure_two_indirect_refs()
Bram Moolenaarf7779c62020-05-03 15:38:16 +02001457 MakeTwoRefs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001458 ReadRef(g:Read)->assert_equal('some')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001459 ExtendRef(g:Extend, 'more')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001460 ReadRef(g:Read)->assert_equal('some more')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001461 ExtendRef(g:Extend, 'even')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001462 ReadRef(g:Read)->assert_equal('some more even')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001463
1464 unlet g:Extend
1465 unlet g:Read
1466enddef
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001467
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001468def MakeArgRefs(theArg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001469 var local = 'loc_val'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001470 g:UseArg = (s) => theArg .. '/' .. local .. '/' .. s
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001471enddef
1472
1473def MakeArgRefsVarargs(theArg: string, ...rest: list<string>)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001474 var local = 'the_loc'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001475 g:UseVararg = (s) => theArg .. '/' .. local .. '/' .. s .. '/' .. join(rest)
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001476enddef
1477
1478def Test_closure_using_argument()
1479 MakeArgRefs('arg_val')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001480 g:UseArg('call_val')->assert_equal('arg_val/loc_val/call_val')
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001481
1482 MakeArgRefsVarargs('arg_val', 'one', 'two')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001483 g:UseVararg('call_val')->assert_equal('arg_val/the_loc/call_val/one two')
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001484
1485 unlet g:UseArg
1486 unlet g:UseVararg
1487enddef
1488
Bram Moolenaar85d5e2b2020-10-10 14:13:01 +02001489def MakeGetAndAppendRefs()
1490 var local = 'a'
1491
1492 def Append(arg: string)
1493 local ..= arg
1494 enddef
1495 g:Append = Append
1496
1497 def Get(): string
1498 return local
1499 enddef
1500 g:Get = Get
1501enddef
1502
1503def Test_closure_append_get()
1504 MakeGetAndAppendRefs()
1505 g:Get()->assert_equal('a')
1506 g:Append('-b')
1507 g:Get()->assert_equal('a-b')
1508 g:Append('-c')
1509 g:Get()->assert_equal('a-b-c')
1510
1511 unlet g:Append
1512 unlet g:Get
1513enddef
Bram Moolenaarb68b3462020-05-06 21:06:30 +02001514
Bram Moolenaar04b12692020-05-04 23:24:44 +02001515def Test_nested_closure()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001516 var local = 'text'
Bram Moolenaar04b12692020-05-04 23:24:44 +02001517 def Closure(arg: string): string
1518 return local .. arg
1519 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001520 Closure('!!!')->assert_equal('text!!!')
Bram Moolenaar04b12692020-05-04 23:24:44 +02001521enddef
1522
Bram Moolenaar6f5b6df2020-05-16 21:20:12 +02001523func GetResult(Ref)
1524 return a:Ref('some')
1525endfunc
1526
1527def Test_call_closure_not_compiled()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001528 var text = 'text'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001529 g:Ref = (s) => s .. text
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001530 GetResult(g:Ref)->assert_equal('sometext')
Bram Moolenaar6f5b6df2020-05-16 21:20:12 +02001531enddef
1532
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001533def Test_double_closure_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001534 var lines =<< trim END
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001535 vim9script
1536 def Func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001537 var name = 0
1538 for i in range(2)
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001539 timer_start(0, () => name)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001540 endfor
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001541 enddef
1542 Func()
1543 END
Bram Moolenaar148ce7a2020-09-23 21:57:23 +02001544 CheckScriptSuccess(lines)
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001545enddef
1546
Bram Moolenaar85d5e2b2020-10-10 14:13:01 +02001547def Test_nested_closure_used()
1548 var lines =<< trim END
1549 vim9script
1550 def Func()
1551 var x = 'hello'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001552 var Closure = () => x
1553 g:Myclosure = () => Closure()
Bram Moolenaar85d5e2b2020-10-10 14:13:01 +02001554 enddef
1555 Func()
1556 assert_equal('hello', g:Myclosure())
1557 END
1558 CheckScriptSuccess(lines)
1559enddef
Bram Moolenaar0876c782020-10-07 19:08:04 +02001560
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02001561def Test_nested_closure_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001562 var lines =<< trim END
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02001563 vim9script
1564 def FuncA()
1565 FuncB(0)
1566 enddef
1567 def FuncB(n: number): list<string>
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001568 return map([0], (_, v) => n)
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02001569 enddef
1570 FuncA()
1571 END
1572 CheckScriptFailure(lines, 'E1012:')
1573enddef
1574
Bram Moolenaarf112f302020-12-20 17:47:52 +01001575def Test_global_closure()
1576 var lines =<< trim END
1577 vim9script
1578 def ReverseEveryNLines(n: number, line1: number, line2: number)
1579 var mods = 'sil keepj keepp lockm '
1580 var range = ':' .. line1 .. ',' .. line2
1581 def g:Offset(): number
1582 var offset = (line('.') - line1 + 1) % n
1583 return offset != 0 ? offset : n
1584 enddef
1585 exe mods .. range .. 'g/^/exe "m .-" .. g:Offset()'
1586 enddef
1587
1588 new
1589 repeat(['aaa', 'bbb', 'ccc'], 3)->setline(1)
1590 ReverseEveryNLines(3, 1, 9)
1591 END
1592 CheckScriptSuccess(lines)
1593 var expected = repeat(['ccc', 'bbb', 'aaa'], 3)
1594 assert_equal(expected, getline(1, 9))
1595 bwipe!
1596enddef
1597
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01001598def Test_global_closure_called_directly()
1599 var lines =<< trim END
1600 vim9script
1601 def Outer()
1602 var x = 1
1603 def g:Inner()
1604 var y = x
1605 x += 1
1606 assert_equal(1, y)
1607 enddef
1608 g:Inner()
1609 assert_equal(2, x)
1610 enddef
1611 Outer()
1612 END
1613 CheckScriptSuccess(lines)
1614 delfunc g:Inner
1615enddef
1616
Bram Moolenaar34c54eb2020-11-25 19:15:19 +01001617def Test_failure_in_called_function()
1618 # this was using the frame index as the return value
1619 var lines =<< trim END
1620 vim9script
1621 au TerminalWinOpen * eval [][0]
1622 def PopupTerm(a: any)
1623 # make sure typvals on stack are string
1624 ['a', 'b', 'c', 'd', 'e', 'f', 'g']->join()
1625 FireEvent()
1626 enddef
1627 def FireEvent()
1628 do TerminalWinOpen
1629 enddef
1630 # use try/catch to make eval fail
1631 try
1632 call PopupTerm(0)
1633 catch
1634 endtry
1635 au! TerminalWinOpen
1636 END
1637 CheckScriptSuccess(lines)
1638enddef
1639
Bram Moolenaar5366e1a2020-10-01 13:01:34 +02001640def Test_nested_lambda()
1641 var lines =<< trim END
1642 vim9script
1643 def Func()
1644 var x = 4
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001645 var Lambda1 = () => 7
1646 var Lambda2 = () => [Lambda1(), x]
Bram Moolenaar5366e1a2020-10-01 13:01:34 +02001647 var res = Lambda2()
1648 assert_equal([7, 4], res)
1649 enddef
1650 Func()
1651 END
1652 CheckScriptSuccess(lines)
1653enddef
1654
Bram Moolenaar52bf81c2020-11-17 18:50:44 +01001655def Shadowed(): list<number>
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001656 var FuncList: list<func: number> = [() => 42]
1657 return FuncList->map((_, Shadowed) => Shadowed())
Bram Moolenaar52bf81c2020-11-17 18:50:44 +01001658enddef
1659
1660def Test_lambda_arg_shadows_func()
1661 assert_equal([42], Shadowed())
1662enddef
1663
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02001664def Line_continuation_in_def(dir: string = ''): string
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001665 var path: string = empty(dir)
1666 \ ? 'empty'
1667 \ : 'full'
1668 return path
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02001669enddef
1670
1671def Test_line_continuation_in_def()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001672 Line_continuation_in_def('.')->assert_equal('full')
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02001673enddef
1674
Bram Moolenaar2ea95b62020-11-19 21:47:56 +01001675def Test_script_var_in_lambda()
1676 var lines =<< trim END
1677 vim9script
1678 var script = 'test'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001679 assert_equal(['test'], map(['one'], () => script))
Bram Moolenaar2ea95b62020-11-19 21:47:56 +01001680 END
1681 CheckScriptSuccess(lines)
1682enddef
1683
Bram Moolenaar5e654232020-09-16 15:22:00 +02001684def Line_continuation_in_lambda(): list<string>
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001685 var x = range(97, 100)
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001686 ->map((_, v) => nr2char(v)
1687 ->toupper())
Bram Moolenaar7a4b8982020-07-08 17:36:21 +02001688 ->reverse()
1689 return x
1690enddef
1691
1692def Test_line_continuation_in_lambda()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001693 Line_continuation_in_lambda()->assert_equal(['D', 'C', 'B', 'A'])
Bram Moolenaar7a4b8982020-07-08 17:36:21 +02001694enddef
1695
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001696func Test_silent_echo()
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001697 CheckScreendump
1698
1699 let lines =<< trim END
1700 vim9script
1701 def EchoNothing()
1702 silent echo ''
1703 enddef
1704 defcompile
1705 END
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001706 call writefile(lines, 'XTest_silent_echo')
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001707
1708 " Check that the balloon shows up after a mouse move
1709 let buf = RunVimInTerminal('-S XTest_silent_echo', {'rows': 6})
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001710 call term_sendkeys(buf, ":abc")
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001711 call VerifyScreenDump(buf, 'Test_vim9_silent_echo', {})
1712
1713 " clean up
1714 call StopVimInTerminal(buf)
1715 call delete('XTest_silent_echo')
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001716endfunc
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001717
Bram Moolenaar171fb922020-10-28 16:54:47 +01001718def SilentlyError()
1719 execute('silent! invalid')
1720 g:did_it = 'yes'
1721enddef
1722
Bram Moolenaar28ee8922020-10-28 20:20:00 +01001723func UserError()
1724 silent! invalid
1725endfunc
1726
1727def SilentlyUserError()
1728 UserError()
1729 g:did_it = 'yes'
1730enddef
Bram Moolenaar171fb922020-10-28 16:54:47 +01001731
1732" This can't be a :def function, because the assert would not be reached.
Bram Moolenaar171fb922020-10-28 16:54:47 +01001733func Test_ignore_silent_error()
1734 let g:did_it = 'no'
1735 call SilentlyError()
1736 call assert_equal('yes', g:did_it)
1737
Bram Moolenaar28ee8922020-10-28 20:20:00 +01001738 let g:did_it = 'no'
1739 call SilentlyUserError()
1740 call assert_equal('yes', g:did_it)
Bram Moolenaar171fb922020-10-28 16:54:47 +01001741
1742 unlet g:did_it
1743endfunc
1744
Bram Moolenaarcd030c42020-10-30 21:49:40 +01001745def Test_ignore_silent_error_in_filter()
1746 var lines =<< trim END
1747 vim9script
1748 def Filter(winid: number, key: string): bool
1749 if key == 'o'
1750 silent! eval [][0]
1751 return true
1752 endif
1753 return popup_filter_menu(winid, key)
1754 enddef
1755
Bram Moolenaare0de1712020-12-02 17:36:54 +01001756 popup_create('popup', {filter: Filter})
Bram Moolenaarcd030c42020-10-30 21:49:40 +01001757 feedkeys("o\r", 'xnt')
1758 END
1759 CheckScriptSuccess(lines)
1760enddef
1761
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001762def Fibonacci(n: number): number
1763 if n < 2
1764 return n
1765 else
1766 return Fibonacci(n - 1) + Fibonacci(n - 2)
1767 endif
1768enddef
1769
Bram Moolenaar985116a2020-07-12 17:31:09 +02001770def Test_recursive_call()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001771 Fibonacci(20)->assert_equal(6765)
Bram Moolenaar985116a2020-07-12 17:31:09 +02001772enddef
1773
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001774def TreeWalk(dir: string): list<any>
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001775 return readdir(dir)->map((_, val) =>
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001776 fnamemodify(dir .. '/' .. val, ':p')->isdirectory()
Bram Moolenaar2bede172020-11-19 18:53:18 +01001777 ? {[val]: TreeWalk(dir .. '/' .. val)}
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001778 : val
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001779 )
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001780enddef
1781
1782def Test_closure_in_map()
1783 mkdir('XclosureDir/tdir', 'p')
1784 writefile(['111'], 'XclosureDir/file1')
1785 writefile(['222'], 'XclosureDir/file2')
1786 writefile(['333'], 'XclosureDir/tdir/file3')
1787
Bram Moolenaare0de1712020-12-02 17:36:54 +01001788 TreeWalk('XclosureDir')->assert_equal(['file1', 'file2', {tdir: ['file3']}])
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001789
1790 delete('XclosureDir', 'rf')
1791enddef
1792
Bram Moolenaar7b5d5442020-10-04 13:42:34 +02001793def Test_invalid_function_name()
1794 var lines =<< trim END
1795 vim9script
1796 def s: list<string>
1797 END
1798 CheckScriptFailure(lines, 'E129:')
1799
1800 lines =<< trim END
1801 vim9script
1802 def g: list<string>
1803 END
1804 CheckScriptFailure(lines, 'E129:')
1805
1806 lines =<< trim END
1807 vim9script
1808 def <SID>: list<string>
1809 END
1810 CheckScriptFailure(lines, 'E884:')
1811
1812 lines =<< trim END
1813 vim9script
1814 def F list<string>
1815 END
1816 CheckScriptFailure(lines, 'E488:')
1817enddef
1818
Bram Moolenaara90afb92020-07-15 22:38:56 +02001819def Test_partial_call()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001820 var Xsetlist = function('setloclist', [0])
Bram Moolenaare0de1712020-12-02 17:36:54 +01001821 Xsetlist([], ' ', {title: 'test'})
1822 getloclist(0, {title: 1})->assert_equal({title: 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02001823
1824 Xsetlist = function('setloclist', [0, [], ' '])
Bram Moolenaare0de1712020-12-02 17:36:54 +01001825 Xsetlist({title: 'test'})
1826 getloclist(0, {title: 1})->assert_equal({title: 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02001827
1828 Xsetlist = function('setqflist')
Bram Moolenaare0de1712020-12-02 17:36:54 +01001829 Xsetlist([], ' ', {title: 'test'})
1830 getqflist({title: 1})->assert_equal({title: 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02001831
1832 Xsetlist = function('setqflist', [[], ' '])
Bram Moolenaare0de1712020-12-02 17:36:54 +01001833 Xsetlist({title: 'test'})
1834 getqflist({title: 1})->assert_equal({title: 'test'})
Bram Moolenaar6abd3dc2020-10-04 14:17:32 +02001835
1836 var Len: func: number = function('len', ['word'])
1837 assert_equal(4, Len())
Bram Moolenaara90afb92020-07-15 22:38:56 +02001838enddef
1839
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001840def Test_cmd_modifier()
1841 tab echo '0'
Bram Moolenaard2c61702020-09-06 15:58:36 +02001842 CheckDefFailure(['5tab echo 3'], 'E16:')
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001843enddef
1844
1845def Test_restore_modifiers()
1846 # check that when compiling a :def function command modifiers are not messed
1847 # up.
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001848 var lines =<< trim END
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001849 vim9script
1850 set eventignore=
1851 autocmd QuickFixCmdPost * copen
1852 def AutocmdsDisabled()
Bram Moolenaar6cf7e3b2020-10-28 14:31:16 +01001853 eval 0
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001854 enddef
1855 func Func()
1856 noautocmd call s:AutocmdsDisabled()
1857 let g:ei_after = &eventignore
1858 endfunc
1859 Func()
1860 END
1861 CheckScriptSuccess(lines)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001862 g:ei_after->assert_equal('')
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001863enddef
1864
Bram Moolenaardfa3d552020-09-10 22:05:08 +02001865def StackTop()
1866 eval 1
1867 eval 2
1868 # call not on fourth line
1869 StackBot()
1870enddef
1871
1872def StackBot()
1873 # throw an error
1874 eval [][0]
1875enddef
1876
1877def Test_callstack_def()
1878 try
1879 StackTop()
1880 catch
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001881 v:throwpoint->assert_match('Test_callstack_def\[2\]..StackTop\[4\]..StackBot, line 2')
Bram Moolenaardfa3d552020-09-10 22:05:08 +02001882 endtry
1883enddef
1884
Bram Moolenaare8211a32020-10-09 22:04:29 +02001885" Re-using spot for variable used in block
1886def Test_block_scoped_var()
1887 var lines =<< trim END
1888 vim9script
1889 def Func()
1890 var x = ['a', 'b', 'c']
1891 if 1
1892 var y = 'x'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001893 map(x, () => y)
Bram Moolenaare8211a32020-10-09 22:04:29 +02001894 endif
1895 var z = x
1896 assert_equal(['x', 'x', 'x'], z)
1897 enddef
1898 Func()
1899 END
1900 CheckScriptSuccess(lines)
1901enddef
1902
Bram Moolenaareeece9e2020-11-20 19:26:48 +01001903def Test_reset_did_emsg()
1904 var lines =<< trim END
1905 @s = 'blah'
1906 au BufWinLeave * #
1907 def Func()
1908 var winid = popup_create('popup', {})
1909 exe '*s'
1910 popup_close(winid)
1911 enddef
1912 Func()
1913 END
1914 CheckScriptFailure(lines, 'E492:', 8)
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001915 delfunc! g:Func
Bram Moolenaareeece9e2020-11-20 19:26:48 +01001916enddef
1917
Bram Moolenaar57f799e2020-12-12 20:42:19 +01001918def Test_did_emsg_reset()
1919 # executing an autocommand resets did_emsg, this should not result in a
1920 # builtin function considered failing
1921 var lines =<< trim END
1922 vim9script
1923 au BufWinLeave * #
1924 def Func()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001925 popup_menu('', {callback: () => popup_create('', {})->popup_close()})
Bram Moolenaar57f799e2020-12-12 20:42:19 +01001926 eval [][0]
1927 enddef
1928 nno <F3> <cmd>call <sid>Func()<cr>
1929 feedkeys("\<F3>\e", 'xt')
1930 END
1931 writefile(lines, 'XemsgReset')
1932 assert_fails('so XemsgReset', ['E684:', 'E684:'], lines, 2)
1933 delete('XemsgReset')
1934 nunmap <F3>
1935 au! BufWinLeave
1936enddef
1937
Bram Moolenaar56602ba2020-12-05 21:22:08 +01001938def Test_abort_with_silent_call()
1939 var lines =<< trim END
1940 vim9script
1941 g:result = 'none'
1942 def Func()
1943 g:result += 3
1944 g:result = 'yes'
1945 enddef
1946 # error is silenced, but function aborts on error
1947 silent! Func()
1948 assert_equal('none', g:result)
1949 unlet g:result
1950 END
1951 CheckScriptSuccess(lines)
1952enddef
1953
Bram Moolenaarf665e972020-12-05 19:17:16 +01001954def Test_continues_with_silent_error()
1955 var lines =<< trim END
1956 vim9script
1957 g:result = 'none'
1958 def Func()
1959 silent! g:result += 3
1960 g:result = 'yes'
1961 enddef
1962 # error is silenced, function does not abort
1963 Func()
1964 assert_equal('yes', g:result)
1965 unlet g:result
1966 END
1967 CheckScriptSuccess(lines)
1968enddef
1969
Bram Moolenaaraf0df472020-12-02 20:51:22 +01001970def Test_abort_even_with_silent()
1971 var lines =<< trim END
1972 vim9script
1973 g:result = 'none'
1974 def Func()
1975 eval {-> ''}() .. '' .. {}['X']
1976 g:result = 'yes'
1977 enddef
Bram Moolenaarf665e972020-12-05 19:17:16 +01001978 silent! Func()
Bram Moolenaaraf0df472020-12-02 20:51:22 +01001979 assert_equal('none', g:result)
Bram Moolenaar4029cab2020-12-05 18:13:27 +01001980 unlet g:result
1981 END
1982 CheckScriptSuccess(lines)
1983enddef
1984
Bram Moolenaarf665e972020-12-05 19:17:16 +01001985def Test_cmdmod_silent_restored()
1986 var lines =<< trim END
1987 vim9script
1988 def Func()
1989 g:result = 'none'
1990 silent! g:result += 3
1991 g:result = 'none'
1992 g:result += 3
1993 enddef
1994 Func()
1995 END
1996 # can't use CheckScriptFailure, it ignores the :silent!
1997 var fname = 'Xdefsilent'
1998 writefile(lines, fname)
1999 var caught = 'no'
2000 try
2001 exe 'source ' .. fname
2002 catch /E1030:/
2003 caught = 'yes'
2004 assert_match('Func, line 4', v:throwpoint)
2005 endtry
2006 assert_equal('yes', caught)
2007 delete(fname)
2008enddef
2009
Bram Moolenaar4029cab2020-12-05 18:13:27 +01002010def Test_dict_member_with_silent()
2011 var lines =<< trim END
2012 vim9script
2013 g:result = 'none'
2014 var d: dict<any>
2015 def Func()
2016 try
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002017 g:result = map([], (_, v) => ({}[v]))->join() .. d['']
Bram Moolenaar4029cab2020-12-05 18:13:27 +01002018 catch
2019 endtry
2020 enddef
2021 silent! Func()
2022 assert_equal('0', g:result)
2023 unlet g:result
Bram Moolenaaraf0df472020-12-02 20:51:22 +01002024 END
2025 CheckScriptSuccess(lines)
2026enddef
2027
Bram Moolenaar5b3d1bb2020-12-22 12:20:08 +01002028def Test_opfunc()
2029 nnoremap <F3> <cmd>set opfunc=Opfunc<cr>g@
2030 def g:Opfunc(_: any): string
2031 setline(1, 'ASDF')
2032 return ''
2033 enddef
2034 new
2035 setline(1, 'asdf')
2036 feedkeys("\<F3>$", 'x')
2037 assert_equal('ASDF', getline(1))
2038
2039 bwipe!
2040 nunmap <F3>
2041enddef
2042
Bram Moolenaar077a4232020-12-22 18:33:27 +01002043" this was crashing on exit
2044def Test_nested_lambda_in_closure()
2045 var lines =<< trim END
2046 vim9script
2047 def Outer()
2048 def g:Inner()
2049 echo map([1, 2, 3], {_, v -> v + 1})
2050 enddef
2051 g:Inner()
2052 enddef
2053 defcompile
2054 writefile(['Done'], 'XnestedDone')
2055 quit
2056 END
2057 if !RunVim([], lines, '--clean')
2058 return
2059 endif
2060 assert_equal(['Done'], readfile('XnestedDone'))
2061 delete('XnestedDone')
2062enddef
2063
2064
Bram Moolenaarf7779c62020-05-03 15:38:16 +02002065
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002066" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker