blob: 20ad3ba8e4e02fea74b69095f69caf1768fdedf9 [file] [log] [blame]
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001" Test various aspects of the Vim9 script language.
2
3source check.vim
Bram Moolenaarad304702020-09-06 18:22:53 +02004source term_util.vim
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02005source view_util.vim
Bram Moolenaar04b12692020-05-04 23:24:44 +02006source vim9.vim
Bram Moolenaar47e7d702020-07-05 18:18:42 +02007source screendump.vim
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02008
9func Test_def_basic()
10 def SomeFunc(): string
11 return 'yes'
12 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +020013 call SomeFunc()->assert_equal('yes')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +020014endfunc
15
Bram Moolenaar2b9b17e2020-10-13 18:38:11 +020016func Test_compiling_error()
17 " use a terminal to see the whole error message
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +020018 CheckRunVimInTerminal
19
Bram Moolenaar2b9b17e2020-10-13 18:38:11 +020020 call TestCompilingError()
21endfunc
22
23def TestCompilingError()
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +020024 var lines =<< trim END
25 vim9script
26 def Fails()
27 echo nothing
28 enddef
29 defcompile
30 END
31 call writefile(lines, 'XTest_compile_error')
32 var buf = RunVimInTerminal('-S XTest_compile_error',
Bram Moolenaare0de1712020-12-02 17:36:54 +010033 {rows: 10, wait_for_ruler: 0})
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +020034 var text = ''
35 for loop in range(100)
36 text = ''
37 for i in range(1, 9)
38 text ..= term_getline(buf, i)
39 endfor
Bram Moolenaar18dc3552020-11-22 14:24:00 +010040 if text =~ 'Variable not found: nothing'
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +020041 break
42 endif
43 sleep 20m
44 endfor
45 assert_match('Error detected while compiling command line.*Fails.*Variable not found: nothing', text)
46
47 # clean up
48 call StopVimInTerminal(buf)
49 call delete('XTest_compile_error')
50enddef
51
Bram Moolenaar0ba48e82020-11-17 18:23:19 +010052def CallRecursive(n: number): number
53 return CallRecursive(n + 1)
54enddef
55
56def CallMapRecursive(l: list<number>): number
Bram Moolenaar2949cfd2020-12-31 21:28:47 +010057 return map(l, (_, v) => CallMapRecursive([v]))[0]
Bram Moolenaar0ba48e82020-11-17 18:23:19 +010058enddef
59
60def Test_funcdepth_error()
61 set maxfuncdepth=10
62
63 var caught = false
64 try
65 CallRecursive(1)
66 catch /E132:/
67 caught = true
68 endtry
69 assert_true(caught)
70
71 caught = false
72 try
73 CallMapRecursive([1])
74 catch /E132:/
75 caught = true
76 endtry
77 assert_true(caught)
78
79 set maxfuncdepth&
80enddef
81
Bram Moolenaar5178b1b2021-01-01 18:43:51 +010082def Test_endfunc_enddef()
83 var lines =<< trim END
84 def Test()
85 echo 'test'
86 endfunc
87 enddef
88 END
89 CheckScriptFailure(lines, 'E1151:', 3)
90
91 lines =<< trim END
92 def Test()
93 func Nested()
94 echo 'test'
95 enddef
96 enddef
97 END
98 CheckScriptFailure(lines, 'E1152:', 4)
99enddef
100
Bram Moolenaarb8ba9b92021-01-01 18:54:34 +0100101def Test_missing_endfunc_enddef()
102 var lines =<< trim END
103 vim9script
104 def Test()
105 echo 'test'
106 endef
107 END
108 CheckScriptFailure(lines, 'E1057:', 2)
109
110 lines =<< trim END
111 vim9script
112 func Some()
113 echo 'test'
114 enfffunc
115 END
116 CheckScriptFailure(lines, 'E126:', 2)
117enddef
118
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200119def ReturnString(): string
120 return 'string'
121enddef
122
123def ReturnNumber(): number
124 return 123
125enddef
126
127let g:notNumber = 'string'
128
129def ReturnGlobal(): number
130 return g:notNumber
131enddef
132
133def Test_return_something()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200134 ReturnString()->assert_equal('string')
135 ReturnNumber()->assert_equal(123)
Bram Moolenaar5e654232020-09-16 15:22:00 +0200136 assert_fails('ReturnGlobal()', 'E1012: Type mismatch; expected number but got string', '', 1, 'ReturnGlobal')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200137enddef
138
Bram Moolenaarefd88552020-06-18 20:50:10 +0200139def Test_missing_return()
140 CheckDefFailure(['def Missing(): number',
141 ' if g:cond',
142 ' echo "no return"',
143 ' else',
144 ' return 0',
145 ' endif'
146 'enddef'], 'E1027:')
147 CheckDefFailure(['def Missing(): number',
148 ' if g:cond',
149 ' return 1',
150 ' else',
151 ' echo "no return"',
152 ' endif'
153 'enddef'], 'E1027:')
154 CheckDefFailure(['def Missing(): number',
155 ' if g:cond',
156 ' return 1',
157 ' else',
158 ' return 2',
159 ' endif'
160 ' return 3'
161 'enddef'], 'E1095:')
162enddef
163
Bram Moolenaar403dc312020-10-17 19:29:51 +0200164def Test_return_bool()
165 var lines =<< trim END
166 vim9script
167 def MenuFilter(id: number, key: string): bool
168 return popup_filter_menu(id, key)
169 enddef
170 def YesnoFilter(id: number, key: string): bool
171 return popup_filter_yesno(id, key)
172 enddef
173 defcompile
174 END
175 CheckScriptSuccess(lines)
176enddef
177
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200178let s:nothing = 0
179def ReturnNothing()
180 s:nothing = 1
181 if true
182 return
183 endif
184 s:nothing = 2
185enddef
186
187def Test_return_nothing()
188 ReturnNothing()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200189 s:nothing->assert_equal(1)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200190enddef
191
192func Increment()
193 let g:counter += 1
194endfunc
195
196def Test_call_ufunc_count()
197 g:counter = 1
198 Increment()
199 Increment()
200 Increment()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200201 # works with and without :call
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200202 g:counter->assert_equal(4)
203 eval g:counter->assert_equal(4)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200204 unlet g:counter
205enddef
206
207def MyVarargs(arg: string, ...rest: list<string>): string
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200208 var res = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200209 for s in rest
210 res ..= ',' .. s
211 endfor
212 return res
213enddef
214
215def Test_call_varargs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200216 MyVarargs('one')->assert_equal('one')
217 MyVarargs('one', 'two')->assert_equal('one,two')
218 MyVarargs('one', 'two', 'three')->assert_equal('one,two,three')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200219enddef
220
221def MyDefaultArgs(name = 'string'): string
222 return name
223enddef
224
Bram Moolenaare30f64b2020-07-15 19:48:20 +0200225def MyDefaultSecond(name: string, second: bool = true): string
226 return second ? name : 'none'
227enddef
228
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200229def Test_call_default_args()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200230 MyDefaultArgs()->assert_equal('string')
231 MyDefaultArgs('one')->assert_equal('one')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200232 assert_fails('MyDefaultArgs("one", "two")', 'E118:', '', 3, 'Test_call_default_args')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200233
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200234 MyDefaultSecond('test')->assert_equal('test')
235 MyDefaultSecond('test', true)->assert_equal('test')
236 MyDefaultSecond('test', false)->assert_equal('none')
Bram Moolenaare30f64b2020-07-15 19:48:20 +0200237
Bram Moolenaar822ba242020-05-24 23:00:18 +0200238 CheckScriptFailure(['def Func(arg: number = asdf)', 'enddef', 'defcompile'], 'E1001:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100239 delfunc g:Func
Bram Moolenaar77072282020-09-16 17:55:40 +0200240 CheckScriptFailure(['def Func(arg: number = "text")', 'enddef', 'defcompile'], 'E1013: Argument 1: type mismatch, expected number but got string')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100241 delfunc g:Func
Bram Moolenaar04b12692020-05-04 23:24:44 +0200242enddef
243
244def Test_nested_function()
245 def Nested(arg: string): string
246 return 'nested ' .. arg
247 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200248 Nested('function')->assert_equal('nested function')
Bram Moolenaar04b12692020-05-04 23:24:44 +0200249
Bram Moolenaar0e65d3d2020-05-05 17:53:16 +0200250 CheckDefFailure(['def Nested()', 'enddef', 'Nested(66)'], 'E118:')
251 CheckDefFailure(['def Nested(arg: string)', 'enddef', 'Nested()'], 'E119:')
252
Bram Moolenaar04b12692020-05-04 23:24:44 +0200253 CheckDefFailure(['func Nested()', 'endfunc'], 'E1086:')
Bram Moolenaarbcbf4132020-08-01 22:35:13 +0200254 CheckDefFailure(['def s:Nested()', 'enddef'], 'E1075:')
255 CheckDefFailure(['def b:Nested()', 'enddef'], 'E1075:')
Bram Moolenaar8b848ca2020-09-10 22:28:01 +0200256
Bram Moolenaar54021752020-12-06 18:50:36 +0100257 var lines =<< trim END
258 def Outer()
259 def Inner()
260 # comment
261 enddef
262 def Inner()
263 enddef
264 enddef
265 END
266 CheckDefFailure(lines, 'E1073:')
267
268 lines =<< trim END
269 def Outer()
270 def Inner()
271 # comment
272 enddef
273 def! Inner()
274 enddef
275 enddef
276 END
277 CheckDefFailure(lines, 'E1117:')
278
279 # nested function inside conditional
280 # TODO: should it work when "thecount" is inside the "if"?
281 lines =<< trim END
282 vim9script
283 var thecount = 0
284 if true
285 def Test(): number
286 def TheFunc(): number
287 thecount += 1
288 return thecount
289 enddef
290 return TheFunc()
291 enddef
292 endif
293 defcompile
294 assert_equal(1, Test())
295 assert_equal(2, Test())
296 END
297 CheckScriptSuccess(lines)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200298enddef
299
Bram Moolenaaradc8e442020-12-31 18:28:18 +0100300def Test_not_nested_function()
301 echo printf('%d',
302 function('len')('xxx'))
303enddef
304
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200305func Test_call_default_args_from_func()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200306 call MyDefaultArgs()->assert_equal('string')
307 call MyDefaultArgs('one')->assert_equal('one')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200308 call assert_fails('call MyDefaultArgs("one", "two")', 'E118:', '', 3, 'Test_call_default_args_from_func')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200309endfunc
310
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200311def Test_nested_global_function()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200312 var lines =<< trim END
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200313 vim9script
314 def Outer()
315 def g:Inner(): string
316 return 'inner'
317 enddef
318 enddef
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200319 defcompile
320 Outer()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200321 g:Inner()->assert_equal('inner')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200322 delfunc g:Inner
323 Outer()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200324 g:Inner()->assert_equal('inner')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200325 delfunc g:Inner
326 Outer()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200327 g:Inner()->assert_equal('inner')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200328 delfunc g:Inner
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200329 END
330 CheckScriptSuccess(lines)
Bram Moolenaar2c79e9d2020-08-01 18:57:52 +0200331
332 lines =<< trim END
333 vim9script
334 def Outer()
335 def g:Inner(): string
336 return 'inner'
337 enddef
338 enddef
339 defcompile
340 Outer()
341 Outer()
342 END
343 CheckScriptFailure(lines, "E122:")
Bram Moolenaarcd45ed02020-12-22 17:35:54 +0100344 delfunc g:Inner
Bram Moolenaarad486a02020-08-01 23:22:18 +0200345
346 lines =<< trim END
347 vim9script
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100348 def Outer()
349 def g:Inner()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100350 echo map([1, 2, 3], (_, v) => v + 1)
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100351 enddef
352 g:Inner()
353 enddef
354 Outer()
355 END
356 CheckScriptSuccess(lines)
357 delfunc g:Inner
358
359 lines =<< trim END
360 vim9script
Bram Moolenaarad486a02020-08-01 23:22:18 +0200361 def Func()
362 echo 'script'
363 enddef
364 def Outer()
365 def Func()
366 echo 'inner'
367 enddef
368 enddef
369 defcompile
370 END
371 CheckScriptFailure(lines, "E1073:")
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200372enddef
373
Bram Moolenaar6abdcf82020-11-22 18:15:44 +0100374def DefListAll()
375 def
376enddef
377
378def DefListOne()
379 def DefListOne
380enddef
381
382def DefListMatches()
383 def /DefList
384enddef
385
386def Test_nested_def_list()
387 var funcs = split(execute('call DefListAll()'), "\n")
388 assert_true(len(funcs) > 10)
389 assert_true(funcs->index('def DefListAll()') >= 0)
390
391 funcs = split(execute('call DefListOne()'), "\n")
392 assert_equal([' def DefListOne()', '1 def DefListOne', ' enddef'], funcs)
393
394 funcs = split(execute('call DefListMatches()'), "\n")
395 assert_true(len(funcs) >= 3)
396 assert_true(funcs->index('def DefListAll()') >= 0)
397 assert_true(funcs->index('def DefListOne()') >= 0)
398 assert_true(funcs->index('def DefListMatches()') >= 0)
Bram Moolenaar54021752020-12-06 18:50:36 +0100399
400 var lines =<< trim END
401 vim9script
402 def Func()
403 def +Func+
404 enddef
405 defcompile
406 END
407 CheckScriptFailure(lines, 'E476:', 1)
Bram Moolenaar6abdcf82020-11-22 18:15:44 +0100408enddef
409
Bram Moolenaar333894b2020-08-01 18:53:07 +0200410def Test_global_local_function()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200411 var lines =<< trim END
Bram Moolenaar333894b2020-08-01 18:53:07 +0200412 vim9script
413 def g:Func(): string
414 return 'global'
415 enddef
416 def Func(): string
417 return 'local'
418 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200419 g:Func()->assert_equal('global')
420 Func()->assert_equal('local')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100421 delfunc g:Func
Bram Moolenaar333894b2020-08-01 18:53:07 +0200422 END
423 CheckScriptSuccess(lines)
Bram Moolenaar035d6e92020-08-11 22:30:42 +0200424
425 lines =<< trim END
426 vim9script
427 def g:Funcy()
428 echo 'funcy'
429 enddef
430 s:Funcy()
431 END
432 CheckScriptFailure(lines, 'E117:')
Bram Moolenaar333894b2020-08-01 18:53:07 +0200433enddef
434
Bram Moolenaar0f769812020-09-12 18:32:34 +0200435def Test_local_function_shadows_global()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200436 var lines =<< trim END
Bram Moolenaar0f769812020-09-12 18:32:34 +0200437 vim9script
438 def g:Gfunc(): string
439 return 'global'
440 enddef
441 def AnotherFunc(): number
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200442 var Gfunc = function('len')
Bram Moolenaar0f769812020-09-12 18:32:34 +0200443 return Gfunc('testing')
444 enddef
445 g:Gfunc()->assert_equal('global')
446 AnotherFunc()->assert_equal(7)
447 delfunc g:Gfunc
448 END
449 CheckScriptSuccess(lines)
450
451 lines =<< trim END
452 vim9script
453 def g:Func(): string
454 return 'global'
455 enddef
456 def AnotherFunc()
457 g:Func = function('len')
458 enddef
459 AnotherFunc()
460 END
461 CheckScriptFailure(lines, 'E705:')
462 delfunc g:Func
463enddef
464
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200465func TakesOneArg(arg)
466 echo a:arg
467endfunc
468
469def Test_call_wrong_args()
Bram Moolenaard2c61702020-09-06 15:58:36 +0200470 CheckDefFailure(['TakesOneArg()'], 'E119:')
471 CheckDefFailure(['TakesOneArg(11, 22)'], 'E118:')
472 CheckDefFailure(['bufnr(xxx)'], 'E1001:')
473 CheckScriptFailure(['def Func(Ref: func(s: string))'], 'E475:')
Bram Moolenaaree8580e2020-08-28 17:19:07 +0200474
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200475 var lines =<< trim END
Bram Moolenaaree8580e2020-08-28 17:19:07 +0200476 vim9script
477 def Func(s: string)
478 echo s
479 enddef
480 Func([])
481 END
Bram Moolenaar77072282020-09-16 17:55:40 +0200482 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected string but got list<unknown>', 5)
Bram Moolenaarb185a402020-09-18 22:42:00 +0200483
484 lines =<< trim END
485 vim9script
486 def FuncOne(nr: number)
487 echo nr
488 enddef
489 def FuncTwo()
490 FuncOne()
491 enddef
492 defcompile
493 END
494 writefile(lines, 'Xscript')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200495 var didCatch = false
Bram Moolenaarb185a402020-09-18 22:42:00 +0200496 try
497 source Xscript
498 catch
499 assert_match('E119: Not enough arguments for function: <SNR>\d\+_FuncOne', v:exception)
500 assert_match('Xscript\[8\]..function <SNR>\d\+_FuncTwo, line 1', v:throwpoint)
501 didCatch = true
502 endtry
503 assert_true(didCatch)
504
505 lines =<< trim END
506 vim9script
507 def FuncOne(nr: number)
508 echo nr
509 enddef
510 def FuncTwo()
511 FuncOne(1, 2)
512 enddef
513 defcompile
514 END
515 writefile(lines, 'Xscript')
516 didCatch = false
517 try
518 source Xscript
519 catch
520 assert_match('E118: Too many arguments for function: <SNR>\d\+_FuncOne', v:exception)
521 assert_match('Xscript\[8\]..function <SNR>\d\+_FuncTwo, line 1', v:throwpoint)
522 didCatch = true
523 endtry
524 assert_true(didCatch)
525
526 delete('Xscript')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200527enddef
528
Bram Moolenaar50824712020-12-20 21:10:17 +0100529def Test_call_funcref_wrong_args()
530 var head =<< trim END
531 vim9script
532 def Func3(a1: string, a2: number, a3: list<number>)
533 echo a1 .. a2 .. a3[0]
534 enddef
535 def Testme()
536 var funcMap: dict<func> = {func: Func3}
537 END
538 var tail =<< trim END
539 enddef
540 Testme()
541 END
542 CheckScriptSuccess(head + ["funcMap['func']('str', 123, [1, 2, 3])"] + tail)
543
544 CheckScriptFailure(head + ["funcMap['func']('str', 123)"] + tail, 'E119:')
545 CheckScriptFailure(head + ["funcMap['func']('str', 123, [1], 4)"] + tail, 'E118:')
546enddef
547
Bram Moolenaarb4d16cb2020-11-05 18:45:46 +0100548def Test_call_lambda_args()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100549 CheckDefFailure(['echo ((i) => 0)()'],
550 'E119: Not enough arguments for function: ((i) => 0)()')
Bram Moolenaarb4d16cb2020-11-05 18:45:46 +0100551
552 var lines =<< trim END
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100553 var Ref = (x: number, y: number) => x + y
Bram Moolenaarb4d16cb2020-11-05 18:45:46 +0100554 echo Ref(1, 'x')
555 END
556 CheckDefFailure(lines, 'E1013: Argument 2: type mismatch, expected number but got string')
557enddef
558
Bram Moolenaar709664c2020-12-12 14:33:41 +0100559def Test_lambda_uses_assigned_var()
560 CheckDefSuccess([
561 'var x: any = "aaa"'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100562 'x = filter(["bbb"], (_, v) => v =~ x)'])
Bram Moolenaar709664c2020-12-12 14:33:41 +0100563enddef
564
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200565" Default arg and varargs
566def MyDefVarargs(one: string, two = 'foo', ...rest: list<string>): string
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200567 var res = one .. ',' .. two
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200568 for s in rest
569 res ..= ',' .. s
570 endfor
571 return res
572enddef
573
574def Test_call_def_varargs()
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200575 assert_fails('MyDefVarargs()', 'E119:', '', 1, 'Test_call_def_varargs')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200576 MyDefVarargs('one')->assert_equal('one,foo')
577 MyDefVarargs('one', 'two')->assert_equal('one,two')
578 MyDefVarargs('one', 'two', 'three')->assert_equal('one,two,three')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200579 CheckDefFailure(['MyDefVarargs("one", 22)'],
Bram Moolenaar77072282020-09-16 17:55:40 +0200580 'E1013: Argument 2: type mismatch, expected string but got number')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200581 CheckDefFailure(['MyDefVarargs("one", "two", 123)'],
Bram Moolenaar77072282020-09-16 17:55:40 +0200582 'E1013: Argument 3: type mismatch, expected string but got number')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200583
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200584 var lines =<< trim END
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200585 vim9script
586 def Func(...l: list<string>)
587 echo l
588 enddef
589 Func('a', 'b', 'c')
590 END
591 CheckScriptSuccess(lines)
592
593 lines =<< trim END
594 vim9script
595 def Func(...l: list<string>)
596 echo l
597 enddef
598 Func()
599 END
600 CheckScriptSuccess(lines)
601
602 lines =<< trim END
603 vim9script
Bram Moolenaar2f8cbc42020-09-16 17:22:59 +0200604 def Func(...l: any)
605 echo l
606 enddef
607 Func(0)
608 END
609 CheckScriptSuccess(lines)
610
611 lines =<< trim END
612 vim9script
Bram Moolenaar28022722020-09-21 22:02:49 +0200613 def Func(..._l: list<string>)
614 echo _l
615 enddef
616 Func('a', 'b', 'c')
617 END
618 CheckScriptSuccess(lines)
619
620 lines =<< trim END
621 vim9script
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200622 def Func(...l: list<string>)
623 echo l
624 enddef
625 Func(1, 2, 3)
626 END
Bram Moolenaar77072282020-09-16 17:55:40 +0200627 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200628
629 lines =<< trim END
630 vim9script
631 def Func(...l: list<string>)
632 echo l
633 enddef
634 Func('a', 9)
635 END
Bram Moolenaar77072282020-09-16 17:55:40 +0200636 CheckScriptFailure(lines, 'E1013: Argument 2: type mismatch')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200637
638 lines =<< trim END
639 vim9script
640 def Func(...l: list<string>)
641 echo l
642 enddef
643 Func(1, 'a')
644 END
Bram Moolenaar77072282020-09-16 17:55:40 +0200645 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200646enddef
647
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200648let s:value = ''
649
650def FuncOneDefArg(opt = 'text')
651 s:value = opt
652enddef
653
654def FuncTwoDefArg(nr = 123, opt = 'text'): string
655 return nr .. opt
656enddef
657
658def FuncVarargs(...arg: list<string>): string
659 return join(arg, ',')
660enddef
661
662def Test_func_type_varargs()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200663 var RefDefArg: func(?string)
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200664 RefDefArg = FuncOneDefArg
665 RefDefArg()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200666 s:value->assert_equal('text')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200667 RefDefArg('some')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200668 s:value->assert_equal('some')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200669
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200670 var RefDef2Arg: func(?number, ?string): string
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200671 RefDef2Arg = FuncTwoDefArg
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200672 RefDef2Arg()->assert_equal('123text')
673 RefDef2Arg(99)->assert_equal('99text')
674 RefDef2Arg(77, 'some')->assert_equal('77some')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200675
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200676 CheckDefFailure(['var RefWrong: func(string?)'], 'E1010:')
677 CheckDefFailure(['var RefWrong: func(?string, string)'], 'E1007:')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200678
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200679 var RefVarargs: func(...list<string>): string
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200680 RefVarargs = FuncVarargs
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200681 RefVarargs()->assert_equal('')
682 RefVarargs('one')->assert_equal('one')
683 RefVarargs('one', 'two')->assert_equal('one,two')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200684
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200685 CheckDefFailure(['var RefWrong: func(...list<string>, string)'], 'E110:')
686 CheckDefFailure(['var RefWrong: func(...list<string>, ?string)'], 'E110:')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200687enddef
688
Bram Moolenaar0b76b422020-04-07 22:05:08 +0200689" Only varargs
690def MyVarargsOnly(...args: list<string>): string
691 return join(args, ',')
692enddef
693
694def Test_call_varargs_only()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200695 MyVarargsOnly()->assert_equal('')
696 MyVarargsOnly('one')->assert_equal('one')
697 MyVarargsOnly('one', 'two')->assert_equal('one,two')
Bram Moolenaar77072282020-09-16 17:55:40 +0200698 CheckDefFailure(['MyVarargsOnly(1)'], 'E1013: Argument 1: type mismatch, expected string but got number')
699 CheckDefFailure(['MyVarargsOnly("one", 2)'], 'E1013: Argument 2: type mismatch, expected string but got number')
Bram Moolenaar0b76b422020-04-07 22:05:08 +0200700enddef
701
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200702def Test_using_var_as_arg()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200703 writefile(['def Func(x: number)', 'var x = 234', 'enddef', 'defcompile'], 'Xdef')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200704 assert_fails('so Xdef', 'E1006:', '', 1, 'Func')
Bram Moolenaard2c61702020-09-06 15:58:36 +0200705 delete('Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200706enddef
707
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200708def DictArg(arg: dict<string>)
709 arg['key'] = 'value'
710enddef
711
712def ListArg(arg: list<string>)
713 arg[0] = 'value'
714enddef
715
716def Test_assign_to_argument()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200717 # works for dict and list
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200718 var d: dict<string> = {}
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200719 DictArg(d)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200720 d['key']->assert_equal('value')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200721 var l: list<string> = []
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200722 ListArg(l)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200723 l[0]->assert_equal('value')
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200724
Bram Moolenaard2c61702020-09-06 15:58:36 +0200725 CheckScriptFailure(['def Func(arg: number)', 'arg = 3', 'enddef', 'defcompile'], 'E1090:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100726 delfunc! g:Func
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200727enddef
728
Bram Moolenaarb816dae2020-09-20 22:04:00 +0200729" These argument names are reserved in legacy functions.
730def WithReservedNames(firstline: string, lastline: string): string
731 return firstline .. lastline
732enddef
733
734def Test_argument_names()
735 assert_equal('OK', WithReservedNames('O', 'K'))
736enddef
737
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200738def Test_call_func_defined_later()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200739 g:DefinedLater('one')->assert_equal('one')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200740 assert_fails('NotDefined("one")', 'E117:', '', 2, 'Test_call_func_defined_later')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200741enddef
742
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200743func DefinedLater(arg)
744 return a:arg
745endfunc
746
747def Test_call_funcref()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200748 g:SomeFunc('abc')->assert_equal(3)
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200749 assert_fails('NotAFunc()', 'E117:', '', 2, 'Test_call_funcref') # comment after call
750 assert_fails('g:NotAFunc()', 'E117:', '', 3, 'Test_call_funcref')
Bram Moolenaar2f1980f2020-07-22 19:30:06 +0200751
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200752 var lines =<< trim END
Bram Moolenaar2f1980f2020-07-22 19:30:06 +0200753 vim9script
754 def RetNumber(): number
755 return 123
756 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200757 var Funcref: func: number = function('RetNumber')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200758 Funcref()->assert_equal(123)
Bram Moolenaar2f1980f2020-07-22 19:30:06 +0200759 END
760 CheckScriptSuccess(lines)
Bram Moolenaar0f60e802020-07-22 20:16:11 +0200761
762 lines =<< trim END
763 vim9script
764 def RetNumber(): number
765 return 123
766 enddef
767 def Bar(F: func: number): number
768 return F()
769 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200770 var Funcref = function('RetNumber')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200771 Bar(Funcref)->assert_equal(123)
Bram Moolenaar0f60e802020-07-22 20:16:11 +0200772 END
773 CheckScriptSuccess(lines)
Bram Moolenaarbfba8652020-07-23 20:09:10 +0200774
775 lines =<< trim END
776 vim9script
777 def UseNumber(nr: number)
778 echo nr
779 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200780 var Funcref: func(number) = function('UseNumber')
Bram Moolenaarbfba8652020-07-23 20:09:10 +0200781 Funcref(123)
782 END
783 CheckScriptSuccess(lines)
Bram Moolenaarb8070e32020-07-23 20:56:04 +0200784
785 lines =<< trim END
786 vim9script
787 def UseNumber(nr: number)
788 echo nr
789 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200790 var Funcref: func(string) = function('UseNumber')
Bram Moolenaarb8070e32020-07-23 20:56:04 +0200791 END
Bram Moolenaar5e654232020-09-16 15:22:00 +0200792 CheckScriptFailure(lines, 'E1012: Type mismatch; expected func(string) but got func(number)')
Bram Moolenaar4fc224c2020-07-26 17:56:25 +0200793
794 lines =<< trim END
795 vim9script
796 def EchoNr(nr = 34)
797 g:echo = nr
798 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200799 var Funcref: func(?number) = function('EchoNr')
Bram Moolenaar4fc224c2020-07-26 17:56:25 +0200800 Funcref()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200801 g:echo->assert_equal(34)
Bram Moolenaar4fc224c2020-07-26 17:56:25 +0200802 Funcref(123)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200803 g:echo->assert_equal(123)
Bram Moolenaar4fc224c2020-07-26 17:56:25 +0200804 END
805 CheckScriptSuccess(lines)
Bram Moolenaarace61322020-07-26 18:16:58 +0200806
807 lines =<< trim END
808 vim9script
809 def EchoList(...l: list<number>)
810 g:echo = l
811 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200812 var Funcref: func(...list<number>) = function('EchoList')
Bram Moolenaarace61322020-07-26 18:16:58 +0200813 Funcref()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200814 g:echo->assert_equal([])
Bram Moolenaarace61322020-07-26 18:16:58 +0200815 Funcref(1, 2, 3)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200816 g:echo->assert_equal([1, 2, 3])
Bram Moolenaarace61322020-07-26 18:16:58 +0200817 END
818 CheckScriptSuccess(lines)
Bram Moolenaar01865ad2020-07-26 18:33:09 +0200819
820 lines =<< trim END
821 vim9script
822 def OptAndVar(nr: number, opt = 12, ...l: list<number>): number
823 g:optarg = opt
824 g:listarg = l
825 return nr
826 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200827 var Funcref: func(number, ?number, ...list<number>): number = function('OptAndVar')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200828 Funcref(10)->assert_equal(10)
829 g:optarg->assert_equal(12)
830 g:listarg->assert_equal([])
Bram Moolenaar01865ad2020-07-26 18:33:09 +0200831
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200832 Funcref(11, 22)->assert_equal(11)
833 g:optarg->assert_equal(22)
834 g:listarg->assert_equal([])
Bram Moolenaar01865ad2020-07-26 18:33:09 +0200835
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200836 Funcref(17, 18, 1, 2, 3)->assert_equal(17)
837 g:optarg->assert_equal(18)
838 g:listarg->assert_equal([1, 2, 3])
Bram Moolenaar01865ad2020-07-26 18:33:09 +0200839 END
840 CheckScriptSuccess(lines)
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200841enddef
842
843let SomeFunc = function('len')
844let NotAFunc = 'text'
845
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200846def CombineFuncrefTypes()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200847 # same arguments, different return type
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200848 var Ref1: func(bool): string
849 var Ref2: func(bool): number
850 var Ref3: func(bool): any
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200851 Ref3 = g:cond ? Ref1 : Ref2
852
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200853 # different number of arguments
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200854 var Refa1: func(bool): number
855 var Refa2: func(bool, number): number
856 var Refa3: func: number
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200857 Refa3 = g:cond ? Refa1 : Refa2
858
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200859 # different argument types
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200860 var Refb1: func(bool, string): number
861 var Refb2: func(string, number): number
862 var Refb3: func(any, any): number
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200863 Refb3 = g:cond ? Refb1 : Refb2
864enddef
865
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200866def FuncWithForwardCall()
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200867 return g:DefinedEvenLater("yes")
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200868enddef
869
870def DefinedEvenLater(arg: string): string
871 return arg
872enddef
873
874def Test_error_in_nested_function()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200875 # Error in called function requires unwinding the call stack.
Bram Moolenaar44d66522020-09-06 22:26:57 +0200876 assert_fails('FuncWithForwardCall()', 'E1096:', '', 1, 'FuncWithForwardCall')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200877enddef
878
879def Test_return_type_wrong()
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200880 CheckScriptFailure([
881 'def Func(): number',
882 'return "a"',
883 'enddef',
884 'defcompile'], 'expected number but got string')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100885 delfunc! g:Func
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200886 CheckScriptFailure([
887 'def Func(): string',
888 'return 1',
889 'enddef',
890 'defcompile'], 'expected string but got number')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100891 delfunc! g:Func
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200892 CheckScriptFailure([
893 'def Func(): void',
894 'return "a"',
895 'enddef',
896 'defcompile'],
897 'E1096: Returning a value in a function without a return type')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100898 delfunc! g:Func
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200899 CheckScriptFailure([
900 'def Func()',
901 'return "a"',
902 'enddef',
903 'defcompile'],
904 'E1096: Returning a value in a function without a return type')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100905 delfunc! g:Func
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200906
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200907 CheckScriptFailure([
908 'def Func(): number',
909 'return',
910 'enddef',
911 'defcompile'], 'E1003:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100912 delfunc! g:Func
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200913
914 CheckScriptFailure(['def Func(): list', 'return []', 'enddef'], 'E1008:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100915 delfunc! g:Func
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200916 CheckScriptFailure(['def Func(): dict', 'return {}', 'enddef'], 'E1008:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100917 delfunc! g:Func
Bram Moolenaaree4e0c12020-04-06 21:35:05 +0200918 CheckScriptFailure(['def Func()', 'return 1'], 'E1057:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100919 delfunc! g:Func
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200920
921 CheckScriptFailure([
922 'vim9script',
923 'def FuncB()',
924 ' return 123',
925 'enddef',
926 'def FuncA()',
927 ' FuncB()',
928 'enddef',
929 'defcompile'], 'E1096:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200930enddef
931
932def Test_arg_type_wrong()
933 CheckScriptFailure(['def Func3(items: list)', 'echo "a"', 'enddef'], 'E1008: Missing <type>')
Bram Moolenaaree4e0c12020-04-06 21:35:05 +0200934 CheckScriptFailure(['def Func4(...)', 'echo "a"', 'enddef'], 'E1055: Missing name after ...')
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +0200935 CheckScriptFailure(['def Func5(items:string)', 'echo "a"'], 'E1069:')
Bram Moolenaar6e949782020-04-13 17:21:00 +0200936 CheckScriptFailure(['def Func5(items)', 'echo "a"'], 'E1077:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200937enddef
938
939def Test_vim9script_call()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200940 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200941 vim9script
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200942 var name = ''
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200943 def MyFunc(arg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200944 name = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200945 enddef
946 MyFunc('foobar')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200947 name->assert_equal('foobar')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200948
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200949 var str = 'barfoo'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200950 str->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200951 name->assert_equal('barfoo')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200952
Bram Moolenaar67979662020-06-20 22:50:47 +0200953 g:value = 'value'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200954 g:value->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200955 name->assert_equal('value')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200956
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200957 var listvar = []
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200958 def ListFunc(arg: list<number>)
959 listvar = arg
960 enddef
961 [1, 2, 3]->ListFunc()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200962 listvar->assert_equal([1, 2, 3])
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200963
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200964 var dictvar = {}
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200965 def DictFunc(arg: dict<number>)
966 dictvar = arg
967 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +0100968 {a: 1, b: 2}->DictFunc()
969 dictvar->assert_equal({a: 1, b: 2})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200970 def CompiledDict()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100971 {a: 3, b: 4}->DictFunc()
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200972 enddef
973 CompiledDict()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100974 dictvar->assert_equal({a: 3, b: 4})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200975
Bram Moolenaare0de1712020-12-02 17:36:54 +0100976 {a: 3, b: 4}->DictFunc()
977 dictvar->assert_equal({a: 3, b: 4})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200978
979 ('text')->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200980 name->assert_equal('text')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200981 ("some")->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200982 name->assert_equal('some')
Bram Moolenaare6b53242020-07-01 17:28:33 +0200983
Bram Moolenaar13e12b82020-07-24 18:47:22 +0200984 # line starting with single quote is not a mark
Bram Moolenaar10409562020-07-29 20:00:38 +0200985 # line starting with double quote can be a method call
Bram Moolenaar3d48e252020-07-15 14:15:52 +0200986 'asdfasdf'->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200987 name->assert_equal('asdfasdf')
Bram Moolenaar10409562020-07-29 20:00:38 +0200988 "xyz"->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200989 name->assert_equal('xyz')
Bram Moolenaar3d48e252020-07-15 14:15:52 +0200990
991 def UseString()
992 'xyork'->MyFunc()
993 enddef
994 UseString()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200995 name->assert_equal('xyork')
Bram Moolenaar3d48e252020-07-15 14:15:52 +0200996
Bram Moolenaar10409562020-07-29 20:00:38 +0200997 def UseString2()
998 "knife"->MyFunc()
999 enddef
1000 UseString2()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001001 name->assert_equal('knife')
Bram Moolenaar10409562020-07-29 20:00:38 +02001002
Bram Moolenaar13e12b82020-07-24 18:47:22 +02001003 # prepending a colon makes it a mark
1004 new
1005 setline(1, ['aaa', 'bbb', 'ccc'])
1006 normal! 3Gmt1G
1007 :'t
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001008 getcurpos()[1]->assert_equal(3)
Bram Moolenaar13e12b82020-07-24 18:47:22 +02001009 bwipe!
1010
Bram Moolenaare6b53242020-07-01 17:28:33 +02001011 MyFunc(
1012 'continued'
1013 )
1014 assert_equal('continued',
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001015 name
Bram Moolenaare6b53242020-07-01 17:28:33 +02001016 )
1017
1018 call MyFunc(
1019 'more'
1020 ..
1021 'lines'
1022 )
1023 assert_equal(
1024 'morelines',
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001025 name)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001026 END
1027 writefile(lines, 'Xcall.vim')
1028 source Xcall.vim
1029 delete('Xcall.vim')
1030enddef
1031
1032def Test_vim9script_call_fail_decl()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001033 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001034 vim9script
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001035 var name = ''
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001036 def MyFunc(arg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001037 var name = 123
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001038 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +02001039 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001040 END
Bram Moolenaar6c4bfe42020-07-23 18:26:30 +02001041 CheckScriptFailure(lines, 'E1054:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001042enddef
1043
Bram Moolenaar65b95452020-07-19 14:03:09 +02001044def Test_vim9script_call_fail_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001045 var lines =<< trim END
Bram Moolenaar65b95452020-07-19 14:03:09 +02001046 vim9script
1047 def MyFunc(arg: string)
1048 echo arg
1049 enddef
1050 MyFunc(1234)
1051 END
Bram Moolenaar77072282020-09-16 17:55:40 +02001052 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected string but got number')
Bram Moolenaar65b95452020-07-19 14:03:09 +02001053enddef
1054
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001055def Test_vim9script_call_fail_const()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001056 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001057 vim9script
1058 const var = ''
1059 def MyFunc(arg: string)
1060 var = 'asdf'
1061 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +02001062 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001063 END
1064 writefile(lines, 'Xcall_const.vim')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001065 assert_fails('source Xcall_const.vim', 'E46:', '', 1, 'MyFunc')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001066 delete('Xcall_const.vim')
Bram Moolenaar3bdc90b2020-12-22 20:35:40 +01001067
1068 lines =<< trim END
1069 const g:Aconst = 77
1070 def Change()
1071 # comment
1072 g:Aconst = 99
1073 enddef
1074 call Change()
1075 unlet g:Aconst
1076 END
Bram Moolenaar1dcf55d2020-12-22 22:07:30 +01001077 CheckScriptFailure(lines, 'E741: Value is locked: Aconst', 2)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001078enddef
1079
1080" Test that inside :function a Python function can be defined, :def is not
1081" recognized.
1082func Test_function_python()
1083 CheckFeature python3
Bram Moolenaar727345e2020-09-27 23:33:59 +02001084 let py = 'python3'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001085 execute py "<< EOF"
1086def do_something():
1087 return 1
1088EOF
1089endfunc
1090
1091def Test_delfunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001092 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001093 vim9script
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02001094 def g:GoneSoon()
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001095 echo 'hello'
1096 enddef
1097
1098 def CallGoneSoon()
1099 GoneSoon()
1100 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +02001101 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001102
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02001103 delfunc g:GoneSoon
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001104 CallGoneSoon()
1105 END
1106 writefile(lines, 'XToDelFunc')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001107 assert_fails('so XToDelFunc', 'E933:', '', 1, 'CallGoneSoon')
1108 assert_fails('so XToDelFunc', 'E933:', '', 1, 'CallGoneSoon')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001109
1110 delete('XToDelFunc')
1111enddef
1112
1113def Test_redef_failure()
Bram Moolenaard2c61702020-09-06 15:58:36 +02001114 writefile(['def Func0(): string', 'return "Func0"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001115 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +02001116 writefile(['def Func1(): string', 'return "Func1"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001117 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +02001118 writefile(['def! Func0(): string', 'enddef', 'defcompile'], 'Xdef')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001119 assert_fails('so Xdef', 'E1027:', '', 1, 'Func0')
Bram Moolenaard2c61702020-09-06 15:58:36 +02001120 writefile(['def Func2(): string', 'return "Func2"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001121 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +02001122 delete('Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001123
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001124 g:Func0()->assert_equal(0)
1125 g:Func1()->assert_equal('Func1')
1126 g:Func2()->assert_equal('Func2')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001127
1128 delfunc! Func0
1129 delfunc! Func1
1130 delfunc! Func2
1131enddef
1132
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +02001133def Test_vim9script_func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001134 var lines =<< trim END
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +02001135 vim9script
1136 func Func(arg)
1137 echo a:arg
1138 endfunc
1139 Func('text')
1140 END
1141 writefile(lines, 'XVim9Func')
1142 so XVim9Func
1143
1144 delete('XVim9Func')
1145enddef
1146
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001147let s:funcResult = 0
1148
1149def FuncNoArgNoRet()
Bram Moolenaar53900992020-08-22 19:02:02 +02001150 s:funcResult = 11
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001151enddef
1152
1153def FuncNoArgRetNumber(): number
Bram Moolenaar53900992020-08-22 19:02:02 +02001154 s:funcResult = 22
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001155 return 1234
1156enddef
1157
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001158def FuncNoArgRetString(): string
Bram Moolenaar53900992020-08-22 19:02:02 +02001159 s:funcResult = 45
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001160 return 'text'
1161enddef
1162
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001163def FuncOneArgNoRet(arg: number)
Bram Moolenaar53900992020-08-22 19:02:02 +02001164 s:funcResult = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001165enddef
1166
1167def FuncOneArgRetNumber(arg: number): number
Bram Moolenaar53900992020-08-22 19:02:02 +02001168 s:funcResult = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001169 return arg
1170enddef
1171
Bram Moolenaar08938ee2020-04-11 23:17:17 +02001172def FuncTwoArgNoRet(one: bool, two: number)
Bram Moolenaar53900992020-08-22 19:02:02 +02001173 s:funcResult = two
Bram Moolenaar08938ee2020-04-11 23:17:17 +02001174enddef
1175
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001176def FuncOneArgRetString(arg: string): string
1177 return arg
1178enddef
1179
Bram Moolenaar89228602020-04-05 22:14:54 +02001180def FuncOneArgRetAny(arg: any): any
1181 return arg
1182enddef
1183
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001184def Test_func_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001185 var Ref1: func()
Bram Moolenaar53900992020-08-22 19:02:02 +02001186 s:funcResult = 0
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001187 Ref1 = FuncNoArgNoRet
1188 Ref1()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001189 s:funcResult->assert_equal(11)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001190
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001191 var Ref2: func
Bram Moolenaar53900992020-08-22 19:02:02 +02001192 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001193 Ref2 = FuncNoArgNoRet
1194 Ref2()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001195 s:funcResult->assert_equal(11)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001196
Bram Moolenaar53900992020-08-22 19:02:02 +02001197 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001198 Ref2 = FuncOneArgNoRet
1199 Ref2(12)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001200 s:funcResult->assert_equal(12)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001201
Bram Moolenaar53900992020-08-22 19:02:02 +02001202 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001203 Ref2 = FuncNoArgRetNumber
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001204 Ref2()->assert_equal(1234)
1205 s:funcResult->assert_equal(22)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001206
Bram Moolenaar53900992020-08-22 19:02:02 +02001207 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001208 Ref2 = FuncOneArgRetNumber
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001209 Ref2(13)->assert_equal(13)
1210 s:funcResult->assert_equal(13)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001211enddef
1212
Bram Moolenaar9978d472020-07-05 16:01:56 +02001213def Test_repeat_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001214 var res = 0
Bram Moolenaar9978d472020-07-05 16:01:56 +02001215 for n in repeat([1], 3)
1216 res += n
1217 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001218 res->assert_equal(3)
Bram Moolenaarfce82b32020-07-05 16:07:21 +02001219
1220 res = 0
1221 for n in add([1, 2], 3)
1222 res += n
1223 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001224 res->assert_equal(6)
Bram Moolenaar9978d472020-07-05 16:01:56 +02001225enddef
1226
Bram Moolenaar846178a2020-07-05 17:04:13 +02001227def Test_argv_return_type()
1228 next fileone filetwo
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001229 var res = ''
Bram Moolenaar846178a2020-07-05 17:04:13 +02001230 for name in argv()
1231 res ..= name
1232 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001233 res->assert_equal('fileonefiletwo')
Bram Moolenaar846178a2020-07-05 17:04:13 +02001234enddef
1235
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001236def Test_func_type_part()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001237 var RefVoid: func: void
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001238 RefVoid = FuncNoArgNoRet
1239 RefVoid = FuncOneArgNoRet
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001240 CheckDefFailure(['var RefVoid: func: void', 'RefVoid = FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func(...) but got func(): number')
1241 CheckDefFailure(['var RefVoid: func: void', 'RefVoid = FuncNoArgRetString'], 'E1012: Type mismatch; expected func(...) but got func(): string')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001242
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001243 var RefAny: func(): any
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001244 RefAny = FuncNoArgRetNumber
1245 RefAny = FuncNoArgRetString
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001246 CheckDefFailure(['var RefAny: func(): any', 'RefAny = FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(): any but got func()')
1247 CheckDefFailure(['var RefAny: func(): any', 'RefAny = FuncOneArgNoRet'], 'E1012: Type mismatch; expected func(): any but got func(number)')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001248
Bram Moolenaar6abd3dc2020-10-04 14:17:32 +02001249 var RefAnyNoArgs: func: any = RefAny
1250
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001251 var RefNr: func: number
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001252 RefNr = FuncNoArgRetNumber
1253 RefNr = FuncOneArgRetNumber
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001254 CheckDefFailure(['var RefNr: func: number', 'RefNr = FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(...): number but got func()')
1255 CheckDefFailure(['var RefNr: func: number', 'RefNr = FuncNoArgRetString'], 'E1012: Type mismatch; expected func(...): number but got func(): string')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001256
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001257 var RefStr: func: string
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001258 RefStr = FuncNoArgRetString
1259 RefStr = FuncOneArgRetString
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001260 CheckDefFailure(['var RefStr: func: string', 'RefStr = FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(...): string but got func()')
1261 CheckDefFailure(['var RefStr: func: string', 'RefStr = FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func(...): string but got func(): number')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001262enddef
1263
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001264def Test_func_type_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001265 CheckDefFailure(['var ref1: func()'], 'E704:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001266
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001267 CheckDefFailure(['var Ref1: func()', 'Ref1 = FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func() but got func(): number')
1268 CheckDefFailure(['var Ref1: func()', 'Ref1 = FuncOneArgNoRet'], 'E1012: Type mismatch; expected func() but got func(number)')
1269 CheckDefFailure(['var Ref1: func()', 'Ref1 = FuncOneArgRetNumber'], 'E1012: Type mismatch; expected func() but got func(number): number')
1270 CheckDefFailure(['var Ref1: func(bool)', 'Ref1 = FuncTwoArgNoRet'], 'E1012: Type mismatch; expected func(bool) but got func(bool, number)')
1271 CheckDefFailure(['var Ref1: func(?bool)', 'Ref1 = FuncTwoArgNoRet'], 'E1012: Type mismatch; expected func(?bool) but got func(bool, number)')
1272 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 +02001273
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001274 CheckDefFailure(['var RefWrong: func(string ,number)'], 'E1068:')
1275 CheckDefFailure(['var RefWrong: func(string,number)'], 'E1069:')
1276 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:')
1277 CheckDefFailure(['var RefWrong: func(bool):string'], 'E1069:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001278enddef
1279
Bram Moolenaar89228602020-04-05 22:14:54 +02001280def Test_func_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001281 var nr: number
Bram Moolenaar89228602020-04-05 22:14:54 +02001282 nr = FuncNoArgRetNumber()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001283 nr->assert_equal(1234)
Bram Moolenaar89228602020-04-05 22:14:54 +02001284
1285 nr = FuncOneArgRetAny(122)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001286 nr->assert_equal(122)
Bram Moolenaar89228602020-04-05 22:14:54 +02001287
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001288 var str: string
Bram Moolenaar89228602020-04-05 22:14:54 +02001289 str = FuncOneArgRetAny('yes')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001290 str->assert_equal('yes')
Bram Moolenaar89228602020-04-05 22:14:54 +02001291
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001292 CheckDefFailure(['var str: string', 'str = FuncNoArgRetNumber()'], 'E1012: Type mismatch; expected string but got number')
Bram Moolenaar89228602020-04-05 22:14:54 +02001293enddef
1294
Bram Moolenaar6abd3dc2020-10-04 14:17:32 +02001295def Test_func_common_type()
1296 def FuncOne(n: number): number
1297 return n
1298 enddef
1299 def FuncTwo(s: string): number
1300 return len(s)
1301 enddef
1302 def FuncThree(n: number, s: string): number
1303 return n + len(s)
1304 enddef
1305 var list = [FuncOne, FuncTwo, FuncThree]
1306 assert_equal(8, list[0](8))
1307 assert_equal(4, list[1]('word'))
1308 assert_equal(7, list[2](3, 'word'))
1309enddef
1310
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001311def MultiLine(
1312 arg1: string,
1313 arg2 = 1234,
1314 ...rest: list<string>
1315 ): string
1316 return arg1 .. arg2 .. join(rest, '-')
1317enddef
1318
Bram Moolenaar2c330432020-04-13 14:41:35 +02001319def MultiLineComment(
1320 arg1: string, # comment
1321 arg2 = 1234, # comment
1322 ...rest: list<string> # comment
1323 ): string # comment
1324 return arg1 .. arg2 .. join(rest, '-')
1325enddef
1326
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001327def Test_multiline()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001328 MultiLine('text')->assert_equal('text1234')
1329 MultiLine('text', 777)->assert_equal('text777')
1330 MultiLine('text', 777, 'one')->assert_equal('text777one')
1331 MultiLine('text', 777, 'one', 'two')->assert_equal('text777one-two')
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001332enddef
1333
Bram Moolenaar23e03252020-04-12 22:22:31 +02001334func Test_multiline_not_vim9()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001335 call MultiLine('text')->assert_equal('text1234')
1336 call MultiLine('text', 777)->assert_equal('text777')
1337 call MultiLine('text', 777, 'one')->assert_equal('text777one')
1338 call MultiLine('text', 777, 'one', 'two')->assert_equal('text777one-two')
Bram Moolenaar23e03252020-04-12 22:22:31 +02001339endfunc
1340
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001341
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02001342" When using CheckScriptFailure() for the below test, E1010 is generated instead
1343" of E1056.
1344func Test_E1056_1059()
1345 let caught_1056 = 0
1346 try
1347 def F():
1348 return 1
1349 enddef
1350 catch /E1056:/
1351 let caught_1056 = 1
1352 endtry
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001353 eval caught_1056->assert_equal(1)
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02001354
1355 let caught_1059 = 0
1356 try
1357 def F5(items : list)
1358 echo 'a'
1359 enddef
1360 catch /E1059:/
1361 let caught_1059 = 1
1362 endtry
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001363 eval caught_1059->assert_equal(1)
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02001364endfunc
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001365
Bram Moolenaar015f4262020-05-05 21:25:22 +02001366func DelMe()
1367 echo 'DelMe'
1368endfunc
1369
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001370def Test_error_reporting()
1371 # comment lines at the start of the function
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001372 var lines =<< trim END
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001373 " comment
1374 def Func()
1375 # comment
1376 # comment
1377 invalid
1378 enddef
1379 defcompile
1380 END
Bram Moolenaar08052222020-09-14 17:04:31 +02001381 writefile(lines, 'Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001382 try
1383 source Xdef
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001384 assert_report('should have failed')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001385 catch /E476:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001386 v:exception->assert_match('Invalid command: invalid')
1387 v:throwpoint->assert_match(', line 3$')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001388 endtry
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001389 delfunc! g:Func
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001390
1391 # comment lines after the start of the function
1392 lines =<< trim END
1393 " comment
1394 def Func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001395 var x = 1234
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001396 # comment
1397 # comment
1398 invalid
1399 enddef
1400 defcompile
1401 END
Bram Moolenaar08052222020-09-14 17:04:31 +02001402 writefile(lines, 'Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001403 try
1404 source Xdef
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001405 assert_report('should have failed')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001406 catch /E476:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001407 v:exception->assert_match('Invalid command: invalid')
1408 v:throwpoint->assert_match(', line 4$')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001409 endtry
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001410 delfunc! g:Func
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001411
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001412 lines =<< trim END
1413 vim9script
1414 def Func()
Bram Moolenaare0de1712020-12-02 17:36:54 +01001415 var db = {foo: 1, bar: 2}
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001416 # comment
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001417 var x = db.asdf
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001418 enddef
1419 defcompile
1420 Func()
1421 END
Bram Moolenaar08052222020-09-14 17:04:31 +02001422 writefile(lines, 'Xdef')
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001423 try
1424 source Xdef
1425 assert_report('should have failed')
1426 catch /E716:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001427 v:throwpoint->assert_match('_Func, line 3$')
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001428 endtry
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001429 delfunc! g:Func
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001430
Bram Moolenaar08052222020-09-14 17:04:31 +02001431 delete('Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001432enddef
1433
Bram Moolenaar015f4262020-05-05 21:25:22 +02001434def Test_deleted_function()
1435 CheckDefExecFailure([
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001436 'var RefMe: func = function("g:DelMe")',
Bram Moolenaar015f4262020-05-05 21:25:22 +02001437 'delfunc g:DelMe',
1438 'echo RefMe()'], 'E117:')
1439enddef
1440
1441def Test_unknown_function()
1442 CheckDefExecFailure([
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001443 'var Ref: func = function("NotExist")',
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +02001444 'delfunc g:NotExist'], 'E700:')
Bram Moolenaar015f4262020-05-05 21:25:22 +02001445enddef
1446
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +02001447def RefFunc(Ref: func(string): string): string
1448 return Ref('more')
1449enddef
1450
1451def Test_closure_simple()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001452 var local = 'some '
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001453 RefFunc((s) => local .. s)->assert_equal('some more')
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +02001454enddef
1455
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001456def MakeRef()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001457 var local = 'some '
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001458 g:Ref = (s) => local .. s
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001459enddef
1460
1461def Test_closure_ref_after_return()
1462 MakeRef()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001463 g:Ref('thing')->assert_equal('some thing')
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001464 unlet g:Ref
1465enddef
1466
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001467def MakeTwoRefs()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001468 var local = ['some']
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001469 g:Extend = (s) => local->add(s)
1470 g:Read = () => local
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001471enddef
1472
1473def Test_closure_two_refs()
1474 MakeTwoRefs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001475 join(g:Read(), ' ')->assert_equal('some')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001476 g:Extend('more')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001477 join(g:Read(), ' ')->assert_equal('some more')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001478 g:Extend('even')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001479 join(g:Read(), ' ')->assert_equal('some more even')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001480
1481 unlet g:Extend
1482 unlet g:Read
1483enddef
1484
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001485def ReadRef(Ref: func(): list<string>): string
1486 return join(Ref(), ' ')
1487enddef
1488
Bram Moolenaar5e654232020-09-16 15:22:00 +02001489def ExtendRef(Ref: func(string): list<string>, add: string)
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001490 Ref(add)
1491enddef
1492
1493def Test_closure_two_indirect_refs()
Bram Moolenaarf7779c62020-05-03 15:38:16 +02001494 MakeTwoRefs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001495 ReadRef(g:Read)->assert_equal('some')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001496 ExtendRef(g:Extend, 'more')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001497 ReadRef(g:Read)->assert_equal('some more')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001498 ExtendRef(g:Extend, 'even')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001499 ReadRef(g:Read)->assert_equal('some more even')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001500
1501 unlet g:Extend
1502 unlet g:Read
1503enddef
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001504
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001505def MakeArgRefs(theArg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001506 var local = 'loc_val'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001507 g:UseArg = (s) => theArg .. '/' .. local .. '/' .. s
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001508enddef
1509
1510def MakeArgRefsVarargs(theArg: string, ...rest: list<string>)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001511 var local = 'the_loc'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001512 g:UseVararg = (s) => theArg .. '/' .. local .. '/' .. s .. '/' .. join(rest)
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001513enddef
1514
1515def Test_closure_using_argument()
1516 MakeArgRefs('arg_val')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001517 g:UseArg('call_val')->assert_equal('arg_val/loc_val/call_val')
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001518
1519 MakeArgRefsVarargs('arg_val', 'one', 'two')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001520 g:UseVararg('call_val')->assert_equal('arg_val/the_loc/call_val/one two')
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001521
1522 unlet g:UseArg
1523 unlet g:UseVararg
1524enddef
1525
Bram Moolenaar85d5e2b2020-10-10 14:13:01 +02001526def MakeGetAndAppendRefs()
1527 var local = 'a'
1528
1529 def Append(arg: string)
1530 local ..= arg
1531 enddef
1532 g:Append = Append
1533
1534 def Get(): string
1535 return local
1536 enddef
1537 g:Get = Get
1538enddef
1539
1540def Test_closure_append_get()
1541 MakeGetAndAppendRefs()
1542 g:Get()->assert_equal('a')
1543 g:Append('-b')
1544 g:Get()->assert_equal('a-b')
1545 g:Append('-c')
1546 g:Get()->assert_equal('a-b-c')
1547
1548 unlet g:Append
1549 unlet g:Get
1550enddef
Bram Moolenaarb68b3462020-05-06 21:06:30 +02001551
Bram Moolenaar04b12692020-05-04 23:24:44 +02001552def Test_nested_closure()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001553 var local = 'text'
Bram Moolenaar04b12692020-05-04 23:24:44 +02001554 def Closure(arg: string): string
1555 return local .. arg
1556 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001557 Closure('!!!')->assert_equal('text!!!')
Bram Moolenaar04b12692020-05-04 23:24:44 +02001558enddef
1559
Bram Moolenaar6f5b6df2020-05-16 21:20:12 +02001560func GetResult(Ref)
1561 return a:Ref('some')
1562endfunc
1563
1564def Test_call_closure_not_compiled()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001565 var text = 'text'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001566 g:Ref = (s) => s .. text
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001567 GetResult(g:Ref)->assert_equal('sometext')
Bram Moolenaar6f5b6df2020-05-16 21:20:12 +02001568enddef
1569
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001570def Test_double_closure_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001571 var lines =<< trim END
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001572 vim9script
1573 def Func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001574 var name = 0
1575 for i in range(2)
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001576 timer_start(0, () => name)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001577 endfor
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001578 enddef
1579 Func()
1580 END
Bram Moolenaar148ce7a2020-09-23 21:57:23 +02001581 CheckScriptSuccess(lines)
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001582enddef
1583
Bram Moolenaar85d5e2b2020-10-10 14:13:01 +02001584def Test_nested_closure_used()
1585 var lines =<< trim END
1586 vim9script
1587 def Func()
1588 var x = 'hello'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001589 var Closure = () => x
1590 g:Myclosure = () => Closure()
Bram Moolenaar85d5e2b2020-10-10 14:13:01 +02001591 enddef
1592 Func()
1593 assert_equal('hello', g:Myclosure())
1594 END
1595 CheckScriptSuccess(lines)
1596enddef
Bram Moolenaar0876c782020-10-07 19:08:04 +02001597
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02001598def Test_nested_closure_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001599 var lines =<< trim END
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02001600 vim9script
1601 def FuncA()
1602 FuncB(0)
1603 enddef
1604 def FuncB(n: number): list<string>
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001605 return map([0], (_, v) => n)
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02001606 enddef
1607 FuncA()
1608 END
1609 CheckScriptFailure(lines, 'E1012:')
1610enddef
1611
Bram Moolenaarf112f302020-12-20 17:47:52 +01001612def Test_global_closure()
1613 var lines =<< trim END
1614 vim9script
1615 def ReverseEveryNLines(n: number, line1: number, line2: number)
1616 var mods = 'sil keepj keepp lockm '
1617 var range = ':' .. line1 .. ',' .. line2
1618 def g:Offset(): number
1619 var offset = (line('.') - line1 + 1) % n
1620 return offset != 0 ? offset : n
1621 enddef
1622 exe mods .. range .. 'g/^/exe "m .-" .. g:Offset()'
1623 enddef
1624
1625 new
1626 repeat(['aaa', 'bbb', 'ccc'], 3)->setline(1)
1627 ReverseEveryNLines(3, 1, 9)
1628 END
1629 CheckScriptSuccess(lines)
1630 var expected = repeat(['ccc', 'bbb', 'aaa'], 3)
1631 assert_equal(expected, getline(1, 9))
1632 bwipe!
1633enddef
1634
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01001635def Test_global_closure_called_directly()
1636 var lines =<< trim END
1637 vim9script
1638 def Outer()
1639 var x = 1
1640 def g:Inner()
1641 var y = x
1642 x += 1
1643 assert_equal(1, y)
1644 enddef
1645 g:Inner()
1646 assert_equal(2, x)
1647 enddef
1648 Outer()
1649 END
1650 CheckScriptSuccess(lines)
1651 delfunc g:Inner
1652enddef
1653
Bram Moolenaar34c54eb2020-11-25 19:15:19 +01001654def Test_failure_in_called_function()
1655 # this was using the frame index as the return value
1656 var lines =<< trim END
1657 vim9script
1658 au TerminalWinOpen * eval [][0]
1659 def PopupTerm(a: any)
1660 # make sure typvals on stack are string
1661 ['a', 'b', 'c', 'd', 'e', 'f', 'g']->join()
1662 FireEvent()
1663 enddef
1664 def FireEvent()
1665 do TerminalWinOpen
1666 enddef
1667 # use try/catch to make eval fail
1668 try
1669 call PopupTerm(0)
1670 catch
1671 endtry
1672 au! TerminalWinOpen
1673 END
1674 CheckScriptSuccess(lines)
1675enddef
1676
Bram Moolenaar5366e1a2020-10-01 13:01:34 +02001677def Test_nested_lambda()
1678 var lines =<< trim END
1679 vim9script
1680 def Func()
1681 var x = 4
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001682 var Lambda1 = () => 7
1683 var Lambda2 = () => [Lambda1(), x]
Bram Moolenaar5366e1a2020-10-01 13:01:34 +02001684 var res = Lambda2()
1685 assert_equal([7, 4], res)
1686 enddef
1687 Func()
1688 END
1689 CheckScriptSuccess(lines)
1690enddef
1691
Bram Moolenaar52bf81c2020-11-17 18:50:44 +01001692def Shadowed(): list<number>
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001693 var FuncList: list<func: number> = [() => 42]
1694 return FuncList->map((_, Shadowed) => Shadowed())
Bram Moolenaar52bf81c2020-11-17 18:50:44 +01001695enddef
1696
1697def Test_lambda_arg_shadows_func()
1698 assert_equal([42], Shadowed())
1699enddef
1700
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02001701def Line_continuation_in_def(dir: string = ''): string
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001702 var path: string = empty(dir)
1703 \ ? 'empty'
1704 \ : 'full'
1705 return path
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02001706enddef
1707
1708def Test_line_continuation_in_def()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001709 Line_continuation_in_def('.')->assert_equal('full')
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02001710enddef
1711
Bram Moolenaar2ea95b62020-11-19 21:47:56 +01001712def Test_script_var_in_lambda()
1713 var lines =<< trim END
1714 vim9script
1715 var script = 'test'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001716 assert_equal(['test'], map(['one'], () => script))
Bram Moolenaar2ea95b62020-11-19 21:47:56 +01001717 END
1718 CheckScriptSuccess(lines)
1719enddef
1720
Bram Moolenaar5e654232020-09-16 15:22:00 +02001721def Line_continuation_in_lambda(): list<string>
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001722 var x = range(97, 100)
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001723 ->map((_, v) => nr2char(v)
1724 ->toupper())
Bram Moolenaar7a4b8982020-07-08 17:36:21 +02001725 ->reverse()
1726 return x
1727enddef
1728
1729def Test_line_continuation_in_lambda()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001730 Line_continuation_in_lambda()->assert_equal(['D', 'C', 'B', 'A'])
Bram Moolenaar7a4b8982020-07-08 17:36:21 +02001731enddef
1732
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001733func Test_silent_echo()
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001734 CheckScreendump
1735
1736 let lines =<< trim END
1737 vim9script
1738 def EchoNothing()
1739 silent echo ''
1740 enddef
1741 defcompile
1742 END
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001743 call writefile(lines, 'XTest_silent_echo')
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001744
1745 " Check that the balloon shows up after a mouse move
1746 let buf = RunVimInTerminal('-S XTest_silent_echo', {'rows': 6})
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001747 call term_sendkeys(buf, ":abc")
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001748 call VerifyScreenDump(buf, 'Test_vim9_silent_echo', {})
1749
1750 " clean up
1751 call StopVimInTerminal(buf)
1752 call delete('XTest_silent_echo')
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001753endfunc
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001754
Bram Moolenaar171fb922020-10-28 16:54:47 +01001755def SilentlyError()
1756 execute('silent! invalid')
1757 g:did_it = 'yes'
1758enddef
1759
Bram Moolenaar28ee8922020-10-28 20:20:00 +01001760func UserError()
1761 silent! invalid
1762endfunc
1763
1764def SilentlyUserError()
1765 UserError()
1766 g:did_it = 'yes'
1767enddef
Bram Moolenaar171fb922020-10-28 16:54:47 +01001768
1769" This can't be a :def function, because the assert would not be reached.
Bram Moolenaar171fb922020-10-28 16:54:47 +01001770func Test_ignore_silent_error()
1771 let g:did_it = 'no'
1772 call SilentlyError()
1773 call assert_equal('yes', g:did_it)
1774
Bram Moolenaar28ee8922020-10-28 20:20:00 +01001775 let g:did_it = 'no'
1776 call SilentlyUserError()
1777 call assert_equal('yes', g:did_it)
Bram Moolenaar171fb922020-10-28 16:54:47 +01001778
1779 unlet g:did_it
1780endfunc
1781
Bram Moolenaarcd030c42020-10-30 21:49:40 +01001782def Test_ignore_silent_error_in_filter()
1783 var lines =<< trim END
1784 vim9script
1785 def Filter(winid: number, key: string): bool
1786 if key == 'o'
1787 silent! eval [][0]
1788 return true
1789 endif
1790 return popup_filter_menu(winid, key)
1791 enddef
1792
Bram Moolenaare0de1712020-12-02 17:36:54 +01001793 popup_create('popup', {filter: Filter})
Bram Moolenaarcd030c42020-10-30 21:49:40 +01001794 feedkeys("o\r", 'xnt')
1795 END
1796 CheckScriptSuccess(lines)
1797enddef
1798
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001799def Fibonacci(n: number): number
1800 if n < 2
1801 return n
1802 else
1803 return Fibonacci(n - 1) + Fibonacci(n - 2)
1804 endif
1805enddef
1806
Bram Moolenaar985116a2020-07-12 17:31:09 +02001807def Test_recursive_call()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001808 Fibonacci(20)->assert_equal(6765)
Bram Moolenaar985116a2020-07-12 17:31:09 +02001809enddef
1810
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001811def TreeWalk(dir: string): list<any>
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001812 return readdir(dir)->map((_, val) =>
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001813 fnamemodify(dir .. '/' .. val, ':p')->isdirectory()
Bram Moolenaar2bede172020-11-19 18:53:18 +01001814 ? {[val]: TreeWalk(dir .. '/' .. val)}
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001815 : val
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001816 )
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001817enddef
1818
1819def Test_closure_in_map()
1820 mkdir('XclosureDir/tdir', 'p')
1821 writefile(['111'], 'XclosureDir/file1')
1822 writefile(['222'], 'XclosureDir/file2')
1823 writefile(['333'], 'XclosureDir/tdir/file3')
1824
Bram Moolenaare0de1712020-12-02 17:36:54 +01001825 TreeWalk('XclosureDir')->assert_equal(['file1', 'file2', {tdir: ['file3']}])
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001826
1827 delete('XclosureDir', 'rf')
1828enddef
1829
Bram Moolenaar7b5d5442020-10-04 13:42:34 +02001830def Test_invalid_function_name()
1831 var lines =<< trim END
1832 vim9script
1833 def s: list<string>
1834 END
1835 CheckScriptFailure(lines, 'E129:')
1836
1837 lines =<< trim END
1838 vim9script
1839 def g: list<string>
1840 END
1841 CheckScriptFailure(lines, 'E129:')
1842
1843 lines =<< trim END
1844 vim9script
1845 def <SID>: list<string>
1846 END
1847 CheckScriptFailure(lines, 'E884:')
1848
1849 lines =<< trim END
1850 vim9script
1851 def F list<string>
1852 END
1853 CheckScriptFailure(lines, 'E488:')
1854enddef
1855
Bram Moolenaara90afb92020-07-15 22:38:56 +02001856def Test_partial_call()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001857 var Xsetlist = function('setloclist', [0])
Bram Moolenaare0de1712020-12-02 17:36:54 +01001858 Xsetlist([], ' ', {title: 'test'})
1859 getloclist(0, {title: 1})->assert_equal({title: 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02001860
1861 Xsetlist = function('setloclist', [0, [], ' '])
Bram Moolenaare0de1712020-12-02 17:36:54 +01001862 Xsetlist({title: 'test'})
1863 getloclist(0, {title: 1})->assert_equal({title: 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02001864
1865 Xsetlist = function('setqflist')
Bram Moolenaare0de1712020-12-02 17:36:54 +01001866 Xsetlist([], ' ', {title: 'test'})
1867 getqflist({title: 1})->assert_equal({title: 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02001868
1869 Xsetlist = function('setqflist', [[], ' '])
Bram Moolenaare0de1712020-12-02 17:36:54 +01001870 Xsetlist({title: 'test'})
1871 getqflist({title: 1})->assert_equal({title: 'test'})
Bram Moolenaar6abd3dc2020-10-04 14:17:32 +02001872
1873 var Len: func: number = function('len', ['word'])
1874 assert_equal(4, Len())
Bram Moolenaara90afb92020-07-15 22:38:56 +02001875enddef
1876
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001877def Test_cmd_modifier()
1878 tab echo '0'
Bram Moolenaard2c61702020-09-06 15:58:36 +02001879 CheckDefFailure(['5tab echo 3'], 'E16:')
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001880enddef
1881
1882def Test_restore_modifiers()
1883 # check that when compiling a :def function command modifiers are not messed
1884 # up.
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001885 var lines =<< trim END
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001886 vim9script
1887 set eventignore=
1888 autocmd QuickFixCmdPost * copen
1889 def AutocmdsDisabled()
Bram Moolenaar6cf7e3b2020-10-28 14:31:16 +01001890 eval 0
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001891 enddef
1892 func Func()
1893 noautocmd call s:AutocmdsDisabled()
1894 let g:ei_after = &eventignore
1895 endfunc
1896 Func()
1897 END
1898 CheckScriptSuccess(lines)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001899 g:ei_after->assert_equal('')
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001900enddef
1901
Bram Moolenaardfa3d552020-09-10 22:05:08 +02001902def StackTop()
1903 eval 1
1904 eval 2
1905 # call not on fourth line
1906 StackBot()
1907enddef
1908
1909def StackBot()
1910 # throw an error
1911 eval [][0]
1912enddef
1913
1914def Test_callstack_def()
1915 try
1916 StackTop()
1917 catch
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001918 v:throwpoint->assert_match('Test_callstack_def\[2\]..StackTop\[4\]..StackBot, line 2')
Bram Moolenaardfa3d552020-09-10 22:05:08 +02001919 endtry
1920enddef
1921
Bram Moolenaare8211a32020-10-09 22:04:29 +02001922" Re-using spot for variable used in block
1923def Test_block_scoped_var()
1924 var lines =<< trim END
1925 vim9script
1926 def Func()
1927 var x = ['a', 'b', 'c']
1928 if 1
1929 var y = 'x'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001930 map(x, () => y)
Bram Moolenaare8211a32020-10-09 22:04:29 +02001931 endif
1932 var z = x
1933 assert_equal(['x', 'x', 'x'], z)
1934 enddef
1935 Func()
1936 END
1937 CheckScriptSuccess(lines)
1938enddef
1939
Bram Moolenaareeece9e2020-11-20 19:26:48 +01001940def Test_reset_did_emsg()
1941 var lines =<< trim END
1942 @s = 'blah'
1943 au BufWinLeave * #
1944 def Func()
1945 var winid = popup_create('popup', {})
1946 exe '*s'
1947 popup_close(winid)
1948 enddef
1949 Func()
1950 END
1951 CheckScriptFailure(lines, 'E492:', 8)
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001952 delfunc! g:Func
Bram Moolenaareeece9e2020-11-20 19:26:48 +01001953enddef
1954
Bram Moolenaar57f799e2020-12-12 20:42:19 +01001955def Test_did_emsg_reset()
1956 # executing an autocommand resets did_emsg, this should not result in a
1957 # builtin function considered failing
1958 var lines =<< trim END
1959 vim9script
1960 au BufWinLeave * #
1961 def Func()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001962 popup_menu('', {callback: () => popup_create('', {})->popup_close()})
Bram Moolenaar57f799e2020-12-12 20:42:19 +01001963 eval [][0]
1964 enddef
1965 nno <F3> <cmd>call <sid>Func()<cr>
1966 feedkeys("\<F3>\e", 'xt')
1967 END
1968 writefile(lines, 'XemsgReset')
1969 assert_fails('so XemsgReset', ['E684:', 'E684:'], lines, 2)
1970 delete('XemsgReset')
1971 nunmap <F3>
1972 au! BufWinLeave
1973enddef
1974
Bram Moolenaar56602ba2020-12-05 21:22:08 +01001975def Test_abort_with_silent_call()
1976 var lines =<< trim END
1977 vim9script
1978 g:result = 'none'
1979 def Func()
1980 g:result += 3
1981 g:result = 'yes'
1982 enddef
1983 # error is silenced, but function aborts on error
1984 silent! Func()
1985 assert_equal('none', g:result)
1986 unlet g:result
1987 END
1988 CheckScriptSuccess(lines)
1989enddef
1990
Bram Moolenaarf665e972020-12-05 19:17:16 +01001991def Test_continues_with_silent_error()
1992 var lines =<< trim END
1993 vim9script
1994 g:result = 'none'
1995 def Func()
1996 silent! g:result += 3
1997 g:result = 'yes'
1998 enddef
1999 # error is silenced, function does not abort
2000 Func()
2001 assert_equal('yes', g:result)
2002 unlet g:result
2003 END
2004 CheckScriptSuccess(lines)
2005enddef
2006
Bram Moolenaaraf0df472020-12-02 20:51:22 +01002007def Test_abort_even_with_silent()
2008 var lines =<< trim END
2009 vim9script
2010 g:result = 'none'
2011 def Func()
2012 eval {-> ''}() .. '' .. {}['X']
2013 g:result = 'yes'
2014 enddef
Bram Moolenaarf665e972020-12-05 19:17:16 +01002015 silent! Func()
Bram Moolenaaraf0df472020-12-02 20:51:22 +01002016 assert_equal('none', g:result)
Bram Moolenaar4029cab2020-12-05 18:13:27 +01002017 unlet g:result
2018 END
2019 CheckScriptSuccess(lines)
2020enddef
2021
Bram Moolenaarf665e972020-12-05 19:17:16 +01002022def Test_cmdmod_silent_restored()
2023 var lines =<< trim END
2024 vim9script
2025 def Func()
2026 g:result = 'none'
2027 silent! g:result += 3
2028 g:result = 'none'
2029 g:result += 3
2030 enddef
2031 Func()
2032 END
2033 # can't use CheckScriptFailure, it ignores the :silent!
2034 var fname = 'Xdefsilent'
2035 writefile(lines, fname)
2036 var caught = 'no'
2037 try
2038 exe 'source ' .. fname
2039 catch /E1030:/
2040 caught = 'yes'
2041 assert_match('Func, line 4', v:throwpoint)
2042 endtry
2043 assert_equal('yes', caught)
2044 delete(fname)
2045enddef
2046
Bram Moolenaar4029cab2020-12-05 18:13:27 +01002047def Test_dict_member_with_silent()
2048 var lines =<< trim END
2049 vim9script
2050 g:result = 'none'
2051 var d: dict<any>
2052 def Func()
2053 try
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002054 g:result = map([], (_, v) => ({}[v]))->join() .. d['']
Bram Moolenaar4029cab2020-12-05 18:13:27 +01002055 catch
2056 endtry
2057 enddef
2058 silent! Func()
2059 assert_equal('0', g:result)
2060 unlet g:result
Bram Moolenaaraf0df472020-12-02 20:51:22 +01002061 END
2062 CheckScriptSuccess(lines)
2063enddef
2064
Bram Moolenaar5b3d1bb2020-12-22 12:20:08 +01002065def Test_opfunc()
2066 nnoremap <F3> <cmd>set opfunc=Opfunc<cr>g@
2067 def g:Opfunc(_: any): string
2068 setline(1, 'ASDF')
2069 return ''
2070 enddef
2071 new
2072 setline(1, 'asdf')
2073 feedkeys("\<F3>$", 'x')
2074 assert_equal('ASDF', getline(1))
2075
2076 bwipe!
2077 nunmap <F3>
2078enddef
2079
Bram Moolenaar077a4232020-12-22 18:33:27 +01002080" this was crashing on exit
2081def Test_nested_lambda_in_closure()
2082 var lines =<< trim END
2083 vim9script
2084 def Outer()
2085 def g:Inner()
2086 echo map([1, 2, 3], {_, v -> v + 1})
2087 enddef
2088 g:Inner()
2089 enddef
2090 defcompile
2091 writefile(['Done'], 'XnestedDone')
2092 quit
2093 END
2094 if !RunVim([], lines, '--clean')
2095 return
2096 endif
2097 assert_equal(['Done'], readfile('XnestedDone'))
2098 delete('XnestedDone')
2099enddef
2100
2101
Bram Moolenaarf7779c62020-05-03 15:38:16 +02002102
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002103" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker