1" 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>
35      return items({k: 'v'})
36    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_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
133def 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
203def 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
217def Test_delete()
218  var res: bool = delete('doesnotexist')
219  assert_equal(true, res)
220enddef
221
222def Test_executable()
223  assert_false(executable(""))
224  assert_false(executable(test_null_string()))
225
226  CheckDefExecFailure(['echo executable(123)'], 'E928:')
227  CheckDefExecFailure(['echo executable(true)'], 'E928:')
228enddef
229
230def Test_exepath()
231  CheckDefExecFailure(['echo exepath(true)'], 'E928:')
232  CheckDefExecFailure(['echo exepath(v:null)'], 'E928:')
233  CheckDefExecFailure(['echo exepath("")'], 'E1142:')
234enddef
235
236def Test_expand()
237  split SomeFile
238  expand('%', true, true)->assert_equal(['SomeFile'])
239  close
240enddef
241
242def Test_extend_arg_types()
243  g:number_one = 1
244  g:string_keep = 'keep'
245  var lines =<< trim END
246      assert_equal([1, 2, 3], extend([1, 2], [3]))
247      assert_equal([3, 1, 2], extend([1, 2], [3], 0))
248      assert_equal([1, 3, 2], extend([1, 2], [3], 1))
249      assert_equal([1, 3, 2], extend([1, 2], [3], g:number_one))
250
251      assert_equal({a: 1, b: 2, c: 3}, extend({a: 1, b: 2}, {c: 3}))
252      assert_equal({a: 1, b: 4}, extend({a: 1, b: 2}, {b: 4}))
253      assert_equal({a: 1, b: 2}, extend({a: 1, b: 2}, {b: 4}, 'keep'))
254      assert_equal({a: 1, b: 2}, extend({a: 1, b: 2}, {b: 4}, g:string_keep))
255
256      var res: list<dict<any>>
257      extend(res, mapnew([1, 2], (_, v) => ({})))
258      assert_equal([{}, {}], res)
259  END
260  CheckDefAndScriptSuccess(lines)
261
262  CheckDefFailure(['extend([1, 2], 3)'], 'E1013: Argument 2: type mismatch, expected list<number> but got number')
263  CheckDefFailure(['extend([1, 2], ["x"])'], 'E1013: Argument 2: type mismatch, expected list<number> but got list<string>')
264  CheckDefFailure(['extend([1, 2], [3], "x")'], 'E1013: Argument 3: type mismatch, expected number but got string')
265
266  CheckDefFailure(['extend({a: 1}, 42)'], 'E1013: Argument 2: type mismatch, expected dict<number> but got number')
267  CheckDefFailure(['extend({a: 1}, {b: "x"})'], 'E1013: Argument 2: type mismatch, expected dict<number> but got dict<string>')
268  CheckDefFailure(['extend({a: 1}, {b: 2}, 1)'], 'E1013: Argument 3: type mismatch, expected string but got number')
269
270  CheckDefFailure(['extend([1], ["b"])'], 'E1013: Argument 2: type mismatch, expected list<number> but got list<string>')
271  CheckDefExecFailure(['extend([1], ["b", 1])'], 'E1013: Argument 2: type mismatch, expected list<number> but got list<any>')
272enddef
273
274def Test_extendnew()
275  assert_equal([1, 2, 'a'], extendnew([1, 2], ['a']))
276  assert_equal({one: 1, two: 'a'}, extendnew({one: 1}, {two: 'a'}))
277
278  CheckDefFailure(['extendnew({a: 1}, 42)'], 'E1013: Argument 2: type mismatch, expected dict<number> but got number')
279  CheckDefFailure(['extendnew({a: 1}, [42])'], 'E1013: Argument 2: type mismatch, expected dict<number> but got list<number>')
280  CheckDefFailure(['extendnew([1, 2], "x")'], 'E1013: Argument 2: type mismatch, expected list<number> but got string')
281  CheckDefFailure(['extendnew([1, 2], {x: 1})'], 'E1013: Argument 2: type mismatch, expected list<number> but got dict<number>')
282enddef
283
284def Test_extend_return_type()
285  var l = extend([1, 2], [3])
286  var res = 0
287  for n in l
288    res += n
289  endfor
290  res->assert_equal(6)
291enddef
292
293func g:ExtendDict(d)
294  call extend(a:d, #{xx: 'x'})
295endfunc
296
297def Test_extend_dict_item_type()
298  var lines =<< trim END
299       var d: dict<number> = {a: 1}
300       extend(d, {b: 2})
301  END
302  CheckDefAndScriptSuccess(lines)
303
304  lines =<< trim END
305       var d: dict<number> = {a: 1}
306       extend(d, {b: 'x'})
307  END
308  CheckDefAndScriptFailure(lines, 'E1013: Argument 2: type mismatch, expected dict<number> but got dict<string>', 2)
309
310  lines =<< trim END
311       var d: dict<number> = {a: 1}
312       g:ExtendDict(d)
313  END
314  CheckDefExecFailure(lines, 'E1012: Type mismatch; expected number but got string', 0)
315  CheckScriptFailure(['vim9script'] + lines, 'E1012:', 1)
316enddef
317
318func g:ExtendList(l)
319  call extend(a:l, ['x'])
320endfunc
321
322def Test_extend_list_item_type()
323  var lines =<< trim END
324       var l: list<number> = [1]
325       extend(l, [2])
326  END
327  CheckDefAndScriptSuccess(lines)
328
329  lines =<< trim END
330       var l: list<number> = [1]
331       extend(l, ['x'])
332  END
333  CheckDefAndScriptFailure(lines, 'E1013: Argument 2: type mismatch, expected list<number> but got list<string>', 2)
334
335  lines =<< trim END
336       var l: list<number> = [1]
337       g:ExtendList(l)
338  END
339  CheckDefExecFailure(lines, 'E1012: Type mismatch; expected number but got string', 0)
340  CheckScriptFailure(['vim9script'] + lines, 'E1012:', 1)
341enddef
342
343def Test_job_info_return_type()
344  if has('job')
345    job_start(&shell)
346    var jobs = job_info()
347    assert_equal('list<job>', typename(jobs))
348    assert_equal('dict<any>', typename(job_info(jobs[0])))
349    job_stop(jobs[0])
350  endif
351enddef
352
353def Test_filereadable()
354  assert_false(filereadable(""))
355  assert_false(filereadable(test_null_string()))
356
357  CheckDefExecFailure(['echo filereadable(123)'], 'E928:')
358  CheckDefExecFailure(['echo filereadable(true)'], 'E928:')
359enddef
360
361def Test_filewritable()
362  assert_false(filewritable(""))
363  assert_false(filewritable(test_null_string()))
364
365  CheckDefExecFailure(['echo filewritable(123)'], 'E928:')
366  CheckDefExecFailure(['echo filewritable(true)'], 'E928:')
367enddef
368
369def Test_finddir()
370  CheckDefExecFailure(['echo finddir(true)'], 'E928:')
371  CheckDefExecFailure(['echo finddir(v:null)'], 'E928:')
372  CheckDefExecFailure(['echo finddir("")'], 'E1142:')
373enddef
374
375def Test_findfile()
376  CheckDefExecFailure(['echo findfile(true)'], 'E928:')
377  CheckDefExecFailure(['echo findfile(v:null)'], 'E928:')
378  CheckDefExecFailure(['echo findfile("")'], 'E1142:')
379enddef
380
381def Test_flattennew()
382  var lines =<< trim END
383      var l = [1, [2, [3, 4]], 5]
384      call assert_equal([1, 2, 3, 4, 5], flattennew(l))
385      call assert_equal([1, [2, [3, 4]], 5], l)
386
387      call assert_equal([1, 2, [3, 4], 5], flattennew(l, 1))
388      call assert_equal([1, [2, [3, 4]], 5], l)
389  END
390  CheckDefAndScriptSuccess(lines)
391
392  lines =<< trim END
393      echo flatten([1, 2, 3])
394  END
395  CheckDefAndScriptFailure(lines, 'E1158:')
396enddef
397
398def Test_fnamemodify()
399  CheckDefSuccess(['echo fnamemodify(test_null_string(), ":p")'])
400  CheckDefSuccess(['echo fnamemodify("", ":p")'])
401  CheckDefSuccess(['echo fnamemodify("file", test_null_string())'])
402  CheckDefSuccess(['echo fnamemodify("file", "")'])
403
404  CheckDefExecFailure(['echo fnamemodify(true, ":p")'], 'E928:')
405  CheckDefExecFailure(['echo fnamemodify(v:null, ":p")'], 'E928:')
406  CheckDefExecFailure(['echo fnamemodify("file", true)'], 'E928:')
407enddef
408
409def Wrong_dict_key_type(items: list<number>): list<number>
410  return filter(items, (_, val) => get({[val]: 1}, 'x'))
411enddef
412
413def Test_filter_wrong_dict_key_type()
414  assert_fails('Wrong_dict_key_type([1, v:null, 3])', 'E1013:')
415enddef
416
417def Test_filter_return_type()
418  var l = filter([1, 2, 3], () => 1)
419  var res = 0
420  for n in l
421    res += n
422  endfor
423  res->assert_equal(6)
424enddef
425
426def Test_filter_missing_argument()
427  var dict = {aa: [1], ab: [2], ac: [3], de: [4]}
428  var res = dict->filter((k) => k =~ 'a' && k !~ 'b')
429  res->assert_equal({aa: [1], ac: [3]})
430enddef
431
432def Test_garbagecollect()
433  garbagecollect(true)
434enddef
435
436def Test_getbufinfo()
437  var bufinfo = getbufinfo(bufnr())
438  getbufinfo('%')->assert_equal(bufinfo)
439
440  edit Xtestfile1
441  hide edit Xtestfile2
442  hide enew
443  getbufinfo({bufloaded: true, buflisted: true, bufmodified: false})
444      ->len()->assert_equal(3)
445  bwipe Xtestfile1 Xtestfile2
446enddef
447
448def Test_getbufline()
449  e SomeFile
450  var buf = bufnr()
451  e #
452  var lines = ['aaa', 'bbb', 'ccc']
453  setbufline(buf, 1, lines)
454  getbufline('#', 1, '$')->assert_equal(lines)
455  getbufline(-1, '$', '$')->assert_equal([])
456  getbufline(-1, 1, '$')->assert_equal([])
457
458  bwipe!
459enddef
460
461def Test_getchangelist()
462  new
463  setline(1, 'some text')
464  var changelist = bufnr()->getchangelist()
465  getchangelist('%')->assert_equal(changelist)
466  bwipe!
467enddef
468
469def Test_getchar()
470  while getchar(0)
471  endwhile
472  getchar(true)->assert_equal(0)
473enddef
474
475def Test_getenv()
476  if getenv('does-not_exist') == ''
477    assert_report('getenv() should return null')
478  endif
479  if getenv('does-not_exist') == null
480  else
481    assert_report('getenv() should return null')
482  endif
483  $SOMEENVVAR = 'some'
484  assert_equal('some', getenv('SOMEENVVAR'))
485  unlet $SOMEENVVAR
486enddef
487
488def Test_getcompletion()
489  set wildignore=*.vim,*~
490  var l = getcompletion('run', 'file', true)
491  l->assert_equal([])
492  set wildignore&
493enddef
494
495def Test_getloclist_return_type()
496  var l = getloclist(1)
497  l->assert_equal([])
498
499  var d = getloclist(1, {items: 0})
500  d->assert_equal({items: []})
501enddef
502
503def Test_getfperm()
504  assert_equal('', getfperm(""))
505  assert_equal('', getfperm(test_null_string()))
506
507  CheckDefExecFailure(['echo getfperm(true)'], 'E928:')
508  CheckDefExecFailure(['echo getfperm(v:null)'], 'E928:')
509enddef
510
511def Test_getfsize()
512  assert_equal(-1, getfsize(""))
513  assert_equal(-1, getfsize(test_null_string()))
514
515  CheckDefExecFailure(['echo getfsize(true)'], 'E928:')
516  CheckDefExecFailure(['echo getfsize(v:null)'], 'E928:')
517enddef
518
519def Test_getftime()
520  assert_equal(-1, getftime(""))
521  assert_equal(-1, getftime(test_null_string()))
522
523  CheckDefExecFailure(['echo getftime(true)'], 'E928:')
524  CheckDefExecFailure(['echo getftime(v:null)'], 'E928:')
525enddef
526
527def Test_getftype()
528  assert_equal('', getftype(""))
529  assert_equal('', getftype(test_null_string()))
530
531  CheckDefExecFailure(['echo getftype(true)'], 'E928:')
532  CheckDefExecFailure(['echo getftype(v:null)'], 'E928:')
533enddef
534
535def Test_getqflist_return_type()
536  var l = getqflist()
537  l->assert_equal([])
538
539  var d = getqflist({items: 0})
540  d->assert_equal({items: []})
541enddef
542
543def Test_getreg()
544  var lines = ['aaa', 'bbb', 'ccc']
545  setreg('a', lines)
546  getreg('a', true, true)->assert_equal(lines)
547  assert_fails('getreg("ab")', 'E1162:')
548enddef
549
550def Test_getreg_return_type()
551  var s1: string = getreg('"')
552  var s2: string = getreg('"', 1)
553  var s3: list<string> = getreg('"', 1, 1)
554enddef
555
556def Test_getreginfo()
557  var text = 'abc'
558  setreg('a', text)
559  getreginfo('a')->assert_equal({regcontents: [text], regtype: 'v', isunnamed: false})
560  assert_fails('getreginfo("ab")', 'E1162:')
561enddef
562
563def Test_getregtype()
564  var lines = ['aaa', 'bbb', 'ccc']
565  setreg('a', lines)
566  getregtype('a')->assert_equal('V')
567  assert_fails('getregtype("ab")', 'E1162:')
568enddef
569
570def Test_glob()
571  glob('runtest.vim', true, true, true)->assert_equal(['runtest.vim'])
572enddef
573
574def Test_globpath()
575  globpath('.', 'runtest.vim', true, true, true)->assert_equal(['./runtest.vim'])
576enddef
577
578def Test_has()
579  has('eval', true)->assert_equal(1)
580enddef
581
582def Test_hasmapto()
583  hasmapto('foobar', 'i', true)->assert_equal(0)
584  iabbrev foo foobar
585  hasmapto('foobar', 'i', true)->assert_equal(1)
586  iunabbrev foo
587enddef
588
589def Test_index()
590  index(['a', 'b', 'a', 'B'], 'b', 2, true)->assert_equal(3)
591enddef
592
593let s:number_one = 1
594let s:number_two = 2
595let s:string_keep = 'keep'
596
597def Test_insert()
598  var l = insert([2, 1], 3)
599  var res = 0
600  for n in l
601    res += n
602  endfor
603  res->assert_equal(6)
604
605  assert_equal([1, 2, 3], insert([2, 3], 1))
606  assert_equal([1, 2, 3], insert([2, 3], s:number_one))
607  assert_equal([1, 2, 3], insert([1, 2], 3, 2))
608  assert_equal([1, 2, 3], insert([1, 2], 3, s:number_two))
609  assert_equal(['a', 'b', 'c'], insert(['b', 'c'], 'a'))
610  assert_equal(0z1234, insert(0z34, 0x12))
611
612  CheckDefFailure(['insert([2, 3], "a")'], 'E1013: Argument 2: type mismatch, expected number but got string', 1)
613  CheckDefFailure(['insert([2, 3], 1, "x")'], 'E1013: Argument 3: type mismatch, expected number but got string', 1)
614enddef
615
616def Test_keys_return_type()
617  const var: list<string> = {a: 1, b: 2}->keys()
618  var->assert_equal(['a', 'b'])
619enddef
620
621def Test_list2str_str2list_utf8()
622  var s = "\u3042\u3044"
623  var l = [0x3042, 0x3044]
624  str2list(s, true)->assert_equal(l)
625  list2str(l, true)->assert_equal(s)
626enddef
627
628def SID(): number
629  return expand('<SID>')
630          ->matchstr('<SNR>\zs\d\+\ze_$')
631          ->str2nr()
632enddef
633
634def Test_map_function_arg()
635  var lines =<< trim END
636      def MapOne(i: number, v: string): string
637        return i .. ':' .. v
638      enddef
639      var l = ['a', 'b', 'c']
640      map(l, MapOne)
641      assert_equal(['0:a', '1:b', '2:c'], l)
642  END
643  CheckDefAndScriptSuccess(lines)
644enddef
645
646def Test_map_item_type()
647  var lines =<< trim END
648      var l = ['a', 'b', 'c']
649      map(l, (k, v) => k .. '/' .. v )
650      assert_equal(['0/a', '1/b', '2/c'], l)
651  END
652  CheckDefAndScriptSuccess(lines)
653
654  lines =<< trim END
655    var l: list<number> = [0]
656    echo map(l, (_, v) => [])
657  END
658  CheckDefExecAndScriptFailure(lines, 'E1012: Type mismatch; expected number but got list<unknown>', 2)
659
660  lines =<< trim END
661    var l: list<number> = range(2)
662    echo map(l, (_, v) => [])
663  END
664  CheckDefExecAndScriptFailure(lines, 'E1012: Type mismatch; expected number but got list<unknown>', 2)
665
666  lines =<< trim END
667    var d: dict<number> = {key: 0}
668    echo map(d, (_, v) => [])
669  END
670  CheckDefExecAndScriptFailure(lines, 'E1012: Type mismatch; expected number but got list<unknown>', 2)
671enddef
672
673def Test_maparg()
674  var lnum = str2nr(expand('<sflnum>'))
675  map foo bar
676  maparg('foo', '', false, true)->assert_equal({
677        lnum: lnum + 1,
678        script: 0,
679        mode: ' ',
680        silent: 0,
681        noremap: 0,
682        lhs: 'foo',
683        lhsraw: 'foo',
684        nowait: 0,
685        expr: 0,
686        sid: SID(),
687        rhs: 'bar',
688        buffer: 0})
689  unmap foo
690enddef
691
692def Test_mapcheck()
693  iabbrev foo foobar
694  mapcheck('foo', 'i', true)->assert_equal('foobar')
695  iunabbrev foo
696enddef
697
698def Test_maparg_mapset()
699  nnoremap <F3> :echo "hit F3"<CR>
700  var mapsave = maparg('<F3>', 'n', false, true)
701  mapset('n', false, mapsave)
702
703  nunmap <F3>
704enddef
705
706def Test_map_failure()
707  CheckFeature job
708
709  var lines =<< trim END
710      vim9script
711      writefile([], 'Xtmpfile')
712      silent e Xtmpfile
713      var d = {[bufnr('%')]: {a: 0}}
714      au BufReadPost * Func()
715      def Func()
716          if d->has_key('')
717          endif
718          eval d[expand('<abuf>')]->mapnew((_, v: dict<job>) => 0)
719      enddef
720      e
721  END
722  CheckScriptFailure(lines, 'E1013:')
723  au! BufReadPost
724  delete('Xtmpfile')
725enddef
726
727def Test_max()
728  g:flag = true
729  var l1: list<number> = g:flag
730          ? [1, max([2, 3])]
731          : [4, 5]
732  assert_equal([1, 3], l1)
733
734  g:flag = false
735  var l2: list<number> = g:flag
736          ? [1, max([2, 3])]
737          : [4, 5]
738  assert_equal([4, 5], l2)
739enddef
740
741def Test_min()
742  g:flag = true
743  var l1: list<number> = g:flag
744          ? [1, min([2, 3])]
745          : [4, 5]
746  assert_equal([1, 2], l1)
747
748  g:flag = false
749  var l2: list<number> = g:flag
750          ? [1, min([2, 3])]
751          : [4, 5]
752  assert_equal([4, 5], l2)
753enddef
754
755def Test_nr2char()
756  nr2char(97, true)->assert_equal('a')
757enddef
758
759def Test_readdir()
760   eval expand('sautest')->readdir((e) => e[0] !=# '.')
761   eval expand('sautest')->readdirex((e) => e.name[0] !=# '.')
762enddef
763
764def Test_readblob()
765  var blob = 0z12341234
766  writefile(blob, 'Xreadblob')
767  var read: blob = readblob('Xreadblob')
768  assert_equal(blob, read)
769
770  var lines =<< trim END
771      var read: list<string> = readblob('Xreadblob')
772  END
773  CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected list<string> but got blob', 1)
774  delete('Xreadblob')
775enddef
776
777def Test_readfile()
778  var text = ['aaa', 'bbb', 'ccc']
779  writefile(text, 'Xreadfile')
780  var read: list<string> = readfile('Xreadfile')
781  assert_equal(text, read)
782
783  var lines =<< trim END
784      var read: dict<string> = readfile('Xreadfile')
785  END
786  CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected dict<string> but got list<string>', 1)
787  delete('Xreadfile')
788enddef
789
790def Test_remove_return_type()
791  var l = remove({one: [1, 2], two: [3, 4]}, 'one')
792  var res = 0
793  for n in l
794    res += n
795  endfor
796  res->assert_equal(3)
797enddef
798
799def Test_reverse_return_type()
800  var l = reverse([1, 2, 3])
801  var res = 0
802  for n in l
803    res += n
804  endfor
805  res->assert_equal(6)
806enddef
807
808def Test_search()
809  new
810  setline(1, ['foo', 'bar'])
811  var val = 0
812  # skip expr returns boolean
813  search('bar', 'W', 0, 0, () => val == 1)->assert_equal(2)
814  :1
815  search('bar', 'W', 0, 0, () => val == 0)->assert_equal(0)
816  # skip expr returns number, only 0 and 1 are accepted
817  :1
818  search('bar', 'W', 0, 0, () => 0)->assert_equal(2)
819  :1
820  search('bar', 'W', 0, 0, () => 1)->assert_equal(0)
821  assert_fails("search('bar', '', 0, 0, () => -1)", 'E1023:')
822  assert_fails("search('bar', '', 0, 0, () => -1)", 'E1023:')
823enddef
824
825def Test_searchcount()
826  new
827  setline(1, "foo bar")
828  :/foo
829  searchcount({recompute: true})
830      ->assert_equal({
831          exact_match: 1,
832          current: 1,
833          total: 1,
834          maxcount: 99,
835          incomplete: 0})
836  bwipe!
837enddef
838
839def Test_set_get_bufline()
840  # similar to Test_setbufline_getbufline()
841  var lines =<< trim END
842      new
843      var b = bufnr('%')
844      hide
845      assert_equal(0, setbufline(b, 1, ['foo', 'bar']))
846      assert_equal(['foo'], getbufline(b, 1))
847      assert_equal(['bar'], getbufline(b, '$'))
848      assert_equal(['foo', 'bar'], getbufline(b, 1, 2))
849      exe "bd!" b
850      assert_equal([], getbufline(b, 1, 2))
851
852      split Xtest
853      setline(1, ['a', 'b', 'c'])
854      b = bufnr('%')
855      wincmd w
856
857      assert_equal(1, setbufline(b, 5, 'x'))
858      assert_equal(1, setbufline(b, 5, ['x']))
859      assert_equal(1, setbufline(b, 5, []))
860      assert_equal(1, setbufline(b, 5, test_null_list()))
861
862      assert_equal(1, 'x'->setbufline(bufnr('$') + 1, 1))
863      assert_equal(1, ['x']->setbufline(bufnr('$') + 1, 1))
864      assert_equal(1, []->setbufline(bufnr('$') + 1, 1))
865      assert_equal(1, test_null_list()->setbufline(bufnr('$') + 1, 1))
866
867      assert_equal(['a', 'b', 'c'], getbufline(b, 1, '$'))
868
869      assert_equal(0, setbufline(b, 4, ['d', 'e']))
870      assert_equal(['c'], b->getbufline(3))
871      assert_equal(['d'], getbufline(b, 4))
872      assert_equal(['e'], getbufline(b, 5))
873      assert_equal([], getbufline(b, 6))
874      assert_equal([], getbufline(b, 2, 1))
875
876      if has('job')
877        setbufline(b, 2, [function('eval'), {key: 123}, test_null_job()])
878        assert_equal(["function('eval')",
879                        "{'key': 123}",
880                        "no process"],
881                        getbufline(b, 2, 4))
882      endif
883
884      exe 'bwipe! ' .. b
885  END
886  CheckDefAndScriptSuccess(lines)
887enddef
888
889def Test_searchdecl()
890  searchdecl('blah', true, true)->assert_equal(1)
891enddef
892
893def Test_setbufvar()
894  setbufvar(bufnr('%'), '&syntax', 'vim')
895  &syntax->assert_equal('vim')
896  setbufvar(bufnr('%'), '&ts', 16)
897  &ts->assert_equal(16)
898  setbufvar(bufnr('%'), '&ai', true)
899  &ai->assert_equal(true)
900  setbufvar(bufnr('%'), '&ft', 'filetype')
901  &ft->assert_equal('filetype')
902
903  settabwinvar(1, 1, '&syntax', 'vam')
904  &syntax->assert_equal('vam')
905  settabwinvar(1, 1, '&ts', 15)
906  &ts->assert_equal(15)
907  setlocal ts=8
908  settabwinvar(1, 1, '&list', false)
909  &list->assert_equal(false)
910  settabwinvar(1, 1, '&list', true)
911  &list->assert_equal(true)
912  setlocal list&
913
914  setbufvar('%', 'myvar', 123)
915  getbufvar('%', 'myvar')->assert_equal(123)
916enddef
917
918def Test_setloclist()
919  var items = [{filename: '/tmp/file', lnum: 1, valid: true}]
920  var what = {items: items}
921  setqflist([], ' ', what)
922  setloclist(0, [], ' ', what)
923enddef
924
925def Test_setreg()
926  setreg('a', ['aaa', 'bbb', 'ccc'])
927  var reginfo = getreginfo('a')
928  setreg('a', reginfo)
929  getreginfo('a')->assert_equal(reginfo)
930  assert_fails('setreg("ab", 0)', 'E1162:')
931enddef
932
933def Test_slice()
934  assert_equal('12345', slice('012345', 1))
935  assert_equal('123', slice('012345', 1, 4))
936  assert_equal('1234', slice('012345', 1, -1))
937  assert_equal('1', slice('012345', 1, -4))
938  assert_equal('', slice('012345', 1, -5))
939  assert_equal('', slice('012345', 1, -6))
940
941  assert_equal([1, 2, 3, 4, 5], slice(range(6), 1))
942  assert_equal([1, 2, 3], slice(range(6), 1, 4))
943  assert_equal([1, 2, 3, 4], slice(range(6), 1, -1))
944  assert_equal([1], slice(range(6), 1, -4))
945  assert_equal([], slice(range(6), 1, -5))
946  assert_equal([], slice(range(6), 1, -6))
947
948  assert_equal(0z1122334455, slice(0z001122334455, 1))
949  assert_equal(0z112233, slice(0z001122334455, 1, 4))
950  assert_equal(0z11223344, slice(0z001122334455, 1, -1))
951  assert_equal(0z11, slice(0z001122334455, 1, -4))
952  assert_equal(0z, slice(0z001122334455, 1, -5))
953  assert_equal(0z, slice(0z001122334455, 1, -6))
954enddef
955
956def Test_spellsuggest()
957  if !has('spell')
958    MissingFeature 'spell'
959  else
960    spellsuggest('marrch', 1, true)->assert_equal(['March'])
961  endif
962enddef
963
964def Test_sort_return_type()
965  var res: list<number>
966  res = [1, 2, 3]->sort()
967enddef
968
969def Test_sort_argument()
970  var lines =<< trim END
971    var res = ['b', 'a', 'c']->sort('i')
972    res->assert_equal(['a', 'b', 'c'])
973
974    def Compare(a: number, b: number): number
975      return a - b
976    enddef
977    var l = [3, 6, 7, 1, 8, 2, 4, 5]
978    sort(l, Compare)
979    assert_equal([1, 2, 3, 4, 5, 6, 7, 8], l)
980  END
981  CheckDefAndScriptSuccess(lines)
982enddef
983
984def Test_split()
985  split('  aa  bb  ', '\W\+', true)->assert_equal(['', 'aa', 'bb', ''])
986enddef
987
988def Run_str2float()
989  if !has('float')
990    MissingFeature 'float'
991  endif
992    str2float("1.00")->assert_equal(1.00)
993    str2float("2e-2")->assert_equal(0.02)
994
995    CheckDefFailure(['echo str2float(123)'], 'E1013:')
996    CheckScriptFailure(['vim9script', 'echo str2float(123)'], 'E1024:')
997  endif
998enddef
999
1000def Test_str2nr()
1001  str2nr("1'000'000", 10, true)->assert_equal(1000000)
1002
1003  CheckDefFailure(['echo str2nr(123)'], 'E1013:')
1004  CheckScriptFailure(['vim9script', 'echo str2nr(123)'], 'E1024:')
1005  CheckDefFailure(['echo str2nr("123", "x")'], 'E1013:')
1006  CheckScriptFailure(['vim9script', 'echo str2nr("123", "x")'], 'E1030:')
1007  CheckDefFailure(['echo str2nr("123", 10, "x")'], 'E1013:')
1008  CheckScriptFailure(['vim9script', 'echo str2nr("123", 10, "x")'], 'E1135:')
1009enddef
1010
1011def Test_strchars()
1012  strchars("A\u20dd", true)->assert_equal(1)
1013enddef
1014
1015def Test_submatch()
1016  var pat = 'A\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)\(.\)'
1017  var Rep = () => range(10)->mapnew((_, v) => submatch(v, true))->string()
1018  var actual = substitute('A123456789', pat, Rep, '')
1019  var expected = "[['A123456789'], ['1'], ['2'], ['3'], ['4'], ['5'], ['6'], ['7'], ['8'], ['9']]"
1020  actual->assert_equal(expected)
1021enddef
1022
1023def Test_synID()
1024  new
1025  setline(1, "text")
1026  synID(1, 1, true)->assert_equal(0)
1027  bwipe!
1028enddef
1029
1030def Test_term_gettty()
1031  if !has('terminal')
1032    MissingFeature 'terminal'
1033  else
1034    var buf = Run_shell_in_terminal({})
1035    term_gettty(buf, true)->assert_notequal('')
1036    StopShellInTerminal(buf)
1037  endif
1038enddef
1039
1040def Test_term_start()
1041  if !has('terminal')
1042    MissingFeature 'terminal'
1043  else
1044    botright new
1045    var winnr = winnr()
1046    term_start(&shell, {curwin: true})
1047    winnr()->assert_equal(winnr)
1048    bwipe!
1049  endif
1050enddef
1051
1052def Test_timer_paused()
1053  var id = timer_start(50, () => 0)
1054  timer_pause(id, true)
1055  var info = timer_info(id)
1056  info[0]['paused']->assert_equal(1)
1057  timer_stop(id)
1058enddef
1059
1060def Test_win_execute()
1061  assert_equal("\n" .. winnr(), win_execute(win_getid(), 'echo winnr()'))
1062  assert_equal('', win_execute(342343, 'echo winnr()'))
1063enddef
1064
1065def Test_win_splitmove()
1066  split
1067  win_splitmove(1, 2, {vertical: true, rightbelow: true})
1068  close
1069enddef
1070
1071def Test_winrestcmd()
1072  split
1073  var cmd = winrestcmd()
1074  wincmd _
1075  exe cmd
1076  assert_equal(cmd, winrestcmd())
1077  close
1078enddef
1079
1080def Test_winsaveview()
1081  var view: dict<number> = winsaveview()
1082
1083  var lines =<< trim END
1084      var view: list<number> = winsaveview()
1085  END
1086  CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected list<number> but got dict<number>', 1)
1087enddef
1088
1089
1090
1091
1092" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker
1093