blob: 6d83ddc515d4fe6854c5f3860f491282a635d11d [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()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +020083 # works with and without :call
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +020084 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
Bram Moolenaare30f64b2020-07-15 19:48:20 +0200107def MyDefaultSecond(name: string, second: bool = true): string
108 return second ? name : 'none'
109enddef
110
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200111def Test_call_default_args()
112 assert_equal('string', MyDefaultArgs())
113 assert_equal('one', MyDefaultArgs('one'))
114 assert_fails('call MyDefaultArgs("one", "two")', 'E118:')
115
Bram Moolenaare30f64b2020-07-15 19:48:20 +0200116 assert_equal('test', MyDefaultSecond('test'))
117 assert_equal('test', MyDefaultSecond('test', true))
118 assert_equal('none', MyDefaultSecond('test', false))
119
Bram Moolenaar822ba242020-05-24 23:00:18 +0200120 CheckScriptFailure(['def Func(arg: number = asdf)', 'enddef', 'defcompile'], 'E1001:')
121 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 +0200122enddef
123
124def Test_nested_function()
125 def Nested(arg: string): string
126 return 'nested ' .. arg
127 enddef
128 assert_equal('nested function', Nested('function'))
129
Bram Moolenaar0e65d3d2020-05-05 17:53:16 +0200130 CheckDefFailure(['def Nested()', 'enddef', 'Nested(66)'], 'E118:')
131 CheckDefFailure(['def Nested(arg: string)', 'enddef', 'Nested()'], 'E119:')
132
Bram Moolenaar04b12692020-05-04 23:24:44 +0200133 CheckDefFailure(['func Nested()', 'endfunc'], 'E1086:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200134enddef
135
136func Test_call_default_args_from_func()
137 call assert_equal('string', MyDefaultArgs())
138 call assert_equal('one', MyDefaultArgs('one'))
139 call assert_fails('call MyDefaultArgs("one", "two")', 'E118:')
140endfunc
141
142func TakesOneArg(arg)
143 echo a:arg
144endfunc
145
146def Test_call_wrong_args()
147 call CheckDefFailure(['TakesOneArg()'], 'E119:')
148 call CheckDefFailure(['TakesOneArg(11, 22)'], 'E118:')
149 call CheckDefFailure(['bufnr(xxx)'], 'E1001:')
Bram Moolenaar1c0d44f2020-05-02 19:04:58 +0200150 call CheckScriptFailure(['def Func(Ref: func(s: string))'], 'E475:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200151enddef
152
153" Default arg and varargs
154def MyDefVarargs(one: string, two = 'foo', ...rest: list<string>): string
155 let res = one .. ',' .. two
156 for s in rest
157 res ..= ',' .. s
158 endfor
159 return res
160enddef
161
162def Test_call_def_varargs()
163 call assert_fails('call MyDefVarargs()', 'E119:')
164 assert_equal('one,foo', MyDefVarargs('one'))
165 assert_equal('one,two', MyDefVarargs('one', 'two'))
166 assert_equal('one,two,three', MyDefVarargs('one', 'two', 'three'))
Bram Moolenaar49cf7cc2020-04-07 22:45:00 +0200167 call CheckDefFailure(['MyDefVarargs("one", 22)'], 'E1013: argument 2: type mismatch, expected string but got number')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200168enddef
169
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200170let s:value = ''
171
172def FuncOneDefArg(opt = 'text')
173 s:value = opt
174enddef
175
176def FuncTwoDefArg(nr = 123, opt = 'text'): string
177 return nr .. opt
178enddef
179
180def FuncVarargs(...arg: list<string>): string
181 return join(arg, ',')
182enddef
183
184def Test_func_type_varargs()
185 let RefDefArg: func(?string)
186 RefDefArg = FuncOneDefArg
187 RefDefArg()
188 assert_equal('text', s:value)
189 RefDefArg('some')
190 assert_equal('some', s:value)
191
192 let RefDef2Arg: func(?number, ?string): string
193 RefDef2Arg = FuncTwoDefArg
194 assert_equal('123text', RefDef2Arg())
195 assert_equal('99text', RefDef2Arg(99))
196 assert_equal('77some', RefDef2Arg(77, 'some'))
197
198 call CheckDefFailure(['let RefWrong: func(string?)'], 'E1010:')
199 call CheckDefFailure(['let RefWrong: func(?string, string)'], 'E1007:')
200
201 let RefVarargs: func(...list<string>): string
202 RefVarargs = FuncVarargs
203 assert_equal('', RefVarargs())
204 assert_equal('one', RefVarargs('one'))
205 assert_equal('one,two', RefVarargs('one', 'two'))
206
207 call CheckDefFailure(['let RefWrong: func(...list<string>, string)'], 'E110:')
208 call CheckDefFailure(['let RefWrong: func(...list<string>, ?string)'], 'E110:')
209enddef
210
Bram Moolenaar0b76b422020-04-07 22:05:08 +0200211" Only varargs
212def MyVarargsOnly(...args: list<string>): string
213 return join(args, ',')
214enddef
215
216def Test_call_varargs_only()
217 assert_equal('', MyVarargsOnly())
218 assert_equal('one', MyVarargsOnly('one'))
219 assert_equal('one,two', MyVarargsOnly('one', 'two'))
220 call CheckDefFailure(['MyVarargsOnly(1)'], 'E1013: argument 1: type mismatch, expected string but got number')
221 call CheckDefFailure(['MyVarargsOnly("one", 2)'], 'E1013: argument 2: type mismatch, expected string but got number')
222enddef
223
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200224def Test_using_var_as_arg()
Bram Moolenaar822ba242020-05-24 23:00:18 +0200225 call writefile(['def Func(x: number)', 'let x = 234', 'enddef', 'defcompile'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200226 call assert_fails('so Xdef', 'E1006:')
227 call delete('Xdef')
228enddef
229
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200230def DictArg(arg: dict<string>)
231 arg['key'] = 'value'
232enddef
233
234def ListArg(arg: list<string>)
235 arg[0] = 'value'
236enddef
237
238def Test_assign_to_argument()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200239 # works for dict and list
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200240 let d: dict<string> = {}
241 DictArg(d)
242 assert_equal('value', d['key'])
243 let l: list<string> = []
244 ListArg(l)
245 assert_equal('value', l[0])
246
Bram Moolenaar822ba242020-05-24 23:00:18 +0200247 call CheckScriptFailure(['def Func(arg: number)', 'arg = 3', 'enddef', 'defcompile'], 'E1090:')
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200248enddef
249
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200250def Test_call_func_defined_later()
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200251 call assert_equal('one', g:DefinedLater('one'))
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200252 call assert_fails('call NotDefined("one")', 'E117:')
253enddef
254
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200255func DefinedLater(arg)
256 return a:arg
257endfunc
258
259def Test_call_funcref()
260 assert_equal(3, g:SomeFunc('abc'))
261 assert_fails('NotAFunc()', 'E117:')
262 assert_fails('g:NotAFunc()', 'E117:')
263enddef
264
265let SomeFunc = function('len')
266let NotAFunc = 'text'
267
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200268def CombineFuncrefTypes()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200269 # same arguments, different return type
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200270 let Ref1: func(bool): string
271 let Ref2: func(bool): number
272 let Ref3: func(bool): any
273 Ref3 = g:cond ? Ref1 : Ref2
274
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200275 # different number of arguments
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200276 let Refa1: func(bool): number
277 let Refa2: func(bool, number): number
278 let Refa3: func: number
279 Refa3 = g:cond ? Refa1 : Refa2
280
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200281 # different argument types
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200282 let Refb1: func(bool, string): number
283 let Refb2: func(string, number): number
284 let Refb3: func(any, any): number
285 Refb3 = g:cond ? Refb1 : Refb2
286enddef
287
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200288def FuncWithForwardCall()
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200289 return g:DefinedEvenLater("yes")
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200290enddef
291
292def DefinedEvenLater(arg: string): string
293 return arg
294enddef
295
296def Test_error_in_nested_function()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200297 # Error in called function requires unwinding the call stack.
Bram Moolenaar05a55512020-07-05 15:52:19 +0200298 assert_fails('call FuncWithForwardCall()', 'E1096')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200299enddef
300
301def Test_return_type_wrong()
Bram Moolenaar822ba242020-05-24 23:00:18 +0200302 CheckScriptFailure(['def Func(): number', 'return "a"', 'enddef', 'defcompile'], 'expected number but got string')
303 CheckScriptFailure(['def Func(): string', 'return 1', 'enddef', 'defcompile'], 'expected string but got number')
Bram Moolenaar05a55512020-07-05 15:52:19 +0200304 CheckScriptFailure(['def Func(): void', 'return "a"', 'enddef', 'defcompile'], 'E1096: Returning a value in a function without a return type')
305 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 +0200306
Bram Moolenaar822ba242020-05-24 23:00:18 +0200307 CheckScriptFailure(['def Func(): number', 'return', 'enddef', 'defcompile'], 'E1003:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200308
309 CheckScriptFailure(['def Func(): list', 'return []', 'enddef'], 'E1008:')
310 CheckScriptFailure(['def Func(): dict', 'return {}', 'enddef'], 'E1008:')
Bram Moolenaaree4e0c12020-04-06 21:35:05 +0200311 CheckScriptFailure(['def Func()', 'return 1'], 'E1057:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200312enddef
313
314def Test_arg_type_wrong()
315 CheckScriptFailure(['def Func3(items: list)', 'echo "a"', 'enddef'], 'E1008: Missing <type>')
Bram Moolenaaree4e0c12020-04-06 21:35:05 +0200316 CheckScriptFailure(['def Func4(...)', 'echo "a"', 'enddef'], 'E1055: Missing name after ...')
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +0200317 CheckScriptFailure(['def Func5(items:string)', 'echo "a"'], 'E1069:')
Bram Moolenaar6e949782020-04-13 17:21:00 +0200318 CheckScriptFailure(['def Func5(items)', 'echo "a"'], 'E1077:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200319enddef
320
321def Test_vim9script_call()
322 let lines =<< trim END
323 vim9script
324 let var = ''
325 def MyFunc(arg: string)
326 var = arg
327 enddef
328 MyFunc('foobar')
329 assert_equal('foobar', var)
330
331 let str = 'barfoo'
332 str->MyFunc()
333 assert_equal('barfoo', var)
334
Bram Moolenaar67979662020-06-20 22:50:47 +0200335 g:value = 'value'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200336 g:value->MyFunc()
337 assert_equal('value', var)
338
339 let listvar = []
340 def ListFunc(arg: list<number>)
341 listvar = arg
342 enddef
343 [1, 2, 3]->ListFunc()
344 assert_equal([1, 2, 3], listvar)
345
346 let dictvar = {}
347 def DictFunc(arg: dict<number>)
348 dictvar = arg
349 enddef
350 {'a': 1, 'b': 2}->DictFunc()
351 assert_equal(#{a: 1, b: 2}, dictvar)
352 def CompiledDict()
353 {'a': 3, 'b': 4}->DictFunc()
354 enddef
355 CompiledDict()
356 assert_equal(#{a: 3, b: 4}, dictvar)
357
358 #{a: 3, b: 4}->DictFunc()
359 assert_equal(#{a: 3, b: 4}, dictvar)
360
361 ('text')->MyFunc()
362 assert_equal('text', var)
363 ("some")->MyFunc()
364 assert_equal('some', var)
Bram Moolenaare6b53242020-07-01 17:28:33 +0200365
Bram Moolenaar3d48e252020-07-15 14:15:52 +0200366 'asdfasdf'->MyFunc()
367 assert_equal('asdfasdf', var)
368
369 def UseString()
370 'xyork'->MyFunc()
371 enddef
372 UseString()
373 assert_equal('xyork', var)
374
Bram Moolenaare6b53242020-07-01 17:28:33 +0200375 MyFunc(
376 'continued'
377 )
378 assert_equal('continued',
379 var
380 )
381
382 call MyFunc(
383 'more'
384 ..
385 'lines'
386 )
387 assert_equal(
388 'morelines',
389 var)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200390 END
391 writefile(lines, 'Xcall.vim')
392 source Xcall.vim
393 delete('Xcall.vim')
394enddef
395
396def Test_vim9script_call_fail_decl()
397 let lines =<< trim END
398 vim9script
399 let var = ''
400 def MyFunc(arg: string)
401 let var = 123
402 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +0200403 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200404 END
405 writefile(lines, 'Xcall_decl.vim')
406 assert_fails('source Xcall_decl.vim', 'E1054:')
407 delete('Xcall_decl.vim')
408enddef
409
Bram Moolenaar65b95452020-07-19 14:03:09 +0200410def Test_vim9script_call_fail_type()
411 let lines =<< trim END
412 vim9script
413 def MyFunc(arg: string)
414 echo arg
415 enddef
416 MyFunc(1234)
417 END
418 CheckScriptFailure(lines, 'E1013: type mismatch, expected string but got number')
419enddef
420
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200421def Test_vim9script_call_fail_const()
422 let lines =<< trim END
423 vim9script
424 const var = ''
425 def MyFunc(arg: string)
426 var = 'asdf'
427 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +0200428 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200429 END
430 writefile(lines, 'Xcall_const.vim')
431 assert_fails('source Xcall_const.vim', 'E46:')
432 delete('Xcall_const.vim')
433enddef
434
435" Test that inside :function a Python function can be defined, :def is not
436" recognized.
437func Test_function_python()
438 CheckFeature python3
439 let py = 'python3'
440 execute py "<< EOF"
441def do_something():
442 return 1
443EOF
444endfunc
445
446def Test_delfunc()
447 let lines =<< trim END
448 vim9script
Bram Moolenaar4c17ad92020-04-27 22:47:51 +0200449 def g:GoneSoon()
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200450 echo 'hello'
451 enddef
452
453 def CallGoneSoon()
454 GoneSoon()
455 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +0200456 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200457
Bram Moolenaar4c17ad92020-04-27 22:47:51 +0200458 delfunc g:GoneSoon
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200459 CallGoneSoon()
460 END
461 writefile(lines, 'XToDelFunc')
462 assert_fails('so XToDelFunc', 'E933')
463 assert_fails('so XToDelFunc', 'E933')
464
465 delete('XToDelFunc')
466enddef
467
468def Test_redef_failure()
469 call writefile(['def Func0(): string', 'return "Func0"', 'enddef'], 'Xdef')
470 so Xdef
471 call writefile(['def Func1(): string', 'return "Func1"', 'enddef'], 'Xdef')
472 so Xdef
Bram Moolenaar822ba242020-05-24 23:00:18 +0200473 call writefile(['def! Func0(): string', 'enddef', 'defcompile'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200474 call assert_fails('so Xdef', 'E1027:')
475 call writefile(['def Func2(): string', 'return "Func2"', 'enddef'], 'Xdef')
476 so Xdef
477 call delete('Xdef')
478
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200479 call assert_equal(0, g:Func0())
480 call assert_equal('Func1', g:Func1())
481 call assert_equal('Func2', g:Func2())
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200482
483 delfunc! Func0
484 delfunc! Func1
485 delfunc! Func2
486enddef
487
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +0200488def Test_vim9script_func()
489 let lines =<< trim END
490 vim9script
491 func Func(arg)
492 echo a:arg
493 endfunc
494 Func('text')
495 END
496 writefile(lines, 'XVim9Func')
497 so XVim9Func
498
499 delete('XVim9Func')
500enddef
501
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200502" Test for internal functions returning different types
503func Test_InternalFuncRetType()
504 let lines =<< trim END
505 def RetFloat(): float
506 return ceil(1.456)
507 enddef
508
509 def RetListAny(): list<any>
510 return items({'k' : 'v'})
511 enddef
512
513 def RetListString(): list<string>
514 return split('a:b:c', ':')
515 enddef
516
517 def RetListDictAny(): list<dict<any>>
518 return getbufinfo()
519 enddef
520
521 def RetDictNumber(): dict<number>
522 return wordcount()
523 enddef
524
525 def RetDictString(): dict<string>
526 return environ()
527 enddef
528 END
529 call writefile(lines, 'Xscript')
530 source Xscript
531
532 call assert_equal(2.0, RetFloat())
533 call assert_equal([['k', 'v']], RetListAny())
534 call assert_equal(['a', 'b', 'c'], RetListString())
535 call assert_notequal([], RetListDictAny())
536 call assert_notequal({}, RetDictNumber())
537 call assert_notequal({}, RetDictString())
538 call delete('Xscript')
539endfunc
540
541" Test for passing too many or too few arguments to internal functions
542func Test_internalfunc_arg_error()
543 let l =<< trim END
544 def! FArgErr(): float
545 return ceil(1.1, 2)
546 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +0200547 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200548 END
549 call writefile(l, 'Xinvalidarg')
550 call assert_fails('so Xinvalidarg', 'E118:')
551 let l =<< trim END
552 def! FArgErr(): float
553 return ceil()
554 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +0200555 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200556 END
557 call writefile(l, 'Xinvalidarg')
558 call assert_fails('so Xinvalidarg', 'E119:')
559 call delete('Xinvalidarg')
560endfunc
561
562let s:funcResult = 0
563
564def FuncNoArgNoRet()
565 funcResult = 11
566enddef
567
568def FuncNoArgRetNumber(): number
569 funcResult = 22
570 return 1234
571enddef
572
Bram Moolenaarec5929d2020-04-07 20:53:39 +0200573def FuncNoArgRetString(): string
574 funcResult = 45
575 return 'text'
576enddef
577
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200578def FuncOneArgNoRet(arg: number)
579 funcResult = arg
580enddef
581
582def FuncOneArgRetNumber(arg: number): number
583 funcResult = arg
584 return arg
585enddef
586
Bram Moolenaar08938ee2020-04-11 23:17:17 +0200587def FuncTwoArgNoRet(one: bool, two: number)
588 funcResult = two
589enddef
590
Bram Moolenaarec5929d2020-04-07 20:53:39 +0200591def FuncOneArgRetString(arg: string): string
592 return arg
593enddef
594
Bram Moolenaar89228602020-04-05 22:14:54 +0200595def FuncOneArgRetAny(arg: any): any
596 return arg
597enddef
598
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200599def Test_func_type()
600 let Ref1: func()
601 funcResult = 0
602 Ref1 = FuncNoArgNoRet
603 Ref1()
604 assert_equal(11, funcResult)
Bram Moolenaar4c683752020-04-05 21:38:23 +0200605
606 let Ref2: func
607 funcResult = 0
608 Ref2 = FuncNoArgNoRet
609 Ref2()
610 assert_equal(11, funcResult)
611
612 funcResult = 0
613 Ref2 = FuncOneArgNoRet
614 Ref2(12)
615 assert_equal(12, funcResult)
616
617 funcResult = 0
618 Ref2 = FuncNoArgRetNumber
619 assert_equal(1234, Ref2())
620 assert_equal(22, funcResult)
621
622 funcResult = 0
623 Ref2 = FuncOneArgRetNumber
624 assert_equal(13, Ref2(13))
625 assert_equal(13, funcResult)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200626enddef
627
Bram Moolenaar9978d472020-07-05 16:01:56 +0200628def Test_repeat_return_type()
629 let res = 0
630 for n in repeat([1], 3)
631 res += n
632 endfor
633 assert_equal(3, res)
Bram Moolenaarfce82b32020-07-05 16:07:21 +0200634
635 res = 0
636 for n in add([1, 2], 3)
637 res += n
638 endfor
639 assert_equal(6, res)
Bram Moolenaar9978d472020-07-05 16:01:56 +0200640enddef
641
Bram Moolenaar846178a2020-07-05 17:04:13 +0200642def Test_argv_return_type()
643 next fileone filetwo
644 let res = ''
645 for name in argv()
646 res ..= name
647 endfor
648 assert_equal('fileonefiletwo', res)
649enddef
650
Bram Moolenaarec5929d2020-04-07 20:53:39 +0200651def Test_func_type_part()
652 let RefVoid: func: void
653 RefVoid = FuncNoArgNoRet
654 RefVoid = FuncOneArgNoRet
655 CheckDefFailure(['let RefVoid: func: void', 'RefVoid = FuncNoArgRetNumber'], 'E1013: type mismatch, expected func() but got func(): number')
Bram Moolenaar6ff71d82020-05-24 23:45:24 +0200656 CheckDefFailure(['let RefVoid: func: void', 'RefVoid = FuncNoArgRetString'], 'E1013: type mismatch, expected func() but got func(): string')
Bram Moolenaarec5929d2020-04-07 20:53:39 +0200657
658 let RefAny: func(): any
659 RefAny = FuncNoArgRetNumber
660 RefAny = FuncNoArgRetString
661 CheckDefFailure(['let RefAny: func(): any', 'RefAny = FuncNoArgNoRet'], 'E1013: type mismatch, expected func(): any but got func()')
662 CheckDefFailure(['let RefAny: func(): any', 'RefAny = FuncOneArgNoRet'], 'E1013: type mismatch, expected func(): any but got func(number)')
663
664 let RefNr: func: number
665 RefNr = FuncNoArgRetNumber
666 RefNr = FuncOneArgRetNumber
667 CheckDefFailure(['let RefNr: func: number', 'RefNr = FuncNoArgNoRet'], 'E1013: type mismatch, expected func(): number but got func()')
Bram Moolenaar6ff71d82020-05-24 23:45:24 +0200668 CheckDefFailure(['let RefNr: func: number', 'RefNr = FuncNoArgRetString'], 'E1013: type mismatch, expected func(): number but got func(): string')
Bram Moolenaarec5929d2020-04-07 20:53:39 +0200669
670 let RefStr: func: string
671 RefStr = FuncNoArgRetString
672 RefStr = FuncOneArgRetString
673 CheckDefFailure(['let RefStr: func: string', 'RefStr = FuncNoArgNoRet'], 'E1013: type mismatch, expected func(): string but got func()')
674 CheckDefFailure(['let RefStr: func: string', 'RefStr = FuncNoArgRetNumber'], 'E1013: type mismatch, expected func(): string but got func(): number')
675enddef
676
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200677def Test_func_type_fails()
678 CheckDefFailure(['let ref1: func()'], 'E704:')
679
680 CheckDefFailure(['let Ref1: func()', 'Ref1 = FuncNoArgRetNumber'], 'E1013: type mismatch, expected func() but got func(): number')
681 CheckDefFailure(['let Ref1: func()', 'Ref1 = FuncOneArgNoRet'], 'E1013: type mismatch, expected func() but got func(number)')
682 CheckDefFailure(['let Ref1: func()', 'Ref1 = FuncOneArgRetNumber'], 'E1013: type mismatch, expected func() but got func(number): number')
Bram Moolenaar6ff71d82020-05-24 23:45:24 +0200683 CheckDefFailure(['let Ref1: func(bool)', 'Ref1 = FuncTwoArgNoRet'], 'E1013: type mismatch, expected func(bool) but got func(bool, number)')
684 CheckDefFailure(['let Ref1: func(?bool)', 'Ref1 = FuncTwoArgNoRet'], 'E1013: type mismatch, expected func(?bool) but got func(bool, number)')
685 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 +0200686
687 call CheckDefFailure(['let RefWrong: func(string ,number)'], 'E1068:')
688 call CheckDefFailure(['let RefWrong: func(string,number)'], 'E1069:')
689 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:')
690 call CheckDefFailure(['let RefWrong: func(bool):string'], 'E1069:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200691enddef
692
Bram Moolenaar89228602020-04-05 22:14:54 +0200693def Test_func_return_type()
694 let nr: number
695 nr = FuncNoArgRetNumber()
696 assert_equal(1234, nr)
697
698 nr = FuncOneArgRetAny(122)
699 assert_equal(122, nr)
700
701 let str: string
702 str = FuncOneArgRetAny('yes')
703 assert_equal('yes', str)
704
705 CheckDefFailure(['let str: string', 'str = FuncNoArgRetNumber()'], 'E1013: type mismatch, expected string but got number')
706enddef
707
Bram Moolenaar5e774c72020-04-12 21:53:00 +0200708def MultiLine(
709 arg1: string,
710 arg2 = 1234,
711 ...rest: list<string>
712 ): string
713 return arg1 .. arg2 .. join(rest, '-')
714enddef
715
Bram Moolenaar2c330432020-04-13 14:41:35 +0200716def MultiLineComment(
717 arg1: string, # comment
718 arg2 = 1234, # comment
719 ...rest: list<string> # comment
720 ): string # comment
721 return arg1 .. arg2 .. join(rest, '-')
722enddef
723
Bram Moolenaar5e774c72020-04-12 21:53:00 +0200724def Test_multiline()
725 assert_equal('text1234', MultiLine('text'))
726 assert_equal('text777', MultiLine('text', 777))
727 assert_equal('text777one', MultiLine('text', 777, 'one'))
728 assert_equal('text777one-two', MultiLine('text', 777, 'one', 'two'))
729enddef
730
Bram Moolenaar23e03252020-04-12 22:22:31 +0200731func Test_multiline_not_vim9()
732 call assert_equal('text1234', MultiLine('text'))
733 call assert_equal('text777', MultiLine('text', 777))
734 call assert_equal('text777one', MultiLine('text', 777, 'one'))
735 call assert_equal('text777one-two', MultiLine('text', 777, 'one', 'two'))
736endfunc
737
Bram Moolenaar5e774c72020-04-12 21:53:00 +0200738
Bram Moolenaaree4e0c12020-04-06 21:35:05 +0200739" When using CheckScriptFailure() for the below test, E1010 is generated instead
740" of E1056.
741func Test_E1056_1059()
742 let caught_1056 = 0
743 try
744 def F():
745 return 1
746 enddef
747 catch /E1056:/
748 let caught_1056 = 1
749 endtry
750 call assert_equal(1, caught_1056)
751
752 let caught_1059 = 0
753 try
754 def F5(items : list)
755 echo 'a'
756 enddef
757 catch /E1059:/
758 let caught_1059 = 1
759 endtry
760 call assert_equal(1, caught_1059)
761endfunc
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200762
Bram Moolenaar015f4262020-05-05 21:25:22 +0200763func DelMe()
764 echo 'DelMe'
765endfunc
766
767def Test_deleted_function()
768 CheckDefExecFailure([
769 'let RefMe: func = function("g:DelMe")',
770 'delfunc g:DelMe',
771 'echo RefMe()'], 'E117:')
772enddef
773
774def Test_unknown_function()
775 CheckDefExecFailure([
776 'let Ref: func = function("NotExist")',
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +0200777 'delfunc g:NotExist'], 'E700:')
Bram Moolenaar015f4262020-05-05 21:25:22 +0200778enddef
779
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +0200780def RefFunc(Ref: func(string): string): string
781 return Ref('more')
782enddef
783
784def Test_closure_simple()
785 let local = 'some '
786 assert_equal('some more', RefFunc({s -> local .. s}))
787enddef
788
Bram Moolenaarbf67ea12020-05-02 17:52:42 +0200789def MakeRef()
790 let local = 'some '
791 g:Ref = {s -> local .. s}
792enddef
793
794def Test_closure_ref_after_return()
795 MakeRef()
796 assert_equal('some thing', g:Ref('thing'))
797 unlet g:Ref
798enddef
799
Bram Moolenaar5adc55c2020-05-02 23:12:58 +0200800def MakeTwoRefs()
801 let local = ['some']
802 g:Extend = {s -> local->add(s)}
803 g:Read = {-> local}
804enddef
805
806def Test_closure_two_refs()
807 MakeTwoRefs()
808 assert_equal('some', join(g:Read(), ' '))
809 g:Extend('more')
810 assert_equal('some more', join(g:Read(), ' '))
811 g:Extend('even')
812 assert_equal('some more even', join(g:Read(), ' '))
813
814 unlet g:Extend
815 unlet g:Read
816enddef
817
Bram Moolenaar5adc55c2020-05-02 23:12:58 +0200818def ReadRef(Ref: func(): list<string>): string
819 return join(Ref(), ' ')
820enddef
821
822def ExtendRef(Ref: func(string), add: string)
823 Ref(add)
824enddef
825
826def Test_closure_two_indirect_refs()
Bram Moolenaarf7779c62020-05-03 15:38:16 +0200827 MakeTwoRefs()
Bram Moolenaar5adc55c2020-05-02 23:12:58 +0200828 assert_equal('some', ReadRef(g:Read))
829 ExtendRef(g:Extend, 'more')
830 assert_equal('some more', ReadRef(g:Read))
831 ExtendRef(g:Extend, 'even')
832 assert_equal('some more even', ReadRef(g:Read))
833
834 unlet g:Extend
835 unlet g:Read
836enddef
Bram Moolenaarbf67ea12020-05-02 17:52:42 +0200837
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +0200838def MakeArgRefs(theArg: string)
839 let local = 'loc_val'
840 g:UseArg = {s -> theArg .. '/' .. local .. '/' .. s}
841enddef
842
843def MakeArgRefsVarargs(theArg: string, ...rest: list<string>)
844 let local = 'the_loc'
845 g:UseVararg = {s -> theArg .. '/' .. local .. '/' .. s .. '/' .. join(rest)}
846enddef
847
848def Test_closure_using_argument()
849 MakeArgRefs('arg_val')
850 assert_equal('arg_val/loc_val/call_val', g:UseArg('call_val'))
851
852 MakeArgRefsVarargs('arg_val', 'one', 'two')
853 assert_equal('arg_val/the_loc/call_val/one two', g:UseVararg('call_val'))
854
855 unlet g:UseArg
856 unlet g:UseVararg
857enddef
858
Bram Moolenaarb68b3462020-05-06 21:06:30 +0200859def MakeGetAndAppendRefs()
860 let local = 'a'
861
862 def Append(arg: string)
863 local ..= arg
864 enddef
865 g:Append = Append
866
867 def Get(): string
868 return local
869 enddef
870 g:Get = Get
871enddef
872
873def Test_closure_append_get()
874 MakeGetAndAppendRefs()
875 assert_equal('a', g:Get())
876 g:Append('-b')
877 assert_equal('a-b', g:Get())
878 g:Append('-c')
879 assert_equal('a-b-c', g:Get())
880
881 unlet g:Append
882 unlet g:Get
883enddef
884
Bram Moolenaar04b12692020-05-04 23:24:44 +0200885def Test_nested_closure()
886 let local = 'text'
887 def Closure(arg: string): string
888 return local .. arg
889 enddef
890 assert_equal('text!!!', Closure('!!!'))
891enddef
892
Bram Moolenaar6f5b6df2020-05-16 21:20:12 +0200893func GetResult(Ref)
894 return a:Ref('some')
895endfunc
896
897def Test_call_closure_not_compiled()
898 let text = 'text'
899 g:Ref = {s -> s .. text}
900 assert_equal('sometext', GetResult(g:Ref))
901enddef
902
Bram Moolenaar865af6b2020-06-18 18:45:49 +0200903def Test_sort_return_type()
904 let res: list<number>
905 res = [1, 2, 3]->sort()
906enddef
907
Bram Moolenaarf151ad12020-06-30 13:38:01 +0200908def Test_getqflist_return_type()
909 let l = getqflist()
910 assert_equal([], l)
911
912 let d = getqflist(#{items: 0})
913 assert_equal(#{items: []}, d)
914enddef
915
916def Test_getloclist_return_type()
917 let l = getloclist(1)
918 assert_equal([], l)
919
920 let d = getloclist(1, #{items: 0})
921 assert_equal(#{items: []}, d)
922enddef
923
Bram Moolenaara66ba012020-07-05 18:41:08 +0200924def Test_copy_return_type()
925 let l = copy([1, 2, 3])
926 let res = 0
927 for n in l
928 res += n
929 endfor
930 assert_equal(6, res)
931
932 let dl = deepcopy([1, 2, 3])
933 res = 0
934 for n in dl
935 res += n
936 endfor
937 assert_equal(6, res)
938enddef
939
Bram Moolenaarb3c019c2020-07-05 20:08:39 +0200940def Test_extend_return_type()
941 let l = extend([1, 2], [3])
942 let res = 0
943 for n in l
944 res += n
945 endfor
946 assert_equal(6, res)
947enddef
948
Bram Moolenaar252e88a2020-07-05 20:47:18 +0200949def Test_insert_return_type()
950 let l = insert([2, 1], 3)
951 let res = 0
952 for n in l
953 res += n
954 endfor
955 assert_equal(6, res)
956enddef
957
Bram Moolenaar67627352020-07-05 21:10:24 +0200958def Test_reverse_return_type()
959 let l = reverse([1, 2, 3])
960 let res = 0
961 for n in l
962 res += n
963 endfor
964 assert_equal(6, res)
965enddef
966
Bram Moolenaarad7c2492020-07-05 20:55:29 +0200967def Test_remove_return_type()
968 let l = remove(#{one: [1, 2], two: [3, 4]}, 'one')
969 let res = 0
970 for n in l
971 res += n
972 endfor
973 assert_equal(3, res)
974enddef
975
Bram Moolenaar0d94ad62020-07-05 20:16:41 +0200976def Test_filter_return_type()
977 let l = filter([1, 2, 3], {-> 1})
978 let res = 0
979 for n in l
980 res += n
981 endfor
982 assert_equal(6, res)
983enddef
984
Bram Moolenaarf1a23682020-07-13 18:55:48 +0200985def Wrong_dict_key_type(items: list<number>): list<number>
986 return filter(items, {_, val -> get({val: 1}, 'x')})
987enddef
988
989def Test_wrong_dict_key_type()
990 assert_fails('Wrong_dict_key_type([1, 2, 3])', 'E1029:')
991enddef
992
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +0200993def Line_continuation_in_def(dir: string = ''): string
994 let path: string = empty(dir)
995 \ ? 'empty'
996 \ : 'full'
997 return path
998enddef
999
1000def Test_line_continuation_in_def()
1001 assert_equal('full', Line_continuation_in_def('.'))
1002enddef
1003
Bram Moolenaar7a4b8982020-07-08 17:36:21 +02001004def Line_continuation_in_lambda(): list<number>
1005 let x = range(97, 100)
Bram Moolenaar914e7ea2020-07-11 15:20:48 +02001006 ->map({_, v -> nr2char(v)
Bram Moolenaar7a4b8982020-07-08 17:36:21 +02001007 ->toupper()})
1008 ->reverse()
1009 return x
1010enddef
1011
1012def Test_line_continuation_in_lambda()
1013 assert_equal(['D', 'C', 'B', 'A'], Line_continuation_in_lambda())
1014enddef
1015
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001016func Test_silent_echo()
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001017 CheckScreendump
1018
1019 let lines =<< trim END
1020 vim9script
1021 def EchoNothing()
1022 silent echo ''
1023 enddef
1024 defcompile
1025 END
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001026 call writefile(lines, 'XTest_silent_echo')
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001027
1028 " Check that the balloon shows up after a mouse move
1029 let buf = RunVimInTerminal('-S XTest_silent_echo', {'rows': 6})
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001030 call term_sendkeys(buf, ":abc")
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001031 call VerifyScreenDump(buf, 'Test_vim9_silent_echo', {})
1032
1033 " clean up
1034 call StopVimInTerminal(buf)
1035 call delete('XTest_silent_echo')
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001036endfunc
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001037
Bram Moolenaar985116a2020-07-12 17:31:09 +02001038def Fibonacci(n: number): number
1039 if n < 2
1040 return n
1041 else
1042 return Fibonacci(n - 1) + Fibonacci(n - 2)
1043 endif
1044enddef
1045
1046def Test_recursive_call()
1047 assert_equal(6765, Fibonacci(20))
1048enddef
1049
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001050def TreeWalk(dir: string): list<any>
1051 return readdir(dir)->map({_, val ->
1052 fnamemodify(dir .. '/' .. val, ':p')->isdirectory()
1053 ? {val : TreeWalk(dir .. '/' .. val)}
1054 : val
1055 })
1056enddef
1057
1058def Test_closure_in_map()
1059 mkdir('XclosureDir/tdir', 'p')
1060 writefile(['111'], 'XclosureDir/file1')
1061 writefile(['222'], 'XclosureDir/file2')
1062 writefile(['333'], 'XclosureDir/tdir/file3')
1063
1064 assert_equal(['file1', 'file2', {'tdir': ['file3']}], TreeWalk('XclosureDir'))
1065
1066 delete('XclosureDir', 'rf')
1067enddef
1068
Bram Moolenaara90afb92020-07-15 22:38:56 +02001069def Test_partial_call()
1070 let Xsetlist = function('setloclist', [0])
1071 Xsetlist([], ' ', {'title': 'test'})
1072 assert_equal({'title': 'test'}, getloclist(0, {'title': 1}))
1073
1074 Xsetlist = function('setloclist', [0, [], ' '])
1075 Xsetlist({'title': 'test'})
1076 assert_equal({'title': 'test'}, getloclist(0, {'title': 1}))
1077
1078 Xsetlist = function('setqflist')
1079 Xsetlist([], ' ', {'title': 'test'})
1080 assert_equal({'title': 'test'}, getqflist({'title': 1}))
1081
1082 Xsetlist = function('setqflist', [[], ' '])
1083 Xsetlist({'title': 'test'})
1084 assert_equal({'title': 'test'}, getqflist({'title': 1}))
1085enddef
1086
Bram Moolenaarf7779c62020-05-03 15:38:16 +02001087
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001088" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker