blob: f9c5a2af2d98c5f425cf66c2132c160cb1c56b73 [file] [log] [blame]
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001" Test various aspects of the Vim9 script language.
2
3source check.vim
4source view_util.vim
Bram Moolenaar04b12692020-05-04 23:24:44 +02005source vim9.vim
Bram Moolenaar47e7d702020-07-05 18:18:42 +02006source screendump.vim
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02007
8func Test_def_basic()
9 def SomeFunc(): string
10 return 'yes'
11 enddef
12 call assert_equal('yes', SomeFunc())
13endfunc
14
15def ReturnString(): string
16 return 'string'
17enddef
18
19def ReturnNumber(): number
20 return 123
21enddef
22
23let g:notNumber = 'string'
24
25def ReturnGlobal(): number
26 return g:notNumber
27enddef
28
29def Test_return_something()
30 assert_equal('string', ReturnString())
31 assert_equal(123, ReturnNumber())
32 assert_fails('call ReturnGlobal()', 'E1029: Expected number but got string')
33enddef
34
Bram Moolenaarefd88552020-06-18 20:50:10 +020035def Test_missing_return()
36 CheckDefFailure(['def Missing(): number',
37 ' if g:cond',
38 ' echo "no return"',
39 ' else',
40 ' return 0',
41 ' endif'
42 'enddef'], 'E1027:')
43 CheckDefFailure(['def Missing(): number',
44 ' if g:cond',
45 ' return 1',
46 ' else',
47 ' echo "no return"',
48 ' endif'
49 'enddef'], 'E1027:')
50 CheckDefFailure(['def Missing(): number',
51 ' if g:cond',
52 ' return 1',
53 ' else',
54 ' return 2',
55 ' endif'
56 ' return 3'
57 'enddef'], 'E1095:')
58enddef
59
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +020060let s:nothing = 0
61def ReturnNothing()
62 s:nothing = 1
63 if true
64 return
65 endif
66 s:nothing = 2
67enddef
68
69def Test_return_nothing()
70 ReturnNothing()
71 assert_equal(1, s:nothing)
72enddef
73
74func Increment()
75 let g:counter += 1
76endfunc
77
78def Test_call_ufunc_count()
79 g:counter = 1
80 Increment()
81 Increment()
82 Increment()
83 " works with and without :call
84 assert_equal(4, g:counter)
85 call assert_equal(4, g:counter)
86 unlet g:counter
87enddef
88
89def MyVarargs(arg: string, ...rest: list<string>): string
90 let res = arg
91 for s in rest
92 res ..= ',' .. s
93 endfor
94 return res
95enddef
96
97def Test_call_varargs()
98 assert_equal('one', MyVarargs('one'))
99 assert_equal('one,two', MyVarargs('one', 'two'))
100 assert_equal('one,two,three', MyVarargs('one', 'two', 'three'))
101enddef
102
103def MyDefaultArgs(name = 'string'): string
104 return name
105enddef
106
107def Test_call_default_args()
108 assert_equal('string', MyDefaultArgs())
109 assert_equal('one', MyDefaultArgs('one'))
110 assert_fails('call MyDefaultArgs("one", "two")', 'E118:')
111
Bram Moolenaar822ba242020-05-24 23:00:18 +0200112 CheckScriptFailure(['def Func(arg: number = asdf)', 'enddef', 'defcompile'], 'E1001:')
113 CheckScriptFailure(['def Func(arg: number = "text")', 'enddef', 'defcompile'], 'E1013: argument 1: type mismatch, expected number but got string')
Bram Moolenaar04b12692020-05-04 23:24:44 +0200114enddef
115
116def Test_nested_function()
117 def Nested(arg: string): string
118 return 'nested ' .. arg
119 enddef
120 assert_equal('nested function', Nested('function'))
121
Bram Moolenaar0e65d3d2020-05-05 17:53:16 +0200122 CheckDefFailure(['def Nested()', 'enddef', 'Nested(66)'], 'E118:')
123 CheckDefFailure(['def Nested(arg: string)', 'enddef', 'Nested()'], 'E119:')
124
Bram Moolenaar04b12692020-05-04 23:24:44 +0200125 CheckDefFailure(['func Nested()', 'endfunc'], 'E1086:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200126enddef
127
128func Test_call_default_args_from_func()
129 call assert_equal('string', MyDefaultArgs())
130 call assert_equal('one', MyDefaultArgs('one'))
131 call assert_fails('call MyDefaultArgs("one", "two")', 'E118:')
132endfunc
133
134func TakesOneArg(arg)
135 echo a:arg
136endfunc
137
138def Test_call_wrong_args()
139 call CheckDefFailure(['TakesOneArg()'], 'E119:')
140 call CheckDefFailure(['TakesOneArg(11, 22)'], 'E118:')
141 call CheckDefFailure(['bufnr(xxx)'], 'E1001:')
Bram Moolenaar1c0d44f2020-05-02 19:04:58 +0200142 call CheckScriptFailure(['def Func(Ref: func(s: string))'], 'E475:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200143enddef
144
145" Default arg and varargs
146def MyDefVarargs(one: string, two = 'foo', ...rest: list<string>): string
147 let res = one .. ',' .. two
148 for s in rest
149 res ..= ',' .. s
150 endfor
151 return res
152enddef
153
154def Test_call_def_varargs()
155 call assert_fails('call MyDefVarargs()', 'E119:')
156 assert_equal('one,foo', MyDefVarargs('one'))
157 assert_equal('one,two', MyDefVarargs('one', 'two'))
158 assert_equal('one,two,three', MyDefVarargs('one', 'two', 'three'))
Bram Moolenaar49cf7cc2020-04-07 22:45:00 +0200159 call CheckDefFailure(['MyDefVarargs("one", 22)'], 'E1013: argument 2: type mismatch, expected string but got number')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200160enddef
161
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200162let s:value = ''
163
164def FuncOneDefArg(opt = 'text')
165 s:value = opt
166enddef
167
168def FuncTwoDefArg(nr = 123, opt = 'text'): string
169 return nr .. opt
170enddef
171
172def FuncVarargs(...arg: list<string>): string
173 return join(arg, ',')
174enddef
175
176def Test_func_type_varargs()
177 let RefDefArg: func(?string)
178 RefDefArg = FuncOneDefArg
179 RefDefArg()
180 assert_equal('text', s:value)
181 RefDefArg('some')
182 assert_equal('some', s:value)
183
184 let RefDef2Arg: func(?number, ?string): string
185 RefDef2Arg = FuncTwoDefArg
186 assert_equal('123text', RefDef2Arg())
187 assert_equal('99text', RefDef2Arg(99))
188 assert_equal('77some', RefDef2Arg(77, 'some'))
189
190 call CheckDefFailure(['let RefWrong: func(string?)'], 'E1010:')
191 call CheckDefFailure(['let RefWrong: func(?string, string)'], 'E1007:')
192
193 let RefVarargs: func(...list<string>): string
194 RefVarargs = FuncVarargs
195 assert_equal('', RefVarargs())
196 assert_equal('one', RefVarargs('one'))
197 assert_equal('one,two', RefVarargs('one', 'two'))
198
199 call CheckDefFailure(['let RefWrong: func(...list<string>, string)'], 'E110:')
200 call CheckDefFailure(['let RefWrong: func(...list<string>, ?string)'], 'E110:')
201enddef
202
Bram Moolenaar0b76b422020-04-07 22:05:08 +0200203" Only varargs
204def MyVarargsOnly(...args: list<string>): string
205 return join(args, ',')
206enddef
207
208def Test_call_varargs_only()
209 assert_equal('', MyVarargsOnly())
210 assert_equal('one', MyVarargsOnly('one'))
211 assert_equal('one,two', MyVarargsOnly('one', 'two'))
212 call CheckDefFailure(['MyVarargsOnly(1)'], 'E1013: argument 1: type mismatch, expected string but got number')
213 call CheckDefFailure(['MyVarargsOnly("one", 2)'], 'E1013: argument 2: type mismatch, expected string but got number')
214enddef
215
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200216def Test_using_var_as_arg()
Bram Moolenaar822ba242020-05-24 23:00:18 +0200217 call writefile(['def Func(x: number)', 'let x = 234', 'enddef', 'defcompile'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200218 call assert_fails('so Xdef', 'E1006:')
219 call delete('Xdef')
220enddef
221
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200222def DictArg(arg: dict<string>)
223 arg['key'] = 'value'
224enddef
225
226def ListArg(arg: list<string>)
227 arg[0] = 'value'
228enddef
229
230def Test_assign_to_argument()
231 " works for dict and list
232 let d: dict<string> = {}
233 DictArg(d)
234 assert_equal('value', d['key'])
235 let l: list<string> = []
236 ListArg(l)
237 assert_equal('value', l[0])
238
Bram Moolenaar822ba242020-05-24 23:00:18 +0200239 call CheckScriptFailure(['def Func(arg: number)', 'arg = 3', 'enddef', 'defcompile'], 'E1090:')
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200240enddef
241
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200242def Test_call_func_defined_later()
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200243 call assert_equal('one', g:DefinedLater('one'))
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200244 call assert_fails('call NotDefined("one")', 'E117:')
245enddef
246
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200247func DefinedLater(arg)
248 return a:arg
249endfunc
250
251def Test_call_funcref()
252 assert_equal(3, g:SomeFunc('abc'))
253 assert_fails('NotAFunc()', 'E117:')
254 assert_fails('g:NotAFunc()', 'E117:')
255enddef
256
257let SomeFunc = function('len')
258let NotAFunc = 'text'
259
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200260def CombineFuncrefTypes()
261 " same arguments, different return type
262 let Ref1: func(bool): string
263 let Ref2: func(bool): number
264 let Ref3: func(bool): any
265 Ref3 = g:cond ? Ref1 : Ref2
266
267 " different number of arguments
268 let Refa1: func(bool): number
269 let Refa2: func(bool, number): number
270 let Refa3: func: number
271 Refa3 = g:cond ? Refa1 : Refa2
272
273 " different argument types
274 let Refb1: func(bool, string): number
275 let Refb2: func(string, number): number
276 let Refb3: func(any, any): number
277 Refb3 = g:cond ? Refb1 : Refb2
278enddef
279
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200280def FuncWithForwardCall()
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200281 return g:DefinedEvenLater("yes")
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200282enddef
283
284def DefinedEvenLater(arg: string): string
285 return arg
286enddef
287
288def Test_error_in_nested_function()
289 " Error in called function requires unwinding the call stack.
Bram Moolenaar05a55512020-07-05 15:52:19 +0200290 assert_fails('call FuncWithForwardCall()', 'E1096')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200291enddef
292
293def Test_return_type_wrong()
Bram Moolenaar822ba242020-05-24 23:00:18 +0200294 CheckScriptFailure(['def Func(): number', 'return "a"', 'enddef', 'defcompile'], 'expected number but got string')
295 CheckScriptFailure(['def Func(): string', 'return 1', 'enddef', 'defcompile'], 'expected string but got number')
Bram Moolenaar05a55512020-07-05 15:52:19 +0200296 CheckScriptFailure(['def Func(): void', 'return "a"', 'enddef', 'defcompile'], 'E1096: Returning a value in a function without a return type')
297 CheckScriptFailure(['def Func()', 'return "a"', 'enddef', 'defcompile'], 'E1096: Returning a value in a function without a return type')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200298
Bram Moolenaar822ba242020-05-24 23:00:18 +0200299 CheckScriptFailure(['def Func(): number', 'return', 'enddef', 'defcompile'], 'E1003:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200300
301 CheckScriptFailure(['def Func(): list', 'return []', 'enddef'], 'E1008:')
302 CheckScriptFailure(['def Func(): dict', 'return {}', 'enddef'], 'E1008:')
Bram Moolenaaree4e0c12020-04-06 21:35:05 +0200303 CheckScriptFailure(['def Func()', 'return 1'], 'E1057:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200304enddef
305
306def Test_arg_type_wrong()
307 CheckScriptFailure(['def Func3(items: list)', 'echo "a"', 'enddef'], 'E1008: Missing <type>')
Bram Moolenaaree4e0c12020-04-06 21:35:05 +0200308 CheckScriptFailure(['def Func4(...)', 'echo "a"', 'enddef'], 'E1055: Missing name after ...')
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +0200309 CheckScriptFailure(['def Func5(items:string)', 'echo "a"'], 'E1069:')
Bram Moolenaar6e949782020-04-13 17:21:00 +0200310 CheckScriptFailure(['def Func5(items)', 'echo "a"'], 'E1077:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200311enddef
312
313def Test_vim9script_call()
314 let lines =<< trim END
315 vim9script
316 let var = ''
317 def MyFunc(arg: string)
318 var = arg
319 enddef
320 MyFunc('foobar')
321 assert_equal('foobar', var)
322
323 let str = 'barfoo'
324 str->MyFunc()
325 assert_equal('barfoo', var)
326
Bram Moolenaar67979662020-06-20 22:50:47 +0200327 g:value = 'value'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200328 g:value->MyFunc()
329 assert_equal('value', var)
330
331 let listvar = []
332 def ListFunc(arg: list<number>)
333 listvar = arg
334 enddef
335 [1, 2, 3]->ListFunc()
336 assert_equal([1, 2, 3], listvar)
337
338 let dictvar = {}
339 def DictFunc(arg: dict<number>)
340 dictvar = arg
341 enddef
342 {'a': 1, 'b': 2}->DictFunc()
343 assert_equal(#{a: 1, b: 2}, dictvar)
344 def CompiledDict()
345 {'a': 3, 'b': 4}->DictFunc()
346 enddef
347 CompiledDict()
348 assert_equal(#{a: 3, b: 4}, dictvar)
349
350 #{a: 3, b: 4}->DictFunc()
351 assert_equal(#{a: 3, b: 4}, dictvar)
352
353 ('text')->MyFunc()
354 assert_equal('text', var)
355 ("some")->MyFunc()
356 assert_equal('some', var)
Bram Moolenaare6b53242020-07-01 17:28:33 +0200357
Bram Moolenaar3d48e252020-07-15 14:15:52 +0200358 'asdfasdf'->MyFunc()
359 assert_equal('asdfasdf', var)
360
361 def UseString()
362 'xyork'->MyFunc()
363 enddef
364 UseString()
365 assert_equal('xyork', var)
366
Bram Moolenaare6b53242020-07-01 17:28:33 +0200367 MyFunc(
368 'continued'
369 )
370 assert_equal('continued',
371 var
372 )
373
374 call MyFunc(
375 'more'
376 ..
377 'lines'
378 )
379 assert_equal(
380 'morelines',
381 var)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200382 END
383 writefile(lines, 'Xcall.vim')
384 source Xcall.vim
385 delete('Xcall.vim')
386enddef
387
388def Test_vim9script_call_fail_decl()
389 let lines =<< trim END
390 vim9script
391 let var = ''
392 def MyFunc(arg: string)
393 let var = 123
394 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +0200395 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200396 END
397 writefile(lines, 'Xcall_decl.vim')
398 assert_fails('source Xcall_decl.vim', 'E1054:')
399 delete('Xcall_decl.vim')
400enddef
401
402def Test_vim9script_call_fail_const()
403 let lines =<< trim END
404 vim9script
405 const var = ''
406 def MyFunc(arg: string)
407 var = 'asdf'
408 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +0200409 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200410 END
411 writefile(lines, 'Xcall_const.vim')
412 assert_fails('source Xcall_const.vim', 'E46:')
413 delete('Xcall_const.vim')
414enddef
415
416" Test that inside :function a Python function can be defined, :def is not
417" recognized.
418func Test_function_python()
419 CheckFeature python3
420 let py = 'python3'
421 execute py "<< EOF"
422def do_something():
423 return 1
424EOF
425endfunc
426
427def Test_delfunc()
428 let lines =<< trim END
429 vim9script
Bram Moolenaar4c17ad92020-04-27 22:47:51 +0200430 def g:GoneSoon()
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200431 echo 'hello'
432 enddef
433
434 def CallGoneSoon()
435 GoneSoon()
436 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +0200437 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200438
Bram Moolenaar4c17ad92020-04-27 22:47:51 +0200439 delfunc g:GoneSoon
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200440 CallGoneSoon()
441 END
442 writefile(lines, 'XToDelFunc')
443 assert_fails('so XToDelFunc', 'E933')
444 assert_fails('so XToDelFunc', 'E933')
445
446 delete('XToDelFunc')
447enddef
448
449def Test_redef_failure()
450 call writefile(['def Func0(): string', 'return "Func0"', 'enddef'], 'Xdef')
451 so Xdef
452 call writefile(['def Func1(): string', 'return "Func1"', 'enddef'], 'Xdef')
453 so Xdef
Bram Moolenaar822ba242020-05-24 23:00:18 +0200454 call writefile(['def! Func0(): string', 'enddef', 'defcompile'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200455 call assert_fails('so Xdef', 'E1027:')
456 call writefile(['def Func2(): string', 'return "Func2"', 'enddef'], 'Xdef')
457 so Xdef
458 call delete('Xdef')
459
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200460 call assert_equal(0, g:Func0())
461 call assert_equal('Func1', g:Func1())
462 call assert_equal('Func2', g:Func2())
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200463
464 delfunc! Func0
465 delfunc! Func1
466 delfunc! Func2
467enddef
468
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +0200469def Test_vim9script_func()
470 let lines =<< trim END
471 vim9script
472 func Func(arg)
473 echo a:arg
474 endfunc
475 Func('text')
476 END
477 writefile(lines, 'XVim9Func')
478 so XVim9Func
479
480 delete('XVim9Func')
481enddef
482
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200483" Test for internal functions returning different types
484func Test_InternalFuncRetType()
485 let lines =<< trim END
486 def RetFloat(): float
487 return ceil(1.456)
488 enddef
489
490 def RetListAny(): list<any>
491 return items({'k' : 'v'})
492 enddef
493
494 def RetListString(): list<string>
495 return split('a:b:c', ':')
496 enddef
497
498 def RetListDictAny(): list<dict<any>>
499 return getbufinfo()
500 enddef
501
502 def RetDictNumber(): dict<number>
503 return wordcount()
504 enddef
505
506 def RetDictString(): dict<string>
507 return environ()
508 enddef
509 END
510 call writefile(lines, 'Xscript')
511 source Xscript
512
513 call assert_equal(2.0, RetFloat())
514 call assert_equal([['k', 'v']], RetListAny())
515 call assert_equal(['a', 'b', 'c'], RetListString())
516 call assert_notequal([], RetListDictAny())
517 call assert_notequal({}, RetDictNumber())
518 call assert_notequal({}, RetDictString())
519 call delete('Xscript')
520endfunc
521
522" Test for passing too many or too few arguments to internal functions
523func Test_internalfunc_arg_error()
524 let l =<< trim END
525 def! FArgErr(): float
526 return ceil(1.1, 2)
527 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +0200528 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200529 END
530 call writefile(l, 'Xinvalidarg')
531 call assert_fails('so Xinvalidarg', 'E118:')
532 let l =<< trim END
533 def! FArgErr(): float
534 return ceil()
535 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +0200536 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200537 END
538 call writefile(l, 'Xinvalidarg')
539 call assert_fails('so Xinvalidarg', 'E119:')
540 call delete('Xinvalidarg')
541endfunc
542
543let s:funcResult = 0
544
545def FuncNoArgNoRet()
546 funcResult = 11
547enddef
548
549def FuncNoArgRetNumber(): number
550 funcResult = 22
551 return 1234
552enddef
553
Bram Moolenaarec5929d2020-04-07 20:53:39 +0200554def FuncNoArgRetString(): string
555 funcResult = 45
556 return 'text'
557enddef
558
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200559def FuncOneArgNoRet(arg: number)
560 funcResult = arg
561enddef
562
563def FuncOneArgRetNumber(arg: number): number
564 funcResult = arg
565 return arg
566enddef
567
Bram Moolenaar08938ee2020-04-11 23:17:17 +0200568def FuncTwoArgNoRet(one: bool, two: number)
569 funcResult = two
570enddef
571
Bram Moolenaarec5929d2020-04-07 20:53:39 +0200572def FuncOneArgRetString(arg: string): string
573 return arg
574enddef
575
Bram Moolenaar89228602020-04-05 22:14:54 +0200576def FuncOneArgRetAny(arg: any): any
577 return arg
578enddef
579
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200580def Test_func_type()
581 let Ref1: func()
582 funcResult = 0
583 Ref1 = FuncNoArgNoRet
584 Ref1()
585 assert_equal(11, funcResult)
Bram Moolenaar4c683752020-04-05 21:38:23 +0200586
587 let Ref2: func
588 funcResult = 0
589 Ref2 = FuncNoArgNoRet
590 Ref2()
591 assert_equal(11, funcResult)
592
593 funcResult = 0
594 Ref2 = FuncOneArgNoRet
595 Ref2(12)
596 assert_equal(12, funcResult)
597
598 funcResult = 0
599 Ref2 = FuncNoArgRetNumber
600 assert_equal(1234, Ref2())
601 assert_equal(22, funcResult)
602
603 funcResult = 0
604 Ref2 = FuncOneArgRetNumber
605 assert_equal(13, Ref2(13))
606 assert_equal(13, funcResult)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200607enddef
608
Bram Moolenaar9978d472020-07-05 16:01:56 +0200609def Test_repeat_return_type()
610 let res = 0
611 for n in repeat([1], 3)
612 res += n
613 endfor
614 assert_equal(3, res)
Bram Moolenaarfce82b32020-07-05 16:07:21 +0200615
616 res = 0
617 for n in add([1, 2], 3)
618 res += n
619 endfor
620 assert_equal(6, res)
Bram Moolenaar9978d472020-07-05 16:01:56 +0200621enddef
622
Bram Moolenaar846178a2020-07-05 17:04:13 +0200623def Test_argv_return_type()
624 next fileone filetwo
625 let res = ''
626 for name in argv()
627 res ..= name
628 endfor
629 assert_equal('fileonefiletwo', res)
630enddef
631
Bram Moolenaarec5929d2020-04-07 20:53:39 +0200632def Test_func_type_part()
633 let RefVoid: func: void
634 RefVoid = FuncNoArgNoRet
635 RefVoid = FuncOneArgNoRet
636 CheckDefFailure(['let RefVoid: func: void', 'RefVoid = FuncNoArgRetNumber'], 'E1013: type mismatch, expected func() but got func(): number')
Bram Moolenaar6ff71d82020-05-24 23:45:24 +0200637 CheckDefFailure(['let RefVoid: func: void', 'RefVoid = FuncNoArgRetString'], 'E1013: type mismatch, expected func() but got func(): string')
Bram Moolenaarec5929d2020-04-07 20:53:39 +0200638
639 let RefAny: func(): any
640 RefAny = FuncNoArgRetNumber
641 RefAny = FuncNoArgRetString
642 CheckDefFailure(['let RefAny: func(): any', 'RefAny = FuncNoArgNoRet'], 'E1013: type mismatch, expected func(): any but got func()')
643 CheckDefFailure(['let RefAny: func(): any', 'RefAny = FuncOneArgNoRet'], 'E1013: type mismatch, expected func(): any but got func(number)')
644
645 let RefNr: func: number
646 RefNr = FuncNoArgRetNumber
647 RefNr = FuncOneArgRetNumber
648 CheckDefFailure(['let RefNr: func: number', 'RefNr = FuncNoArgNoRet'], 'E1013: type mismatch, expected func(): number but got func()')
Bram Moolenaar6ff71d82020-05-24 23:45:24 +0200649 CheckDefFailure(['let RefNr: func: number', 'RefNr = FuncNoArgRetString'], 'E1013: type mismatch, expected func(): number but got func(): string')
Bram Moolenaarec5929d2020-04-07 20:53:39 +0200650
651 let RefStr: func: string
652 RefStr = FuncNoArgRetString
653 RefStr = FuncOneArgRetString
654 CheckDefFailure(['let RefStr: func: string', 'RefStr = FuncNoArgNoRet'], 'E1013: type mismatch, expected func(): string but got func()')
655 CheckDefFailure(['let RefStr: func: string', 'RefStr = FuncNoArgRetNumber'], 'E1013: type mismatch, expected func(): string but got func(): number')
656enddef
657
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200658def Test_func_type_fails()
659 CheckDefFailure(['let ref1: func()'], 'E704:')
660
661 CheckDefFailure(['let Ref1: func()', 'Ref1 = FuncNoArgRetNumber'], 'E1013: type mismatch, expected func() but got func(): number')
662 CheckDefFailure(['let Ref1: func()', 'Ref1 = FuncOneArgNoRet'], 'E1013: type mismatch, expected func() but got func(number)')
663 CheckDefFailure(['let Ref1: func()', 'Ref1 = FuncOneArgRetNumber'], 'E1013: type mismatch, expected func() but got func(number): number')
Bram Moolenaar6ff71d82020-05-24 23:45:24 +0200664 CheckDefFailure(['let Ref1: func(bool)', 'Ref1 = FuncTwoArgNoRet'], 'E1013: type mismatch, expected func(bool) but got func(bool, number)')
665 CheckDefFailure(['let Ref1: func(?bool)', 'Ref1 = FuncTwoArgNoRet'], 'E1013: type mismatch, expected func(?bool) but got func(bool, number)')
666 CheckDefFailure(['let Ref1: func(...bool)', 'Ref1 = FuncTwoArgNoRet'], 'E1013: type mismatch, expected func(...bool) but got func(bool, number)')
Bram Moolenaar08938ee2020-04-11 23:17:17 +0200667
668 call CheckDefFailure(['let RefWrong: func(string ,number)'], 'E1068:')
669 call CheckDefFailure(['let RefWrong: func(string,number)'], 'E1069:')
670 call CheckDefFailure(['let RefWrong: func(bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool)'], 'E740:')
671 call CheckDefFailure(['let RefWrong: func(bool):string'], 'E1069:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200672enddef
673
Bram Moolenaar89228602020-04-05 22:14:54 +0200674def Test_func_return_type()
675 let nr: number
676 nr = FuncNoArgRetNumber()
677 assert_equal(1234, nr)
678
679 nr = FuncOneArgRetAny(122)
680 assert_equal(122, nr)
681
682 let str: string
683 str = FuncOneArgRetAny('yes')
684 assert_equal('yes', str)
685
686 CheckDefFailure(['let str: string', 'str = FuncNoArgRetNumber()'], 'E1013: type mismatch, expected string but got number')
687enddef
688
Bram Moolenaar5e774c72020-04-12 21:53:00 +0200689def MultiLine(
690 arg1: string,
691 arg2 = 1234,
692 ...rest: list<string>
693 ): string
694 return arg1 .. arg2 .. join(rest, '-')
695enddef
696
Bram Moolenaar2c330432020-04-13 14:41:35 +0200697def MultiLineComment(
698 arg1: string, # comment
699 arg2 = 1234, # comment
700 ...rest: list<string> # comment
701 ): string # comment
702 return arg1 .. arg2 .. join(rest, '-')
703enddef
704
Bram Moolenaar5e774c72020-04-12 21:53:00 +0200705def Test_multiline()
706 assert_equal('text1234', MultiLine('text'))
707 assert_equal('text777', MultiLine('text', 777))
708 assert_equal('text777one', MultiLine('text', 777, 'one'))
709 assert_equal('text777one-two', MultiLine('text', 777, 'one', 'two'))
710enddef
711
Bram Moolenaar23e03252020-04-12 22:22:31 +0200712func Test_multiline_not_vim9()
713 call assert_equal('text1234', MultiLine('text'))
714 call assert_equal('text777', MultiLine('text', 777))
715 call assert_equal('text777one', MultiLine('text', 777, 'one'))
716 call assert_equal('text777one-two', MultiLine('text', 777, 'one', 'two'))
717endfunc
718
Bram Moolenaar5e774c72020-04-12 21:53:00 +0200719
Bram Moolenaaree4e0c12020-04-06 21:35:05 +0200720" When using CheckScriptFailure() for the below test, E1010 is generated instead
721" of E1056.
722func Test_E1056_1059()
723 let caught_1056 = 0
724 try
725 def F():
726 return 1
727 enddef
728 catch /E1056:/
729 let caught_1056 = 1
730 endtry
731 call assert_equal(1, caught_1056)
732
733 let caught_1059 = 0
734 try
735 def F5(items : list)
736 echo 'a'
737 enddef
738 catch /E1059:/
739 let caught_1059 = 1
740 endtry
741 call assert_equal(1, caught_1059)
742endfunc
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200743
Bram Moolenaar015f4262020-05-05 21:25:22 +0200744func DelMe()
745 echo 'DelMe'
746endfunc
747
748def Test_deleted_function()
749 CheckDefExecFailure([
750 'let RefMe: func = function("g:DelMe")',
751 'delfunc g:DelMe',
752 'echo RefMe()'], 'E117:')
753enddef
754
755def Test_unknown_function()
756 CheckDefExecFailure([
757 'let Ref: func = function("NotExist")',
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200758 'delfunc g:NotExist'], 'E700:')
Bram Moolenaar015f4262020-05-05 21:25:22 +0200759enddef
760
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +0200761def RefFunc(Ref: func(string): string): string
762 return Ref('more')
763enddef
764
765def Test_closure_simple()
766 let local = 'some '
767 assert_equal('some more', RefFunc({s -> local .. s}))
768enddef
769
Bram Moolenaarbf67ea12020-05-02 17:52:42 +0200770def MakeRef()
771 let local = 'some '
772 g:Ref = {s -> local .. s}
773enddef
774
775def Test_closure_ref_after_return()
776 MakeRef()
777 assert_equal('some thing', g:Ref('thing'))
778 unlet g:Ref
779enddef
780
Bram Moolenaar5adc55c2020-05-02 23:12:58 +0200781def MakeTwoRefs()
782 let local = ['some']
783 g:Extend = {s -> local->add(s)}
784 g:Read = {-> local}
785enddef
786
787def Test_closure_two_refs()
788 MakeTwoRefs()
789 assert_equal('some', join(g:Read(), ' '))
790 g:Extend('more')
791 assert_equal('some more', join(g:Read(), ' '))
792 g:Extend('even')
793 assert_equal('some more even', join(g:Read(), ' '))
794
795 unlet g:Extend
796 unlet g:Read
797enddef
798
Bram Moolenaar5adc55c2020-05-02 23:12:58 +0200799def ReadRef(Ref: func(): list<string>): string
800 return join(Ref(), ' ')
801enddef
802
803def ExtendRef(Ref: func(string), add: string)
804 Ref(add)
805enddef
806
807def Test_closure_two_indirect_refs()
Bram Moolenaarf7779c62020-05-03 15:38:16 +0200808 MakeTwoRefs()
Bram Moolenaar5adc55c2020-05-02 23:12:58 +0200809 assert_equal('some', ReadRef(g:Read))
810 ExtendRef(g:Extend, 'more')
811 assert_equal('some more', ReadRef(g:Read))
812 ExtendRef(g:Extend, 'even')
813 assert_equal('some more even', ReadRef(g:Read))
814
815 unlet g:Extend
816 unlet g:Read
817enddef
Bram Moolenaarbf67ea12020-05-02 17:52:42 +0200818
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +0200819def MakeArgRefs(theArg: string)
820 let local = 'loc_val'
821 g:UseArg = {s -> theArg .. '/' .. local .. '/' .. s}
822enddef
823
824def MakeArgRefsVarargs(theArg: string, ...rest: list<string>)
825 let local = 'the_loc'
826 g:UseVararg = {s -> theArg .. '/' .. local .. '/' .. s .. '/' .. join(rest)}
827enddef
828
829def Test_closure_using_argument()
830 MakeArgRefs('arg_val')
831 assert_equal('arg_val/loc_val/call_val', g:UseArg('call_val'))
832
833 MakeArgRefsVarargs('arg_val', 'one', 'two')
834 assert_equal('arg_val/the_loc/call_val/one two', g:UseVararg('call_val'))
835
836 unlet g:UseArg
837 unlet g:UseVararg
838enddef
839
Bram Moolenaarb68b3462020-05-06 21:06:30 +0200840def MakeGetAndAppendRefs()
841 let local = 'a'
842
843 def Append(arg: string)
844 local ..= arg
845 enddef
846 g:Append = Append
847
848 def Get(): string
849 return local
850 enddef
851 g:Get = Get
852enddef
853
854def Test_closure_append_get()
855 MakeGetAndAppendRefs()
856 assert_equal('a', g:Get())
857 g:Append('-b')
858 assert_equal('a-b', g:Get())
859 g:Append('-c')
860 assert_equal('a-b-c', g:Get())
861
862 unlet g:Append
863 unlet g:Get
864enddef
865
Bram Moolenaar04b12692020-05-04 23:24:44 +0200866def Test_nested_closure()
867 let local = 'text'
868 def Closure(arg: string): string
869 return local .. arg
870 enddef
871 assert_equal('text!!!', Closure('!!!'))
872enddef
873
Bram Moolenaar6f5b6df2020-05-16 21:20:12 +0200874func GetResult(Ref)
875 return a:Ref('some')
876endfunc
877
878def Test_call_closure_not_compiled()
879 let text = 'text'
880 g:Ref = {s -> s .. text}
881 assert_equal('sometext', GetResult(g:Ref))
882enddef
883
Bram Moolenaar865af6b2020-06-18 18:45:49 +0200884def Test_sort_return_type()
885 let res: list<number>
886 res = [1, 2, 3]->sort()
887enddef
888
Bram Moolenaarf151ad12020-06-30 13:38:01 +0200889def Test_getqflist_return_type()
890 let l = getqflist()
891 assert_equal([], l)
892
893 let d = getqflist(#{items: 0})
894 assert_equal(#{items: []}, d)
895enddef
896
897def Test_getloclist_return_type()
898 let l = getloclist(1)
899 assert_equal([], l)
900
901 let d = getloclist(1, #{items: 0})
902 assert_equal(#{items: []}, d)
903enddef
904
Bram Moolenaara66ba012020-07-05 18:41:08 +0200905def Test_copy_return_type()
906 let l = copy([1, 2, 3])
907 let res = 0
908 for n in l
909 res += n
910 endfor
911 assert_equal(6, res)
912
913 let dl = deepcopy([1, 2, 3])
914 res = 0
915 for n in dl
916 res += n
917 endfor
918 assert_equal(6, res)
919enddef
920
Bram Moolenaarb3c019c2020-07-05 20:08:39 +0200921def Test_extend_return_type()
922 let l = extend([1, 2], [3])
923 let res = 0
924 for n in l
925 res += n
926 endfor
927 assert_equal(6, res)
928enddef
929
Bram Moolenaar252e88a2020-07-05 20:47:18 +0200930def Test_insert_return_type()
931 let l = insert([2, 1], 3)
932 let res = 0
933 for n in l
934 res += n
935 endfor
936 assert_equal(6, res)
937enddef
938
Bram Moolenaar67627352020-07-05 21:10:24 +0200939def Test_reverse_return_type()
940 let l = reverse([1, 2, 3])
941 let res = 0
942 for n in l
943 res += n
944 endfor
945 assert_equal(6, res)
946enddef
947
Bram Moolenaarad7c2492020-07-05 20:55:29 +0200948def Test_remove_return_type()
949 let l = remove(#{one: [1, 2], two: [3, 4]}, 'one')
950 let res = 0
951 for n in l
952 res += n
953 endfor
954 assert_equal(3, res)
955enddef
956
Bram Moolenaar0d94ad62020-07-05 20:16:41 +0200957def Test_filter_return_type()
958 let l = filter([1, 2, 3], {-> 1})
959 let res = 0
960 for n in l
961 res += n
962 endfor
963 assert_equal(6, res)
964enddef
965
Bram Moolenaarf1a23682020-07-13 18:55:48 +0200966def Wrong_dict_key_type(items: list<number>): list<number>
967 return filter(items, {_, val -> get({val: 1}, 'x')})
968enddef
969
970def Test_wrong_dict_key_type()
971 assert_fails('Wrong_dict_key_type([1, 2, 3])', 'E1029:')
972enddef
973
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +0200974def Line_continuation_in_def(dir: string = ''): string
975 let path: string = empty(dir)
976 \ ? 'empty'
977 \ : 'full'
978 return path
979enddef
980
981def Test_line_continuation_in_def()
982 assert_equal('full', Line_continuation_in_def('.'))
983enddef
984
Bram Moolenaar7a4b8982020-07-08 17:36:21 +0200985def Line_continuation_in_lambda(): list<number>
986 let x = range(97, 100)
Bram Moolenaar914e7ea2020-07-11 15:20:48 +0200987 ->map({_, v -> nr2char(v)
Bram Moolenaar7a4b8982020-07-08 17:36:21 +0200988 ->toupper()})
989 ->reverse()
990 return x
991enddef
992
993def Test_line_continuation_in_lambda()
994 assert_equal(['D', 'C', 'B', 'A'], Line_continuation_in_lambda())
995enddef
996
Bram Moolenaar8f510af2020-07-05 18:48:23 +0200997func Test_silent_echo()
Bram Moolenaar47e7d702020-07-05 18:18:42 +0200998 CheckScreendump
999
1000 let lines =<< trim END
1001 vim9script
1002 def EchoNothing()
1003 silent echo ''
1004 enddef
1005 defcompile
1006 END
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001007 call writefile(lines, 'XTest_silent_echo')
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001008
1009 " Check that the balloon shows up after a mouse move
1010 let buf = RunVimInTerminal('-S XTest_silent_echo', {'rows': 6})
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001011 call term_sendkeys(buf, ":abc")
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001012 call VerifyScreenDump(buf, 'Test_vim9_silent_echo', {})
1013
1014 " clean up
1015 call StopVimInTerminal(buf)
1016 call delete('XTest_silent_echo')
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001017endfunc
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001018
Bram Moolenaar985116a2020-07-12 17:31:09 +02001019def Fibonacci(n: number): number
1020 if n < 2
1021 return n
1022 else
1023 return Fibonacci(n - 1) + Fibonacci(n - 2)
1024 endif
1025enddef
1026
1027def Test_recursive_call()
1028 assert_equal(6765, Fibonacci(20))
1029enddef
1030
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001031def TreeWalk(dir: string): list<any>
1032 return readdir(dir)->map({_, val ->
1033 fnamemodify(dir .. '/' .. val, ':p')->isdirectory()
1034 ? {val : TreeWalk(dir .. '/' .. val)}
1035 : val
1036 })
1037enddef
1038
1039def Test_closure_in_map()
1040 mkdir('XclosureDir/tdir', 'p')
1041 writefile(['111'], 'XclosureDir/file1')
1042 writefile(['222'], 'XclosureDir/file2')
1043 writefile(['333'], 'XclosureDir/tdir/file3')
1044
1045 assert_equal(['file1', 'file2', {'tdir': ['file3']}], TreeWalk('XclosureDir'))
1046
1047 delete('XclosureDir', 'rf')
1048enddef
1049
Bram Moolenaarf7779c62020-05-03 15:38:16 +02001050
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001051" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker