blob: eadd1d1c3f91389dd59db82759bdb5ef4d4f1696 [file] [log] [blame]
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001" Test various aspects of the Vim9 script language.
2
3source check.vim
Bram Moolenaarad304702020-09-06 18:22:53 +02004source term_util.vim
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02005source view_util.vim
Bram Moolenaar04b12692020-05-04 23:24:44 +02006source vim9.vim
Bram Moolenaar47e7d702020-07-05 18:18:42 +02007source screendump.vim
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02008
9func Test_def_basic()
10 def SomeFunc(): string
11 return 'yes'
12 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +020013 call SomeFunc()->assert_equal('yes')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +020014endfunc
15
Bram Moolenaar2b9b17e2020-10-13 18:38:11 +020016func Test_compiling_error()
17 " use a terminal to see the whole error message
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +020018 CheckRunVimInTerminal
19
Bram Moolenaar2b9b17e2020-10-13 18:38:11 +020020 call TestCompilingError()
21endfunc
22
23def TestCompilingError()
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +020024 var lines =<< trim END
25 vim9script
26 def Fails()
27 echo nothing
28 enddef
29 defcompile
30 END
31 call writefile(lines, 'XTest_compile_error')
32 var buf = RunVimInTerminal('-S XTest_compile_error',
Bram Moolenaare0de1712020-12-02 17:36:54 +010033 {rows: 10, wait_for_ruler: 0})
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +020034 var text = ''
35 for loop in range(100)
36 text = ''
37 for i in range(1, 9)
38 text ..= term_getline(buf, i)
39 endfor
Bram Moolenaar18dc3552020-11-22 14:24:00 +010040 if text =~ 'Variable not found: nothing'
Bram Moolenaarf4e8cdd2020-10-12 22:07:13 +020041 break
42 endif
43 sleep 20m
44 endfor
45 assert_match('Error detected while compiling command line.*Fails.*Variable not found: nothing', text)
46
47 # clean up
48 call StopVimInTerminal(buf)
49 call delete('XTest_compile_error')
50enddef
51
Bram Moolenaar0ba48e82020-11-17 18:23:19 +010052def CallRecursive(n: number): number
53 return CallRecursive(n + 1)
54enddef
55
56def CallMapRecursive(l: list<number>): number
57 return map(l, {_, v -> CallMapRecursive([v])})[0]
58enddef
59
60def Test_funcdepth_error()
61 set maxfuncdepth=10
62
63 var caught = false
64 try
65 CallRecursive(1)
66 catch /E132:/
67 caught = true
68 endtry
69 assert_true(caught)
70
71 caught = false
72 try
73 CallMapRecursive([1])
74 catch /E132:/
75 caught = true
76 endtry
77 assert_true(caught)
78
79 set maxfuncdepth&
80enddef
81
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +020082def ReturnString(): string
83 return 'string'
84enddef
85
86def ReturnNumber(): number
87 return 123
88enddef
89
90let g:notNumber = 'string'
91
92def ReturnGlobal(): number
93 return g:notNumber
94enddef
95
96def Test_return_something()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +020097 ReturnString()->assert_equal('string')
98 ReturnNumber()->assert_equal(123)
Bram Moolenaar5e654232020-09-16 15:22:00 +020099 assert_fails('ReturnGlobal()', 'E1012: Type mismatch; expected number but got string', '', 1, 'ReturnGlobal')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200100enddef
101
Bram Moolenaarefd88552020-06-18 20:50:10 +0200102def Test_missing_return()
103 CheckDefFailure(['def Missing(): number',
104 ' if g:cond',
105 ' echo "no return"',
106 ' else',
107 ' return 0',
108 ' endif'
109 'enddef'], 'E1027:')
110 CheckDefFailure(['def Missing(): number',
111 ' if g:cond',
112 ' return 1',
113 ' else',
114 ' echo "no return"',
115 ' endif'
116 'enddef'], 'E1027:')
117 CheckDefFailure(['def Missing(): number',
118 ' if g:cond',
119 ' return 1',
120 ' else',
121 ' return 2',
122 ' endif'
123 ' return 3'
124 'enddef'], 'E1095:')
125enddef
126
Bram Moolenaar403dc312020-10-17 19:29:51 +0200127def Test_return_bool()
128 var lines =<< trim END
129 vim9script
130 def MenuFilter(id: number, key: string): bool
131 return popup_filter_menu(id, key)
132 enddef
133 def YesnoFilter(id: number, key: string): bool
134 return popup_filter_yesno(id, key)
135 enddef
136 defcompile
137 END
138 CheckScriptSuccess(lines)
139enddef
140
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200141let s:nothing = 0
142def ReturnNothing()
143 s:nothing = 1
144 if true
145 return
146 endif
147 s:nothing = 2
148enddef
149
150def Test_return_nothing()
151 ReturnNothing()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200152 s:nothing->assert_equal(1)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200153enddef
154
155func Increment()
156 let g:counter += 1
157endfunc
158
159def Test_call_ufunc_count()
160 g:counter = 1
161 Increment()
162 Increment()
163 Increment()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200164 # works with and without :call
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200165 g:counter->assert_equal(4)
166 eval g:counter->assert_equal(4)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200167 unlet g:counter
168enddef
169
170def MyVarargs(arg: string, ...rest: list<string>): string
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200171 var res = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200172 for s in rest
173 res ..= ',' .. s
174 endfor
175 return res
176enddef
177
178def Test_call_varargs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200179 MyVarargs('one')->assert_equal('one')
180 MyVarargs('one', 'two')->assert_equal('one,two')
181 MyVarargs('one', 'two', 'three')->assert_equal('one,two,three')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200182enddef
183
184def MyDefaultArgs(name = 'string'): string
185 return name
186enddef
187
Bram Moolenaare30f64b2020-07-15 19:48:20 +0200188def MyDefaultSecond(name: string, second: bool = true): string
189 return second ? name : 'none'
190enddef
191
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200192def Test_call_default_args()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200193 MyDefaultArgs()->assert_equal('string')
194 MyDefaultArgs('one')->assert_equal('one')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200195 assert_fails('MyDefaultArgs("one", "two")', 'E118:', '', 3, 'Test_call_default_args')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200196
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200197 MyDefaultSecond('test')->assert_equal('test')
198 MyDefaultSecond('test', true)->assert_equal('test')
199 MyDefaultSecond('test', false)->assert_equal('none')
Bram Moolenaare30f64b2020-07-15 19:48:20 +0200200
Bram Moolenaar822ba242020-05-24 23:00:18 +0200201 CheckScriptFailure(['def Func(arg: number = asdf)', 'enddef', 'defcompile'], 'E1001:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100202 delfunc g:Func
Bram Moolenaar77072282020-09-16 17:55:40 +0200203 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 +0100204 delfunc g:Func
Bram Moolenaar04b12692020-05-04 23:24:44 +0200205enddef
206
207def Test_nested_function()
208 def Nested(arg: string): string
209 return 'nested ' .. arg
210 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200211 Nested('function')->assert_equal('nested function')
Bram Moolenaar04b12692020-05-04 23:24:44 +0200212
Bram Moolenaar0e65d3d2020-05-05 17:53:16 +0200213 CheckDefFailure(['def Nested()', 'enddef', 'Nested(66)'], 'E118:')
214 CheckDefFailure(['def Nested(arg: string)', 'enddef', 'Nested()'], 'E119:')
215
Bram Moolenaar04b12692020-05-04 23:24:44 +0200216 CheckDefFailure(['func Nested()', 'endfunc'], 'E1086:')
Bram Moolenaarbcbf4132020-08-01 22:35:13 +0200217 CheckDefFailure(['def s:Nested()', 'enddef'], 'E1075:')
218 CheckDefFailure(['def b:Nested()', 'enddef'], 'E1075:')
Bram Moolenaar8b848ca2020-09-10 22:28:01 +0200219
Bram Moolenaar54021752020-12-06 18:50:36 +0100220 var lines =<< trim END
221 def Outer()
222 def Inner()
223 # comment
224 enddef
225 def Inner()
226 enddef
227 enddef
228 END
229 CheckDefFailure(lines, 'E1073:')
230
231 lines =<< trim END
232 def Outer()
233 def Inner()
234 # comment
235 enddef
236 def! Inner()
237 enddef
238 enddef
239 END
240 CheckDefFailure(lines, 'E1117:')
241
242 # nested function inside conditional
243 # TODO: should it work when "thecount" is inside the "if"?
244 lines =<< trim END
245 vim9script
246 var thecount = 0
247 if true
248 def Test(): number
249 def TheFunc(): number
250 thecount += 1
251 return thecount
252 enddef
253 return TheFunc()
254 enddef
255 endif
256 defcompile
257 assert_equal(1, Test())
258 assert_equal(2, Test())
259 END
260 CheckScriptSuccess(lines)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200261enddef
262
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200263func Test_call_default_args_from_func()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200264 call MyDefaultArgs()->assert_equal('string')
265 call MyDefaultArgs('one')->assert_equal('one')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200266 call assert_fails('call MyDefaultArgs("one", "two")', 'E118:', '', 3, 'Test_call_default_args_from_func')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200267endfunc
268
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200269def Test_nested_global_function()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200270 var lines =<< trim END
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200271 vim9script
272 def Outer()
273 def g:Inner(): string
274 return 'inner'
275 enddef
276 enddef
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200277 defcompile
278 Outer()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200279 g:Inner()->assert_equal('inner')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200280 delfunc g:Inner
281 Outer()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200282 g:Inner()->assert_equal('inner')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200283 delfunc g:Inner
284 Outer()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200285 g:Inner()->assert_equal('inner')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200286 delfunc g:Inner
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200287 END
288 CheckScriptSuccess(lines)
Bram Moolenaar2c79e9d2020-08-01 18:57:52 +0200289
290 lines =<< trim END
291 vim9script
292 def Outer()
293 def g:Inner(): string
294 return 'inner'
295 enddef
296 enddef
297 defcompile
298 Outer()
299 Outer()
300 END
301 CheckScriptFailure(lines, "E122:")
Bram Moolenaarcd45ed02020-12-22 17:35:54 +0100302 delfunc g:Inner
Bram Moolenaarad486a02020-08-01 23:22:18 +0200303
304 lines =<< trim END
305 vim9script
Bram Moolenaar58a52f22020-12-22 18:56:55 +0100306 def Outer()
307 def g:Inner()
308 echo map([1, 2, 3], {_, v -> v + 1})
309 enddef
310 g:Inner()
311 enddef
312 Outer()
313 END
314 CheckScriptSuccess(lines)
315 delfunc g:Inner
316
317 lines =<< trim END
318 vim9script
Bram Moolenaarad486a02020-08-01 23:22:18 +0200319 def Func()
320 echo 'script'
321 enddef
322 def Outer()
323 def Func()
324 echo 'inner'
325 enddef
326 enddef
327 defcompile
328 END
329 CheckScriptFailure(lines, "E1073:")
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200330enddef
331
Bram Moolenaar6abdcf82020-11-22 18:15:44 +0100332def DefListAll()
333 def
334enddef
335
336def DefListOne()
337 def DefListOne
338enddef
339
340def DefListMatches()
341 def /DefList
342enddef
343
344def Test_nested_def_list()
345 var funcs = split(execute('call DefListAll()'), "\n")
346 assert_true(len(funcs) > 10)
347 assert_true(funcs->index('def DefListAll()') >= 0)
348
349 funcs = split(execute('call DefListOne()'), "\n")
350 assert_equal([' def DefListOne()', '1 def DefListOne', ' enddef'], funcs)
351
352 funcs = split(execute('call DefListMatches()'), "\n")
353 assert_true(len(funcs) >= 3)
354 assert_true(funcs->index('def DefListAll()') >= 0)
355 assert_true(funcs->index('def DefListOne()') >= 0)
356 assert_true(funcs->index('def DefListMatches()') >= 0)
Bram Moolenaar54021752020-12-06 18:50:36 +0100357
358 var lines =<< trim END
359 vim9script
360 def Func()
361 def +Func+
362 enddef
363 defcompile
364 END
365 CheckScriptFailure(lines, 'E476:', 1)
Bram Moolenaar6abdcf82020-11-22 18:15:44 +0100366enddef
367
Bram Moolenaar333894b2020-08-01 18:53:07 +0200368def Test_global_local_function()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200369 var lines =<< trim END
Bram Moolenaar333894b2020-08-01 18:53:07 +0200370 vim9script
371 def g:Func(): string
372 return 'global'
373 enddef
374 def Func(): string
375 return 'local'
376 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200377 g:Func()->assert_equal('global')
378 Func()->assert_equal('local')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100379 delfunc g:Func
Bram Moolenaar333894b2020-08-01 18:53:07 +0200380 END
381 CheckScriptSuccess(lines)
Bram Moolenaar035d6e92020-08-11 22:30:42 +0200382
383 lines =<< trim END
384 vim9script
385 def g:Funcy()
386 echo 'funcy'
387 enddef
388 s:Funcy()
389 END
390 CheckScriptFailure(lines, 'E117:')
Bram Moolenaar333894b2020-08-01 18:53:07 +0200391enddef
392
Bram Moolenaar0f769812020-09-12 18:32:34 +0200393def Test_local_function_shadows_global()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200394 var lines =<< trim END
Bram Moolenaar0f769812020-09-12 18:32:34 +0200395 vim9script
396 def g:Gfunc(): string
397 return 'global'
398 enddef
399 def AnotherFunc(): number
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200400 var Gfunc = function('len')
Bram Moolenaar0f769812020-09-12 18:32:34 +0200401 return Gfunc('testing')
402 enddef
403 g:Gfunc()->assert_equal('global')
404 AnotherFunc()->assert_equal(7)
405 delfunc g:Gfunc
406 END
407 CheckScriptSuccess(lines)
408
409 lines =<< trim END
410 vim9script
411 def g:Func(): string
412 return 'global'
413 enddef
414 def AnotherFunc()
415 g:Func = function('len')
416 enddef
417 AnotherFunc()
418 END
419 CheckScriptFailure(lines, 'E705:')
420 delfunc g:Func
421enddef
422
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200423func TakesOneArg(arg)
424 echo a:arg
425endfunc
426
427def Test_call_wrong_args()
Bram Moolenaard2c61702020-09-06 15:58:36 +0200428 CheckDefFailure(['TakesOneArg()'], 'E119:')
429 CheckDefFailure(['TakesOneArg(11, 22)'], 'E118:')
430 CheckDefFailure(['bufnr(xxx)'], 'E1001:')
431 CheckScriptFailure(['def Func(Ref: func(s: string))'], 'E475:')
Bram Moolenaaree8580e2020-08-28 17:19:07 +0200432
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200433 var lines =<< trim END
Bram Moolenaaree8580e2020-08-28 17:19:07 +0200434 vim9script
435 def Func(s: string)
436 echo s
437 enddef
438 Func([])
439 END
Bram Moolenaar77072282020-09-16 17:55:40 +0200440 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected string but got list<unknown>', 5)
Bram Moolenaarb185a402020-09-18 22:42:00 +0200441
442 lines =<< trim END
443 vim9script
444 def FuncOne(nr: number)
445 echo nr
446 enddef
447 def FuncTwo()
448 FuncOne()
449 enddef
450 defcompile
451 END
452 writefile(lines, 'Xscript')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200453 var didCatch = false
Bram Moolenaarb185a402020-09-18 22:42:00 +0200454 try
455 source Xscript
456 catch
457 assert_match('E119: Not enough arguments for function: <SNR>\d\+_FuncOne', v:exception)
458 assert_match('Xscript\[8\]..function <SNR>\d\+_FuncTwo, line 1', v:throwpoint)
459 didCatch = true
460 endtry
461 assert_true(didCatch)
462
463 lines =<< trim END
464 vim9script
465 def FuncOne(nr: number)
466 echo nr
467 enddef
468 def FuncTwo()
469 FuncOne(1, 2)
470 enddef
471 defcompile
472 END
473 writefile(lines, 'Xscript')
474 didCatch = false
475 try
476 source Xscript
477 catch
478 assert_match('E118: Too many arguments for function: <SNR>\d\+_FuncOne', v:exception)
479 assert_match('Xscript\[8\]..function <SNR>\d\+_FuncTwo, line 1', v:throwpoint)
480 didCatch = true
481 endtry
482 assert_true(didCatch)
483
484 delete('Xscript')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200485enddef
486
Bram Moolenaar50824712020-12-20 21:10:17 +0100487def Test_call_funcref_wrong_args()
488 var head =<< trim END
489 vim9script
490 def Func3(a1: string, a2: number, a3: list<number>)
491 echo a1 .. a2 .. a3[0]
492 enddef
493 def Testme()
494 var funcMap: dict<func> = {func: Func3}
495 END
496 var tail =<< trim END
497 enddef
498 Testme()
499 END
500 CheckScriptSuccess(head + ["funcMap['func']('str', 123, [1, 2, 3])"] + tail)
501
502 CheckScriptFailure(head + ["funcMap['func']('str', 123)"] + tail, 'E119:')
503 CheckScriptFailure(head + ["funcMap['func']('str', 123, [1], 4)"] + tail, 'E118:')
504enddef
505
Bram Moolenaarb4d16cb2020-11-05 18:45:46 +0100506def Test_call_lambda_args()
507 CheckDefFailure(['echo {i -> 0}()'],
508 'E119: Not enough arguments for function: {i -> 0}()')
509
510 var lines =<< trim END
511 var Ref = {x: number, y: number -> x + y}
512 echo Ref(1, 'x')
513 END
514 CheckDefFailure(lines, 'E1013: Argument 2: type mismatch, expected number but got string')
515enddef
516
Bram Moolenaar709664c2020-12-12 14:33:41 +0100517def Test_lambda_uses_assigned_var()
518 CheckDefSuccess([
519 'var x: any = "aaa"'
520 'x = filter(["bbb"], {_, v -> v =~ x})'])
521enddef
522
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200523" Default arg and varargs
524def MyDefVarargs(one: string, two = 'foo', ...rest: list<string>): string
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200525 var res = one .. ',' .. two
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200526 for s in rest
527 res ..= ',' .. s
528 endfor
529 return res
530enddef
531
532def Test_call_def_varargs()
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200533 assert_fails('MyDefVarargs()', 'E119:', '', 1, 'Test_call_def_varargs')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200534 MyDefVarargs('one')->assert_equal('one,foo')
535 MyDefVarargs('one', 'two')->assert_equal('one,two')
536 MyDefVarargs('one', 'two', 'three')->assert_equal('one,two,three')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200537 CheckDefFailure(['MyDefVarargs("one", 22)'],
Bram Moolenaar77072282020-09-16 17:55:40 +0200538 'E1013: Argument 2: type mismatch, expected string but got number')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200539 CheckDefFailure(['MyDefVarargs("one", "two", 123)'],
Bram Moolenaar77072282020-09-16 17:55:40 +0200540 'E1013: Argument 3: type mismatch, expected string but got number')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200541
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200542 var lines =<< trim END
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200543 vim9script
544 def Func(...l: list<string>)
545 echo l
546 enddef
547 Func('a', 'b', 'c')
548 END
549 CheckScriptSuccess(lines)
550
551 lines =<< trim END
552 vim9script
553 def Func(...l: list<string>)
554 echo l
555 enddef
556 Func()
557 END
558 CheckScriptSuccess(lines)
559
560 lines =<< trim END
561 vim9script
Bram Moolenaar2f8cbc42020-09-16 17:22:59 +0200562 def Func(...l: any)
563 echo l
564 enddef
565 Func(0)
566 END
567 CheckScriptSuccess(lines)
568
569 lines =<< trim END
570 vim9script
Bram Moolenaar28022722020-09-21 22:02:49 +0200571 def Func(..._l: list<string>)
572 echo _l
573 enddef
574 Func('a', 'b', 'c')
575 END
576 CheckScriptSuccess(lines)
577
578 lines =<< trim END
579 vim9script
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200580 def Func(...l: list<string>)
581 echo l
582 enddef
583 Func(1, 2, 3)
584 END
Bram Moolenaar77072282020-09-16 17:55:40 +0200585 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200586
587 lines =<< trim END
588 vim9script
589 def Func(...l: list<string>)
590 echo l
591 enddef
592 Func('a', 9)
593 END
Bram Moolenaar77072282020-09-16 17:55:40 +0200594 CheckScriptFailure(lines, 'E1013: Argument 2: type mismatch')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200595
596 lines =<< trim END
597 vim9script
598 def Func(...l: list<string>)
599 echo l
600 enddef
601 Func(1, 'a')
602 END
Bram Moolenaar77072282020-09-16 17:55:40 +0200603 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200604enddef
605
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200606let s:value = ''
607
608def FuncOneDefArg(opt = 'text')
609 s:value = opt
610enddef
611
612def FuncTwoDefArg(nr = 123, opt = 'text'): string
613 return nr .. opt
614enddef
615
616def FuncVarargs(...arg: list<string>): string
617 return join(arg, ',')
618enddef
619
620def Test_func_type_varargs()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200621 var RefDefArg: func(?string)
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200622 RefDefArg = FuncOneDefArg
623 RefDefArg()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200624 s:value->assert_equal('text')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200625 RefDefArg('some')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200626 s:value->assert_equal('some')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200627
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200628 var RefDef2Arg: func(?number, ?string): string
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200629 RefDef2Arg = FuncTwoDefArg
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200630 RefDef2Arg()->assert_equal('123text')
631 RefDef2Arg(99)->assert_equal('99text')
632 RefDef2Arg(77, 'some')->assert_equal('77some')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200633
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200634 CheckDefFailure(['var RefWrong: func(string?)'], 'E1010:')
635 CheckDefFailure(['var RefWrong: func(?string, string)'], 'E1007:')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200636
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200637 var RefVarargs: func(...list<string>): string
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200638 RefVarargs = FuncVarargs
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200639 RefVarargs()->assert_equal('')
640 RefVarargs('one')->assert_equal('one')
641 RefVarargs('one', 'two')->assert_equal('one,two')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200642
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200643 CheckDefFailure(['var RefWrong: func(...list<string>, string)'], 'E110:')
644 CheckDefFailure(['var RefWrong: func(...list<string>, ?string)'], 'E110:')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200645enddef
646
Bram Moolenaar0b76b422020-04-07 22:05:08 +0200647" Only varargs
648def MyVarargsOnly(...args: list<string>): string
649 return join(args, ',')
650enddef
651
652def Test_call_varargs_only()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200653 MyVarargsOnly()->assert_equal('')
654 MyVarargsOnly('one')->assert_equal('one')
655 MyVarargsOnly('one', 'two')->assert_equal('one,two')
Bram Moolenaar77072282020-09-16 17:55:40 +0200656 CheckDefFailure(['MyVarargsOnly(1)'], 'E1013: Argument 1: type mismatch, expected string but got number')
657 CheckDefFailure(['MyVarargsOnly("one", 2)'], 'E1013: Argument 2: type mismatch, expected string but got number')
Bram Moolenaar0b76b422020-04-07 22:05:08 +0200658enddef
659
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200660def Test_using_var_as_arg()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200661 writefile(['def Func(x: number)', 'var x = 234', 'enddef', 'defcompile'], 'Xdef')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200662 assert_fails('so Xdef', 'E1006:', '', 1, 'Func')
Bram Moolenaard2c61702020-09-06 15:58:36 +0200663 delete('Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200664enddef
665
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200666def DictArg(arg: dict<string>)
667 arg['key'] = 'value'
668enddef
669
670def ListArg(arg: list<string>)
671 arg[0] = 'value'
672enddef
673
674def Test_assign_to_argument()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200675 # works for dict and list
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200676 var d: dict<string> = {}
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200677 DictArg(d)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200678 d['key']->assert_equal('value')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200679 var l: list<string> = []
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200680 ListArg(l)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200681 l[0]->assert_equal('value')
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200682
Bram Moolenaard2c61702020-09-06 15:58:36 +0200683 CheckScriptFailure(['def Func(arg: number)', 'arg = 3', 'enddef', 'defcompile'], 'E1090:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100684 delfunc! g:Func
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200685enddef
686
Bram Moolenaarb816dae2020-09-20 22:04:00 +0200687" These argument names are reserved in legacy functions.
688def WithReservedNames(firstline: string, lastline: string): string
689 return firstline .. lastline
690enddef
691
692def Test_argument_names()
693 assert_equal('OK', WithReservedNames('O', 'K'))
694enddef
695
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200696def Test_call_func_defined_later()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200697 g:DefinedLater('one')->assert_equal('one')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200698 assert_fails('NotDefined("one")', 'E117:', '', 2, 'Test_call_func_defined_later')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200699enddef
700
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200701func DefinedLater(arg)
702 return a:arg
703endfunc
704
705def Test_call_funcref()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200706 g:SomeFunc('abc')->assert_equal(3)
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200707 assert_fails('NotAFunc()', 'E117:', '', 2, 'Test_call_funcref') # comment after call
708 assert_fails('g:NotAFunc()', 'E117:', '', 3, 'Test_call_funcref')
Bram Moolenaar2f1980f2020-07-22 19:30:06 +0200709
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200710 var lines =<< trim END
Bram Moolenaar2f1980f2020-07-22 19:30:06 +0200711 vim9script
712 def RetNumber(): number
713 return 123
714 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200715 var Funcref: func: number = function('RetNumber')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200716 Funcref()->assert_equal(123)
Bram Moolenaar2f1980f2020-07-22 19:30:06 +0200717 END
718 CheckScriptSuccess(lines)
Bram Moolenaar0f60e802020-07-22 20:16:11 +0200719
720 lines =<< trim END
721 vim9script
722 def RetNumber(): number
723 return 123
724 enddef
725 def Bar(F: func: number): number
726 return F()
727 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200728 var Funcref = function('RetNumber')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200729 Bar(Funcref)->assert_equal(123)
Bram Moolenaar0f60e802020-07-22 20:16:11 +0200730 END
731 CheckScriptSuccess(lines)
Bram Moolenaarbfba8652020-07-23 20:09:10 +0200732
733 lines =<< trim END
734 vim9script
735 def UseNumber(nr: number)
736 echo nr
737 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200738 var Funcref: func(number) = function('UseNumber')
Bram Moolenaarbfba8652020-07-23 20:09:10 +0200739 Funcref(123)
740 END
741 CheckScriptSuccess(lines)
Bram Moolenaarb8070e32020-07-23 20:56:04 +0200742
743 lines =<< trim END
744 vim9script
745 def UseNumber(nr: number)
746 echo nr
747 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200748 var Funcref: func(string) = function('UseNumber')
Bram Moolenaarb8070e32020-07-23 20:56:04 +0200749 END
Bram Moolenaar5e654232020-09-16 15:22:00 +0200750 CheckScriptFailure(lines, 'E1012: Type mismatch; expected func(string) but got func(number)')
Bram Moolenaar4fc224c2020-07-26 17:56:25 +0200751
752 lines =<< trim END
753 vim9script
754 def EchoNr(nr = 34)
755 g:echo = nr
756 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200757 var Funcref: func(?number) = function('EchoNr')
Bram Moolenaar4fc224c2020-07-26 17:56:25 +0200758 Funcref()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200759 g:echo->assert_equal(34)
Bram Moolenaar4fc224c2020-07-26 17:56:25 +0200760 Funcref(123)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200761 g:echo->assert_equal(123)
Bram Moolenaar4fc224c2020-07-26 17:56:25 +0200762 END
763 CheckScriptSuccess(lines)
Bram Moolenaarace61322020-07-26 18:16:58 +0200764
765 lines =<< trim END
766 vim9script
767 def EchoList(...l: list<number>)
768 g:echo = l
769 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200770 var Funcref: func(...list<number>) = function('EchoList')
Bram Moolenaarace61322020-07-26 18:16:58 +0200771 Funcref()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200772 g:echo->assert_equal([])
Bram Moolenaarace61322020-07-26 18:16:58 +0200773 Funcref(1, 2, 3)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200774 g:echo->assert_equal([1, 2, 3])
Bram Moolenaarace61322020-07-26 18:16:58 +0200775 END
776 CheckScriptSuccess(lines)
Bram Moolenaar01865ad2020-07-26 18:33:09 +0200777
778 lines =<< trim END
779 vim9script
780 def OptAndVar(nr: number, opt = 12, ...l: list<number>): number
781 g:optarg = opt
782 g:listarg = l
783 return nr
784 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200785 var Funcref: func(number, ?number, ...list<number>): number = function('OptAndVar')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200786 Funcref(10)->assert_equal(10)
787 g:optarg->assert_equal(12)
788 g:listarg->assert_equal([])
Bram Moolenaar01865ad2020-07-26 18:33:09 +0200789
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200790 Funcref(11, 22)->assert_equal(11)
791 g:optarg->assert_equal(22)
792 g:listarg->assert_equal([])
Bram Moolenaar01865ad2020-07-26 18:33:09 +0200793
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200794 Funcref(17, 18, 1, 2, 3)->assert_equal(17)
795 g:optarg->assert_equal(18)
796 g:listarg->assert_equal([1, 2, 3])
Bram Moolenaar01865ad2020-07-26 18:33:09 +0200797 END
798 CheckScriptSuccess(lines)
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200799enddef
800
801let SomeFunc = function('len')
802let NotAFunc = 'text'
803
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200804def CombineFuncrefTypes()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200805 # same arguments, different return type
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200806 var Ref1: func(bool): string
807 var Ref2: func(bool): number
808 var Ref3: func(bool): any
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200809 Ref3 = g:cond ? Ref1 : Ref2
810
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200811 # different number of arguments
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200812 var Refa1: func(bool): number
813 var Refa2: func(bool, number): number
814 var Refa3: func: number
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200815 Refa3 = g:cond ? Refa1 : Refa2
816
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200817 # different argument types
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200818 var Refb1: func(bool, string): number
819 var Refb2: func(string, number): number
820 var Refb3: func(any, any): number
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200821 Refb3 = g:cond ? Refb1 : Refb2
822enddef
823
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200824def FuncWithForwardCall()
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200825 return g:DefinedEvenLater("yes")
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200826enddef
827
828def DefinedEvenLater(arg: string): string
829 return arg
830enddef
831
832def Test_error_in_nested_function()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200833 # Error in called function requires unwinding the call stack.
Bram Moolenaar44d66522020-09-06 22:26:57 +0200834 assert_fails('FuncWithForwardCall()', 'E1096:', '', 1, 'FuncWithForwardCall')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200835enddef
836
837def Test_return_type_wrong()
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200838 CheckScriptFailure([
839 'def Func(): number',
840 'return "a"',
841 'enddef',
842 'defcompile'], 'expected number but got string')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100843 delfunc! g:Func
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200844 CheckScriptFailure([
845 'def Func(): string',
846 'return 1',
847 'enddef',
848 'defcompile'], 'expected string but got number')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100849 delfunc! g:Func
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200850 CheckScriptFailure([
851 'def Func(): void',
852 'return "a"',
853 'enddef',
854 'defcompile'],
855 'E1096: Returning a value in a function without a return type')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100856 delfunc! g:Func
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200857 CheckScriptFailure([
858 'def Func()',
859 'return "a"',
860 'enddef',
861 'defcompile'],
862 'E1096: Returning a value in a function without a return type')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100863 delfunc! g:Func
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200864
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200865 CheckScriptFailure([
866 'def Func(): number',
867 'return',
868 'enddef',
869 'defcompile'], 'E1003:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100870 delfunc! g:Func
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200871
872 CheckScriptFailure(['def Func(): list', 'return []', 'enddef'], 'E1008:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100873 delfunc! g:Func
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200874 CheckScriptFailure(['def Func(): dict', 'return {}', 'enddef'], 'E1008:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100875 delfunc! g:Func
Bram Moolenaaree4e0c12020-04-06 21:35:05 +0200876 CheckScriptFailure(['def Func()', 'return 1'], 'E1057:')
Bram Moolenaar2d870f82020-12-05 13:41:01 +0100877 delfunc! g:Func
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200878
879 CheckScriptFailure([
880 'vim9script',
881 'def FuncB()',
882 ' return 123',
883 'enddef',
884 'def FuncA()',
885 ' FuncB()',
886 'enddef',
887 'defcompile'], 'E1096:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200888enddef
889
890def Test_arg_type_wrong()
891 CheckScriptFailure(['def Func3(items: list)', 'echo "a"', 'enddef'], 'E1008: Missing <type>')
Bram Moolenaaree4e0c12020-04-06 21:35:05 +0200892 CheckScriptFailure(['def Func4(...)', 'echo "a"', 'enddef'], 'E1055: Missing name after ...')
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +0200893 CheckScriptFailure(['def Func5(items:string)', 'echo "a"'], 'E1069:')
Bram Moolenaar6e949782020-04-13 17:21:00 +0200894 CheckScriptFailure(['def Func5(items)', 'echo "a"'], 'E1077:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200895enddef
896
897def Test_vim9script_call()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200898 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200899 vim9script
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200900 var name = ''
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200901 def MyFunc(arg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200902 name = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200903 enddef
904 MyFunc('foobar')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200905 name->assert_equal('foobar')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200906
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200907 var str = 'barfoo'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200908 str->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200909 name->assert_equal('barfoo')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200910
Bram Moolenaar67979662020-06-20 22:50:47 +0200911 g:value = 'value'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200912 g:value->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200913 name->assert_equal('value')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200914
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200915 var listvar = []
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200916 def ListFunc(arg: list<number>)
917 listvar = arg
918 enddef
919 [1, 2, 3]->ListFunc()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200920 listvar->assert_equal([1, 2, 3])
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200921
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200922 var dictvar = {}
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200923 def DictFunc(arg: dict<number>)
924 dictvar = arg
925 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +0100926 {a: 1, b: 2}->DictFunc()
927 dictvar->assert_equal({a: 1, b: 2})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200928 def CompiledDict()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100929 {a: 3, b: 4}->DictFunc()
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200930 enddef
931 CompiledDict()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100932 dictvar->assert_equal({a: 3, b: 4})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200933
Bram Moolenaare0de1712020-12-02 17:36:54 +0100934 {a: 3, b: 4}->DictFunc()
935 dictvar->assert_equal({a: 3, b: 4})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200936
937 ('text')->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200938 name->assert_equal('text')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200939 ("some")->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200940 name->assert_equal('some')
Bram Moolenaare6b53242020-07-01 17:28:33 +0200941
Bram Moolenaar13e12b82020-07-24 18:47:22 +0200942 # line starting with single quote is not a mark
Bram Moolenaar10409562020-07-29 20:00:38 +0200943 # line starting with double quote can be a method call
Bram Moolenaar3d48e252020-07-15 14:15:52 +0200944 'asdfasdf'->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200945 name->assert_equal('asdfasdf')
Bram Moolenaar10409562020-07-29 20:00:38 +0200946 "xyz"->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200947 name->assert_equal('xyz')
Bram Moolenaar3d48e252020-07-15 14:15:52 +0200948
949 def UseString()
950 'xyork'->MyFunc()
951 enddef
952 UseString()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200953 name->assert_equal('xyork')
Bram Moolenaar3d48e252020-07-15 14:15:52 +0200954
Bram Moolenaar10409562020-07-29 20:00:38 +0200955 def UseString2()
956 "knife"->MyFunc()
957 enddef
958 UseString2()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200959 name->assert_equal('knife')
Bram Moolenaar10409562020-07-29 20:00:38 +0200960
Bram Moolenaar13e12b82020-07-24 18:47:22 +0200961 # prepending a colon makes it a mark
962 new
963 setline(1, ['aaa', 'bbb', 'ccc'])
964 normal! 3Gmt1G
965 :'t
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200966 getcurpos()[1]->assert_equal(3)
Bram Moolenaar13e12b82020-07-24 18:47:22 +0200967 bwipe!
968
Bram Moolenaare6b53242020-07-01 17:28:33 +0200969 MyFunc(
970 'continued'
971 )
972 assert_equal('continued',
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200973 name
Bram Moolenaare6b53242020-07-01 17:28:33 +0200974 )
975
976 call MyFunc(
977 'more'
978 ..
979 'lines'
980 )
981 assert_equal(
982 'morelines',
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200983 name)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200984 END
985 writefile(lines, 'Xcall.vim')
986 source Xcall.vim
987 delete('Xcall.vim')
988enddef
989
990def Test_vim9script_call_fail_decl()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200991 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200992 vim9script
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200993 var name = ''
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200994 def MyFunc(arg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200995 var name = 123
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200996 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +0200997 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200998 END
Bram Moolenaar6c4bfe42020-07-23 18:26:30 +0200999 CheckScriptFailure(lines, 'E1054:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001000enddef
1001
Bram Moolenaar65b95452020-07-19 14:03:09 +02001002def Test_vim9script_call_fail_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001003 var lines =<< trim END
Bram Moolenaar65b95452020-07-19 14:03:09 +02001004 vim9script
1005 def MyFunc(arg: string)
1006 echo arg
1007 enddef
1008 MyFunc(1234)
1009 END
Bram Moolenaar77072282020-09-16 17:55:40 +02001010 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected string but got number')
Bram Moolenaar65b95452020-07-19 14:03:09 +02001011enddef
1012
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001013def Test_vim9script_call_fail_const()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001014 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001015 vim9script
1016 const var = ''
1017 def MyFunc(arg: string)
1018 var = 'asdf'
1019 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +02001020 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001021 END
1022 writefile(lines, 'Xcall_const.vim')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001023 assert_fails('source Xcall_const.vim', 'E46:', '', 1, 'MyFunc')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001024 delete('Xcall_const.vim')
1025enddef
1026
1027" Test that inside :function a Python function can be defined, :def is not
1028" recognized.
1029func Test_function_python()
1030 CheckFeature python3
Bram Moolenaar727345e2020-09-27 23:33:59 +02001031 let py = 'python3'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001032 execute py "<< EOF"
1033def do_something():
1034 return 1
1035EOF
1036endfunc
1037
1038def Test_delfunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001039 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001040 vim9script
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02001041 def g:GoneSoon()
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001042 echo 'hello'
1043 enddef
1044
1045 def CallGoneSoon()
1046 GoneSoon()
1047 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +02001048 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001049
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02001050 delfunc g:GoneSoon
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001051 CallGoneSoon()
1052 END
1053 writefile(lines, 'XToDelFunc')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001054 assert_fails('so XToDelFunc', 'E933:', '', 1, 'CallGoneSoon')
1055 assert_fails('so XToDelFunc', 'E933:', '', 1, 'CallGoneSoon')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001056
1057 delete('XToDelFunc')
1058enddef
1059
1060def Test_redef_failure()
Bram Moolenaard2c61702020-09-06 15:58:36 +02001061 writefile(['def Func0(): string', 'return "Func0"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001062 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +02001063 writefile(['def Func1(): string', 'return "Func1"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001064 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +02001065 writefile(['def! Func0(): string', 'enddef', 'defcompile'], 'Xdef')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001066 assert_fails('so Xdef', 'E1027:', '', 1, 'Func0')
Bram Moolenaard2c61702020-09-06 15:58:36 +02001067 writefile(['def Func2(): string', 'return "Func2"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001068 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +02001069 delete('Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001070
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001071 g:Func0()->assert_equal(0)
1072 g:Func1()->assert_equal('Func1')
1073 g:Func2()->assert_equal('Func2')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001074
1075 delfunc! Func0
1076 delfunc! Func1
1077 delfunc! Func2
1078enddef
1079
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +02001080def Test_vim9script_func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001081 var lines =<< trim END
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +02001082 vim9script
1083 func Func(arg)
1084 echo a:arg
1085 endfunc
1086 Func('text')
1087 END
1088 writefile(lines, 'XVim9Func')
1089 so XVim9Func
1090
1091 delete('XVim9Func')
1092enddef
1093
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001094let s:funcResult = 0
1095
1096def FuncNoArgNoRet()
Bram Moolenaar53900992020-08-22 19:02:02 +02001097 s:funcResult = 11
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001098enddef
1099
1100def FuncNoArgRetNumber(): number
Bram Moolenaar53900992020-08-22 19:02:02 +02001101 s:funcResult = 22
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001102 return 1234
1103enddef
1104
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001105def FuncNoArgRetString(): string
Bram Moolenaar53900992020-08-22 19:02:02 +02001106 s:funcResult = 45
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001107 return 'text'
1108enddef
1109
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001110def FuncOneArgNoRet(arg: number)
Bram Moolenaar53900992020-08-22 19:02:02 +02001111 s:funcResult = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001112enddef
1113
1114def FuncOneArgRetNumber(arg: number): number
Bram Moolenaar53900992020-08-22 19:02:02 +02001115 s:funcResult = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001116 return arg
1117enddef
1118
Bram Moolenaar08938ee2020-04-11 23:17:17 +02001119def FuncTwoArgNoRet(one: bool, two: number)
Bram Moolenaar53900992020-08-22 19:02:02 +02001120 s:funcResult = two
Bram Moolenaar08938ee2020-04-11 23:17:17 +02001121enddef
1122
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001123def FuncOneArgRetString(arg: string): string
1124 return arg
1125enddef
1126
Bram Moolenaar89228602020-04-05 22:14:54 +02001127def FuncOneArgRetAny(arg: any): any
1128 return arg
1129enddef
1130
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001131def Test_func_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001132 var Ref1: func()
Bram Moolenaar53900992020-08-22 19:02:02 +02001133 s:funcResult = 0
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001134 Ref1 = FuncNoArgNoRet
1135 Ref1()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001136 s:funcResult->assert_equal(11)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001137
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001138 var Ref2: func
Bram Moolenaar53900992020-08-22 19:02:02 +02001139 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001140 Ref2 = FuncNoArgNoRet
1141 Ref2()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001142 s:funcResult->assert_equal(11)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001143
Bram Moolenaar53900992020-08-22 19:02:02 +02001144 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001145 Ref2 = FuncOneArgNoRet
1146 Ref2(12)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001147 s:funcResult->assert_equal(12)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001148
Bram Moolenaar53900992020-08-22 19:02:02 +02001149 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001150 Ref2 = FuncNoArgRetNumber
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001151 Ref2()->assert_equal(1234)
1152 s:funcResult->assert_equal(22)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001153
Bram Moolenaar53900992020-08-22 19:02:02 +02001154 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001155 Ref2 = FuncOneArgRetNumber
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001156 Ref2(13)->assert_equal(13)
1157 s:funcResult->assert_equal(13)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001158enddef
1159
Bram Moolenaar9978d472020-07-05 16:01:56 +02001160def Test_repeat_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001161 var res = 0
Bram Moolenaar9978d472020-07-05 16:01:56 +02001162 for n in repeat([1], 3)
1163 res += n
1164 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001165 res->assert_equal(3)
Bram Moolenaarfce82b32020-07-05 16:07:21 +02001166
1167 res = 0
1168 for n in add([1, 2], 3)
1169 res += n
1170 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001171 res->assert_equal(6)
Bram Moolenaar9978d472020-07-05 16:01:56 +02001172enddef
1173
Bram Moolenaar846178a2020-07-05 17:04:13 +02001174def Test_argv_return_type()
1175 next fileone filetwo
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001176 var res = ''
Bram Moolenaar846178a2020-07-05 17:04:13 +02001177 for name in argv()
1178 res ..= name
1179 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001180 res->assert_equal('fileonefiletwo')
Bram Moolenaar846178a2020-07-05 17:04:13 +02001181enddef
1182
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001183def Test_func_type_part()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001184 var RefVoid: func: void
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001185 RefVoid = FuncNoArgNoRet
1186 RefVoid = FuncOneArgNoRet
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001187 CheckDefFailure(['var RefVoid: func: void', 'RefVoid = FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func(...) but got func(): number')
1188 CheckDefFailure(['var RefVoid: func: void', 'RefVoid = FuncNoArgRetString'], 'E1012: Type mismatch; expected func(...) but got func(): string')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001189
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001190 var RefAny: func(): any
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001191 RefAny = FuncNoArgRetNumber
1192 RefAny = FuncNoArgRetString
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001193 CheckDefFailure(['var RefAny: func(): any', 'RefAny = FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(): any but got func()')
1194 CheckDefFailure(['var RefAny: func(): any', 'RefAny = FuncOneArgNoRet'], 'E1012: Type mismatch; expected func(): any but got func(number)')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001195
Bram Moolenaar6abd3dc2020-10-04 14:17:32 +02001196 var RefAnyNoArgs: func: any = RefAny
1197
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001198 var RefNr: func: number
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001199 RefNr = FuncNoArgRetNumber
1200 RefNr = FuncOneArgRetNumber
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001201 CheckDefFailure(['var RefNr: func: number', 'RefNr = FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(...): number but got func()')
1202 CheckDefFailure(['var RefNr: func: number', 'RefNr = FuncNoArgRetString'], 'E1012: Type mismatch; expected func(...): number but got func(): string')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001203
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001204 var RefStr: func: string
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001205 RefStr = FuncNoArgRetString
1206 RefStr = FuncOneArgRetString
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001207 CheckDefFailure(['var RefStr: func: string', 'RefStr = FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(...): string but got func()')
1208 CheckDefFailure(['var RefStr: func: string', 'RefStr = FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func(...): string but got func(): number')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001209enddef
1210
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001211def Test_func_type_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001212 CheckDefFailure(['var ref1: func()'], 'E704:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001213
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001214 CheckDefFailure(['var Ref1: func()', 'Ref1 = FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func() but got func(): number')
1215 CheckDefFailure(['var Ref1: func()', 'Ref1 = FuncOneArgNoRet'], 'E1012: Type mismatch; expected func() but got func(number)')
1216 CheckDefFailure(['var Ref1: func()', 'Ref1 = FuncOneArgRetNumber'], 'E1012: Type mismatch; expected func() but got func(number): number')
1217 CheckDefFailure(['var Ref1: func(bool)', 'Ref1 = FuncTwoArgNoRet'], 'E1012: Type mismatch; expected func(bool) but got func(bool, number)')
1218 CheckDefFailure(['var Ref1: func(?bool)', 'Ref1 = FuncTwoArgNoRet'], 'E1012: Type mismatch; expected func(?bool) but got func(bool, number)')
1219 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 +02001220
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001221 CheckDefFailure(['var RefWrong: func(string ,number)'], 'E1068:')
1222 CheckDefFailure(['var RefWrong: func(string,number)'], 'E1069:')
1223 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:')
1224 CheckDefFailure(['var RefWrong: func(bool):string'], 'E1069:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001225enddef
1226
Bram Moolenaar89228602020-04-05 22:14:54 +02001227def Test_func_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001228 var nr: number
Bram Moolenaar89228602020-04-05 22:14:54 +02001229 nr = FuncNoArgRetNumber()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001230 nr->assert_equal(1234)
Bram Moolenaar89228602020-04-05 22:14:54 +02001231
1232 nr = FuncOneArgRetAny(122)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001233 nr->assert_equal(122)
Bram Moolenaar89228602020-04-05 22:14:54 +02001234
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001235 var str: string
Bram Moolenaar89228602020-04-05 22:14:54 +02001236 str = FuncOneArgRetAny('yes')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001237 str->assert_equal('yes')
Bram Moolenaar89228602020-04-05 22:14:54 +02001238
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001239 CheckDefFailure(['var str: string', 'str = FuncNoArgRetNumber()'], 'E1012: Type mismatch; expected string but got number')
Bram Moolenaar89228602020-04-05 22:14:54 +02001240enddef
1241
Bram Moolenaar6abd3dc2020-10-04 14:17:32 +02001242def Test_func_common_type()
1243 def FuncOne(n: number): number
1244 return n
1245 enddef
1246 def FuncTwo(s: string): number
1247 return len(s)
1248 enddef
1249 def FuncThree(n: number, s: string): number
1250 return n + len(s)
1251 enddef
1252 var list = [FuncOne, FuncTwo, FuncThree]
1253 assert_equal(8, list[0](8))
1254 assert_equal(4, list[1]('word'))
1255 assert_equal(7, list[2](3, 'word'))
1256enddef
1257
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001258def MultiLine(
1259 arg1: string,
1260 arg2 = 1234,
1261 ...rest: list<string>
1262 ): string
1263 return arg1 .. arg2 .. join(rest, '-')
1264enddef
1265
Bram Moolenaar2c330432020-04-13 14:41:35 +02001266def MultiLineComment(
1267 arg1: string, # comment
1268 arg2 = 1234, # comment
1269 ...rest: list<string> # comment
1270 ): string # comment
1271 return arg1 .. arg2 .. join(rest, '-')
1272enddef
1273
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001274def Test_multiline()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001275 MultiLine('text')->assert_equal('text1234')
1276 MultiLine('text', 777)->assert_equal('text777')
1277 MultiLine('text', 777, 'one')->assert_equal('text777one')
1278 MultiLine('text', 777, 'one', 'two')->assert_equal('text777one-two')
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001279enddef
1280
Bram Moolenaar23e03252020-04-12 22:22:31 +02001281func Test_multiline_not_vim9()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001282 call MultiLine('text')->assert_equal('text1234')
1283 call MultiLine('text', 777)->assert_equal('text777')
1284 call MultiLine('text', 777, 'one')->assert_equal('text777one')
1285 call MultiLine('text', 777, 'one', 'two')->assert_equal('text777one-two')
Bram Moolenaar23e03252020-04-12 22:22:31 +02001286endfunc
1287
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001288
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02001289" When using CheckScriptFailure() for the below test, E1010 is generated instead
1290" of E1056.
1291func Test_E1056_1059()
1292 let caught_1056 = 0
1293 try
1294 def F():
1295 return 1
1296 enddef
1297 catch /E1056:/
1298 let caught_1056 = 1
1299 endtry
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001300 eval caught_1056->assert_equal(1)
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02001301
1302 let caught_1059 = 0
1303 try
1304 def F5(items : list)
1305 echo 'a'
1306 enddef
1307 catch /E1059:/
1308 let caught_1059 = 1
1309 endtry
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001310 eval caught_1059->assert_equal(1)
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02001311endfunc
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001312
Bram Moolenaar015f4262020-05-05 21:25:22 +02001313func DelMe()
1314 echo 'DelMe'
1315endfunc
1316
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001317def Test_error_reporting()
1318 # comment lines at the start of the function
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001319 var lines =<< trim END
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001320 " comment
1321 def Func()
1322 # comment
1323 # comment
1324 invalid
1325 enddef
1326 defcompile
1327 END
Bram Moolenaar08052222020-09-14 17:04:31 +02001328 writefile(lines, 'Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001329 try
1330 source Xdef
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001331 assert_report('should have failed')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001332 catch /E476:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001333 v:exception->assert_match('Invalid command: invalid')
1334 v:throwpoint->assert_match(', line 3$')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001335 endtry
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001336 delfunc! g:Func
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001337
1338 # comment lines after the start of the function
1339 lines =<< trim END
1340 " comment
1341 def Func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001342 var x = 1234
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001343 # comment
1344 # comment
1345 invalid
1346 enddef
1347 defcompile
1348 END
Bram Moolenaar08052222020-09-14 17:04:31 +02001349 writefile(lines, 'Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001350 try
1351 source Xdef
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001352 assert_report('should have failed')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001353 catch /E476:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001354 v:exception->assert_match('Invalid command: invalid')
1355 v:throwpoint->assert_match(', line 4$')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001356 endtry
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001357 delfunc! g:Func
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001358
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001359 lines =<< trim END
1360 vim9script
1361 def Func()
Bram Moolenaare0de1712020-12-02 17:36:54 +01001362 var db = {foo: 1, bar: 2}
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001363 # comment
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001364 var x = db.asdf
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001365 enddef
1366 defcompile
1367 Func()
1368 END
Bram Moolenaar08052222020-09-14 17:04:31 +02001369 writefile(lines, 'Xdef')
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001370 try
1371 source Xdef
1372 assert_report('should have failed')
1373 catch /E716:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001374 v:throwpoint->assert_match('_Func, line 3$')
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001375 endtry
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001376 delfunc! g:Func
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001377
Bram Moolenaar08052222020-09-14 17:04:31 +02001378 delete('Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001379enddef
1380
Bram Moolenaar015f4262020-05-05 21:25:22 +02001381def Test_deleted_function()
1382 CheckDefExecFailure([
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001383 'var RefMe: func = function("g:DelMe")',
Bram Moolenaar015f4262020-05-05 21:25:22 +02001384 'delfunc g:DelMe',
1385 'echo RefMe()'], 'E117:')
1386enddef
1387
1388def Test_unknown_function()
1389 CheckDefExecFailure([
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001390 'var Ref: func = function("NotExist")',
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +02001391 'delfunc g:NotExist'], 'E700:')
Bram Moolenaar015f4262020-05-05 21:25:22 +02001392enddef
1393
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +02001394def RefFunc(Ref: func(string): string): string
1395 return Ref('more')
1396enddef
1397
1398def Test_closure_simple()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001399 var local = 'some '
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001400 RefFunc({s -> local .. s})->assert_equal('some more')
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +02001401enddef
1402
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001403def MakeRef()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001404 var local = 'some '
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001405 g:Ref = {s -> local .. s}
1406enddef
1407
1408def Test_closure_ref_after_return()
1409 MakeRef()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001410 g:Ref('thing')->assert_equal('some thing')
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001411 unlet g:Ref
1412enddef
1413
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001414def MakeTwoRefs()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001415 var local = ['some']
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001416 g:Extend = {s -> local->add(s)}
1417 g:Read = {-> local}
1418enddef
1419
1420def Test_closure_two_refs()
1421 MakeTwoRefs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001422 join(g:Read(), ' ')->assert_equal('some')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001423 g:Extend('more')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001424 join(g:Read(), ' ')->assert_equal('some more')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001425 g:Extend('even')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001426 join(g:Read(), ' ')->assert_equal('some more even')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001427
1428 unlet g:Extend
1429 unlet g:Read
1430enddef
1431
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001432def ReadRef(Ref: func(): list<string>): string
1433 return join(Ref(), ' ')
1434enddef
1435
Bram Moolenaar5e654232020-09-16 15:22:00 +02001436def ExtendRef(Ref: func(string): list<string>, add: string)
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001437 Ref(add)
1438enddef
1439
1440def Test_closure_two_indirect_refs()
Bram Moolenaarf7779c62020-05-03 15:38:16 +02001441 MakeTwoRefs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001442 ReadRef(g:Read)->assert_equal('some')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001443 ExtendRef(g:Extend, 'more')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001444 ReadRef(g:Read)->assert_equal('some more')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001445 ExtendRef(g:Extend, 'even')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001446 ReadRef(g:Read)->assert_equal('some more even')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001447
1448 unlet g:Extend
1449 unlet g:Read
1450enddef
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001451
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001452def MakeArgRefs(theArg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001453 var local = 'loc_val'
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001454 g:UseArg = {s -> theArg .. '/' .. local .. '/' .. s}
1455enddef
1456
1457def MakeArgRefsVarargs(theArg: string, ...rest: list<string>)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001458 var local = 'the_loc'
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001459 g:UseVararg = {s -> theArg .. '/' .. local .. '/' .. s .. '/' .. join(rest)}
1460enddef
1461
1462def Test_closure_using_argument()
1463 MakeArgRefs('arg_val')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001464 g:UseArg('call_val')->assert_equal('arg_val/loc_val/call_val')
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001465
1466 MakeArgRefsVarargs('arg_val', 'one', 'two')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001467 g:UseVararg('call_val')->assert_equal('arg_val/the_loc/call_val/one two')
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001468
1469 unlet g:UseArg
1470 unlet g:UseVararg
1471enddef
1472
Bram Moolenaar85d5e2b2020-10-10 14:13:01 +02001473def MakeGetAndAppendRefs()
1474 var local = 'a'
1475
1476 def Append(arg: string)
1477 local ..= arg
1478 enddef
1479 g:Append = Append
1480
1481 def Get(): string
1482 return local
1483 enddef
1484 g:Get = Get
1485enddef
1486
1487def Test_closure_append_get()
1488 MakeGetAndAppendRefs()
1489 g:Get()->assert_equal('a')
1490 g:Append('-b')
1491 g:Get()->assert_equal('a-b')
1492 g:Append('-c')
1493 g:Get()->assert_equal('a-b-c')
1494
1495 unlet g:Append
1496 unlet g:Get
1497enddef
Bram Moolenaarb68b3462020-05-06 21:06:30 +02001498
Bram Moolenaar04b12692020-05-04 23:24:44 +02001499def Test_nested_closure()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001500 var local = 'text'
Bram Moolenaar04b12692020-05-04 23:24:44 +02001501 def Closure(arg: string): string
1502 return local .. arg
1503 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001504 Closure('!!!')->assert_equal('text!!!')
Bram Moolenaar04b12692020-05-04 23:24:44 +02001505enddef
1506
Bram Moolenaar6f5b6df2020-05-16 21:20:12 +02001507func GetResult(Ref)
1508 return a:Ref('some')
1509endfunc
1510
1511def Test_call_closure_not_compiled()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001512 var text = 'text'
Bram Moolenaar6f5b6df2020-05-16 21:20:12 +02001513 g:Ref = {s -> s .. text}
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001514 GetResult(g:Ref)->assert_equal('sometext')
Bram Moolenaar6f5b6df2020-05-16 21:20:12 +02001515enddef
1516
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001517def Test_double_closure_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001518 var lines =<< trim END
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001519 vim9script
1520 def Func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001521 var name = 0
1522 for i in range(2)
1523 timer_start(0, {-> name})
1524 endfor
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001525 enddef
1526 Func()
1527 END
Bram Moolenaar148ce7a2020-09-23 21:57:23 +02001528 CheckScriptSuccess(lines)
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001529enddef
1530
Bram Moolenaar85d5e2b2020-10-10 14:13:01 +02001531def Test_nested_closure_used()
1532 var lines =<< trim END
1533 vim9script
1534 def Func()
1535 var x = 'hello'
1536 var Closure = {-> x}
1537 g:Myclosure = {-> Closure()}
1538 enddef
1539 Func()
1540 assert_equal('hello', g:Myclosure())
1541 END
1542 CheckScriptSuccess(lines)
1543enddef
Bram Moolenaar0876c782020-10-07 19:08:04 +02001544
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02001545def Test_nested_closure_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001546 var lines =<< trim END
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02001547 vim9script
1548 def FuncA()
1549 FuncB(0)
1550 enddef
1551 def FuncB(n: number): list<string>
1552 return map([0], {_, v -> n})
1553 enddef
1554 FuncA()
1555 END
1556 CheckScriptFailure(lines, 'E1012:')
1557enddef
1558
Bram Moolenaarf112f302020-12-20 17:47:52 +01001559def Test_global_closure()
1560 var lines =<< trim END
1561 vim9script
1562 def ReverseEveryNLines(n: number, line1: number, line2: number)
1563 var mods = 'sil keepj keepp lockm '
1564 var range = ':' .. line1 .. ',' .. line2
1565 def g:Offset(): number
1566 var offset = (line('.') - line1 + 1) % n
1567 return offset != 0 ? offset : n
1568 enddef
1569 exe mods .. range .. 'g/^/exe "m .-" .. g:Offset()'
1570 enddef
1571
1572 new
1573 repeat(['aaa', 'bbb', 'ccc'], 3)->setline(1)
1574 ReverseEveryNLines(3, 1, 9)
1575 END
1576 CheckScriptSuccess(lines)
1577 var expected = repeat(['ccc', 'bbb', 'aaa'], 3)
1578 assert_equal(expected, getline(1, 9))
1579 bwipe!
1580enddef
1581
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01001582def Test_global_closure_called_directly()
1583 var lines =<< trim END
1584 vim9script
1585 def Outer()
1586 var x = 1
1587 def g:Inner()
1588 var y = x
1589 x += 1
1590 assert_equal(1, y)
1591 enddef
1592 g:Inner()
1593 assert_equal(2, x)
1594 enddef
1595 Outer()
1596 END
1597 CheckScriptSuccess(lines)
1598 delfunc g:Inner
1599enddef
1600
Bram Moolenaar34c54eb2020-11-25 19:15:19 +01001601def Test_failure_in_called_function()
1602 # this was using the frame index as the return value
1603 var lines =<< trim END
1604 vim9script
1605 au TerminalWinOpen * eval [][0]
1606 def PopupTerm(a: any)
1607 # make sure typvals on stack are string
1608 ['a', 'b', 'c', 'd', 'e', 'f', 'g']->join()
1609 FireEvent()
1610 enddef
1611 def FireEvent()
1612 do TerminalWinOpen
1613 enddef
1614 # use try/catch to make eval fail
1615 try
1616 call PopupTerm(0)
1617 catch
1618 endtry
1619 au! TerminalWinOpen
1620 END
1621 CheckScriptSuccess(lines)
1622enddef
1623
Bram Moolenaar5366e1a2020-10-01 13:01:34 +02001624def Test_nested_lambda()
1625 var lines =<< trim END
1626 vim9script
1627 def Func()
1628 var x = 4
1629 var Lambda1 = {-> 7}
1630 var Lambda2 = {-> [Lambda1(), x]}
1631 var res = Lambda2()
1632 assert_equal([7, 4], res)
1633 enddef
1634 Func()
1635 END
1636 CheckScriptSuccess(lines)
1637enddef
1638
Bram Moolenaar52bf81c2020-11-17 18:50:44 +01001639def Shadowed(): list<number>
1640 var FuncList: list<func: number> = [{ -> 42}]
1641 return FuncList->map({_, Shadowed -> Shadowed()})
1642enddef
1643
1644def Test_lambda_arg_shadows_func()
1645 assert_equal([42], Shadowed())
1646enddef
1647
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02001648def Line_continuation_in_def(dir: string = ''): string
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001649 var path: string = empty(dir)
1650 \ ? 'empty'
1651 \ : 'full'
1652 return path
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02001653enddef
1654
1655def Test_line_continuation_in_def()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001656 Line_continuation_in_def('.')->assert_equal('full')
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02001657enddef
1658
Bram Moolenaar2ea95b62020-11-19 21:47:56 +01001659def Test_script_var_in_lambda()
1660 var lines =<< trim END
1661 vim9script
1662 var script = 'test'
1663 assert_equal(['test'], map(['one'], {-> script}))
1664 END
1665 CheckScriptSuccess(lines)
1666enddef
1667
Bram Moolenaar5e654232020-09-16 15:22:00 +02001668def Line_continuation_in_lambda(): list<string>
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001669 var x = range(97, 100)
Bram Moolenaar914e7ea2020-07-11 15:20:48 +02001670 ->map({_, v -> nr2char(v)
Bram Moolenaar7a4b8982020-07-08 17:36:21 +02001671 ->toupper()})
1672 ->reverse()
1673 return x
1674enddef
1675
1676def Test_line_continuation_in_lambda()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001677 Line_continuation_in_lambda()->assert_equal(['D', 'C', 'B', 'A'])
Bram Moolenaar7a4b8982020-07-08 17:36:21 +02001678enddef
1679
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001680func Test_silent_echo()
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001681 CheckScreendump
1682
1683 let lines =<< trim END
1684 vim9script
1685 def EchoNothing()
1686 silent echo ''
1687 enddef
1688 defcompile
1689 END
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001690 call writefile(lines, 'XTest_silent_echo')
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001691
1692 " Check that the balloon shows up after a mouse move
1693 let buf = RunVimInTerminal('-S XTest_silent_echo', {'rows': 6})
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001694 call term_sendkeys(buf, ":abc")
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001695 call VerifyScreenDump(buf, 'Test_vim9_silent_echo', {})
1696
1697 " clean up
1698 call StopVimInTerminal(buf)
1699 call delete('XTest_silent_echo')
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001700endfunc
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001701
Bram Moolenaar171fb922020-10-28 16:54:47 +01001702def SilentlyError()
1703 execute('silent! invalid')
1704 g:did_it = 'yes'
1705enddef
1706
Bram Moolenaar28ee8922020-10-28 20:20:00 +01001707func UserError()
1708 silent! invalid
1709endfunc
1710
1711def SilentlyUserError()
1712 UserError()
1713 g:did_it = 'yes'
1714enddef
Bram Moolenaar171fb922020-10-28 16:54:47 +01001715
1716" This can't be a :def function, because the assert would not be reached.
Bram Moolenaar171fb922020-10-28 16:54:47 +01001717func Test_ignore_silent_error()
1718 let g:did_it = 'no'
1719 call SilentlyError()
1720 call assert_equal('yes', g:did_it)
1721
Bram Moolenaar28ee8922020-10-28 20:20:00 +01001722 let g:did_it = 'no'
1723 call SilentlyUserError()
1724 call assert_equal('yes', g:did_it)
Bram Moolenaar171fb922020-10-28 16:54:47 +01001725
1726 unlet g:did_it
1727endfunc
1728
Bram Moolenaarcd030c42020-10-30 21:49:40 +01001729def Test_ignore_silent_error_in_filter()
1730 var lines =<< trim END
1731 vim9script
1732 def Filter(winid: number, key: string): bool
1733 if key == 'o'
1734 silent! eval [][0]
1735 return true
1736 endif
1737 return popup_filter_menu(winid, key)
1738 enddef
1739
Bram Moolenaare0de1712020-12-02 17:36:54 +01001740 popup_create('popup', {filter: Filter})
Bram Moolenaarcd030c42020-10-30 21:49:40 +01001741 feedkeys("o\r", 'xnt')
1742 END
1743 CheckScriptSuccess(lines)
1744enddef
1745
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001746def Fibonacci(n: number): number
1747 if n < 2
1748 return n
1749 else
1750 return Fibonacci(n - 1) + Fibonacci(n - 2)
1751 endif
1752enddef
1753
Bram Moolenaar985116a2020-07-12 17:31:09 +02001754def Test_recursive_call()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001755 Fibonacci(20)->assert_equal(6765)
Bram Moolenaar985116a2020-07-12 17:31:09 +02001756enddef
1757
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001758def TreeWalk(dir: string): list<any>
1759 return readdir(dir)->map({_, val ->
1760 fnamemodify(dir .. '/' .. val, ':p')->isdirectory()
Bram Moolenaar2bede172020-11-19 18:53:18 +01001761 ? {[val]: TreeWalk(dir .. '/' .. val)}
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001762 : val
1763 })
1764enddef
1765
1766def Test_closure_in_map()
1767 mkdir('XclosureDir/tdir', 'p')
1768 writefile(['111'], 'XclosureDir/file1')
1769 writefile(['222'], 'XclosureDir/file2')
1770 writefile(['333'], 'XclosureDir/tdir/file3')
1771
Bram Moolenaare0de1712020-12-02 17:36:54 +01001772 TreeWalk('XclosureDir')->assert_equal(['file1', 'file2', {tdir: ['file3']}])
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001773
1774 delete('XclosureDir', 'rf')
1775enddef
1776
Bram Moolenaar7b5d5442020-10-04 13:42:34 +02001777def Test_invalid_function_name()
1778 var lines =<< trim END
1779 vim9script
1780 def s: list<string>
1781 END
1782 CheckScriptFailure(lines, 'E129:')
1783
1784 lines =<< trim END
1785 vim9script
1786 def g: list<string>
1787 END
1788 CheckScriptFailure(lines, 'E129:')
1789
1790 lines =<< trim END
1791 vim9script
1792 def <SID>: list<string>
1793 END
1794 CheckScriptFailure(lines, 'E884:')
1795
1796 lines =<< trim END
1797 vim9script
1798 def F list<string>
1799 END
1800 CheckScriptFailure(lines, 'E488:')
1801enddef
1802
Bram Moolenaara90afb92020-07-15 22:38:56 +02001803def Test_partial_call()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001804 var Xsetlist = function('setloclist', [0])
Bram Moolenaare0de1712020-12-02 17:36:54 +01001805 Xsetlist([], ' ', {title: 'test'})
1806 getloclist(0, {title: 1})->assert_equal({title: 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02001807
1808 Xsetlist = function('setloclist', [0, [], ' '])
Bram Moolenaare0de1712020-12-02 17:36:54 +01001809 Xsetlist({title: 'test'})
1810 getloclist(0, {title: 1})->assert_equal({title: 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02001811
1812 Xsetlist = function('setqflist')
Bram Moolenaare0de1712020-12-02 17:36:54 +01001813 Xsetlist([], ' ', {title: 'test'})
1814 getqflist({title: 1})->assert_equal({title: 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02001815
1816 Xsetlist = function('setqflist', [[], ' '])
Bram Moolenaare0de1712020-12-02 17:36:54 +01001817 Xsetlist({title: 'test'})
1818 getqflist({title: 1})->assert_equal({title: 'test'})
Bram Moolenaar6abd3dc2020-10-04 14:17:32 +02001819
1820 var Len: func: number = function('len', ['word'])
1821 assert_equal(4, Len())
Bram Moolenaara90afb92020-07-15 22:38:56 +02001822enddef
1823
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001824def Test_cmd_modifier()
1825 tab echo '0'
Bram Moolenaard2c61702020-09-06 15:58:36 +02001826 CheckDefFailure(['5tab echo 3'], 'E16:')
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001827enddef
1828
1829def Test_restore_modifiers()
1830 # check that when compiling a :def function command modifiers are not messed
1831 # up.
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001832 var lines =<< trim END
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001833 vim9script
1834 set eventignore=
1835 autocmd QuickFixCmdPost * copen
1836 def AutocmdsDisabled()
Bram Moolenaar6cf7e3b2020-10-28 14:31:16 +01001837 eval 0
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001838 enddef
1839 func Func()
1840 noautocmd call s:AutocmdsDisabled()
1841 let g:ei_after = &eventignore
1842 endfunc
1843 Func()
1844 END
1845 CheckScriptSuccess(lines)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001846 g:ei_after->assert_equal('')
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001847enddef
1848
Bram Moolenaardfa3d552020-09-10 22:05:08 +02001849def StackTop()
1850 eval 1
1851 eval 2
1852 # call not on fourth line
1853 StackBot()
1854enddef
1855
1856def StackBot()
1857 # throw an error
1858 eval [][0]
1859enddef
1860
1861def Test_callstack_def()
1862 try
1863 StackTop()
1864 catch
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001865 v:throwpoint->assert_match('Test_callstack_def\[2\]..StackTop\[4\]..StackBot, line 2')
Bram Moolenaardfa3d552020-09-10 22:05:08 +02001866 endtry
1867enddef
1868
Bram Moolenaare8211a32020-10-09 22:04:29 +02001869" Re-using spot for variable used in block
1870def Test_block_scoped_var()
1871 var lines =<< trim END
1872 vim9script
1873 def Func()
1874 var x = ['a', 'b', 'c']
1875 if 1
1876 var y = 'x'
1877 map(x, {-> y})
1878 endif
1879 var z = x
1880 assert_equal(['x', 'x', 'x'], z)
1881 enddef
1882 Func()
1883 END
1884 CheckScriptSuccess(lines)
1885enddef
1886
Bram Moolenaareeece9e2020-11-20 19:26:48 +01001887def Test_reset_did_emsg()
1888 var lines =<< trim END
1889 @s = 'blah'
1890 au BufWinLeave * #
1891 def Func()
1892 var winid = popup_create('popup', {})
1893 exe '*s'
1894 popup_close(winid)
1895 enddef
1896 Func()
1897 END
1898 CheckScriptFailure(lines, 'E492:', 8)
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001899 delfunc! g:Func
Bram Moolenaareeece9e2020-11-20 19:26:48 +01001900enddef
1901
Bram Moolenaar57f799e2020-12-12 20:42:19 +01001902def Test_did_emsg_reset()
1903 # executing an autocommand resets did_emsg, this should not result in a
1904 # builtin function considered failing
1905 var lines =<< trim END
1906 vim9script
1907 au BufWinLeave * #
1908 def Func()
1909 popup_menu('', {callback: {-> popup_create('', {})->popup_close()}})
1910 eval [][0]
1911 enddef
1912 nno <F3> <cmd>call <sid>Func()<cr>
1913 feedkeys("\<F3>\e", 'xt')
1914 END
1915 writefile(lines, 'XemsgReset')
1916 assert_fails('so XemsgReset', ['E684:', 'E684:'], lines, 2)
1917 delete('XemsgReset')
1918 nunmap <F3>
1919 au! BufWinLeave
1920enddef
1921
Bram Moolenaar56602ba2020-12-05 21:22:08 +01001922def Test_abort_with_silent_call()
1923 var lines =<< trim END
1924 vim9script
1925 g:result = 'none'
1926 def Func()
1927 g:result += 3
1928 g:result = 'yes'
1929 enddef
1930 # error is silenced, but function aborts on error
1931 silent! Func()
1932 assert_equal('none', g:result)
1933 unlet g:result
1934 END
1935 CheckScriptSuccess(lines)
1936enddef
1937
Bram Moolenaarf665e972020-12-05 19:17:16 +01001938def Test_continues_with_silent_error()
1939 var lines =<< trim END
1940 vim9script
1941 g:result = 'none'
1942 def Func()
1943 silent! g:result += 3
1944 g:result = 'yes'
1945 enddef
1946 # error is silenced, function does not abort
1947 Func()
1948 assert_equal('yes', g:result)
1949 unlet g:result
1950 END
1951 CheckScriptSuccess(lines)
1952enddef
1953
Bram Moolenaaraf0df472020-12-02 20:51:22 +01001954def Test_abort_even_with_silent()
1955 var lines =<< trim END
1956 vim9script
1957 g:result = 'none'
1958 def Func()
1959 eval {-> ''}() .. '' .. {}['X']
1960 g:result = 'yes'
1961 enddef
Bram Moolenaarf665e972020-12-05 19:17:16 +01001962 silent! Func()
Bram Moolenaaraf0df472020-12-02 20:51:22 +01001963 assert_equal('none', g:result)
Bram Moolenaar4029cab2020-12-05 18:13:27 +01001964 unlet g:result
1965 END
1966 CheckScriptSuccess(lines)
1967enddef
1968
Bram Moolenaarf665e972020-12-05 19:17:16 +01001969def Test_cmdmod_silent_restored()
1970 var lines =<< trim END
1971 vim9script
1972 def Func()
1973 g:result = 'none'
1974 silent! g:result += 3
1975 g:result = 'none'
1976 g:result += 3
1977 enddef
1978 Func()
1979 END
1980 # can't use CheckScriptFailure, it ignores the :silent!
1981 var fname = 'Xdefsilent'
1982 writefile(lines, fname)
1983 var caught = 'no'
1984 try
1985 exe 'source ' .. fname
1986 catch /E1030:/
1987 caught = 'yes'
1988 assert_match('Func, line 4', v:throwpoint)
1989 endtry
1990 assert_equal('yes', caught)
1991 delete(fname)
1992enddef
1993
Bram Moolenaar4029cab2020-12-05 18:13:27 +01001994def Test_dict_member_with_silent()
1995 var lines =<< trim END
1996 vim9script
1997 g:result = 'none'
1998 var d: dict<any>
1999 def Func()
2000 try
2001 g:result = map([], {_, v -> {}[v]})->join() .. d['']
2002 catch
2003 endtry
2004 enddef
2005 silent! Func()
2006 assert_equal('0', g:result)
2007 unlet g:result
Bram Moolenaaraf0df472020-12-02 20:51:22 +01002008 END
2009 CheckScriptSuccess(lines)
2010enddef
2011
Bram Moolenaar5b3d1bb2020-12-22 12:20:08 +01002012def Test_opfunc()
2013 nnoremap <F3> <cmd>set opfunc=Opfunc<cr>g@
2014 def g:Opfunc(_: any): string
2015 setline(1, 'ASDF')
2016 return ''
2017 enddef
2018 new
2019 setline(1, 'asdf')
2020 feedkeys("\<F3>$", 'x')
2021 assert_equal('ASDF', getline(1))
2022
2023 bwipe!
2024 nunmap <F3>
2025enddef
2026
Bram Moolenaar077a4232020-12-22 18:33:27 +01002027" this was crashing on exit
2028def Test_nested_lambda_in_closure()
2029 var lines =<< trim END
2030 vim9script
2031 def Outer()
2032 def g:Inner()
2033 echo map([1, 2, 3], {_, v -> v + 1})
2034 enddef
2035 g:Inner()
2036 enddef
2037 defcompile
2038 writefile(['Done'], 'XnestedDone')
2039 quit
2040 END
2041 if !RunVim([], lines, '--clean')
2042 return
2043 endif
2044 assert_equal(['Done'], readfile('XnestedDone'))
2045 delete('XnestedDone')
2046enddef
2047
2048
Bram Moolenaarf7779c62020-05-03 15:38:16 +02002049
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002050" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker