blob: 61d129ec235e6337974be133b2859f250a9941ed [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
118def Test_bufname()
119 split SomeFile
120 bufname('%')->assert_equal('SomeFile')
121 edit OtherFile
122 bufname('#')->assert_equal('SomeFile')
123 close
124enddef
125
126def Test_bufnr()
127 var buf = bufnr()
128 bufnr('%')->assert_equal(buf)
129
130 buf = bufnr('Xdummy', true)
131 buf->assert_notequal(-1)
132 exe 'bwipe! ' .. buf
133enddef
134
135def Test_bufwinid()
136 var origwin = win_getid()
137 below split SomeFile
138 var SomeFileID = win_getid()
139 below split OtherFile
140 below split SomeFile
141 bufwinid('SomeFile')->assert_equal(SomeFileID)
142
143 win_gotoid(origwin)
144 only
145 bwipe SomeFile
146 bwipe OtherFile
147enddef
148
149def Test_call_call()
150 var l = [3, 2, 1]
151 call('reverse', [l])
152 l->assert_equal([1, 2, 3])
153enddef
154
155def Test_char2nr()
156 char2nr('あ', true)->assert_equal(12354)
157enddef
158
159def Test_col()
160 new
161 setline(1, 'asdf')
162 col([1, '$'])->assert_equal(5)
163enddef
164
165def Test_copy_return_type()
166 var l = copy([1, 2, 3])
167 var res = 0
168 for n in l
169 res += n
170 endfor
171 res->assert_equal(6)
172
173 var dl = deepcopy([1, 2, 3])
174 res = 0
175 for n in dl
176 res += n
177 endfor
178 res->assert_equal(6)
179
180 dl = deepcopy([1, 2, 3], true)
181enddef
182
183def Test_count()
184 count('ABC ABC ABC', 'b', true)->assert_equal(3)
185 count('ABC ABC ABC', 'b', false)->assert_equal(0)
186enddef
187
Bram Moolenaar9a963372020-12-21 21:58:46 +0100188def Test_cursor()
189 new
190 setline(1, range(4))
191 cursor(2, 1)
192 assert_equal(2, getcurpos()[1])
193 cursor('$', 1)
194 assert_equal(4, getcurpos()[1])
195
196 var lines =<< trim END
197 cursor('2', 1)
198 END
199 CheckDefExecAndScriptFailure(lines, 'E475:')
200enddef
201
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100202def Test_executable()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100203 assert_false(executable(""))
204 assert_false(executable(test_null_string()))
205
206 CheckDefExecFailure(['echo executable(123)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100207 CheckDefExecFailure(['echo executable(true)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100208enddef
209
210def Test_exepath()
211 CheckDefExecFailure(['echo exepath(true)'], 'E928:')
212 CheckDefExecFailure(['echo exepath(v:null)'], 'E928:')
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100213 CheckDefExecFailure(['echo exepath("")'], 'E1142:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100214enddef
215
Bram Moolenaar94738d82020-10-21 14:25:07 +0200216def Test_expand()
217 split SomeFile
218 expand('%', true, true)->assert_equal(['SomeFile'])
219 close
220enddef
221
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100222def Test_extend_arg_types()
223 assert_equal([1, 2, 3], extend([1, 2], [3]))
224 assert_equal([3, 1, 2], extend([1, 2], [3], 0))
225 assert_equal([1, 3, 2], extend([1, 2], [3], 1))
Bram Moolenaar193f6202020-11-16 20:08:35 +0100226 assert_equal([1, 3, 2], extend([1, 2], [3], s:number_one))
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100227
Bram Moolenaare0de1712020-12-02 17:36:54 +0100228 assert_equal({a: 1, b: 2, c: 3}, extend({a: 1, b: 2}, {c: 3}))
229 assert_equal({a: 1, b: 4}, extend({a: 1, b: 2}, {b: 4}))
230 assert_equal({a: 1, b: 2}, extend({a: 1, b: 2}, {b: 4}, 'keep'))
231 assert_equal({a: 1, b: 2}, extend({a: 1, b: 2}, {b: 4}, s:string_keep))
Bram Moolenaar193f6202020-11-16 20:08:35 +0100232
233 var res: list<dict<any>>
Bram Moolenaar75ab91f2021-01-10 22:42:50 +0100234 extend(res, mapnew([1, 2], (_, v) => ({})))
Bram Moolenaar193f6202020-11-16 20:08:35 +0100235 assert_equal([{}, {}], res)
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100236
237 CheckDefFailure(['extend([1, 2], 3)'], 'E1013: Argument 2: type mismatch, expected list<number> but got number')
238 CheckDefFailure(['extend([1, 2], ["x"])'], 'E1013: Argument 2: type mismatch, expected list<number> but got list<string>')
239 CheckDefFailure(['extend([1, 2], [3], "x")'], 'E1013: Argument 3: type mismatch, expected number but got string')
240
Bram Moolenaare0de1712020-12-02 17:36:54 +0100241 CheckDefFailure(['extend({a: 1}, 42)'], 'E1013: Argument 2: type mismatch, expected dict<number> but got number')
242 CheckDefFailure(['extend({a: 1}, {b: "x"})'], 'E1013: Argument 2: type mismatch, expected dict<number> but got dict<string>')
243 CheckDefFailure(['extend({a: 1}, {b: 2}, 1)'], 'E1013: Argument 3: type mismatch, expected string but got number')
Bram Moolenaar351ead02021-01-16 16:07:01 +0100244
245 CheckDefFailure(['extend([1], ["b"])'], 'E1013: Argument 2: type mismatch, expected list<number> but got list<string>')
246 CheckDefExecFailure(['extend([1], ["b", 1])'], 'E1012: Type mismatch; expected list<number> but got list<any>')
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100247enddef
248
Bram Moolenaarb0e6b512021-01-12 20:23:40 +0100249def Test_extendnew()
250 assert_equal([1, 2, 'a'], extendnew([1, 2], ['a']))
251 assert_equal({one: 1, two: 'a'}, extendnew({one: 1}, {two: 'a'}))
252
253 CheckDefFailure(['extendnew({a: 1}, 42)'], 'E1013: Argument 2: type mismatch, expected dict<number> but got number')
254 CheckDefFailure(['extendnew({a: 1}, [42])'], 'E1013: Argument 2: type mismatch, expected dict<number> but got list<number>')
255 CheckDefFailure(['extendnew([1, 2], "x")'], 'E1013: Argument 2: type mismatch, expected list<number> but got string')
256 CheckDefFailure(['extendnew([1, 2], {x: 1})'], 'E1013: Argument 2: type mismatch, expected list<number> but got dict<number>')
257enddef
258
Bram Moolenaar94738d82020-10-21 14:25:07 +0200259def Test_extend_return_type()
260 var l = extend([1, 2], [3])
261 var res = 0
262 for n in l
263 res += n
264 endfor
265 res->assert_equal(6)
266enddef
267
Bram Moolenaaraa210a32021-01-02 15:41:03 +0100268func g:ExtendDict(d)
269 call extend(a:d, #{xx: 'x'})
270endfunc
271
272def Test_extend_dict_item_type()
273 var lines =<< trim END
274 var d: dict<number> = {a: 1}
275 extend(d, {b: 2})
276 END
277 CheckDefAndScriptSuccess(lines)
278
279 lines =<< trim END
280 var d: dict<number> = {a: 1}
281 extend(d, {b: 'x'})
282 END
283 CheckDefFailure(lines, 'E1013: Argument 2: type mismatch, expected dict<number> but got dict<string>', 2)
284 CheckScriptFailure(['vim9script'] + lines, 'E1012:', 3)
285
286 lines =<< trim END
287 var d: dict<number> = {a: 1}
288 g:ExtendDict(d)
289 END
290 CheckDefExecFailure(lines, 'E1012: Type mismatch; expected number but got string', 0)
291 CheckScriptFailure(['vim9script'] + lines, 'E1012:', 1)
292enddef
293
294func g:ExtendList(l)
295 call extend(a:l, ['x'])
296endfunc
297
298def Test_extend_list_item_type()
299 var lines =<< trim END
300 var l: list<number> = [1]
301 extend(l, [2])
302 END
303 CheckDefAndScriptSuccess(lines)
304
305 lines =<< trim END
306 var l: list<number> = [1]
307 extend(l, ['x'])
308 END
309 CheckDefFailure(lines, 'E1013: Argument 2: type mismatch, expected list<number> but got list<string>', 2)
310 CheckScriptFailure(['vim9script'] + lines, 'E1012:', 3)
311
312 lines =<< trim END
313 var l: list<number> = [1]
314 g:ExtendList(l)
315 END
316 CheckDefExecFailure(lines, 'E1012: Type mismatch; expected number but got string', 0)
317 CheckScriptFailure(['vim9script'] + lines, 'E1012:', 1)
318enddef
Bram Moolenaar94738d82020-10-21 14:25:07 +0200319
Bram Moolenaar64ed4d42021-01-12 21:22:31 +0100320def Test_job_info_return_type()
321 if has('job')
322 job_start(&shell)
323 var jobs = job_info()
Bram Moolenaara47e05f2021-01-12 21:49:00 +0100324 assert_equal('list<job>', typename(jobs))
325 assert_equal('dict<any>', typename(job_info(jobs[0])))
Bram Moolenaar64ed4d42021-01-12 21:22:31 +0100326 job_stop(jobs[0])
327 endif
328enddef
329
Bram Moolenaar94738d82020-10-21 14:25:07 +0200330def Wrong_dict_key_type(items: list<number>): list<number>
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100331 return filter(items, (_, val) => get({[val]: 1}, 'x'))
Bram Moolenaar94738d82020-10-21 14:25:07 +0200332enddef
333
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100334def Test_filereadable()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100335 assert_false(filereadable(""))
336 assert_false(filereadable(test_null_string()))
337
338 CheckDefExecFailure(['echo filereadable(123)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100339 CheckDefExecFailure(['echo filereadable(true)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100340enddef
341
342def Test_filewritable()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100343 assert_false(filewritable(""))
344 assert_false(filewritable(test_null_string()))
345
346 CheckDefExecFailure(['echo filewritable(123)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100347 CheckDefExecFailure(['echo filewritable(true)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100348enddef
349
350def Test_finddir()
351 CheckDefExecFailure(['echo finddir(true)'], 'E928:')
352 CheckDefExecFailure(['echo finddir(v:null)'], 'E928:')
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100353 CheckDefExecFailure(['echo finddir("")'], 'E1142:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100354enddef
355
356def Test_findfile()
357 CheckDefExecFailure(['echo findfile(true)'], 'E928:')
358 CheckDefExecFailure(['echo findfile(v:null)'], 'E928:')
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100359 CheckDefExecFailure(['echo findfile("")'], 'E1142:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100360enddef
361
362def Test_fnamemodify()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100363 CheckDefSuccess(['echo fnamemodify(test_null_string(), ":p")'])
364 CheckDefSuccess(['echo fnamemodify("", ":p")'])
365 CheckDefSuccess(['echo fnamemodify("file", test_null_string())'])
366 CheckDefSuccess(['echo fnamemodify("file", "")'])
367
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100368 CheckDefExecFailure(['echo fnamemodify(true, ":p")'], 'E928:')
369 CheckDefExecFailure(['echo fnamemodify(v:null, ":p")'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100370 CheckDefExecFailure(['echo fnamemodify("file", true)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100371enddef
372
Bram Moolenaar94738d82020-10-21 14:25:07 +0200373def Test_filter_wrong_dict_key_type()
374 assert_fails('Wrong_dict_key_type([1, 2, 3])', 'E1012:')
375enddef
376
377def Test_filter_return_type()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100378 var l = filter([1, 2, 3], () => 1)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200379 var res = 0
380 for n in l
381 res += n
382 endfor
383 res->assert_equal(6)
384enddef
385
Bram Moolenaarfc0e8f52020-12-25 20:24:51 +0100386def Test_filter_missing_argument()
387 var dict = {aa: [1], ab: [2], ac: [3], de: [4]}
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100388 var res = dict->filter((k) => k =~ 'a' && k !~ 'b')
Bram Moolenaarfc0e8f52020-12-25 20:24:51 +0100389 res->assert_equal({aa: [1], ac: [3]})
390enddef
Bram Moolenaar94738d82020-10-21 14:25:07 +0200391
392def Test_garbagecollect()
393 garbagecollect(true)
394enddef
395
396def Test_getbufinfo()
397 var bufinfo = getbufinfo(bufnr())
398 getbufinfo('%')->assert_equal(bufinfo)
399
400 edit Xtestfile1
401 hide edit Xtestfile2
402 hide enew
Bram Moolenaare0de1712020-12-02 17:36:54 +0100403 getbufinfo({bufloaded: true, buflisted: true, bufmodified: false})
Bram Moolenaar94738d82020-10-21 14:25:07 +0200404 ->len()->assert_equal(3)
405 bwipe Xtestfile1 Xtestfile2
406enddef
407
408def Test_getbufline()
409 e SomeFile
410 var buf = bufnr()
411 e #
412 var lines = ['aaa', 'bbb', 'ccc']
413 setbufline(buf, 1, lines)
414 getbufline('#', 1, '$')->assert_equal(lines)
Bram Moolenaare6e70a12020-10-22 18:23:38 +0200415 getbufline(-1, '$', '$')->assert_equal([])
416 getbufline(-1, 1, '$')->assert_equal([])
Bram Moolenaar94738d82020-10-21 14:25:07 +0200417
418 bwipe!
419enddef
420
421def Test_getchangelist()
422 new
423 setline(1, 'some text')
424 var changelist = bufnr()->getchangelist()
425 getchangelist('%')->assert_equal(changelist)
426 bwipe!
427enddef
428
429def Test_getchar()
430 while getchar(0)
431 endwhile
432 getchar(true)->assert_equal(0)
433enddef
434
435def Test_getcompletion()
436 set wildignore=*.vim,*~
437 var l = getcompletion('run', 'file', true)
438 l->assert_equal([])
439 set wildignore&
440enddef
441
442def Test_getloclist_return_type()
443 var l = getloclist(1)
444 l->assert_equal([])
445
Bram Moolenaare0de1712020-12-02 17:36:54 +0100446 var d = getloclist(1, {items: 0})
447 d->assert_equal({items: []})
Bram Moolenaar94738d82020-10-21 14:25:07 +0200448enddef
449
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100450def Test_getfperm()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100451 assert_equal('', getfperm(""))
452 assert_equal('', getfperm(test_null_string()))
453
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100454 CheckDefExecFailure(['echo getfperm(true)'], 'E928:')
455 CheckDefExecFailure(['echo getfperm(v:null)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100456enddef
457
458def Test_getfsize()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100459 assert_equal(-1, getfsize(""))
460 assert_equal(-1, getfsize(test_null_string()))
461
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100462 CheckDefExecFailure(['echo getfsize(true)'], 'E928:')
463 CheckDefExecFailure(['echo getfsize(v:null)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100464enddef
465
466def Test_getftime()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100467 assert_equal(-1, getftime(""))
468 assert_equal(-1, getftime(test_null_string()))
469
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100470 CheckDefExecFailure(['echo getftime(true)'], 'E928:')
471 CheckDefExecFailure(['echo getftime(v:null)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100472enddef
473
474def Test_getftype()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100475 assert_equal('', getftype(""))
476 assert_equal('', getftype(test_null_string()))
477
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100478 CheckDefExecFailure(['echo getftype(true)'], 'E928:')
479 CheckDefExecFailure(['echo getftype(v:null)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100480enddef
481
Bram Moolenaar94738d82020-10-21 14:25:07 +0200482def Test_getqflist_return_type()
483 var l = getqflist()
484 l->assert_equal([])
485
Bram Moolenaare0de1712020-12-02 17:36:54 +0100486 var d = getqflist({items: 0})
487 d->assert_equal({items: []})
Bram Moolenaar94738d82020-10-21 14:25:07 +0200488enddef
489
490def Test_getreg()
491 var lines = ['aaa', 'bbb', 'ccc']
492 setreg('a', lines)
493 getreg('a', true, true)->assert_equal(lines)
494enddef
495
496def Test_getreg_return_type()
497 var s1: string = getreg('"')
498 var s2: string = getreg('"', 1)
499 var s3: list<string> = getreg('"', 1, 1)
500enddef
501
502def Test_glob()
503 glob('runtest.vim', true, true, true)->assert_equal(['runtest.vim'])
504enddef
505
506def Test_globpath()
507 globpath('.', 'runtest.vim', true, true, true)->assert_equal(['./runtest.vim'])
508enddef
509
510def Test_has()
511 has('eval', true)->assert_equal(1)
512enddef
513
514def Test_hasmapto()
515 hasmapto('foobar', 'i', true)->assert_equal(0)
516 iabbrev foo foobar
517 hasmapto('foobar', 'i', true)->assert_equal(1)
518 iunabbrev foo
519enddef
520
521def Test_index()
522 index(['a', 'b', 'a', 'B'], 'b', 2, true)->assert_equal(3)
523enddef
524
Bram Moolenaar193f6202020-11-16 20:08:35 +0100525let s:number_one = 1
526let s:number_two = 2
527let s:string_keep = 'keep'
528
Bram Moolenaarca174532020-10-21 16:42:22 +0200529def Test_insert()
Bram Moolenaar94738d82020-10-21 14:25:07 +0200530 var l = insert([2, 1], 3)
531 var res = 0
532 for n in l
533 res += n
534 endfor
535 res->assert_equal(6)
Bram Moolenaarca174532020-10-21 16:42:22 +0200536
537 assert_equal([1, 2, 3], insert([2, 3], 1))
Bram Moolenaar193f6202020-11-16 20:08:35 +0100538 assert_equal([1, 2, 3], insert([2, 3], s:number_one))
Bram Moolenaarca174532020-10-21 16:42:22 +0200539 assert_equal([1, 2, 3], insert([1, 2], 3, 2))
Bram Moolenaar193f6202020-11-16 20:08:35 +0100540 assert_equal([1, 2, 3], insert([1, 2], 3, s:number_two))
Bram Moolenaarca174532020-10-21 16:42:22 +0200541 assert_equal(['a', 'b', 'c'], insert(['b', 'c'], 'a'))
542 assert_equal(0z1234, insert(0z34, 0x12))
Bram Moolenaar193f6202020-11-16 20:08:35 +0100543
Bram Moolenaarca174532020-10-21 16:42:22 +0200544 CheckDefFailure(['insert([2, 3], "a")'], 'E1013: Argument 2: type mismatch, expected number but got string', 1)
545 CheckDefFailure(['insert([2, 3], 1, "x")'], 'E1013: Argument 3: type mismatch, expected number but got string', 1)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200546enddef
547
548def Test_keys_return_type()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100549 const var: list<string> = {a: 1, b: 2}->keys()
Bram Moolenaar94738d82020-10-21 14:25:07 +0200550 var->assert_equal(['a', 'b'])
551enddef
552
553def Test_list2str_str2list_utf8()
554 var s = "\u3042\u3044"
555 var l = [0x3042, 0x3044]
556 str2list(s, true)->assert_equal(l)
557 list2str(l, true)->assert_equal(s)
558enddef
559
560def SID(): number
561 return expand('<SID>')
562 ->matchstr('<SNR>\zs\d\+\ze_$')
563 ->str2nr()
564enddef
565
Bram Moolenaar70250fb2021-01-16 19:01:53 +0100566def Test_map_function_arg()
567 var lines =<< trim END
568 def MapOne(i: number, v: string): string
569 return i .. ':' .. v
570 enddef
571 var l = ['a', 'b', 'c']
572 map(l, MapOne)
573 assert_equal(['0:a', '1:b', '2:c'], l)
574 END
575 CheckDefAndScriptSuccess(lines)
576enddef
577
578def Test_map_item_type()
579 var lines =<< trim END
580 var l = ['a', 'b', 'c']
581 map(l, (k, v) => k .. '/' .. v )
582 assert_equal(['0/a', '1/b', '2/c'], l)
583 END
584 CheckDefAndScriptSuccess(lines)
585
586 lines =<< trim END
587 var l: list<number> = [0]
588 echo map(l, (_, v) => [])
589 END
590 CheckDefExecAndScriptFailure(lines, 'E1012: Type mismatch; expected number but got list<unknown>', 2)
591
592 lines =<< trim END
593 var l: list<number> = range(2)
594 echo map(l, (_, v) => [])
595 END
596 CheckDefExecAndScriptFailure(lines, 'E1012: Type mismatch; expected number but got list<unknown>', 2)
597
598 lines =<< trim END
599 var d: dict<number> = {key: 0}
600 echo map(d, (_, v) => [])
601 END
602 CheckDefExecAndScriptFailure(lines, 'E1012: Type mismatch; expected number but got list<unknown>', 2)
603enddef
604
Bram Moolenaar94738d82020-10-21 14:25:07 +0200605def Test_maparg()
606 var lnum = str2nr(expand('<sflnum>'))
607 map foo bar
Bram Moolenaare0de1712020-12-02 17:36:54 +0100608 maparg('foo', '', false, true)->assert_equal({
Bram Moolenaar94738d82020-10-21 14:25:07 +0200609 lnum: lnum + 1,
610 script: 0,
611 mode: ' ',
612 silent: 0,
613 noremap: 0,
614 lhs: 'foo',
615 lhsraw: 'foo',
616 nowait: 0,
617 expr: 0,
618 sid: SID(),
619 rhs: 'bar',
620 buffer: 0})
621 unmap foo
622enddef
623
624def Test_mapcheck()
625 iabbrev foo foobar
626 mapcheck('foo', 'i', true)->assert_equal('foobar')
627 iunabbrev foo
628enddef
629
630def Test_maparg_mapset()
631 nnoremap <F3> :echo "hit F3"<CR>
632 var mapsave = maparg('<F3>', 'n', false, true)
633 mapset('n', false, mapsave)
634
635 nunmap <F3>
636enddef
637
638def Test_nr2char()
639 nr2char(97, true)->assert_equal('a')
640enddef
641
642def Test_readdir()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100643 eval expand('sautest')->readdir((e) => e[0] !=# '.')
644 eval expand('sautest')->readdirex((e) => e.name[0] !=# '.')
Bram Moolenaar94738d82020-10-21 14:25:07 +0200645enddef
646
Bram Moolenaarc423ad72021-01-13 20:38:03 +0100647def Test_readblob()
648 var blob = 0z12341234
649 writefile(blob, 'Xreadblob')
650 var read: blob = readblob('Xreadblob')
651 assert_equal(blob, read)
652
653 var lines =<< trim END
654 var read: list<string> = readblob('Xreadblob')
655 END
656 CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected list<string> but got blob', 1)
657 delete('Xreadblob')
658enddef
659
660def Test_readfile()
661 var text = ['aaa', 'bbb', 'ccc']
662 writefile(text, 'Xreadfile')
663 var read: list<string> = readfile('Xreadfile')
664 assert_equal(text, read)
665
666 var lines =<< trim END
667 var read: dict<string> = readfile('Xreadfile')
668 END
669 CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected dict<string> but got list<string>', 1)
670 delete('Xreadfile')
671enddef
672
Bram Moolenaar94738d82020-10-21 14:25:07 +0200673def Test_remove_return_type()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100674 var l = remove({one: [1, 2], two: [3, 4]}, 'one')
Bram Moolenaar94738d82020-10-21 14:25:07 +0200675 var res = 0
676 for n in l
677 res += n
678 endfor
679 res->assert_equal(3)
680enddef
681
682def Test_reverse_return_type()
683 var l = reverse([1, 2, 3])
684 var res = 0
685 for n in l
686 res += n
687 endfor
688 res->assert_equal(6)
689enddef
690
691def Test_search()
692 new
693 setline(1, ['foo', 'bar'])
694 var val = 0
695 # skip expr returns boolean
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100696 search('bar', 'W', 0, 0, () => val == 1)->assert_equal(2)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200697 :1
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100698 search('bar', 'W', 0, 0, () => val == 0)->assert_equal(0)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200699 # skip expr returns number, only 0 and 1 are accepted
700 :1
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100701 search('bar', 'W', 0, 0, () => 0)->assert_equal(2)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200702 :1
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100703 search('bar', 'W', 0, 0, () => 1)->assert_equal(0)
704 assert_fails("search('bar', '', 0, 0, () => -1)", 'E1023:')
705 assert_fails("search('bar', '', 0, 0, () => -1)", 'E1023:')
Bram Moolenaar94738d82020-10-21 14:25:07 +0200706enddef
707
708def Test_searchcount()
709 new
710 setline(1, "foo bar")
711 :/foo
Bram Moolenaare0de1712020-12-02 17:36:54 +0100712 searchcount({recompute: true})
713 ->assert_equal({
Bram Moolenaar94738d82020-10-21 14:25:07 +0200714 exact_match: 1,
715 current: 1,
716 total: 1,
717 maxcount: 99,
718 incomplete: 0})
719 bwipe!
720enddef
721
722def Test_searchdecl()
723 searchdecl('blah', true, true)->assert_equal(1)
724enddef
725
726def Test_setbufvar()
727 setbufvar(bufnr('%'), '&syntax', 'vim')
728 &syntax->assert_equal('vim')
729 setbufvar(bufnr('%'), '&ts', 16)
730 &ts->assert_equal(16)
Bram Moolenaar31a201a2021-01-03 14:47:25 +0100731 setbufvar(bufnr('%'), '&ai', true)
732 &ai->assert_equal(true)
733 setbufvar(bufnr('%'), '&ft', 'filetype')
734 &ft->assert_equal('filetype')
735
Bram Moolenaar94738d82020-10-21 14:25:07 +0200736 settabwinvar(1, 1, '&syntax', 'vam')
737 &syntax->assert_equal('vam')
738 settabwinvar(1, 1, '&ts', 15)
739 &ts->assert_equal(15)
740 setlocal ts=8
Bram Moolenaarb0d81822021-01-03 15:55:10 +0100741 settabwinvar(1, 1, '&list', false)
742 &list->assert_equal(false)
Bram Moolenaar31a201a2021-01-03 14:47:25 +0100743 settabwinvar(1, 1, '&list', true)
744 &list->assert_equal(true)
745 setlocal list&
Bram Moolenaar94738d82020-10-21 14:25:07 +0200746
747 setbufvar('%', 'myvar', 123)
748 getbufvar('%', 'myvar')->assert_equal(123)
749enddef
750
751def Test_setloclist()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100752 var items = [{filename: '/tmp/file', lnum: 1, valid: true}]
753 var what = {items: items}
Bram Moolenaar94738d82020-10-21 14:25:07 +0200754 setqflist([], ' ', what)
755 setloclist(0, [], ' ', what)
756enddef
757
758def Test_setreg()
759 setreg('a', ['aaa', 'bbb', 'ccc'])
760 var reginfo = getreginfo('a')
761 setreg('a', reginfo)
762 getreginfo('a')->assert_equal(reginfo)
763enddef
764
Bram Moolenaar6601b622021-01-13 21:47:15 +0100765def Test_slice()
766 assert_equal('12345', slice('012345', 1))
767 assert_equal('123', slice('012345', 1, 4))
768 assert_equal('1234', slice('012345', 1, -1))
769 assert_equal('1', slice('012345', 1, -4))
770 assert_equal('', slice('012345', 1, -5))
771 assert_equal('', slice('012345', 1, -6))
772
773 assert_equal([1, 2, 3, 4, 5], slice(range(6), 1))
774 assert_equal([1, 2, 3], slice(range(6), 1, 4))
775 assert_equal([1, 2, 3, 4], slice(range(6), 1, -1))
776 assert_equal([1], slice(range(6), 1, -4))
777 assert_equal([], slice(range(6), 1, -5))
778 assert_equal([], slice(range(6), 1, -6))
779
780 assert_equal(0z1122334455, slice(0z001122334455, 1))
781 assert_equal(0z112233, slice(0z001122334455, 1, 4))
782 assert_equal(0z11223344, slice(0z001122334455, 1, -1))
783 assert_equal(0z11, slice(0z001122334455, 1, -4))
784 assert_equal(0z, slice(0z001122334455, 1, -5))
785 assert_equal(0z, slice(0z001122334455, 1, -6))
786enddef
787
Bram Moolenaar94738d82020-10-21 14:25:07 +0200788def Test_spellsuggest()
789 if !has('spell')
790 MissingFeature 'spell'
791 else
792 spellsuggest('marrch', 1, true)->assert_equal(['March'])
793 endif
794enddef
795
796def Test_sort_return_type()
797 var res: list<number>
798 res = [1, 2, 3]->sort()
799enddef
800
801def Test_sort_argument()
Bram Moolenaar08cf0c02020-12-05 21:47:06 +0100802 var lines =<< trim END
803 var res = ['b', 'a', 'c']->sort('i')
804 res->assert_equal(['a', 'b', 'c'])
805
806 def Compare(a: number, b: number): number
807 return a - b
808 enddef
809 var l = [3, 6, 7, 1, 8, 2, 4, 5]
810 sort(l, Compare)
811 assert_equal([1, 2, 3, 4, 5, 6, 7, 8], l)
812 END
813 CheckDefAndScriptSuccess(lines)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200814enddef
815
816def Test_split()
817 split(' aa bb ', '\W\+', true)->assert_equal(['', 'aa', 'bb', ''])
818enddef
819
820def Test_str2nr()
821 str2nr("1'000'000", 10, true)->assert_equal(1000000)
822enddef
823
824def Test_strchars()
825 strchars("A\u20dd", true)->assert_equal(1)
826enddef
827
828def Test_submatch()
829 var pat = 'A\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)'
Bram Moolenaar75ab91f2021-01-10 22:42:50 +0100830 var Rep = () => range(10)->mapnew((_, v) => submatch(v, true))->string()
Bram Moolenaar94738d82020-10-21 14:25:07 +0200831 var actual = substitute('A123456789', pat, Rep, '')
832 var expected = "[['A123456789'], ['1'], ['2'], ['3'], ['4'], ['5'], ['6'], ['7'], ['8'], ['9']]"
833 actual->assert_equal(expected)
834enddef
835
836def Test_synID()
837 new
838 setline(1, "text")
839 synID(1, 1, true)->assert_equal(0)
840 bwipe!
841enddef
842
843def Test_term_gettty()
844 if !has('terminal')
845 MissingFeature 'terminal'
846 else
847 var buf = Run_shell_in_terminal({})
848 term_gettty(buf, true)->assert_notequal('')
849 StopShellInTerminal(buf)
850 endif
851enddef
852
853def Test_term_start()
854 if !has('terminal')
855 MissingFeature 'terminal'
856 else
857 botright new
858 var winnr = winnr()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100859 term_start(&shell, {curwin: true})
Bram Moolenaar94738d82020-10-21 14:25:07 +0200860 winnr()->assert_equal(winnr)
861 bwipe!
862 endif
863enddef
864
865def Test_timer_paused()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100866 var id = timer_start(50, () => 0)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200867 timer_pause(id, true)
868 var info = timer_info(id)
869 info[0]['paused']->assert_equal(1)
870 timer_stop(id)
871enddef
872
Bram Moolenaar37487e12021-01-12 22:08:53 +0100873def Test_win_execute()
874 assert_equal("\n" .. winnr(), win_execute(win_getid(), 'echo winnr()'))
875 assert_equal('', win_execute(342343, 'echo winnr()'))
876enddef
877
Bram Moolenaar94738d82020-10-21 14:25:07 +0200878def Test_win_splitmove()
879 split
Bram Moolenaare0de1712020-12-02 17:36:54 +0100880 win_splitmove(1, 2, {vertical: true, rightbelow: true})
Bram Moolenaar94738d82020-10-21 14:25:07 +0200881 close
882enddef
883
Bram Moolenaar285b15f2020-12-29 20:25:19 +0100884def Test_winrestcmd()
885 split
886 var cmd = winrestcmd()
887 wincmd _
888 exe cmd
889 assert_equal(cmd, winrestcmd())
890 close
891enddef
892
Bram Moolenaar43b69b32021-01-07 20:23:33 +0100893def Test_winsaveview()
894 var view: dict<number> = winsaveview()
895
896 var lines =<< trim END
897 var view: list<number> = winsaveview()
898 END
899 CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected list<number> but got dict<number>', 1)
900enddef
901
902
Bram Moolenaar285b15f2020-12-29 20:25:19 +0100903
Bram Moolenaar94738d82020-10-21 14:25:07 +0200904
905" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker