blob: 50dc2a0735edf114773f1e88abe14177740337b5 [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()
243 assert_equal([1, 2, 3], extend([1, 2], [3]))
244 assert_equal([3, 1, 2], extend([1, 2], [3], 0))
245 assert_equal([1, 3, 2], extend([1, 2], [3], 1))
Bram Moolenaar193f6202020-11-16 20:08:35 +0100246 assert_equal([1, 3, 2], extend([1, 2], [3], s:number_one))
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100247
Bram Moolenaare0de1712020-12-02 17:36:54 +0100248 assert_equal({a: 1, b: 2, c: 3}, extend({a: 1, b: 2}, {c: 3}))
249 assert_equal({a: 1, b: 4}, extend({a: 1, b: 2}, {b: 4}))
250 assert_equal({a: 1, b: 2}, extend({a: 1, b: 2}, {b: 4}, 'keep'))
251 assert_equal({a: 1, b: 2}, extend({a: 1, b: 2}, {b: 4}, s:string_keep))
Bram Moolenaar193f6202020-11-16 20:08:35 +0100252
253 var res: list<dict<any>>
Bram Moolenaar75ab91f2021-01-10 22:42:50 +0100254 extend(res, mapnew([1, 2], (_, v) => ({})))
Bram Moolenaar193f6202020-11-16 20:08:35 +0100255 assert_equal([{}, {}], res)
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100256
257 CheckDefFailure(['extend([1, 2], 3)'], 'E1013: Argument 2: type mismatch, expected list<number> but got number')
258 CheckDefFailure(['extend([1, 2], ["x"])'], 'E1013: Argument 2: type mismatch, expected list<number> but got list<string>')
259 CheckDefFailure(['extend([1, 2], [3], "x")'], 'E1013: Argument 3: type mismatch, expected number but got string')
260
Bram Moolenaare0de1712020-12-02 17:36:54 +0100261 CheckDefFailure(['extend({a: 1}, 42)'], 'E1013: Argument 2: type mismatch, expected dict<number> but got number')
262 CheckDefFailure(['extend({a: 1}, {b: "x"})'], 'E1013: Argument 2: type mismatch, expected dict<number> but got dict<string>')
263 CheckDefFailure(['extend({a: 1}, {b: 2}, 1)'], 'E1013: Argument 3: type mismatch, expected string but got number')
Bram Moolenaar351ead02021-01-16 16:07:01 +0100264
265 CheckDefFailure(['extend([1], ["b"])'], 'E1013: Argument 2: type mismatch, expected list<number> but got list<string>')
266 CheckDefExecFailure(['extend([1], ["b", 1])'], 'E1012: Type mismatch; expected list<number> but got list<any>')
Bram Moolenaarfbcbffe2020-10-31 19:33:38 +0100267enddef
268
Bram Moolenaarb0e6b512021-01-12 20:23:40 +0100269def Test_extendnew()
270 assert_equal([1, 2, 'a'], extendnew([1, 2], ['a']))
271 assert_equal({one: 1, two: 'a'}, extendnew({one: 1}, {two: 'a'}))
272
273 CheckDefFailure(['extendnew({a: 1}, 42)'], 'E1013: Argument 2: type mismatch, expected dict<number> but got number')
274 CheckDefFailure(['extendnew({a: 1}, [42])'], 'E1013: Argument 2: type mismatch, expected dict<number> but got list<number>')
275 CheckDefFailure(['extendnew([1, 2], "x")'], 'E1013: Argument 2: type mismatch, expected list<number> but got string')
276 CheckDefFailure(['extendnew([1, 2], {x: 1})'], 'E1013: Argument 2: type mismatch, expected list<number> but got dict<number>')
277enddef
278
Bram Moolenaar94738d82020-10-21 14:25:07 +0200279def Test_extend_return_type()
280 var l = extend([1, 2], [3])
281 var res = 0
282 for n in l
283 res += n
284 endfor
285 res->assert_equal(6)
286enddef
287
Bram Moolenaaraa210a32021-01-02 15:41:03 +0100288func g:ExtendDict(d)
289 call extend(a:d, #{xx: 'x'})
290endfunc
291
292def Test_extend_dict_item_type()
293 var lines =<< trim END
294 var d: dict<number> = {a: 1}
295 extend(d, {b: 2})
296 END
297 CheckDefAndScriptSuccess(lines)
298
299 lines =<< trim END
300 var d: dict<number> = {a: 1}
301 extend(d, {b: 'x'})
302 END
303 CheckDefFailure(lines, 'E1013: Argument 2: type mismatch, expected dict<number> but got dict<string>', 2)
304 CheckScriptFailure(['vim9script'] + lines, 'E1012:', 3)
305
306 lines =<< trim END
307 var d: dict<number> = {a: 1}
308 g:ExtendDict(d)
309 END
310 CheckDefExecFailure(lines, 'E1012: Type mismatch; expected number but got string', 0)
311 CheckScriptFailure(['vim9script'] + lines, 'E1012:', 1)
312enddef
313
314func g:ExtendList(l)
315 call extend(a:l, ['x'])
316endfunc
317
318def Test_extend_list_item_type()
319 var lines =<< trim END
320 var l: list<number> = [1]
321 extend(l, [2])
322 END
323 CheckDefAndScriptSuccess(lines)
324
325 lines =<< trim END
326 var l: list<number> = [1]
327 extend(l, ['x'])
328 END
329 CheckDefFailure(lines, 'E1013: Argument 2: type mismatch, expected list<number> but got list<string>', 2)
330 CheckScriptFailure(['vim9script'] + lines, 'E1012:', 3)
331
332 lines =<< trim END
333 var l: list<number> = [1]
334 g:ExtendList(l)
335 END
336 CheckDefExecFailure(lines, 'E1012: Type mismatch; expected number but got string', 0)
337 CheckScriptFailure(['vim9script'] + lines, 'E1012:', 1)
338enddef
Bram Moolenaar94738d82020-10-21 14:25:07 +0200339
Bram Moolenaar64ed4d42021-01-12 21:22:31 +0100340def Test_job_info_return_type()
341 if has('job')
342 job_start(&shell)
343 var jobs = job_info()
Bram Moolenaara47e05f2021-01-12 21:49:00 +0100344 assert_equal('list<job>', typename(jobs))
345 assert_equal('dict<any>', typename(job_info(jobs[0])))
Bram Moolenaar64ed4d42021-01-12 21:22:31 +0100346 job_stop(jobs[0])
347 endif
348enddef
349
Bram Moolenaar94738d82020-10-21 14:25:07 +0200350def Wrong_dict_key_type(items: list<number>): list<number>
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100351 return filter(items, (_, val) => get({[val]: 1}, 'x'))
Bram Moolenaar94738d82020-10-21 14:25:07 +0200352enddef
353
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100354def Test_filereadable()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100355 assert_false(filereadable(""))
356 assert_false(filereadable(test_null_string()))
357
358 CheckDefExecFailure(['echo filereadable(123)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100359 CheckDefExecFailure(['echo filereadable(true)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100360enddef
361
362def Test_filewritable()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100363 assert_false(filewritable(""))
364 assert_false(filewritable(test_null_string()))
365
366 CheckDefExecFailure(['echo filewritable(123)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100367 CheckDefExecFailure(['echo filewritable(true)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100368enddef
369
370def Test_finddir()
371 CheckDefExecFailure(['echo finddir(true)'], 'E928:')
372 CheckDefExecFailure(['echo finddir(v:null)'], 'E928:')
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100373 CheckDefExecFailure(['echo finddir("")'], 'E1142:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100374enddef
375
376def Test_findfile()
377 CheckDefExecFailure(['echo findfile(true)'], 'E928:')
378 CheckDefExecFailure(['echo findfile(v:null)'], 'E928:')
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100379 CheckDefExecFailure(['echo findfile("")'], 'E1142:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100380enddef
381
382def Test_fnamemodify()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100383 CheckDefSuccess(['echo fnamemodify(test_null_string(), ":p")'])
384 CheckDefSuccess(['echo fnamemodify("", ":p")'])
385 CheckDefSuccess(['echo fnamemodify("file", test_null_string())'])
386 CheckDefSuccess(['echo fnamemodify("file", "")'])
387
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100388 CheckDefExecFailure(['echo fnamemodify(true, ":p")'], 'E928:')
389 CheckDefExecFailure(['echo fnamemodify(v:null, ":p")'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100390 CheckDefExecFailure(['echo fnamemodify("file", true)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100391enddef
392
Bram Moolenaar94738d82020-10-21 14:25:07 +0200393def Test_filter_wrong_dict_key_type()
394 assert_fails('Wrong_dict_key_type([1, 2, 3])', 'E1012:')
395enddef
396
397def Test_filter_return_type()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100398 var l = filter([1, 2, 3], () => 1)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200399 var res = 0
400 for n in l
401 res += n
402 endfor
403 res->assert_equal(6)
404enddef
405
Bram Moolenaarfc0e8f52020-12-25 20:24:51 +0100406def Test_filter_missing_argument()
407 var dict = {aa: [1], ab: [2], ac: [3], de: [4]}
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100408 var res = dict->filter((k) => k =~ 'a' && k !~ 'b')
Bram Moolenaarfc0e8f52020-12-25 20:24:51 +0100409 res->assert_equal({aa: [1], ac: [3]})
410enddef
Bram Moolenaar94738d82020-10-21 14:25:07 +0200411
412def Test_garbagecollect()
413 garbagecollect(true)
414enddef
415
416def Test_getbufinfo()
417 var bufinfo = getbufinfo(bufnr())
418 getbufinfo('%')->assert_equal(bufinfo)
419
420 edit Xtestfile1
421 hide edit Xtestfile2
422 hide enew
Bram Moolenaare0de1712020-12-02 17:36:54 +0100423 getbufinfo({bufloaded: true, buflisted: true, bufmodified: false})
Bram Moolenaar94738d82020-10-21 14:25:07 +0200424 ->len()->assert_equal(3)
425 bwipe Xtestfile1 Xtestfile2
426enddef
427
428def Test_getbufline()
429 e SomeFile
430 var buf = bufnr()
431 e #
432 var lines = ['aaa', 'bbb', 'ccc']
433 setbufline(buf, 1, lines)
434 getbufline('#', 1, '$')->assert_equal(lines)
Bram Moolenaare6e70a12020-10-22 18:23:38 +0200435 getbufline(-1, '$', '$')->assert_equal([])
436 getbufline(-1, 1, '$')->assert_equal([])
Bram Moolenaar94738d82020-10-21 14:25:07 +0200437
438 bwipe!
439enddef
440
441def Test_getchangelist()
442 new
443 setline(1, 'some text')
444 var changelist = bufnr()->getchangelist()
445 getchangelist('%')->assert_equal(changelist)
446 bwipe!
447enddef
448
449def Test_getchar()
450 while getchar(0)
451 endwhile
452 getchar(true)->assert_equal(0)
453enddef
454
455def Test_getcompletion()
456 set wildignore=*.vim,*~
457 var l = getcompletion('run', 'file', true)
458 l->assert_equal([])
459 set wildignore&
460enddef
461
462def Test_getloclist_return_type()
463 var l = getloclist(1)
464 l->assert_equal([])
465
Bram Moolenaare0de1712020-12-02 17:36:54 +0100466 var d = getloclist(1, {items: 0})
467 d->assert_equal({items: []})
Bram Moolenaar94738d82020-10-21 14:25:07 +0200468enddef
469
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100470def Test_getfperm()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100471 assert_equal('', getfperm(""))
472 assert_equal('', getfperm(test_null_string()))
473
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100474 CheckDefExecFailure(['echo getfperm(true)'], 'E928:')
475 CheckDefExecFailure(['echo getfperm(v:null)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100476enddef
477
478def Test_getfsize()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100479 assert_equal(-1, getfsize(""))
480 assert_equal(-1, getfsize(test_null_string()))
481
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100482 CheckDefExecFailure(['echo getfsize(true)'], 'E928:')
483 CheckDefExecFailure(['echo getfsize(v:null)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100484enddef
485
486def Test_getftime()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100487 assert_equal(-1, getftime(""))
488 assert_equal(-1, getftime(test_null_string()))
489
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100490 CheckDefExecFailure(['echo getftime(true)'], 'E928:')
491 CheckDefExecFailure(['echo getftime(v:null)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100492enddef
493
494def Test_getftype()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100495 assert_equal('', getftype(""))
496 assert_equal('', getftype(test_null_string()))
497
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100498 CheckDefExecFailure(['echo getftype(true)'], 'E928:')
499 CheckDefExecFailure(['echo getftype(v:null)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100500enddef
501
Bram Moolenaar94738d82020-10-21 14:25:07 +0200502def Test_getqflist_return_type()
503 var l = getqflist()
504 l->assert_equal([])
505
Bram Moolenaare0de1712020-12-02 17:36:54 +0100506 var d = getqflist({items: 0})
507 d->assert_equal({items: []})
Bram Moolenaar94738d82020-10-21 14:25:07 +0200508enddef
509
510def Test_getreg()
511 var lines = ['aaa', 'bbb', 'ccc']
512 setreg('a', lines)
513 getreg('a', true, true)->assert_equal(lines)
514enddef
515
516def Test_getreg_return_type()
517 var s1: string = getreg('"')
518 var s2: string = getreg('"', 1)
519 var s3: list<string> = getreg('"', 1, 1)
520enddef
521
522def Test_glob()
523 glob('runtest.vim', true, true, true)->assert_equal(['runtest.vim'])
524enddef
525
526def Test_globpath()
527 globpath('.', 'runtest.vim', true, true, true)->assert_equal(['./runtest.vim'])
528enddef
529
530def Test_has()
531 has('eval', true)->assert_equal(1)
532enddef
533
534def Test_hasmapto()
535 hasmapto('foobar', 'i', true)->assert_equal(0)
536 iabbrev foo foobar
537 hasmapto('foobar', 'i', true)->assert_equal(1)
538 iunabbrev foo
539enddef
540
541def Test_index()
542 index(['a', 'b', 'a', 'B'], 'b', 2, true)->assert_equal(3)
543enddef
544
Bram Moolenaar193f6202020-11-16 20:08:35 +0100545let s:number_one = 1
546let s:number_two = 2
547let s:string_keep = 'keep'
548
Bram Moolenaarca174532020-10-21 16:42:22 +0200549def Test_insert()
Bram Moolenaar94738d82020-10-21 14:25:07 +0200550 var l = insert([2, 1], 3)
551 var res = 0
552 for n in l
553 res += n
554 endfor
555 res->assert_equal(6)
Bram Moolenaarca174532020-10-21 16:42:22 +0200556
557 assert_equal([1, 2, 3], insert([2, 3], 1))
Bram Moolenaar193f6202020-11-16 20:08:35 +0100558 assert_equal([1, 2, 3], insert([2, 3], s:number_one))
Bram Moolenaarca174532020-10-21 16:42:22 +0200559 assert_equal([1, 2, 3], insert([1, 2], 3, 2))
Bram Moolenaar193f6202020-11-16 20:08:35 +0100560 assert_equal([1, 2, 3], insert([1, 2], 3, s:number_two))
Bram Moolenaarca174532020-10-21 16:42:22 +0200561 assert_equal(['a', 'b', 'c'], insert(['b', 'c'], 'a'))
562 assert_equal(0z1234, insert(0z34, 0x12))
Bram Moolenaar193f6202020-11-16 20:08:35 +0100563
Bram Moolenaarca174532020-10-21 16:42:22 +0200564 CheckDefFailure(['insert([2, 3], "a")'], 'E1013: Argument 2: type mismatch, expected number but got string', 1)
565 CheckDefFailure(['insert([2, 3], 1, "x")'], 'E1013: Argument 3: type mismatch, expected number but got string', 1)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200566enddef
567
568def Test_keys_return_type()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100569 const var: list<string> = {a: 1, b: 2}->keys()
Bram Moolenaar94738d82020-10-21 14:25:07 +0200570 var->assert_equal(['a', 'b'])
571enddef
572
573def Test_list2str_str2list_utf8()
574 var s = "\u3042\u3044"
575 var l = [0x3042, 0x3044]
576 str2list(s, true)->assert_equal(l)
577 list2str(l, true)->assert_equal(s)
578enddef
579
580def SID(): number
581 return expand('<SID>')
582 ->matchstr('<SNR>\zs\d\+\ze_$')
583 ->str2nr()
584enddef
585
Bram Moolenaar70250fb2021-01-16 19:01:53 +0100586def Test_map_function_arg()
587 var lines =<< trim END
588 def MapOne(i: number, v: string): string
589 return i .. ':' .. v
590 enddef
591 var l = ['a', 'b', 'c']
592 map(l, MapOne)
593 assert_equal(['0:a', '1:b', '2:c'], l)
594 END
595 CheckDefAndScriptSuccess(lines)
596enddef
597
598def Test_map_item_type()
599 var lines =<< trim END
600 var l = ['a', 'b', 'c']
601 map(l, (k, v) => k .. '/' .. v )
602 assert_equal(['0/a', '1/b', '2/c'], l)
603 END
604 CheckDefAndScriptSuccess(lines)
605
606 lines =<< trim END
607 var l: list<number> = [0]
608 echo map(l, (_, v) => [])
609 END
610 CheckDefExecAndScriptFailure(lines, 'E1012: Type mismatch; expected number but got list<unknown>', 2)
611
612 lines =<< trim END
613 var l: list<number> = range(2)
614 echo map(l, (_, v) => [])
615 END
616 CheckDefExecAndScriptFailure(lines, 'E1012: Type mismatch; expected number but got list<unknown>', 2)
617
618 lines =<< trim END
619 var d: dict<number> = {key: 0}
620 echo map(d, (_, v) => [])
621 END
622 CheckDefExecAndScriptFailure(lines, 'E1012: Type mismatch; expected number but got list<unknown>', 2)
623enddef
624
Bram Moolenaar94738d82020-10-21 14:25:07 +0200625def Test_maparg()
626 var lnum = str2nr(expand('<sflnum>'))
627 map foo bar
Bram Moolenaare0de1712020-12-02 17:36:54 +0100628 maparg('foo', '', false, true)->assert_equal({
Bram Moolenaar94738d82020-10-21 14:25:07 +0200629 lnum: lnum + 1,
630 script: 0,
631 mode: ' ',
632 silent: 0,
633 noremap: 0,
634 lhs: 'foo',
635 lhsraw: 'foo',
636 nowait: 0,
637 expr: 0,
638 sid: SID(),
639 rhs: 'bar',
640 buffer: 0})
641 unmap foo
642enddef
643
644def Test_mapcheck()
645 iabbrev foo foobar
646 mapcheck('foo', 'i', true)->assert_equal('foobar')
647 iunabbrev foo
648enddef
649
650def Test_maparg_mapset()
651 nnoremap <F3> :echo "hit F3"<CR>
652 var mapsave = maparg('<F3>', 'n', false, true)
653 mapset('n', false, mapsave)
654
655 nunmap <F3>
656enddef
657
658def Test_nr2char()
659 nr2char(97, true)->assert_equal('a')
660enddef
661
662def Test_readdir()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100663 eval expand('sautest')->readdir((e) => e[0] !=# '.')
664 eval expand('sautest')->readdirex((e) => e.name[0] !=# '.')
Bram Moolenaar94738d82020-10-21 14:25:07 +0200665enddef
666
Bram Moolenaarc423ad72021-01-13 20:38:03 +0100667def Test_readblob()
668 var blob = 0z12341234
669 writefile(blob, 'Xreadblob')
670 var read: blob = readblob('Xreadblob')
671 assert_equal(blob, read)
672
673 var lines =<< trim END
674 var read: list<string> = readblob('Xreadblob')
675 END
676 CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected list<string> but got blob', 1)
677 delete('Xreadblob')
678enddef
679
680def Test_readfile()
681 var text = ['aaa', 'bbb', 'ccc']
682 writefile(text, 'Xreadfile')
683 var read: list<string> = readfile('Xreadfile')
684 assert_equal(text, read)
685
686 var lines =<< trim END
687 var read: dict<string> = readfile('Xreadfile')
688 END
689 CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected dict<string> but got list<string>', 1)
690 delete('Xreadfile')
691enddef
692
Bram Moolenaar94738d82020-10-21 14:25:07 +0200693def Test_remove_return_type()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100694 var l = remove({one: [1, 2], two: [3, 4]}, 'one')
Bram Moolenaar94738d82020-10-21 14:25:07 +0200695 var res = 0
696 for n in l
697 res += n
698 endfor
699 res->assert_equal(3)
700enddef
701
702def Test_reverse_return_type()
703 var l = reverse([1, 2, 3])
704 var res = 0
705 for n in l
706 res += n
707 endfor
708 res->assert_equal(6)
709enddef
710
711def Test_search()
712 new
713 setline(1, ['foo', 'bar'])
714 var val = 0
715 # skip expr returns boolean
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100716 search('bar', 'W', 0, 0, () => val == 1)->assert_equal(2)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200717 :1
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100718 search('bar', 'W', 0, 0, () => val == 0)->assert_equal(0)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200719 # skip expr returns number, only 0 and 1 are accepted
720 :1
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100721 search('bar', 'W', 0, 0, () => 0)->assert_equal(2)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200722 :1
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100723 search('bar', 'W', 0, 0, () => 1)->assert_equal(0)
724 assert_fails("search('bar', '', 0, 0, () => -1)", 'E1023:')
725 assert_fails("search('bar', '', 0, 0, () => -1)", 'E1023:')
Bram Moolenaar94738d82020-10-21 14:25:07 +0200726enddef
727
728def Test_searchcount()
729 new
730 setline(1, "foo bar")
731 :/foo
Bram Moolenaare0de1712020-12-02 17:36:54 +0100732 searchcount({recompute: true})
733 ->assert_equal({
Bram Moolenaar94738d82020-10-21 14:25:07 +0200734 exact_match: 1,
735 current: 1,
736 total: 1,
737 maxcount: 99,
738 incomplete: 0})
739 bwipe!
740enddef
741
742def Test_searchdecl()
743 searchdecl('blah', true, true)->assert_equal(1)
744enddef
745
746def Test_setbufvar()
747 setbufvar(bufnr('%'), '&syntax', 'vim')
748 &syntax->assert_equal('vim')
749 setbufvar(bufnr('%'), '&ts', 16)
750 &ts->assert_equal(16)
Bram Moolenaar31a201a2021-01-03 14:47:25 +0100751 setbufvar(bufnr('%'), '&ai', true)
752 &ai->assert_equal(true)
753 setbufvar(bufnr('%'), '&ft', 'filetype')
754 &ft->assert_equal('filetype')
755
Bram Moolenaar94738d82020-10-21 14:25:07 +0200756 settabwinvar(1, 1, '&syntax', 'vam')
757 &syntax->assert_equal('vam')
758 settabwinvar(1, 1, '&ts', 15)
759 &ts->assert_equal(15)
760 setlocal ts=8
Bram Moolenaarb0d81822021-01-03 15:55:10 +0100761 settabwinvar(1, 1, '&list', false)
762 &list->assert_equal(false)
Bram Moolenaar31a201a2021-01-03 14:47:25 +0100763 settabwinvar(1, 1, '&list', true)
764 &list->assert_equal(true)
765 setlocal list&
Bram Moolenaar94738d82020-10-21 14:25:07 +0200766
767 setbufvar('%', 'myvar', 123)
768 getbufvar('%', 'myvar')->assert_equal(123)
769enddef
770
771def Test_setloclist()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100772 var items = [{filename: '/tmp/file', lnum: 1, valid: true}]
773 var what = {items: items}
Bram Moolenaar94738d82020-10-21 14:25:07 +0200774 setqflist([], ' ', what)
775 setloclist(0, [], ' ', what)
776enddef
777
778def Test_setreg()
779 setreg('a', ['aaa', 'bbb', 'ccc'])
780 var reginfo = getreginfo('a')
781 setreg('a', reginfo)
782 getreginfo('a')->assert_equal(reginfo)
783enddef
784
Bram Moolenaar6601b622021-01-13 21:47:15 +0100785def Test_slice()
786 assert_equal('12345', slice('012345', 1))
787 assert_equal('123', slice('012345', 1, 4))
788 assert_equal('1234', slice('012345', 1, -1))
789 assert_equal('1', slice('012345', 1, -4))
790 assert_equal('', slice('012345', 1, -5))
791 assert_equal('', slice('012345', 1, -6))
792
793 assert_equal([1, 2, 3, 4, 5], slice(range(6), 1))
794 assert_equal([1, 2, 3], slice(range(6), 1, 4))
795 assert_equal([1, 2, 3, 4], slice(range(6), 1, -1))
796 assert_equal([1], slice(range(6), 1, -4))
797 assert_equal([], slice(range(6), 1, -5))
798 assert_equal([], slice(range(6), 1, -6))
799
800 assert_equal(0z1122334455, slice(0z001122334455, 1))
801 assert_equal(0z112233, slice(0z001122334455, 1, 4))
802 assert_equal(0z11223344, slice(0z001122334455, 1, -1))
803 assert_equal(0z11, slice(0z001122334455, 1, -4))
804 assert_equal(0z, slice(0z001122334455, 1, -5))
805 assert_equal(0z, slice(0z001122334455, 1, -6))
806enddef
807
Bram Moolenaar94738d82020-10-21 14:25:07 +0200808def Test_spellsuggest()
809 if !has('spell')
810 MissingFeature 'spell'
811 else
812 spellsuggest('marrch', 1, true)->assert_equal(['March'])
813 endif
814enddef
815
816def Test_sort_return_type()
817 var res: list<number>
818 res = [1, 2, 3]->sort()
819enddef
820
821def Test_sort_argument()
Bram Moolenaar08cf0c02020-12-05 21:47:06 +0100822 var lines =<< trim END
823 var res = ['b', 'a', 'c']->sort('i')
824 res->assert_equal(['a', 'b', 'c'])
825
826 def Compare(a: number, b: number): number
827 return a - b
828 enddef
829 var l = [3, 6, 7, 1, 8, 2, 4, 5]
830 sort(l, Compare)
831 assert_equal([1, 2, 3, 4, 5, 6, 7, 8], l)
832 END
833 CheckDefAndScriptSuccess(lines)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200834enddef
835
836def Test_split()
837 split(' aa bb ', '\W\+', true)->assert_equal(['', 'aa', 'bb', ''])
838enddef
839
840def Test_str2nr()
841 str2nr("1'000'000", 10, true)->assert_equal(1000000)
842enddef
843
844def Test_strchars()
845 strchars("A\u20dd", true)->assert_equal(1)
846enddef
847
848def Test_submatch()
849 var pat = 'A\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)'
Bram Moolenaar75ab91f2021-01-10 22:42:50 +0100850 var Rep = () => range(10)->mapnew((_, v) => submatch(v, true))->string()
Bram Moolenaar94738d82020-10-21 14:25:07 +0200851 var actual = substitute('A123456789', pat, Rep, '')
852 var expected = "[['A123456789'], ['1'], ['2'], ['3'], ['4'], ['5'], ['6'], ['7'], ['8'], ['9']]"
853 actual->assert_equal(expected)
854enddef
855
856def Test_synID()
857 new
858 setline(1, "text")
859 synID(1, 1, true)->assert_equal(0)
860 bwipe!
861enddef
862
863def Test_term_gettty()
864 if !has('terminal')
865 MissingFeature 'terminal'
866 else
867 var buf = Run_shell_in_terminal({})
868 term_gettty(buf, true)->assert_notequal('')
869 StopShellInTerminal(buf)
870 endif
871enddef
872
873def Test_term_start()
874 if !has('terminal')
875 MissingFeature 'terminal'
876 else
877 botright new
878 var winnr = winnr()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100879 term_start(&shell, {curwin: true})
Bram Moolenaar94738d82020-10-21 14:25:07 +0200880 winnr()->assert_equal(winnr)
881 bwipe!
882 endif
883enddef
884
885def Test_timer_paused()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100886 var id = timer_start(50, () => 0)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200887 timer_pause(id, true)
888 var info = timer_info(id)
889 info[0]['paused']->assert_equal(1)
890 timer_stop(id)
891enddef
892
Bram Moolenaar37487e12021-01-12 22:08:53 +0100893def Test_win_execute()
894 assert_equal("\n" .. winnr(), win_execute(win_getid(), 'echo winnr()'))
895 assert_equal('', win_execute(342343, 'echo winnr()'))
896enddef
897
Bram Moolenaar94738d82020-10-21 14:25:07 +0200898def Test_win_splitmove()
899 split
Bram Moolenaare0de1712020-12-02 17:36:54 +0100900 win_splitmove(1, 2, {vertical: true, rightbelow: true})
Bram Moolenaar94738d82020-10-21 14:25:07 +0200901 close
902enddef
903
Bram Moolenaar285b15f2020-12-29 20:25:19 +0100904def Test_winrestcmd()
905 split
906 var cmd = winrestcmd()
907 wincmd _
908 exe cmd
909 assert_equal(cmd, winrestcmd())
910 close
911enddef
912
Bram Moolenaar43b69b32021-01-07 20:23:33 +0100913def Test_winsaveview()
914 var view: dict<number> = winsaveview()
915
916 var lines =<< trim END
917 var view: list<number> = winsaveview()
918 END
919 CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected list<number> but got dict<number>', 1)
920enddef
921
922
Bram Moolenaar285b15f2020-12-29 20:25:19 +0100923
Bram Moolenaar94738d82020-10-21 14:25:07 +0200924
925" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker