blob: 4f741bb45d923d7441be7f1e5317376d3c7361e6 [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
16def ReturnString(): string
17 return 'string'
18enddef
19
20def ReturnNumber(): number
21 return 123
22enddef
23
24let g:notNumber = 'string'
25
26def ReturnGlobal(): number
27 return g:notNumber
28enddef
29
30def Test_return_something()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +020031 ReturnString()->assert_equal('string')
32 ReturnNumber()->assert_equal(123)
Bram Moolenaar5e654232020-09-16 15:22:00 +020033 assert_fails('ReturnGlobal()', 'E1012: Type mismatch; expected number but got string', '', 1, 'ReturnGlobal')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +020034enddef
35
Bram Moolenaarefd88552020-06-18 20:50:10 +020036def Test_missing_return()
37 CheckDefFailure(['def Missing(): number',
38 ' if g:cond',
39 ' echo "no return"',
40 ' else',
41 ' return 0',
42 ' endif'
43 'enddef'], 'E1027:')
44 CheckDefFailure(['def Missing(): number',
45 ' if g:cond',
46 ' return 1',
47 ' else',
48 ' echo "no return"',
49 ' endif'
50 'enddef'], 'E1027:')
51 CheckDefFailure(['def Missing(): number',
52 ' if g:cond',
53 ' return 1',
54 ' else',
55 ' return 2',
56 ' endif'
57 ' return 3'
58 'enddef'], 'E1095:')
59enddef
60
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +020061let s:nothing = 0
62def ReturnNothing()
63 s:nothing = 1
64 if true
65 return
66 endif
67 s:nothing = 2
68enddef
69
70def Test_return_nothing()
71 ReturnNothing()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +020072 s:nothing->assert_equal(1)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +020073enddef
74
75func Increment()
76 let g:counter += 1
77endfunc
78
79def Test_call_ufunc_count()
80 g:counter = 1
81 Increment()
82 Increment()
83 Increment()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +020084 # works with and without :call
Bram Moolenaarc0c71e92020-09-11 19:09:48 +020085 g:counter->assert_equal(4)
86 eval g:counter->assert_equal(4)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +020087 unlet g:counter
88enddef
89
90def MyVarargs(arg: string, ...rest: list<string>): string
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +020091 var res = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +020092 for s in rest
93 res ..= ',' .. s
94 endfor
95 return res
96enddef
97
98def Test_call_varargs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +020099 MyVarargs('one')->assert_equal('one')
100 MyVarargs('one', 'two')->assert_equal('one,two')
101 MyVarargs('one', 'two', 'three')->assert_equal('one,two,three')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200102enddef
103
104def MyDefaultArgs(name = 'string'): string
105 return name
106enddef
107
Bram Moolenaare30f64b2020-07-15 19:48:20 +0200108def MyDefaultSecond(name: string, second: bool = true): string
109 return second ? name : 'none'
110enddef
111
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200112def Test_call_default_args()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200113 MyDefaultArgs()->assert_equal('string')
114 MyDefaultArgs('one')->assert_equal('one')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200115 assert_fails('MyDefaultArgs("one", "two")', 'E118:', '', 3, 'Test_call_default_args')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200116
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200117 MyDefaultSecond('test')->assert_equal('test')
118 MyDefaultSecond('test', true)->assert_equal('test')
119 MyDefaultSecond('test', false)->assert_equal('none')
Bram Moolenaare30f64b2020-07-15 19:48:20 +0200120
Bram Moolenaar822ba242020-05-24 23:00:18 +0200121 CheckScriptFailure(['def Func(arg: number = asdf)', 'enddef', 'defcompile'], 'E1001:')
Bram Moolenaar77072282020-09-16 17:55:40 +0200122 CheckScriptFailure(['def Func(arg: number = "text")', 'enddef', 'defcompile'], 'E1013: Argument 1: type mismatch, expected number but got string')
Bram Moolenaar04b12692020-05-04 23:24:44 +0200123enddef
124
125def Test_nested_function()
126 def Nested(arg: string): string
127 return 'nested ' .. arg
128 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200129 Nested('function')->assert_equal('nested function')
Bram Moolenaar04b12692020-05-04 23:24:44 +0200130
Bram Moolenaar0e65d3d2020-05-05 17:53:16 +0200131 CheckDefFailure(['def Nested()', 'enddef', 'Nested(66)'], 'E118:')
132 CheckDefFailure(['def Nested(arg: string)', 'enddef', 'Nested()'], 'E119:')
133
Bram Moolenaar04b12692020-05-04 23:24:44 +0200134 CheckDefFailure(['func Nested()', 'endfunc'], 'E1086:')
Bram Moolenaarbcbf4132020-08-01 22:35:13 +0200135 CheckDefFailure(['def s:Nested()', 'enddef'], 'E1075:')
136 CheckDefFailure(['def b:Nested()', 'enddef'], 'E1075:')
Bram Moolenaar8b848ca2020-09-10 22:28:01 +0200137
138 CheckDefFailure([
139 'def Outer()',
140 ' def Inner()',
141 ' # comment',
142 ' enddef',
143 ' def Inner()',
144 ' enddef',
145 'enddef'], 'E1073:')
146 CheckDefFailure([
147 'def Outer()',
148 ' def Inner()',
149 ' # comment',
150 ' enddef',
151 ' def! Inner()',
152 ' enddef',
153 'enddef'], 'E1117:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200154enddef
155
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200156func Test_call_default_args_from_func()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200157 call MyDefaultArgs()->assert_equal('string')
158 call MyDefaultArgs('one')->assert_equal('one')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200159 call assert_fails('call MyDefaultArgs("one", "two")', 'E118:', '', 3, 'Test_call_default_args_from_func')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200160endfunc
161
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200162def Test_nested_global_function()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200163 var lines =<< trim END
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200164 vim9script
165 def Outer()
166 def g:Inner(): string
167 return 'inner'
168 enddef
169 enddef
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200170 defcompile
171 Outer()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200172 g:Inner()->assert_equal('inner')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200173 delfunc g:Inner
174 Outer()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200175 g:Inner()->assert_equal('inner')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200176 delfunc g:Inner
177 Outer()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200178 g:Inner()->assert_equal('inner')
Bram Moolenaaraf8edbb2020-08-01 00:03:09 +0200179 delfunc g:Inner
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200180 END
181 CheckScriptSuccess(lines)
Bram Moolenaar2c79e9d2020-08-01 18:57:52 +0200182
183 lines =<< trim END
184 vim9script
185 def Outer()
186 def g:Inner(): string
187 return 'inner'
188 enddef
189 enddef
190 defcompile
191 Outer()
192 Outer()
193 END
194 CheckScriptFailure(lines, "E122:")
Bram Moolenaarad486a02020-08-01 23:22:18 +0200195
196 lines =<< trim END
197 vim9script
198 def Func()
199 echo 'script'
200 enddef
201 def Outer()
202 def Func()
203 echo 'inner'
204 enddef
205 enddef
206 defcompile
207 END
208 CheckScriptFailure(lines, "E1073:")
Bram Moolenaar38ddf332020-07-31 22:05:04 +0200209enddef
210
Bram Moolenaar333894b2020-08-01 18:53:07 +0200211def Test_global_local_function()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200212 var lines =<< trim END
Bram Moolenaar333894b2020-08-01 18:53:07 +0200213 vim9script
214 def g:Func(): string
215 return 'global'
216 enddef
217 def Func(): string
218 return 'local'
219 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200220 g:Func()->assert_equal('global')
221 Func()->assert_equal('local')
Bram Moolenaar333894b2020-08-01 18:53:07 +0200222 END
223 CheckScriptSuccess(lines)
Bram Moolenaar035d6e92020-08-11 22:30:42 +0200224
225 lines =<< trim END
226 vim9script
227 def g:Funcy()
228 echo 'funcy'
229 enddef
230 s:Funcy()
231 END
232 CheckScriptFailure(lines, 'E117:')
Bram Moolenaar333894b2020-08-01 18:53:07 +0200233enddef
234
Bram Moolenaar0f769812020-09-12 18:32:34 +0200235def Test_local_function_shadows_global()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200236 var lines =<< trim END
Bram Moolenaar0f769812020-09-12 18:32:34 +0200237 vim9script
238 def g:Gfunc(): string
239 return 'global'
240 enddef
241 def AnotherFunc(): number
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200242 var Gfunc = function('len')
Bram Moolenaar0f769812020-09-12 18:32:34 +0200243 return Gfunc('testing')
244 enddef
245 g:Gfunc()->assert_equal('global')
246 AnotherFunc()->assert_equal(7)
247 delfunc g:Gfunc
248 END
249 CheckScriptSuccess(lines)
250
251 lines =<< trim END
252 vim9script
253 def g:Func(): string
254 return 'global'
255 enddef
256 def AnotherFunc()
257 g:Func = function('len')
258 enddef
259 AnotherFunc()
260 END
261 CheckScriptFailure(lines, 'E705:')
262 delfunc g:Func
263enddef
264
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200265func TakesOneArg(arg)
266 echo a:arg
267endfunc
268
269def Test_call_wrong_args()
Bram Moolenaard2c61702020-09-06 15:58:36 +0200270 CheckDefFailure(['TakesOneArg()'], 'E119:')
271 CheckDefFailure(['TakesOneArg(11, 22)'], 'E118:')
272 CheckDefFailure(['bufnr(xxx)'], 'E1001:')
273 CheckScriptFailure(['def Func(Ref: func(s: string))'], 'E475:')
Bram Moolenaaree8580e2020-08-28 17:19:07 +0200274
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200275 var lines =<< trim END
Bram Moolenaaree8580e2020-08-28 17:19:07 +0200276 vim9script
277 def Func(s: string)
278 echo s
279 enddef
280 Func([])
281 END
Bram Moolenaar77072282020-09-16 17:55:40 +0200282 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected string but got list<unknown>', 5)
Bram Moolenaarb185a402020-09-18 22:42:00 +0200283
284 lines =<< trim END
285 vim9script
286 def FuncOne(nr: number)
287 echo nr
288 enddef
289 def FuncTwo()
290 FuncOne()
291 enddef
292 defcompile
293 END
294 writefile(lines, 'Xscript')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200295 var didCatch = false
Bram Moolenaarb185a402020-09-18 22:42:00 +0200296 try
297 source Xscript
298 catch
299 assert_match('E119: Not enough arguments for function: <SNR>\d\+_FuncOne', v:exception)
300 assert_match('Xscript\[8\]..function <SNR>\d\+_FuncTwo, line 1', v:throwpoint)
301 didCatch = true
302 endtry
303 assert_true(didCatch)
304
305 lines =<< trim END
306 vim9script
307 def FuncOne(nr: number)
308 echo nr
309 enddef
310 def FuncTwo()
311 FuncOne(1, 2)
312 enddef
313 defcompile
314 END
315 writefile(lines, 'Xscript')
316 didCatch = false
317 try
318 source Xscript
319 catch
320 assert_match('E118: Too many arguments for function: <SNR>\d\+_FuncOne', v:exception)
321 assert_match('Xscript\[8\]..function <SNR>\d\+_FuncTwo, line 1', v:throwpoint)
322 didCatch = true
323 endtry
324 assert_true(didCatch)
325
326 delete('Xscript')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200327enddef
328
329" Default arg and varargs
330def MyDefVarargs(one: string, two = 'foo', ...rest: list<string>): string
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200331 var res = one .. ',' .. two
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200332 for s in rest
333 res ..= ',' .. s
334 endfor
335 return res
336enddef
337
338def Test_call_def_varargs()
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200339 assert_fails('MyDefVarargs()', 'E119:', '', 1, 'Test_call_def_varargs')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200340 MyDefVarargs('one')->assert_equal('one,foo')
341 MyDefVarargs('one', 'two')->assert_equal('one,two')
342 MyDefVarargs('one', 'two', 'three')->assert_equal('one,two,three')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200343 CheckDefFailure(['MyDefVarargs("one", 22)'],
Bram Moolenaar77072282020-09-16 17:55:40 +0200344 'E1013: Argument 2: type mismatch, expected string but got number')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200345 CheckDefFailure(['MyDefVarargs("one", "two", 123)'],
Bram Moolenaar77072282020-09-16 17:55:40 +0200346 'E1013: Argument 3: type mismatch, expected string but got number')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200347
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200348 var lines =<< trim END
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200349 vim9script
350 def Func(...l: list<string>)
351 echo l
352 enddef
353 Func('a', 'b', 'c')
354 END
355 CheckScriptSuccess(lines)
356
357 lines =<< trim END
358 vim9script
359 def Func(...l: list<string>)
360 echo l
361 enddef
362 Func()
363 END
364 CheckScriptSuccess(lines)
365
366 lines =<< trim END
367 vim9script
Bram Moolenaar2f8cbc42020-09-16 17:22:59 +0200368 def Func(...l: any)
369 echo l
370 enddef
371 Func(0)
372 END
373 CheckScriptSuccess(lines)
374
375 lines =<< trim END
376 vim9script
Bram Moolenaar28022722020-09-21 22:02:49 +0200377 def Func(..._l: list<string>)
378 echo _l
379 enddef
380 Func('a', 'b', 'c')
381 END
382 CheckScriptSuccess(lines)
383
384 lines =<< trim END
385 vim9script
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200386 def Func(...l: list<string>)
387 echo l
388 enddef
389 Func(1, 2, 3)
390 END
Bram Moolenaar77072282020-09-16 17:55:40 +0200391 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200392
393 lines =<< trim END
394 vim9script
395 def Func(...l: list<string>)
396 echo l
397 enddef
398 Func('a', 9)
399 END
Bram Moolenaar77072282020-09-16 17:55:40 +0200400 CheckScriptFailure(lines, 'E1013: Argument 2: type mismatch')
Bram Moolenaar24aa48b2020-07-25 16:33:02 +0200401
402 lines =<< trim END
403 vim9script
404 def Func(...l: list<string>)
405 echo l
406 enddef
407 Func(1, 'a')
408 END
Bram Moolenaar77072282020-09-16 17:55:40 +0200409 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200410enddef
411
Bram Moolenaar575f24b2020-08-12 14:21:11 +0200412def Test_call_call()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200413 var l = [3, 2, 1]
Bram Moolenaar575f24b2020-08-12 14:21:11 +0200414 call('reverse', [l])
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200415 l->assert_equal([1, 2, 3])
Bram Moolenaar575f24b2020-08-12 14:21:11 +0200416enddef
417
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200418let s:value = ''
419
420def FuncOneDefArg(opt = 'text')
421 s:value = opt
422enddef
423
424def FuncTwoDefArg(nr = 123, opt = 'text'): string
425 return nr .. opt
426enddef
427
428def FuncVarargs(...arg: list<string>): string
429 return join(arg, ',')
430enddef
431
432def Test_func_type_varargs()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200433 var RefDefArg: func(?string)
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200434 RefDefArg = FuncOneDefArg
435 RefDefArg()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200436 s:value->assert_equal('text')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200437 RefDefArg('some')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200438 s:value->assert_equal('some')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200439
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200440 var RefDef2Arg: func(?number, ?string): string
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200441 RefDef2Arg = FuncTwoDefArg
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200442 RefDef2Arg()->assert_equal('123text')
443 RefDef2Arg(99)->assert_equal('99text')
444 RefDef2Arg(77, 'some')->assert_equal('77some')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200445
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200446 CheckDefFailure(['var RefWrong: func(string?)'], 'E1010:')
447 CheckDefFailure(['var RefWrong: func(?string, string)'], 'E1007:')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200448
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200449 var RefVarargs: func(...list<string>): string
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200450 RefVarargs = FuncVarargs
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200451 RefVarargs()->assert_equal('')
452 RefVarargs('one')->assert_equal('one')
453 RefVarargs('one', 'two')->assert_equal('one,two')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200454
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200455 CheckDefFailure(['var RefWrong: func(...list<string>, string)'], 'E110:')
456 CheckDefFailure(['var RefWrong: func(...list<string>, ?string)'], 'E110:')
Bram Moolenaar1378fbc2020-04-11 20:50:33 +0200457enddef
458
Bram Moolenaar0b76b422020-04-07 22:05:08 +0200459" Only varargs
460def MyVarargsOnly(...args: list<string>): string
461 return join(args, ',')
462enddef
463
464def Test_call_varargs_only()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200465 MyVarargsOnly()->assert_equal('')
466 MyVarargsOnly('one')->assert_equal('one')
467 MyVarargsOnly('one', 'two')->assert_equal('one,two')
Bram Moolenaar77072282020-09-16 17:55:40 +0200468 CheckDefFailure(['MyVarargsOnly(1)'], 'E1013: Argument 1: type mismatch, expected string but got number')
469 CheckDefFailure(['MyVarargsOnly("one", 2)'], 'E1013: Argument 2: type mismatch, expected string but got number')
Bram Moolenaar0b76b422020-04-07 22:05:08 +0200470enddef
471
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200472def Test_using_var_as_arg()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200473 writefile(['def Func(x: number)', 'var x = 234', 'enddef', 'defcompile'], 'Xdef')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200474 assert_fails('so Xdef', 'E1006:', '', 1, 'Func')
Bram Moolenaard2c61702020-09-06 15:58:36 +0200475 delete('Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200476enddef
477
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200478def DictArg(arg: dict<string>)
479 arg['key'] = 'value'
480enddef
481
482def ListArg(arg: list<string>)
483 arg[0] = 'value'
484enddef
485
486def Test_assign_to_argument()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200487 # works for dict and list
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200488 var d: dict<string> = {}
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200489 DictArg(d)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200490 d['key']->assert_equal('value')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200491 var l: list<string> = []
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200492 ListArg(l)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200493 l[0]->assert_equal('value')
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200494
Bram Moolenaard2c61702020-09-06 15:58:36 +0200495 CheckScriptFailure(['def Func(arg: number)', 'arg = 3', 'enddef', 'defcompile'], 'E1090:')
Bram Moolenaarcb2bdb12020-05-10 22:53:56 +0200496enddef
497
Bram Moolenaarb816dae2020-09-20 22:04:00 +0200498" These argument names are reserved in legacy functions.
499def WithReservedNames(firstline: string, lastline: string): string
500 return firstline .. lastline
501enddef
502
503def Test_argument_names()
504 assert_equal('OK', WithReservedNames('O', 'K'))
505enddef
506
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200507def Test_call_func_defined_later()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200508 g:DefinedLater('one')->assert_equal('one')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200509 assert_fails('NotDefined("one")', 'E117:', '', 2, 'Test_call_func_defined_later')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200510enddef
511
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200512func DefinedLater(arg)
513 return a:arg
514endfunc
515
516def Test_call_funcref()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200517 g:SomeFunc('abc')->assert_equal(3)
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200518 assert_fails('NotAFunc()', 'E117:', '', 2, 'Test_call_funcref') # comment after call
519 assert_fails('g:NotAFunc()', 'E117:', '', 3, 'Test_call_funcref')
Bram Moolenaar2f1980f2020-07-22 19:30:06 +0200520
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200521 var lines =<< trim END
Bram Moolenaar2f1980f2020-07-22 19:30:06 +0200522 vim9script
523 def RetNumber(): number
524 return 123
525 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200526 var Funcref: func: number = function('RetNumber')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200527 Funcref()->assert_equal(123)
Bram Moolenaar2f1980f2020-07-22 19:30:06 +0200528 END
529 CheckScriptSuccess(lines)
Bram Moolenaar0f60e802020-07-22 20:16:11 +0200530
531 lines =<< trim END
532 vim9script
533 def RetNumber(): number
534 return 123
535 enddef
536 def Bar(F: func: number): number
537 return F()
538 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200539 var Funcref = function('RetNumber')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200540 Bar(Funcref)->assert_equal(123)
Bram Moolenaar0f60e802020-07-22 20:16:11 +0200541 END
542 CheckScriptSuccess(lines)
Bram Moolenaarbfba8652020-07-23 20:09:10 +0200543
544 lines =<< trim END
545 vim9script
546 def UseNumber(nr: number)
547 echo nr
548 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200549 var Funcref: func(number) = function('UseNumber')
Bram Moolenaarbfba8652020-07-23 20:09:10 +0200550 Funcref(123)
551 END
552 CheckScriptSuccess(lines)
Bram Moolenaarb8070e32020-07-23 20:56:04 +0200553
554 lines =<< trim END
555 vim9script
556 def UseNumber(nr: number)
557 echo nr
558 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200559 var Funcref: func(string) = function('UseNumber')
Bram Moolenaarb8070e32020-07-23 20:56:04 +0200560 END
Bram Moolenaar5e654232020-09-16 15:22:00 +0200561 CheckScriptFailure(lines, 'E1012: Type mismatch; expected func(string) but got func(number)')
Bram Moolenaar4fc224c2020-07-26 17:56:25 +0200562
563 lines =<< trim END
564 vim9script
565 def EchoNr(nr = 34)
566 g:echo = nr
567 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200568 var Funcref: func(?number) = function('EchoNr')
Bram Moolenaar4fc224c2020-07-26 17:56:25 +0200569 Funcref()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200570 g:echo->assert_equal(34)
Bram Moolenaar4fc224c2020-07-26 17:56:25 +0200571 Funcref(123)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200572 g:echo->assert_equal(123)
Bram Moolenaar4fc224c2020-07-26 17:56:25 +0200573 END
574 CheckScriptSuccess(lines)
Bram Moolenaarace61322020-07-26 18:16:58 +0200575
576 lines =<< trim END
577 vim9script
578 def EchoList(...l: list<number>)
579 g:echo = l
580 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200581 var Funcref: func(...list<number>) = function('EchoList')
Bram Moolenaarace61322020-07-26 18:16:58 +0200582 Funcref()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200583 g:echo->assert_equal([])
Bram Moolenaarace61322020-07-26 18:16:58 +0200584 Funcref(1, 2, 3)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200585 g:echo->assert_equal([1, 2, 3])
Bram Moolenaarace61322020-07-26 18:16:58 +0200586 END
587 CheckScriptSuccess(lines)
Bram Moolenaar01865ad2020-07-26 18:33:09 +0200588
589 lines =<< trim END
590 vim9script
591 def OptAndVar(nr: number, opt = 12, ...l: list<number>): number
592 g:optarg = opt
593 g:listarg = l
594 return nr
595 enddef
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200596 var Funcref: func(number, ?number, ...list<number>): number = function('OptAndVar')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200597 Funcref(10)->assert_equal(10)
598 g:optarg->assert_equal(12)
599 g:listarg->assert_equal([])
Bram Moolenaar01865ad2020-07-26 18:33:09 +0200600
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200601 Funcref(11, 22)->assert_equal(11)
602 g:optarg->assert_equal(22)
603 g:listarg->assert_equal([])
Bram Moolenaar01865ad2020-07-26 18:33:09 +0200604
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200605 Funcref(17, 18, 1, 2, 3)->assert_equal(17)
606 g:optarg->assert_equal(18)
607 g:listarg->assert_equal([1, 2, 3])
Bram Moolenaar01865ad2020-07-26 18:33:09 +0200608 END
609 CheckScriptSuccess(lines)
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200610enddef
611
612let SomeFunc = function('len')
613let NotAFunc = 'text'
614
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200615def CombineFuncrefTypes()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200616 # same arguments, different return type
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200617 var Ref1: func(bool): string
618 var Ref2: func(bool): number
619 var Ref3: func(bool): any
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200620 Ref3 = g:cond ? Ref1 : Ref2
621
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200622 # different number of arguments
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200623 var Refa1: func(bool): number
624 var Refa2: func(bool, number): number
625 var Refa3: func: number
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200626 Refa3 = g:cond ? Refa1 : Refa2
627
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200628 # different argument types
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200629 var Refb1: func(bool, string): number
630 var Refb2: func(string, number): number
631 var Refb3: func(any, any): number
Bram Moolenaar99aaf0c2020-04-12 14:39:53 +0200632 Refb3 = g:cond ? Refb1 : Refb2
633enddef
634
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200635def FuncWithForwardCall()
Bram Moolenaar1df8b3f2020-04-23 18:13:23 +0200636 return g:DefinedEvenLater("yes")
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200637enddef
638
639def DefinedEvenLater(arg: string): string
640 return arg
641enddef
642
643def Test_error_in_nested_function()
Bram Moolenaarf5be8cd2020-07-17 20:36:00 +0200644 # Error in called function requires unwinding the call stack.
Bram Moolenaar44d66522020-09-06 22:26:57 +0200645 assert_fails('FuncWithForwardCall()', 'E1096:', '', 1, 'FuncWithForwardCall')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200646enddef
647
648def Test_return_type_wrong()
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200649 CheckScriptFailure([
650 'def Func(): number',
651 'return "a"',
652 'enddef',
653 'defcompile'], 'expected number but got string')
654 CheckScriptFailure([
655 'def Func(): string',
656 'return 1',
657 'enddef',
658 'defcompile'], 'expected string but got number')
659 CheckScriptFailure([
660 'def Func(): void',
661 'return "a"',
662 'enddef',
663 'defcompile'],
664 'E1096: Returning a value in a function without a return type')
665 CheckScriptFailure([
666 'def Func()',
667 'return "a"',
668 'enddef',
669 'defcompile'],
670 'E1096: Returning a value in a function without a return type')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200671
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200672 CheckScriptFailure([
673 'def Func(): number',
674 'return',
675 'enddef',
676 'defcompile'], 'E1003:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200677
678 CheckScriptFailure(['def Func(): list', 'return []', 'enddef'], 'E1008:')
679 CheckScriptFailure(['def Func(): dict', 'return {}', 'enddef'], 'E1008:')
Bram Moolenaaree4e0c12020-04-06 21:35:05 +0200680 CheckScriptFailure(['def Func()', 'return 1'], 'E1057:')
Bram Moolenaar5a849da2020-08-08 16:47:30 +0200681
682 CheckScriptFailure([
683 'vim9script',
684 'def FuncB()',
685 ' return 123',
686 'enddef',
687 'def FuncA()',
688 ' FuncB()',
689 'enddef',
690 'defcompile'], 'E1096:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200691enddef
692
693def Test_arg_type_wrong()
694 CheckScriptFailure(['def Func3(items: list)', 'echo "a"', 'enddef'], 'E1008: Missing <type>')
Bram Moolenaaree4e0c12020-04-06 21:35:05 +0200695 CheckScriptFailure(['def Func4(...)', 'echo "a"', 'enddef'], 'E1055: Missing name after ...')
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +0200696 CheckScriptFailure(['def Func5(items:string)', 'echo "a"'], 'E1069:')
Bram Moolenaar6e949782020-04-13 17:21:00 +0200697 CheckScriptFailure(['def Func5(items)', 'echo "a"'], 'E1077:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200698enddef
699
700def Test_vim9script_call()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200701 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200702 vim9script
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200703 var name = ''
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200704 def MyFunc(arg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200705 name = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200706 enddef
707 MyFunc('foobar')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200708 name->assert_equal('foobar')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200709
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200710 var str = 'barfoo'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200711 str->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200712 name->assert_equal('barfoo')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200713
Bram Moolenaar67979662020-06-20 22:50:47 +0200714 g:value = 'value'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200715 g:value->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200716 name->assert_equal('value')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200717
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200718 var listvar = []
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200719 def ListFunc(arg: list<number>)
720 listvar = arg
721 enddef
722 [1, 2, 3]->ListFunc()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200723 listvar->assert_equal([1, 2, 3])
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200724
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200725 var dictvar = {}
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200726 def DictFunc(arg: dict<number>)
727 dictvar = arg
728 enddef
729 {'a': 1, 'b': 2}->DictFunc()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200730 dictvar->assert_equal(#{a: 1, b: 2})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200731 def CompiledDict()
732 {'a': 3, 'b': 4}->DictFunc()
733 enddef
734 CompiledDict()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200735 dictvar->assert_equal(#{a: 3, b: 4})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200736
737 #{a: 3, b: 4}->DictFunc()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200738 dictvar->assert_equal(#{a: 3, b: 4})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200739
740 ('text')->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200741 name->assert_equal('text')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200742 ("some")->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200743 name->assert_equal('some')
Bram Moolenaare6b53242020-07-01 17:28:33 +0200744
Bram Moolenaar13e12b82020-07-24 18:47:22 +0200745 # line starting with single quote is not a mark
Bram Moolenaar10409562020-07-29 20:00:38 +0200746 # line starting with double quote can be a method call
Bram Moolenaar3d48e252020-07-15 14:15:52 +0200747 'asdfasdf'->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200748 name->assert_equal('asdfasdf')
Bram Moolenaar10409562020-07-29 20:00:38 +0200749 "xyz"->MyFunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200750 name->assert_equal('xyz')
Bram Moolenaar3d48e252020-07-15 14:15:52 +0200751
752 def UseString()
753 'xyork'->MyFunc()
754 enddef
755 UseString()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200756 name->assert_equal('xyork')
Bram Moolenaar3d48e252020-07-15 14:15:52 +0200757
Bram Moolenaar10409562020-07-29 20:00:38 +0200758 def UseString2()
759 "knife"->MyFunc()
760 enddef
761 UseString2()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200762 name->assert_equal('knife')
Bram Moolenaar10409562020-07-29 20:00:38 +0200763
Bram Moolenaar13e12b82020-07-24 18:47:22 +0200764 # prepending a colon makes it a mark
765 new
766 setline(1, ['aaa', 'bbb', 'ccc'])
767 normal! 3Gmt1G
768 :'t
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200769 getcurpos()[1]->assert_equal(3)
Bram Moolenaar13e12b82020-07-24 18:47:22 +0200770 bwipe!
771
Bram Moolenaare6b53242020-07-01 17:28:33 +0200772 MyFunc(
773 'continued'
774 )
775 assert_equal('continued',
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200776 name
Bram Moolenaare6b53242020-07-01 17:28:33 +0200777 )
778
779 call MyFunc(
780 'more'
781 ..
782 'lines'
783 )
784 assert_equal(
785 'morelines',
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200786 name)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200787 END
788 writefile(lines, 'Xcall.vim')
789 source Xcall.vim
790 delete('Xcall.vim')
791enddef
792
793def Test_vim9script_call_fail_decl()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200794 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200795 vim9script
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200796 var name = ''
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200797 def MyFunc(arg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200798 var name = 123
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200799 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +0200800 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200801 END
Bram Moolenaar6c4bfe42020-07-23 18:26:30 +0200802 CheckScriptFailure(lines, 'E1054:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200803enddef
804
Bram Moolenaar65b95452020-07-19 14:03:09 +0200805def Test_vim9script_call_fail_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200806 var lines =<< trim END
Bram Moolenaar65b95452020-07-19 14:03:09 +0200807 vim9script
808 def MyFunc(arg: string)
809 echo arg
810 enddef
811 MyFunc(1234)
812 END
Bram Moolenaar77072282020-09-16 17:55:40 +0200813 CheckScriptFailure(lines, 'E1013: Argument 1: type mismatch, expected string but got number')
Bram Moolenaar65b95452020-07-19 14:03:09 +0200814enddef
815
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200816def Test_vim9script_call_fail_const()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200817 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200818 vim9script
819 const var = ''
820 def MyFunc(arg: string)
821 var = 'asdf'
822 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +0200823 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200824 END
825 writefile(lines, 'Xcall_const.vim')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200826 assert_fails('source Xcall_const.vim', 'E46:', '', 1, 'MyFunc')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200827 delete('Xcall_const.vim')
828enddef
829
830" Test that inside :function a Python function can be defined, :def is not
831" recognized.
832func Test_function_python()
833 CheckFeature python3
Bram Moolenaar727345e2020-09-27 23:33:59 +0200834 let py = 'python3'
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200835 execute py "<< EOF"
836def do_something():
837 return 1
838EOF
839endfunc
840
841def Test_delfunc()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200842 var lines =<< trim END
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200843 vim9script
Bram Moolenaar4c17ad92020-04-27 22:47:51 +0200844 def g:GoneSoon()
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200845 echo 'hello'
846 enddef
847
848 def CallGoneSoon()
849 GoneSoon()
850 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +0200851 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200852
Bram Moolenaar4c17ad92020-04-27 22:47:51 +0200853 delfunc g:GoneSoon
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200854 CallGoneSoon()
855 END
856 writefile(lines, 'XToDelFunc')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200857 assert_fails('so XToDelFunc', 'E933:', '', 1, 'CallGoneSoon')
858 assert_fails('so XToDelFunc', 'E933:', '', 1, 'CallGoneSoon')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200859
860 delete('XToDelFunc')
861enddef
862
863def Test_redef_failure()
Bram Moolenaard2c61702020-09-06 15:58:36 +0200864 writefile(['def Func0(): string', 'return "Func0"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200865 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +0200866 writefile(['def Func1(): string', 'return "Func1"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200867 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +0200868 writefile(['def! Func0(): string', 'enddef', 'defcompile'], 'Xdef')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200869 assert_fails('so Xdef', 'E1027:', '', 1, 'Func0')
Bram Moolenaard2c61702020-09-06 15:58:36 +0200870 writefile(['def Func2(): string', 'return "Func2"', 'enddef'], 'Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200871 so Xdef
Bram Moolenaard2c61702020-09-06 15:58:36 +0200872 delete('Xdef')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200873
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200874 g:Func0()->assert_equal(0)
875 g:Func1()->assert_equal('Func1')
876 g:Func2()->assert_equal('Func2')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200877
878 delfunc! Func0
879 delfunc! Func1
880 delfunc! Func2
881enddef
882
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +0200883def Test_vim9script_func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200884 var lines =<< trim END
Bram Moolenaarf93c7fe2020-04-23 22:16:53 +0200885 vim9script
886 func Func(arg)
887 echo a:arg
888 endfunc
889 Func('text')
890 END
891 writefile(lines, 'XVim9Func')
892 so XVim9Func
893
894 delete('XVim9Func')
895enddef
896
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200897" Test for internal functions returning different types
898func Test_InternalFuncRetType()
899 let lines =<< trim END
900 def RetFloat(): float
901 return ceil(1.456)
902 enddef
903
904 def RetListAny(): list<any>
Bram Moolenaar17a836c2020-08-12 17:35:58 +0200905 return items({'k': 'v'})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200906 enddef
907
908 def RetListString(): list<string>
909 return split('a:b:c', ':')
910 enddef
911
912 def RetListDictAny(): list<dict<any>>
913 return getbufinfo()
914 enddef
915
916 def RetDictNumber(): dict<number>
917 return wordcount()
918 enddef
919
920 def RetDictString(): dict<string>
921 return environ()
922 enddef
923 END
924 call writefile(lines, 'Xscript')
925 source Xscript
926
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200927 call RetFloat()->assert_equal(2.0)
928 call RetListAny()->assert_equal([['k', 'v']])
929 call RetListString()->assert_equal(['a', 'b', 'c'])
930 call RetListDictAny()->assert_notequal([])
931 call RetDictNumber()->assert_notequal({})
932 call RetDictString()->assert_notequal({})
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200933 call delete('Xscript')
934endfunc
935
936" Test for passing too many or too few arguments to internal functions
937func Test_internalfunc_arg_error()
938 let l =<< trim END
939 def! FArgErr(): float
940 return ceil(1.1, 2)
941 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +0200942 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200943 END
944 call writefile(l, 'Xinvalidarg')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200945 call assert_fails('so Xinvalidarg', 'E118:', '', 1, 'FArgErr')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200946 let l =<< trim END
947 def! FArgErr(): float
948 return ceil()
949 enddef
Bram Moolenaar822ba242020-05-24 23:00:18 +0200950 defcompile
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200951 END
952 call writefile(l, 'Xinvalidarg')
Bram Moolenaar9bd5d872020-09-06 21:47:48 +0200953 call assert_fails('so Xinvalidarg', 'E119:', '', 1, 'FArgErr')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200954 call delete('Xinvalidarg')
955endfunc
956
957let s:funcResult = 0
958
959def FuncNoArgNoRet()
Bram Moolenaar53900992020-08-22 19:02:02 +0200960 s:funcResult = 11
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200961enddef
962
963def FuncNoArgRetNumber(): number
Bram Moolenaar53900992020-08-22 19:02:02 +0200964 s:funcResult = 22
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200965 return 1234
966enddef
967
Bram Moolenaarec5929d2020-04-07 20:53:39 +0200968def FuncNoArgRetString(): string
Bram Moolenaar53900992020-08-22 19:02:02 +0200969 s:funcResult = 45
Bram Moolenaarec5929d2020-04-07 20:53:39 +0200970 return 'text'
971enddef
972
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200973def FuncOneArgNoRet(arg: number)
Bram Moolenaar53900992020-08-22 19:02:02 +0200974 s:funcResult = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200975enddef
976
977def FuncOneArgRetNumber(arg: number): number
Bram Moolenaar53900992020-08-22 19:02:02 +0200978 s:funcResult = arg
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200979 return arg
980enddef
981
Bram Moolenaar08938ee2020-04-11 23:17:17 +0200982def FuncTwoArgNoRet(one: bool, two: number)
Bram Moolenaar53900992020-08-22 19:02:02 +0200983 s:funcResult = two
Bram Moolenaar08938ee2020-04-11 23:17:17 +0200984enddef
985
Bram Moolenaarec5929d2020-04-07 20:53:39 +0200986def FuncOneArgRetString(arg: string): string
987 return arg
988enddef
989
Bram Moolenaar89228602020-04-05 22:14:54 +0200990def FuncOneArgRetAny(arg: any): any
991 return arg
992enddef
993
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200994def Test_func_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +0200995 var Ref1: func()
Bram Moolenaar53900992020-08-22 19:02:02 +0200996 s:funcResult = 0
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +0200997 Ref1 = FuncNoArgNoRet
998 Ref1()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +0200999 s:funcResult->assert_equal(11)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001000
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001001 var Ref2: func
Bram Moolenaar53900992020-08-22 19:02:02 +02001002 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001003 Ref2 = FuncNoArgNoRet
1004 Ref2()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001005 s:funcResult->assert_equal(11)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001006
Bram Moolenaar53900992020-08-22 19:02:02 +02001007 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001008 Ref2 = FuncOneArgNoRet
1009 Ref2(12)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001010 s:funcResult->assert_equal(12)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001011
Bram Moolenaar53900992020-08-22 19:02:02 +02001012 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001013 Ref2 = FuncNoArgRetNumber
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001014 Ref2()->assert_equal(1234)
1015 s:funcResult->assert_equal(22)
Bram Moolenaar4c683752020-04-05 21:38:23 +02001016
Bram Moolenaar53900992020-08-22 19:02:02 +02001017 s:funcResult = 0
Bram Moolenaar4c683752020-04-05 21:38:23 +02001018 Ref2 = FuncOneArgRetNumber
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001019 Ref2(13)->assert_equal(13)
1020 s:funcResult->assert_equal(13)
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001021enddef
1022
Bram Moolenaar9978d472020-07-05 16:01:56 +02001023def Test_repeat_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001024 var res = 0
Bram Moolenaar9978d472020-07-05 16:01:56 +02001025 for n in repeat([1], 3)
1026 res += n
1027 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001028 res->assert_equal(3)
Bram Moolenaarfce82b32020-07-05 16:07:21 +02001029
1030 res = 0
1031 for n in add([1, 2], 3)
1032 res += n
1033 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001034 res->assert_equal(6)
Bram Moolenaar9978d472020-07-05 16:01:56 +02001035enddef
1036
Bram Moolenaar846178a2020-07-05 17:04:13 +02001037def Test_argv_return_type()
1038 next fileone filetwo
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001039 var res = ''
Bram Moolenaar846178a2020-07-05 17:04:13 +02001040 for name in argv()
1041 res ..= name
1042 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001043 res->assert_equal('fileonefiletwo')
Bram Moolenaar846178a2020-07-05 17:04:13 +02001044enddef
1045
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001046def Test_func_type_part()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001047 var RefVoid: func: void
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001048 RefVoid = FuncNoArgNoRet
1049 RefVoid = FuncOneArgNoRet
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001050 CheckDefFailure(['var RefVoid: func: void', 'RefVoid = FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func(...) but got func(): number')
1051 CheckDefFailure(['var RefVoid: func: void', 'RefVoid = FuncNoArgRetString'], 'E1012: Type mismatch; expected func(...) but got func(): string')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001052
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001053 var RefAny: func(): any
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001054 RefAny = FuncNoArgRetNumber
1055 RefAny = FuncNoArgRetString
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001056 CheckDefFailure(['var RefAny: func(): any', 'RefAny = FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(): any but got func()')
1057 CheckDefFailure(['var RefAny: func(): any', 'RefAny = FuncOneArgNoRet'], 'E1012: Type mismatch; expected func(): any but got func(number)')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001058
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001059 var RefNr: func: number
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001060 RefNr = FuncNoArgRetNumber
1061 RefNr = FuncOneArgRetNumber
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001062 CheckDefFailure(['var RefNr: func: number', 'RefNr = FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(...): number but got func()')
1063 CheckDefFailure(['var RefNr: func: number', 'RefNr = FuncNoArgRetString'], 'E1012: Type mismatch; expected func(...): number but got func(): string')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001064
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001065 var RefStr: func: string
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001066 RefStr = FuncNoArgRetString
1067 RefStr = FuncOneArgRetString
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001068 CheckDefFailure(['var RefStr: func: string', 'RefStr = FuncNoArgNoRet'], 'E1012: Type mismatch; expected func(...): string but got func()')
1069 CheckDefFailure(['var RefStr: func: string', 'RefStr = FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func(...): string but got func(): number')
Bram Moolenaarec5929d2020-04-07 20:53:39 +02001070enddef
1071
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001072def Test_func_type_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001073 CheckDefFailure(['var ref1: func()'], 'E704:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001074
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001075 CheckDefFailure(['var Ref1: func()', 'Ref1 = FuncNoArgRetNumber'], 'E1012: Type mismatch; expected func() but got func(): number')
1076 CheckDefFailure(['var Ref1: func()', 'Ref1 = FuncOneArgNoRet'], 'E1012: Type mismatch; expected func() but got func(number)')
1077 CheckDefFailure(['var Ref1: func()', 'Ref1 = FuncOneArgRetNumber'], 'E1012: Type mismatch; expected func() but got func(number): number')
1078 CheckDefFailure(['var Ref1: func(bool)', 'Ref1 = FuncTwoArgNoRet'], 'E1012: Type mismatch; expected func(bool) but got func(bool, number)')
1079 CheckDefFailure(['var Ref1: func(?bool)', 'Ref1 = FuncTwoArgNoRet'], 'E1012: Type mismatch; expected func(?bool) but got func(bool, number)')
1080 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 +02001081
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001082 CheckDefFailure(['var RefWrong: func(string ,number)'], 'E1068:')
1083 CheckDefFailure(['var RefWrong: func(string,number)'], 'E1069:')
1084 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:')
1085 CheckDefFailure(['var RefWrong: func(bool):string'], 'E1069:')
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001086enddef
1087
Bram Moolenaar89228602020-04-05 22:14:54 +02001088def Test_func_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001089 var nr: number
Bram Moolenaar89228602020-04-05 22:14:54 +02001090 nr = FuncNoArgRetNumber()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001091 nr->assert_equal(1234)
Bram Moolenaar89228602020-04-05 22:14:54 +02001092
1093 nr = FuncOneArgRetAny(122)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001094 nr->assert_equal(122)
Bram Moolenaar89228602020-04-05 22:14:54 +02001095
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001096 var str: string
Bram Moolenaar89228602020-04-05 22:14:54 +02001097 str = FuncOneArgRetAny('yes')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001098 str->assert_equal('yes')
Bram Moolenaar89228602020-04-05 22:14:54 +02001099
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001100 CheckDefFailure(['var str: string', 'str = FuncNoArgRetNumber()'], 'E1012: Type mismatch; expected string but got number')
Bram Moolenaar89228602020-04-05 22:14:54 +02001101enddef
1102
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001103def MultiLine(
1104 arg1: string,
1105 arg2 = 1234,
1106 ...rest: list<string>
1107 ): string
1108 return arg1 .. arg2 .. join(rest, '-')
1109enddef
1110
Bram Moolenaar2c330432020-04-13 14:41:35 +02001111def MultiLineComment(
1112 arg1: string, # comment
1113 arg2 = 1234, # comment
1114 ...rest: list<string> # comment
1115 ): string # comment
1116 return arg1 .. arg2 .. join(rest, '-')
1117enddef
1118
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001119def Test_multiline()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001120 MultiLine('text')->assert_equal('text1234')
1121 MultiLine('text', 777)->assert_equal('text777')
1122 MultiLine('text', 777, 'one')->assert_equal('text777one')
1123 MultiLine('text', 777, 'one', 'two')->assert_equal('text777one-two')
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001124enddef
1125
Bram Moolenaar23e03252020-04-12 22:22:31 +02001126func Test_multiline_not_vim9()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001127 call MultiLine('text')->assert_equal('text1234')
1128 call MultiLine('text', 777)->assert_equal('text777')
1129 call MultiLine('text', 777, 'one')->assert_equal('text777one')
1130 call MultiLine('text', 777, 'one', 'two')->assert_equal('text777one-two')
Bram Moolenaar23e03252020-04-12 22:22:31 +02001131endfunc
1132
Bram Moolenaar5e774c72020-04-12 21:53:00 +02001133
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02001134" When using CheckScriptFailure() for the below test, E1010 is generated instead
1135" of E1056.
1136func Test_E1056_1059()
1137 let caught_1056 = 0
1138 try
1139 def F():
1140 return 1
1141 enddef
1142 catch /E1056:/
1143 let caught_1056 = 1
1144 endtry
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001145 eval caught_1056->assert_equal(1)
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02001146
1147 let caught_1059 = 0
1148 try
1149 def F5(items : list)
1150 echo 'a'
1151 enddef
1152 catch /E1059:/
1153 let caught_1059 = 1
1154 endtry
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001155 eval caught_1059->assert_equal(1)
Bram Moolenaaree4e0c12020-04-06 21:35:05 +02001156endfunc
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001157
Bram Moolenaar015f4262020-05-05 21:25:22 +02001158func DelMe()
1159 echo 'DelMe'
1160endfunc
1161
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001162def Test_error_reporting()
1163 # comment lines at the start of the function
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001164 var lines =<< trim END
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001165 " comment
1166 def Func()
1167 # comment
1168 # comment
1169 invalid
1170 enddef
1171 defcompile
1172 END
Bram Moolenaar08052222020-09-14 17:04:31 +02001173 writefile(lines, 'Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001174 try
1175 source Xdef
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001176 assert_report('should have failed')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001177 catch /E476:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001178 v:exception->assert_match('Invalid command: invalid')
1179 v:throwpoint->assert_match(', line 3$')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001180 endtry
1181
1182 # comment lines after the start of the function
1183 lines =<< trim END
1184 " comment
1185 def Func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001186 var x = 1234
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001187 # comment
1188 # comment
1189 invalid
1190 enddef
1191 defcompile
1192 END
Bram Moolenaar08052222020-09-14 17:04:31 +02001193 writefile(lines, 'Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001194 try
1195 source Xdef
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001196 assert_report('should have failed')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001197 catch /E476:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001198 v:exception->assert_match('Invalid command: invalid')
1199 v:throwpoint->assert_match(', line 4$')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001200 endtry
1201
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001202 lines =<< trim END
1203 vim9script
1204 def Func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001205 var db = #{foo: 1, bar: 2}
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001206 # comment
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001207 var x = db.asdf
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001208 enddef
1209 defcompile
1210 Func()
1211 END
Bram Moolenaar08052222020-09-14 17:04:31 +02001212 writefile(lines, 'Xdef')
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001213 try
1214 source Xdef
1215 assert_report('should have failed')
1216 catch /E716:/
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001217 v:throwpoint->assert_match('_Func, line 3$')
Bram Moolenaar7517ffd2020-08-14 18:35:07 +02001218 endtry
1219
Bram Moolenaar08052222020-09-14 17:04:31 +02001220 delete('Xdef')
Bram Moolenaarbf8feb52020-08-08 14:26:31 +02001221enddef
1222
Bram Moolenaar015f4262020-05-05 21:25:22 +02001223def Test_deleted_function()
1224 CheckDefExecFailure([
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001225 'var RefMe: func = function("g:DelMe")',
Bram Moolenaar015f4262020-05-05 21:25:22 +02001226 'delfunc g:DelMe',
1227 'echo RefMe()'], 'E117:')
1228enddef
1229
1230def Test_unknown_function()
1231 CheckDefExecFailure([
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001232 'var Ref: func = function("NotExist")',
Bram Moolenaar9b7bf9e2020-07-11 22:14:59 +02001233 'delfunc g:NotExist'], 'E700:')
Bram Moolenaar015f4262020-05-05 21:25:22 +02001234enddef
1235
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +02001236def RefFunc(Ref: func(string): string): string
1237 return Ref('more')
1238enddef
1239
1240def Test_closure_simple()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001241 var local = 'some '
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001242 RefFunc({s -> local .. s})->assert_equal('some more')
Bram Moolenaarc8cd2b32020-05-01 19:29:08 +02001243enddef
1244
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001245def MakeRef()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001246 var local = 'some '
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001247 g:Ref = {s -> local .. s}
1248enddef
1249
1250def Test_closure_ref_after_return()
1251 MakeRef()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001252 g:Ref('thing')->assert_equal('some thing')
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001253 unlet g:Ref
1254enddef
1255
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001256def MakeTwoRefs()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001257 var local = ['some']
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001258 g:Extend = {s -> local->add(s)}
1259 g:Read = {-> local}
1260enddef
1261
1262def Test_closure_two_refs()
1263 MakeTwoRefs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001264 join(g:Read(), ' ')->assert_equal('some')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001265 g:Extend('more')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001266 join(g:Read(), ' ')->assert_equal('some more')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001267 g:Extend('even')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001268 join(g:Read(), ' ')->assert_equal('some more even')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001269
1270 unlet g:Extend
1271 unlet g:Read
1272enddef
1273
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001274def ReadRef(Ref: func(): list<string>): string
1275 return join(Ref(), ' ')
1276enddef
1277
Bram Moolenaar5e654232020-09-16 15:22:00 +02001278def ExtendRef(Ref: func(string): list<string>, add: string)
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001279 Ref(add)
1280enddef
1281
1282def Test_closure_two_indirect_refs()
Bram Moolenaarf7779c62020-05-03 15:38:16 +02001283 MakeTwoRefs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001284 ReadRef(g:Read)->assert_equal('some')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001285 ExtendRef(g:Extend, 'more')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001286 ReadRef(g:Read)->assert_equal('some more')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001287 ExtendRef(g:Extend, 'even')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001288 ReadRef(g:Read)->assert_equal('some more even')
Bram Moolenaar5adc55c2020-05-02 23:12:58 +02001289
1290 unlet g:Extend
1291 unlet g:Read
1292enddef
Bram Moolenaarbf67ea12020-05-02 17:52:42 +02001293
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001294def MakeArgRefs(theArg: string)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001295 var local = 'loc_val'
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001296 g:UseArg = {s -> theArg .. '/' .. local .. '/' .. s}
1297enddef
1298
1299def MakeArgRefsVarargs(theArg: string, ...rest: list<string>)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001300 var local = 'the_loc'
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001301 g:UseVararg = {s -> theArg .. '/' .. local .. '/' .. s .. '/' .. join(rest)}
1302enddef
1303
1304def Test_closure_using_argument()
1305 MakeArgRefs('arg_val')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001306 g:UseArg('call_val')->assert_equal('arg_val/loc_val/call_val')
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001307
1308 MakeArgRefsVarargs('arg_val', 'one', 'two')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001309 g:UseVararg('call_val')->assert_equal('arg_val/the_loc/call_val/one two')
Bram Moolenaar2fd4cd72020-05-03 22:30:49 +02001310
1311 unlet g:UseArg
1312 unlet g:UseVararg
1313enddef
1314
Bram Moolenaarb68b3462020-05-06 21:06:30 +02001315def MakeGetAndAppendRefs()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001316 var local = 'a'
Bram Moolenaarb68b3462020-05-06 21:06:30 +02001317
1318 def Append(arg: string)
1319 local ..= arg
1320 enddef
1321 g:Append = Append
1322
1323 def Get(): string
1324 return local
1325 enddef
1326 g:Get = Get
1327enddef
1328
1329def Test_closure_append_get()
1330 MakeGetAndAppendRefs()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001331 g:Get()->assert_equal('a')
Bram Moolenaarb68b3462020-05-06 21:06:30 +02001332 g:Append('-b')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001333 g:Get()->assert_equal('a-b')
Bram Moolenaarb68b3462020-05-06 21:06:30 +02001334 g:Append('-c')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001335 g:Get()->assert_equal('a-b-c')
Bram Moolenaarb68b3462020-05-06 21:06:30 +02001336
1337 unlet g:Append
1338 unlet g:Get
1339enddef
1340
Bram Moolenaar04b12692020-05-04 23:24:44 +02001341def Test_nested_closure()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001342 var local = 'text'
Bram Moolenaar04b12692020-05-04 23:24:44 +02001343 def Closure(arg: string): string
1344 return local .. arg
1345 enddef
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001346 Closure('!!!')->assert_equal('text!!!')
Bram Moolenaar04b12692020-05-04 23:24:44 +02001347enddef
1348
Bram Moolenaar6f5b6df2020-05-16 21:20:12 +02001349func GetResult(Ref)
1350 return a:Ref('some')
1351endfunc
1352
1353def Test_call_closure_not_compiled()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001354 var text = 'text'
Bram Moolenaar6f5b6df2020-05-16 21:20:12 +02001355 g:Ref = {s -> s .. text}
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001356 GetResult(g:Ref)->assert_equal('sometext')
Bram Moolenaar6f5b6df2020-05-16 21:20:12 +02001357enddef
1358
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001359def Test_double_closure_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001360 var lines =<< trim END
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001361 vim9script
1362 def Func()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001363 var name = 0
1364 for i in range(2)
1365 timer_start(0, {-> name})
1366 endfor
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001367 enddef
1368 Func()
1369 END
Bram Moolenaar148ce7a2020-09-23 21:57:23 +02001370 CheckScriptSuccess(lines)
Bram Moolenaar7cbfaa52020-09-18 21:25:32 +02001371enddef
1372
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02001373def Test_nested_closure_fails()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001374 var lines =<< trim END
Bram Moolenaarc70bdab2020-09-26 19:59:38 +02001375 vim9script
1376 def FuncA()
1377 FuncB(0)
1378 enddef
1379 def FuncB(n: number): list<string>
1380 return map([0], {_, v -> n})
1381 enddef
1382 FuncA()
1383 END
1384 CheckScriptFailure(lines, 'E1012:')
1385enddef
1386
Bram Moolenaar5366e1a2020-10-01 13:01:34 +02001387def Test_nested_lambda()
1388 var lines =<< trim END
1389 vim9script
1390 def Func()
1391 var x = 4
1392 var Lambda1 = {-> 7}
1393 var Lambda2 = {-> [Lambda1(), x]}
1394 var res = Lambda2()
1395 assert_equal([7, 4], res)
1396 enddef
1397 Func()
1398 END
1399 CheckScriptSuccess(lines)
1400enddef
1401
Bram Moolenaar865af6b2020-06-18 18:45:49 +02001402def Test_sort_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001403 var res: list<number>
Bram Moolenaar865af6b2020-06-18 18:45:49 +02001404 res = [1, 2, 3]->sort()
1405enddef
1406
Bram Moolenaar08e51f42020-09-16 23:23:36 +02001407def Test_sort_argument()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001408 var res = ['b', 'a', 'c']->sort('i')
Bram Moolenaar08e51f42020-09-16 23:23:36 +02001409 res->assert_equal(['a', 'b', 'c'])
1410enddef
1411
Bram Moolenaarf151ad12020-06-30 13:38:01 +02001412def Test_getqflist_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001413 var l = getqflist()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001414 l->assert_equal([])
Bram Moolenaarf151ad12020-06-30 13:38:01 +02001415
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001416 var d = getqflist(#{items: 0})
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001417 d->assert_equal(#{items: []})
Bram Moolenaarf151ad12020-06-30 13:38:01 +02001418enddef
1419
1420def Test_getloclist_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001421 var l = getloclist(1)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001422 l->assert_equal([])
Bram Moolenaarf151ad12020-06-30 13:38:01 +02001423
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001424 var d = getloclist(1, #{items: 0})
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001425 d->assert_equal(#{items: []})
Bram Moolenaarf151ad12020-06-30 13:38:01 +02001426enddef
1427
Bram Moolenaara66ba012020-07-05 18:41:08 +02001428def Test_copy_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001429 var l = copy([1, 2, 3])
1430 var res = 0
Bram Moolenaara66ba012020-07-05 18:41:08 +02001431 for n in l
1432 res += n
1433 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001434 res->assert_equal(6)
Bram Moolenaara66ba012020-07-05 18:41:08 +02001435
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001436 var dl = deepcopy([1, 2, 3])
Bram Moolenaara66ba012020-07-05 18:41:08 +02001437 res = 0
1438 for n in dl
1439 res += n
1440 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001441 res->assert_equal(6)
Bram Moolenaar44b4a242020-09-05 17:18:28 +02001442
1443 dl = deepcopy([1, 2, 3], true)
Bram Moolenaara66ba012020-07-05 18:41:08 +02001444enddef
1445
Bram Moolenaarb3c019c2020-07-05 20:08:39 +02001446def Test_extend_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001447 var l = extend([1, 2], [3])
1448 var res = 0
Bram Moolenaarb3c019c2020-07-05 20:08:39 +02001449 for n in l
1450 res += n
1451 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001452 res->assert_equal(6)
Bram Moolenaarb3c019c2020-07-05 20:08:39 +02001453enddef
1454
Bram Moolenaar2df47312020-09-05 17:30:44 +02001455def Test_garbagecollect()
1456 garbagecollect(true)
1457enddef
1458
Bram Moolenaar252e88a2020-07-05 20:47:18 +02001459def Test_insert_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001460 var l = insert([2, 1], 3)
1461 var res = 0
Bram Moolenaar252e88a2020-07-05 20:47:18 +02001462 for n in l
1463 res += n
1464 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001465 res->assert_equal(6)
Bram Moolenaar252e88a2020-07-05 20:47:18 +02001466enddef
1467
Bram Moolenaar32f335f2020-08-14 18:56:45 +02001468def Test_keys_return_type()
1469 const var: list<string> = #{a: 1, b: 2}->keys()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001470 var->assert_equal(['a', 'b'])
Bram Moolenaar32f335f2020-08-14 18:56:45 +02001471enddef
1472
Bram Moolenaar67627352020-07-05 21:10:24 +02001473def Test_reverse_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001474 var l = reverse([1, 2, 3])
1475 var res = 0
Bram Moolenaar67627352020-07-05 21:10:24 +02001476 for n in l
1477 res += n
1478 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001479 res->assert_equal(6)
Bram Moolenaar67627352020-07-05 21:10:24 +02001480enddef
1481
Bram Moolenaarad7c2492020-07-05 20:55:29 +02001482def Test_remove_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001483 var l = remove(#{one: [1, 2], two: [3, 4]}, 'one')
1484 var res = 0
Bram Moolenaarad7c2492020-07-05 20:55:29 +02001485 for n in l
1486 res += n
1487 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001488 res->assert_equal(3)
Bram Moolenaarad7c2492020-07-05 20:55:29 +02001489enddef
1490
Bram Moolenaar0d94ad62020-07-05 20:16:41 +02001491def Test_filter_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001492 var l = filter([1, 2, 3], {-> 1})
1493 var res = 0
Bram Moolenaar0d94ad62020-07-05 20:16:41 +02001494 for n in l
1495 res += n
1496 endfor
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001497 res->assert_equal(6)
Bram Moolenaar0d94ad62020-07-05 20:16:41 +02001498enddef
1499
Bram Moolenaarf39397e2020-08-17 22:21:36 +02001500def Test_bufnr()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001501 var buf = bufnr()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001502 bufnr('%')->assert_equal(buf)
Bram Moolenaarfe136c92020-09-04 18:35:26 +02001503
1504 buf = bufnr('Xdummy', true)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001505 buf->assert_notequal(-1)
Bram Moolenaarfe136c92020-09-04 18:35:26 +02001506 exe 'bwipe! ' .. buf
Bram Moolenaarf39397e2020-08-17 22:21:36 +02001507enddef
1508
Bram Moolenaarec65d772020-08-20 22:29:12 +02001509def Test_col()
1510 new
1511 setline(1, 'asdf')
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001512 col([1, '$'])->assert_equal(5)
Bram Moolenaarec65d772020-08-20 22:29:12 +02001513enddef
1514
Bram Moolenaar24f77502020-09-04 19:50:57 +02001515def Test_char2nr()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001516 char2nr('あ', true)->assert_equal(12354)
Bram Moolenaar24f77502020-09-04 19:50:57 +02001517enddef
1518
Bram Moolenaar3d945cc2020-08-06 21:26:59 +02001519def Test_getreg_return_type()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001520 var s1: string = getreg('"')
1521 var s2: string = getreg('"', 1)
1522 var s3: list<string> = getreg('"', 1, 1)
Bram Moolenaar3d945cc2020-08-06 21:26:59 +02001523enddef
1524
Bram Moolenaarf1a23682020-07-13 18:55:48 +02001525def Wrong_dict_key_type(items: list<number>): list<number>
1526 return filter(items, {_, val -> get({val: 1}, 'x')})
1527enddef
1528
1529def Test_wrong_dict_key_type()
Bram Moolenaar5e654232020-09-16 15:22:00 +02001530 assert_fails('Wrong_dict_key_type([1, 2, 3])', 'E1012:')
Bram Moolenaarf1a23682020-07-13 18:55:48 +02001531enddef
1532
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02001533def Line_continuation_in_def(dir: string = ''): string
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001534 var path: string = empty(dir)
1535 \ ? 'empty'
1536 \ : 'full'
1537 return path
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02001538enddef
1539
1540def Test_line_continuation_in_def()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001541 Line_continuation_in_def('.')->assert_equal('full')
Bram Moolenaaracd4c5e2020-06-22 19:39:03 +02001542enddef
1543
Bram Moolenaar5e654232020-09-16 15:22:00 +02001544def Line_continuation_in_lambda(): list<string>
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001545 var x = range(97, 100)
Bram Moolenaar914e7ea2020-07-11 15:20:48 +02001546 ->map({_, v -> nr2char(v)
Bram Moolenaar7a4b8982020-07-08 17:36:21 +02001547 ->toupper()})
1548 ->reverse()
1549 return x
1550enddef
1551
1552def Test_line_continuation_in_lambda()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001553 Line_continuation_in_lambda()->assert_equal(['D', 'C', 'B', 'A'])
Bram Moolenaar7a4b8982020-07-08 17:36:21 +02001554enddef
1555
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001556func Test_silent_echo()
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001557 CheckScreendump
1558
1559 let lines =<< trim END
1560 vim9script
1561 def EchoNothing()
1562 silent echo ''
1563 enddef
1564 defcompile
1565 END
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001566 call writefile(lines, 'XTest_silent_echo')
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001567
1568 " Check that the balloon shows up after a mouse move
1569 let buf = RunVimInTerminal('-S XTest_silent_echo', {'rows': 6})
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001570 call term_sendkeys(buf, ":abc")
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001571 call VerifyScreenDump(buf, 'Test_vim9_silent_echo', {})
1572
1573 " clean up
1574 call StopVimInTerminal(buf)
1575 call delete('XTest_silent_echo')
Bram Moolenaar8f510af2020-07-05 18:48:23 +02001576endfunc
Bram Moolenaar47e7d702020-07-05 18:18:42 +02001577
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001578""""""" builtin functions that behave differently in Vim9
Bram Moolenaare15eebd2020-08-18 19:11:38 +02001579
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001580def Test_bufname()
1581 split SomeFile
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001582 bufname('%')->assert_equal('SomeFile')
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001583 edit OtherFile
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001584 bufname('#')->assert_equal('SomeFile')
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001585 close
Bram Moolenaar191929b2020-08-19 21:20:49 +02001586enddef
1587
Bram Moolenaara5d38412020-09-02 21:02:35 +02001588def Test_bufwinid()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001589 var origwin = win_getid()
Bram Moolenaara5d38412020-09-02 21:02:35 +02001590 below split SomeFile
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001591 var SomeFileID = win_getid()
Bram Moolenaara5d38412020-09-02 21:02:35 +02001592 below split OtherFile
1593 below split SomeFile
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001594 bufwinid('SomeFile')->assert_equal(SomeFileID)
Bram Moolenaara5d38412020-09-02 21:02:35 +02001595
1596 win_gotoid(origwin)
1597 only
1598 bwipe SomeFile
1599 bwipe OtherFile
1600enddef
1601
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001602def Test_count()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001603 count('ABC ABC ABC', 'b', true)->assert_equal(3)
1604 count('ABC ABC ABC', 'b', false)->assert_equal(0)
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001605enddef
1606
1607def Test_expand()
1608 split SomeFile
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001609 expand('%', true, true)->assert_equal(['SomeFile'])
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001610 close
1611enddef
1612
1613def Test_getbufinfo()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001614 var bufinfo = getbufinfo(bufnr())
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001615 getbufinfo('%')->assert_equal(bufinfo)
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001616
1617 edit Xtestfile1
1618 hide edit Xtestfile2
1619 hide enew
1620 getbufinfo(#{bufloaded: true, buflisted: true, bufmodified: false})
1621 ->len()->assert_equal(3)
1622 bwipe Xtestfile1 Xtestfile2
1623enddef
1624
Bram Moolenaara5d38412020-09-02 21:02:35 +02001625def Test_getbufline()
1626 e SomeFile
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001627 var buf = bufnr()
Bram Moolenaara5d38412020-09-02 21:02:35 +02001628 e #
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001629 var lines = ['aaa', 'bbb', 'ccc']
Bram Moolenaara5d38412020-09-02 21:02:35 +02001630 setbufline(buf, 1, lines)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001631 getbufline('#', 1, '$')->assert_equal(lines)
Bram Moolenaara5d38412020-09-02 21:02:35 +02001632
1633 bwipe!
1634enddef
1635
1636def Test_getchangelist()
1637 new
1638 setline(1, 'some text')
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001639 var changelist = bufnr()->getchangelist()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001640 getchangelist('%')->assert_equal(changelist)
Bram Moolenaara5d38412020-09-02 21:02:35 +02001641 bwipe!
1642enddef
1643
Bram Moolenaarc08cc722020-09-05 17:51:23 +02001644def Test_getchar()
Bram Moolenaar636c5d52020-09-05 18:48:57 +02001645 while getchar(0)
1646 endwhile
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001647 getchar(true)->assert_equal(0)
Bram Moolenaarc08cc722020-09-05 17:51:23 +02001648enddef
1649
Bram Moolenaard217a872020-09-05 18:31:33 +02001650def Test_getcompletion()
1651 set wildignore=*.vim,*~
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001652 var l = getcompletion('run', 'file', true)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001653 l->assert_equal([])
Bram Moolenaard217a872020-09-05 18:31:33 +02001654 set wildignore&
1655enddef
1656
Bram Moolenaar67ff97d2020-09-02 21:45:54 +02001657def Test_getreg()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001658 var lines = ['aaa', 'bbb', 'ccc']
Bram Moolenaar67ff97d2020-09-02 21:45:54 +02001659 setreg('a', lines)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001660 getreg('a', true, true)->assert_equal(lines)
Bram Moolenaar67ff97d2020-09-02 21:45:54 +02001661enddef
1662
Bram Moolenaar5892ea12020-09-02 21:53:11 +02001663def Test_glob()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001664 glob('runtest.vim', true, true, true)->assert_equal(['runtest.vim'])
Bram Moolenaar5892ea12020-09-02 21:53:11 +02001665enddef
1666
Bram Moolenaarf966ce52020-09-02 21:57:07 +02001667def Test_globpath()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001668 globpath('.', 'runtest.vim', true, true, true)->assert_equal(['./runtest.vim'])
Bram Moolenaarf966ce52020-09-02 21:57:07 +02001669enddef
1670
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001671def Test_has()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001672 has('eval', true)->assert_equal(1)
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001673enddef
1674
Bram Moolenaar04d594b2020-09-02 22:25:35 +02001675def Test_hasmapto()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001676 hasmapto('foobar', 'i', true)->assert_equal(0)
Bram Moolenaar04d594b2020-09-02 22:25:35 +02001677 iabbrev foo foobar
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001678 hasmapto('foobar', 'i', true)->assert_equal(1)
Bram Moolenaar04d594b2020-09-02 22:25:35 +02001679 iunabbrev foo
1680enddef
1681
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001682def Test_index()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001683 index(['a', 'b', 'a', 'B'], 'b', 2, true)->assert_equal(3)
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001684enddef
1685
1686def Test_list2str_str2list_utf8()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001687 var s = "\u3042\u3044"
1688 var l = [0x3042, 0x3044]
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001689 str2list(s, true)->assert_equal(l)
1690 list2str(l, true)->assert_equal(s)
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001691enddef
1692
Bram Moolenaar04d594b2020-09-02 22:25:35 +02001693def SID(): number
1694 return expand('<SID>')
1695 ->matchstr('<SNR>\zs\d\+\ze_$')
1696 ->str2nr()
1697enddef
1698
1699def Test_maparg()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001700 var lnum = str2nr(expand('<sflnum>'))
Bram Moolenaar04d594b2020-09-02 22:25:35 +02001701 map foo bar
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001702 maparg('foo', '', false, true)->assert_equal(#{
Bram Moolenaar04d594b2020-09-02 22:25:35 +02001703 lnum: lnum + 1,
1704 script: 0,
1705 mode: ' ',
1706 silent: 0,
1707 noremap: 0,
1708 lhs: 'foo',
1709 lhsraw: 'foo',
1710 nowait: 0,
1711 expr: 0,
1712 sid: SID(),
1713 rhs: 'bar',
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001714 buffer: 0})
Bram Moolenaar04d594b2020-09-02 22:25:35 +02001715 unmap foo
1716enddef
1717
1718def Test_mapcheck()
1719 iabbrev foo foobar
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001720 mapcheck('foo', 'i', true)->assert_equal('foobar')
Bram Moolenaar04d594b2020-09-02 22:25:35 +02001721 iunabbrev foo
1722enddef
1723
Bram Moolenaar74273e62020-10-01 21:37:21 +02001724def Test_maparg_mapset()
1725 nnoremap <F3> :echo "hit F3"<CR>
1726 var mapsave = maparg('<F3>', 'n', false, true)
1727 mapset('n', false, mapsave)
1728
1729 nunmap <F3>
1730enddef
1731
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001732def Test_nr2char()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001733 nr2char(97, true)->assert_equal('a')
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001734enddef
1735
1736def Test_readdir()
1737 eval expand('sautest')->readdir({e -> e[0] !=# '.'})
1738 eval expand('sautest')->readdirex({e -> e.name[0] !=# '.'})
1739enddef
1740
1741def Test_search()
1742 new
1743 setline(1, ['foo', 'bar'])
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001744 var val = 0
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001745 # skip expr returns boolean
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001746 search('bar', 'W', 0, 0, {-> val == 1})->assert_equal(2)
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001747 :1
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001748 search('bar', 'W', 0, 0, {-> val == 0})->assert_equal(0)
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001749 # skip expr returns number, only 0 and 1 are accepted
1750 :1
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001751 search('bar', 'W', 0, 0, {-> 0})->assert_equal(2)
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001752 :1
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001753 search('bar', 'W', 0, 0, {-> 1})->assert_equal(0)
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001754 assert_fails("search('bar', '', 0, 0, {-> -1})", 'E1023:')
1755 assert_fails("search('bar', '', 0, 0, {-> -1})", 'E1023:')
1756enddef
1757
1758def Test_searchcount()
1759 new
1760 setline(1, "foo bar")
1761 :/foo
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001762 searchcount(#{recompute: true})
1763 ->assert_equal(#{
1764 exact_match: 1,
1765 current: 1,
1766 total: 1,
1767 maxcount: 99,
1768 incomplete: 0})
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001769 bwipe!
1770enddef
1771
1772def Test_searchdecl()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001773 searchdecl('blah', true, true)->assert_equal(1)
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001774enddef
1775
1776def Test_setbufvar()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001777 setbufvar(bufnr('%'), '&syntax', 'vim')
1778 &syntax->assert_equal('vim')
1779 setbufvar(bufnr('%'), '&ts', 16)
1780 &ts->assert_equal(16)
1781 settabwinvar(1, 1, '&syntax', 'vam')
1782 &syntax->assert_equal('vam')
1783 settabwinvar(1, 1, '&ts', 15)
1784 &ts->assert_equal(15)
1785 setlocal ts=8
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001786
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001787 setbufvar('%', 'myvar', 123)
1788 getbufvar('%', 'myvar')->assert_equal(123)
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001789enddef
1790
Bram Moolenaar401f0c02020-09-05 22:37:39 +02001791def Test_setloclist()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001792 var items = [#{filename: '/tmp/file', lnum: 1, valid: true}]
1793 var what = #{items: items}
Bram Moolenaar401f0c02020-09-05 22:37:39 +02001794 setqflist([], ' ', what)
1795 setloclist(0, [], ' ', what)
1796enddef
1797
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001798def Test_setreg()
1799 setreg('a', ['aaa', 'bbb', 'ccc'])
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001800 var reginfo = getreginfo('a')
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001801 setreg('a', reginfo)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001802 getreginfo('a')->assert_equal(reginfo)
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001803enddef
1804
Bram Moolenaar7c27f332020-09-05 22:45:55 +02001805def Test_spellsuggest()
1806 if !has('spell')
1807 MissingFeature 'spell'
1808 else
1809 spellsuggest('marrch', 1, true)->assert_equal(['March'])
1810 endif
1811enddef
1812
Bram Moolenaar3986b942020-09-06 16:09:04 +02001813def Test_split()
1814 split(' aa bb ', '\W\+', true)->assert_equal(['', 'aa', 'bb', ''])
1815enddef
1816
1817def Test_str2nr()
1818 str2nr("1'000'000", 10, true)->assert_equal(1000000)
1819enddef
1820
1821def Test_strchars()
1822 strchars("A\u20dd", true)->assert_equal(1)
1823enddef
1824
Bram Moolenaarad304702020-09-06 18:22:53 +02001825def Test_submatch()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001826 var pat = 'A\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)'
1827 var Rep = {-> range(10)->map({_, v -> submatch(v, true)})->string()}
1828 var actual = substitute('A123456789', pat, Rep, '')
1829 var expected = "[['A123456789'], ['1'], ['2'], ['3'], ['4'], ['5'], ['6'], ['7'], ['8'], ['9']]"
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001830 actual->assert_equal(expected)
Bram Moolenaarad304702020-09-06 18:22:53 +02001831enddef
1832
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001833def Test_synID()
1834 new
1835 setline(1, "text")
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001836 synID(1, 1, true)->assert_equal(0)
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001837 bwipe!
1838enddef
1839
Bram Moolenaarad304702020-09-06 18:22:53 +02001840def Test_term_gettty()
Bram Moolenaar63969ef2020-09-06 20:06:59 +02001841 if !has('terminal')
1842 MissingFeature 'terminal'
1843 else
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001844 var buf = Run_shell_in_terminal({})
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001845 term_gettty(buf, true)->assert_notequal('')
Bram Moolenaar63969ef2020-09-06 20:06:59 +02001846 StopShellInTerminal(buf)
1847 endif
Bram Moolenaarad304702020-09-06 18:22:53 +02001848enddef
1849
1850def Test_term_start()
Bram Moolenaar63969ef2020-09-06 20:06:59 +02001851 if !has('terminal')
1852 MissingFeature 'terminal'
1853 else
1854 botright new
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001855 var winnr = winnr()
Bram Moolenaar63969ef2020-09-06 20:06:59 +02001856 term_start(&shell, #{curwin: true})
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001857 winnr()->assert_equal(winnr)
Bram Moolenaar63969ef2020-09-06 20:06:59 +02001858 bwipe!
1859 endif
Bram Moolenaarad304702020-09-06 18:22:53 +02001860enddef
1861
Bram Moolenaar418155d2020-09-06 18:39:38 +02001862def Test_timer_paused()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001863 var id = timer_start(50, {-> 0})
Bram Moolenaar418155d2020-09-06 18:39:38 +02001864 timer_pause(id, true)
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001865 var info = timer_info(id)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001866 info[0]['paused']->assert_equal(1)
Bram Moolenaar418155d2020-09-06 18:39:38 +02001867 timer_stop(id)
1868enddef
1869
Bram Moolenaar4b9bd692020-09-05 21:57:53 +02001870def Test_win_splitmove()
1871 split
1872 win_splitmove(1, 2, #{vertical: true, rightbelow: true})
1873 close
1874enddef
1875
1876""""""" end of builtin functions
1877
1878def Fibonacci(n: number): number
1879 if n < 2
1880 return n
1881 else
1882 return Fibonacci(n - 1) + Fibonacci(n - 2)
1883 endif
1884enddef
1885
Bram Moolenaar985116a2020-07-12 17:31:09 +02001886def Test_recursive_call()
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001887 Fibonacci(20)->assert_equal(6765)
Bram Moolenaar985116a2020-07-12 17:31:09 +02001888enddef
1889
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001890def TreeWalk(dir: string): list<any>
1891 return readdir(dir)->map({_, val ->
1892 fnamemodify(dir .. '/' .. val, ':p')->isdirectory()
Bram Moolenaarbb1b5e22020-08-05 10:53:21 +02001893 ? {val: TreeWalk(dir .. '/' .. val)}
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001894 : val
1895 })
1896enddef
1897
1898def Test_closure_in_map()
1899 mkdir('XclosureDir/tdir', 'p')
1900 writefile(['111'], 'XclosureDir/file1')
1901 writefile(['222'], 'XclosureDir/file2')
1902 writefile(['333'], 'XclosureDir/tdir/file3')
1903
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001904 TreeWalk('XclosureDir')->assert_equal(['file1', 'file2', {'tdir': ['file3']}])
Bram Moolenaar08f7a412020-07-13 20:41:08 +02001905
1906 delete('XclosureDir', 'rf')
1907enddef
1908
Bram Moolenaar7b5d5442020-10-04 13:42:34 +02001909def Test_invalid_function_name()
1910 var lines =<< trim END
1911 vim9script
1912 def s: list<string>
1913 END
1914 CheckScriptFailure(lines, 'E129:')
1915
1916 lines =<< trim END
1917 vim9script
1918 def g: list<string>
1919 END
1920 CheckScriptFailure(lines, 'E129:')
1921
1922 lines =<< trim END
1923 vim9script
1924 def <SID>: list<string>
1925 END
1926 CheckScriptFailure(lines, 'E884:')
1927
1928 lines =<< trim END
1929 vim9script
1930 def F list<string>
1931 END
1932 CheckScriptFailure(lines, 'E488:')
1933enddef
1934
Bram Moolenaara90afb92020-07-15 22:38:56 +02001935def Test_partial_call()
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001936 var Xsetlist = function('setloclist', [0])
Bram Moolenaara90afb92020-07-15 22:38:56 +02001937 Xsetlist([], ' ', {'title': 'test'})
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001938 getloclist(0, {'title': 1})->assert_equal({'title': 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02001939
1940 Xsetlist = function('setloclist', [0, [], ' '])
1941 Xsetlist({'title': 'test'})
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001942 getloclist(0, {'title': 1})->assert_equal({'title': 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02001943
1944 Xsetlist = function('setqflist')
1945 Xsetlist([], ' ', {'title': 'test'})
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001946 getqflist({'title': 1})->assert_equal({'title': 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02001947
1948 Xsetlist = function('setqflist', [[], ' '])
1949 Xsetlist({'title': 'test'})
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001950 getqflist({'title': 1})->assert_equal({'title': 'test'})
Bram Moolenaara90afb92020-07-15 22:38:56 +02001951enddef
1952
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001953def Test_cmd_modifier()
1954 tab echo '0'
Bram Moolenaard2c61702020-09-06 15:58:36 +02001955 CheckDefFailure(['5tab echo 3'], 'E16:')
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001956enddef
1957
1958def Test_restore_modifiers()
1959 # check that when compiling a :def function command modifiers are not messed
1960 # up.
Bram Moolenaar7a9cbca2020-09-27 22:47:05 +02001961 var lines =<< trim END
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001962 vim9script
1963 set eventignore=
1964 autocmd QuickFixCmdPost * copen
1965 def AutocmdsDisabled()
1966 eval 0
1967 enddef
1968 func Func()
1969 noautocmd call s:AutocmdsDisabled()
1970 let g:ei_after = &eventignore
1971 endfunc
1972 Func()
1973 END
1974 CheckScriptSuccess(lines)
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001975 g:ei_after->assert_equal('')
Bram Moolenaar2dd0a2c2020-08-08 15:10:27 +02001976enddef
1977
Bram Moolenaardfa3d552020-09-10 22:05:08 +02001978def StackTop()
1979 eval 1
1980 eval 2
1981 # call not on fourth line
1982 StackBot()
1983enddef
1984
1985def StackBot()
1986 # throw an error
1987 eval [][0]
1988enddef
1989
1990def Test_callstack_def()
1991 try
1992 StackTop()
1993 catch
Bram Moolenaarc0c71e92020-09-11 19:09:48 +02001994 v:throwpoint->assert_match('Test_callstack_def\[2\]..StackTop\[4\]..StackBot, line 2')
Bram Moolenaardfa3d552020-09-10 22:05:08 +02001995 endtry
1996enddef
1997
Bram Moolenaarf7779c62020-05-03 15:38:16 +02001998
Bram Moolenaar5deeb3f2020-04-05 17:08:17 +02001999" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker