blob: 83b9931f24daf33c53fb4f5a2fcd04e92f96a18e [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 Moolenaarfbcbffe2020-10-31 19:33:38 +0100244enddef
245
Bram Moolenaarb0e6b512021-01-12 20:23:40 +0100246def Test_extendnew()
247 assert_equal([1, 2, 'a'], extendnew([1, 2], ['a']))
248 assert_equal({one: 1, two: 'a'}, extendnew({one: 1}, {two: 'a'}))
249
250 CheckDefFailure(['extendnew({a: 1}, 42)'], 'E1013: Argument 2: type mismatch, expected dict<number> but got number')
251 CheckDefFailure(['extendnew({a: 1}, [42])'], 'E1013: Argument 2: type mismatch, expected dict<number> but got list<number>')
252 CheckDefFailure(['extendnew([1, 2], "x")'], 'E1013: Argument 2: type mismatch, expected list<number> but got string')
253 CheckDefFailure(['extendnew([1, 2], {x: 1})'], 'E1013: Argument 2: type mismatch, expected list<number> but got dict<number>')
254enddef
255
Bram Moolenaar94738d82020-10-21 14:25:07 +0200256def Test_extend_return_type()
257 var l = extend([1, 2], [3])
258 var res = 0
259 for n in l
260 res += n
261 endfor
262 res->assert_equal(6)
263enddef
264
Bram Moolenaaraa210a32021-01-02 15:41:03 +0100265func g:ExtendDict(d)
266 call extend(a:d, #{xx: 'x'})
267endfunc
268
269def Test_extend_dict_item_type()
270 var lines =<< trim END
271 var d: dict<number> = {a: 1}
272 extend(d, {b: 2})
273 END
274 CheckDefAndScriptSuccess(lines)
275
276 lines =<< trim END
277 var d: dict<number> = {a: 1}
278 extend(d, {b: 'x'})
279 END
280 CheckDefFailure(lines, 'E1013: Argument 2: type mismatch, expected dict<number> but got dict<string>', 2)
281 CheckScriptFailure(['vim9script'] + lines, 'E1012:', 3)
282
283 lines =<< trim END
284 var d: dict<number> = {a: 1}
285 g:ExtendDict(d)
286 END
287 CheckDefExecFailure(lines, 'E1012: Type mismatch; expected number but got string', 0)
288 CheckScriptFailure(['vim9script'] + lines, 'E1012:', 1)
289enddef
290
291func g:ExtendList(l)
292 call extend(a:l, ['x'])
293endfunc
294
295def Test_extend_list_item_type()
296 var lines =<< trim END
297 var l: list<number> = [1]
298 extend(l, [2])
299 END
300 CheckDefAndScriptSuccess(lines)
301
302 lines =<< trim END
303 var l: list<number> = [1]
304 extend(l, ['x'])
305 END
306 CheckDefFailure(lines, 'E1013: Argument 2: type mismatch, expected list<number> but got list<string>', 2)
307 CheckScriptFailure(['vim9script'] + lines, 'E1012:', 3)
308
309 lines =<< trim END
310 var l: list<number> = [1]
311 g:ExtendList(l)
312 END
313 CheckDefExecFailure(lines, 'E1012: Type mismatch; expected number but got string', 0)
314 CheckScriptFailure(['vim9script'] + lines, 'E1012:', 1)
315enddef
Bram Moolenaar94738d82020-10-21 14:25:07 +0200316
Bram Moolenaar64ed4d42021-01-12 21:22:31 +0100317def Test_job_info_return_type()
318 if has('job')
319 job_start(&shell)
320 var jobs = job_info()
321 assert_equal(v:t_list, type(jobs))
322 assert_equal(v:t_dict, type(job_info(jobs[0])))
323 job_stop(jobs[0])
324 endif
325enddef
326
Bram Moolenaar94738d82020-10-21 14:25:07 +0200327def Wrong_dict_key_type(items: list<number>): list<number>
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100328 return filter(items, (_, val) => get({[val]: 1}, 'x'))
Bram Moolenaar94738d82020-10-21 14:25:07 +0200329enddef
330
Bram Moolenaar08cf0c02020-12-05 21:47:06 +0100331def Test_map_function_arg()
332 var lines =<< trim END
333 def MapOne(i: number, v: string): string
334 return i .. ':' .. v
335 enddef
336 var l = ['a', 'b', 'c']
337 map(l, MapOne)
338 assert_equal(['0:a', '1:b', '2:c'], l)
339 END
340 CheckDefAndScriptSuccess(lines)
341enddef
342
Bram Moolenaar75ab91f2021-01-10 22:42:50 +0100343def Test_map_item_type()
344 var lines =<< trim END
345 var l = ['a', 'b', 'c']
346 map(l, (k, v) => k .. '/' .. v )
347 assert_equal(['0/a', '1/b', '2/c'], l)
348 END
349 CheckDefAndScriptSuccess(lines)
350enddef
351
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100352def Test_filereadable()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100353 assert_false(filereadable(""))
354 assert_false(filereadable(test_null_string()))
355
356 CheckDefExecFailure(['echo filereadable(123)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100357 CheckDefExecFailure(['echo filereadable(true)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100358enddef
359
360def Test_filewritable()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100361 assert_false(filewritable(""))
362 assert_false(filewritable(test_null_string()))
363
364 CheckDefExecFailure(['echo filewritable(123)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100365 CheckDefExecFailure(['echo filewritable(true)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100366enddef
367
368def Test_finddir()
369 CheckDefExecFailure(['echo finddir(true)'], 'E928:')
370 CheckDefExecFailure(['echo finddir(v:null)'], 'E928:')
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100371 CheckDefExecFailure(['echo finddir("")'], 'E1142:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100372enddef
373
374def Test_findfile()
375 CheckDefExecFailure(['echo findfile(true)'], 'E928:')
376 CheckDefExecFailure(['echo findfile(v:null)'], 'E928:')
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100377 CheckDefExecFailure(['echo findfile("")'], 'E1142:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100378enddef
379
380def Test_fnamemodify()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100381 CheckDefSuccess(['echo fnamemodify(test_null_string(), ":p")'])
382 CheckDefSuccess(['echo fnamemodify("", ":p")'])
383 CheckDefSuccess(['echo fnamemodify("file", test_null_string())'])
384 CheckDefSuccess(['echo fnamemodify("file", "")'])
385
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100386 CheckDefExecFailure(['echo fnamemodify(true, ":p")'], 'E928:')
387 CheckDefExecFailure(['echo fnamemodify(v:null, ":p")'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100388 CheckDefExecFailure(['echo fnamemodify("file", true)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100389enddef
390
Bram Moolenaar94738d82020-10-21 14:25:07 +0200391def Test_filter_wrong_dict_key_type()
392 assert_fails('Wrong_dict_key_type([1, 2, 3])', 'E1012:')
393enddef
394
395def Test_filter_return_type()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100396 var l = filter([1, 2, 3], () => 1)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200397 var res = 0
398 for n in l
399 res += n
400 endfor
401 res->assert_equal(6)
402enddef
403
Bram Moolenaarfc0e8f52020-12-25 20:24:51 +0100404def Test_filter_missing_argument()
405 var dict = {aa: [1], ab: [2], ac: [3], de: [4]}
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100406 var res = dict->filter((k) => k =~ 'a' && k !~ 'b')
Bram Moolenaarfc0e8f52020-12-25 20:24:51 +0100407 res->assert_equal({aa: [1], ac: [3]})
408enddef
Bram Moolenaar94738d82020-10-21 14:25:07 +0200409
410def Test_garbagecollect()
411 garbagecollect(true)
412enddef
413
414def Test_getbufinfo()
415 var bufinfo = getbufinfo(bufnr())
416 getbufinfo('%')->assert_equal(bufinfo)
417
418 edit Xtestfile1
419 hide edit Xtestfile2
420 hide enew
Bram Moolenaare0de1712020-12-02 17:36:54 +0100421 getbufinfo({bufloaded: true, buflisted: true, bufmodified: false})
Bram Moolenaar94738d82020-10-21 14:25:07 +0200422 ->len()->assert_equal(3)
423 bwipe Xtestfile1 Xtestfile2
424enddef
425
426def Test_getbufline()
427 e SomeFile
428 var buf = bufnr()
429 e #
430 var lines = ['aaa', 'bbb', 'ccc']
431 setbufline(buf, 1, lines)
432 getbufline('#', 1, '$')->assert_equal(lines)
Bram Moolenaare6e70a12020-10-22 18:23:38 +0200433 getbufline(-1, '$', '$')->assert_equal([])
434 getbufline(-1, 1, '$')->assert_equal([])
Bram Moolenaar94738d82020-10-21 14:25:07 +0200435
436 bwipe!
437enddef
438
439def Test_getchangelist()
440 new
441 setline(1, 'some text')
442 var changelist = bufnr()->getchangelist()
443 getchangelist('%')->assert_equal(changelist)
444 bwipe!
445enddef
446
447def Test_getchar()
448 while getchar(0)
449 endwhile
450 getchar(true)->assert_equal(0)
451enddef
452
453def Test_getcompletion()
454 set wildignore=*.vim,*~
455 var l = getcompletion('run', 'file', true)
456 l->assert_equal([])
457 set wildignore&
458enddef
459
460def Test_getloclist_return_type()
461 var l = getloclist(1)
462 l->assert_equal([])
463
Bram Moolenaare0de1712020-12-02 17:36:54 +0100464 var d = getloclist(1, {items: 0})
465 d->assert_equal({items: []})
Bram Moolenaar94738d82020-10-21 14:25:07 +0200466enddef
467
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100468def Test_getfperm()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100469 assert_equal('', getfperm(""))
470 assert_equal('', getfperm(test_null_string()))
471
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100472 CheckDefExecFailure(['echo getfperm(true)'], 'E928:')
473 CheckDefExecFailure(['echo getfperm(v:null)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100474enddef
475
476def Test_getfsize()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100477 assert_equal(-1, getfsize(""))
478 assert_equal(-1, getfsize(test_null_string()))
479
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100480 CheckDefExecFailure(['echo getfsize(true)'], 'E928:')
481 CheckDefExecFailure(['echo getfsize(v:null)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100482enddef
483
484def Test_getftime()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100485 assert_equal(-1, getftime(""))
486 assert_equal(-1, getftime(test_null_string()))
487
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100488 CheckDefExecFailure(['echo getftime(true)'], 'E928:')
489 CheckDefExecFailure(['echo getftime(v:null)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100490enddef
491
492def Test_getftype()
Bram Moolenaar2a9d5d32020-12-12 18:58:40 +0100493 assert_equal('', getftype(""))
494 assert_equal('', getftype(test_null_string()))
495
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100496 CheckDefExecFailure(['echo getftype(true)'], 'E928:')
497 CheckDefExecFailure(['echo getftype(v:null)'], 'E928:')
Bram Moolenaar7bb4e742020-12-09 12:41:50 +0100498enddef
499
Bram Moolenaar94738d82020-10-21 14:25:07 +0200500def Test_getqflist_return_type()
501 var l = getqflist()
502 l->assert_equal([])
503
Bram Moolenaare0de1712020-12-02 17:36:54 +0100504 var d = getqflist({items: 0})
505 d->assert_equal({items: []})
Bram Moolenaar94738d82020-10-21 14:25:07 +0200506enddef
507
508def Test_getreg()
509 var lines = ['aaa', 'bbb', 'ccc']
510 setreg('a', lines)
511 getreg('a', true, true)->assert_equal(lines)
512enddef
513
514def Test_getreg_return_type()
515 var s1: string = getreg('"')
516 var s2: string = getreg('"', 1)
517 var s3: list<string> = getreg('"', 1, 1)
518enddef
519
520def Test_glob()
521 glob('runtest.vim', true, true, true)->assert_equal(['runtest.vim'])
522enddef
523
524def Test_globpath()
525 globpath('.', 'runtest.vim', true, true, true)->assert_equal(['./runtest.vim'])
526enddef
527
528def Test_has()
529 has('eval', true)->assert_equal(1)
530enddef
531
532def Test_hasmapto()
533 hasmapto('foobar', 'i', true)->assert_equal(0)
534 iabbrev foo foobar
535 hasmapto('foobar', 'i', true)->assert_equal(1)
536 iunabbrev foo
537enddef
538
539def Test_index()
540 index(['a', 'b', 'a', 'B'], 'b', 2, true)->assert_equal(3)
541enddef
542
Bram Moolenaar193f6202020-11-16 20:08:35 +0100543let s:number_one = 1
544let s:number_two = 2
545let s:string_keep = 'keep'
546
Bram Moolenaarca174532020-10-21 16:42:22 +0200547def Test_insert()
Bram Moolenaar94738d82020-10-21 14:25:07 +0200548 var l = insert([2, 1], 3)
549 var res = 0
550 for n in l
551 res += n
552 endfor
553 res->assert_equal(6)
Bram Moolenaarca174532020-10-21 16:42:22 +0200554
555 assert_equal([1, 2, 3], insert([2, 3], 1))
Bram Moolenaar193f6202020-11-16 20:08:35 +0100556 assert_equal([1, 2, 3], insert([2, 3], s:number_one))
Bram Moolenaarca174532020-10-21 16:42:22 +0200557 assert_equal([1, 2, 3], insert([1, 2], 3, 2))
Bram Moolenaar193f6202020-11-16 20:08:35 +0100558 assert_equal([1, 2, 3], insert([1, 2], 3, s:number_two))
Bram Moolenaarca174532020-10-21 16:42:22 +0200559 assert_equal(['a', 'b', 'c'], insert(['b', 'c'], 'a'))
560 assert_equal(0z1234, insert(0z34, 0x12))
Bram Moolenaar193f6202020-11-16 20:08:35 +0100561
Bram Moolenaarca174532020-10-21 16:42:22 +0200562 CheckDefFailure(['insert([2, 3], "a")'], 'E1013: Argument 2: type mismatch, expected number but got string', 1)
563 CheckDefFailure(['insert([2, 3], 1, "x")'], 'E1013: Argument 3: type mismatch, expected number but got string', 1)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200564enddef
565
566def Test_keys_return_type()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100567 const var: list<string> = {a: 1, b: 2}->keys()
Bram Moolenaar94738d82020-10-21 14:25:07 +0200568 var->assert_equal(['a', 'b'])
569enddef
570
571def Test_list2str_str2list_utf8()
572 var s = "\u3042\u3044"
573 var l = [0x3042, 0x3044]
574 str2list(s, true)->assert_equal(l)
575 list2str(l, true)->assert_equal(s)
576enddef
577
578def SID(): number
579 return expand('<SID>')
580 ->matchstr('<SNR>\zs\d\+\ze_$')
581 ->str2nr()
582enddef
583
584def Test_maparg()
585 var lnum = str2nr(expand('<sflnum>'))
586 map foo bar
Bram Moolenaare0de1712020-12-02 17:36:54 +0100587 maparg('foo', '', false, true)->assert_equal({
Bram Moolenaar94738d82020-10-21 14:25:07 +0200588 lnum: lnum + 1,
589 script: 0,
590 mode: ' ',
591 silent: 0,
592 noremap: 0,
593 lhs: 'foo',
594 lhsraw: 'foo',
595 nowait: 0,
596 expr: 0,
597 sid: SID(),
598 rhs: 'bar',
599 buffer: 0})
600 unmap foo
601enddef
602
603def Test_mapcheck()
604 iabbrev foo foobar
605 mapcheck('foo', 'i', true)->assert_equal('foobar')
606 iunabbrev foo
607enddef
608
609def Test_maparg_mapset()
610 nnoremap <F3> :echo "hit F3"<CR>
611 var mapsave = maparg('<F3>', 'n', false, true)
612 mapset('n', false, mapsave)
613
614 nunmap <F3>
615enddef
616
617def Test_nr2char()
618 nr2char(97, true)->assert_equal('a')
619enddef
620
621def Test_readdir()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100622 eval expand('sautest')->readdir((e) => e[0] !=# '.')
623 eval expand('sautest')->readdirex((e) => e.name[0] !=# '.')
Bram Moolenaar94738d82020-10-21 14:25:07 +0200624enddef
625
626def Test_remove_return_type()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100627 var l = remove({one: [1, 2], two: [3, 4]}, 'one')
Bram Moolenaar94738d82020-10-21 14:25:07 +0200628 var res = 0
629 for n in l
630 res += n
631 endfor
632 res->assert_equal(3)
633enddef
634
635def Test_reverse_return_type()
636 var l = reverse([1, 2, 3])
637 var res = 0
638 for n in l
639 res += n
640 endfor
641 res->assert_equal(6)
642enddef
643
644def Test_search()
645 new
646 setline(1, ['foo', 'bar'])
647 var val = 0
648 # skip expr returns boolean
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100649 search('bar', 'W', 0, 0, () => val == 1)->assert_equal(2)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200650 :1
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100651 search('bar', 'W', 0, 0, () => val == 0)->assert_equal(0)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200652 # skip expr returns number, only 0 and 1 are accepted
653 :1
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100654 search('bar', 'W', 0, 0, () => 0)->assert_equal(2)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200655 :1
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100656 search('bar', 'W', 0, 0, () => 1)->assert_equal(0)
657 assert_fails("search('bar', '', 0, 0, () => -1)", 'E1023:')
658 assert_fails("search('bar', '', 0, 0, () => -1)", 'E1023:')
Bram Moolenaar94738d82020-10-21 14:25:07 +0200659enddef
660
661def Test_searchcount()
662 new
663 setline(1, "foo bar")
664 :/foo
Bram Moolenaare0de1712020-12-02 17:36:54 +0100665 searchcount({recompute: true})
666 ->assert_equal({
Bram Moolenaar94738d82020-10-21 14:25:07 +0200667 exact_match: 1,
668 current: 1,
669 total: 1,
670 maxcount: 99,
671 incomplete: 0})
672 bwipe!
673enddef
674
675def Test_searchdecl()
676 searchdecl('blah', true, true)->assert_equal(1)
677enddef
678
679def Test_setbufvar()
680 setbufvar(bufnr('%'), '&syntax', 'vim')
681 &syntax->assert_equal('vim')
682 setbufvar(bufnr('%'), '&ts', 16)
683 &ts->assert_equal(16)
Bram Moolenaar31a201a2021-01-03 14:47:25 +0100684 setbufvar(bufnr('%'), '&ai', true)
685 &ai->assert_equal(true)
686 setbufvar(bufnr('%'), '&ft', 'filetype')
687 &ft->assert_equal('filetype')
688
Bram Moolenaar94738d82020-10-21 14:25:07 +0200689 settabwinvar(1, 1, '&syntax', 'vam')
690 &syntax->assert_equal('vam')
691 settabwinvar(1, 1, '&ts', 15)
692 &ts->assert_equal(15)
693 setlocal ts=8
Bram Moolenaarb0d81822021-01-03 15:55:10 +0100694 settabwinvar(1, 1, '&list', false)
695 &list->assert_equal(false)
Bram Moolenaar31a201a2021-01-03 14:47:25 +0100696 settabwinvar(1, 1, '&list', true)
697 &list->assert_equal(true)
698 setlocal list&
Bram Moolenaar94738d82020-10-21 14:25:07 +0200699
700 setbufvar('%', 'myvar', 123)
701 getbufvar('%', 'myvar')->assert_equal(123)
702enddef
703
704def Test_setloclist()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100705 var items = [{filename: '/tmp/file', lnum: 1, valid: true}]
706 var what = {items: items}
Bram Moolenaar94738d82020-10-21 14:25:07 +0200707 setqflist([], ' ', what)
708 setloclist(0, [], ' ', what)
709enddef
710
711def Test_setreg()
712 setreg('a', ['aaa', 'bbb', 'ccc'])
713 var reginfo = getreginfo('a')
714 setreg('a', reginfo)
715 getreginfo('a')->assert_equal(reginfo)
716enddef
717
718def Test_spellsuggest()
719 if !has('spell')
720 MissingFeature 'spell'
721 else
722 spellsuggest('marrch', 1, true)->assert_equal(['March'])
723 endif
724enddef
725
726def Test_sort_return_type()
727 var res: list<number>
728 res = [1, 2, 3]->sort()
729enddef
730
731def Test_sort_argument()
Bram Moolenaar08cf0c02020-12-05 21:47:06 +0100732 var lines =<< trim END
733 var res = ['b', 'a', 'c']->sort('i')
734 res->assert_equal(['a', 'b', 'c'])
735
736 def Compare(a: number, b: number): number
737 return a - b
738 enddef
739 var l = [3, 6, 7, 1, 8, 2, 4, 5]
740 sort(l, Compare)
741 assert_equal([1, 2, 3, 4, 5, 6, 7, 8], l)
742 END
743 CheckDefAndScriptSuccess(lines)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200744enddef
745
746def Test_split()
747 split(' aa bb ', '\W\+', true)->assert_equal(['', 'aa', 'bb', ''])
748enddef
749
750def Test_str2nr()
751 str2nr("1'000'000", 10, true)->assert_equal(1000000)
752enddef
753
754def Test_strchars()
755 strchars("A\u20dd", true)->assert_equal(1)
756enddef
757
758def Test_submatch()
759 var pat = 'A\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)'
Bram Moolenaar75ab91f2021-01-10 22:42:50 +0100760 var Rep = () => range(10)->mapnew((_, v) => submatch(v, true))->string()
Bram Moolenaar94738d82020-10-21 14:25:07 +0200761 var actual = substitute('A123456789', pat, Rep, '')
762 var expected = "[['A123456789'], ['1'], ['2'], ['3'], ['4'], ['5'], ['6'], ['7'], ['8'], ['9']]"
763 actual->assert_equal(expected)
764enddef
765
766def Test_synID()
767 new
768 setline(1, "text")
769 synID(1, 1, true)->assert_equal(0)
770 bwipe!
771enddef
772
773def Test_term_gettty()
774 if !has('terminal')
775 MissingFeature 'terminal'
776 else
777 var buf = Run_shell_in_terminal({})
778 term_gettty(buf, true)->assert_notequal('')
779 StopShellInTerminal(buf)
780 endif
781enddef
782
783def Test_term_start()
784 if !has('terminal')
785 MissingFeature 'terminal'
786 else
787 botright new
788 var winnr = winnr()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100789 term_start(&shell, {curwin: true})
Bram Moolenaar94738d82020-10-21 14:25:07 +0200790 winnr()->assert_equal(winnr)
791 bwipe!
792 endif
793enddef
794
795def Test_timer_paused()
Bram Moolenaar2949cfd2020-12-31 21:28:47 +0100796 var id = timer_start(50, () => 0)
Bram Moolenaar94738d82020-10-21 14:25:07 +0200797 timer_pause(id, true)
798 var info = timer_info(id)
799 info[0]['paused']->assert_equal(1)
800 timer_stop(id)
801enddef
802
803def Test_win_splitmove()
804 split
Bram Moolenaare0de1712020-12-02 17:36:54 +0100805 win_splitmove(1, 2, {vertical: true, rightbelow: true})
Bram Moolenaar94738d82020-10-21 14:25:07 +0200806 close
807enddef
808
Bram Moolenaar285b15f2020-12-29 20:25:19 +0100809def Test_winrestcmd()
810 split
811 var cmd = winrestcmd()
812 wincmd _
813 exe cmd
814 assert_equal(cmd, winrestcmd())
815 close
816enddef
817
Bram Moolenaar43b69b32021-01-07 20:23:33 +0100818def Test_winsaveview()
819 var view: dict<number> = winsaveview()
820
821 var lines =<< trim END
822 var view: list<number> = winsaveview()
823 END
824 CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected list<number> but got dict<number>', 1)
825enddef
826
827
Bram Moolenaar285b15f2020-12-29 20:25:19 +0100828
Bram Moolenaar94738d82020-10-21 14:25:07 +0200829
830" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker