blob: 180070e0aa3fec4a1180e357c978378e30d83017 [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 Moolenaar08cf0c02020-12-05 21:47:06 +0100334def Test_map_function_arg()
335 var lines =<< trim END
336 def MapOne(i: number, v: string): string
337 return i .. ':' .. v
338 enddef
339 var l = ['a', 'b', 'c']
340 map(l, MapOne)
341 assert_equal(['0:a', '1:b', '2:c'], l)
342 END
343 CheckDefAndScriptSuccess(lines)
344enddef
345
Bram Moolenaar75ab91f2021-01-10 22:42:50 +0100346def Test_map_item_type()
347 var lines =<< trim END
348 var l = ['a', 'b', 'c']
349 map(l, (k, v) => k .. '/' .. v )
350 assert_equal(['0/a', '1/b', '2/c'], l)
351 END
352 CheckDefAndScriptSuccess(lines)
353enddef
354
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100355def Test_filereadable()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100356 assert_false(filereadable(""))
357 assert_false(filereadable(test_null_string()))
358
359 CheckDefExecFailure(['echo filereadable(123)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100360 CheckDefExecFailure(['echo filereadable(true)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100361enddef
362
363def Test_filewritable()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100364 assert_false(filewritable(""))
365 assert_false(filewritable(test_null_string()))
366
367 CheckDefExecFailure(['echo filewritable(123)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100368 CheckDefExecFailure(['echo filewritable(true)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100369enddef
370
371def Test_finddir()
372 CheckDefExecFailure(['echo finddir(true)'], 'E928:')
373 CheckDefExecFailure(['echo finddir(v:null)'], 'E928:')
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100374 CheckDefExecFailure(['echo finddir("")'], 'E1142:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100375enddef
376
377def Test_findfile()
378 CheckDefExecFailure(['echo findfile(true)'], 'E928:')
379 CheckDefExecFailure(['echo findfile(v:null)'], 'E928:')
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100380 CheckDefExecFailure(['echo findfile("")'], 'E1142:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100381enddef
382
383def Test_fnamemodify()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100384 CheckDefSuccess(['echo fnamemodify(test_null_string(), ":p")'])
385 CheckDefSuccess(['echo fnamemodify("", ":p")'])
386 CheckDefSuccess(['echo fnamemodify("file", test_null_string())'])
387 CheckDefSuccess(['echo fnamemodify("file", "")'])
388
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100389 CheckDefExecFailure(['echo fnamemodify(true, ":p")'], 'E928:')
390 CheckDefExecFailure(['echo fnamemodify(v:null, ":p")'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100391 CheckDefExecFailure(['echo fnamemodify("file", true)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100392enddef
393
Bram Moolenaar94738d82020-10-21 14:25:07 +0200394def Test_filter_wrong_dict_key_type()
395 assert_fails('Wrong_dict_key_type([1, 2, 3])', 'E1012:')
396enddef
397
398def Test_filter_return_type()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100399 var l = filter([1, 2, 3], () => 1)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200400 var res = 0
401 for n in l
402 res += n
403 endfor
404 res->assert_equal(6)
405enddef
406
Bram Moolenaarfc0e8f52020-12-25 20:24:51 +0100407def Test_filter_missing_argument()
408 var dict = {aa: [1], ab: [2], ac: [3], de: [4]}
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100409 var res = dict->filter((k) => k =~ 'a' && k !~ 'b')
Bram Moolenaarfc0e8f52020-12-25 20:24:51 +0100410 res->assert_equal({aa: [1], ac: [3]})
411enddef
Bram Moolenaar94738d82020-10-21 14:25:07 +0200412
413def Test_garbagecollect()
414 garbagecollect(true)
415enddef
416
417def Test_getbufinfo()
418 var bufinfo = getbufinfo(bufnr())
419 getbufinfo('%')->assert_equal(bufinfo)
420
421 edit Xtestfile1
422 hide edit Xtestfile2
423 hide enew
Bram Moolenaare0de1712020-12-02 17:36:54 +0100424 getbufinfo({bufloaded: true, buflisted: true, bufmodified: false})
Bram Moolenaar94738d82020-10-21 14:25:07 +0200425 ->len()->assert_equal(3)
426 bwipe Xtestfile1 Xtestfile2
427enddef
428
429def Test_getbufline()
430 e SomeFile
431 var buf = bufnr()
432 e #
433 var lines = ['aaa', 'bbb', 'ccc']
434 setbufline(buf, 1, lines)
435 getbufline('#', 1, '$')->assert_equal(lines)
Bram Moolenaare6e70a12020-10-22 18:23:38 +0200436 getbufline(-1, '$', '$')->assert_equal([])
437 getbufline(-1, 1, '$')->assert_equal([])
Bram Moolenaar94738d82020-10-21 14:25:07 +0200438
439 bwipe!
440enddef
441
442def Test_getchangelist()
443 new
444 setline(1, 'some text')
445 var changelist = bufnr()->getchangelist()
446 getchangelist('%')->assert_equal(changelist)
447 bwipe!
448enddef
449
450def Test_getchar()
451 while getchar(0)
452 endwhile
453 getchar(true)->assert_equal(0)
454enddef
455
456def Test_getcompletion()
457 set wildignore=*.vim,*~
458 var l = getcompletion('run', 'file', true)
459 l->assert_equal([])
460 set wildignore&
461enddef
462
463def Test_getloclist_return_type()
464 var l = getloclist(1)
465 l->assert_equal([])
466
Bram Moolenaare0de1712020-12-02 17:36:54 +0100467 var d = getloclist(1, {items: 0})
468 d->assert_equal({items: []})
Bram Moolenaar94738d82020-10-21 14:25:07 +0200469enddef
470
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100471def Test_getfperm()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100472 assert_equal('', getfperm(""))
473 assert_equal('', getfperm(test_null_string()))
474
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100475 CheckDefExecFailure(['echo getfperm(true)'], 'E928:')
476 CheckDefExecFailure(['echo getfperm(v:null)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100477enddef
478
479def Test_getfsize()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100480 assert_equal(-1, getfsize(""))
481 assert_equal(-1, getfsize(test_null_string()))
482
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100483 CheckDefExecFailure(['echo getfsize(true)'], 'E928:')
484 CheckDefExecFailure(['echo getfsize(v:null)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100485enddef
486
487def Test_getftime()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100488 assert_equal(-1, getftime(""))
489 assert_equal(-1, getftime(test_null_string()))
490
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100491 CheckDefExecFailure(['echo getftime(true)'], 'E928:')
492 CheckDefExecFailure(['echo getftime(v:null)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100493enddef
494
495def Test_getftype()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100496 assert_equal('', getftype(""))
497 assert_equal('', getftype(test_null_string()))
498
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100499 CheckDefExecFailure(['echo getftype(true)'], 'E928:')
500 CheckDefExecFailure(['echo getftype(v:null)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100501enddef
502
Bram Moolenaar94738d82020-10-21 14:25:07 +0200503def Test_getqflist_return_type()
504 var l = getqflist()
505 l->assert_equal([])
506
Bram Moolenaare0de1712020-12-02 17:36:54 +0100507 var d = getqflist({items: 0})
508 d->assert_equal({items: []})
Bram Moolenaar94738d82020-10-21 14:25:07 +0200509enddef
510
511def Test_getreg()
512 var lines = ['aaa', 'bbb', 'ccc']
513 setreg('a', lines)
514 getreg('a', true, true)->assert_equal(lines)
515enddef
516
517def Test_getreg_return_type()
518 var s1: string = getreg('"')
519 var s2: string = getreg('"', 1)
520 var s3: list<string> = getreg('"', 1, 1)
521enddef
522
523def Test_glob()
524 glob('runtest.vim', true, true, true)->assert_equal(['runtest.vim'])
525enddef
526
527def Test_globpath()
528 globpath('.', 'runtest.vim', true, true, true)->assert_equal(['./runtest.vim'])
529enddef
530
531def Test_has()
532 has('eval', true)->assert_equal(1)
533enddef
534
535def Test_hasmapto()
536 hasmapto('foobar', 'i', true)->assert_equal(0)
537 iabbrev foo foobar
538 hasmapto('foobar', 'i', true)->assert_equal(1)
539 iunabbrev foo
540enddef
541
542def Test_index()
543 index(['a', 'b', 'a', 'B'], 'b', 2, true)->assert_equal(3)
544enddef
545
Bram Moolenaar193f6202020-11-16 20:08:35 +0100546let s:number_one = 1
547let s:number_two = 2
548let s:string_keep = 'keep'
549
Bram Moolenaarca174532020-10-21 16:42:22 +0200550def Test_insert()
Bram Moolenaar94738d82020-10-21 14:25:07 +0200551 var l = insert([2, 1], 3)
552 var res = 0
553 for n in l
554 res += n
555 endfor
556 res->assert_equal(6)
Bram Moolenaarca174532020-10-21 16:42:22 +0200557
558 assert_equal([1, 2, 3], insert([2, 3], 1))
Bram Moolenaar193f6202020-11-16 20:08:35 +0100559 assert_equal([1, 2, 3], insert([2, 3], s:number_one))
Bram Moolenaarca174532020-10-21 16:42:22 +0200560 assert_equal([1, 2, 3], insert([1, 2], 3, 2))
Bram Moolenaar193f6202020-11-16 20:08:35 +0100561 assert_equal([1, 2, 3], insert([1, 2], 3, s:number_two))
Bram Moolenaarca174532020-10-21 16:42:22 +0200562 assert_equal(['a', 'b', 'c'], insert(['b', 'c'], 'a'))
563 assert_equal(0z1234, insert(0z34, 0x12))
Bram Moolenaar193f6202020-11-16 20:08:35 +0100564
Bram Moolenaarca174532020-10-21 16:42:22 +0200565 CheckDefFailure(['insert([2, 3], "a")'], 'E1013: Argument 2: type mismatch, expected number but got string', 1)
566 CheckDefFailure(['insert([2, 3], 1, "x")'], 'E1013: Argument 3: type mismatch, expected number but got string', 1)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200567enddef
568
569def Test_keys_return_type()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100570 const var: list<string> = {a: 1, b: 2}->keys()
Bram Moolenaar94738d82020-10-21 14:25:07 +0200571 var->assert_equal(['a', 'b'])
572enddef
573
574def Test_list2str_str2list_utf8()
575 var s = "\u3042\u3044"
576 var l = [0x3042, 0x3044]
577 str2list(s, true)->assert_equal(l)
578 list2str(l, true)->assert_equal(s)
579enddef
580
581def SID(): number
582 return expand('<SID>')
583 ->matchstr('<SNR>\zs\d\+\ze_$')
584 ->str2nr()
585enddef
586
587def Test_maparg()
588 var lnum = str2nr(expand('<sflnum>'))
589 map foo bar
Bram Moolenaare0de1712020-12-02 17:36:54 +0100590 maparg('foo', '', false, true)->assert_equal({
Bram Moolenaar94738d82020-10-21 14:25:07 +0200591 lnum: lnum + 1,
592 script: 0,
593 mode: ' ',
594 silent: 0,
595 noremap: 0,
596 lhs: 'foo',
597 lhsraw: 'foo',
598 nowait: 0,
599 expr: 0,
600 sid: SID(),
601 rhs: 'bar',
602 buffer: 0})
603 unmap foo
604enddef
605
606def Test_mapcheck()
607 iabbrev foo foobar
608 mapcheck('foo', 'i', true)->assert_equal('foobar')
609 iunabbrev foo
610enddef
611
612def Test_maparg_mapset()
613 nnoremap <F3> :echo "hit F3"<CR>
614 var mapsave = maparg('<F3>', 'n', false, true)
615 mapset('n', false, mapsave)
616
617 nunmap <F3>
618enddef
619
620def Test_nr2char()
621 nr2char(97, true)->assert_equal('a')
622enddef
623
624def Test_readdir()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100625 eval expand('sautest')->readdir((e) => e[0] !=# '.')
626 eval expand('sautest')->readdirex((e) => e.name[0] !=# '.')
Bram Moolenaar94738d82020-10-21 14:25:07 +0200627enddef
628
Bram Moolenaarc423ad72021-01-13 20:38:03 +0100629def Test_readblob()
630 var blob = 0z12341234
631 writefile(blob, 'Xreadblob')
632 var read: blob = readblob('Xreadblob')
633 assert_equal(blob, read)
634
635 var lines =<< trim END
636 var read: list<string> = readblob('Xreadblob')
637 END
638 CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected list<string> but got blob', 1)
639 delete('Xreadblob')
640enddef
641
642def Test_readfile()
643 var text = ['aaa', 'bbb', 'ccc']
644 writefile(text, 'Xreadfile')
645 var read: list<string> = readfile('Xreadfile')
646 assert_equal(text, read)
647
648 var lines =<< trim END
649 var read: dict<string> = readfile('Xreadfile')
650 END
651 CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected dict<string> but got list<string>', 1)
652 delete('Xreadfile')
653enddef
654
Bram Moolenaar94738d82020-10-21 14:25:07 +0200655def Test_remove_return_type()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100656 var l = remove({one: [1, 2], two: [3, 4]}, 'one')
Bram Moolenaar94738d82020-10-21 14:25:07 +0200657 var res = 0
658 for n in l
659 res += n
660 endfor
661 res->assert_equal(3)
662enddef
663
664def Test_reverse_return_type()
665 var l = reverse([1, 2, 3])
666 var res = 0
667 for n in l
668 res += n
669 endfor
670 res->assert_equal(6)
671enddef
672
673def Test_search()
674 new
675 setline(1, ['foo', 'bar'])
676 var val = 0
677 # skip expr returns boolean
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100678 search('bar', 'W', 0, 0, () => val == 1)->assert_equal(2)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200679 :1
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100680 search('bar', 'W', 0, 0, () => val == 0)->assert_equal(0)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200681 # skip expr returns number, only 0 and 1 are accepted
682 :1
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100683 search('bar', 'W', 0, 0, () => 0)->assert_equal(2)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200684 :1
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100685 search('bar', 'W', 0, 0, () => 1)->assert_equal(0)
686 assert_fails("search('bar', '', 0, 0, () => -1)", 'E1023:')
687 assert_fails("search('bar', '', 0, 0, () => -1)", 'E1023:')
Bram Moolenaar94738d82020-10-21 14:25:07 +0200688enddef
689
690def Test_searchcount()
691 new
692 setline(1, "foo bar")
693 :/foo
Bram Moolenaare0de1712020-12-02 17:36:54 +0100694 searchcount({recompute: true})
695 ->assert_equal({
Bram Moolenaar94738d82020-10-21 14:25:07 +0200696 exact_match: 1,
697 current: 1,
698 total: 1,
699 maxcount: 99,
700 incomplete: 0})
701 bwipe!
702enddef
703
704def Test_searchdecl()
705 searchdecl('blah', true, true)->assert_equal(1)
706enddef
707
708def Test_setbufvar()
709 setbufvar(bufnr('%'), '&syntax', 'vim')
710 &syntax->assert_equal('vim')
711 setbufvar(bufnr('%'), '&ts', 16)
712 &ts->assert_equal(16)
Bram Moolenaar31a201a2021-01-03 14:47:25 +0100713 setbufvar(bufnr('%'), '&ai', true)
714 &ai->assert_equal(true)
715 setbufvar(bufnr('%'), '&ft', 'filetype')
716 &ft->assert_equal('filetype')
717
Bram Moolenaar94738d82020-10-21 14:25:07 +0200718 settabwinvar(1, 1, '&syntax', 'vam')
719 &syntax->assert_equal('vam')
720 settabwinvar(1, 1, '&ts', 15)
721 &ts->assert_equal(15)
722 setlocal ts=8
Bram Moolenaarb0d81822021-01-03 15:55:10 +0100723 settabwinvar(1, 1, '&list', false)
724 &list->assert_equal(false)
Bram Moolenaar31a201a2021-01-03 14:47:25 +0100725 settabwinvar(1, 1, '&list', true)
726 &list->assert_equal(true)
727 setlocal list&
Bram Moolenaar94738d82020-10-21 14:25:07 +0200728
729 setbufvar('%', 'myvar', 123)
730 getbufvar('%', 'myvar')->assert_equal(123)
731enddef
732
733def Test_setloclist()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100734 var items = [{filename: '/tmp/file', lnum: 1, valid: true}]
735 var what = {items: items}
Bram Moolenaar94738d82020-10-21 14:25:07 +0200736 setqflist([], ' ', what)
737 setloclist(0, [], ' ', what)
738enddef
739
740def Test_setreg()
741 setreg('a', ['aaa', 'bbb', 'ccc'])
742 var reginfo = getreginfo('a')
743 setreg('a', reginfo)
744 getreginfo('a')->assert_equal(reginfo)
745enddef
746
Bram Moolenaar6601b622021-01-13 21:47:15 +0100747def Test_slice()
748 assert_equal('12345', slice('012345', 1))
749 assert_equal('123', slice('012345', 1, 4))
750 assert_equal('1234', slice('012345', 1, -1))
751 assert_equal('1', slice('012345', 1, -4))
752 assert_equal('', slice('012345', 1, -5))
753 assert_equal('', slice('012345', 1, -6))
754
755 assert_equal([1, 2, 3, 4, 5], slice(range(6), 1))
756 assert_equal([1, 2, 3], slice(range(6), 1, 4))
757 assert_equal([1, 2, 3, 4], slice(range(6), 1, -1))
758 assert_equal([1], slice(range(6), 1, -4))
759 assert_equal([], slice(range(6), 1, -5))
760 assert_equal([], slice(range(6), 1, -6))
761
762 assert_equal(0z1122334455, slice(0z001122334455, 1))
763 assert_equal(0z112233, slice(0z001122334455, 1, 4))
764 assert_equal(0z11223344, slice(0z001122334455, 1, -1))
765 assert_equal(0z11, slice(0z001122334455, 1, -4))
766 assert_equal(0z, slice(0z001122334455, 1, -5))
767 assert_equal(0z, slice(0z001122334455, 1, -6))
768enddef
769
Bram Moolenaar94738d82020-10-21 14:25:07 +0200770def Test_spellsuggest()
771 if !has('spell')
772 MissingFeature 'spell'
773 else
774 spellsuggest('marrch', 1, true)->assert_equal(['March'])
775 endif
776enddef
777
778def Test_sort_return_type()
779 var res: list<number>
780 res = [1, 2, 3]->sort()
781enddef
782
783def Test_sort_argument()
Bram Moolenaar08cf0c02020-12-05 21:47:06 +0100784 var lines =<< trim END
785 var res = ['b', 'a', 'c']->sort('i')
786 res->assert_equal(['a', 'b', 'c'])
787
788 def Compare(a: number, b: number): number
789 return a - b
790 enddef
791 var l = [3, 6, 7, 1, 8, 2, 4, 5]
792 sort(l, Compare)
793 assert_equal([1, 2, 3, 4, 5, 6, 7, 8], l)
794 END
795 CheckDefAndScriptSuccess(lines)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200796enddef
797
798def Test_split()
799 split(' aa bb ', '\W\+', true)->assert_equal(['', 'aa', 'bb', ''])
800enddef
801
802def Test_str2nr()
803 str2nr("1'000'000", 10, true)->assert_equal(1000000)
804enddef
805
806def Test_strchars()
807 strchars("A\u20dd", true)->assert_equal(1)
808enddef
809
810def Test_submatch()
811 var pat = 'A\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)'
Bram Moolenaar75ab91f2021-01-10 22:42:50 +0100812 var Rep = () => range(10)->mapnew((_, v) => submatch(v, true))->string()
Bram Moolenaar94738d82020-10-21 14:25:07 +0200813 var actual = substitute('A123456789', pat, Rep, '')
814 var expected = "[['A123456789'], ['1'], ['2'], ['3'], ['4'], ['5'], ['6'], ['7'], ['8'], ['9']]"
815 actual->assert_equal(expected)
816enddef
817
818def Test_synID()
819 new
820 setline(1, "text")
821 synID(1, 1, true)->assert_equal(0)
822 bwipe!
823enddef
824
825def Test_term_gettty()
826 if !has('terminal')
827 MissingFeature 'terminal'
828 else
829 var buf = Run_shell_in_terminal({})
830 term_gettty(buf, true)->assert_notequal('')
831 StopShellInTerminal(buf)
832 endif
833enddef
834
835def Test_term_start()
836 if !has('terminal')
837 MissingFeature 'terminal'
838 else
839 botright new
840 var winnr = winnr()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100841 term_start(&shell, {curwin: true})
Bram Moolenaar94738d82020-10-21 14:25:07 +0200842 winnr()->assert_equal(winnr)
843 bwipe!
844 endif
845enddef
846
847def Test_timer_paused()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100848 var id = timer_start(50, () => 0)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200849 timer_pause(id, true)
850 var info = timer_info(id)
851 info[0]['paused']->assert_equal(1)
852 timer_stop(id)
853enddef
854
Bram Moolenaar37487e12021-01-12 22:08:53 +0100855def Test_win_execute()
856 assert_equal("\n" .. winnr(), win_execute(win_getid(), 'echo winnr()'))
857 assert_equal('', win_execute(342343, 'echo winnr()'))
858enddef
859
Bram Moolenaar94738d82020-10-21 14:25:07 +0200860def Test_win_splitmove()
861 split
Bram Moolenaare0de1712020-12-02 17:36:54 +0100862 win_splitmove(1, 2, {vertical: true, rightbelow: true})
Bram Moolenaar94738d82020-10-21 14:25:07 +0200863 close
864enddef
865
Bram Moolenaar285b15f2020-12-29 20:25:19 +0100866def Test_winrestcmd()
867 split
868 var cmd = winrestcmd()
869 wincmd _
870 exe cmd
871 assert_equal(cmd, winrestcmd())
872 close
873enddef
874
Bram Moolenaar43b69b32021-01-07 20:23:33 +0100875def Test_winsaveview()
876 var view: dict<number> = winsaveview()
877
878 var lines =<< trim END
879 var view: list<number> = winsaveview()
880 END
881 CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected list<number> but got dict<number>', 1)
882enddef
883
884
Bram Moolenaar285b15f2020-12-29 20:25:19 +0100885
Bram Moolenaar94738d82020-10-21 14:25:07 +0200886
887" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker