blob: ee97e56ea682183215d5f00046b45fa5a9011210 [file] [log] [blame]
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001" Test various aspects of the Vim9 script language.
2
3source check.vim
Bram Moolenaarad304702020-09-06 18:22:53 +02004source term_util.vim
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02005source view_util.vim
Bram Moolenaar04b12692020-05-04 23:24:44 +02006source vim9.vim
Bram Moolenaar47e7d702020-07-05 18:18:42 +02007source screendump.vim
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02008
9func Test_def_basic()
10 def SomeFunc(): string
11 return 'yes'
12 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +020013 call SomeFunc()->assert_equal('yes')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +020014endfunc
15
Bram Moolenaar2b9b17e2020-10-13 18:38:11 +020016func Test_compiling_error()
17 " use a terminal to see the whole error message
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +020018 CheckRunVimInTerminal
19
Bram Moolenaar2b9b17e2020-10-13 18:38:11 +020020 call TestCompilingError()
Bram Moolenaare8c46602021-04-05 22:27:37 +020021 call TestCompilingErrorInTry()
Bram Moolenaar2b9b17e2020-10-13 18:38:11 +020022endfunc
23
24def TestCompilingError()
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +020025 var lines =<< trim END
26 vim9script
27 def Fails()
28 echo nothing
29 enddef
30 defcompile
31 END
Bram Moolenaare8c46602021-04-05 22:27:37 +020032 writefile(lines, 'XTest_compile_error')
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +020033 var buf = RunVimInTerminal('-S XTest_compile_error',
Bram Moolenaare0de1712020-12-02 17:36:54 +010034 {rows: 10, wait_for_ruler: 0})
Bram Moolenaare8c46602021-04-05 22:27:37 +020035 WaitForAssert(() => assert_match('Error detected while compiling command line.*Fails.*Variable not found: nothing',
Bram Moolenaar03dfde22021-02-14 13:17:22 +010036 Term_getlines(buf, range(1, 9))))
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +020037
38 # clean up
Bram Moolenaare8c46602021-04-05 22:27:37 +020039 StopVimInTerminal(buf)
40 delete('XTest_compile_error')
41enddef
42
43def TestCompilingErrorInTry()
44 var dir = 'Xdir/autoload'
45 mkdir(dir, 'p')
46
47 var lines =<< trim END
48 vim9script
49 def script#OnlyCompiled()
50 g:runtime = 'yes'
51 invalid
52 enddef
53 END
54 writefile(lines, dir .. '/script.vim')
55
56 lines =<< trim END
57 vim9script
58 todo
59 try
60 script#OnlyCompiled()
61 catch /nothing/
62 endtry
63 END
64 lines[1] = 'set rtp=' .. getcwd() .. '/Xdir'
65 writefile(lines, 'XTest_compile_error')
66
67 var buf = RunVimInTerminal('-S XTest_compile_error', {rows: 10, wait_for_ruler: 0})
68 WaitForAssert(() => assert_match('Error detected while compiling command line.*function script#OnlyCompiled.*Invalid command: invalid',
69 Term_getlines(buf, range(1, 9))))
70
71 # clean up
72 StopVimInTerminal(buf)
73 delete('XTest_compile_error')
74 delete('Xdir', 'rf')
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +020075enddef
76
Bram Moolenaarb55d6182021-06-08 22:01:53 +020077def Test_compile_error_in_called_function()
78 var lines =<< trim END
79 vim9script
80 var n: number
81 def Foo()
82 &hls = n
83 enddef
84 def Bar()
85 Foo()
86 enddef
87 silent! Foo()
88 Bar()
89 END
90 CheckScriptFailureList(lines, ['E1012:', 'E1191:'])
91enddef
92
Bram Moolenaar22f17a22021-06-21 20:48:58 +020093def Test_wrong_function_name()
94 var lines =<< trim END
95 vim9script
96 func _Foo()
97 echo 'foo'
98 endfunc
99 END
100 CheckScriptFailure(lines, 'E128:')
101
102 lines =<< trim END
103 vim9script
104 def _Foo()
105 echo 'foo'
106 enddef
107 END
108 CheckScriptFailure(lines, 'E128:')
109enddef
110
Bram Moolenaarf48b2fa2021-04-12 22:02:36 +0200111def Test_autoload_name_mismatch()
112 var dir = 'Xdir/autoload'
113 mkdir(dir, 'p')
114
115 var lines =<< trim END
116 vim9script
117 def scriptX#Function()
118 # comment
119 g:runtime = 'yes'
120 enddef
121 END
122 writefile(lines, dir .. '/script.vim')
123
124 var save_rtp = &rtp
125 exe 'set rtp=' .. getcwd() .. '/Xdir'
126 lines =<< trim END
127 call script#Function()
128 END
129 CheckScriptFailure(lines, 'E746:', 2)
130
131 &rtp = save_rtp
132 delete(dir, 'rf')
133enddef
134
Bram Moolenaarf0a40692021-06-11 22:05:47 +0200135def Test_autoload_names()
136 var dir = 'Xdir/autoload'
137 mkdir(dir, 'p')
138
139 var lines =<< trim END
140 func foobar#function()
141 return 'yes'
142 endfunc
143 let foobar#var = 'no'
144 END
145 writefile(lines, dir .. '/foobar.vim')
146
147 var save_rtp = &rtp
148 exe 'set rtp=' .. getcwd() .. '/Xdir'
149
150 lines =<< trim END
151 assert_equal('yes', foobar#function())
152 var Function = foobar#function
153 assert_equal('yes', Function())
154
155 assert_equal('no', foobar#var)
156 END
157 CheckDefAndScriptSuccess(lines)
158
159 &rtp = save_rtp
160 delete(dir, 'rf')
161enddef
162
Bram Moolenaar0ba48e82020-11-17 18:23:19 +0100163def CallRecursive(n: number): number
164 return CallRecursive(n + 1)
165enddef
166
167def CallMapRecursive(l: list<number>): number
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100168 return map(l, (_, v) => CallMapRecursive([v]))[0]
Bram Moolenaar0ba48e82020-11-17 18:23:19 +0100169enddef
170
171def Test_funcdepth_error()
172 set maxfuncdepth=10
173
174 var caught = false
175 try
176 CallRecursive(1)
177 catch /E132:/
178 caught = true
179 endtry
180 assert_true(caught)
181
182 caught = false
183 try
184 CallMapRecursive([1])
185 catch /E132:/
186 caught = true
187 endtry
188 assert_true(caught)
189
190 set maxfuncdepth&
191enddef
192
Bram Moolenaar5178b1b2021-01-01 18:43:51 +0100193def Test_endfunc_enddef()
194 var lines =<< trim END
195 def Test()
196 echo 'test'
197 endfunc
198 enddef
199 END
200 CheckScriptFailure(lines, 'E1151:', 3)
201
202 lines =<< trim END
203 def Test()
204 func Nested()
205 echo 'test'
206 enddef
207 enddef
208 END
209 CheckScriptFailure(lines, 'E1152:', 4)
Bram Moolenaar49f1e9e2021-03-22 20:49:02 +0100210
211 lines =<< trim END
212 def Ok()
213 echo 'hello'
214 enddef | echo 'there'
215 def Bad()
216 echo 'hello'
217 enddef there
218 END
219 CheckScriptFailure(lines, 'E1173: Text found after enddef: there', 6)
Bram Moolenaar5178b1b2021-01-01 18:43:51 +0100220enddef
221
Bram Moolenaarb8ba9b92021-01-01 18:54:34 +0100222def Test_missing_endfunc_enddef()
223 var lines =<< trim END
224 vim9script
225 def Test()
226 echo 'test'
227 endef
228 END
229 CheckScriptFailure(lines, 'E1057:', 2)
230
231 lines =<< trim END
232 vim9script
233 func Some()
234 echo 'test'
235 enfffunc
236 END
237 CheckScriptFailure(lines, 'E126:', 2)
238enddef
239
Bram Moolenaar4efd9942021-01-24 21:14:20 +0100240def Test_white_space_before_paren()
241 var lines =<< trim END
242 vim9script
243 def Test ()
244 echo 'test'
245 enddef
246 END
247 CheckScriptFailure(lines, 'E1068:', 2)
248
249 lines =<< trim END
250 vim9script
251 func Test ()
252 echo 'test'
253 endfunc
254 END
255 CheckScriptFailure(lines, 'E1068:', 2)
256
257 lines =<< trim END
258 def Test ()
259 echo 'test'
260 enddef
261 END
262 CheckScriptFailure(lines, 'E1068:', 1)
263
264 lines =<< trim END
265 func Test ()
266 echo 'test'
267 endfunc
268 END
269 CheckScriptSuccess(lines)
270enddef
271
Bram Moolenaar832ea892021-01-08 21:55:26 +0100272def Test_enddef_dict_key()
273 var d = {
274 enddef: 'x',
275 endfunc: 'y',
276 }
277 assert_equal({enddef: 'x', endfunc: 'y'}, d)
278enddef
279
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200280def ReturnString(): string
281 return 'string'
282enddef
283
284def ReturnNumber(): number
285 return 123
286enddef
287
288let g:notNumber = 'string'
289
290def ReturnGlobal(): number
291 return g:notNumber
292enddef
293
294def Test_return_something()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200295 ReturnString()->assert_equal('string')
296 ReturnNumber()->assert_equal(123)
Bram Moolenaar5e654232020-09-16 15:22:00 +0200297 assert_fails('ReturnGlobal()', 'E1012: Type mismatch; expected number but got string', '', 1, 'ReturnGlobal')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200298enddef
299
Bram Moolenaare32e5162021-01-21 20:21:29 +0100300def Test_check_argument_type()
301 var lines =<< trim END
302 vim9script
303 def Val(a: number, b: number): number
304 return 0
305 enddef
306 def Func()
307 var x: any = true
308 Val(0, x)
309 enddef
310 disass Func
311 Func()
312 END
313 CheckScriptFailure(lines, 'E1013: Argument 2: type mismatch, expected number but got bool', 2)
314enddef
315
Bram Moolenaarefd88552020-06-18 20:50:10 +0200316def Test_missing_return()
317 CheckDefFailure(['def Missing(): number',
318 ' if g:cond',
319 ' echo "no return"',
320 ' else',
321 ' return 0',
322 ' endif'
323 'enddef'], 'E1027:')
324 CheckDefFailure(['def Missing(): number',
325 ' if g:cond',
326 ' return 1',
327 ' else',
328 ' echo "no return"',
329 ' endif'
330 'enddef'], 'E1027:')
331 CheckDefFailure(['def Missing(): number',
332 ' if g:cond',
333 ' return 1',
334 ' else',
335 ' return 2',
336 ' endif'
337 ' return 3'
338 'enddef'], 'E1095:')
339enddef
340
Bram Moolenaar403dc312020-10-17 19:29:51 +0200341def Test_return_bool()
342 var lines =<< trim END
343 vim9script
344 def MenuFilter(id: number, key: string): bool
345 return popup_filter_menu(id, key)
346 enddef
347 def YesnoFilter(id: number, key: string): bool
348 return popup_filter_yesno(id, key)
349 enddef
350 defcompile
351 END
352 CheckScriptSuccess(lines)
353enddef
354
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200355let s:nothing = 0
356def ReturnNothing()
357 s:nothing = 1
358 if true
359 return
360 endif
361 s:nothing = 2
362enddef
363
364def Test_return_nothing()
365 ReturnNothing()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200366 s:nothing->assert_equal(1)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200367enddef
368
Bram Moolenaar648ea762021-01-15 19:04:32 +0100369def Test_return_invalid()
370 var lines =<< trim END
371 vim9script
372 def Func(): invalid
373 return xxx
374 enddef
375 defcompile
376 END
377 CheckScriptFailure(lines, 'E1010:', 2)
Bram Moolenaar31842cd2021-02-12 22:10:21 +0100378
379 lines =<< trim END
380 vim9script
381 def Test(Fun: func(number): number): list<number>
382 return map([1, 2, 3], (_, i) => Fun(i))
383 enddef
384 defcompile
385 def Inc(nr: number): nr
386 return nr + 2
387 enddef
388 echo Test(Inc)
389 END
390 # doing this twice was leaking memory
391 CheckScriptFailure(lines, 'E1010:')
392 CheckScriptFailure(lines, 'E1010:')
Bram Moolenaar648ea762021-01-15 19:04:32 +0100393enddef
394
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200395func Increment()
396 let g:counter += 1
397endfunc
398
399def Test_call_ufunc_count()
400 g:counter = 1
401 Increment()
402 Increment()
403 Increment()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200404 # works with and without :call
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200405 g:counter->assert_equal(4)
406 eval g:counter->assert_equal(4)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200407 unlet g:counter
408enddef
409
410def MyVarargs(arg: string, ...rest: list<string>): string
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200411 var res = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200412 for s in rest
413 res ..= ',' .. s
414 endfor
415 return res
416enddef
417
418def Test_call_varargs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200419 MyVarargs('one')->assert_equal('one')
420 MyVarargs('one', 'two')->assert_equal('one,two')
421 MyVarargs('one', 'two', 'three')->assert_equal('one,two,three')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200422enddef
423
424def MyDefaultArgs(name = 'string'): string
425 return name
426enddef
427
Bram Moolenaare30f64b2020-07-15 19:48:20 +0200428def MyDefaultSecond(name: string, second: bool = true): string
429 return second ? name : 'none'
430enddef
431
Bram Moolenaar38a3bfa2021-03-29 22:14:55 +0200432
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200433def Test_call_default_args()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200434 MyDefaultArgs()->assert_equal('string')
Bram Moolenaar38a3bfa2021-03-29 22:14:55 +0200435 MyDefaultArgs(v:none)->assert_equal('string')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200436 MyDefaultArgs('one')->assert_equal('one')
Bram Moolenaar38a3bfa2021-03-29 22:14:55 +0200437 assert_fails('MyDefaultArgs("one", "two")', 'E118:', '', 4, 'Test_call_default_args')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200438
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200439 MyDefaultSecond('test')->assert_equal('test')
440 MyDefaultSecond('test', true)->assert_equal('test')
441 MyDefaultSecond('test', false)->assert_equal('none')
Bram Moolenaare30f64b2020-07-15 19:48:20 +0200442
Bram Moolenaar38a3bfa2021-03-29 22:14:55 +0200443 var lines =<< trim END
444 def MyDefaultThird(name: string, aa = 'aa', bb = 'bb'): string
445 return name .. aa .. bb
446 enddef
447
448 MyDefaultThird('->')->assert_equal('->aabb')
449 MyDefaultThird('->', v:none)->assert_equal('->aabb')
450 MyDefaultThird('->', 'xx')->assert_equal('->xxbb')
451 MyDefaultThird('->', v:none, v:none)->assert_equal('->aabb')
452 MyDefaultThird('->', 'xx', v:none)->assert_equal('->xxbb')
453 MyDefaultThird('->', v:none, 'yy')->assert_equal('->aayy')
454 MyDefaultThird('->', 'xx', 'yy')->assert_equal('->xxyy')
455 END
456 CheckDefAndScriptSuccess(lines)
457
Bram Moolenaar822ba242020-05-24 23:00:18 +0200458 CheckScriptFailure(['def Func(arg: number = asdf)', 'enddef', 'defcompile'], 'E1001:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100459 delfunc g:Func
Bram Moolenaar77072282020-09-16 17:55:40 +0200460 CheckScriptFailure(['def Func(arg: number = "text")', 'enddef', 'defcompile'], 'E1013: Argument 1: type mismatch, expected number but got string')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100461 delfunc g:Func
Yegappan Lakshmanan34fcb692021-05-25 20:14:00 +0200462 CheckDefFailure(['def Func(x: number = )', 'enddef'], 'E15:')
Bram Moolenaar12bce952021-03-11 20:04:04 +0100463
Bram Moolenaar38a3bfa2021-03-29 22:14:55 +0200464 lines =<< trim END
Bram Moolenaar12bce952021-03-11 20:04:04 +0100465 vim9script
466 def Func(a = b == 0 ? 1 : 2, b = 0)
467 enddef
468 defcompile
469 END
470 CheckScriptFailure(lines, 'E1001: Variable not found: b')
Bram Moolenaar04b12692020-05-04 23:24:44 +0200471enddef
472
Bram Moolenaarcef12702021-01-04 14:09:43 +0100473def FuncWithComment( # comment
474 a: number, #comment
475 b: bool, # comment
476 c: string) #comment
477 assert_equal(4, a)
478 assert_equal(true, b)
479 assert_equal('yes', c)
480enddef
481
482def Test_func_with_comments()
483 FuncWithComment(4, true, 'yes')
484
485 var lines =<< trim END
486 def Func(# comment
487 arg: string)
488 enddef
489 END
490 CheckScriptFailure(lines, 'E125:', 1)
491
492 lines =<< trim END
493 def Func(
494 arg: string# comment
495 )
496 enddef
497 END
498 CheckScriptFailure(lines, 'E475:', 2)
499
500 lines =<< trim END
501 def Func(
502 arg: string
503 )# comment
504 enddef
505 END
506 CheckScriptFailure(lines, 'E488:', 3)
507enddef
508
Bram Moolenaar04b12692020-05-04 23:24:44 +0200509def Test_nested_function()
510 def Nested(arg: string): string
511 return 'nested ' .. arg
512 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200513 Nested('function')->assert_equal('nested function')
Bram Moolenaar04b12692020-05-04 23:24:44 +0200514
Bram Moolenaar0e65d3d2020-05-05 17:53:16 +0200515 CheckDefFailure(['def Nested()', 'enddef', 'Nested(66)'], 'E118:')
516 CheckDefFailure(['def Nested(arg: string)', 'enddef', 'Nested()'], 'E119:')
517
Bram Moolenaar04b12692020-05-04 23:24:44 +0200518 CheckDefFailure(['func Nested()', 'endfunc'], 'E1086:')
Bram Moolenaarbcbf4132020-08-01 22:35:13 +0200519 CheckDefFailure(['def s:Nested()', 'enddef'], 'E1075:')
520 CheckDefFailure(['def b:Nested()', 'enddef'], 'E1075:')
Bram Moolenaar8b848ca2020-09-10 22:28:01 +0200521
Bram Moolenaar54021752020-12-06 18:50:36 +0100522 var lines =<< trim END
523 def Outer()
524 def Inner()
525 # comment
526 enddef
527 def Inner()
528 enddef
529 enddef
530 END
531 CheckDefFailure(lines, 'E1073:')
532
533 lines =<< trim END
534 def Outer()
535 def Inner()
536 # comment
537 enddef
538 def! Inner()
539 enddef
540 enddef
541 END
542 CheckDefFailure(lines, 'E1117:')
543
544 # nested function inside conditional
Bram Moolenaar54021752020-12-06 18:50:36 +0100545 lines =<< trim END
546 vim9script
547 var thecount = 0
548 if true
549 def Test(): number
550 def TheFunc(): number
551 thecount += 1
552 return thecount
553 enddef
554 return TheFunc()
555 enddef
556 endif
557 defcompile
558 assert_equal(1, Test())
559 assert_equal(2, Test())
560 END
561 CheckScriptSuccess(lines)
Bram Moolenaar8863bda2021-03-17 18:42:08 +0100562
563 # also works when "thecount" is inside the "if" block
564 lines =<< trim END
565 vim9script
566 if true
567 var thecount = 0
568 def Test(): number
569 def TheFunc(): number
570 thecount += 1
571 return thecount
572 enddef
573 return TheFunc()
574 enddef
575 endif
576 defcompile
577 assert_equal(1, Test())
578 assert_equal(2, Test())
579 END
580 CheckScriptSuccess(lines)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200581enddef
582
Bram Moolenaaradc8e442020-12-31 18:28:18 +0100583def Test_not_nested_function()
584 echo printf('%d',
585 function('len')('xxx'))
586enddef
587
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200588func Test_call_default_args_from_func()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200589 call MyDefaultArgs()->assert_equal('string')
590 call MyDefaultArgs('one')->assert_equal('one')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200591 call assert_fails('call MyDefaultArgs("one", "two")', 'E118:', '', 3, 'Test_call_default_args_from_func')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200592endfunc
593
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200594def Test_nested_global_function()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200595 var lines =<< trim END
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200596 vim9script
597 def Outer()
598 def g:Inner(): string
599 return 'inner'
600 enddef
601 enddef
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200602 defcompile
603 Outer()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200604 g:Inner()->assert_equal('inner')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200605 delfunc g:Inner
606 Outer()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200607 g:Inner()->assert_equal('inner')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200608 delfunc g:Inner
609 Outer()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200610 g:Inner()->assert_equal('inner')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200611 delfunc g:Inner
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200612 END
613 CheckScriptSuccess(lines)
Bram Moolenaar2c79e9d2020-08-01 18:57:52 +0200614
615 lines =<< trim END
616 vim9script
617 def Outer()
618 def g:Inner(): string
619 return 'inner'
620 enddef
621 enddef
622 defcompile
623 Outer()
624 Outer()
625 END
626 CheckScriptFailure(lines, "E122:")
Bram Moolenaarcd45ed02020-12-22 17:35:54 +0100627 delfunc g:Inner
Bram Moolenaarad486a02020-08-01 23:22:18 +0200628
629 lines =<< trim END
630 vim9script
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100631 def Outer()
632 def g:Inner()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100633 echo map([1, 2, 3], (_, v) => v + 1)
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100634 enddef
635 g:Inner()
636 enddef
637 Outer()
638 END
639 CheckScriptSuccess(lines)
640 delfunc g:Inner
641
642 lines =<< trim END
643 vim9script
Bram Moolenaarad486a02020-08-01 23:22:18 +0200644 def Func()
645 echo 'script'
646 enddef
647 def Outer()
648 def Func()
649 echo 'inner'
650 enddef
651 enddef
652 defcompile
653 END
654 CheckScriptFailure(lines, "E1073:")
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200655enddef
656
Bram Moolenaar6abdcf82020-11-22 18:15:44 +0100657def DefListAll()
658 def
659enddef
660
661def DefListOne()
662 def DefListOne
663enddef
664
665def DefListMatches()
666 def /DefList
667enddef
668
669def Test_nested_def_list()
670 var funcs = split(execute('call DefListAll()'), "\n")
671 assert_true(len(funcs) > 10)
672 assert_true(funcs->index('def DefListAll()') >= 0)
673
674 funcs = split(execute('call DefListOne()'), "\n")
675 assert_equal([' def DefListOne()', '1 def DefListOne', ' enddef'], funcs)
676
677 funcs = split(execute('call DefListMatches()'), "\n")
678 assert_true(len(funcs) >= 3)
679 assert_true(funcs->index('def DefListAll()') >= 0)
680 assert_true(funcs->index('def DefListOne()') >= 0)
681 assert_true(funcs->index('def DefListMatches()') >= 0)
Bram Moolenaar54021752020-12-06 18:50:36 +0100682
683 var lines =<< trim END
684 vim9script
685 def Func()
686 def +Func+
687 enddef
688 defcompile
689 END
690 CheckScriptFailure(lines, 'E476:', 1)
Bram Moolenaar6abdcf82020-11-22 18:15:44 +0100691enddef
692
Bram Moolenaar333894b2020-08-01 18:53:07 +0200693def Test_global_local_function()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200694 var lines =<< trim END
Bram Moolenaar333894b2020-08-01 18:53:07 +0200695 vim9script
696 def g:Func(): string
697 return 'global'
698 enddef
699 def Func(): string
700 return 'local'
701 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200702 g:Func()->assert_equal('global')
703 Func()->assert_equal('local')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100704 delfunc g:Func
Bram Moolenaar333894b2020-08-01 18:53:07 +0200705 END
706 CheckScriptSuccess(lines)
Bram Moolenaar035d6e92020-08-11 22:30:42 +0200707
708 lines =<< trim END
709 vim9script
710 def g:Funcy()
711 echo 'funcy'
712 enddef
713 s:Funcy()
714 END
715 CheckScriptFailure(lines, 'E117:')
Bram Moolenaar333894b2020-08-01 18:53:07 +0200716enddef
717
Bram Moolenaar0f769812020-09-12 18:32:34 +0200718def Test_local_function_shadows_global()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200719 var lines =<< trim END
Bram Moolenaar0f769812020-09-12 18:32:34 +0200720 vim9script
721 def g:Gfunc(): string
722 return 'global'
723 enddef
724 def AnotherFunc(): number
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200725 var Gfunc = function('len')
Bram Moolenaar0f769812020-09-12 18:32:34 +0200726 return Gfunc('testing')
727 enddef
728 g:Gfunc()->assert_equal('global')
729 AnotherFunc()->assert_equal(7)
730 delfunc g:Gfunc
731 END
732 CheckScriptSuccess(lines)
733
734 lines =<< trim END
735 vim9script
736 def g:Func(): string
737 return 'global'
738 enddef
739 def AnotherFunc()
740 g:Func = function('len')
741 enddef
742 AnotherFunc()
743 END
744 CheckScriptFailure(lines, 'E705:')
745 delfunc g:Func
Bram Moolenaar0865b152021-04-05 15:38:51 +0200746
747 # global function is found without g: prefix
748 lines =<< trim END
749 vim9script
750 def g:Func(): string
751 return 'global'
752 enddef
753 def AnotherFunc(): string
754 return Func()
755 enddef
756 assert_equal('global', AnotherFunc())
757 delfunc g:Func
758 END
759 CheckScriptSuccess(lines)
760
761 lines =<< trim END
762 vim9script
763 def g:Func(): string
764 return 'global'
765 enddef
766 assert_equal('global', Func())
767 delfunc g:Func
768 END
769 CheckScriptSuccess(lines)
Bram Moolenaar0f769812020-09-12 18:32:34 +0200770enddef
771
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200772func TakesOneArg(arg)
773 echo a:arg
774endfunc
775
776def Test_call_wrong_args()
Bram Moolenaard2c61702020-09-06 15:58:36 +0200777 CheckDefFailure(['TakesOneArg()'], 'E119:')
778 CheckDefFailure(['TakesOneArg(11, 22)'], 'E118:')
779 CheckDefFailure(['bufnr(xxx)'], 'E1001:')
780 CheckScriptFailure(['def Func(Ref: func(s: string))'], 'E475:')
Bram Moolenaaree8580e2020-08-28 17:19:07 +0200781
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200782 var lines =<< trim END
Bram Moolenaaree8580e2020-08-28 17:19:07 +0200783 vim9script
784 def Func(s: string)
785 echo s
786 enddef
787 Func([])
788 END
Bram Moolenaar77072282020-09-16 17:55:40 +0200789 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected string but got list<unknown>', 5)
Bram Moolenaarb185a402020-09-18 22:42:00 +0200790
791 lines =<< trim END
792 vim9script
Bram Moolenaarb4893b82021-02-21 22:20:24 +0100793 var name = 'piet'
794 def FuncOne(name: string)
795 echo nr
796 enddef
797 END
Bram Moolenaar057e84a2021-02-28 16:55:11 +0100798 CheckScriptFailure(lines, 'E1168:')
Bram Moolenaarb4893b82021-02-21 22:20:24 +0100799
800 lines =<< trim END
801 vim9script
Bram Moolenaarb185a402020-09-18 22:42:00 +0200802 def FuncOne(nr: number)
803 echo nr
804 enddef
805 def FuncTwo()
806 FuncOne()
807 enddef
808 defcompile
809 END
810 writefile(lines, 'Xscript')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200811 var didCatch = false
Bram Moolenaarb185a402020-09-18 22:42:00 +0200812 try
813 source Xscript
814 catch
815 assert_match('E119: Not enough arguments for function: <SNR>\d\+_FuncOne', v:exception)
816 assert_match('Xscript\[8\]..function <SNR>\d\+_FuncTwo, line 1', v:throwpoint)
817 didCatch = true
818 endtry
819 assert_true(didCatch)
820
821 lines =<< trim END
822 vim9script
823 def FuncOne(nr: number)
824 echo nr
825 enddef
826 def FuncTwo()
827 FuncOne(1, 2)
828 enddef
829 defcompile
830 END
831 writefile(lines, 'Xscript')
832 didCatch = false
833 try
834 source Xscript
835 catch
836 assert_match('E118: Too many arguments for function: <SNR>\d\+_FuncOne', v:exception)
837 assert_match('Xscript\[8\]..function <SNR>\d\+_FuncTwo, line 1', v:throwpoint)
838 didCatch = true
839 endtry
840 assert_true(didCatch)
841
842 delete('Xscript')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200843enddef
844
Bram Moolenaar50824712020-12-20 21:10:17 +0100845def Test_call_funcref_wrong_args()
846 var head =<< trim END
847 vim9script
848 def Func3(a1: string, a2: number, a3: list<number>)
849 echo a1 .. a2 .. a3[0]
850 enddef
851 def Testme()
852 var funcMap: dict<func> = {func: Func3}
853 END
854 var tail =<< trim END
855 enddef
856 Testme()
857 END
858 CheckScriptSuccess(head + ["funcMap['func']('str', 123, [1, 2, 3])"] + tail)
859
860 CheckScriptFailure(head + ["funcMap['func']('str', 123)"] + tail, 'E119:')
861 CheckScriptFailure(head + ["funcMap['func']('str', 123, [1], 4)"] + tail, 'E118:')
Bram Moolenaar32b3f822021-01-06 21:59:39 +0100862
863 var lines =<< trim END
864 vim9script
865 var Ref: func(number): any
866 Ref = (j) => !j
867 echo Ref(false)
868 END
869 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected number but got bool', 4)
870
871 lines =<< trim END
872 vim9script
873 var Ref: func(number): any
874 Ref = (j) => !j
875 call Ref(false)
876 END
877 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected number but got bool', 4)
Bram Moolenaar50824712020-12-20 21:10:17 +0100878enddef
879
Bram Moolenaarb4d16cb2020-11-05 18:45:46 +0100880def Test_call_lambda_args()
Bram Moolenaar2a389082021-04-09 20:24:31 +0200881 var lines =<< trim END
882 var Callback = (..._) => 'anything'
883 assert_equal('anything', Callback())
884 assert_equal('anything', Callback(1))
885 assert_equal('anything', Callback('a', 2))
Bram Moolenaar1088b692021-04-09 22:12:44 +0200886
887 assert_equal('xyz', ((a: string): string => a)('xyz'))
Bram Moolenaar2a389082021-04-09 20:24:31 +0200888 END
889 CheckDefAndScriptSuccess(lines)
890
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100891 CheckDefFailure(['echo ((i) => 0)()'],
892 'E119: Not enough arguments for function: ((i) => 0)()')
Bram Moolenaarb4d16cb2020-11-05 18:45:46 +0100893
Bram Moolenaar2a389082021-04-09 20:24:31 +0200894 lines =<< trim END
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100895 var Ref = (x: number, y: number) => x + y
Bram Moolenaarb4d16cb2020-11-05 18:45:46 +0100896 echo Ref(1, 'x')
897 END
898 CheckDefFailure(lines, 'E1013: Argument 2: type mismatch, expected number but got string')
Bram Moolenaare68b02a2021-01-03 13:09:51 +0100899
900 lines =<< trim END
901 var Ref: func(job, string, number)
902 Ref = (x, y) => 0
903 END
904 CheckDefAndScriptFailure(lines, 'E1012:')
905
906 lines =<< trim END
907 var Ref: func(job, string)
908 Ref = (x, y, z) => 0
909 END
910 CheckDefAndScriptFailure(lines, 'E1012:')
Bram Moolenaar057e84a2021-02-28 16:55:11 +0100911
912 lines =<< trim END
913 var one = 1
914 var l = [1, 2, 3]
915 echo map(l, (one) => one)
916 END
917 CheckDefFailure(lines, 'E1167:')
918 CheckScriptFailure(['vim9script'] + lines, 'E1168:')
919
920 lines =<< trim END
Bram Moolenaar14ded112021-06-26 19:25:49 +0200921 var Ref: func(any, ?any): bool
922 Ref = (_, y = 1) => false
923 END
924 CheckDefAndScriptFailure(lines, 'E1172:')
925
926 lines =<< trim END
Bram Moolenaar057e84a2021-02-28 16:55:11 +0100927 def ShadowLocal()
928 var one = 1
929 var l = [1, 2, 3]
930 echo map(l, (one) => one)
931 enddef
932 END
933 CheckDefFailure(lines, 'E1167:')
934
935 lines =<< trim END
936 def Shadowarg(one: number)
937 var l = [1, 2, 3]
938 echo map(l, (one) => one)
939 enddef
940 END
941 CheckDefFailure(lines, 'E1167:')
Bram Moolenaar767034c2021-04-09 17:24:52 +0200942
943 lines =<< trim END
944 echo ((a) => a)('aa', 'bb')
945 END
946 CheckDefAndScriptFailure(lines, 'E118:', 1)
Bram Moolenaarb4d16cb2020-11-05 18:45:46 +0100947enddef
948
Bram Moolenaar5f91e742021-03-17 21:29:29 +0100949def FilterWithCond(x: string, Cond: func(string): bool): bool
950 return Cond(x)
951enddef
952
Bram Moolenaar0346b792021-01-31 22:18:29 +0100953def Test_lambda_return_type()
954 var lines =<< trim END
955 var Ref = (): => 123
956 END
957 CheckDefAndScriptFailure(lines, 'E1157:', 1)
Bram Moolenaar5f91e742021-03-17 21:29:29 +0100958
Yegappan Lakshmanan611728f2021-05-24 15:15:47 +0200959 # no space before the return type
960 lines =<< trim END
961 var Ref = (x):number => x + 1
962 END
963 CheckDefAndScriptFailure(lines, 'E1069:', 1)
964
Bram Moolenaar5f91e742021-03-17 21:29:29 +0100965 # this works
966 for x in ['foo', 'boo']
967 echo FilterWithCond(x, (v) => v =~ '^b')
968 endfor
969
970 # this fails
971 lines =<< trim END
972 echo FilterWithCond('foo', (v) => v .. '^b')
973 END
974 CheckDefAndScriptFailure(lines, 'E1013: Argument 2: type mismatch, expected func(string): bool but got func(any): string', 1)
Bram Moolenaara9931532021-06-12 15:58:16 +0200975
976 lines =<< trim END
977 var Lambda1 = (x) => {
978 return x
979 }
980 assert_equal('asdf', Lambda1('asdf'))
981 var Lambda2 = (x): string => {
982 return x
983 }
984 assert_equal('foo', Lambda2('foo'))
985 END
986 CheckDefAndScriptSuccess(lines)
987
988 lines =<< trim END
989 var Lambda = (x): string => {
990 return x
991 }
992 echo Lambda(['foo'])
993 END
994 CheckDefExecAndScriptFailure(lines, 'E1012:')
Bram Moolenaar0346b792021-01-31 22:18:29 +0100995enddef
996
Bram Moolenaar709664c2020-12-12 14:33:41 +0100997def Test_lambda_uses_assigned_var()
998 CheckDefSuccess([
999 'var x: any = "aaa"'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001000 'x = filter(["bbb"], (_, v) => v =~ x)'])
Bram Moolenaar709664c2020-12-12 14:33:41 +01001001enddef
1002
Bram Moolenaar18062fc2021-03-05 21:35:47 +01001003def Test_pass_legacy_lambda_to_def_func()
1004 var lines =<< trim END
1005 vim9script
1006 func Foo()
1007 eval s:Bar({x -> 0})
1008 endfunc
1009 def Bar(y: any)
1010 enddef
1011 Foo()
1012 END
1013 CheckScriptSuccess(lines)
Bram Moolenaar831bdf82021-06-22 19:32:17 +02001014
1015 lines =<< trim END
1016 vim9script
1017 def g:TestFunc(f: func())
1018 enddef
1019 legacy call g:TestFunc({-> 0})
1020 delfunc g:TestFunc
1021
1022 def g:TestFunc(f: func(number))
1023 enddef
1024 legacy call g:TestFunc({nr -> 0})
1025 delfunc g:TestFunc
1026 END
1027 CheckScriptSuccess(lines)
Bram Moolenaar18062fc2021-03-05 21:35:47 +01001028enddef
1029
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001030" Default arg and varargs
1031def MyDefVarargs(one: string, two = 'foo', ...rest: list<string>): string
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001032 var res = one .. ',' .. two
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001033 for s in rest
1034 res ..= ',' .. s
1035 endfor
1036 return res
1037enddef
1038
1039def Test_call_def_varargs()
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001040 assert_fails('MyDefVarargs()', 'E119:', '', 1, 'Test_call_def_varargs')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001041 MyDefVarargs('one')->assert_equal('one,foo')
1042 MyDefVarargs('one', 'two')->assert_equal('one,two')
1043 MyDefVarargs('one', 'two', 'three')->assert_equal('one,two,three')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +02001044 CheckDefFailure(['MyDefVarargs("one", 22)'],
Bram Moolenaar77072282020-09-16 17:55:40 +02001045 'E1013: Argument 2: type mismatch, expected string but got number')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +02001046 CheckDefFailure(['MyDefVarargs("one", "two", 123)'],
Bram Moolenaar77072282020-09-16 17:55:40 +02001047 'E1013: Argument 3: type mismatch, expected string but got number')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +02001048
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001049 var lines =<< trim END
Bram Moolenaar24aa48b2020-07-25 16:33:02 +02001050 vim9script
1051 def Func(...l: list<string>)
1052 echo l
1053 enddef
1054 Func('a', 'b', 'c')
1055 END
1056 CheckScriptSuccess(lines)
1057
1058 lines =<< trim END
1059 vim9script
1060 def Func(...l: list<string>)
1061 echo l
1062 enddef
1063 Func()
1064 END
1065 CheckScriptSuccess(lines)
1066
1067 lines =<< trim END
1068 vim9script
Bram Moolenaar2a389082021-04-09 20:24:31 +02001069 def Func(...l: list<any>)
Bram Moolenaar2f8cbc42020-09-16 17:22:59 +02001070 echo l
1071 enddef
1072 Func(0)
1073 END
1074 CheckScriptSuccess(lines)
1075
1076 lines =<< trim END
1077 vim9script
Bram Moolenaar2a389082021-04-09 20:24:31 +02001078 def Func(...l: any)
1079 echo l
1080 enddef
1081 Func(0)
1082 END
1083 CheckScriptFailure(lines, 'E1180:', 2)
1084
1085 lines =<< trim END
1086 vim9script
Bram Moolenaar28022722020-09-21 22:02:49 +02001087 def Func(..._l: list<string>)
1088 echo _l
1089 enddef
1090 Func('a', 'b', 'c')
1091 END
1092 CheckScriptSuccess(lines)
1093
1094 lines =<< trim END
1095 vim9script
Bram Moolenaar24aa48b2020-07-25 16:33:02 +02001096 def Func(...l: list<string>)
1097 echo l
1098 enddef
1099 Func(1, 2, 3)
1100 END
Bram Moolenaar77072282020-09-16 17:55:40 +02001101 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +02001102
1103 lines =<< trim END
1104 vim9script
1105 def Func(...l: list<string>)
1106 echo l
1107 enddef
1108 Func('a', 9)
1109 END
Bram Moolenaar77072282020-09-16 17:55:40 +02001110 CheckScriptFailure(lines, 'E1013: Argument 2: type mismatch')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +02001111
1112 lines =<< trim END
1113 vim9script
1114 def Func(...l: list<string>)
1115 echo l
1116 enddef
1117 Func(1, 'a')
1118 END
Bram Moolenaar77072282020-09-16 17:55:40 +02001119 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch')
Bram Moolenaar4f53b792021-02-07 15:59:49 +01001120
1121 lines =<< trim END
1122 vim9script
1123 def Func( # some comment
1124 ...l = []
1125 )
1126 echo l
1127 enddef
1128 END
1129 CheckScriptFailure(lines, 'E1160:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001130enddef
1131
Bram Moolenaar1378fbc2020-04-11 20:50:33 +02001132let s:value = ''
1133
1134def FuncOneDefArg(opt = 'text')
1135 s:value = opt
1136enddef
1137
1138def FuncTwoDefArg(nr = 123, opt = 'text'): string
1139 return nr .. opt
1140enddef
1141
1142def FuncVarargs(...arg: list<string>): string
1143 return join(arg, ',')
1144enddef
1145
1146def Test_func_type_varargs()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001147 var RefDefArg: func(?string)
Bram Moolenaar1378fbc2020-04-11 20:50:33 +02001148 RefDefArg = FuncOneDefArg
1149 RefDefArg()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001150 s:value->assert_equal('text')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +02001151 RefDefArg('some')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001152 s:value->assert_equal('some')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +02001153
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001154 var RefDef2Arg: func(?number, ?string): string
Bram Moolenaar1378fbc2020-04-11 20:50:33 +02001155 RefDef2Arg = FuncTwoDefArg
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001156 RefDef2Arg()->assert_equal('123text')
1157 RefDef2Arg(99)->assert_equal('99text')
1158 RefDef2Arg(77, 'some')->assert_equal('77some')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +02001159
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001160 CheckDefFailure(['var RefWrong: func(string?)'], 'E1010:')
1161 CheckDefFailure(['var RefWrong: func(?string, string)'], 'E1007:')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +02001162
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001163 var RefVarargs: func(...list<string>): string
Bram Moolenaar1378fbc2020-04-11 20:50:33 +02001164 RefVarargs = FuncVarargs
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001165 RefVarargs()->assert_equal('')
1166 RefVarargs('one')->assert_equal('one')
1167 RefVarargs('one', 'two')->assert_equal('one,two')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +02001168
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001169 CheckDefFailure(['var RefWrong: func(...list<string>, string)'], 'E110:')
1170 CheckDefFailure(['var RefWrong: func(...list<string>, ?string)'], 'E110:')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +02001171enddef
1172
Bram Moolenaar0b76b422020-04-07 22:05:08 +02001173" Only varargs
1174def MyVarargsOnly(...args: list<string>): string
1175 return join(args, ',')
1176enddef
1177
1178def Test_call_varargs_only()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001179 MyVarargsOnly()->assert_equal('')
1180 MyVarargsOnly('one')->assert_equal('one')
1181 MyVarargsOnly('one', 'two')->assert_equal('one,two')
Bram Moolenaar77072282020-09-16 17:55:40 +02001182 CheckDefFailure(['MyVarargsOnly(1)'], 'E1013: Argument 1: type mismatch, expected string but got number')
1183 CheckDefFailure(['MyVarargsOnly("one", 2)'], 'E1013: Argument 2: type mismatch, expected string but got number')
Bram Moolenaar0b76b422020-04-07 22:05:08 +02001184enddef
1185
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001186def Test_using_var_as_arg()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001187 writefile(['def Func(x: number)', 'var x = 234', 'enddef', 'defcompile'], 'Xdef')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001188 assert_fails('so Xdef', 'E1006:', '', 1, 'Func')
Bram Moolenaard2c61702020-09-06 15:58:36 +02001189 delete('Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001190enddef
1191
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +02001192def DictArg(arg: dict<string>)
1193 arg['key'] = 'value'
1194enddef
1195
1196def ListArg(arg: list<string>)
1197 arg[0] = 'value'
1198enddef
1199
1200def Test_assign_to_argument()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +02001201 # works for dict and list
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001202 var d: dict<string> = {}
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +02001203 DictArg(d)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001204 d['key']->assert_equal('value')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001205 var l: list<string> = []
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +02001206 ListArg(l)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001207 l[0]->assert_equal('value')
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +02001208
Bram Moolenaard2c61702020-09-06 15:58:36 +02001209 CheckScriptFailure(['def Func(arg: number)', 'arg = 3', 'enddef', 'defcompile'], 'E1090:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001210 delfunc! g:Func
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +02001211enddef
1212
Bram Moolenaarb816dae2020-09-20 22:04:00 +02001213" These argument names are reserved in legacy functions.
1214def WithReservedNames(firstline: string, lastline: string): string
1215 return firstline .. lastline
1216enddef
1217
1218def Test_argument_names()
1219 assert_equal('OK', WithReservedNames('O', 'K'))
1220enddef
1221
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001222def Test_call_func_defined_later()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001223 g:DefinedLater('one')->assert_equal('one')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001224 assert_fails('NotDefined("one")', 'E117:', '', 2, 'Test_call_func_defined_later')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001225enddef
1226
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +02001227func DefinedLater(arg)
1228 return a:arg
1229endfunc
1230
1231def Test_call_funcref()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001232 g:SomeFunc('abc')->assert_equal(3)
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001233 assert_fails('NotAFunc()', 'E117:', '', 2, 'Test_call_funcref') # comment after call
1234 assert_fails('g:NotAFunc()', 'E117:', '', 3, 'Test_call_funcref')
Bram Moolenaar2f1980f2020-07-22 19:30:06 +02001235
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001236 var lines =<< trim END
Bram Moolenaar2f1980f2020-07-22 19:30:06 +02001237 vim9script
1238 def RetNumber(): number
1239 return 123
1240 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001241 var Funcref: func: number = function('RetNumber')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001242 Funcref()->assert_equal(123)
Bram Moolenaar2f1980f2020-07-22 19:30:06 +02001243 END
1244 CheckScriptSuccess(lines)
Bram Moolenaar0f60e802020-07-22 20:16:11 +02001245
1246 lines =<< trim END
1247 vim9script
1248 def RetNumber(): number
1249 return 123
1250 enddef
1251 def Bar(F: func: number): number
1252 return F()
1253 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001254 var Funcref = function('RetNumber')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001255 Bar(Funcref)->assert_equal(123)
Bram Moolenaar0f60e802020-07-22 20:16:11 +02001256 END
1257 CheckScriptSuccess(lines)
Bram Moolenaarbfba8652020-07-23 20:09:10 +02001258
1259 lines =<< trim END
1260 vim9script
1261 def UseNumber(nr: number)
1262 echo nr
1263 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001264 var Funcref: func(number) = function('UseNumber')
Bram Moolenaarbfba8652020-07-23 20:09:10 +02001265 Funcref(123)
1266 END
1267 CheckScriptSuccess(lines)
Bram Moolenaarb8070e32020-07-23 20:56:04 +02001268
1269 lines =<< trim END
1270 vim9script
1271 def UseNumber(nr: number)
1272 echo nr
1273 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001274 var Funcref: func(string) = function('UseNumber')
Bram Moolenaarb8070e32020-07-23 20:56:04 +02001275 END
Bram Moolenaar5e654232020-09-16 15:22:00 +02001276 CheckScriptFailure(lines, 'E1012: Type mismatch; expected func(string) but got func(number)')
Bram Moolenaar4fc224c2020-07-26 17:56:25 +02001277
1278 lines =<< trim END
1279 vim9script
1280 def EchoNr(nr = 34)
1281 g:echo = nr
1282 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001283 var Funcref: func(?number) = function('EchoNr')
Bram Moolenaar4fc224c2020-07-26 17:56:25 +02001284 Funcref()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001285 g:echo->assert_equal(34)
Bram Moolenaar4fc224c2020-07-26 17:56:25 +02001286 Funcref(123)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001287 g:echo->assert_equal(123)
Bram Moolenaar4fc224c2020-07-26 17:56:25 +02001288 END
1289 CheckScriptSuccess(lines)
Bram Moolenaarace61322020-07-26 18:16:58 +02001290
1291 lines =<< trim END
1292 vim9script
1293 def EchoList(...l: list<number>)
1294 g:echo = l
1295 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001296 var Funcref: func(...list<number>) = function('EchoList')
Bram Moolenaarace61322020-07-26 18:16:58 +02001297 Funcref()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001298 g:echo->assert_equal([])
Bram Moolenaarace61322020-07-26 18:16:58 +02001299 Funcref(1, 2, 3)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001300 g:echo->assert_equal([1, 2, 3])
Bram Moolenaarace61322020-07-26 18:16:58 +02001301 END
1302 CheckScriptSuccess(lines)
Bram Moolenaar01865ad2020-07-26 18:33:09 +02001303
1304 lines =<< trim END
1305 vim9script
1306 def OptAndVar(nr: number, opt = 12, ...l: list<number>): number
1307 g:optarg = opt
1308 g:listarg = l
1309 return nr
1310 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001311 var Funcref: func(number, ?number, ...list<number>): number = function('OptAndVar')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001312 Funcref(10)->assert_equal(10)
1313 g:optarg->assert_equal(12)
1314 g:listarg->assert_equal([])
Bram Moolenaar01865ad2020-07-26 18:33:09 +02001315
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001316 Funcref(11, 22)->assert_equal(11)
1317 g:optarg->assert_equal(22)
1318 g:listarg->assert_equal([])
Bram Moolenaar01865ad2020-07-26 18:33:09 +02001319
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001320 Funcref(17, 18, 1, 2, 3)->assert_equal(17)
1321 g:optarg->assert_equal(18)
1322 g:listarg->assert_equal([1, 2, 3])
Bram Moolenaar01865ad2020-07-26 18:33:09 +02001323 END
1324 CheckScriptSuccess(lines)
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +02001325enddef
1326
1327let SomeFunc = function('len')
1328let NotAFunc = 'text'
1329
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +02001330def CombineFuncrefTypes()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +02001331 # same arguments, different return type
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001332 var Ref1: func(bool): string
1333 var Ref2: func(bool): number
1334 var Ref3: func(bool): any
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +02001335 Ref3 = g:cond ? Ref1 : Ref2
1336
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +02001337 # different number of arguments
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001338 var Refa1: func(bool): number
1339 var Refa2: func(bool, number): number
1340 var Refa3: func: number
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +02001341 Refa3 = g:cond ? Refa1 : Refa2
1342
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +02001343 # different argument types
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001344 var Refb1: func(bool, string): number
1345 var Refb2: func(string, number): number
1346 var Refb3: func(any, any): number
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +02001347 Refb3 = g:cond ? Refb1 : Refb2
1348enddef
1349
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001350def FuncWithForwardCall()
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +02001351 return g:DefinedEvenLater("yes")
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001352enddef
1353
1354def DefinedEvenLater(arg: string): string
1355 return arg
1356enddef
1357
1358def Test_error_in_nested_function()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +02001359 # Error in called function requires unwinding the call stack.
Bram Moolenaar44d66522020-09-06 22:26:57 +02001360 assert_fails('FuncWithForwardCall()', 'E1096:', '', 1, 'FuncWithForwardCall')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001361enddef
1362
1363def Test_return_type_wrong()
Bram Moolenaar5a849da2020-08-08 16:47:30 +02001364 CheckScriptFailure([
1365 'def Func(): number',
1366 'return "a"',
1367 'enddef',
1368 'defcompile'], 'expected number but got string')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001369 delfunc! g:Func
Bram Moolenaar5a849da2020-08-08 16:47:30 +02001370 CheckScriptFailure([
1371 'def Func(): string',
1372 'return 1',
1373 'enddef',
1374 'defcompile'], 'expected string but got number')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001375 delfunc! g:Func
Bram Moolenaar5a849da2020-08-08 16:47:30 +02001376 CheckScriptFailure([
1377 'def Func(): void',
1378 'return "a"',
1379 'enddef',
1380 'defcompile'],
1381 'E1096: Returning a value in a function without a return type')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001382 delfunc! g:Func
Bram Moolenaar5a849da2020-08-08 16:47:30 +02001383 CheckScriptFailure([
1384 'def Func()',
1385 'return "a"',
1386 'enddef',
1387 'defcompile'],
1388 'E1096: Returning a value in a function without a return type')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001389 delfunc! g:Func
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001390
Bram Moolenaar5a849da2020-08-08 16:47:30 +02001391 CheckScriptFailure([
1392 'def Func(): number',
1393 'return',
1394 'enddef',
1395 'defcompile'], 'E1003:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001396 delfunc! g:Func
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001397
1398 CheckScriptFailure(['def Func(): list', 'return []', 'enddef'], 'E1008:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001399 delfunc! g:Func
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001400 CheckScriptFailure(['def Func(): dict', 'return {}', 'enddef'], 'E1008:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001401 delfunc! g:Func
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02001402 CheckScriptFailure(['def Func()', 'return 1'], 'E1057:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001403 delfunc! g:Func
Bram Moolenaar5a849da2020-08-08 16:47:30 +02001404
1405 CheckScriptFailure([
1406 'vim9script',
1407 'def FuncB()',
1408 ' return 123',
1409 'enddef',
1410 'def FuncA()',
1411 ' FuncB()',
1412 'enddef',
1413 'defcompile'], 'E1096:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001414enddef
1415
1416def Test_arg_type_wrong()
1417 CheckScriptFailure(['def Func3(items: list)', 'echo "a"', 'enddef'], 'E1008: Missing <type>')
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02001418 CheckScriptFailure(['def Func4(...)', 'echo "a"', 'enddef'], 'E1055: Missing name after ...')
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +02001419 CheckScriptFailure(['def Func5(items:string)', 'echo "a"'], 'E1069:')
Bram Moolenaar6e949782020-04-13 17:21:00 +02001420 CheckScriptFailure(['def Func5(items)', 'echo "a"'], 'E1077:')
Yegappan Lakshmanan34fcb692021-05-25 20:14:00 +02001421 CheckScriptFailure(['def Func6(...x:list<number>)', 'echo "a"', 'enddef'], 'E1069:')
1422 CheckScriptFailure(['def Func7(...x: int)', 'echo "a"', 'enddef'], 'E1010:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001423enddef
1424
Bram Moolenaar86cdb8a2021-04-06 19:01:03 +02001425def Test_white_space_before_comma()
1426 var lines =<< trim END
1427 vim9script
1428 def Func(a: number , b: number)
1429 enddef
1430 END
1431 CheckScriptFailure(lines, 'E1068:')
Yegappan Lakshmanan611728f2021-05-24 15:15:47 +02001432 call assert_fails('vim9cmd echo stridx("a" .. "b" , "a")', 'E1068:')
Bram Moolenaar86cdb8a2021-04-06 19:01:03 +02001433enddef
1434
Bram Moolenaar608d78f2021-03-06 22:33:12 +01001435def Test_white_space_after_comma()
1436 var lines =<< trim END
1437 vim9script
1438 def Func(a: number,b: number)
1439 enddef
1440 END
1441 CheckScriptFailure(lines, 'E1069:')
1442
1443 # OK in legacy function
1444 lines =<< trim END
1445 vim9script
1446 func Func(a,b)
1447 endfunc
1448 END
1449 CheckScriptSuccess(lines)
1450enddef
1451
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001452def Test_vim9script_call()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001453 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001454 vim9script
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001455 var name = ''
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001456 def MyFunc(arg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001457 name = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001458 enddef
1459 MyFunc('foobar')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001460 name->assert_equal('foobar')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001461
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001462 var str = 'barfoo'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001463 str->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001464 name->assert_equal('barfoo')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001465
Bram Moolenaar67979662020-06-20 22:50:47 +02001466 g:value = 'value'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001467 g:value->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001468 name->assert_equal('value')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001469
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001470 var listvar = []
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001471 def ListFunc(arg: list<number>)
1472 listvar = arg
1473 enddef
1474 [1, 2, 3]->ListFunc()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001475 listvar->assert_equal([1, 2, 3])
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001476
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001477 var dictvar = {}
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001478 def DictFunc(arg: dict<number>)
1479 dictvar = arg
1480 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +01001481 {a: 1, b: 2}->DictFunc()
1482 dictvar->assert_equal({a: 1, b: 2})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001483 def CompiledDict()
Bram Moolenaare0de1712020-12-02 17:36:54 +01001484 {a: 3, b: 4}->DictFunc()
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001485 enddef
1486 CompiledDict()
Bram Moolenaare0de1712020-12-02 17:36:54 +01001487 dictvar->assert_equal({a: 3, b: 4})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001488
Bram Moolenaare0de1712020-12-02 17:36:54 +01001489 {a: 3, b: 4}->DictFunc()
1490 dictvar->assert_equal({a: 3, b: 4})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001491
1492 ('text')->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001493 name->assert_equal('text')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001494 ("some")->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001495 name->assert_equal('some')
Bram Moolenaare6b53242020-07-01 17:28:33 +02001496
Bram Moolenaar13e12b82020-07-24 18:47:22 +02001497 # line starting with single quote is not a mark
Bram Moolenaar10409562020-07-29 20:00:38 +02001498 # line starting with double quote can be a method call
Bram Moolenaar3d48e252020-07-15 14:15:52 +02001499 'asdfasdf'->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001500 name->assert_equal('asdfasdf')
Bram Moolenaar10409562020-07-29 20:00:38 +02001501 "xyz"->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001502 name->assert_equal('xyz')
Bram Moolenaar3d48e252020-07-15 14:15:52 +02001503
1504 def UseString()
1505 'xyork'->MyFunc()
1506 enddef
1507 UseString()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001508 name->assert_equal('xyork')
Bram Moolenaar3d48e252020-07-15 14:15:52 +02001509
Bram Moolenaar10409562020-07-29 20:00:38 +02001510 def UseString2()
1511 "knife"->MyFunc()
1512 enddef
1513 UseString2()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001514 name->assert_equal('knife')
Bram Moolenaar10409562020-07-29 20:00:38 +02001515
Bram Moolenaar13e12b82020-07-24 18:47:22 +02001516 # prepending a colon makes it a mark
1517 new
1518 setline(1, ['aaa', 'bbb', 'ccc'])
1519 normal! 3Gmt1G
1520 :'t
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001521 getcurpos()[1]->assert_equal(3)
Bram Moolenaar13e12b82020-07-24 18:47:22 +02001522 bwipe!
1523
Bram Moolenaare6b53242020-07-01 17:28:33 +02001524 MyFunc(
1525 'continued'
1526 )
1527 assert_equal('continued',
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001528 name
Bram Moolenaare6b53242020-07-01 17:28:33 +02001529 )
1530
1531 call MyFunc(
1532 'more'
1533 ..
1534 'lines'
1535 )
1536 assert_equal(
1537 'morelines',
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001538 name)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001539 END
1540 writefile(lines, 'Xcall.vim')
1541 source Xcall.vim
1542 delete('Xcall.vim')
1543enddef
1544
1545def Test_vim9script_call_fail_decl()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001546 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001547 vim9script
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001548 var name = ''
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001549 def MyFunc(arg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001550 var name = 123
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001551 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +02001552 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001553 END
Bram Moolenaar6c4bfe42020-07-23 18:26:30 +02001554 CheckScriptFailure(lines, 'E1054:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001555enddef
1556
Bram Moolenaar65b95452020-07-19 14:03:09 +02001557def Test_vim9script_call_fail_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001558 var lines =<< trim END
Bram Moolenaar65b95452020-07-19 14:03:09 +02001559 vim9script
1560 def MyFunc(arg: string)
1561 echo arg
1562 enddef
1563 MyFunc(1234)
1564 END
Bram Moolenaar77072282020-09-16 17:55:40 +02001565 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected string but got number')
Bram Moolenaar65b95452020-07-19 14:03:09 +02001566enddef
1567
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001568def Test_vim9script_call_fail_const()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001569 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001570 vim9script
1571 const var = ''
1572 def MyFunc(arg: string)
1573 var = 'asdf'
1574 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +02001575 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001576 END
1577 writefile(lines, 'Xcall_const.vim')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001578 assert_fails('source Xcall_const.vim', 'E46:', '', 1, 'MyFunc')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001579 delete('Xcall_const.vim')
Bram Moolenaar3bdc90b2020-12-22 20:35:40 +01001580
1581 lines =<< trim END
1582 const g:Aconst = 77
1583 def Change()
1584 # comment
1585 g:Aconst = 99
1586 enddef
1587 call Change()
1588 unlet g:Aconst
1589 END
Bram Moolenaar1dcf55d2020-12-22 22:07:30 +01001590 CheckScriptFailure(lines, 'E741: Value is locked: Aconst', 2)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001591enddef
1592
1593" Test that inside :function a Python function can be defined, :def is not
1594" recognized.
1595func Test_function_python()
1596 CheckFeature python3
Bram Moolenaar727345e2020-09-27 23:33:59 +02001597 let py = 'python3'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001598 execute py "<< EOF"
1599def do_something():
1600 return 1
1601EOF
1602endfunc
1603
1604def Test_delfunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001605 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001606 vim9script
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02001607 def g:GoneSoon()
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001608 echo 'hello'
1609 enddef
1610
1611 def CallGoneSoon()
1612 GoneSoon()
1613 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +02001614 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001615
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02001616 delfunc g:GoneSoon
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001617 CallGoneSoon()
1618 END
1619 writefile(lines, 'XToDelFunc')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001620 assert_fails('so XToDelFunc', 'E933:', '', 1, 'CallGoneSoon')
1621 assert_fails('so XToDelFunc', 'E933:', '', 1, 'CallGoneSoon')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001622
1623 delete('XToDelFunc')
1624enddef
1625
1626def Test_redef_failure()
Bram Moolenaard2c61702020-09-06 15:58:36 +02001627 writefile(['def Func0(): string', 'return "Func0"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001628 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +02001629 writefile(['def Func1(): string', 'return "Func1"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001630 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +02001631 writefile(['def! Func0(): string', 'enddef', 'defcompile'], 'Xdef')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001632 assert_fails('so Xdef', 'E1027:', '', 1, 'Func0')
Bram Moolenaard2c61702020-09-06 15:58:36 +02001633 writefile(['def Func2(): string', 'return "Func2"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001634 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +02001635 delete('Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001636
Bram Moolenaar701cc6c2021-04-10 13:33:48 +02001637 assert_fails('g:Func0()', 'E1091:')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001638 g:Func1()->assert_equal('Func1')
1639 g:Func2()->assert_equal('Func2')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001640
1641 delfunc! Func0
1642 delfunc! Func1
1643 delfunc! Func2
1644enddef
1645
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +02001646def Test_vim9script_func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001647 var lines =<< trim END
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +02001648 vim9script
1649 func Func(arg)
1650 echo a:arg
1651 endfunc
1652 Func('text')
1653 END
1654 writefile(lines, 'XVim9Func')
1655 so XVim9Func
1656
1657 delete('XVim9Func')
1658enddef
1659
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001660let s:funcResult = 0
1661
1662def FuncNoArgNoRet()
Bram Moolenaar53900992020-08-22 19:02:02 +02001663 s:funcResult = 11
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001664enddef
1665
1666def FuncNoArgRetNumber(): number
Bram Moolenaar53900992020-08-22 19:02:02 +02001667 s:funcResult = 22
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001668 return 1234
1669enddef
1670
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001671def FuncNoArgRetString(): string
Bram Moolenaar53900992020-08-22 19:02:02 +02001672 s:funcResult = 45
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001673 return 'text'
1674enddef
1675
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001676def FuncOneArgNoRet(arg: number)
Bram Moolenaar53900992020-08-22 19:02:02 +02001677 s:funcResult = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001678enddef
1679
1680def FuncOneArgRetNumber(arg: number): number
Bram Moolenaar53900992020-08-22 19:02:02 +02001681 s:funcResult = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001682 return arg
1683enddef
1684
Bram Moolenaar08938ee2020-04-11 23:17:17 +02001685def FuncTwoArgNoRet(one: bool, two: number)
Bram Moolenaar53900992020-08-22 19:02:02 +02001686 s:funcResult = two
Bram Moolenaar08938ee2020-04-11 23:17:17 +02001687enddef
1688
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001689def FuncOneArgRetString(arg: string): string
1690 return arg
1691enddef
1692
Bram Moolenaar89228602020-04-05 22:14:54 +02001693def FuncOneArgRetAny(arg: any): any
1694 return arg
1695enddef
1696
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001697def Test_func_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001698 var Ref1: func()
Bram Moolenaar53900992020-08-22 19:02:02 +02001699 s:funcResult = 0
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001700 Ref1 = FuncNoArgNoRet
1701 Ref1()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001702 s:funcResult->assert_equal(11)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001703
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001704 var Ref2: func
Bram Moolenaar53900992020-08-22 19:02:02 +02001705 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001706 Ref2 = FuncNoArgNoRet
1707 Ref2()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001708 s:funcResult->assert_equal(11)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001709
Bram Moolenaar53900992020-08-22 19:02:02 +02001710 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001711 Ref2 = FuncOneArgNoRet
1712 Ref2(12)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001713 s:funcResult->assert_equal(12)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001714
Bram Moolenaar53900992020-08-22 19:02:02 +02001715 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001716 Ref2 = FuncNoArgRetNumber
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001717 Ref2()->assert_equal(1234)
1718 s:funcResult->assert_equal(22)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001719
Bram Moolenaar53900992020-08-22 19:02:02 +02001720 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001721 Ref2 = FuncOneArgRetNumber
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001722 Ref2(13)->assert_equal(13)
1723 s:funcResult->assert_equal(13)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001724enddef
1725
Bram Moolenaar9978d472020-07-05 16:01:56 +02001726def Test_repeat_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001727 var res = 0
Bram Moolenaar9978d472020-07-05 16:01:56 +02001728 for n in repeat([1], 3)
1729 res += n
1730 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001731 res->assert_equal(3)
Bram Moolenaarfce82b32020-07-05 16:07:21 +02001732
1733 res = 0
1734 for n in add([1, 2], 3)
1735 res += n
1736 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001737 res->assert_equal(6)
Bram Moolenaar9978d472020-07-05 16:01:56 +02001738enddef
1739
Bram Moolenaar846178a2020-07-05 17:04:13 +02001740def Test_argv_return_type()
1741 next fileone filetwo
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001742 var res = ''
Bram Moolenaar846178a2020-07-05 17:04:13 +02001743 for name in argv()
1744 res ..= name
1745 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001746 res->assert_equal('fileonefiletwo')
Bram Moolenaar846178a2020-07-05 17:04:13 +02001747enddef
1748
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001749def Test_func_type_part()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001750 var RefVoid: func: void
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001751 RefVoid = FuncNoArgNoRet
1752 RefVoid = FuncOneArgNoRet
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001753 CheckDefFailure(['var RefVoid: func: void', 'RefVoid = FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func(...) but got func(): number')
1754 CheckDefFailure(['var RefVoid: func: void', 'RefVoid = FuncNoArgRetString'], 'E1012: Type mismatch; expected func(...) but got func(): string')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001755
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001756 var RefAny: func(): any
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001757 RefAny = FuncNoArgRetNumber
1758 RefAny = FuncNoArgRetString
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001759 CheckDefFailure(['var RefAny: func(): any', 'RefAny = FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(): any but got func()')
1760 CheckDefFailure(['var RefAny: func(): any', 'RefAny = FuncOneArgNoRet'], 'E1012: Type mismatch; expected func(): any but got func(number)')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001761
Bram Moolenaar6abd3dc2020-10-04 14:17:32 +02001762 var RefAnyNoArgs: func: any = RefAny
1763
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001764 var RefNr: func: number
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001765 RefNr = FuncNoArgRetNumber
1766 RefNr = FuncOneArgRetNumber
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001767 CheckDefFailure(['var RefNr: func: number', 'RefNr = FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(...): number but got func()')
1768 CheckDefFailure(['var RefNr: func: number', 'RefNr = FuncNoArgRetString'], 'E1012: Type mismatch; expected func(...): number but got func(): string')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001769
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001770 var RefStr: func: string
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001771 RefStr = FuncNoArgRetString
1772 RefStr = FuncOneArgRetString
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001773 CheckDefFailure(['var RefStr: func: string', 'RefStr = FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(...): string but got func()')
1774 CheckDefFailure(['var RefStr: func: string', 'RefStr = FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func(...): string but got func(): number')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001775enddef
1776
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001777def Test_func_type_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001778 CheckDefFailure(['var ref1: func()'], 'E704:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001779
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001780 CheckDefFailure(['var Ref1: func()', 'Ref1 = FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func() but got func(): number')
1781 CheckDefFailure(['var Ref1: func()', 'Ref1 = FuncOneArgNoRet'], 'E1012: Type mismatch; expected func() but got func(number)')
1782 CheckDefFailure(['var Ref1: func()', 'Ref1 = FuncOneArgRetNumber'], 'E1012: Type mismatch; expected func() but got func(number): number')
1783 CheckDefFailure(['var Ref1: func(bool)', 'Ref1 = FuncTwoArgNoRet'], 'E1012: Type mismatch; expected func(bool) but got func(bool, number)')
1784 CheckDefFailure(['var Ref1: func(?bool)', 'Ref1 = FuncTwoArgNoRet'], 'E1012: Type mismatch; expected func(?bool) but got func(bool, number)')
1785 CheckDefFailure(['var Ref1: func(...bool)', 'Ref1 = FuncTwoArgNoRet'], 'E1012: Type mismatch; expected func(...bool) but got func(bool, number)')
Bram Moolenaar08938ee2020-04-11 23:17:17 +02001786
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001787 CheckDefFailure(['var RefWrong: func(string ,number)'], 'E1068:')
1788 CheckDefFailure(['var RefWrong: func(string,number)'], 'E1069:')
1789 CheckDefFailure(['var RefWrong: func(bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool, bool)'], 'E1005:')
1790 CheckDefFailure(['var RefWrong: func(bool):string'], 'E1069:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001791enddef
1792
Bram Moolenaar89228602020-04-05 22:14:54 +02001793def Test_func_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001794 var nr: number
Bram Moolenaar89228602020-04-05 22:14:54 +02001795 nr = FuncNoArgRetNumber()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001796 nr->assert_equal(1234)
Bram Moolenaar89228602020-04-05 22:14:54 +02001797
1798 nr = FuncOneArgRetAny(122)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001799 nr->assert_equal(122)
Bram Moolenaar89228602020-04-05 22:14:54 +02001800
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001801 var str: string
Bram Moolenaar89228602020-04-05 22:14:54 +02001802 str = FuncOneArgRetAny('yes')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001803 str->assert_equal('yes')
Bram Moolenaar89228602020-04-05 22:14:54 +02001804
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001805 CheckDefFailure(['var str: string', 'str = FuncNoArgRetNumber()'], 'E1012: Type mismatch; expected string but got number')
Bram Moolenaar89228602020-04-05 22:14:54 +02001806enddef
1807
Bram Moolenaar6abd3dc2020-10-04 14:17:32 +02001808def Test_func_common_type()
1809 def FuncOne(n: number): number
1810 return n
1811 enddef
1812 def FuncTwo(s: string): number
1813 return len(s)
1814 enddef
1815 def FuncThree(n: number, s: string): number
1816 return n + len(s)
1817 enddef
1818 var list = [FuncOne, FuncTwo, FuncThree]
1819 assert_equal(8, list[0](8))
1820 assert_equal(4, list[1]('word'))
1821 assert_equal(7, list[2](3, 'word'))
1822enddef
1823
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001824def MultiLine(
1825 arg1: string,
1826 arg2 = 1234,
1827 ...rest: list<string>
1828 ): string
1829 return arg1 .. arg2 .. join(rest, '-')
1830enddef
1831
Bram Moolenaar2c330432020-04-13 14:41:35 +02001832def MultiLineComment(
1833 arg1: string, # comment
1834 arg2 = 1234, # comment
1835 ...rest: list<string> # comment
1836 ): string # comment
1837 return arg1 .. arg2 .. join(rest, '-')
1838enddef
1839
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001840def Test_multiline()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001841 MultiLine('text')->assert_equal('text1234')
1842 MultiLine('text', 777)->assert_equal('text777')
1843 MultiLine('text', 777, 'one')->assert_equal('text777one')
1844 MultiLine('text', 777, 'one', 'two')->assert_equal('text777one-two')
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001845enddef
1846
Bram Moolenaar23e03252020-04-12 22:22:31 +02001847func Test_multiline_not_vim9()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001848 call MultiLine('text')->assert_equal('text1234')
1849 call MultiLine('text', 777)->assert_equal('text777')
1850 call MultiLine('text', 777, 'one')->assert_equal('text777one')
1851 call MultiLine('text', 777, 'one', 'two')->assert_equal('text777one-two')
Bram Moolenaar23e03252020-04-12 22:22:31 +02001852endfunc
1853
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001854
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02001855" When using CheckScriptFailure() for the below test, E1010 is generated instead
1856" of E1056.
1857func Test_E1056_1059()
1858 let caught_1056 = 0
1859 try
1860 def F():
1861 return 1
1862 enddef
1863 catch /E1056:/
1864 let caught_1056 = 1
1865 endtry
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001866 eval caught_1056->assert_equal(1)
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02001867
1868 let caught_1059 = 0
1869 try
1870 def F5(items : list)
1871 echo 'a'
1872 enddef
1873 catch /E1059:/
1874 let caught_1059 = 1
1875 endtry
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001876 eval caught_1059->assert_equal(1)
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02001877endfunc
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001878
Bram Moolenaar015f4262020-05-05 21:25:22 +02001879func DelMe()
1880 echo 'DelMe'
1881endfunc
1882
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001883def Test_error_reporting()
1884 # comment lines at the start of the function
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001885 var lines =<< trim END
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001886 " comment
1887 def Func()
1888 # comment
1889 # comment
1890 invalid
1891 enddef
1892 defcompile
1893 END
Bram Moolenaar08052222020-09-14 17:04:31 +02001894 writefile(lines, 'Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001895 try
1896 source Xdef
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001897 assert_report('should have failed')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001898 catch /E476:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001899 v:exception->assert_match('Invalid command: invalid')
1900 v:throwpoint->assert_match(', line 3$')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001901 endtry
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001902 delfunc! g:Func
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001903
1904 # comment lines after the start of the function
1905 lines =<< trim END
1906 " comment
1907 def Func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001908 var x = 1234
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001909 # comment
1910 # comment
1911 invalid
1912 enddef
1913 defcompile
1914 END
Bram Moolenaar08052222020-09-14 17:04:31 +02001915 writefile(lines, 'Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001916 try
1917 source Xdef
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001918 assert_report('should have failed')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001919 catch /E476:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001920 v:exception->assert_match('Invalid command: invalid')
1921 v:throwpoint->assert_match(', line 4$')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001922 endtry
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001923 delfunc! g:Func
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001924
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001925 lines =<< trim END
1926 vim9script
1927 def Func()
Bram Moolenaare0de1712020-12-02 17:36:54 +01001928 var db = {foo: 1, bar: 2}
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001929 # comment
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001930 var x = db.asdf
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001931 enddef
1932 defcompile
1933 Func()
1934 END
Bram Moolenaar08052222020-09-14 17:04:31 +02001935 writefile(lines, 'Xdef')
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001936 try
1937 source Xdef
1938 assert_report('should have failed')
1939 catch /E716:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001940 v:throwpoint->assert_match('_Func, line 3$')
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001941 endtry
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001942 delfunc! g:Func
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001943
Bram Moolenaar08052222020-09-14 17:04:31 +02001944 delete('Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001945enddef
1946
Bram Moolenaar015f4262020-05-05 21:25:22 +02001947def Test_deleted_function()
1948 CheckDefExecFailure([
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001949 'var RefMe: func = function("g:DelMe")',
Bram Moolenaar015f4262020-05-05 21:25:22 +02001950 'delfunc g:DelMe',
1951 'echo RefMe()'], 'E117:')
1952enddef
1953
1954def Test_unknown_function()
1955 CheckDefExecFailure([
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001956 'var Ref: func = function("NotExist")',
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +02001957 'delfunc g:NotExist'], 'E700:')
Bram Moolenaar015f4262020-05-05 21:25:22 +02001958enddef
1959
Bram Moolenaar328eac22021-01-07 19:23:08 +01001960def RefFunc(Ref: func(any): any): string
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +02001961 return Ref('more')
1962enddef
1963
1964def Test_closure_simple()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001965 var local = 'some '
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001966 RefFunc((s) => local .. s)->assert_equal('some more')
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +02001967enddef
1968
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001969def MakeRef()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001970 var local = 'some '
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001971 g:Ref = (s) => local .. s
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001972enddef
1973
1974def Test_closure_ref_after_return()
1975 MakeRef()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001976 g:Ref('thing')->assert_equal('some thing')
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001977 unlet g:Ref
1978enddef
1979
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001980def MakeTwoRefs()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001981 var local = ['some']
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001982 g:Extend = (s) => local->add(s)
1983 g:Read = () => local
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001984enddef
1985
1986def Test_closure_two_refs()
1987 MakeTwoRefs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001988 join(g:Read(), ' ')->assert_equal('some')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001989 g:Extend('more')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001990 join(g:Read(), ' ')->assert_equal('some more')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001991 g:Extend('even')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001992 join(g:Read(), ' ')->assert_equal('some more even')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001993
1994 unlet g:Extend
1995 unlet g:Read
1996enddef
1997
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001998def ReadRef(Ref: func(): list<string>): string
1999 return join(Ref(), ' ')
2000enddef
2001
Bram Moolenaar5e654232020-09-16 15:22:00 +02002002def ExtendRef(Ref: func(string): list<string>, add: string)
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02002003 Ref(add)
2004enddef
2005
2006def Test_closure_two_indirect_refs()
Bram Moolenaarf7779c62020-05-03 15:38:16 +02002007 MakeTwoRefs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002008 ReadRef(g:Read)->assert_equal('some')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02002009 ExtendRef(g:Extend, 'more')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002010 ReadRef(g:Read)->assert_equal('some more')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02002011 ExtendRef(g:Extend, 'even')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002012 ReadRef(g:Read)->assert_equal('some more even')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02002013
2014 unlet g:Extend
2015 unlet g:Read
2016enddef
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02002017
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02002018def MakeArgRefs(theArg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002019 var local = 'loc_val'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002020 g:UseArg = (s) => theArg .. '/' .. local .. '/' .. s
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02002021enddef
2022
2023def MakeArgRefsVarargs(theArg: string, ...rest: list<string>)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002024 var local = 'the_loc'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002025 g:UseVararg = (s) => theArg .. '/' .. local .. '/' .. s .. '/' .. join(rest)
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02002026enddef
2027
2028def Test_closure_using_argument()
2029 MakeArgRefs('arg_val')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002030 g:UseArg('call_val')->assert_equal('arg_val/loc_val/call_val')
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02002031
2032 MakeArgRefsVarargs('arg_val', 'one', 'two')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002033 g:UseVararg('call_val')->assert_equal('arg_val/the_loc/call_val/one two')
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02002034
2035 unlet g:UseArg
2036 unlet g:UseVararg
Bram Moolenaar44ec21c2021-02-12 21:50:57 +01002037
2038 var lines =<< trim END
2039 vim9script
2040 def Test(Fun: func(number): number): list<number>
2041 return map([1, 2, 3], (_, i) => Fun(i))
2042 enddef
2043 def Inc(nr: number): number
2044 return nr + 2
2045 enddef
2046 assert_equal([3, 4, 5], Test(Inc))
2047 END
2048 CheckScriptSuccess(lines)
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02002049enddef
2050
Bram Moolenaar85d5e2b2020-10-10 14:13:01 +02002051def MakeGetAndAppendRefs()
2052 var local = 'a'
2053
2054 def Append(arg: string)
2055 local ..= arg
2056 enddef
2057 g:Append = Append
2058
2059 def Get(): string
2060 return local
2061 enddef
2062 g:Get = Get
2063enddef
2064
2065def Test_closure_append_get()
2066 MakeGetAndAppendRefs()
2067 g:Get()->assert_equal('a')
2068 g:Append('-b')
2069 g:Get()->assert_equal('a-b')
2070 g:Append('-c')
2071 g:Get()->assert_equal('a-b-c')
2072
2073 unlet g:Append
2074 unlet g:Get
2075enddef
Bram Moolenaarb68b3462020-05-06 21:06:30 +02002076
Bram Moolenaar04b12692020-05-04 23:24:44 +02002077def Test_nested_closure()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002078 var local = 'text'
Bram Moolenaar04b12692020-05-04 23:24:44 +02002079 def Closure(arg: string): string
2080 return local .. arg
2081 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002082 Closure('!!!')->assert_equal('text!!!')
Bram Moolenaar04b12692020-05-04 23:24:44 +02002083enddef
2084
Bram Moolenaar6f5b6df2020-05-16 21:20:12 +02002085func GetResult(Ref)
2086 return a:Ref('some')
2087endfunc
2088
2089def Test_call_closure_not_compiled()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002090 var text = 'text'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002091 g:Ref = (s) => s .. text
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002092 GetResult(g:Ref)->assert_equal('sometext')
Bram Moolenaar6f5b6df2020-05-16 21:20:12 +02002093enddef
2094
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02002095def Test_double_closure_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002096 var lines =<< trim END
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02002097 vim9script
2098 def Func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002099 var name = 0
2100 for i in range(2)
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002101 timer_start(0, () => name)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002102 endfor
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02002103 enddef
2104 Func()
2105 END
Bram Moolenaar148ce7a2020-09-23 21:57:23 +02002106 CheckScriptSuccess(lines)
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02002107enddef
2108
Bram Moolenaar85d5e2b2020-10-10 14:13:01 +02002109def Test_nested_closure_used()
2110 var lines =<< trim END
2111 vim9script
2112 def Func()
2113 var x = 'hello'
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002114 var Closure = () => x
2115 g:Myclosure = () => Closure()
Bram Moolenaar85d5e2b2020-10-10 14:13:01 +02002116 enddef
2117 Func()
2118 assert_equal('hello', g:Myclosure())
2119 END
2120 CheckScriptSuccess(lines)
2121enddef
Bram Moolenaar0876c782020-10-07 19:08:04 +02002122
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02002123def Test_nested_closure_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002124 var lines =<< trim END
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02002125 vim9script
2126 def FuncA()
2127 FuncB(0)
2128 enddef
2129 def FuncB(n: number): list<string>
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002130 return map([0], (_, v) => n)
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02002131 enddef
2132 FuncA()
2133 END
2134 CheckScriptFailure(lines, 'E1012:')
2135enddef
2136
Bram Moolenaarf112f302020-12-20 17:47:52 +01002137def Test_global_closure()
2138 var lines =<< trim END
2139 vim9script
2140 def ReverseEveryNLines(n: number, line1: number, line2: number)
2141 var mods = 'sil keepj keepp lockm '
2142 var range = ':' .. line1 .. ',' .. line2
2143 def g:Offset(): number
2144 var offset = (line('.') - line1 + 1) % n
2145 return offset != 0 ? offset : n
2146 enddef
2147 exe mods .. range .. 'g/^/exe "m .-" .. g:Offset()'
2148 enddef
2149
2150 new
2151 repeat(['aaa', 'bbb', 'ccc'], 3)->setline(1)
2152 ReverseEveryNLines(3, 1, 9)
2153 END
2154 CheckScriptSuccess(lines)
2155 var expected = repeat(['ccc', 'bbb', 'aaa'], 3)
2156 assert_equal(expected, getline(1, 9))
2157 bwipe!
2158enddef
2159
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01002160def Test_global_closure_called_directly()
2161 var lines =<< trim END
2162 vim9script
2163 def Outer()
2164 var x = 1
2165 def g:Inner()
2166 var y = x
2167 x += 1
2168 assert_equal(1, y)
2169 enddef
2170 g:Inner()
2171 assert_equal(2, x)
2172 enddef
2173 Outer()
2174 END
2175 CheckScriptSuccess(lines)
2176 delfunc g:Inner
2177enddef
2178
Bram Moolenaar34c54eb2020-11-25 19:15:19 +01002179def Test_failure_in_called_function()
2180 # this was using the frame index as the return value
2181 var lines =<< trim END
2182 vim9script
2183 au TerminalWinOpen * eval [][0]
2184 def PopupTerm(a: any)
2185 # make sure typvals on stack are string
2186 ['a', 'b', 'c', 'd', 'e', 'f', 'g']->join()
2187 FireEvent()
2188 enddef
2189 def FireEvent()
2190 do TerminalWinOpen
2191 enddef
2192 # use try/catch to make eval fail
2193 try
2194 call PopupTerm(0)
2195 catch
2196 endtry
2197 au! TerminalWinOpen
2198 END
2199 CheckScriptSuccess(lines)
2200enddef
2201
Bram Moolenaar5366e1a2020-10-01 13:01:34 +02002202def Test_nested_lambda()
2203 var lines =<< trim END
2204 vim9script
2205 def Func()
2206 var x = 4
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002207 var Lambda1 = () => 7
2208 var Lambda2 = () => [Lambda1(), x]
Bram Moolenaar5366e1a2020-10-01 13:01:34 +02002209 var res = Lambda2()
2210 assert_equal([7, 4], res)
2211 enddef
2212 Func()
2213 END
2214 CheckScriptSuccess(lines)
2215enddef
2216
Bram Moolenaarc04f2a42021-06-09 19:30:03 +02002217def Test_double_nested_lambda()
2218 var lines =<< trim END
2219 vim9script
2220 def F(head: string): func(string): func(string): string
2221 return (sep: string): func(string): string => ((tail: string): string => {
2222 return head .. sep .. tail
2223 })
2224 enddef
2225 assert_equal('hello-there', F('hello')('-')('there'))
2226 END
2227 CheckScriptSuccess(lines)
2228enddef
2229
Bram Moolenaar074f84c2021-05-18 11:47:44 +02002230def Test_nested_inline_lambda()
2231 # TODO: use the "text" argument
2232 var lines =<< trim END
2233 vim9script
2234 def F(text: string): func(string): func(string): string
2235 return (arg: string): func(string): string => ((sep: string): string => {
2236 return sep .. arg
2237 })
2238 enddef
2239 assert_equal('--there', F('unused')('there')('--'))
2240 END
2241 CheckScriptSuccess(lines)
2242enddef
2243
Bram Moolenaar52bf81c2020-11-17 18:50:44 +01002244def Shadowed(): list<number>
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002245 var FuncList: list<func: number> = [() => 42]
Bram Moolenaar75ab91f2021-01-10 22:42:50 +01002246 return FuncList->mapnew((_, Shadowed) => Shadowed())
Bram Moolenaar52bf81c2020-11-17 18:50:44 +01002247enddef
2248
2249def Test_lambda_arg_shadows_func()
2250 assert_equal([42], Shadowed())
2251enddef
2252
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02002253def Line_continuation_in_def(dir: string = ''): string
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002254 var path: string = empty(dir)
2255 \ ? 'empty'
2256 \ : 'full'
2257 return path
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02002258enddef
2259
2260def Test_line_continuation_in_def()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002261 Line_continuation_in_def('.')->assert_equal('full')
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02002262enddef
2263
Bram Moolenaar2ea95b62020-11-19 21:47:56 +01002264def Test_script_var_in_lambda()
2265 var lines =<< trim END
2266 vim9script
2267 var script = 'test'
Bram Moolenaarbb8a7ce2021-04-10 20:10:26 +02002268 assert_equal(['test'], map(['one'], (_, _) => script))
Bram Moolenaar2ea95b62020-11-19 21:47:56 +01002269 END
2270 CheckScriptSuccess(lines)
2271enddef
2272
Bram Moolenaar5e654232020-09-16 15:22:00 +02002273def Line_continuation_in_lambda(): list<string>
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002274 var x = range(97, 100)
Bram Moolenaar75ab91f2021-01-10 22:42:50 +01002275 ->mapnew((_, v) => nr2char(v)
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002276 ->toupper())
Bram Moolenaar7a4b8982020-07-08 17:36:21 +02002277 ->reverse()
2278 return x
2279enddef
2280
2281def Test_line_continuation_in_lambda()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002282 Line_continuation_in_lambda()->assert_equal(['D', 'C', 'B', 'A'])
Bram Moolenaarf898f7c2021-01-16 18:09:52 +01002283
2284 var lines =<< trim END
2285 vim9script
2286 var res = [{n: 1, m: 2, s: 'xxx'}]
2287 ->mapnew((_, v: dict<any>): string => printf('%d:%d:%s',
2288 v.n,
2289 v.m,
2290 substitute(v.s, '.*', 'yyy', '')
2291 ))
2292 assert_equal(['1:2:yyy'], res)
2293 END
2294 CheckScriptSuccess(lines)
Bram Moolenaar7a4b8982020-07-08 17:36:21 +02002295enddef
2296
Bram Moolenaarb6571982021-01-08 22:24:19 +01002297def Test_list_lambda()
2298 timer_start(1000, (_) => 0)
2299 var body = execute(timer_info()[0].callback
2300 ->string()
2301 ->substitute("('", ' ', '')
2302 ->substitute("')", '', '')
2303 ->substitute('function\zs', ' ', ''))
Bram Moolenaar767034c2021-04-09 17:24:52 +02002304 assert_match('def <lambda>\d\+(_: any): number\n1 return 0\n enddef', body)
Bram Moolenaarb6571982021-01-08 22:24:19 +01002305enddef
2306
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02002307def Test_legacy_lambda()
2308 legacy echo {x -> 'hello ' .. x}('foo')
Bram Moolenaardc4c2302021-04-25 13:54:42 +02002309
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02002310 var lines =<< trim END
2311 echo {x -> 'hello ' .. x}('foo')
2312 END
2313 CheckDefAndScriptFailure(lines, 'E720:')
Bram Moolenaardc4c2302021-04-25 13:54:42 +02002314
2315 lines =<< trim END
2316 vim9script
2317 def Func()
2318 echo (() => 'no error')()
2319 enddef
2320 legacy call s:Func()
2321 END
2322 CheckScriptSuccess(lines)
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02002323enddef
2324
Bram Moolenaarce024c32021-06-26 13:00:49 +02002325def Test_legacy()
2326 var lines =<< trim END
2327 vim9script
2328 func g:LegacyFunction()
2329 let g:legacyvar = 1
2330 endfunc
2331 def Testit()
2332 legacy call g:LegacyFunction()
2333 enddef
2334 Testit()
2335 assert_equal(1, g:legacyvar)
2336 unlet g:legacyvar
2337 delfunc g:LegacyFunction
2338 END
2339 CheckScriptSuccess(lines)
2340enddef
2341
Bram Moolenaarc3cb1c92021-06-02 16:47:53 +02002342def Test_legacy_errors()
2343 for cmd in ['if', 'elseif', 'else', 'endif',
2344 'for', 'endfor', 'continue', 'break',
2345 'while', 'endwhile',
2346 'try', 'catch', 'finally', 'endtry']
2347 CheckDefFailure(['legacy ' .. cmd .. ' expr'], 'E1189:')
2348 endfor
2349enddef
2350
Bram Moolenaarab360522021-01-10 14:02:28 +01002351def DoFilterThis(a: string): list<string>
2352 # closure nested inside another closure using argument
2353 var Filter = (l) => filter(l, (_, v) => stridx(v, a) == 0)
2354 return ['x', 'y', 'a', 'x2', 'c']->Filter()
2355enddef
2356
2357def Test_nested_closure_using_argument()
2358 assert_equal(['x', 'x2'], DoFilterThis('x'))
2359enddef
2360
Bram Moolenaar0186e582021-01-10 18:33:11 +01002361def Test_triple_nested_closure()
2362 var what = 'x'
2363 var Match = (val: string, cmp: string): bool => stridx(val, cmp) == 0
2364 var Filter = (l) => filter(l, (_, v) => Match(v, what))
2365 assert_equal(['x', 'x2'], ['x', 'y', 'a', 'x2', 'c']->Filter())
2366enddef
2367
Bram Moolenaar8f510af2020-07-05 18:48:23 +02002368func Test_silent_echo()
Bram Moolenaar47e7d702020-07-05 18:18:42 +02002369 CheckScreendump
2370
2371 let lines =<< trim END
2372 vim9script
2373 def EchoNothing()
2374 silent echo ''
2375 enddef
2376 defcompile
2377 END
Bram Moolenaar8f510af2020-07-05 18:48:23 +02002378 call writefile(lines, 'XTest_silent_echo')
Bram Moolenaar47e7d702020-07-05 18:18:42 +02002379
2380 " Check that the balloon shows up after a mouse move
2381 let buf = RunVimInTerminal('-S XTest_silent_echo', {'rows': 6})
Bram Moolenaar8f510af2020-07-05 18:48:23 +02002382 call term_sendkeys(buf, ":abc")
Bram Moolenaar47e7d702020-07-05 18:18:42 +02002383 call VerifyScreenDump(buf, 'Test_vim9_silent_echo', {})
2384
2385 " clean up
2386 call StopVimInTerminal(buf)
2387 call delete('XTest_silent_echo')
Bram Moolenaar8f510af2020-07-05 18:48:23 +02002388endfunc
Bram Moolenaar47e7d702020-07-05 18:18:42 +02002389
Bram Moolenaar171fb922020-10-28 16:54:47 +01002390def SilentlyError()
2391 execute('silent! invalid')
2392 g:did_it = 'yes'
2393enddef
2394
Bram Moolenaar28ee8922020-10-28 20:20:00 +01002395func UserError()
2396 silent! invalid
2397endfunc
2398
2399def SilentlyUserError()
2400 UserError()
2401 g:did_it = 'yes'
2402enddef
Bram Moolenaar171fb922020-10-28 16:54:47 +01002403
2404" This can't be a :def function, because the assert would not be reached.
Bram Moolenaar171fb922020-10-28 16:54:47 +01002405func Test_ignore_silent_error()
2406 let g:did_it = 'no'
2407 call SilentlyError()
2408 call assert_equal('yes', g:did_it)
2409
Bram Moolenaar28ee8922020-10-28 20:20:00 +01002410 let g:did_it = 'no'
2411 call SilentlyUserError()
2412 call assert_equal('yes', g:did_it)
Bram Moolenaar171fb922020-10-28 16:54:47 +01002413
2414 unlet g:did_it
2415endfunc
2416
Bram Moolenaarcd030c42020-10-30 21:49:40 +01002417def Test_ignore_silent_error_in_filter()
2418 var lines =<< trim END
2419 vim9script
2420 def Filter(winid: number, key: string): bool
2421 if key == 'o'
2422 silent! eval [][0]
2423 return true
2424 endif
2425 return popup_filter_menu(winid, key)
2426 enddef
2427
Bram Moolenaare0de1712020-12-02 17:36:54 +01002428 popup_create('popup', {filter: Filter})
Bram Moolenaarcd030c42020-10-30 21:49:40 +01002429 feedkeys("o\r", 'xnt')
2430 END
2431 CheckScriptSuccess(lines)
2432enddef
2433
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02002434def Fibonacci(n: number): number
2435 if n < 2
2436 return n
2437 else
2438 return Fibonacci(n - 1) + Fibonacci(n - 2)
2439 endif
2440enddef
2441
Bram Moolenaar985116a2020-07-12 17:31:09 +02002442def Test_recursive_call()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002443 Fibonacci(20)->assert_equal(6765)
Bram Moolenaar985116a2020-07-12 17:31:09 +02002444enddef
2445
Bram Moolenaar08f7a412020-07-13 20:41:08 +02002446def TreeWalk(dir: string): list<any>
Bram Moolenaar75ab91f2021-01-10 22:42:50 +01002447 return readdir(dir)->mapnew((_, val) =>
Bram Moolenaar08f7a412020-07-13 20:41:08 +02002448 fnamemodify(dir .. '/' .. val, ':p')->isdirectory()
Bram Moolenaar2bede172020-11-19 18:53:18 +01002449 ? {[val]: TreeWalk(dir .. '/' .. val)}
Bram Moolenaar08f7a412020-07-13 20:41:08 +02002450 : val
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002451 )
Bram Moolenaar08f7a412020-07-13 20:41:08 +02002452enddef
2453
2454def Test_closure_in_map()
2455 mkdir('XclosureDir/tdir', 'p')
2456 writefile(['111'], 'XclosureDir/file1')
2457 writefile(['222'], 'XclosureDir/file2')
2458 writefile(['333'], 'XclosureDir/tdir/file3')
2459
Bram Moolenaare0de1712020-12-02 17:36:54 +01002460 TreeWalk('XclosureDir')->assert_equal(['file1', 'file2', {tdir: ['file3']}])
Bram Moolenaar08f7a412020-07-13 20:41:08 +02002461
2462 delete('XclosureDir', 'rf')
2463enddef
2464
Bram Moolenaar7b5d5442020-10-04 13:42:34 +02002465def Test_invalid_function_name()
2466 var lines =<< trim END
2467 vim9script
2468 def s: list<string>
2469 END
2470 CheckScriptFailure(lines, 'E129:')
2471
2472 lines =<< trim END
2473 vim9script
2474 def g: list<string>
2475 END
2476 CheckScriptFailure(lines, 'E129:')
2477
2478 lines =<< trim END
2479 vim9script
2480 def <SID>: list<string>
2481 END
2482 CheckScriptFailure(lines, 'E884:')
2483
2484 lines =<< trim END
2485 vim9script
2486 def F list<string>
2487 END
2488 CheckScriptFailure(lines, 'E488:')
2489enddef
2490
Bram Moolenaara90afb92020-07-15 22:38:56 +02002491def Test_partial_call()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002492 var Xsetlist = function('setloclist', [0])
Bram Moolenaare0de1712020-12-02 17:36:54 +01002493 Xsetlist([], ' ', {title: 'test'})
2494 getloclist(0, {title: 1})->assert_equal({title: 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02002495
2496 Xsetlist = function('setloclist', [0, [], ' '])
Bram Moolenaare0de1712020-12-02 17:36:54 +01002497 Xsetlist({title: 'test'})
2498 getloclist(0, {title: 1})->assert_equal({title: 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02002499
2500 Xsetlist = function('setqflist')
Bram Moolenaare0de1712020-12-02 17:36:54 +01002501 Xsetlist([], ' ', {title: 'test'})
2502 getqflist({title: 1})->assert_equal({title: 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02002503
2504 Xsetlist = function('setqflist', [[], ' '])
Bram Moolenaare0de1712020-12-02 17:36:54 +01002505 Xsetlist({title: 'test'})
2506 getqflist({title: 1})->assert_equal({title: 'test'})
Bram Moolenaar6abd3dc2020-10-04 14:17:32 +02002507
2508 var Len: func: number = function('len', ['word'])
2509 assert_equal(4, Len())
Bram Moolenaara90afb92020-07-15 22:38:56 +02002510enddef
2511
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02002512def Test_cmd_modifier()
2513 tab echo '0'
Bram Moolenaard2c61702020-09-06 15:58:36 +02002514 CheckDefFailure(['5tab echo 3'], 'E16:')
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02002515enddef
2516
2517def Test_restore_modifiers()
2518 # check that when compiling a :def function command modifiers are not messed
2519 # up.
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02002520 var lines =<< trim END
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02002521 vim9script
2522 set eventignore=
2523 autocmd QuickFixCmdPost * copen
2524 def AutocmdsDisabled()
Bram Moolenaar6cf7e3b2020-10-28 14:31:16 +01002525 eval 0
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02002526 enddef
2527 func Func()
2528 noautocmd call s:AutocmdsDisabled()
2529 let g:ei_after = &eventignore
2530 endfunc
2531 Func()
2532 END
2533 CheckScriptSuccess(lines)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002534 g:ei_after->assert_equal('')
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02002535enddef
2536
Bram Moolenaardfa3d552020-09-10 22:05:08 +02002537def StackTop()
2538 eval 1
2539 eval 2
2540 # call not on fourth line
2541 StackBot()
2542enddef
2543
2544def StackBot()
2545 # throw an error
2546 eval [][0]
2547enddef
2548
2549def Test_callstack_def()
2550 try
2551 StackTop()
2552 catch
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02002553 v:throwpoint->assert_match('Test_callstack_def\[2\]..StackTop\[4\]..StackBot, line 2')
Bram Moolenaardfa3d552020-09-10 22:05:08 +02002554 endtry
2555enddef
2556
Bram Moolenaare8211a32020-10-09 22:04:29 +02002557" Re-using spot for variable used in block
2558def Test_block_scoped_var()
2559 var lines =<< trim END
2560 vim9script
2561 def Func()
2562 var x = ['a', 'b', 'c']
2563 if 1
2564 var y = 'x'
Bram Moolenaarbb8a7ce2021-04-10 20:10:26 +02002565 map(x, (_, _) => y)
Bram Moolenaare8211a32020-10-09 22:04:29 +02002566 endif
2567 var z = x
2568 assert_equal(['x', 'x', 'x'], z)
2569 enddef
2570 Func()
2571 END
2572 CheckScriptSuccess(lines)
2573enddef
2574
Bram Moolenaareeece9e2020-11-20 19:26:48 +01002575def Test_reset_did_emsg()
2576 var lines =<< trim END
2577 @s = 'blah'
2578 au BufWinLeave * #
2579 def Func()
2580 var winid = popup_create('popup', {})
2581 exe '*s'
2582 popup_close(winid)
2583 enddef
2584 Func()
2585 END
2586 CheckScriptFailure(lines, 'E492:', 8)
Bram Moolenaar2d870f82020-12-05 13:41:01 +01002587 delfunc! g:Func
Bram Moolenaareeece9e2020-11-20 19:26:48 +01002588enddef
2589
Bram Moolenaar57f799e2020-12-12 20:42:19 +01002590def Test_did_emsg_reset()
2591 # executing an autocommand resets did_emsg, this should not result in a
2592 # builtin function considered failing
2593 var lines =<< trim END
2594 vim9script
2595 au BufWinLeave * #
2596 def Func()
Bram Moolenaar767034c2021-04-09 17:24:52 +02002597 popup_menu('', {callback: (a, b) => popup_create('', {})->popup_close()})
Bram Moolenaar57f799e2020-12-12 20:42:19 +01002598 eval [][0]
2599 enddef
2600 nno <F3> <cmd>call <sid>Func()<cr>
2601 feedkeys("\<F3>\e", 'xt')
2602 END
2603 writefile(lines, 'XemsgReset')
2604 assert_fails('so XemsgReset', ['E684:', 'E684:'], lines, 2)
2605 delete('XemsgReset')
2606 nunmap <F3>
2607 au! BufWinLeave
2608enddef
2609
Bram Moolenaar56602ba2020-12-05 21:22:08 +01002610def Test_abort_with_silent_call()
2611 var lines =<< trim END
2612 vim9script
2613 g:result = 'none'
2614 def Func()
2615 g:result += 3
2616 g:result = 'yes'
2617 enddef
2618 # error is silenced, but function aborts on error
2619 silent! Func()
2620 assert_equal('none', g:result)
2621 unlet g:result
2622 END
2623 CheckScriptSuccess(lines)
2624enddef
2625
Bram Moolenaarf665e972020-12-05 19:17:16 +01002626def Test_continues_with_silent_error()
2627 var lines =<< trim END
2628 vim9script
2629 g:result = 'none'
2630 def Func()
2631 silent! g:result += 3
2632 g:result = 'yes'
2633 enddef
2634 # error is silenced, function does not abort
2635 Func()
2636 assert_equal('yes', g:result)
2637 unlet g:result
2638 END
2639 CheckScriptSuccess(lines)
2640enddef
2641
Bram Moolenaaraf0df472020-12-02 20:51:22 +01002642def Test_abort_even_with_silent()
2643 var lines =<< trim END
2644 vim9script
2645 g:result = 'none'
2646 def Func()
2647 eval {-> ''}() .. '' .. {}['X']
2648 g:result = 'yes'
2649 enddef
Bram Moolenaarf665e972020-12-05 19:17:16 +01002650 silent! Func()
Bram Moolenaaraf0df472020-12-02 20:51:22 +01002651 assert_equal('none', g:result)
Bram Moolenaar4029cab2020-12-05 18:13:27 +01002652 unlet g:result
2653 END
2654 CheckScriptSuccess(lines)
2655enddef
2656
Bram Moolenaarf665e972020-12-05 19:17:16 +01002657def Test_cmdmod_silent_restored()
2658 var lines =<< trim END
2659 vim9script
2660 def Func()
2661 g:result = 'none'
2662 silent! g:result += 3
2663 g:result = 'none'
2664 g:result += 3
2665 enddef
2666 Func()
2667 END
2668 # can't use CheckScriptFailure, it ignores the :silent!
2669 var fname = 'Xdefsilent'
2670 writefile(lines, fname)
2671 var caught = 'no'
2672 try
2673 exe 'source ' .. fname
2674 catch /E1030:/
2675 caught = 'yes'
2676 assert_match('Func, line 4', v:throwpoint)
2677 endtry
2678 assert_equal('yes', caught)
2679 delete(fname)
2680enddef
2681
Bram Moolenaar2fecb532021-03-24 22:00:56 +01002682def Test_cmdmod_silent_nested()
2683 var lines =<< trim END
2684 vim9script
2685 var result = ''
2686
2687 def Error()
2688 result ..= 'Eb'
2689 eval [][0]
2690 result ..= 'Ea'
2691 enddef
2692
2693 def Crash()
2694 result ..= 'Cb'
2695 sil! Error()
2696 result ..= 'Ca'
2697 enddef
2698
2699 Crash()
2700 assert_equal('CbEbEaCa', result)
2701 END
2702 CheckScriptSuccess(lines)
2703enddef
2704
Bram Moolenaar4029cab2020-12-05 18:13:27 +01002705def Test_dict_member_with_silent()
2706 var lines =<< trim END
2707 vim9script
2708 g:result = 'none'
2709 var d: dict<any>
2710 def Func()
2711 try
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01002712 g:result = map([], (_, v) => ({}[v]))->join() .. d['']
Bram Moolenaar4029cab2020-12-05 18:13:27 +01002713 catch
2714 endtry
2715 enddef
2716 silent! Func()
2717 assert_equal('0', g:result)
2718 unlet g:result
Bram Moolenaaraf0df472020-12-02 20:51:22 +01002719 END
2720 CheckScriptSuccess(lines)
2721enddef
2722
Bram Moolenaarf9041332021-01-21 19:41:16 +01002723def Test_skip_cmds_with_silent()
2724 var lines =<< trim END
2725 vim9script
2726
2727 def Func(b: bool)
2728 Crash()
2729 enddef
2730
2731 def Crash()
2732 sil! :/not found/d _
2733 sil! :/not found/put _
2734 enddef
2735
2736 Func(true)
2737 END
2738 CheckScriptSuccess(lines)
2739enddef
2740
Bram Moolenaar5b3d1bb2020-12-22 12:20:08 +01002741def Test_opfunc()
2742 nnoremap <F3> <cmd>set opfunc=Opfunc<cr>g@
2743 def g:Opfunc(_: any): string
2744 setline(1, 'ASDF')
2745 return ''
2746 enddef
2747 new
2748 setline(1, 'asdf')
2749 feedkeys("\<F3>$", 'x')
2750 assert_equal('ASDF', getline(1))
2751
2752 bwipe!
2753 nunmap <F3>
2754enddef
2755
Bram Moolenaar077a4232020-12-22 18:33:27 +01002756" this was crashing on exit
2757def Test_nested_lambda_in_closure()
2758 var lines =<< trim END
2759 vim9script
Bram Moolenaar227c58a2021-04-28 20:40:44 +02002760 command WriteDone writefile(['Done'], 'XnestedDone')
Bram Moolenaar077a4232020-12-22 18:33:27 +01002761 def Outer()
2762 def g:Inner()
2763 echo map([1, 2, 3], {_, v -> v + 1})
2764 enddef
2765 g:Inner()
2766 enddef
2767 defcompile
Bram Moolenaar227c58a2021-04-28 20:40:44 +02002768 # not reached
Bram Moolenaar077a4232020-12-22 18:33:27 +01002769 END
Bram Moolenaar227c58a2021-04-28 20:40:44 +02002770 if !RunVim([], lines, '--clean -c WriteDone -c quit')
Bram Moolenaar077a4232020-12-22 18:33:27 +01002771 return
2772 endif
2773 assert_equal(['Done'], readfile('XnestedDone'))
2774 delete('XnestedDone')
2775enddef
2776
Bram Moolenaar04947cc2021-03-06 19:26:46 +01002777def Test_check_func_arg_types()
2778 var lines =<< trim END
2779 vim9script
2780 def F1(x: string): string
2781 return x
2782 enddef
2783
2784 def F2(x: number): number
2785 return x + 1
2786 enddef
2787
2788 def G(g: func): dict<func>
2789 return {f: g}
2790 enddef
2791
2792 def H(d: dict<func>): string
2793 return d.f('a')
2794 enddef
2795 END
2796
2797 CheckScriptSuccess(lines + ['echo H(G(F1))'])
2798 CheckScriptFailure(lines + ['echo H(G(F2))'], 'E1013:')
2799enddef
2800
Bram Moolenaar701cc6c2021-04-10 13:33:48 +02002801def Test_compile_error()
2802 var lines =<< trim END
2803 def g:Broken()
2804 echo 'a' + {}
2805 enddef
2806 call g:Broken()
2807 END
2808 # First call: compilation error
2809 CheckScriptFailure(lines, 'E1051: Wrong argument type for +')
2810
2811 # Second call won't try compiling again
2812 assert_fails('call g:Broken()', 'E1091: Function is not compiled: Broken')
Bram Moolenaar599410c2021-04-10 14:03:43 +02002813 delfunc g:Broken
2814
2815 # No error when compiling with :silent!
2816 lines =<< trim END
2817 def g:Broken()
2818 echo 'a' + []
2819 enddef
2820 silent! defcompile
2821 END
2822 CheckScriptSuccess(lines)
2823
2824 # Calling the function won't try compiling again
2825 assert_fails('call g:Broken()', 'E1091: Function is not compiled: Broken')
2826 delfunc g:Broken
Bram Moolenaar701cc6c2021-04-10 13:33:48 +02002827enddef
2828
Bram Moolenaar962c43b2021-04-10 17:18:09 +02002829def Test_ignored_argument()
2830 var lines =<< trim END
2831 vim9script
2832 def Ignore(_, _): string
2833 return 'yes'
2834 enddef
2835 assert_equal('yes', Ignore(1, 2))
2836
2837 func Ok(_)
2838 return a:_
2839 endfunc
2840 assert_equal('ok', Ok('ok'))
2841
2842 func Oktoo()
2843 let _ = 'too'
2844 return _
2845 endfunc
2846 assert_equal('too', Oktoo())
Bram Moolenaarda479c72021-04-10 21:01:38 +02002847
2848 assert_equal([[1], [2], [3]], range(3)->mapnew((_, v) => [v]->map((_, w) => w + 1)))
Bram Moolenaar962c43b2021-04-10 17:18:09 +02002849 END
2850 CheckScriptSuccess(lines)
2851
2852 lines =<< trim END
2853 def Ignore(_: string): string
2854 return _
2855 enddef
2856 defcompile
2857 END
2858 CheckScriptFailure(lines, 'E1181:', 1)
2859
2860 lines =<< trim END
2861 var _ = 1
2862 END
2863 CheckDefAndScriptFailure(lines, 'E1181:', 1)
Yegappan Lakshmanan34fcb692021-05-25 20:14:00 +02002864
2865 lines =<< trim END
2866 var x = _
2867 END
2868 CheckDefAndScriptFailure(lines, 'E1181:', 1)
Bram Moolenaar962c43b2021-04-10 17:18:09 +02002869enddef
2870
Bram Moolenaarbb8a7ce2021-04-10 20:10:26 +02002871def Test_too_many_arguments()
2872 var lines =<< trim END
2873 echo [0, 1, 2]->map(() => 123)
2874 END
2875 CheckDefExecAndScriptFailure(lines, 'E1106: 2 arguments too many', 1)
2876
2877 lines =<< trim END
2878 echo [0, 1, 2]->map((_) => 123)
2879 END
2880 CheckDefExecAndScriptFailure(lines, 'E1106: One argument too many', 1)
2881enddef
Bram Moolenaar077a4232020-12-22 18:33:27 +01002882
Bram Moolenaara6aa1642021-04-23 19:32:23 +02002883def Test_closing_brace_at_start_of_line()
2884 var lines =<< trim END
2885 def Func()
2886 enddef
2887 Func(
2888 )
2889 END
2890 call CheckDefAndScriptSuccess(lines)
2891enddef
2892
Bram Moolenaar20677332021-06-06 17:02:53 +02002893if has('python3')
2894 def Test_python3_heredoc()
2895 py3 << trim EOF
2896 import vim
2897 vim.vars['didit'] = 'yes'
2898 EOF
2899 assert_equal('yes', g:didit)
2900
2901 python3 << trim EOF
2902 import vim
2903 vim.vars['didit'] = 'again'
2904 EOF
2905 assert_equal('again', g:didit)
2906 enddef
2907endif
2908
2909" This messes up syntax highlight, keep near the end.
2910if has('lua')
2911 def Test_lua_heredoc()
2912 g:d = {}
2913 lua << trim EOF
2914 x = vim.eval('g:d')
2915 x['key'] = 'val'
2916 EOF
2917 assert_equal('val', g:d.key)
2918 enddef
2919endif
2920
Bram Moolenaarf7779c62020-05-03 15:38:16 +02002921
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002922" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker