blob: 8ec3db14829f1c1976e041172569aa9736f1b65b [file] [log] [blame]
Bram Moolenaar94738d82020-10-21 14:25:07 +02001" Test using builtin functions in the Vim9 script language.
2
3source check.vim
4source vim9.vim
5
6" Test for passing too many or too few arguments to builtin functions
7func Test_internalfunc_arg_error()
8 let l =<< trim END
9 def! FArgErr(): float
10 return ceil(1.1, 2)
11 enddef
12 defcompile
13 END
14 call writefile(l, 'Xinvalidarg')
15 call assert_fails('so Xinvalidarg', 'E118:', '', 1, 'FArgErr')
16 let l =<< trim END
17 def! FArgErr(): float
18 return ceil()
19 enddef
20 defcompile
21 END
22 call writefile(l, 'Xinvalidarg')
23 call assert_fails('so Xinvalidarg', 'E119:', '', 1, 'FArgErr')
24 call delete('Xinvalidarg')
25endfunc
26
27" Test for builtin functions returning different types
28func Test_InternalFuncRetType()
29 let lines =<< trim END
30 def RetFloat(): float
31 return ceil(1.456)
32 enddef
33
34 def RetListAny(): list<any>
Bram Moolenaare0de1712020-12-02 17:36:54 +010035 return items({k: 'v'})
Bram Moolenaar94738d82020-10-21 14:25:07 +020036 enddef
37
38 def RetListString(): list<string>
39 return split('a:b:c', ':')
40 enddef
41
42 def RetListDictAny(): list<dict<any>>
43 return getbufinfo()
44 enddef
45
46 def RetDictNumber(): dict<number>
47 return wordcount()
48 enddef
49
50 def RetDictString(): dict<string>
51 return environ()
52 enddef
53 END
54 call writefile(lines, 'Xscript')
55 source Xscript
56
57 call RetFloat()->assert_equal(2.0)
58 call RetListAny()->assert_equal([['k', 'v']])
59 call RetListString()->assert_equal(['a', 'b', 'c'])
60 call RetListDictAny()->assert_notequal([])
61 call RetDictNumber()->assert_notequal({})
62 call RetDictString()->assert_notequal({})
63 call delete('Xscript')
64endfunc
65
66def Test_abs()
67 assert_equal(0, abs(0))
68 assert_equal(2, abs(-2))
69 assert_equal(3, abs(3))
70 CheckDefFailure(['abs("text")'], 'E1013: Argument 1: type mismatch, expected number but got string', 1)
71 if has('float')
72 assert_equal(0, abs(0))
73 assert_equal(2.0, abs(-2.0))
74 assert_equal(3.0, abs(3.0))
75 endif
76enddef
77
78def Test_add_list()
79 var l: list<number> # defaults to empty list
80 add(l, 9)
81 assert_equal([9], l)
82
83 var lines =<< trim END
84 var l: list<number>
85 add(l, "x")
86 END
87 CheckDefFailure(lines, 'E1012:', 2)
88
89 lines =<< trim END
90 var l: list<number> = test_null_list()
91 add(l, 123)
92 END
93 CheckDefExecFailure(lines, 'E1130:', 2)
94enddef
95
96def Test_add_blob()
97 var b1: blob = 0z12
98 add(b1, 0x34)
99 assert_equal(0z1234, b1)
100
101 var b2: blob # defaults to empty blob
102 add(b2, 0x67)
103 assert_equal(0z67, b2)
104
105 var lines =<< trim END
106 var b: blob
107 add(b, "x")
108 END
109 CheckDefFailure(lines, 'E1012:', 2)
110
111 lines =<< trim END
112 var b: blob = test_null_blob()
113 add(b, 123)
114 END
115 CheckDefExecFailure(lines, 'E1131:', 2)
116enddef
117
Bram Moolenaar3af15ab2021-01-17 16:16:23 +0100118def Test_append()
119 new
120 setline(1, range(3))
121 var res1: number = append(1, 'one')
122 assert_equal(0, res1)
123 var res2: bool = append(3, 'two')
124 assert_equal(false, res2)
125 assert_equal(['0', 'one', '1', 'two', '2'], getline(1, 6))
126enddef
127
128def Test_buflisted()
129 var res: bool = buflisted('asdf')
130 assert_equal(false, res)
131enddef
132
Bram Moolenaar94738d82020-10-21 14:25:07 +0200133def Test_bufname()
134 split SomeFile
135 bufname('%')->assert_equal('SomeFile')
136 edit OtherFile
137 bufname('#')->assert_equal('SomeFile')
138 close
139enddef
140
141def Test_bufnr()
142 var buf = bufnr()
143 bufnr('%')->assert_equal(buf)
144
145 buf = bufnr('Xdummy', true)
146 buf->assert_notequal(-1)
147 exe 'bwipe! ' .. buf
148enddef
149
150def Test_bufwinid()
151 var origwin = win_getid()
152 below split SomeFile
153 var SomeFileID = win_getid()
154 below split OtherFile
155 below split SomeFile
156 bufwinid('SomeFile')->assert_equal(SomeFileID)
157
158 win_gotoid(origwin)
159 only
160 bwipe SomeFile
161 bwipe OtherFile
162enddef
163
164def Test_call_call()
165 var l = [3, 2, 1]
166 call('reverse', [l])
167 l->assert_equal([1, 2, 3])
168enddef
169
170def Test_char2nr()
171 char2nr('あ', true)->assert_equal(12354)
172enddef
173
174def Test_col()
175 new
176 setline(1, 'asdf')
177 col([1, '$'])->assert_equal(5)
178enddef
179
180def Test_copy_return_type()
181 var l = copy([1, 2, 3])
182 var res = 0
183 for n in l
184 res += n
185 endfor
186 res->assert_equal(6)
187
188 var dl = deepcopy([1, 2, 3])
189 res = 0
190 for n in dl
191 res += n
192 endfor
193 res->assert_equal(6)
194
195 dl = deepcopy([1, 2, 3], true)
196enddef
197
198def Test_count()
199 count('ABC ABC ABC', 'b', true)->assert_equal(3)
200 count('ABC ABC ABC', 'b', false)->assert_equal(0)
201enddef
202
Bram Moolenaar9a963372020-12-21 21:58:46 +0100203def Test_cursor()
204 new
205 setline(1, range(4))
206 cursor(2, 1)
207 assert_equal(2, getcurpos()[1])
208 cursor('$', 1)
209 assert_equal(4, getcurpos()[1])
210
211 var lines =<< trim END
212 cursor('2', 1)
213 END
214 CheckDefExecAndScriptFailure(lines, 'E475:')
215enddef
216
Bram Moolenaar3af15ab2021-01-17 16:16:23 +0100217def Test_delete()
218 var res: bool = delete('doesnotexist')
219 assert_equal(true, res)
220enddef
221
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100222def Test_executable()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100223 assert_false(executable(""))
224 assert_false(executable(test_null_string()))
225
226 CheckDefExecFailure(['echo executable(123)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100227 CheckDefExecFailure(['echo executable(true)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100228enddef
229
230def Test_exepath()
231 CheckDefExecFailure(['echo exepath(true)'], 'E928:')
232 CheckDefExecFailure(['echo exepath(v:null)'], 'E928:')
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100233 CheckDefExecFailure(['echo exepath("")'], 'E1142:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100234enddef
235
Bram Moolenaar94738d82020-10-21 14:25:07 +0200236def Test_expand()
237 split SomeFile
238 expand('%', true, true)->assert_equal(['SomeFile'])
239 close
240enddef
241
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100242def Test_extend_arg_types()
Bram Moolenaarc03f5c62021-01-31 17:48:30 +0100243 g:number_one = 1
244 g:string_keep = 'keep'
245 var lines =<< trim END
246 assert_equal([1, 2, 3], extend([1, 2], [3]))
247 assert_equal([3, 1, 2], extend([1, 2], [3], 0))
248 assert_equal([1, 3, 2], extend([1, 2], [3], 1))
249 assert_equal([1, 3, 2], extend([1, 2], [3], g:number_one))
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100250
Bram Moolenaarc03f5c62021-01-31 17:48:30 +0100251 assert_equal({a: 1, b: 2, c: 3}, extend({a: 1, b: 2}, {c: 3}))
252 assert_equal({a: 1, b: 4}, extend({a: 1, b: 2}, {b: 4}))
253 assert_equal({a: 1, b: 2}, extend({a: 1, b: 2}, {b: 4}, 'keep'))
254 assert_equal({a: 1, b: 2}, extend({a: 1, b: 2}, {b: 4}, g:string_keep))
Bram Moolenaar193f6202020-11-16 20:08:35 +0100255
Bram Moolenaarc03f5c62021-01-31 17:48:30 +0100256 var res: list<dict<any>>
257 extend(res, mapnew([1, 2], (_, v) => ({})))
258 assert_equal([{}, {}], res)
259 END
260 CheckDefAndScriptSuccess(lines)
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100261
262 CheckDefFailure(['extend([1, 2], 3)'], 'E1013: Argument 2: type mismatch, expected list<number> but got number')
263 CheckDefFailure(['extend([1, 2], ["x"])'], 'E1013: Argument 2: type mismatch, expected list<number> but got list<string>')
264 CheckDefFailure(['extend([1, 2], [3], "x")'], 'E1013: Argument 3: type mismatch, expected number but got string')
265
Bram Moolenaare0de1712020-12-02 17:36:54 +0100266 CheckDefFailure(['extend({a: 1}, 42)'], 'E1013: Argument 2: type mismatch, expected dict<number> but got number')
267 CheckDefFailure(['extend({a: 1}, {b: "x"})'], 'E1013: Argument 2: type mismatch, expected dict<number> but got dict<string>')
268 CheckDefFailure(['extend({a: 1}, {b: 2}, 1)'], 'E1013: Argument 3: type mismatch, expected string but got number')
Bram Moolenaar351ead02021-01-16 16:07:01 +0100269
270 CheckDefFailure(['extend([1], ["b"])'], 'E1013: Argument 2: type mismatch, expected list<number> but got list<string>')
Bram Moolenaare32e5162021-01-21 20:21:29 +0100271 CheckDefExecFailure(['extend([1], ["b", 1])'], 'E1013: Argument 2: type mismatch, expected list<number> but got list<any>')
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100272enddef
273
Bram Moolenaarb0e6b512021-01-12 20:23:40 +0100274def Test_extendnew()
275 assert_equal([1, 2, 'a'], extendnew([1, 2], ['a']))
276 assert_equal({one: 1, two: 'a'}, extendnew({one: 1}, {two: 'a'}))
277
278 CheckDefFailure(['extendnew({a: 1}, 42)'], 'E1013: Argument 2: type mismatch, expected dict<number> but got number')
279 CheckDefFailure(['extendnew({a: 1}, [42])'], 'E1013: Argument 2: type mismatch, expected dict<number> but got list<number>')
280 CheckDefFailure(['extendnew([1, 2], "x")'], 'E1013: Argument 2: type mismatch, expected list<number> but got string')
281 CheckDefFailure(['extendnew([1, 2], {x: 1})'], 'E1013: Argument 2: type mismatch, expected list<number> but got dict<number>')
282enddef
283
Bram Moolenaar94738d82020-10-21 14:25:07 +0200284def Test_extend_return_type()
285 var l = extend([1, 2], [3])
286 var res = 0
287 for n in l
288 res += n
289 endfor
290 res->assert_equal(6)
291enddef
292
Bram Moolenaaraa210a32021-01-02 15:41:03 +0100293func g:ExtendDict(d)
294 call extend(a:d, #{xx: 'x'})
295endfunc
296
297def Test_extend_dict_item_type()
298 var lines =<< trim END
299 var d: dict<number> = {a: 1}
300 extend(d, {b: 2})
301 END
302 CheckDefAndScriptSuccess(lines)
303
304 lines =<< trim END
305 var d: dict<number> = {a: 1}
306 extend(d, {b: 'x'})
307 END
Bram Moolenaarc03f5c62021-01-31 17:48:30 +0100308 CheckDefAndScriptFailure(lines, 'E1013: Argument 2: type mismatch, expected dict<number> but got dict<string>', 2)
Bram Moolenaaraa210a32021-01-02 15:41:03 +0100309
310 lines =<< trim END
311 var d: dict<number> = {a: 1}
312 g:ExtendDict(d)
313 END
314 CheckDefExecFailure(lines, 'E1012: Type mismatch; expected number but got string', 0)
315 CheckScriptFailure(['vim9script'] + lines, 'E1012:', 1)
316enddef
317
318func g:ExtendList(l)
319 call extend(a:l, ['x'])
320endfunc
321
322def Test_extend_list_item_type()
323 var lines =<< trim END
324 var l: list<number> = [1]
325 extend(l, [2])
326 END
327 CheckDefAndScriptSuccess(lines)
328
329 lines =<< trim END
330 var l: list<number> = [1]
331 extend(l, ['x'])
332 END
Bram Moolenaarc03f5c62021-01-31 17:48:30 +0100333 CheckDefAndScriptFailure(lines, 'E1013: Argument 2: type mismatch, expected list<number> but got list<string>', 2)
Bram Moolenaaraa210a32021-01-02 15:41:03 +0100334
335 lines =<< trim END
336 var l: list<number> = [1]
337 g:ExtendList(l)
338 END
339 CheckDefExecFailure(lines, 'E1012: Type mismatch; expected number but got string', 0)
340 CheckScriptFailure(['vim9script'] + lines, 'E1012:', 1)
341enddef
Bram Moolenaar94738d82020-10-21 14:25:07 +0200342
Bram Moolenaar64ed4d42021-01-12 21:22:31 +0100343def Test_job_info_return_type()
344 if has('job')
345 job_start(&shell)
346 var jobs = job_info()
Bram Moolenaara47e05f2021-01-12 21:49:00 +0100347 assert_equal('list<job>', typename(jobs))
348 assert_equal('dict<any>', typename(job_info(jobs[0])))
Bram Moolenaar64ed4d42021-01-12 21:22:31 +0100349 job_stop(jobs[0])
350 endif
351enddef
352
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100353def Test_filereadable()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100354 assert_false(filereadable(""))
355 assert_false(filereadable(test_null_string()))
356
357 CheckDefExecFailure(['echo filereadable(123)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100358 CheckDefExecFailure(['echo filereadable(true)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100359enddef
360
361def Test_filewritable()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100362 assert_false(filewritable(""))
363 assert_false(filewritable(test_null_string()))
364
365 CheckDefExecFailure(['echo filewritable(123)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100366 CheckDefExecFailure(['echo filewritable(true)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100367enddef
368
369def Test_finddir()
370 CheckDefExecFailure(['echo finddir(true)'], 'E928:')
371 CheckDefExecFailure(['echo finddir(v:null)'], 'E928:')
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100372 CheckDefExecFailure(['echo finddir("")'], 'E1142:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100373enddef
374
375def Test_findfile()
376 CheckDefExecFailure(['echo findfile(true)'], 'E928:')
377 CheckDefExecFailure(['echo findfile(v:null)'], 'E928:')
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100378 CheckDefExecFailure(['echo findfile("")'], 'E1142:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100379enddef
380
Bram Moolenaar3b690062021-02-01 20:14:51 +0100381def Test_flattennew()
382 var lines =<< trim END
383 var l = [1, [2, [3, 4]], 5]
384 call assert_equal([1, 2, 3, 4, 5], flattennew(l))
385 call assert_equal([1, [2, [3, 4]], 5], l)
386
387 call assert_equal([1, 2, [3, 4], 5], flattennew(l, 1))
388 call assert_equal([1, [2, [3, 4]], 5], l)
389 END
390 CheckDefAndScriptSuccess(lines)
391
392 lines =<< trim END
393 echo flatten([1, 2, 3])
394 END
395 CheckDefAndScriptFailure(lines, 'E1158:')
396enddef
397
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100398def Test_fnamemodify()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100399 CheckDefSuccess(['echo fnamemodify(test_null_string(), ":p")'])
400 CheckDefSuccess(['echo fnamemodify("", ":p")'])
401 CheckDefSuccess(['echo fnamemodify("file", test_null_string())'])
402 CheckDefSuccess(['echo fnamemodify("file", "")'])
403
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100404 CheckDefExecFailure(['echo fnamemodify(true, ":p")'], 'E928:')
405 CheckDefExecFailure(['echo fnamemodify(v:null, ":p")'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100406 CheckDefExecFailure(['echo fnamemodify("file", true)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100407enddef
408
Bram Moolenaar2e5910b2021-02-03 17:41:24 +0100409def Wrong_dict_key_type(items: list<number>): list<number>
410 return filter(items, (_, val) => get({[val]: 1}, 'x'))
411enddef
412
Bram Moolenaar94738d82020-10-21 14:25:07 +0200413def Test_filter_wrong_dict_key_type()
Bram Moolenaar2e5910b2021-02-03 17:41:24 +0100414 assert_fails('Wrong_dict_key_type([1, v:null, 3])', 'E1013:')
Bram Moolenaar94738d82020-10-21 14:25:07 +0200415enddef
416
417def Test_filter_return_type()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100418 var l = filter([1, 2, 3], () => 1)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200419 var res = 0
420 for n in l
421 res += n
422 endfor
423 res->assert_equal(6)
424enddef
425
Bram Moolenaarfc0e8f52020-12-25 20:24:51 +0100426def Test_filter_missing_argument()
427 var dict = {aa: [1], ab: [2], ac: [3], de: [4]}
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100428 var res = dict->filter((k) => k =~ 'a' && k !~ 'b')
Bram Moolenaarfc0e8f52020-12-25 20:24:51 +0100429 res->assert_equal({aa: [1], ac: [3]})
430enddef
Bram Moolenaar94738d82020-10-21 14:25:07 +0200431
432def Test_garbagecollect()
433 garbagecollect(true)
434enddef
435
436def Test_getbufinfo()
437 var bufinfo = getbufinfo(bufnr())
438 getbufinfo('%')->assert_equal(bufinfo)
439
440 edit Xtestfile1
441 hide edit Xtestfile2
442 hide enew
Bram Moolenaare0de1712020-12-02 17:36:54 +0100443 getbufinfo({bufloaded: true, buflisted: true, bufmodified: false})
Bram Moolenaar94738d82020-10-21 14:25:07 +0200444 ->len()->assert_equal(3)
445 bwipe Xtestfile1 Xtestfile2
446enddef
447
448def Test_getbufline()
449 e SomeFile
450 var buf = bufnr()
451 e #
452 var lines = ['aaa', 'bbb', 'ccc']
453 setbufline(buf, 1, lines)
454 getbufline('#', 1, '$')->assert_equal(lines)
Bram Moolenaare6e70a12020-10-22 18:23:38 +0200455 getbufline(-1, '$', '$')->assert_equal([])
456 getbufline(-1, 1, '$')->assert_equal([])
Bram Moolenaar94738d82020-10-21 14:25:07 +0200457
458 bwipe!
459enddef
460
461def Test_getchangelist()
462 new
463 setline(1, 'some text')
464 var changelist = bufnr()->getchangelist()
465 getchangelist('%')->assert_equal(changelist)
466 bwipe!
467enddef
468
469def Test_getchar()
470 while getchar(0)
471 endwhile
472 getchar(true)->assert_equal(0)
473enddef
474
Bram Moolenaar7ad67d12021-03-10 16:08:26 +0100475def Test_getenv()
476 if getenv('does-not_exist') == ''
477 assert_report('getenv() should return null')
478 endif
479 if getenv('does-not_exist') == null
480 else
481 assert_report('getenv() should return null')
482 endif
483 $SOMEENVVAR = 'some'
484 assert_equal('some', getenv('SOMEENVVAR'))
485 unlet $SOMEENVVAR
486enddef
487
Bram Moolenaar94738d82020-10-21 14:25:07 +0200488def Test_getcompletion()
489 set wildignore=*.vim,*~
490 var l = getcompletion('run', 'file', true)
491 l->assert_equal([])
492 set wildignore&
493enddef
494
495def Test_getloclist_return_type()
496 var l = getloclist(1)
497 l->assert_equal([])
498
Bram Moolenaare0de1712020-12-02 17:36:54 +0100499 var d = getloclist(1, {items: 0})
500 d->assert_equal({items: []})
Bram Moolenaar94738d82020-10-21 14:25:07 +0200501enddef
502
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100503def Test_getfperm()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100504 assert_equal('', getfperm(""))
505 assert_equal('', getfperm(test_null_string()))
506
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100507 CheckDefExecFailure(['echo getfperm(true)'], 'E928:')
508 CheckDefExecFailure(['echo getfperm(v:null)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100509enddef
510
511def Test_getfsize()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100512 assert_equal(-1, getfsize(""))
513 assert_equal(-1, getfsize(test_null_string()))
514
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100515 CheckDefExecFailure(['echo getfsize(true)'], 'E928:')
516 CheckDefExecFailure(['echo getfsize(v:null)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100517enddef
518
519def Test_getftime()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100520 assert_equal(-1, getftime(""))
521 assert_equal(-1, getftime(test_null_string()))
522
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100523 CheckDefExecFailure(['echo getftime(true)'], 'E928:')
524 CheckDefExecFailure(['echo getftime(v:null)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100525enddef
526
527def Test_getftype()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100528 assert_equal('', getftype(""))
529 assert_equal('', getftype(test_null_string()))
530
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100531 CheckDefExecFailure(['echo getftype(true)'], 'E928:')
532 CheckDefExecFailure(['echo getftype(v:null)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100533enddef
534
Bram Moolenaar94738d82020-10-21 14:25:07 +0200535def Test_getqflist_return_type()
536 var l = getqflist()
537 l->assert_equal([])
538
Bram Moolenaare0de1712020-12-02 17:36:54 +0100539 var d = getqflist({items: 0})
540 d->assert_equal({items: []})
Bram Moolenaar94738d82020-10-21 14:25:07 +0200541enddef
542
543def Test_getreg()
544 var lines = ['aaa', 'bbb', 'ccc']
545 setreg('a', lines)
546 getreg('a', true, true)->assert_equal(lines)
Bram Moolenaar418a29f2021-02-10 22:23:41 +0100547 assert_fails('getreg("ab")', 'E1162:')
Bram Moolenaar94738d82020-10-21 14:25:07 +0200548enddef
549
550def Test_getreg_return_type()
551 var s1: string = getreg('"')
552 var s2: string = getreg('"', 1)
553 var s3: list<string> = getreg('"', 1, 1)
554enddef
555
Bram Moolenaar418a29f2021-02-10 22:23:41 +0100556def Test_getreginfo()
557 var text = 'abc'
558 setreg('a', text)
559 getreginfo('a')->assert_equal({regcontents: [text], regtype: 'v', isunnamed: false})
560 assert_fails('getreginfo("ab")', 'E1162:')
561enddef
562
563def Test_getregtype()
564 var lines = ['aaa', 'bbb', 'ccc']
565 setreg('a', lines)
566 getregtype('a')->assert_equal('V')
567 assert_fails('getregtype("ab")', 'E1162:')
568enddef
569
Bram Moolenaar94738d82020-10-21 14:25:07 +0200570def Test_glob()
571 glob('runtest.vim', true, true, true)->assert_equal(['runtest.vim'])
572enddef
573
574def Test_globpath()
575 globpath('.', 'runtest.vim', true, true, true)->assert_equal(['./runtest.vim'])
576enddef
577
578def Test_has()
579 has('eval', true)->assert_equal(1)
580enddef
581
582def Test_hasmapto()
583 hasmapto('foobar', 'i', true)->assert_equal(0)
584 iabbrev foo foobar
585 hasmapto('foobar', 'i', true)->assert_equal(1)
586 iunabbrev foo
587enddef
588
589def Test_index()
590 index(['a', 'b', 'a', 'B'], 'b', 2, true)->assert_equal(3)
591enddef
592
Bram Moolenaar193f6202020-11-16 20:08:35 +0100593let s:number_one = 1
594let s:number_two = 2
595let s:string_keep = 'keep'
596
Bram Moolenaarca174532020-10-21 16:42:22 +0200597def Test_insert()
Bram Moolenaar94738d82020-10-21 14:25:07 +0200598 var l = insert([2, 1], 3)
599 var res = 0
600 for n in l
601 res += n
602 endfor
603 res->assert_equal(6)
Bram Moolenaarca174532020-10-21 16:42:22 +0200604
605 assert_equal([1, 2, 3], insert([2, 3], 1))
Bram Moolenaar193f6202020-11-16 20:08:35 +0100606 assert_equal([1, 2, 3], insert([2, 3], s:number_one))
Bram Moolenaarca174532020-10-21 16:42:22 +0200607 assert_equal([1, 2, 3], insert([1, 2], 3, 2))
Bram Moolenaar193f6202020-11-16 20:08:35 +0100608 assert_equal([1, 2, 3], insert([1, 2], 3, s:number_two))
Bram Moolenaarca174532020-10-21 16:42:22 +0200609 assert_equal(['a', 'b', 'c'], insert(['b', 'c'], 'a'))
610 assert_equal(0z1234, insert(0z34, 0x12))
Bram Moolenaar193f6202020-11-16 20:08:35 +0100611
Bram Moolenaarca174532020-10-21 16:42:22 +0200612 CheckDefFailure(['insert([2, 3], "a")'], 'E1013: Argument 2: type mismatch, expected number but got string', 1)
613 CheckDefFailure(['insert([2, 3], 1, "x")'], 'E1013: Argument 3: type mismatch, expected number but got string', 1)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200614enddef
615
616def Test_keys_return_type()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100617 const var: list<string> = {a: 1, b: 2}->keys()
Bram Moolenaar94738d82020-10-21 14:25:07 +0200618 var->assert_equal(['a', 'b'])
619enddef
620
621def Test_list2str_str2list_utf8()
622 var s = "\u3042\u3044"
623 var l = [0x3042, 0x3044]
624 str2list(s, true)->assert_equal(l)
625 list2str(l, true)->assert_equal(s)
626enddef
627
628def SID(): number
629 return expand('<SID>')
630 ->matchstr('<SNR>\zs\d\+\ze_$')
631 ->str2nr()
632enddef
633
Bram Moolenaar70250fb2021-01-16 19:01:53 +0100634def Test_map_function_arg()
635 var lines =<< trim END
636 def MapOne(i: number, v: string): string
637 return i .. ':' .. v
638 enddef
639 var l = ['a', 'b', 'c']
640 map(l, MapOne)
641 assert_equal(['0:a', '1:b', '2:c'], l)
642 END
643 CheckDefAndScriptSuccess(lines)
644enddef
645
646def Test_map_item_type()
647 var lines =<< trim END
648 var l = ['a', 'b', 'c']
649 map(l, (k, v) => k .. '/' .. v )
650 assert_equal(['0/a', '1/b', '2/c'], l)
651 END
652 CheckDefAndScriptSuccess(lines)
653
654 lines =<< trim END
655 var l: list<number> = [0]
656 echo map(l, (_, v) => [])
657 END
658 CheckDefExecAndScriptFailure(lines, 'E1012: Type mismatch; expected number but got list<unknown>', 2)
659
660 lines =<< trim END
661 var l: list<number> = range(2)
662 echo map(l, (_, v) => [])
663 END
664 CheckDefExecAndScriptFailure(lines, 'E1012: Type mismatch; expected number but got list<unknown>', 2)
665
666 lines =<< trim END
667 var d: dict<number> = {key: 0}
668 echo map(d, (_, v) => [])
669 END
670 CheckDefExecAndScriptFailure(lines, 'E1012: Type mismatch; expected number but got list<unknown>', 2)
671enddef
672
Bram Moolenaar94738d82020-10-21 14:25:07 +0200673def Test_maparg()
674 var lnum = str2nr(expand('<sflnum>'))
675 map foo bar
Bram Moolenaare0de1712020-12-02 17:36:54 +0100676 maparg('foo', '', false, true)->assert_equal({
Bram Moolenaar94738d82020-10-21 14:25:07 +0200677 lnum: lnum + 1,
678 script: 0,
679 mode: ' ',
680 silent: 0,
681 noremap: 0,
682 lhs: 'foo',
683 lhsraw: 'foo',
684 nowait: 0,
685 expr: 0,
686 sid: SID(),
687 rhs: 'bar',
688 buffer: 0})
689 unmap foo
690enddef
691
692def Test_mapcheck()
693 iabbrev foo foobar
694 mapcheck('foo', 'i', true)->assert_equal('foobar')
695 iunabbrev foo
696enddef
697
698def Test_maparg_mapset()
699 nnoremap <F3> :echo "hit F3"<CR>
700 var mapsave = maparg('<F3>', 'n', false, true)
701 mapset('n', false, mapsave)
702
703 nunmap <F3>
704enddef
705
Bram Moolenaar027c4ab2021-02-21 16:20:18 +0100706def Test_map_failure()
707 CheckFeature job
708
709 var lines =<< trim END
710 vim9script
711 writefile([], 'Xtmpfile')
712 silent e Xtmpfile
713 var d = {[bufnr('%')]: {a: 0}}
714 au BufReadPost * Func()
715 def Func()
716 if d->has_key('')
717 endif
718 eval d[expand('<abuf>')]->mapnew((_, v: dict<job>) => 0)
719 enddef
720 e
721 END
722 CheckScriptFailure(lines, 'E1013:')
723 au! BufReadPost
724 delete('Xtmpfile')
725enddef
726
Bram Moolenaar9ae37052021-01-22 22:31:10 +0100727def Test_max()
728 g:flag = true
729 var l1: list<number> = g:flag
730 ? [1, max([2, 3])]
731 : [4, 5]
732 assert_equal([1, 3], l1)
733
734 g:flag = false
735 var l2: list<number> = g:flag
736 ? [1, max([2, 3])]
737 : [4, 5]
738 assert_equal([4, 5], l2)
739enddef
740
741def Test_min()
742 g:flag = true
743 var l1: list<number> = g:flag
744 ? [1, min([2, 3])]
745 : [4, 5]
746 assert_equal([1, 2], l1)
747
748 g:flag = false
749 var l2: list<number> = g:flag
750 ? [1, min([2, 3])]
751 : [4, 5]
752 assert_equal([4, 5], l2)
753enddef
754
Bram Moolenaar94738d82020-10-21 14:25:07 +0200755def Test_nr2char()
756 nr2char(97, true)->assert_equal('a')
757enddef
758
759def Test_readdir()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100760 eval expand('sautest')->readdir((e) => e[0] !=# '.')
761 eval expand('sautest')->readdirex((e) => e.name[0] !=# '.')
Bram Moolenaar94738d82020-10-21 14:25:07 +0200762enddef
763
Bram Moolenaarc423ad72021-01-13 20:38:03 +0100764def Test_readblob()
765 var blob = 0z12341234
766 writefile(blob, 'Xreadblob')
767 var read: blob = readblob('Xreadblob')
768 assert_equal(blob, read)
769
770 var lines =<< trim END
771 var read: list<string> = readblob('Xreadblob')
772 END
773 CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected list<string> but got blob', 1)
774 delete('Xreadblob')
775enddef
776
777def Test_readfile()
778 var text = ['aaa', 'bbb', 'ccc']
779 writefile(text, 'Xreadfile')
780 var read: list<string> = readfile('Xreadfile')
781 assert_equal(text, read)
782
783 var lines =<< trim END
784 var read: dict<string> = readfile('Xreadfile')
785 END
786 CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected dict<string> but got list<string>', 1)
787 delete('Xreadfile')
788enddef
789
Bram Moolenaar94738d82020-10-21 14:25:07 +0200790def Test_remove_return_type()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100791 var l = remove({one: [1, 2], two: [3, 4]}, 'one')
Bram Moolenaar94738d82020-10-21 14:25:07 +0200792 var res = 0
793 for n in l
794 res += n
795 endfor
796 res->assert_equal(3)
797enddef
798
799def Test_reverse_return_type()
800 var l = reverse([1, 2, 3])
801 var res = 0
802 for n in l
803 res += n
804 endfor
805 res->assert_equal(6)
806enddef
807
808def Test_search()
809 new
810 setline(1, ['foo', 'bar'])
811 var val = 0
812 # skip expr returns boolean
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100813 search('bar', 'W', 0, 0, () => val == 1)->assert_equal(2)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200814 :1
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100815 search('bar', 'W', 0, 0, () => val == 0)->assert_equal(0)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200816 # skip expr returns number, only 0 and 1 are accepted
817 :1
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100818 search('bar', 'W', 0, 0, () => 0)->assert_equal(2)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200819 :1
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100820 search('bar', 'W', 0, 0, () => 1)->assert_equal(0)
821 assert_fails("search('bar', '', 0, 0, () => -1)", 'E1023:')
822 assert_fails("search('bar', '', 0, 0, () => -1)", 'E1023:')
Bram Moolenaar94738d82020-10-21 14:25:07 +0200823enddef
824
825def Test_searchcount()
826 new
827 setline(1, "foo bar")
828 :/foo
Bram Moolenaare0de1712020-12-02 17:36:54 +0100829 searchcount({recompute: true})
830 ->assert_equal({
Bram Moolenaar94738d82020-10-21 14:25:07 +0200831 exact_match: 1,
832 current: 1,
833 total: 1,
834 maxcount: 99,
835 incomplete: 0})
836 bwipe!
837enddef
838
Bram Moolenaar34453202021-01-31 13:08:38 +0100839def Test_set_get_bufline()
840 # similar to Test_setbufline_getbufline()
841 var lines =<< trim END
842 new
843 var b = bufnr('%')
844 hide
845 assert_equal(0, setbufline(b, 1, ['foo', 'bar']))
846 assert_equal(['foo'], getbufline(b, 1))
847 assert_equal(['bar'], getbufline(b, '$'))
848 assert_equal(['foo', 'bar'], getbufline(b, 1, 2))
849 exe "bd!" b
850 assert_equal([], getbufline(b, 1, 2))
851
852 split Xtest
853 setline(1, ['a', 'b', 'c'])
854 b = bufnr('%')
855 wincmd w
856
857 assert_equal(1, setbufline(b, 5, 'x'))
858 assert_equal(1, setbufline(b, 5, ['x']))
859 assert_equal(1, setbufline(b, 5, []))
860 assert_equal(1, setbufline(b, 5, test_null_list()))
861
862 assert_equal(1, 'x'->setbufline(bufnr('$') + 1, 1))
863 assert_equal(1, ['x']->setbufline(bufnr('$') + 1, 1))
864 assert_equal(1, []->setbufline(bufnr('$') + 1, 1))
865 assert_equal(1, test_null_list()->setbufline(bufnr('$') + 1, 1))
866
867 assert_equal(['a', 'b', 'c'], getbufline(b, 1, '$'))
868
869 assert_equal(0, setbufline(b, 4, ['d', 'e']))
870 assert_equal(['c'], b->getbufline(3))
871 assert_equal(['d'], getbufline(b, 4))
872 assert_equal(['e'], getbufline(b, 5))
873 assert_equal([], getbufline(b, 6))
874 assert_equal([], getbufline(b, 2, 1))
875
Bram Moolenaar00385112021-02-07 14:31:06 +0100876 if has('job')
877 setbufline(b, 2, [function('eval'), {key: 123}, test_null_job()])
878 assert_equal(["function('eval')",
879 "{'key': 123}",
880 "no process"],
881 getbufline(b, 2, 4))
882 endif
Bram Moolenaar34453202021-01-31 13:08:38 +0100883
884 exe 'bwipe! ' .. b
885 END
886 CheckDefAndScriptSuccess(lines)
887enddef
888
Bram Moolenaar94738d82020-10-21 14:25:07 +0200889def Test_searchdecl()
890 searchdecl('blah', true, true)->assert_equal(1)
891enddef
892
893def Test_setbufvar()
894 setbufvar(bufnr('%'), '&syntax', 'vim')
895 &syntax->assert_equal('vim')
896 setbufvar(bufnr('%'), '&ts', 16)
897 &ts->assert_equal(16)
Bram Moolenaar31a201a2021-01-03 14:47:25 +0100898 setbufvar(bufnr('%'), '&ai', true)
899 &ai->assert_equal(true)
900 setbufvar(bufnr('%'), '&ft', 'filetype')
901 &ft->assert_equal('filetype')
902
Bram Moolenaar94738d82020-10-21 14:25:07 +0200903 settabwinvar(1, 1, '&syntax', 'vam')
904 &syntax->assert_equal('vam')
905 settabwinvar(1, 1, '&ts', 15)
906 &ts->assert_equal(15)
907 setlocal ts=8
Bram Moolenaarb0d81822021-01-03 15:55:10 +0100908 settabwinvar(1, 1, '&list', false)
909 &list->assert_equal(false)
Bram Moolenaar31a201a2021-01-03 14:47:25 +0100910 settabwinvar(1, 1, '&list', true)
911 &list->assert_equal(true)
912 setlocal list&
Bram Moolenaar94738d82020-10-21 14:25:07 +0200913
914 setbufvar('%', 'myvar', 123)
915 getbufvar('%', 'myvar')->assert_equal(123)
916enddef
917
918def Test_setloclist()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100919 var items = [{filename: '/tmp/file', lnum: 1, valid: true}]
920 var what = {items: items}
Bram Moolenaar94738d82020-10-21 14:25:07 +0200921 setqflist([], ' ', what)
922 setloclist(0, [], ' ', what)
923enddef
924
925def Test_setreg()
926 setreg('a', ['aaa', 'bbb', 'ccc'])
927 var reginfo = getreginfo('a')
928 setreg('a', reginfo)
929 getreginfo('a')->assert_equal(reginfo)
Bram Moolenaar418a29f2021-02-10 22:23:41 +0100930 assert_fails('setreg("ab", 0)', 'E1162:')
Bram Moolenaar94738d82020-10-21 14:25:07 +0200931enddef
932
Bram Moolenaar6601b622021-01-13 21:47:15 +0100933def Test_slice()
934 assert_equal('12345', slice('012345', 1))
935 assert_equal('123', slice('012345', 1, 4))
936 assert_equal('1234', slice('012345', 1, -1))
937 assert_equal('1', slice('012345', 1, -4))
938 assert_equal('', slice('012345', 1, -5))
939 assert_equal('', slice('012345', 1, -6))
940
941 assert_equal([1, 2, 3, 4, 5], slice(range(6), 1))
942 assert_equal([1, 2, 3], slice(range(6), 1, 4))
943 assert_equal([1, 2, 3, 4], slice(range(6), 1, -1))
944 assert_equal([1], slice(range(6), 1, -4))
945 assert_equal([], slice(range(6), 1, -5))
946 assert_equal([], slice(range(6), 1, -6))
947
948 assert_equal(0z1122334455, slice(0z001122334455, 1))
949 assert_equal(0z112233, slice(0z001122334455, 1, 4))
950 assert_equal(0z11223344, slice(0z001122334455, 1, -1))
951 assert_equal(0z11, slice(0z001122334455, 1, -4))
952 assert_equal(0z, slice(0z001122334455, 1, -5))
953 assert_equal(0z, slice(0z001122334455, 1, -6))
954enddef
955
Bram Moolenaar94738d82020-10-21 14:25:07 +0200956def Test_spellsuggest()
957 if !has('spell')
958 MissingFeature 'spell'
959 else
960 spellsuggest('marrch', 1, true)->assert_equal(['March'])
961 endif
962enddef
963
964def Test_sort_return_type()
965 var res: list<number>
966 res = [1, 2, 3]->sort()
967enddef
968
969def Test_sort_argument()
Bram Moolenaar08cf0c02020-12-05 21:47:06 +0100970 var lines =<< trim END
971 var res = ['b', 'a', 'c']->sort('i')
972 res->assert_equal(['a', 'b', 'c'])
973
974 def Compare(a: number, b: number): number
975 return a - b
976 enddef
977 var l = [3, 6, 7, 1, 8, 2, 4, 5]
978 sort(l, Compare)
979 assert_equal([1, 2, 3, 4, 5, 6, 7, 8], l)
980 END
981 CheckDefAndScriptSuccess(lines)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200982enddef
983
984def Test_split()
985 split(' aa bb ', '\W\+', true)->assert_equal(['', 'aa', 'bb', ''])
986enddef
987
Bram Moolenaar80ad3e22021-01-31 20:48:58 +0100988def Run_str2float()
989 if !has('float')
990 MissingFeature 'float'
991 endif
992 str2float("1.00")->assert_equal(1.00)
993 str2float("2e-2")->assert_equal(0.02)
994
995 CheckDefFailure(['echo str2float(123)'], 'E1013:')
996 CheckScriptFailure(['vim9script', 'echo str2float(123)'], 'E1024:')
997 endif
998enddef
999
Bram Moolenaar94738d82020-10-21 14:25:07 +02001000def Test_str2nr()
1001 str2nr("1'000'000", 10, true)->assert_equal(1000000)
Bram Moolenaarf2b26bc2021-01-30 23:05:11 +01001002
1003 CheckDefFailure(['echo str2nr(123)'], 'E1013:')
1004 CheckScriptFailure(['vim9script', 'echo str2nr(123)'], 'E1024:')
1005 CheckDefFailure(['echo str2nr("123", "x")'], 'E1013:')
1006 CheckScriptFailure(['vim9script', 'echo str2nr("123", "x")'], 'E1030:')
1007 CheckDefFailure(['echo str2nr("123", 10, "x")'], 'E1013:')
1008 CheckScriptFailure(['vim9script', 'echo str2nr("123", 10, "x")'], 'E1135:')
Bram Moolenaar94738d82020-10-21 14:25:07 +02001009enddef
1010
1011def Test_strchars()
1012 strchars("A\u20dd", true)->assert_equal(1)
1013enddef
1014
1015def Test_submatch()
1016 var pat = 'A\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)'
Bram Moolenaar75ab91f2021-01-10 22:42:50 +01001017 var Rep = () => range(10)->mapnew((_, v) => submatch(v, true))->string()
Bram Moolenaar94738d82020-10-21 14:25:07 +02001018 var actual = substitute('A123456789', pat, Rep, '')
1019 var expected = "[['A123456789'], ['1'], ['2'], ['3'], ['4'], ['5'], ['6'], ['7'], ['8'], ['9']]"
1020 actual->assert_equal(expected)
1021enddef
1022
1023def Test_synID()
1024 new
1025 setline(1, "text")
1026 synID(1, 1, true)->assert_equal(0)
1027 bwipe!
1028enddef
1029
1030def Test_term_gettty()
1031 if !has('terminal')
1032 MissingFeature 'terminal'
1033 else
1034 var buf = Run_shell_in_terminal({})
1035 term_gettty(buf, true)->assert_notequal('')
1036 StopShellInTerminal(buf)
1037 endif
1038enddef
1039
1040def Test_term_start()
1041 if !has('terminal')
1042 MissingFeature 'terminal'
1043 else
1044 botright new
1045 var winnr = winnr()
Bram Moolenaare0de1712020-12-02 17:36:54 +01001046 term_start(&shell, {curwin: true})
Bram Moolenaar94738d82020-10-21 14:25:07 +02001047 winnr()->assert_equal(winnr)
1048 bwipe!
1049 endif
1050enddef
1051
1052def Test_timer_paused()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001053 var id = timer_start(50, () => 0)
Bram Moolenaar94738d82020-10-21 14:25:07 +02001054 timer_pause(id, true)
1055 var info = timer_info(id)
1056 info[0]['paused']->assert_equal(1)
1057 timer_stop(id)
1058enddef
1059
Bram Moolenaar37487e12021-01-12 22:08:53 +01001060def Test_win_execute()
1061 assert_equal("\n" .. winnr(), win_execute(win_getid(), 'echo winnr()'))
1062 assert_equal('', win_execute(342343, 'echo winnr()'))
1063enddef
1064
Bram Moolenaar94738d82020-10-21 14:25:07 +02001065def Test_win_splitmove()
1066 split
Bram Moolenaare0de1712020-12-02 17:36:54 +01001067 win_splitmove(1, 2, {vertical: true, rightbelow: true})
Bram Moolenaar94738d82020-10-21 14:25:07 +02001068 close
1069enddef
1070
Bram Moolenaar285b15f2020-12-29 20:25:19 +01001071def Test_winrestcmd()
1072 split
1073 var cmd = winrestcmd()
1074 wincmd _
1075 exe cmd
1076 assert_equal(cmd, winrestcmd())
1077 close
1078enddef
1079
Bram Moolenaar43b69b32021-01-07 20:23:33 +01001080def Test_winsaveview()
1081 var view: dict<number> = winsaveview()
1082
1083 var lines =<< trim END
1084 var view: list<number> = winsaveview()
1085 END
1086 CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected list<number> but got dict<number>', 1)
1087enddef
1088
1089
Bram Moolenaar285b15f2020-12-29 20:25:19 +01001090
Bram Moolenaar94738d82020-10-21 14:25:07 +02001091
1092" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker