blob: e8d712f0a741706dec6d376ef3d44fbc87321558 [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 Moolenaar5deeb3f2020-04-05 17:08:17 +02006
7func Test_def_basic()
8 def SomeFunc(): string
9 return 'yes'
10 enddef
11 call assert_equal('yes', SomeFunc())
12endfunc
13
14def ReturnString(): string
15 return 'string'
16enddef
17
18def ReturnNumber(): number
19 return 123
20enddef
21
22let g:notNumber = 'string'
23
24def ReturnGlobal(): number
25 return g:notNumber
26enddef
27
28def Test_return_something()
29 assert_equal('string', ReturnString())
30 assert_equal(123, ReturnNumber())
31 assert_fails('call ReturnGlobal()', 'E1029: Expected number but got string')
32enddef
33
Bram Moolenaarefd88552020-06-18 20:50:10 +020034def Test_missing_return()
35 CheckDefFailure(['def Missing(): number',
36 ' if g:cond',
37 ' echo "no return"',
38 ' else',
39 ' return 0',
40 ' endif'
41 'enddef'], 'E1027:')
42 CheckDefFailure(['def Missing(): number',
43 ' if g:cond',
44 ' return 1',
45 ' else',
46 ' echo "no return"',
47 ' endif'
48 'enddef'], 'E1027:')
49 CheckDefFailure(['def Missing(): number',
50 ' if g:cond',
51 ' return 1',
52 ' else',
53 ' return 2',
54 ' endif'
55 ' return 3'
56 'enddef'], 'E1095:')
57enddef
58
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +020059let s:nothing = 0
60def ReturnNothing()
61 s:nothing = 1
62 if true
63 return
64 endif
65 s:nothing = 2
66enddef
67
68def Test_return_nothing()
69 ReturnNothing()
70 assert_equal(1, s:nothing)
71enddef
72
73func Increment()
74 let g:counter += 1
75endfunc
76
77def Test_call_ufunc_count()
78 g:counter = 1
79 Increment()
80 Increment()
81 Increment()
82 " works with and without :call
83 assert_equal(4, g:counter)
84 call assert_equal(4, g:counter)
85 unlet g:counter
86enddef
87
88def MyVarargs(arg: string, ...rest: list<string>): string
89 let res = arg
90 for s in rest
91 res ..= ',' .. s
92 endfor
93 return res
94enddef
95
96def Test_call_varargs()
97 assert_equal('one', MyVarargs('one'))
98 assert_equal('one,two', MyVarargs('one', 'two'))
99 assert_equal('one,two,three', MyVarargs('one', 'two', 'three'))
100enddef
101
102def MyDefaultArgs(name = 'string'): string
103 return name
104enddef
105
106def Test_call_default_args()
107 assert_equal('string', MyDefaultArgs())
108 assert_equal('one', MyDefaultArgs('one'))
109 assert_fails('call MyDefaultArgs("one", "two")', 'E118:')
110
Bram Moolenaar822ba242020-05-24 23:00:18 +0200111 CheckScriptFailure(['def Func(arg: number = asdf)', 'enddef', 'defcompile'], 'E1001:')
112 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 +0200113enddef
114
115def Test_nested_function()
116 def Nested(arg: string): string
117 return 'nested ' .. arg
118 enddef
119 assert_equal('nested function', Nested('function'))
120
Bram Moolenaar0e65d3d2020-05-05 17:53:16 +0200121 CheckDefFailure(['def Nested()', 'enddef', 'Nested(66)'], 'E118:')
122 CheckDefFailure(['def Nested(arg: string)', 'enddef', 'Nested()'], 'E119:')
123
Bram Moolenaar04b12692020-05-04 23:24:44 +0200124 CheckDefFailure(['func Nested()', 'endfunc'], 'E1086:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200125enddef
126
127func Test_call_default_args_from_func()
128 call assert_equal('string', MyDefaultArgs())
129 call assert_equal('one', MyDefaultArgs('one'))
130 call assert_fails('call MyDefaultArgs("one", "two")', 'E118:')
131endfunc
132
133func TakesOneArg(arg)
134 echo a:arg
135endfunc
136
137def Test_call_wrong_args()
138 call CheckDefFailure(['TakesOneArg()'], 'E119:')
139 call CheckDefFailure(['TakesOneArg(11, 22)'], 'E118:')
140 call CheckDefFailure(['bufnr(xxx)'], 'E1001:')
Bram Moolenaar1c0d44f2020-05-02 19:04:58 +0200141 call CheckScriptFailure(['def Func(Ref: func(s: string))'], 'E475:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200142enddef
143
144" Default arg and varargs
145def MyDefVarargs(one: string, two = 'foo', ...rest: list<string>): string
146 let res = one .. ',' .. two
147 for s in rest
148 res ..= ',' .. s
149 endfor
150 return res
151enddef
152
153def Test_call_def_varargs()
154 call assert_fails('call MyDefVarargs()', 'E119:')
155 assert_equal('one,foo', MyDefVarargs('one'))
156 assert_equal('one,two', MyDefVarargs('one', 'two'))
157 assert_equal('one,two,three', MyDefVarargs('one', 'two', 'three'))
Bram Moolenaar49cf7cc2020-04-07 22:45:00 +0200158 call CheckDefFailure(['MyDefVarargs("one", 22)'], 'E1013: argument 2: type mismatch, expected string but got number')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200159enddef
160
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200161let s:value = ''
162
163def FuncOneDefArg(opt = 'text')
164 s:value = opt
165enddef
166
167def FuncTwoDefArg(nr = 123, opt = 'text'): string
168 return nr .. opt
169enddef
170
171def FuncVarargs(...arg: list<string>): string
172 return join(arg, ',')
173enddef
174
175def Test_func_type_varargs()
176 let RefDefArg: func(?string)
177 RefDefArg = FuncOneDefArg
178 RefDefArg()
179 assert_equal('text', s:value)
180 RefDefArg('some')
181 assert_equal('some', s:value)
182
183 let RefDef2Arg: func(?number, ?string): string
184 RefDef2Arg = FuncTwoDefArg
185 assert_equal('123text', RefDef2Arg())
186 assert_equal('99text', RefDef2Arg(99))
187 assert_equal('77some', RefDef2Arg(77, 'some'))
188
189 call CheckDefFailure(['let RefWrong: func(string?)'], 'E1010:')
190 call CheckDefFailure(['let RefWrong: func(?string, string)'], 'E1007:')
191
192 let RefVarargs: func(...list<string>): string
193 RefVarargs = FuncVarargs
194 assert_equal('', RefVarargs())
195 assert_equal('one', RefVarargs('one'))
196 assert_equal('one,two', RefVarargs('one', 'two'))
197
198 call CheckDefFailure(['let RefWrong: func(...list<string>, string)'], 'E110:')
199 call CheckDefFailure(['let RefWrong: func(...list<string>, ?string)'], 'E110:')
200enddef
201
Bram Moolenaar0b76b422020-04-07 22:05:08 +0200202" Only varargs
203def MyVarargsOnly(...args: list<string>): string
204 return join(args, ',')
205enddef
206
207def Test_call_varargs_only()
208 assert_equal('', MyVarargsOnly())
209 assert_equal('one', MyVarargsOnly('one'))
210 assert_equal('one,two', MyVarargsOnly('one', 'two'))
211 call CheckDefFailure(['MyVarargsOnly(1)'], 'E1013: argument 1: type mismatch, expected string but got number')
212 call CheckDefFailure(['MyVarargsOnly("one", 2)'], 'E1013: argument 2: type mismatch, expected string but got number')
213enddef
214
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200215def Test_using_var_as_arg()
Bram Moolenaar822ba242020-05-24 23:00:18 +0200216 call writefile(['def Func(x: number)', 'let x = 234', 'enddef', 'defcompile'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200217 call assert_fails('so Xdef', 'E1006:')
218 call delete('Xdef')
219enddef
220
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200221def DictArg(arg: dict<string>)
222 arg['key'] = 'value'
223enddef
224
225def ListArg(arg: list<string>)
226 arg[0] = 'value'
227enddef
228
229def Test_assign_to_argument()
230 " works for dict and list
231 let d: dict<string> = {}
232 DictArg(d)
233 assert_equal('value', d['key'])
234 let l: list<string> = []
235 ListArg(l)
236 assert_equal('value', l[0])
237
Bram Moolenaar822ba242020-05-24 23:00:18 +0200238 call CheckScriptFailure(['def Func(arg: number)', 'arg = 3', 'enddef', 'defcompile'], 'E1090:')
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200239enddef
240
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200241def Test_call_func_defined_later()
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200242 call assert_equal('one', g:DefinedLater('one'))
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200243 call assert_fails('call NotDefined("one")', 'E117:')
244enddef
245
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200246func DefinedLater(arg)
247 return a:arg
248endfunc
249
250def Test_call_funcref()
251 assert_equal(3, g:SomeFunc('abc'))
252 assert_fails('NotAFunc()', 'E117:')
253 assert_fails('g:NotAFunc()', 'E117:')
254enddef
255
256let SomeFunc = function('len')
257let NotAFunc = 'text'
258
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200259def CombineFuncrefTypes()
260 " same arguments, different return type
261 let Ref1: func(bool): string
262 let Ref2: func(bool): number
263 let Ref3: func(bool): any
264 Ref3 = g:cond ? Ref1 : Ref2
265
266 " different number of arguments
267 let Refa1: func(bool): number
268 let Refa2: func(bool, number): number
269 let Refa3: func: number
270 Refa3 = g:cond ? Refa1 : Refa2
271
272 " different argument types
273 let Refb1: func(bool, string): number
274 let Refb2: func(string, number): number
275 let Refb3: func(any, any): number
276 Refb3 = g:cond ? Refb1 : Refb2
277enddef
278
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200279def FuncWithForwardCall()
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200280 return g:DefinedEvenLater("yes")
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200281enddef
282
283def DefinedEvenLater(arg: string): string
284 return arg
285enddef
286
287def Test_error_in_nested_function()
288 " Error in called function requires unwinding the call stack.
Bram Moolenaar05a55512020-07-05 15:52:19 +0200289 assert_fails('call FuncWithForwardCall()', 'E1096')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200290enddef
291
292def Test_return_type_wrong()
Bram Moolenaar822ba242020-05-24 23:00:18 +0200293 CheckScriptFailure(['def Func(): number', 'return "a"', 'enddef', 'defcompile'], 'expected number but got string')
294 CheckScriptFailure(['def Func(): string', 'return 1', 'enddef', 'defcompile'], 'expected string but got number')
Bram Moolenaar05a55512020-07-05 15:52:19 +0200295 CheckScriptFailure(['def Func(): void', 'return "a"', 'enddef', 'defcompile'], 'E1096: Returning a value in a function without a return type')
296 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 +0200297
Bram Moolenaar822ba242020-05-24 23:00:18 +0200298 CheckScriptFailure(['def Func(): number', 'return', 'enddef', 'defcompile'], 'E1003:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200299
300 CheckScriptFailure(['def Func(): list', 'return []', 'enddef'], 'E1008:')
301 CheckScriptFailure(['def Func(): dict', 'return {}', 'enddef'], 'E1008:')
Bram Moolenaaree4e0c12020-04-06 21:35:05 +0200302 CheckScriptFailure(['def Func()', 'return 1'], 'E1057:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200303enddef
304
305def Test_arg_type_wrong()
306 CheckScriptFailure(['def Func3(items: list)', 'echo "a"', 'enddef'], 'E1008: Missing <type>')
Bram Moolenaaree4e0c12020-04-06 21:35:05 +0200307 CheckScriptFailure(['def Func4(...)', 'echo "a"', 'enddef'], 'E1055: Missing name after ...')
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +0200308 CheckScriptFailure(['def Func5(items:string)', 'echo "a"'], 'E1069:')
Bram Moolenaar6e949782020-04-13 17:21:00 +0200309 CheckScriptFailure(['def Func5(items)', 'echo "a"'], 'E1077:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200310enddef
311
312def Test_vim9script_call()
313 let lines =<< trim END
314 vim9script
315 let var = ''
316 def MyFunc(arg: string)
317 var = arg
318 enddef
319 MyFunc('foobar')
320 assert_equal('foobar', var)
321
322 let str = 'barfoo'
323 str->MyFunc()
324 assert_equal('barfoo', var)
325
Bram Moolenaar67979662020-06-20 22:50:47 +0200326 g:value = 'value'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200327 g:value->MyFunc()
328 assert_equal('value', var)
329
330 let listvar = []
331 def ListFunc(arg: list<number>)
332 listvar = arg
333 enddef
334 [1, 2, 3]->ListFunc()
335 assert_equal([1, 2, 3], listvar)
336
337 let dictvar = {}
338 def DictFunc(arg: dict<number>)
339 dictvar = arg
340 enddef
341 {'a': 1, 'b': 2}->DictFunc()
342 assert_equal(#{a: 1, b: 2}, dictvar)
343 def CompiledDict()
344 {'a': 3, 'b': 4}->DictFunc()
345 enddef
346 CompiledDict()
347 assert_equal(#{a: 3, b: 4}, dictvar)
348
349 #{a: 3, b: 4}->DictFunc()
350 assert_equal(#{a: 3, b: 4}, dictvar)
351
352 ('text')->MyFunc()
353 assert_equal('text', var)
354 ("some")->MyFunc()
355 assert_equal('some', var)
Bram Moolenaare6b53242020-07-01 17:28:33 +0200356
357 MyFunc(
358 'continued'
359 )
360 assert_equal('continued',
361 var
362 )
363
364 call MyFunc(
365 'more'
366 ..
367 'lines'
368 )
369 assert_equal(
370 'morelines',
371 var)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200372 END
373 writefile(lines, 'Xcall.vim')
374 source Xcall.vim
375 delete('Xcall.vim')
376enddef
377
378def Test_vim9script_call_fail_decl()
379 let lines =<< trim END
380 vim9script
381 let var = ''
382 def MyFunc(arg: string)
383 let var = 123
384 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +0200385 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200386 END
387 writefile(lines, 'Xcall_decl.vim')
388 assert_fails('source Xcall_decl.vim', 'E1054:')
389 delete('Xcall_decl.vim')
390enddef
391
392def Test_vim9script_call_fail_const()
393 let lines =<< trim END
394 vim9script
395 const var = ''
396 def MyFunc(arg: string)
397 var = 'asdf'
398 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +0200399 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200400 END
401 writefile(lines, 'Xcall_const.vim')
402 assert_fails('source Xcall_const.vim', 'E46:')
403 delete('Xcall_const.vim')
404enddef
405
406" Test that inside :function a Python function can be defined, :def is not
407" recognized.
408func Test_function_python()
409 CheckFeature python3
410 let py = 'python3'
411 execute py "<< EOF"
412def do_something():
413 return 1
414EOF
415endfunc
416
417def Test_delfunc()
418 let lines =<< trim END
419 vim9script
Bram Moolenaar4c17ad92020-04-27 22:47:51 +0200420 def g:GoneSoon()
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200421 echo 'hello'
422 enddef
423
424 def CallGoneSoon()
425 GoneSoon()
426 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +0200427 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200428
Bram Moolenaar4c17ad92020-04-27 22:47:51 +0200429 delfunc g:GoneSoon
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200430 CallGoneSoon()
431 END
432 writefile(lines, 'XToDelFunc')
433 assert_fails('so XToDelFunc', 'E933')
434 assert_fails('so XToDelFunc', 'E933')
435
436 delete('XToDelFunc')
437enddef
438
439def Test_redef_failure()
440 call writefile(['def Func0(): string', 'return "Func0"', 'enddef'], 'Xdef')
441 so Xdef
442 call writefile(['def Func1(): string', 'return "Func1"', 'enddef'], 'Xdef')
443 so Xdef
Bram Moolenaar822ba242020-05-24 23:00:18 +0200444 call writefile(['def! Func0(): string', 'enddef', 'defcompile'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200445 call assert_fails('so Xdef', 'E1027:')
446 call writefile(['def Func2(): string', 'return "Func2"', 'enddef'], 'Xdef')
447 so Xdef
448 call delete('Xdef')
449
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200450 call assert_equal(0, g:Func0())
451 call assert_equal('Func1', g:Func1())
452 call assert_equal('Func2', g:Func2())
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200453
454 delfunc! Func0
455 delfunc! Func1
456 delfunc! Func2
457enddef
458
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +0200459def Test_vim9script_func()
460 let lines =<< trim END
461 vim9script
462 func Func(arg)
463 echo a:arg
464 endfunc
465 Func('text')
466 END
467 writefile(lines, 'XVim9Func')
468 so XVim9Func
469
470 delete('XVim9Func')
471enddef
472
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200473" Test for internal functions returning different types
474func Test_InternalFuncRetType()
475 let lines =<< trim END
476 def RetFloat(): float
477 return ceil(1.456)
478 enddef
479
480 def RetListAny(): list<any>
481 return items({'k' : 'v'})
482 enddef
483
484 def RetListString(): list<string>
485 return split('a:b:c', ':')
486 enddef
487
488 def RetListDictAny(): list<dict<any>>
489 return getbufinfo()
490 enddef
491
492 def RetDictNumber(): dict<number>
493 return wordcount()
494 enddef
495
496 def RetDictString(): dict<string>
497 return environ()
498 enddef
499 END
500 call writefile(lines, 'Xscript')
501 source Xscript
502
503 call assert_equal(2.0, RetFloat())
504 call assert_equal([['k', 'v']], RetListAny())
505 call assert_equal(['a', 'b', 'c'], RetListString())
506 call assert_notequal([], RetListDictAny())
507 call assert_notequal({}, RetDictNumber())
508 call assert_notequal({}, RetDictString())
509 call delete('Xscript')
510endfunc
511
512" Test for passing too many or too few arguments to internal functions
513func Test_internalfunc_arg_error()
514 let l =<< trim END
515 def! FArgErr(): float
516 return ceil(1.1, 2)
517 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +0200518 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200519 END
520 call writefile(l, 'Xinvalidarg')
521 call assert_fails('so Xinvalidarg', 'E118:')
522 let l =<< trim END
523 def! FArgErr(): float
524 return ceil()
525 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +0200526 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200527 END
528 call writefile(l, 'Xinvalidarg')
529 call assert_fails('so Xinvalidarg', 'E119:')
530 call delete('Xinvalidarg')
531endfunc
532
533let s:funcResult = 0
534
535def FuncNoArgNoRet()
536 funcResult = 11
537enddef
538
539def FuncNoArgRetNumber(): number
540 funcResult = 22
541 return 1234
542enddef
543
Bram Moolenaarec5929d2020-04-07 20:53:39 +0200544def FuncNoArgRetString(): string
545 funcResult = 45
546 return 'text'
547enddef
548
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200549def FuncOneArgNoRet(arg: number)
550 funcResult = arg
551enddef
552
553def FuncOneArgRetNumber(arg: number): number
554 funcResult = arg
555 return arg
556enddef
557
Bram Moolenaar08938ee2020-04-11 23:17:17 +0200558def FuncTwoArgNoRet(one: bool, two: number)
559 funcResult = two
560enddef
561
Bram Moolenaarec5929d2020-04-07 20:53:39 +0200562def FuncOneArgRetString(arg: string): string
563 return arg
564enddef
565
Bram Moolenaar89228602020-04-05 22:14:54 +0200566def FuncOneArgRetAny(arg: any): any
567 return arg
568enddef
569
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200570def Test_func_type()
571 let Ref1: func()
572 funcResult = 0
573 Ref1 = FuncNoArgNoRet
574 Ref1()
575 assert_equal(11, funcResult)
Bram Moolenaar4c683752020-04-05 21:38:23 +0200576
577 let Ref2: func
578 funcResult = 0
579 Ref2 = FuncNoArgNoRet
580 Ref2()
581 assert_equal(11, funcResult)
582
583 funcResult = 0
584 Ref2 = FuncOneArgNoRet
585 Ref2(12)
586 assert_equal(12, funcResult)
587
588 funcResult = 0
589 Ref2 = FuncNoArgRetNumber
590 assert_equal(1234, Ref2())
591 assert_equal(22, funcResult)
592
593 funcResult = 0
594 Ref2 = FuncOneArgRetNumber
595 assert_equal(13, Ref2(13))
596 assert_equal(13, funcResult)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200597enddef
598
Bram Moolenaar9978d472020-07-05 16:01:56 +0200599def Test_repeat_return_type()
600 let res = 0
601 for n in repeat([1], 3)
602 res += n
603 endfor
604 assert_equal(3, res)
605enddef
606
Bram Moolenaarec5929d2020-04-07 20:53:39 +0200607def Test_func_type_part()
608 let RefVoid: func: void
609 RefVoid = FuncNoArgNoRet
610 RefVoid = FuncOneArgNoRet
611 CheckDefFailure(['let RefVoid: func: void', 'RefVoid = FuncNoArgRetNumber'], 'E1013: type mismatch, expected func() but got func(): number')
Bram Moolenaar6ff71d82020-05-24 23:45:24 +0200612 CheckDefFailure(['let RefVoid: func: void', 'RefVoid = FuncNoArgRetString'], 'E1013: type mismatch, expected func() but got func(): string')
Bram Moolenaarec5929d2020-04-07 20:53:39 +0200613
614 let RefAny: func(): any
615 RefAny = FuncNoArgRetNumber
616 RefAny = FuncNoArgRetString
617 CheckDefFailure(['let RefAny: func(): any', 'RefAny = FuncNoArgNoRet'], 'E1013: type mismatch, expected func(): any but got func()')
618 CheckDefFailure(['let RefAny: func(): any', 'RefAny = FuncOneArgNoRet'], 'E1013: type mismatch, expected func(): any but got func(number)')
619
620 let RefNr: func: number
621 RefNr = FuncNoArgRetNumber
622 RefNr = FuncOneArgRetNumber
623 CheckDefFailure(['let RefNr: func: number', 'RefNr = FuncNoArgNoRet'], 'E1013: type mismatch, expected func(): number but got func()')
Bram Moolenaar6ff71d82020-05-24 23:45:24 +0200624 CheckDefFailure(['let RefNr: func: number', 'RefNr = FuncNoArgRetString'], 'E1013: type mismatch, expected func(): number but got func(): string')
Bram Moolenaarec5929d2020-04-07 20:53:39 +0200625
626 let RefStr: func: string
627 RefStr = FuncNoArgRetString
628 RefStr = FuncOneArgRetString
629 CheckDefFailure(['let RefStr: func: string', 'RefStr = FuncNoArgNoRet'], 'E1013: type mismatch, expected func(): string but got func()')
630 CheckDefFailure(['let RefStr: func: string', 'RefStr = FuncNoArgRetNumber'], 'E1013: type mismatch, expected func(): string but got func(): number')
631enddef
632
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200633def Test_func_type_fails()
634 CheckDefFailure(['let ref1: func()'], 'E704:')
635
636 CheckDefFailure(['let Ref1: func()', 'Ref1 = FuncNoArgRetNumber'], 'E1013: type mismatch, expected func() but got func(): number')
637 CheckDefFailure(['let Ref1: func()', 'Ref1 = FuncOneArgNoRet'], 'E1013: type mismatch, expected func() but got func(number)')
638 CheckDefFailure(['let Ref1: func()', 'Ref1 = FuncOneArgRetNumber'], 'E1013: type mismatch, expected func() but got func(number): number')
Bram Moolenaar6ff71d82020-05-24 23:45:24 +0200639 CheckDefFailure(['let Ref1: func(bool)', 'Ref1 = FuncTwoArgNoRet'], 'E1013: type mismatch, expected func(bool) but got func(bool, number)')
640 CheckDefFailure(['let Ref1: func(?bool)', 'Ref1 = FuncTwoArgNoRet'], 'E1013: type mismatch, expected func(?bool) but got func(bool, number)')
641 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 +0200642
643 call CheckDefFailure(['let RefWrong: func(string ,number)'], 'E1068:')
644 call CheckDefFailure(['let RefWrong: func(string,number)'], 'E1069:')
645 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:')
646 call CheckDefFailure(['let RefWrong: func(bool):string'], 'E1069:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200647enddef
648
Bram Moolenaar89228602020-04-05 22:14:54 +0200649def Test_func_return_type()
650 let nr: number
651 nr = FuncNoArgRetNumber()
652 assert_equal(1234, nr)
653
654 nr = FuncOneArgRetAny(122)
655 assert_equal(122, nr)
656
657 let str: string
658 str = FuncOneArgRetAny('yes')
659 assert_equal('yes', str)
660
661 CheckDefFailure(['let str: string', 'str = FuncNoArgRetNumber()'], 'E1013: type mismatch, expected string but got number')
662enddef
663
Bram Moolenaar5e774c72020-04-12 21:53:00 +0200664def MultiLine(
665 arg1: string,
666 arg2 = 1234,
667 ...rest: list<string>
668 ): string
669 return arg1 .. arg2 .. join(rest, '-')
670enddef
671
Bram Moolenaar2c330432020-04-13 14:41:35 +0200672def MultiLineComment(
673 arg1: string, # comment
674 arg2 = 1234, # comment
675 ...rest: list<string> # comment
676 ): string # comment
677 return arg1 .. arg2 .. join(rest, '-')
678enddef
679
Bram Moolenaar5e774c72020-04-12 21:53:00 +0200680def Test_multiline()
681 assert_equal('text1234', MultiLine('text'))
682 assert_equal('text777', MultiLine('text', 777))
683 assert_equal('text777one', MultiLine('text', 777, 'one'))
684 assert_equal('text777one-two', MultiLine('text', 777, 'one', 'two'))
685enddef
686
Bram Moolenaar23e03252020-04-12 22:22:31 +0200687func Test_multiline_not_vim9()
688 call assert_equal('text1234', MultiLine('text'))
689 call assert_equal('text777', MultiLine('text', 777))
690 call assert_equal('text777one', MultiLine('text', 777, 'one'))
691 call assert_equal('text777one-two', MultiLine('text', 777, 'one', 'two'))
692endfunc
693
Bram Moolenaar5e774c72020-04-12 21:53:00 +0200694
Bram Moolenaaree4e0c12020-04-06 21:35:05 +0200695" When using CheckScriptFailure() for the below test, E1010 is generated instead
696" of E1056.
697func Test_E1056_1059()
698 let caught_1056 = 0
699 try
700 def F():
701 return 1
702 enddef
703 catch /E1056:/
704 let caught_1056 = 1
705 endtry
706 call assert_equal(1, caught_1056)
707
708 let caught_1059 = 0
709 try
710 def F5(items : list)
711 echo 'a'
712 enddef
713 catch /E1059:/
714 let caught_1059 = 1
715 endtry
716 call assert_equal(1, caught_1059)
717endfunc
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200718
Bram Moolenaar015f4262020-05-05 21:25:22 +0200719func DelMe()
720 echo 'DelMe'
721endfunc
722
723def Test_deleted_function()
724 CheckDefExecFailure([
725 'let RefMe: func = function("g:DelMe")',
726 'delfunc g:DelMe',
727 'echo RefMe()'], 'E117:')
728enddef
729
730def Test_unknown_function()
731 CheckDefExecFailure([
732 'let Ref: func = function("NotExist")',
Bram Moolenaarf3915862020-05-07 14:37:19 +0200733 'delfunc g:NotExist'], 'E130:')
Bram Moolenaar015f4262020-05-05 21:25:22 +0200734enddef
735
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +0200736def RefFunc(Ref: func(string): string): string
737 return Ref('more')
738enddef
739
740def Test_closure_simple()
741 let local = 'some '
742 assert_equal('some more', RefFunc({s -> local .. s}))
743enddef
744
Bram Moolenaarbf67ea12020-05-02 17:52:42 +0200745def MakeRef()
746 let local = 'some '
747 g:Ref = {s -> local .. s}
748enddef
749
750def Test_closure_ref_after_return()
751 MakeRef()
752 assert_equal('some thing', g:Ref('thing'))
753 unlet g:Ref
754enddef
755
Bram Moolenaar5adc55c2020-05-02 23:12:58 +0200756def MakeTwoRefs()
757 let local = ['some']
758 g:Extend = {s -> local->add(s)}
759 g:Read = {-> local}
760enddef
761
762def Test_closure_two_refs()
763 MakeTwoRefs()
764 assert_equal('some', join(g:Read(), ' '))
765 g:Extend('more')
766 assert_equal('some more', join(g:Read(), ' '))
767 g:Extend('even')
768 assert_equal('some more even', join(g:Read(), ' '))
769
770 unlet g:Extend
771 unlet g:Read
772enddef
773
Bram Moolenaar5adc55c2020-05-02 23:12:58 +0200774def ReadRef(Ref: func(): list<string>): string
775 return join(Ref(), ' ')
776enddef
777
778def ExtendRef(Ref: func(string), add: string)
779 Ref(add)
780enddef
781
782def Test_closure_two_indirect_refs()
Bram Moolenaarf7779c62020-05-03 15:38:16 +0200783 MakeTwoRefs()
Bram Moolenaar5adc55c2020-05-02 23:12:58 +0200784 assert_equal('some', ReadRef(g:Read))
785 ExtendRef(g:Extend, 'more')
786 assert_equal('some more', ReadRef(g:Read))
787 ExtendRef(g:Extend, 'even')
788 assert_equal('some more even', ReadRef(g:Read))
789
790 unlet g:Extend
791 unlet g:Read
792enddef
Bram Moolenaarbf67ea12020-05-02 17:52:42 +0200793
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +0200794def MakeArgRefs(theArg: string)
795 let local = 'loc_val'
796 g:UseArg = {s -> theArg .. '/' .. local .. '/' .. s}
797enddef
798
799def MakeArgRefsVarargs(theArg: string, ...rest: list<string>)
800 let local = 'the_loc'
801 g:UseVararg = {s -> theArg .. '/' .. local .. '/' .. s .. '/' .. join(rest)}
802enddef
803
804def Test_closure_using_argument()
805 MakeArgRefs('arg_val')
806 assert_equal('arg_val/loc_val/call_val', g:UseArg('call_val'))
807
808 MakeArgRefsVarargs('arg_val', 'one', 'two')
809 assert_equal('arg_val/the_loc/call_val/one two', g:UseVararg('call_val'))
810
811 unlet g:UseArg
812 unlet g:UseVararg
813enddef
814
Bram Moolenaarb68b3462020-05-06 21:06:30 +0200815def MakeGetAndAppendRefs()
816 let local = 'a'
817
818 def Append(arg: string)
819 local ..= arg
820 enddef
821 g:Append = Append
822
823 def Get(): string
824 return local
825 enddef
826 g:Get = Get
827enddef
828
829def Test_closure_append_get()
830 MakeGetAndAppendRefs()
831 assert_equal('a', g:Get())
832 g:Append('-b')
833 assert_equal('a-b', g:Get())
834 g:Append('-c')
835 assert_equal('a-b-c', g:Get())
836
837 unlet g:Append
838 unlet g:Get
839enddef
840
Bram Moolenaar04b12692020-05-04 23:24:44 +0200841def Test_nested_closure()
842 let local = 'text'
843 def Closure(arg: string): string
844 return local .. arg
845 enddef
846 assert_equal('text!!!', Closure('!!!'))
847enddef
848
Bram Moolenaar6f5b6df2020-05-16 21:20:12 +0200849func GetResult(Ref)
850 return a:Ref('some')
851endfunc
852
853def Test_call_closure_not_compiled()
854 let text = 'text'
855 g:Ref = {s -> s .. text}
856 assert_equal('sometext', GetResult(g:Ref))
857enddef
858
Bram Moolenaar865af6b2020-06-18 18:45:49 +0200859def Test_sort_return_type()
860 let res: list<number>
861 res = [1, 2, 3]->sort()
862enddef
863
Bram Moolenaarf151ad12020-06-30 13:38:01 +0200864def Test_getqflist_return_type()
865 let l = getqflist()
866 assert_equal([], l)
867
868 let d = getqflist(#{items: 0})
869 assert_equal(#{items: []}, d)
870enddef
871
872def Test_getloclist_return_type()
873 let l = getloclist(1)
874 assert_equal([], l)
875
876 let d = getloclist(1, #{items: 0})
877 assert_equal(#{items: []}, d)
878enddef
879
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +0200880def Line_continuation_in_def(dir: string = ''): string
881 let path: string = empty(dir)
882 \ ? 'empty'
883 \ : 'full'
884 return path
885enddef
886
887def Test_line_continuation_in_def()
888 assert_equal('full', Line_continuation_in_def('.'))
889enddef
890
Bram Moolenaarf7779c62020-05-03 15:38:16 +0200891
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200892" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker