blob: bdced0257b60e44b15c00f26ea5f3402d2195d44 [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')
Bram Moolenaar3bdc90b2020-12-22 20:35:40 +01001025
1026 lines =<< trim END
1027 const g:Aconst = 77
1028 def Change()
1029 # comment
1030 g:Aconst = 99
1031 enddef
1032 call Change()
1033 unlet g:Aconst
1034 END
Bram Moolenaar1dcf55d2020-12-22 22:07:30 +01001035 CheckScriptFailure(lines, 'E741: Value is locked: Aconst', 2)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001036enddef
1037
1038" Test that inside :function a Python function can be defined, :def is not
1039" recognized.
1040func Test_function_python()
1041 CheckFeature python3
Bram Moolenaar727345e2020-09-27 23:33:59 +02001042 let py = 'python3'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001043 execute py "<< EOF"
1044def do_something():
1045 return 1
1046EOF
1047endfunc
1048
1049def Test_delfunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001050 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001051 vim9script
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02001052 def g:GoneSoon()
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001053 echo 'hello'
1054 enddef
1055
1056 def CallGoneSoon()
1057 GoneSoon()
1058 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +02001059 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001060
Bram Moolenaar4c17ad92020-04-27 22:47:51 +02001061 delfunc g:GoneSoon
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001062 CallGoneSoon()
1063 END
1064 writefile(lines, 'XToDelFunc')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001065 assert_fails('so XToDelFunc', 'E933:', '', 1, 'CallGoneSoon')
1066 assert_fails('so XToDelFunc', 'E933:', '', 1, 'CallGoneSoon')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001067
1068 delete('XToDelFunc')
1069enddef
1070
1071def Test_redef_failure()
Bram Moolenaard2c61702020-09-06 15:58:36 +02001072 writefile(['def Func0(): string', 'return "Func0"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001073 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +02001074 writefile(['def Func1(): string', 'return "Func1"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001075 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +02001076 writefile(['def! Func0(): string', 'enddef', 'defcompile'], 'Xdef')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +02001077 assert_fails('so Xdef', 'E1027:', '', 1, 'Func0')
Bram Moolenaard2c61702020-09-06 15:58:36 +02001078 writefile(['def Func2(): string', 'return "Func2"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001079 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +02001080 delete('Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001081
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001082 g:Func0()->assert_equal(0)
1083 g:Func1()->assert_equal('Func1')
1084 g:Func2()->assert_equal('Func2')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001085
1086 delfunc! Func0
1087 delfunc! Func1
1088 delfunc! Func2
1089enddef
1090
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +02001091def Test_vim9script_func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001092 var lines =<< trim END
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +02001093 vim9script
1094 func Func(arg)
1095 echo a:arg
1096 endfunc
1097 Func('text')
1098 END
1099 writefile(lines, 'XVim9Func')
1100 so XVim9Func
1101
1102 delete('XVim9Func')
1103enddef
1104
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001105let s:funcResult = 0
1106
1107def FuncNoArgNoRet()
Bram Moolenaar53900992020-08-22 19:02:02 +02001108 s:funcResult = 11
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001109enddef
1110
1111def FuncNoArgRetNumber(): number
Bram Moolenaar53900992020-08-22 19:02:02 +02001112 s:funcResult = 22
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001113 return 1234
1114enddef
1115
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001116def FuncNoArgRetString(): string
Bram Moolenaar53900992020-08-22 19:02:02 +02001117 s:funcResult = 45
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001118 return 'text'
1119enddef
1120
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001121def FuncOneArgNoRet(arg: number)
Bram Moolenaar53900992020-08-22 19:02:02 +02001122 s:funcResult = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001123enddef
1124
1125def FuncOneArgRetNumber(arg: number): number
Bram Moolenaar53900992020-08-22 19:02:02 +02001126 s:funcResult = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001127 return arg
1128enddef
1129
Bram Moolenaar08938ee2020-04-11 23:17:17 +02001130def FuncTwoArgNoRet(one: bool, two: number)
Bram Moolenaar53900992020-08-22 19:02:02 +02001131 s:funcResult = two
Bram Moolenaar08938ee2020-04-11 23:17:17 +02001132enddef
1133
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001134def FuncOneArgRetString(arg: string): string
1135 return arg
1136enddef
1137
Bram Moolenaar89228602020-04-05 22:14:54 +02001138def FuncOneArgRetAny(arg: any): any
1139 return arg
1140enddef
1141
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001142def Test_func_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001143 var Ref1: func()
Bram Moolenaar53900992020-08-22 19:02:02 +02001144 s:funcResult = 0
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001145 Ref1 = FuncNoArgNoRet
1146 Ref1()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001147 s:funcResult->assert_equal(11)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001148
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001149 var Ref2: func
Bram Moolenaar53900992020-08-22 19:02:02 +02001150 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001151 Ref2 = FuncNoArgNoRet
1152 Ref2()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001153 s:funcResult->assert_equal(11)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001154
Bram Moolenaar53900992020-08-22 19:02:02 +02001155 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001156 Ref2 = FuncOneArgNoRet
1157 Ref2(12)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001158 s:funcResult->assert_equal(12)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001159
Bram Moolenaar53900992020-08-22 19:02:02 +02001160 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001161 Ref2 = FuncNoArgRetNumber
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001162 Ref2()->assert_equal(1234)
1163 s:funcResult->assert_equal(22)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001164
Bram Moolenaar53900992020-08-22 19:02:02 +02001165 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001166 Ref2 = FuncOneArgRetNumber
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001167 Ref2(13)->assert_equal(13)
1168 s:funcResult->assert_equal(13)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001169enddef
1170
Bram Moolenaar9978d472020-07-05 16:01:56 +02001171def Test_repeat_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001172 var res = 0
Bram Moolenaar9978d472020-07-05 16:01:56 +02001173 for n in repeat([1], 3)
1174 res += n
1175 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001176 res->assert_equal(3)
Bram Moolenaarfce82b32020-07-05 16:07:21 +02001177
1178 res = 0
1179 for n in add([1, 2], 3)
1180 res += n
1181 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001182 res->assert_equal(6)
Bram Moolenaar9978d472020-07-05 16:01:56 +02001183enddef
1184
Bram Moolenaar846178a2020-07-05 17:04:13 +02001185def Test_argv_return_type()
1186 next fileone filetwo
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001187 var res = ''
Bram Moolenaar846178a2020-07-05 17:04:13 +02001188 for name in argv()
1189 res ..= name
1190 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001191 res->assert_equal('fileonefiletwo')
Bram Moolenaar846178a2020-07-05 17:04:13 +02001192enddef
1193
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001194def Test_func_type_part()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001195 var RefVoid: func: void
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001196 RefVoid = FuncNoArgNoRet
1197 RefVoid = FuncOneArgNoRet
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001198 CheckDefFailure(['var RefVoid: func: void', 'RefVoid = FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func(...) but got func(): number')
1199 CheckDefFailure(['var RefVoid: func: void', 'RefVoid = FuncNoArgRetString'], 'E1012: Type mismatch; expected func(...) but got func(): string')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001200
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001201 var RefAny: func(): any
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001202 RefAny = FuncNoArgRetNumber
1203 RefAny = FuncNoArgRetString
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001204 CheckDefFailure(['var RefAny: func(): any', 'RefAny = FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(): any but got func()')
1205 CheckDefFailure(['var RefAny: func(): any', 'RefAny = FuncOneArgNoRet'], 'E1012: Type mismatch; expected func(): any but got func(number)')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001206
Bram Moolenaar6abd3dc2020-10-04 14:17:32 +02001207 var RefAnyNoArgs: func: any = RefAny
1208
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001209 var RefNr: func: number
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001210 RefNr = FuncNoArgRetNumber
1211 RefNr = FuncOneArgRetNumber
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001212 CheckDefFailure(['var RefNr: func: number', 'RefNr = FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(...): number but got func()')
1213 CheckDefFailure(['var RefNr: func: number', 'RefNr = FuncNoArgRetString'], 'E1012: Type mismatch; expected func(...): number but got func(): string')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001214
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001215 var RefStr: func: string
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001216 RefStr = FuncNoArgRetString
1217 RefStr = FuncOneArgRetString
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001218 CheckDefFailure(['var RefStr: func: string', 'RefStr = FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(...): string but got func()')
1219 CheckDefFailure(['var RefStr: func: string', 'RefStr = FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func(...): string but got func(): number')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001220enddef
1221
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001222def Test_func_type_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001223 CheckDefFailure(['var ref1: func()'], 'E704:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001224
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001225 CheckDefFailure(['var Ref1: func()', 'Ref1 = FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func() but got func(): number')
1226 CheckDefFailure(['var Ref1: func()', 'Ref1 = FuncOneArgNoRet'], 'E1012: Type mismatch; expected func() but got func(number)')
1227 CheckDefFailure(['var Ref1: func()', 'Ref1 = FuncOneArgRetNumber'], 'E1012: Type mismatch; expected func() but got func(number): number')
1228 CheckDefFailure(['var Ref1: func(bool)', 'Ref1 = FuncTwoArgNoRet'], 'E1012: Type mismatch; expected func(bool) but got func(bool, number)')
1229 CheckDefFailure(['var Ref1: func(?bool)', 'Ref1 = FuncTwoArgNoRet'], 'E1012: Type mismatch; expected func(?bool) but got func(bool, number)')
1230 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 +02001231
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001232 CheckDefFailure(['var RefWrong: func(string ,number)'], 'E1068:')
1233 CheckDefFailure(['var RefWrong: func(string,number)'], 'E1069:')
1234 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:')
1235 CheckDefFailure(['var RefWrong: func(bool):string'], 'E1069:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001236enddef
1237
Bram Moolenaar89228602020-04-05 22:14:54 +02001238def Test_func_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001239 var nr: number
Bram Moolenaar89228602020-04-05 22:14:54 +02001240 nr = FuncNoArgRetNumber()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001241 nr->assert_equal(1234)
Bram Moolenaar89228602020-04-05 22:14:54 +02001242
1243 nr = FuncOneArgRetAny(122)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001244 nr->assert_equal(122)
Bram Moolenaar89228602020-04-05 22:14:54 +02001245
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001246 var str: string
Bram Moolenaar89228602020-04-05 22:14:54 +02001247 str = FuncOneArgRetAny('yes')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001248 str->assert_equal('yes')
Bram Moolenaar89228602020-04-05 22:14:54 +02001249
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001250 CheckDefFailure(['var str: string', 'str = FuncNoArgRetNumber()'], 'E1012: Type mismatch; expected string but got number')
Bram Moolenaar89228602020-04-05 22:14:54 +02001251enddef
1252
Bram Moolenaar6abd3dc2020-10-04 14:17:32 +02001253def Test_func_common_type()
1254 def FuncOne(n: number): number
1255 return n
1256 enddef
1257 def FuncTwo(s: string): number
1258 return len(s)
1259 enddef
1260 def FuncThree(n: number, s: string): number
1261 return n + len(s)
1262 enddef
1263 var list = [FuncOne, FuncTwo, FuncThree]
1264 assert_equal(8, list[0](8))
1265 assert_equal(4, list[1]('word'))
1266 assert_equal(7, list[2](3, 'word'))
1267enddef
1268
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001269def MultiLine(
1270 arg1: string,
1271 arg2 = 1234,
1272 ...rest: list<string>
1273 ): string
1274 return arg1 .. arg2 .. join(rest, '-')
1275enddef
1276
Bram Moolenaar2c330432020-04-13 14:41:35 +02001277def MultiLineComment(
1278 arg1: string, # comment
1279 arg2 = 1234, # comment
1280 ...rest: list<string> # comment
1281 ): string # comment
1282 return arg1 .. arg2 .. join(rest, '-')
1283enddef
1284
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001285def Test_multiline()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001286 MultiLine('text')->assert_equal('text1234')
1287 MultiLine('text', 777)->assert_equal('text777')
1288 MultiLine('text', 777, 'one')->assert_equal('text777one')
1289 MultiLine('text', 777, 'one', 'two')->assert_equal('text777one-two')
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001290enddef
1291
Bram Moolenaar23e03252020-04-12 22:22:31 +02001292func Test_multiline_not_vim9()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001293 call MultiLine('text')->assert_equal('text1234')
1294 call MultiLine('text', 777)->assert_equal('text777')
1295 call MultiLine('text', 777, 'one')->assert_equal('text777one')
1296 call MultiLine('text', 777, 'one', 'two')->assert_equal('text777one-two')
Bram Moolenaar23e03252020-04-12 22:22:31 +02001297endfunc
1298
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001299
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02001300" When using CheckScriptFailure() for the below test, E1010 is generated instead
1301" of E1056.
1302func Test_E1056_1059()
1303 let caught_1056 = 0
1304 try
1305 def F():
1306 return 1
1307 enddef
1308 catch /E1056:/
1309 let caught_1056 = 1
1310 endtry
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001311 eval caught_1056->assert_equal(1)
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02001312
1313 let caught_1059 = 0
1314 try
1315 def F5(items : list)
1316 echo 'a'
1317 enddef
1318 catch /E1059:/
1319 let caught_1059 = 1
1320 endtry
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001321 eval caught_1059->assert_equal(1)
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02001322endfunc
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001323
Bram Moolenaar015f4262020-05-05 21:25:22 +02001324func DelMe()
1325 echo 'DelMe'
1326endfunc
1327
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001328def Test_error_reporting()
1329 # comment lines at the start of the function
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001330 var lines =<< trim END
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001331 " comment
1332 def Func()
1333 # comment
1334 # comment
1335 invalid
1336 enddef
1337 defcompile
1338 END
Bram Moolenaar08052222020-09-14 17:04:31 +02001339 writefile(lines, 'Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001340 try
1341 source Xdef
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001342 assert_report('should have failed')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001343 catch /E476:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001344 v:exception->assert_match('Invalid command: invalid')
1345 v:throwpoint->assert_match(', line 3$')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001346 endtry
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001347 delfunc! g:Func
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001348
1349 # comment lines after the start of the function
1350 lines =<< trim END
1351 " comment
1352 def Func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001353 var x = 1234
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001354 # comment
1355 # comment
1356 invalid
1357 enddef
1358 defcompile
1359 END
Bram Moolenaar08052222020-09-14 17:04:31 +02001360 writefile(lines, 'Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001361 try
1362 source Xdef
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001363 assert_report('should have failed')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001364 catch /E476:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001365 v:exception->assert_match('Invalid command: invalid')
1366 v:throwpoint->assert_match(', line 4$')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001367 endtry
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001368 delfunc! g:Func
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001369
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001370 lines =<< trim END
1371 vim9script
1372 def Func()
Bram Moolenaare0de1712020-12-02 17:36:54 +01001373 var db = {foo: 1, bar: 2}
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001374 # comment
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001375 var x = db.asdf
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001376 enddef
1377 defcompile
1378 Func()
1379 END
Bram Moolenaar08052222020-09-14 17:04:31 +02001380 writefile(lines, 'Xdef')
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001381 try
1382 source Xdef
1383 assert_report('should have failed')
1384 catch /E716:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001385 v:throwpoint->assert_match('_Func, line 3$')
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001386 endtry
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001387 delfunc! g:Func
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001388
Bram Moolenaar08052222020-09-14 17:04:31 +02001389 delete('Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001390enddef
1391
Bram Moolenaar015f4262020-05-05 21:25:22 +02001392def Test_deleted_function()
1393 CheckDefExecFailure([
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001394 'var RefMe: func = function("g:DelMe")',
Bram Moolenaar015f4262020-05-05 21:25:22 +02001395 'delfunc g:DelMe',
1396 'echo RefMe()'], 'E117:')
1397enddef
1398
1399def Test_unknown_function()
1400 CheckDefExecFailure([
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001401 'var Ref: func = function("NotExist")',
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +02001402 'delfunc g:NotExist'], 'E700:')
Bram Moolenaar015f4262020-05-05 21:25:22 +02001403enddef
1404
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +02001405def RefFunc(Ref: func(string): string): string
1406 return Ref('more')
1407enddef
1408
1409def Test_closure_simple()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001410 var local = 'some '
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001411 RefFunc({s -> local .. s})->assert_equal('some more')
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +02001412enddef
1413
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001414def MakeRef()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001415 var local = 'some '
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001416 g:Ref = {s -> local .. s}
1417enddef
1418
1419def Test_closure_ref_after_return()
1420 MakeRef()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001421 g:Ref('thing')->assert_equal('some thing')
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001422 unlet g:Ref
1423enddef
1424
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001425def MakeTwoRefs()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001426 var local = ['some']
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001427 g:Extend = {s -> local->add(s)}
1428 g:Read = {-> local}
1429enddef
1430
1431def Test_closure_two_refs()
1432 MakeTwoRefs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001433 join(g:Read(), ' ')->assert_equal('some')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001434 g:Extend('more')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001435 join(g:Read(), ' ')->assert_equal('some more')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001436 g:Extend('even')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001437 join(g:Read(), ' ')->assert_equal('some more even')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001438
1439 unlet g:Extend
1440 unlet g:Read
1441enddef
1442
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001443def ReadRef(Ref: func(): list<string>): string
1444 return join(Ref(), ' ')
1445enddef
1446
Bram Moolenaar5e654232020-09-16 15:22:00 +02001447def ExtendRef(Ref: func(string): list<string>, add: string)
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001448 Ref(add)
1449enddef
1450
1451def Test_closure_two_indirect_refs()
Bram Moolenaarf7779c62020-05-03 15:38:16 +02001452 MakeTwoRefs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001453 ReadRef(g:Read)->assert_equal('some')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001454 ExtendRef(g:Extend, 'more')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001455 ReadRef(g:Read)->assert_equal('some more')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001456 ExtendRef(g:Extend, 'even')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001457 ReadRef(g:Read)->assert_equal('some more even')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001458
1459 unlet g:Extend
1460 unlet g:Read
1461enddef
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001462
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001463def MakeArgRefs(theArg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001464 var local = 'loc_val'
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001465 g:UseArg = {s -> theArg .. '/' .. local .. '/' .. s}
1466enddef
1467
1468def MakeArgRefsVarargs(theArg: string, ...rest: list<string>)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001469 var local = 'the_loc'
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001470 g:UseVararg = {s -> theArg .. '/' .. local .. '/' .. s .. '/' .. join(rest)}
1471enddef
1472
1473def Test_closure_using_argument()
1474 MakeArgRefs('arg_val')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001475 g:UseArg('call_val')->assert_equal('arg_val/loc_val/call_val')
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001476
1477 MakeArgRefsVarargs('arg_val', 'one', 'two')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001478 g:UseVararg('call_val')->assert_equal('arg_val/the_loc/call_val/one two')
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001479
1480 unlet g:UseArg
1481 unlet g:UseVararg
1482enddef
1483
Bram Moolenaar85d5e2b2020-10-10 14:13:01 +02001484def MakeGetAndAppendRefs()
1485 var local = 'a'
1486
1487 def Append(arg: string)
1488 local ..= arg
1489 enddef
1490 g:Append = Append
1491
1492 def Get(): string
1493 return local
1494 enddef
1495 g:Get = Get
1496enddef
1497
1498def Test_closure_append_get()
1499 MakeGetAndAppendRefs()
1500 g:Get()->assert_equal('a')
1501 g:Append('-b')
1502 g:Get()->assert_equal('a-b')
1503 g:Append('-c')
1504 g:Get()->assert_equal('a-b-c')
1505
1506 unlet g:Append
1507 unlet g:Get
1508enddef
Bram Moolenaarb68b3462020-05-06 21:06:30 +02001509
Bram Moolenaar04b12692020-05-04 23:24:44 +02001510def Test_nested_closure()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001511 var local = 'text'
Bram Moolenaar04b12692020-05-04 23:24:44 +02001512 def Closure(arg: string): string
1513 return local .. arg
1514 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001515 Closure('!!!')->assert_equal('text!!!')
Bram Moolenaar04b12692020-05-04 23:24:44 +02001516enddef
1517
Bram Moolenaar6f5b6df2020-05-16 21:20:12 +02001518func GetResult(Ref)
1519 return a:Ref('some')
1520endfunc
1521
1522def Test_call_closure_not_compiled()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001523 var text = 'text'
Bram Moolenaar6f5b6df2020-05-16 21:20:12 +02001524 g:Ref = {s -> s .. text}
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001525 GetResult(g:Ref)->assert_equal('sometext')
Bram Moolenaar6f5b6df2020-05-16 21:20:12 +02001526enddef
1527
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001528def Test_double_closure_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001529 var lines =<< trim END
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001530 vim9script
1531 def Func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001532 var name = 0
1533 for i in range(2)
1534 timer_start(0, {-> name})
1535 endfor
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001536 enddef
1537 Func()
1538 END
Bram Moolenaar148ce7a2020-09-23 21:57:23 +02001539 CheckScriptSuccess(lines)
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001540enddef
1541
Bram Moolenaar85d5e2b2020-10-10 14:13:01 +02001542def Test_nested_closure_used()
1543 var lines =<< trim END
1544 vim9script
1545 def Func()
1546 var x = 'hello'
1547 var Closure = {-> x}
1548 g:Myclosure = {-> Closure()}
1549 enddef
1550 Func()
1551 assert_equal('hello', g:Myclosure())
1552 END
1553 CheckScriptSuccess(lines)
1554enddef
Bram Moolenaar0876c782020-10-07 19:08:04 +02001555
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02001556def Test_nested_closure_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001557 var lines =<< trim END
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02001558 vim9script
1559 def FuncA()
1560 FuncB(0)
1561 enddef
1562 def FuncB(n: number): list<string>
1563 return map([0], {_, v -> n})
1564 enddef
1565 FuncA()
1566 END
1567 CheckScriptFailure(lines, 'E1012:')
1568enddef
1569
Bram Moolenaarf112f302020-12-20 17:47:52 +01001570def Test_global_closure()
1571 var lines =<< trim END
1572 vim9script
1573 def ReverseEveryNLines(n: number, line1: number, line2: number)
1574 var mods = 'sil keepj keepp lockm '
1575 var range = ':' .. line1 .. ',' .. line2
1576 def g:Offset(): number
1577 var offset = (line('.') - line1 + 1) % n
1578 return offset != 0 ? offset : n
1579 enddef
1580 exe mods .. range .. 'g/^/exe "m .-" .. g:Offset()'
1581 enddef
1582
1583 new
1584 repeat(['aaa', 'bbb', 'ccc'], 3)->setline(1)
1585 ReverseEveryNLines(3, 1, 9)
1586 END
1587 CheckScriptSuccess(lines)
1588 var expected = repeat(['ccc', 'bbb', 'aaa'], 3)
1589 assert_equal(expected, getline(1, 9))
1590 bwipe!
1591enddef
1592
Bram Moolenaarcd45ed02020-12-22 17:35:54 +01001593def Test_global_closure_called_directly()
1594 var lines =<< trim END
1595 vim9script
1596 def Outer()
1597 var x = 1
1598 def g:Inner()
1599 var y = x
1600 x += 1
1601 assert_equal(1, y)
1602 enddef
1603 g:Inner()
1604 assert_equal(2, x)
1605 enddef
1606 Outer()
1607 END
1608 CheckScriptSuccess(lines)
1609 delfunc g:Inner
1610enddef
1611
Bram Moolenaar34c54eb2020-11-25 19:15:19 +01001612def Test_failure_in_called_function()
1613 # this was using the frame index as the return value
1614 var lines =<< trim END
1615 vim9script
1616 au TerminalWinOpen * eval [][0]
1617 def PopupTerm(a: any)
1618 # make sure typvals on stack are string
1619 ['a', 'b', 'c', 'd', 'e', 'f', 'g']->join()
1620 FireEvent()
1621 enddef
1622 def FireEvent()
1623 do TerminalWinOpen
1624 enddef
1625 # use try/catch to make eval fail
1626 try
1627 call PopupTerm(0)
1628 catch
1629 endtry
1630 au! TerminalWinOpen
1631 END
1632 CheckScriptSuccess(lines)
1633enddef
1634
Bram Moolenaar5366e1a2020-10-01 13:01:34 +02001635def Test_nested_lambda()
1636 var lines =<< trim END
1637 vim9script
1638 def Func()
1639 var x = 4
1640 var Lambda1 = {-> 7}
1641 var Lambda2 = {-> [Lambda1(), x]}
1642 var res = Lambda2()
1643 assert_equal([7, 4], res)
1644 enddef
1645 Func()
1646 END
1647 CheckScriptSuccess(lines)
1648enddef
1649
Bram Moolenaar52bf81c2020-11-17 18:50:44 +01001650def Shadowed(): list<number>
1651 var FuncList: list<func: number> = [{ -> 42}]
1652 return FuncList->map({_, Shadowed -> Shadowed()})
1653enddef
1654
1655def Test_lambda_arg_shadows_func()
1656 assert_equal([42], Shadowed())
1657enddef
1658
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02001659def Line_continuation_in_def(dir: string = ''): string
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001660 var path: string = empty(dir)
1661 \ ? 'empty'
1662 \ : 'full'
1663 return path
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02001664enddef
1665
1666def Test_line_continuation_in_def()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001667 Line_continuation_in_def('.')->assert_equal('full')
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02001668enddef
1669
Bram Moolenaar2ea95b62020-11-19 21:47:56 +01001670def Test_script_var_in_lambda()
1671 var lines =<< trim END
1672 vim9script
1673 var script = 'test'
1674 assert_equal(['test'], map(['one'], {-> script}))
1675 END
1676 CheckScriptSuccess(lines)
1677enddef
1678
Bram Moolenaar5e654232020-09-16 15:22:00 +02001679def Line_continuation_in_lambda(): list<string>
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001680 var x = range(97, 100)
Bram Moolenaar914e7ea2020-07-11 15:20:48 +02001681 ->map({_, v -> nr2char(v)
Bram Moolenaar7a4b8982020-07-08 17:36:21 +02001682 ->toupper()})
1683 ->reverse()
1684 return x
1685enddef
1686
1687def Test_line_continuation_in_lambda()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001688 Line_continuation_in_lambda()->assert_equal(['D', 'C', 'B', 'A'])
Bram Moolenaar7a4b8982020-07-08 17:36:21 +02001689enddef
1690
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001691func Test_silent_echo()
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001692 CheckScreendump
1693
1694 let lines =<< trim END
1695 vim9script
1696 def EchoNothing()
1697 silent echo ''
1698 enddef
1699 defcompile
1700 END
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001701 call writefile(lines, 'XTest_silent_echo')
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001702
1703 " Check that the balloon shows up after a mouse move
1704 let buf = RunVimInTerminal('-S XTest_silent_echo', {'rows': 6})
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001705 call term_sendkeys(buf, ":abc")
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001706 call VerifyScreenDump(buf, 'Test_vim9_silent_echo', {})
1707
1708 " clean up
1709 call StopVimInTerminal(buf)
1710 call delete('XTest_silent_echo')
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001711endfunc
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001712
Bram Moolenaar171fb922020-10-28 16:54:47 +01001713def SilentlyError()
1714 execute('silent! invalid')
1715 g:did_it = 'yes'
1716enddef
1717
Bram Moolenaar28ee8922020-10-28 20:20:00 +01001718func UserError()
1719 silent! invalid
1720endfunc
1721
1722def SilentlyUserError()
1723 UserError()
1724 g:did_it = 'yes'
1725enddef
Bram Moolenaar171fb922020-10-28 16:54:47 +01001726
1727" This can't be a :def function, because the assert would not be reached.
Bram Moolenaar171fb922020-10-28 16:54:47 +01001728func Test_ignore_silent_error()
1729 let g:did_it = 'no'
1730 call SilentlyError()
1731 call assert_equal('yes', g:did_it)
1732
Bram Moolenaar28ee8922020-10-28 20:20:00 +01001733 let g:did_it = 'no'
1734 call SilentlyUserError()
1735 call assert_equal('yes', g:did_it)
Bram Moolenaar171fb922020-10-28 16:54:47 +01001736
1737 unlet g:did_it
1738endfunc
1739
Bram Moolenaarcd030c42020-10-30 21:49:40 +01001740def Test_ignore_silent_error_in_filter()
1741 var lines =<< trim END
1742 vim9script
1743 def Filter(winid: number, key: string): bool
1744 if key == 'o'
1745 silent! eval [][0]
1746 return true
1747 endif
1748 return popup_filter_menu(winid, key)
1749 enddef
1750
Bram Moolenaare0de1712020-12-02 17:36:54 +01001751 popup_create('popup', {filter: Filter})
Bram Moolenaarcd030c42020-10-30 21:49:40 +01001752 feedkeys("o\r", 'xnt')
1753 END
1754 CheckScriptSuccess(lines)
1755enddef
1756
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001757def Fibonacci(n: number): number
1758 if n < 2
1759 return n
1760 else
1761 return Fibonacci(n - 1) + Fibonacci(n - 2)
1762 endif
1763enddef
1764
Bram Moolenaar985116a2020-07-12 17:31:09 +02001765def Test_recursive_call()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001766 Fibonacci(20)->assert_equal(6765)
Bram Moolenaar985116a2020-07-12 17:31:09 +02001767enddef
1768
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001769def TreeWalk(dir: string): list<any>
1770 return readdir(dir)->map({_, val ->
1771 fnamemodify(dir .. '/' .. val, ':p')->isdirectory()
Bram Moolenaar2bede172020-11-19 18:53:18 +01001772 ? {[val]: TreeWalk(dir .. '/' .. val)}
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001773 : val
1774 })
1775enddef
1776
1777def Test_closure_in_map()
1778 mkdir('XclosureDir/tdir', 'p')
1779 writefile(['111'], 'XclosureDir/file1')
1780 writefile(['222'], 'XclosureDir/file2')
1781 writefile(['333'], 'XclosureDir/tdir/file3')
1782
Bram Moolenaare0de1712020-12-02 17:36:54 +01001783 TreeWalk('XclosureDir')->assert_equal(['file1', 'file2', {tdir: ['file3']}])
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001784
1785 delete('XclosureDir', 'rf')
1786enddef
1787
Bram Moolenaar7b5d5442020-10-04 13:42:34 +02001788def Test_invalid_function_name()
1789 var lines =<< trim END
1790 vim9script
1791 def s: list<string>
1792 END
1793 CheckScriptFailure(lines, 'E129:')
1794
1795 lines =<< trim END
1796 vim9script
1797 def g: list<string>
1798 END
1799 CheckScriptFailure(lines, 'E129:')
1800
1801 lines =<< trim END
1802 vim9script
1803 def <SID>: list<string>
1804 END
1805 CheckScriptFailure(lines, 'E884:')
1806
1807 lines =<< trim END
1808 vim9script
1809 def F list<string>
1810 END
1811 CheckScriptFailure(lines, 'E488:')
1812enddef
1813
Bram Moolenaara90afb92020-07-15 22:38:56 +02001814def Test_partial_call()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001815 var Xsetlist = function('setloclist', [0])
Bram Moolenaare0de1712020-12-02 17:36:54 +01001816 Xsetlist([], ' ', {title: 'test'})
1817 getloclist(0, {title: 1})->assert_equal({title: 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02001818
1819 Xsetlist = function('setloclist', [0, [], ' '])
Bram Moolenaare0de1712020-12-02 17:36:54 +01001820 Xsetlist({title: 'test'})
1821 getloclist(0, {title: 1})->assert_equal({title: 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02001822
1823 Xsetlist = function('setqflist')
Bram Moolenaare0de1712020-12-02 17:36:54 +01001824 Xsetlist([], ' ', {title: 'test'})
1825 getqflist({title: 1})->assert_equal({title: 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02001826
1827 Xsetlist = function('setqflist', [[], ' '])
Bram Moolenaare0de1712020-12-02 17:36:54 +01001828 Xsetlist({title: 'test'})
1829 getqflist({title: 1})->assert_equal({title: 'test'})
Bram Moolenaar6abd3dc2020-10-04 14:17:32 +02001830
1831 var Len: func: number = function('len', ['word'])
1832 assert_equal(4, Len())
Bram Moolenaara90afb92020-07-15 22:38:56 +02001833enddef
1834
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001835def Test_cmd_modifier()
1836 tab echo '0'
Bram Moolenaard2c61702020-09-06 15:58:36 +02001837 CheckDefFailure(['5tab echo 3'], 'E16:')
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001838enddef
1839
1840def Test_restore_modifiers()
1841 # check that when compiling a :def function command modifiers are not messed
1842 # up.
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001843 var lines =<< trim END
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001844 vim9script
1845 set eventignore=
1846 autocmd QuickFixCmdPost * copen
1847 def AutocmdsDisabled()
Bram Moolenaar6cf7e3b2020-10-28 14:31:16 +01001848 eval 0
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001849 enddef
1850 func Func()
1851 noautocmd call s:AutocmdsDisabled()
1852 let g:ei_after = &eventignore
1853 endfunc
1854 Func()
1855 END
1856 CheckScriptSuccess(lines)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001857 g:ei_after->assert_equal('')
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001858enddef
1859
Bram Moolenaardfa3d552020-09-10 22:05:08 +02001860def StackTop()
1861 eval 1
1862 eval 2
1863 # call not on fourth line
1864 StackBot()
1865enddef
1866
1867def StackBot()
1868 # throw an error
1869 eval [][0]
1870enddef
1871
1872def Test_callstack_def()
1873 try
1874 StackTop()
1875 catch
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001876 v:throwpoint->assert_match('Test_callstack_def\[2\]..StackTop\[4\]..StackBot, line 2')
Bram Moolenaardfa3d552020-09-10 22:05:08 +02001877 endtry
1878enddef
1879
Bram Moolenaare8211a32020-10-09 22:04:29 +02001880" Re-using spot for variable used in block
1881def Test_block_scoped_var()
1882 var lines =<< trim END
1883 vim9script
1884 def Func()
1885 var x = ['a', 'b', 'c']
1886 if 1
1887 var y = 'x'
1888 map(x, {-> y})
1889 endif
1890 var z = x
1891 assert_equal(['x', 'x', 'x'], z)
1892 enddef
1893 Func()
1894 END
1895 CheckScriptSuccess(lines)
1896enddef
1897
Bram Moolenaareeece9e2020-11-20 19:26:48 +01001898def Test_reset_did_emsg()
1899 var lines =<< trim END
1900 @s = 'blah'
1901 au BufWinLeave * #
1902 def Func()
1903 var winid = popup_create('popup', {})
1904 exe '*s'
1905 popup_close(winid)
1906 enddef
1907 Func()
1908 END
1909 CheckScriptFailure(lines, 'E492:', 8)
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001910 delfunc! g:Func
Bram Moolenaareeece9e2020-11-20 19:26:48 +01001911enddef
1912
Bram Moolenaar57f799e2020-12-12 20:42:19 +01001913def Test_did_emsg_reset()
1914 # executing an autocommand resets did_emsg, this should not result in a
1915 # builtin function considered failing
1916 var lines =<< trim END
1917 vim9script
1918 au BufWinLeave * #
1919 def Func()
1920 popup_menu('', {callback: {-> popup_create('', {})->popup_close()}})
1921 eval [][0]
1922 enddef
1923 nno <F3> <cmd>call <sid>Func()<cr>
1924 feedkeys("\<F3>\e", 'xt')
1925 END
1926 writefile(lines, 'XemsgReset')
1927 assert_fails('so XemsgReset', ['E684:', 'E684:'], lines, 2)
1928 delete('XemsgReset')
1929 nunmap <F3>
1930 au! BufWinLeave
1931enddef
1932
Bram Moolenaar56602ba2020-12-05 21:22:08 +01001933def Test_abort_with_silent_call()
1934 var lines =<< trim END
1935 vim9script
1936 g:result = 'none'
1937 def Func()
1938 g:result += 3
1939 g:result = 'yes'
1940 enddef
1941 # error is silenced, but function aborts on error
1942 silent! Func()
1943 assert_equal('none', g:result)
1944 unlet g:result
1945 END
1946 CheckScriptSuccess(lines)
1947enddef
1948
Bram Moolenaarf665e972020-12-05 19:17:16 +01001949def Test_continues_with_silent_error()
1950 var lines =<< trim END
1951 vim9script
1952 g:result = 'none'
1953 def Func()
1954 silent! g:result += 3
1955 g:result = 'yes'
1956 enddef
1957 # error is silenced, function does not abort
1958 Func()
1959 assert_equal('yes', g:result)
1960 unlet g:result
1961 END
1962 CheckScriptSuccess(lines)
1963enddef
1964
Bram Moolenaaraf0df472020-12-02 20:51:22 +01001965def Test_abort_even_with_silent()
1966 var lines =<< trim END
1967 vim9script
1968 g:result = 'none'
1969 def Func()
1970 eval {-> ''}() .. '' .. {}['X']
1971 g:result = 'yes'
1972 enddef
Bram Moolenaarf665e972020-12-05 19:17:16 +01001973 silent! Func()
Bram Moolenaaraf0df472020-12-02 20:51:22 +01001974 assert_equal('none', g:result)
Bram Moolenaar4029cab2020-12-05 18:13:27 +01001975 unlet g:result
1976 END
1977 CheckScriptSuccess(lines)
1978enddef
1979
Bram Moolenaarf665e972020-12-05 19:17:16 +01001980def Test_cmdmod_silent_restored()
1981 var lines =<< trim END
1982 vim9script
1983 def Func()
1984 g:result = 'none'
1985 silent! g:result += 3
1986 g:result = 'none'
1987 g:result += 3
1988 enddef
1989 Func()
1990 END
1991 # can't use CheckScriptFailure, it ignores the :silent!
1992 var fname = 'Xdefsilent'
1993 writefile(lines, fname)
1994 var caught = 'no'
1995 try
1996 exe 'source ' .. fname
1997 catch /E1030:/
1998 caught = 'yes'
1999 assert_match('Func, line 4', v:throwpoint)
2000 endtry
2001 assert_equal('yes', caught)
2002 delete(fname)
2003enddef
2004
Bram Moolenaar4029cab2020-12-05 18:13:27 +01002005def Test_dict_member_with_silent()
2006 var lines =<< trim END
2007 vim9script
2008 g:result = 'none'
2009 var d: dict<any>
2010 def Func()
2011 try
2012 g:result = map([], {_, v -> {}[v]})->join() .. d['']
2013 catch
2014 endtry
2015 enddef
2016 silent! Func()
2017 assert_equal('0', g:result)
2018 unlet g:result
Bram Moolenaaraf0df472020-12-02 20:51:22 +01002019 END
2020 CheckScriptSuccess(lines)
2021enddef
2022
Bram Moolenaar5b3d1bb2020-12-22 12:20:08 +01002023def Test_opfunc()
2024 nnoremap <F3> <cmd>set opfunc=Opfunc<cr>g@
2025 def g:Opfunc(_: any): string
2026 setline(1, 'ASDF')
2027 return ''
2028 enddef
2029 new
2030 setline(1, 'asdf')
2031 feedkeys("\<F3>$", 'x')
2032 assert_equal('ASDF', getline(1))
2033
2034 bwipe!
2035 nunmap <F3>
2036enddef
2037
Bram Moolenaar077a4232020-12-22 18:33:27 +01002038" this was crashing on exit
2039def Test_nested_lambda_in_closure()
2040 var lines =<< trim END
2041 vim9script
2042 def Outer()
2043 def g:Inner()
2044 echo map([1, 2, 3], {_, v -> v + 1})
2045 enddef
2046 g:Inner()
2047 enddef
2048 defcompile
2049 writefile(['Done'], 'XnestedDone')
2050 quit
2051 END
2052 if !RunVim([], lines, '--clean')
2053 return
2054 endif
2055 assert_equal(['Done'], readfile('XnestedDone'))
2056 delete('XnestedDone')
2057enddef
2058
2059
Bram Moolenaarf7779c62020-05-03 15:38:16 +02002060
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02002061" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker