1" Test various aspects of the Vim9 script language.
2
3source check.vim
4source term_util.vim
5source view_util.vim
6source vim9.vim
7source shared.vim
8source screendump.vim
9
10def Test_range_only()
11  new
12  setline(1, ['blah', 'Blah'])
13  :/Blah/
14  assert_equal(2, getcurpos()[1])
15  bwipe!
16
17  # without range commands use current line
18  new
19  setline(1, ['one', 'two', 'three'])
20  :2
21  print
22  assert_equal('two', Screenline(&lines))
23  :3
24  list
25  assert_equal('three$', Screenline(&lines))
26
27  # missing command does not print the line
28  var lines =<< trim END
29    vim9script
30    :1|
31    assert_equal('three$', Screenline(&lines))
32    :|
33    assert_equal('three$', Screenline(&lines))
34  END
35  CheckScriptSuccess(lines)
36
37  bwipe!
38
39  # won't generate anything
40  if false
41    :123
42  endif
43enddef
44
45let g:alist = [7]
46let g:astring = 'text'
47let g:anumber = 123
48
49def Test_delfunction()
50  # Check function is defined in script namespace
51  CheckScriptSuccess([
52      'vim9script',
53      'func CheckMe()',
54      '  return 123',
55      'endfunc',
56      'assert_equal(123, s:CheckMe())',
57      ])
58
59  # Check function in script namespace cannot be deleted
60  CheckScriptFailure([
61      'vim9script',
62      'func DeleteMe1()',
63      'endfunc',
64      'delfunction DeleteMe1',
65      ], 'E1084:')
66  CheckScriptFailure([
67      'vim9script',
68      'func DeleteMe2()',
69      'endfunc',
70      'def DoThat()',
71      '  delfunction DeleteMe2',
72      'enddef',
73      'DoThat()',
74      ], 'E1084:')
75  CheckScriptFailure([
76      'vim9script',
77      'def DeleteMe3()',
78      'enddef',
79      'delfunction DeleteMe3',
80      ], 'E1084:')
81  CheckScriptFailure([
82      'vim9script',
83      'def DeleteMe4()',
84      'enddef',
85      'def DoThat()',
86      '  delfunction DeleteMe4',
87      'enddef',
88      'DoThat()',
89      ], 'E1084:')
90
91  # Check that global :def function can be replaced and deleted
92  var lines =<< trim END
93      vim9script
94      def g:Global(): string
95        return "yes"
96      enddef
97      assert_equal("yes", g:Global())
98      def! g:Global(): string
99        return "no"
100      enddef
101      assert_equal("no", g:Global())
102      delfunc g:Global
103      assert_false(exists('*g:Global'))
104  END
105  CheckScriptSuccess(lines)
106
107  # Check that global function can be replaced by a :def function and deleted
108  lines =<< trim END
109      vim9script
110      func g:Global()
111        return "yes"
112      endfunc
113      assert_equal("yes", g:Global())
114      def! g:Global(): string
115        return "no"
116      enddef
117      assert_equal("no", g:Global())
118      delfunc g:Global
119      assert_false(exists('*g:Global'))
120  END
121  CheckScriptSuccess(lines)
122
123  # Check that global :def function can be replaced by a function and deleted
124  lines =<< trim END
125      vim9script
126      def g:Global(): string
127        return "yes"
128      enddef
129      assert_equal("yes", g:Global())
130      func! g:Global()
131        return "no"
132      endfunc
133      assert_equal("no", g:Global())
134      delfunc g:Global
135      assert_false(exists('*g:Global'))
136  END
137  CheckScriptSuccess(lines)
138enddef
139
140def Test_wrong_type()
141  CheckDefFailure(['var name: list<nothing>'], 'E1010:')
142  CheckDefFailure(['var name: list<list<nothing>>'], 'E1010:')
143  CheckDefFailure(['var name: dict<nothing>'], 'E1010:')
144  CheckDefFailure(['var name: dict<dict<nothing>>'], 'E1010:')
145
146  CheckDefFailure(['var name: dict<number'], 'E1009:')
147  CheckDefFailure(['var name: dict<list<number>'], 'E1009:')
148
149  CheckDefFailure(['var name: ally'], 'E1010:')
150  CheckDefFailure(['var name: bram'], 'E1010:')
151  CheckDefFailure(['var name: cathy'], 'E1010:')
152  CheckDefFailure(['var name: dom'], 'E1010:')
153  CheckDefFailure(['var name: freddy'], 'E1010:')
154  CheckDefFailure(['var name: john'], 'E1010:')
155  CheckDefFailure(['var name: larry'], 'E1010:')
156  CheckDefFailure(['var name: ned'], 'E1010:')
157  CheckDefFailure(['var name: pam'], 'E1010:')
158  CheckDefFailure(['var name: sam'], 'E1010:')
159  CheckDefFailure(['var name: vim'], 'E1010:')
160
161  CheckDefFailure(['var Ref: number', 'Ref()'], 'E1085:')
162  CheckDefFailure(['var Ref: string', 'var res = Ref()'], 'E1085:')
163enddef
164
165def Test_script_wrong_type()
166  var lines =<< trim END
167      vim9script
168      var s:dict: dict<string>
169      s:dict['a'] = ['x']
170  END
171  CheckScriptFailure(lines, 'E1012: Type mismatch; expected string but got list<string>', 3)
172enddef
173
174def Test_const()
175  CheckDefFailure(['final name = 234', 'name = 99'], 'E1018:')
176  CheckDefFailure(['final one = 234', 'var one = 99'], 'E1017:')
177  CheckDefFailure(['final list = [1, 2]', 'var list = [3, 4]'], 'E1017:')
178  CheckDefFailure(['final two'], 'E1125:')
179  CheckDefFailure(['final &option'], 'E996:')
180
181  var lines =<< trim END
182    final list = [1, 2, 3]
183    list[0] = 4
184    list->assert_equal([4, 2, 3])
185    const other = [5, 6, 7]
186    other->assert_equal([5, 6, 7])
187
188    var varlist = [7, 8]
189    const constlist = [1, varlist, 3]
190    varlist[0] = 77
191    # TODO: does not work yet
192    # constlist[1][1] = 88
193    var cl = constlist[1]
194    cl[1] = 88
195    constlist->assert_equal([1, [77, 88], 3])
196
197    var vardict = {five: 5, six: 6}
198    const constdict = {one: 1, two: vardict, three: 3}
199    vardict['five'] = 55
200    # TODO: does not work yet
201    # constdict['two']['six'] = 66
202    var cd = constdict['two']
203    cd['six'] = 66
204    constdict->assert_equal({one: 1, two: {five: 55, six: 66}, three: 3})
205  END
206  CheckDefAndScriptSuccess(lines)
207enddef
208
209def Test_const_bang()
210  var lines =<< trim END
211      const var = 234
212      var = 99
213  END
214  CheckDefExecFailure(lines, 'E1018:', 2)
215  CheckScriptFailure(['vim9script'] + lines, 'E46:', 3)
216
217  lines =<< trim END
218      const ll = [2, 3, 4]
219      ll[0] = 99
220  END
221  CheckDefExecFailure(lines, 'E1119:', 2)
222  CheckScriptFailure(['vim9script'] + lines, 'E741:', 3)
223
224  lines =<< trim END
225      const ll = [2, 3, 4]
226      ll[3] = 99
227  END
228  CheckDefExecFailure(lines, 'E1118:', 2)
229  CheckScriptFailure(['vim9script'] + lines, 'E684:', 3)
230
231  lines =<< trim END
232      const dd = {one: 1, two: 2}
233      dd["one"] = 99
234  END
235  CheckDefExecFailure(lines, 'E1121:', 2)
236  CheckScriptFailure(['vim9script'] + lines, 'E741:', 3)
237
238  lines =<< trim END
239      const dd = {one: 1, two: 2}
240      dd["three"] = 99
241  END
242  CheckDefExecFailure(lines, 'E1120:')
243  CheckScriptFailure(['vim9script'] + lines, 'E741:', 3)
244enddef
245
246def Test_range_no_colon()
247  CheckDefFailure(['%s/a/b/'], 'E1050:')
248  CheckDefFailure(['+ s/a/b/'], 'E1050:')
249  CheckDefFailure(['- s/a/b/'], 'E1050:')
250  CheckDefFailure(['. s/a/b/'], 'E1050:')
251enddef
252
253
254def Test_block()
255  var outer = 1
256  {
257    var inner = 2
258    assert_equal(1, outer)
259    assert_equal(2, inner)
260  }
261  assert_equal(1, outer)
262
263  {|echo 'yes'|}
264enddef
265
266def Test_block_failure()
267  CheckDefFailure(['{', 'var inner = 1', '}', 'echo inner'], 'E1001:')
268  CheckDefFailure(['}'], 'E1025:')
269  CheckDefFailure(['{', 'echo 1'], 'E1026:')
270enddef
271
272def Test_block_local_vars()
273  var lines =<< trim END
274      vim9script
275      v:testing = 1
276      if true
277        var text = ['hello']
278        def SayHello(): list<string>
279          return text
280        enddef
281        def SetText(v: string)
282          text = [v]
283        enddef
284      endif
285
286      if true
287        var text = ['again']
288        def SayAgain(): list<string>
289          return text
290        enddef
291      endif
292
293      # test that the "text" variables are not cleaned up
294      test_garbagecollect_now()
295
296      defcompile
297
298      assert_equal(['hello'], SayHello())
299      assert_equal(['again'], SayAgain())
300
301      SetText('foobar')
302      assert_equal(['foobar'], SayHello())
303
304      call writefile(['ok'], 'Xdidit')
305      qall!
306  END
307
308  # need to execute this with a separate Vim instance to avoid the current
309  # context gets garbage collected.
310  writefile(lines, 'Xscript')
311  RunVim([], [], '-S Xscript')
312  assert_equal(['ok'], readfile('Xdidit'))
313
314  delete('Xscript')
315  delete('Xdidit')
316enddef
317
318def Test_block_local_vars_with_func()
319  var lines =<< trim END
320      vim9script
321      if true
322        var foo = 'foo'
323        if true
324          var bar = 'bar'
325          def Func(): list<string>
326            return [foo, bar]
327          enddef
328        endif
329      endif
330      # function is compiled here, after blocks have finished, can still access
331      # "foo" and "bar"
332      assert_equal(['foo', 'bar'], Func())
333  END
334  CheckScriptSuccess(lines)
335enddef
336
337func g:NoSuchFunc()
338  echo 'none'
339endfunc
340
341def Test_try_catch_throw()
342  var l = []
343  try # comment
344    add(l, '1')
345    throw 'wrong'
346    add(l, '2')
347  catch # comment
348    add(l, v:exception)
349  finally # comment
350    add(l, '3')
351  endtry # comment
352  assert_equal(['1', 'wrong', '3'], l)
353
354  l = []
355  try
356    try
357      add(l, '1')
358      throw 'wrong'
359      add(l, '2')
360    catch /right/
361      add(l, v:exception)
362    endtry
363  catch /wrong/
364    add(l, 'caught')
365  fina
366    add(l, 'finally')
367  endtry
368  assert_equal(['1', 'caught', 'finally'], l)
369
370  var n: number
371  try
372    n = l[3]
373  catch /E684:/
374    n = 99
375  endtry
376  assert_equal(99, n)
377
378  var done = 'no'
379  if 0
380    try | catch | endtry
381  else
382    done = 'yes'
383  endif
384  assert_equal('yes', done)
385
386  done = 'no'
387  if 1
388    done = 'yes'
389  else
390    try | catch | endtry
391    done = 'never'
392  endif
393  assert_equal('yes', done)
394
395  if 1
396  else
397    try | catch /pat/ | endtry
398    try | catch /pat/
399    endtry
400    try
401    catch /pat/ | endtry
402    try
403    catch /pat/
404    endtry
405  endif
406
407  try
408    # string slice returns a string, not a number
409    n = g:astring[3]
410  catch /E1012:/
411    n = 77
412  endtry
413  assert_equal(77, n)
414
415  try
416    n = l[g:astring]
417  catch /E1012:/
418    n = 88
419  endtry
420  assert_equal(88, n)
421
422  try
423    n = s:does_not_exist
424  catch /E121:/
425    n = 111
426  endtry
427  assert_equal(111, n)
428
429  try
430    n = g:does_not_exist
431  catch /E121:/
432    n = 121
433  endtry
434  assert_equal(121, n)
435
436  var d = {one: 1}
437  try
438    n = d[g:astring]
439  catch /E716:/
440    n = 222
441  endtry
442  assert_equal(222, n)
443
444  try
445    n = -g:astring
446  catch /E39:/
447    n = 233
448  endtry
449  assert_equal(233, n)
450
451  try
452    n = +g:astring
453  catch /E1030:/
454    n = 244
455  endtry
456  assert_equal(244, n)
457
458  try
459    n = +g:alist
460  catch /E745:/
461    n = 255
462  endtry
463  assert_equal(255, n)
464
465  var nd: dict<any>
466  try
467    nd = {[g:alist]: 1}
468  catch /E1105:/
469    n = 266
470  endtry
471  assert_equal(266, n)
472
473  try
474    [n] = [1, 2, 3]
475  catch /E1093:/
476    n = 277
477  endtry
478  assert_equal(277, n)
479
480  try
481    &ts = g:astring
482  catch /E1012:/
483    n = 288
484  endtry
485  assert_equal(288, n)
486
487  try
488    &backspace = 'asdf'
489  catch /E474:/
490    n = 299
491  endtry
492  assert_equal(299, n)
493
494  l = [1]
495  try
496    l[3] = 3
497  catch /E684:/
498    n = 300
499  endtry
500  assert_equal(300, n)
501
502  try
503    unlet g:does_not_exist
504  catch /E108:/
505    n = 322
506  endtry
507  assert_equal(322, n)
508
509  try
510    d = {text: 1, [g:astring]: 2}
511  catch /E721:/
512    n = 333
513  endtry
514  assert_equal(333, n)
515
516  try
517    l = DeletedFunc()
518  catch /E933:/
519    n = 344
520  endtry
521  assert_equal(344, n)
522
523  try
524    echo len(v:true)
525  catch /E701:/
526    n = 355
527  endtry
528  assert_equal(355, n)
529
530  var P = function('g:NoSuchFunc')
531  delfunc g:NoSuchFunc
532  try
533    echo P()
534  catch /E117:/
535    n = 366
536  endtry
537  assert_equal(366, n)
538
539  try
540    echo g:NoSuchFunc()
541  catch /E117:/
542    n = 377
543  endtry
544  assert_equal(377, n)
545
546  try
547    echo g:alist + 4
548  catch /E745:/
549    n = 388
550  endtry
551  assert_equal(388, n)
552
553  try
554    echo 4 + g:alist
555  catch /E745:/
556    n = 399
557  endtry
558  assert_equal(399, n)
559
560  try
561    echo g:alist.member
562  catch /E715:/
563    n = 400
564  endtry
565  assert_equal(400, n)
566
567  try
568    echo d.member
569  catch /E716:/
570    n = 411
571  endtry
572  assert_equal(411, n)
573
574  var counter = 0
575  for i in range(4)
576    try
577      eval [][0]
578    catch
579    endtry
580    counter += 1
581  endfor
582  assert_equal(4, counter)
583
584  # return in finally after empty catch
585  def ReturnInFinally(): number
586    try
587    finally
588      return 4
589    endtry
590    return 2
591  enddef
592  assert_equal(4, ReturnInFinally())
593
594  var lines =<< trim END
595      vim9script
596      try
597        acos('0.5')
598          ->setline(1)
599      catch
600        g:caught = v:exception
601      endtry
602  END
603  CheckScriptSuccess(lines)
604  assert_match('E808: Number or Float required', g:caught)
605  unlet g:caught
606enddef
607
608" :while at the very start of a function that :continue jumps to
609def TryContinueFunc()
610 while g:Count < 2
611   g:sequence ..= 't'
612    try
613      echoerr 'Test'
614    catch
615      g:Count += 1
616      g:sequence ..= 'c'
617      continue
618    endtry
619    g:sequence ..= 'e'
620    g:Count += 1
621  endwhile
622enddef
623
624def Test_continue_in_try_in_while()
625  g:Count = 0
626  g:sequence = ''
627  TryContinueFunc()
628  assert_equal('tctc', g:sequence)
629  unlet g:Count
630  unlet g:sequence
631enddef
632
633def Test_nocatch_return_in_try()
634  # return in try block returns normally
635  def ReturnInTry(): string
636    try
637      return '"some message"'
638    catch
639    endtry
640    return 'not reached'
641  enddef
642  exe 'echoerr ' .. ReturnInTry()
643enddef
644
645def Test_cnext_works_in_catch()
646  var lines =<< trim END
647      vim9script
648      au BufEnter * eval 0
649      writefile(['text'], 'Xfile1')
650      writefile(['text'], 'Xfile2')
651      var items = [
652          {lnum: 1, filename: 'Xfile1', valid: true},
653          {lnum: 1, filename: 'Xfile2', valid: true}
654        ]
655      setqflist([], ' ', {items: items})
656      cwindow
657
658      def CnextOrCfirst()
659        # if cnext fails, cfirst is used
660        try
661          cnext
662        catch
663          cfirst
664        endtry
665      enddef
666
667      CnextOrCfirst()
668      CnextOrCfirst()
669      writefile([getqflist({idx: 0}).idx], 'Xresult')
670      qall
671  END
672  writefile(lines, 'XCatchCnext')
673  RunVim([], [], '--clean -S XCatchCnext')
674  assert_equal(['1'], readfile('Xresult'))
675
676  delete('Xfile1')
677  delete('Xfile2')
678  delete('XCatchCnext')
679  delete('Xresult')
680enddef
681
682def Test_throw_skipped()
683  if 0
684    throw dontgethere
685  endif
686enddef
687
688def Test_nocatch_throw_silenced()
689  var lines =<< trim END
690    vim9script
691    def Func()
692      throw 'error'
693    enddef
694    silent! Func()
695  END
696  writefile(lines, 'XthrowSilenced')
697  source XthrowSilenced
698  delete('XthrowSilenced')
699enddef
700
701def DeletedFunc(): list<any>
702  return ['delete me']
703enddef
704defcompile
705delfunc DeletedFunc
706
707def ThrowFromDef()
708  throw "getout" # comment
709enddef
710
711func CatchInFunc()
712  try
713    call ThrowFromDef()
714  catch
715    let g:thrown_func = v:exception
716  endtry
717endfunc
718
719def CatchInDef()
720  try
721    ThrowFromDef()
722  catch
723    g:thrown_def = v:exception
724  endtry
725enddef
726
727def ReturnFinally(): string
728  try
729    return 'intry'
730  finall
731    g:in_finally = 'finally'
732  endtry
733  return 'end'
734enddef
735
736def Test_try_catch_nested()
737  CatchInFunc()
738  assert_equal('getout', g:thrown_func)
739
740  CatchInDef()
741  assert_equal('getout', g:thrown_def)
742
743  assert_equal('intry', ReturnFinally())
744  assert_equal('finally', g:in_finally)
745enddef
746
747def TryOne(): number
748  try
749    return 0
750  catch
751  endtry
752  return 0
753enddef
754
755def TryTwo(n: number): string
756  try
757    var x = {}
758  catch
759  endtry
760  return 'text'
761enddef
762
763def Test_try_catch_twice()
764  assert_equal('text', TryOne()->TryTwo())
765enddef
766
767def Test_try_catch_match()
768  var seq = 'a'
769  try
770    throw 'something'
771  catch /nothing/
772    seq ..= 'x'
773  catch /some/
774    seq ..= 'b'
775  catch /asdf/
776    seq ..= 'x'
777  catch ?a\?sdf?
778    seq ..= 'y'
779  finally
780    seq ..= 'c'
781  endtry
782  assert_equal('abc', seq)
783enddef
784
785def Test_try_catch_fails()
786  CheckDefFailure(['catch'], 'E603:')
787  CheckDefFailure(['try', 'echo 0', 'catch', 'catch'], 'E1033:')
788  CheckDefFailure(['try', 'echo 0', 'catch /pat'], 'E1067:')
789  CheckDefFailure(['finally'], 'E606:')
790  CheckDefFailure(['try', 'echo 0', 'finally', 'echo 1', 'finally'], 'E607:')
791  CheckDefFailure(['endtry'], 'E602:')
792  CheckDefFailure(['while 1', 'endtry'], 'E170:')
793  CheckDefFailure(['for i in range(5)', 'endtry'], 'E170:')
794  CheckDefFailure(['if 1', 'endtry'], 'E171:')
795  CheckDefFailure(['try', 'echo 1', 'endtry'], 'E1032:')
796
797  CheckDefFailure(['throw'], 'E1143:')
798  CheckDefFailure(['throw xxx'], 'E1001:')
799enddef
800
801def Try_catch_skipped()
802  var l = []
803  try
804  finally
805  endtry
806
807  if 1
808  else
809    try
810    endtry
811  endif
812enddef
813
814" The skipped try/endtry was updating the wrong instruction.
815def Test_try_catch_skipped()
816  var instr = execute('disassemble Try_catch_skipped')
817  assert_match("NEWLIST size 0\n", instr)
818enddef
819
820
821
822def Test_throw_vimscript()
823  # only checks line continuation
824  var lines =<< trim END
825      vim9script
826      try
827        throw 'one'
828              .. 'two'
829      catch
830        assert_equal('onetwo', v:exception)
831      endtry
832  END
833  CheckScriptSuccess(lines)
834
835  lines =<< trim END
836    vim9script
837    @r = ''
838    def Func()
839      throw @r
840    enddef
841    var result = ''
842    try
843      Func()
844    catch /E1129:/
845      result = 'caught'
846    endtry
847    assert_equal('caught', result)
848  END
849  CheckScriptSuccess(lines)
850enddef
851
852def Test_error_in_nested_function()
853  # an error in a nested :function aborts executing in the calling :def function
854  var lines =<< trim END
855      vim9script
856      def Func()
857        Error()
858        g:test_var = 1
859      enddef
860      func Error() abort
861        eval [][0]
862      endfunc
863      Func()
864  END
865  g:test_var = 0
866  CheckScriptFailure(lines, 'E684:')
867  assert_equal(0, g:test_var)
868enddef
869
870def Test_abort_after_error()
871  var lines =<< trim END
872      vim9script
873      while true
874        echo notfound
875      endwhile
876      g:gotthere = true
877  END
878  g:gotthere = false
879  CheckScriptFailure(lines, 'E121:')
880  assert_false(g:gotthere)
881  unlet g:gotthere
882enddef
883
884def Test_cexpr_vimscript()
885  # only checks line continuation
886  set errorformat=File\ %f\ line\ %l
887  var lines =<< trim END
888      vim9script
889      cexpr 'File'
890                .. ' someFile' ..
891                   ' line 19'
892      assert_equal(19, getqflist()[0].lnum)
893  END
894  CheckScriptSuccess(lines)
895  set errorformat&
896enddef
897
898def Test_statusline_syntax()
899  # legacy syntax is used for 'statusline'
900  var lines =<< trim END
901      vim9script
902      func g:Status()
903        return '%{"x" is# "x"}'
904      endfunc
905      set laststatus=2 statusline=%!Status()
906      redrawstatus
907      set laststatus statusline=
908  END
909  CheckScriptSuccess(lines)
910enddef
911
912def Test_list_vimscript()
913  # checks line continuation and comments
914  var lines =<< trim END
915      vim9script
916      var mylist = [
917            'one',
918            # comment
919            'two', # empty line follows
920
921            'three',
922            ]
923      assert_equal(['one', 'two', 'three'], mylist)
924  END
925  CheckScriptSuccess(lines)
926
927  # check all lines from heredoc are kept
928  lines =<< trim END
929      # comment 1
930      two
931      # comment 3
932
933      five
934      # comment 6
935  END
936  assert_equal(['# comment 1', 'two', '# comment 3', '', 'five', '# comment 6'], lines)
937
938  lines =<< trim END
939    [{
940      a: 0}]->string()->assert_equal("[{'a': 0}]")
941  END
942  CheckDefAndScriptSuccess(lines)
943enddef
944
945if has('channel')
946  let someJob = test_null_job()
947
948  def FuncWithError()
949    echomsg g:someJob
950  enddef
951
952  func Test_convert_emsg_to_exception()
953    try
954      call FuncWithError()
955    catch
956      call assert_match('Vim:E908:', v:exception)
957    endtry
958  endfunc
959endif
960
961let s:export_script_lines =<< trim END
962  vim9script
963  var name: string = 'bob'
964  def Concat(arg: string): string
965    return name .. arg
966  enddef
967  g:result = Concat('bie')
968  g:localname = name
969
970  export const CONST = 1234
971  export var exported = 9876
972  export var exp_name = 'John'
973  export def Exported(): string
974    return 'Exported'
975  enddef
976  export final theList = [1]
977END
978
979def Undo_export_script_lines()
980  unlet g:result
981  unlet g:localname
982enddef
983
984def Test_vim9_import_export()
985  var import_script_lines =<< trim END
986    vim9script
987    import {exported, Exported} from './Xexport.vim'
988    g:imported = exported
989    exported += 3
990    g:imported_added = exported
991    g:imported_func = Exported()
992
993    def GetExported(): string
994      var local_dict = {ref: Exported}
995      return local_dict.ref()
996    enddef
997    g:funcref_result = GetExported()
998
999    import {exp_name} from './Xexport.vim'
1000    g:imported_name = exp_name
1001    exp_name ..= ' Doe'
1002    g:imported_name_appended = exp_name
1003    g:imported_later = exported
1004
1005    import theList from './Xexport.vim'
1006    theList->add(2)
1007    assert_equal([1, 2], theList)
1008  END
1009
1010  writefile(import_script_lines, 'Ximport.vim')
1011  writefile(s:export_script_lines, 'Xexport.vim')
1012
1013  source Ximport.vim
1014
1015  assert_equal('bobbie', g:result)
1016  assert_equal('bob', g:localname)
1017  assert_equal(9876, g:imported)
1018  assert_equal(9879, g:imported_added)
1019  assert_equal(9879, g:imported_later)
1020  assert_equal('Exported', g:imported_func)
1021  assert_equal('Exported', g:funcref_result)
1022  assert_equal('John', g:imported_name)
1023  assert_equal('John Doe', g:imported_name_appended)
1024  assert_false(exists('g:name'))
1025
1026  Undo_export_script_lines()
1027  unlet g:imported
1028  unlet g:imported_added
1029  unlet g:imported_later
1030  unlet g:imported_func
1031  unlet g:imported_name g:imported_name_appended
1032  delete('Ximport.vim')
1033
1034  # similar, with line breaks
1035  var import_line_break_script_lines =<< trim END
1036    vim9script
1037    import {
1038        exported,
1039        Exported,
1040        }
1041        from
1042        './Xexport.vim'
1043    g:imported = exported
1044    exported += 5
1045    g:imported_added = exported
1046    g:imported_func = Exported()
1047  END
1048  writefile(import_line_break_script_lines, 'Ximport_lbr.vim')
1049  source Ximport_lbr.vim
1050
1051  assert_equal(9876, g:imported)
1052  assert_equal(9881, g:imported_added)
1053  assert_equal('Exported', g:imported_func)
1054
1055  # exported script not sourced again
1056  assert_false(exists('g:result'))
1057  unlet g:imported
1058  unlet g:imported_added
1059  unlet g:imported_func
1060  delete('Ximport_lbr.vim')
1061
1062  # import inside :def function
1063  var import_in_def_lines =<< trim END
1064    vim9script
1065    def ImportInDef()
1066      import exported from './Xexport.vim'
1067      g:imported = exported
1068      exported += 7
1069      g:imported_added = exported
1070    enddef
1071    ImportInDef()
1072  END
1073  writefile(import_in_def_lines, 'Ximport2.vim')
1074  source Ximport2.vim
1075  # TODO: this should be 9879
1076  assert_equal(9876, g:imported)
1077  assert_equal(9883, g:imported_added)
1078  unlet g:imported
1079  unlet g:imported_added
1080  delete('Ximport2.vim')
1081
1082  var import_star_as_lines =<< trim END
1083    vim9script
1084    import * as Export from './Xexport.vim'
1085    def UseExport()
1086      g:imported_def = Export.exported
1087    enddef
1088    g:imported_script = Export.exported
1089    assert_equal(1, exists('Export.exported'))
1090    assert_equal(0, exists('Export.notexported'))
1091    UseExport()
1092  END
1093  writefile(import_star_as_lines, 'Ximport.vim')
1094  source Ximport.vim
1095  assert_equal(9883, g:imported_def)
1096  assert_equal(9883, g:imported_script)
1097
1098  var import_star_as_lines_no_dot =<< trim END
1099    vim9script
1100    import * as Export from './Xexport.vim'
1101    def Func()
1102      var dummy = 1
1103      var imported = Export + dummy
1104    enddef
1105    defcompile
1106  END
1107  writefile(import_star_as_lines_no_dot, 'Ximport.vim')
1108  assert_fails('source Ximport.vim', 'E1060:', '', 2, 'Func')
1109
1110  var import_star_as_lines_dot_space =<< trim END
1111    vim9script
1112    import * as Export from './Xexport.vim'
1113    def Func()
1114      var imported = Export . exported
1115    enddef
1116    defcompile
1117  END
1118  writefile(import_star_as_lines_dot_space, 'Ximport.vim')
1119  assert_fails('source Ximport.vim', 'E1074:', '', 1, 'Func')
1120
1121  var import_star_as_duplicated =<< trim END
1122    vim9script
1123    import * as Export from './Xexport.vim'
1124    var some = 'other'
1125    import * as Export from './Xexport.vim'
1126    defcompile
1127  END
1128  writefile(import_star_as_duplicated, 'Ximport.vim')
1129  assert_fails('source Ximport.vim', 'E1073:', '', 4, 'Ximport.vim')
1130
1131  var import_star_as_lines_script_no_dot =<< trim END
1132    vim9script
1133    import * as Export from './Xexport.vim'
1134    g:imported_script = Export exported
1135  END
1136  writefile(import_star_as_lines_script_no_dot, 'Ximport.vim')
1137  assert_fails('source Ximport.vim', 'E1029:')
1138
1139  var import_star_as_lines_script_space_after_dot =<< trim END
1140    vim9script
1141    import * as Export from './Xexport.vim'
1142    g:imported_script = Export. exported
1143  END
1144  writefile(import_star_as_lines_script_space_after_dot, 'Ximport.vim')
1145  assert_fails('source Ximport.vim', 'E1074:')
1146
1147  var import_star_as_lines_missing_name =<< trim END
1148    vim9script
1149    import * as Export from './Xexport.vim'
1150    def Func()
1151      var imported = Export.
1152    enddef
1153    defcompile
1154  END
1155  writefile(import_star_as_lines_missing_name, 'Ximport.vim')
1156  assert_fails('source Ximport.vim', 'E1048:', '', 1, 'Func')
1157
1158  var import_star_as_lbr_lines =<< trim END
1159    vim9script
1160    import *
1161        as Export
1162        from
1163        './Xexport.vim'
1164    def UseExport()
1165      g:imported = Export.exported
1166    enddef
1167    UseExport()
1168  END
1169  writefile(import_star_as_lbr_lines, 'Ximport.vim')
1170  source Ximport.vim
1171  assert_equal(9883, g:imported)
1172
1173  var import_star_lines =<< trim END
1174    vim9script
1175    import * from './Xexport.vim'
1176  END
1177  writefile(import_star_lines, 'Ximport.vim')
1178  assert_fails('source Ximport.vim', 'E1045:', '', 2, 'Ximport.vim')
1179
1180  # try to import something that exists but is not exported
1181  var import_not_exported_lines =<< trim END
1182    vim9script
1183    import name from './Xexport.vim'
1184  END
1185  writefile(import_not_exported_lines, 'Ximport.vim')
1186  assert_fails('source Ximport.vim', 'E1049:', '', 2, 'Ximport.vim')
1187
1188  # try to import something that is already defined
1189  var import_already_defined =<< trim END
1190    vim9script
1191    var exported = 'something'
1192    import exported from './Xexport.vim'
1193  END
1194  writefile(import_already_defined, 'Ximport.vim')
1195  assert_fails('source Ximport.vim', 'E1054:', '', 3, 'Ximport.vim')
1196
1197  # try to import something that is already defined
1198  import_already_defined =<< trim END
1199    vim9script
1200    var exported = 'something'
1201    import * as exported from './Xexport.vim'
1202  END
1203  writefile(import_already_defined, 'Ximport.vim')
1204  assert_fails('source Ximport.vim', 'E1054:', '', 3, 'Ximport.vim')
1205
1206  # try to import something that is already defined
1207  import_already_defined =<< trim END
1208    vim9script
1209    var exported = 'something'
1210    import {exported} from './Xexport.vim'
1211  END
1212  writefile(import_already_defined, 'Ximport.vim')
1213  assert_fails('source Ximport.vim', 'E1054:', '', 3, 'Ximport.vim')
1214
1215  # try changing an imported const
1216  var import_assign_to_const =<< trim END
1217    vim9script
1218    import CONST from './Xexport.vim'
1219    def Assign()
1220      CONST = 987
1221    enddef
1222    defcompile
1223  END
1224  writefile(import_assign_to_const, 'Ximport.vim')
1225  assert_fails('source Ximport.vim', 'E46:', '', 1, '_Assign')
1226
1227  # try changing an imported final
1228  var import_assign_to_final =<< trim END
1229    vim9script
1230    import theList from './Xexport.vim'
1231    def Assign()
1232      theList = [2]
1233    enddef
1234    defcompile
1235  END
1236  writefile(import_assign_to_final, 'Ximport.vim')
1237  assert_fails('source Ximport.vim', 'E46:', '', 1, '_Assign')
1238
1239  # import a very long name, requires making a copy
1240  var import_long_name_lines =<< trim END
1241    vim9script
1242    import name012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789 from './Xexport.vim'
1243  END
1244  writefile(import_long_name_lines, 'Ximport.vim')
1245  assert_fails('source Ximport.vim', 'E1048:', '', 2, 'Ximport.vim')
1246
1247  var import_no_from_lines =<< trim END
1248    vim9script
1249    import name './Xexport.vim'
1250  END
1251  writefile(import_no_from_lines, 'Ximport.vim')
1252  assert_fails('source Ximport.vim', 'E1070:', '', 2, 'Ximport.vim')
1253
1254  var import_invalid_string_lines =<< trim END
1255    vim9script
1256    import name from Xexport.vim
1257  END
1258  writefile(import_invalid_string_lines, 'Ximport.vim')
1259  assert_fails('source Ximport.vim', 'E1071:', '', 2, 'Ximport.vim')
1260
1261  var import_wrong_name_lines =<< trim END
1262    vim9script
1263    import name from './XnoExport.vim'
1264  END
1265  writefile(import_wrong_name_lines, 'Ximport.vim')
1266  assert_fails('source Ximport.vim', 'E1053:', '', 2, 'Ximport.vim')
1267
1268  var import_missing_comma_lines =<< trim END
1269    vim9script
1270    import {exported name} from './Xexport.vim'
1271  END
1272  writefile(import_missing_comma_lines, 'Ximport3.vim')
1273  assert_fails('source Ximport3.vim', 'E1046:', '', 2, 'Ximport3.vim')
1274
1275  delete('Ximport.vim')
1276  delete('Ximport3.vim')
1277  delete('Xexport.vim')
1278
1279  # Check that in a Vim9 script 'cpo' is set to the Vim default.
1280  # Flags added or removed are also applied to the restored value.
1281  set cpo=abcd
1282  var lines =<< trim END
1283    vim9script
1284    g:cpo_in_vim9script = &cpo
1285    set cpo+=f
1286    set cpo-=c
1287    g:cpo_after_vim9script = &cpo
1288  END
1289  writefile(lines, 'Xvim9_script')
1290  source Xvim9_script
1291  assert_equal('fabd', &cpo)
1292  set cpo&vim
1293  assert_equal(&cpo, g:cpo_in_vim9script)
1294  var newcpo = substitute(&cpo, 'c', '', '') .. 'f'
1295  assert_equal(newcpo, g:cpo_after_vim9script)
1296
1297  delete('Xvim9_script')
1298enddef
1299
1300def Test_import_as()
1301  var export_lines =<< trim END
1302    vim9script
1303    export var one = 1
1304    export var yes = 'yes'
1305  END
1306  writefile(export_lines, 'XexportAs')
1307
1308  var import_lines =<< trim END
1309    vim9script
1310    var one = 'notused'
1311    var yes = 777
1312    import one as thatOne from './XexportAs'
1313    assert_equal(1, thatOne)
1314    import yes as yesYes from './XexportAs'
1315    assert_equal('yes', yesYes)
1316  END
1317  CheckScriptSuccess(import_lines)
1318
1319  import_lines =<< trim END
1320    vim9script
1321    import {one as thatOne, yes as yesYes} from './XexportAs'
1322    assert_equal(1, thatOne)
1323    assert_equal('yes', yesYes)
1324    assert_fails('echo one', 'E121:')
1325    assert_fails('echo yes', 'E121:')
1326  END
1327  CheckScriptSuccess(import_lines)
1328
1329  delete('XexportAs')
1330enddef
1331
1332func g:Trigger()
1333  source Ximport.vim
1334  return "echo 'yes'\<CR>"
1335endfunc
1336
1337def Test_import_export_expr_map()
1338  # check that :import and :export work when buffer is locked
1339  var export_lines =<< trim END
1340    vim9script
1341    export def That(): string
1342      return 'yes'
1343    enddef
1344  END
1345  writefile(export_lines, 'Xexport_that.vim')
1346
1347  var import_lines =<< trim END
1348    vim9script
1349    import That from './Xexport_that.vim'
1350    assert_equal('yes', That())
1351  END
1352  writefile(import_lines, 'Ximport.vim')
1353
1354  nnoremap <expr> trigger g:Trigger()
1355  feedkeys('trigger', "xt")
1356
1357  delete('Xexport_that.vim')
1358  delete('Ximport.vim')
1359  nunmap trigger
1360enddef
1361
1362def Test_import_in_filetype()
1363  # check that :import works when the buffer is locked
1364  mkdir('ftplugin', 'p')
1365  var export_lines =<< trim END
1366    vim9script
1367    export var That = 'yes'
1368  END
1369  writefile(export_lines, 'ftplugin/Xexport_ft.vim')
1370
1371  var import_lines =<< trim END
1372    vim9script
1373    import That from './Xexport_ft.vim'
1374    assert_equal('yes', That)
1375    g:did_load_mytpe = 1
1376  END
1377  writefile(import_lines, 'ftplugin/qf.vim')
1378
1379  var save_rtp = &rtp
1380  &rtp = getcwd() .. ',' .. &rtp
1381
1382  filetype plugin on
1383  copen
1384  assert_equal(1, g:did_load_mytpe)
1385
1386  quit!
1387  delete('Xexport_ft.vim')
1388  delete('ftplugin', 'rf')
1389  &rtp = save_rtp
1390enddef
1391
1392def Test_use_import_in_mapping()
1393  var lines =<< trim END
1394      vim9script
1395      export def Funcx()
1396        g:result = 42
1397      enddef
1398  END
1399  writefile(lines, 'XsomeExport.vim')
1400  lines =<< trim END
1401      vim9script
1402      import Funcx from './XsomeExport.vim'
1403      nnoremap <F3> :call <sid>Funcx()<cr>
1404  END
1405  writefile(lines, 'Xmapscript.vim')
1406
1407  source Xmapscript.vim
1408  feedkeys("\<F3>", "xt")
1409  assert_equal(42, g:result)
1410
1411  unlet g:result
1412  delete('XsomeExport.vim')
1413  delete('Xmapscript.vim')
1414  nunmap <F3>
1415enddef
1416
1417def Test_vim9script_mix()
1418  var lines =<< trim END
1419    if has(g:feature)
1420      " legacy script
1421      let g:legacy = 1
1422      finish
1423    endif
1424    vim9script
1425    g:legacy = 0
1426  END
1427  g:feature = 'eval'
1428  g:legacy = -1
1429  CheckScriptSuccess(lines)
1430  assert_equal(1, g:legacy)
1431
1432  g:feature = 'noteval'
1433  g:legacy = -1
1434  CheckScriptSuccess(lines)
1435  assert_equal(0, g:legacy)
1436enddef
1437
1438def Test_vim9script_fails()
1439  CheckScriptFailure(['scriptversion 2', 'vim9script'], 'E1039:')
1440  CheckScriptFailure(['vim9script', 'scriptversion 2'], 'E1040:')
1441  CheckScriptFailure(['export var some = 123'], 'E1042:')
1442  CheckScriptFailure(['import some from "./Xexport.vim"'], 'E1048:')
1443  CheckScriptFailure(['vim9script', 'export var g:some'], 'E1022:')
1444  CheckScriptFailure(['vim9script', 'export echo 134'], 'E1043:')
1445
1446  CheckScriptFailure(['vim9script', 'var str: string', 'str = 1234'], 'E1012:')
1447  CheckScriptFailure(['vim9script', 'const str = "asdf"', 'str = "xxx"'], 'E46:')
1448
1449  assert_fails('vim9script', 'E1038:')
1450  assert_fails('export something', 'E1043:')
1451enddef
1452
1453func Test_import_fails_without_script()
1454  CheckRunVimInTerminal
1455
1456  " call indirectly to avoid compilation error for missing functions
1457  call Run_Test_import_fails_on_command_line()
1458endfunc
1459
1460def Run_Test_import_fails_on_command_line()
1461  var export =<< trim END
1462    vim9script
1463    export def Foo(): number
1464        return 0
1465    enddef
1466  END
1467  writefile(export, 'XexportCmd.vim')
1468
1469  var buf = RunVimInTerminal('-c "import Foo from ''./XexportCmd.vim''"', {
1470                rows: 6, wait_for_ruler: 0})
1471  WaitForAssert(() => assert_match('^E1094:', term_getline(buf, 5)))
1472
1473  delete('XexportCmd.vim')
1474  StopVimInTerminal(buf)
1475enddef
1476
1477def Test_vim9script_reload_noclear()
1478  var lines =<< trim END
1479    vim9script
1480    export var exported = 'thexport'
1481  END
1482  writefile(lines, 'XExportReload')
1483  lines =<< trim END
1484    vim9script noclear
1485    g:loadCount += 1
1486    var s:reloaded = 'init'
1487    import exported from './XExportReload'
1488
1489    def Again(): string
1490      return 'again'
1491    enddef
1492
1493    if exists('s:loaded') | finish | endif
1494    var s:loaded = true
1495
1496    var s:notReloaded = 'yes'
1497    s:reloaded = 'first'
1498    def g:Values(): list<string>
1499      return [s:reloaded, s:notReloaded, Again(), Once(), exported]
1500    enddef
1501
1502    def Once(): string
1503      return 'once'
1504    enddef
1505  END
1506  writefile(lines, 'XReloaded')
1507  g:loadCount = 0
1508  source XReloaded
1509  assert_equal(1, g:loadCount)
1510  assert_equal(['first', 'yes', 'again', 'once', 'thexport'], g:Values())
1511  source XReloaded
1512  assert_equal(2, g:loadCount)
1513  assert_equal(['init', 'yes', 'again', 'once', 'thexport'], g:Values())
1514  source XReloaded
1515  assert_equal(3, g:loadCount)
1516  assert_equal(['init', 'yes', 'again', 'once', 'thexport'], g:Values())
1517
1518  delete('XReloaded')
1519  delete('XExportReload')
1520  delfunc g:Values
1521  unlet g:loadCount
1522enddef
1523
1524def Test_vim9script_reload_import()
1525  var lines =<< trim END
1526    vim9script
1527    const var = ''
1528    var valone = 1234
1529    def MyFunc(arg: string)
1530       valone = 5678
1531    enddef
1532  END
1533  var morelines =<< trim END
1534    var valtwo = 222
1535    export def GetValtwo(): number
1536      return valtwo
1537    enddef
1538  END
1539  writefile(lines + morelines, 'Xreload.vim')
1540  source Xreload.vim
1541  source Xreload.vim
1542  source Xreload.vim
1543
1544  var testlines =<< trim END
1545    vim9script
1546    def TheFunc()
1547      import GetValtwo from './Xreload.vim'
1548      assert_equal(222, GetValtwo())
1549    enddef
1550    TheFunc()
1551  END
1552  writefile(testlines, 'Ximport.vim')
1553  source Ximport.vim
1554
1555  # Test that when not using "morelines" GetValtwo() and valtwo are still
1556  # defined, because import doesn't reload a script.
1557  writefile(lines, 'Xreload.vim')
1558  source Ximport.vim
1559
1560  # cannot declare a var twice
1561  lines =<< trim END
1562    vim9script
1563    var valone = 1234
1564    var valone = 5678
1565  END
1566  writefile(lines, 'Xreload.vim')
1567  assert_fails('source Xreload.vim', 'E1041:', '', 3, 'Xreload.vim')
1568
1569  delete('Xreload.vim')
1570  delete('Ximport.vim')
1571enddef
1572
1573" if a script is reloaded with a script-local variable that changed its type, a
1574" compiled function using that variable must fail.
1575def Test_script_reload_change_type()
1576  var lines =<< trim END
1577    vim9script noclear
1578    var str = 'string'
1579    def g:GetStr(): string
1580      return str .. 'xxx'
1581    enddef
1582  END
1583  writefile(lines, 'Xreload.vim')
1584  source Xreload.vim
1585  echo g:GetStr()
1586
1587  lines =<< trim END
1588    vim9script noclear
1589    var str = 1234
1590  END
1591  writefile(lines, 'Xreload.vim')
1592  source Xreload.vim
1593  assert_fails('echo g:GetStr()', 'E1150:')
1594
1595  delfunc g:GetStr
1596  delete('Xreload.vim')
1597enddef
1598
1599" Define CallFunc so that the test can be compiled
1600command CallFunc echo 'nop'
1601
1602def Test_script_reload_from_function()
1603  var lines =<< trim END
1604      vim9script
1605
1606      if exists('g:loaded')
1607        finish
1608      endif
1609      g:loaded = 1
1610      delcommand CallFunc
1611      command CallFunc Func()
1612      def Func()
1613        so XreloadFunc.vim
1614        g:didTheFunc = 1
1615      enddef
1616  END
1617  writefile(lines, 'XreloadFunc.vim')
1618  source XreloadFunc.vim
1619  CallFunc
1620  assert_equal(1, g:didTheFunc)
1621
1622  delete('XreloadFunc.vim')
1623  delcommand CallFunc
1624  unlet g:loaded
1625  unlet g:didTheFunc
1626enddef
1627
1628def Test_script_var_shadows_function()
1629  var lines =<< trim END
1630      vim9script
1631      def Func(): number
1632        return 123
1633      enddef
1634      var Func = 1
1635  END
1636  CheckScriptFailure(lines, 'E1041:', 5)
1637enddef
1638
1639def s:RetSome(): string
1640  return 'some'
1641enddef
1642
1643" Not exported function that is referenced needs to be accessed by the
1644" script-local name.
1645def Test_vim9script_funcref()
1646  var sortlines =<< trim END
1647      vim9script
1648      def Compare(i1: number, i2: number): number
1649        return i2 - i1
1650      enddef
1651
1652      export def FastSort(): list<number>
1653        return range(5)->sort(Compare)
1654      enddef
1655
1656      export def GetString(arg: string): string
1657        return arg
1658      enddef
1659  END
1660  writefile(sortlines, 'Xsort.vim')
1661
1662  var lines =<< trim END
1663    vim9script
1664    import FastSort from './Xsort.vim'
1665    def Test()
1666      g:result = FastSort()
1667    enddef
1668    Test()
1669
1670    # using a function imported with "as"
1671    import * as anAlias from './Xsort.vim'
1672    assert_equal('yes', anAlias.GetString('yes'))
1673
1674    # using the function from a compiled function
1675    def TestMore(): string
1676      var s = s:anAlias.GetString('foo')
1677      return s .. anAlias.GetString('bar')
1678    enddef
1679    assert_equal('foobar', TestMore())
1680
1681    # error when using a function that isn't exported
1682    assert_fails('anAlias.Compare(1, 2)', 'E1049:')
1683  END
1684  writefile(lines, 'Xscript.vim')
1685
1686  source Xscript.vim
1687  assert_equal([4, 3, 2, 1, 0], g:result)
1688
1689  unlet g:result
1690  delete('Xsort.vim')
1691  delete('Xscript.vim')
1692
1693  var Funcref = function('s:RetSome')
1694  assert_equal('some', Funcref())
1695enddef
1696
1697" Check that when searching for "FilterFunc" it finds the import in the
1698" script where FastFilter() is called from, both as a string and as a direct
1699" function reference.
1700def Test_vim9script_funcref_other_script()
1701  var filterLines =<< trim END
1702    vim9script
1703    export def FilterFunc(idx: number, val: number): bool
1704      return idx % 2 == 1
1705    enddef
1706    export def FastFilter(): list<number>
1707      return range(10)->filter('FilterFunc')
1708    enddef
1709    export def FastFilterDirect(): list<number>
1710      return range(10)->filter(FilterFunc)
1711    enddef
1712  END
1713  writefile(filterLines, 'Xfilter.vim')
1714
1715  var lines =<< trim END
1716    vim9script
1717    import {FilterFunc, FastFilter, FastFilterDirect} from './Xfilter.vim'
1718    def Test()
1719      var x: list<number> = FastFilter()
1720    enddef
1721    Test()
1722    def TestDirect()
1723      var x: list<number> = FastFilterDirect()
1724    enddef
1725    TestDirect()
1726  END
1727  CheckScriptSuccess(lines)
1728  delete('Xfilter.vim')
1729enddef
1730
1731def Test_vim9script_reload_delfunc()
1732  var first_lines =<< trim END
1733    vim9script
1734    def FuncYes(): string
1735      return 'yes'
1736    enddef
1737  END
1738  var withno_lines =<< trim END
1739    def FuncNo(): string
1740      return 'no'
1741    enddef
1742    def g:DoCheck(no_exists: bool)
1743      assert_equal('yes', FuncYes())
1744      assert_equal('no', FuncNo())
1745    enddef
1746  END
1747  var nono_lines =<< trim END
1748    def g:DoCheck(no_exists: bool)
1749      assert_equal('yes', FuncYes())
1750      assert_fails('FuncNo()', 'E117:', '', 2, 'DoCheck')
1751    enddef
1752  END
1753
1754  # FuncNo() is defined
1755  writefile(first_lines + withno_lines, 'Xreloaded.vim')
1756  source Xreloaded.vim
1757  g:DoCheck(true)
1758
1759  # FuncNo() is not redefined
1760  writefile(first_lines + nono_lines, 'Xreloaded.vim')
1761  source Xreloaded.vim
1762  g:DoCheck(false)
1763
1764  # FuncNo() is back
1765  writefile(first_lines + withno_lines, 'Xreloaded.vim')
1766  source Xreloaded.vim
1767  g:DoCheck(false)
1768
1769  delete('Xreloaded.vim')
1770enddef
1771
1772def Test_vim9script_reload_delvar()
1773  # write the script with a script-local variable
1774  var lines =<< trim END
1775    vim9script
1776    var name = 'string'
1777  END
1778  writefile(lines, 'XreloadVar.vim')
1779  source XreloadVar.vim
1780
1781  # now write the script using the same variable locally - works
1782  lines =<< trim END
1783    vim9script
1784    def Func()
1785      var name = 'string'
1786    enddef
1787  END
1788  writefile(lines, 'XreloadVar.vim')
1789  source XreloadVar.vim
1790
1791  delete('XreloadVar.vim')
1792enddef
1793
1794def Test_import_absolute()
1795  var import_lines = [
1796        'vim9script',
1797        'import exported from "' .. escape(getcwd(), '\') .. '/Xexport_abs.vim"',
1798        'def UseExported()',
1799        '  g:imported_abs = exported',
1800        '  exported = 8888',
1801        '  g:imported_after = exported',
1802        'enddef',
1803        'UseExported()',
1804        'g:import_disassembled = execute("disass UseExported")',
1805        ]
1806  writefile(import_lines, 'Ximport_abs.vim')
1807  writefile(s:export_script_lines, 'Xexport_abs.vim')
1808
1809  source Ximport_abs.vim
1810
1811  assert_equal(9876, g:imported_abs)
1812  assert_equal(8888, g:imported_after)
1813  assert_match('<SNR>\d\+_UseExported\_s*' ..
1814          'g:imported_abs = exported\_s*' ..
1815          '0 LOADSCRIPT exported-2 from .*Xexport_abs.vim\_s*' ..
1816          '1 STOREG g:imported_abs\_s*' ..
1817          'exported = 8888\_s*' ..
1818          '2 PUSHNR 8888\_s*' ..
1819          '3 STORESCRIPT exported-2 in .*Xexport_abs.vim\_s*' ..
1820          'g:imported_after = exported\_s*' ..
1821          '4 LOADSCRIPT exported-2 from .*Xexport_abs.vim\_s*' ..
1822          '5 STOREG g:imported_after',
1823        g:import_disassembled)
1824
1825  Undo_export_script_lines()
1826  unlet g:imported_abs
1827  unlet g:import_disassembled
1828
1829  delete('Ximport_abs.vim')
1830  delete('Xexport_abs.vim')
1831enddef
1832
1833def Test_import_rtp()
1834  var import_lines = [
1835        'vim9script',
1836        'import exported from "Xexport_rtp.vim"',
1837        'g:imported_rtp = exported',
1838        ]
1839  writefile(import_lines, 'Ximport_rtp.vim')
1840  mkdir('import')
1841  writefile(s:export_script_lines, 'import/Xexport_rtp.vim')
1842
1843  var save_rtp = &rtp
1844  &rtp = getcwd()
1845  source Ximport_rtp.vim
1846  &rtp = save_rtp
1847
1848  assert_equal(9876, g:imported_rtp)
1849
1850  Undo_export_script_lines()
1851  unlet g:imported_rtp
1852  delete('Ximport_rtp.vim')
1853  delete('import', 'rf')
1854enddef
1855
1856def Test_import_compile_error()
1857  var export_lines = [
1858        'vim9script',
1859        'export def ExpFunc(): string',
1860        '  return notDefined',
1861        'enddef',
1862        ]
1863  writefile(export_lines, 'Xexported.vim')
1864
1865  var import_lines = [
1866        'vim9script',
1867        'import ExpFunc from "./Xexported.vim"',
1868        'def ImpFunc()',
1869        '  echo ExpFunc()',
1870        'enddef',
1871        'defcompile',
1872        ]
1873  writefile(import_lines, 'Ximport.vim')
1874
1875  try
1876    source Ximport.vim
1877  catch /E1001/
1878    # Error should be fore the Xexported.vim file.
1879    assert_match('E1001: Variable not found: notDefined', v:exception)
1880    assert_match('function <SNR>\d\+_ImpFunc\[1\]..<SNR>\d\+_ExpFunc, line 1', v:throwpoint)
1881  endtry
1882
1883  delete('Xexported.vim')
1884  delete('Ximport.vim')
1885enddef
1886
1887def Test_func_redefine_error()
1888  var lines = [
1889        'vim9script',
1890        'def Func()',
1891        '  eval [][0]',
1892        'enddef',
1893        'Func()',
1894        ]
1895  writefile(lines, 'Xtestscript.vim')
1896
1897  for count in range(3)
1898    try
1899      source Xtestscript.vim
1900    catch /E684/
1901      # function name should contain <SNR> every time
1902      assert_match('E684: list index out of range', v:exception)
1903      assert_match('function <SNR>\d\+_Func, line 1', v:throwpoint)
1904    endtry
1905  endfor
1906
1907  delete('Xtestscript.vim')
1908enddef
1909
1910def Test_func_overrules_import_fails()
1911  var export_lines =<< trim END
1912      vim9script
1913      export def Func()
1914        echo 'imported'
1915      enddef
1916  END
1917  writefile(export_lines, 'XexportedFunc.vim')
1918
1919  var lines =<< trim END
1920    vim9script
1921    import Func from './XexportedFunc.vim'
1922    def Func()
1923      echo 'local to function'
1924    enddef
1925  END
1926  CheckScriptFailure(lines, 'E1073:')
1927
1928  lines =<< trim END
1929    vim9script
1930    import Func from './XexportedFunc.vim'
1931    def Outer()
1932      def Func()
1933        echo 'local to function'
1934      enddef
1935    enddef
1936    defcompile
1937  END
1938  CheckScriptFailure(lines, 'E1073:')
1939
1940  delete('XexportedFunc.vim')
1941enddef
1942
1943def Test_func_redefine_fails()
1944  var lines =<< trim END
1945    vim9script
1946    def Func()
1947      echo 'one'
1948    enddef
1949    def Func()
1950      echo 'two'
1951    enddef
1952  END
1953  CheckScriptFailure(lines, 'E1073:')
1954
1955  lines =<< trim END
1956    vim9script
1957    def Foo(): string
1958      return 'foo'
1959      enddef
1960    def Func()
1961      var  Foo = {-> 'lambda'}
1962    enddef
1963    defcompile
1964  END
1965  CheckScriptFailure(lines, 'E1073:')
1966enddef
1967
1968def Test_fixed_size_list()
1969  # will be allocated as one piece of memory, check that changes work
1970  var l = [1, 2, 3, 4]
1971  l->remove(0)
1972  l->add(5)
1973  l->insert(99, 1)
1974  assert_equal([2, 99, 3, 4, 5], l)
1975enddef
1976
1977def Test_no_insert_xit()
1978  CheckDefExecFailure(['a = 1'], 'E1100:')
1979  CheckDefExecFailure(['c = 1'], 'E1100:')
1980  CheckDefExecFailure(['i = 1'], 'E1100:')
1981  CheckDefExecFailure(['t = 1'], 'E1100:')
1982  CheckDefExecFailure(['x = 1'], 'E1100:')
1983
1984  CheckScriptFailure(['vim9script', 'a = 1'], 'E488:')
1985  CheckScriptFailure(['vim9script', 'a'], 'E1100:')
1986  CheckScriptFailure(['vim9script', 'c = 1'], 'E488:')
1987  CheckScriptFailure(['vim9script', 'c'], 'E1100:')
1988  CheckScriptFailure(['vim9script', 'i = 1'], 'E488:')
1989  CheckScriptFailure(['vim9script', 'i'], 'E1100:')
1990  CheckScriptFailure(['vim9script', 'o = 1'], 'E1100:')
1991  CheckScriptFailure(['vim9script', 'o'], 'E1100:')
1992  CheckScriptFailure(['vim9script', 't'], 'E1100:')
1993  CheckScriptFailure(['vim9script', 't = 1'], 'E1100:')
1994  CheckScriptFailure(['vim9script', 'x = 1'], 'E1100:')
1995enddef
1996
1997def IfElse(what: number): string
1998  var res = ''
1999  if what == 1
2000    res = "one"
2001  elseif what == 2
2002    res = "two"
2003  else
2004    res = "three"
2005  endif
2006  return res
2007enddef
2008
2009def Test_if_elseif_else()
2010  assert_equal('one', IfElse(1))
2011  assert_equal('two', IfElse(2))
2012  assert_equal('three', IfElse(3))
2013enddef
2014
2015def Test_if_elseif_else_fails()
2016  CheckDefFailure(['elseif true'], 'E582:')
2017  CheckDefFailure(['else'], 'E581:')
2018  CheckDefFailure(['endif'], 'E580:')
2019  CheckDefFailure(['if g:abool', 'elseif xxx'], 'E1001:')
2020  CheckDefFailure(['if true', 'echo 1'], 'E171:')
2021
2022  var lines =<< trim END
2023      var s = ''
2024      if s = ''
2025      endif
2026  END
2027  CheckDefFailure(lines, 'E488:')
2028
2029  lines =<< trim END
2030      var s = ''
2031      if s == ''
2032      elseif s = ''
2033      endif
2034  END
2035  CheckDefFailure(lines, 'E488:')
2036enddef
2037
2038let g:bool_true = v:true
2039let g:bool_false = v:false
2040
2041def Test_if_const_expr()
2042  var res = false
2043  if true ? true : false
2044    res = true
2045  endif
2046  assert_equal(true, res)
2047
2048  g:glob = 2
2049  if false
2050    execute('g:glob = 3')
2051  endif
2052  assert_equal(2, g:glob)
2053  if true
2054    execute('g:glob = 3')
2055  endif
2056  assert_equal(3, g:glob)
2057
2058  res = false
2059  if g:bool_true ? true : false
2060    res = true
2061  endif
2062  assert_equal(true, res)
2063
2064  res = false
2065  if true ? g:bool_true : false
2066    res = true
2067  endif
2068  assert_equal(true, res)
2069
2070  res = false
2071  if true ? true : g:bool_false
2072    res = true
2073  endif
2074  assert_equal(true, res)
2075
2076  res = false
2077  if true ? false : true
2078    res = true
2079  endif
2080  assert_equal(false, res)
2081
2082  res = false
2083  if false ? false : true
2084    res = true
2085  endif
2086  assert_equal(true, res)
2087
2088  res = false
2089  if false ? true : false
2090    res = true
2091  endif
2092  assert_equal(false, res)
2093
2094  res = false
2095  if has('xyz') ? true : false
2096    res = true
2097  endif
2098  assert_equal(false, res)
2099
2100  res = false
2101  if true && true
2102    res = true
2103  endif
2104  assert_equal(true, res)
2105
2106  res = false
2107  if true && false
2108    res = true
2109  endif
2110  assert_equal(false, res)
2111
2112  res = false
2113  if g:bool_true && false
2114    res = true
2115  endif
2116  assert_equal(false, res)
2117
2118  res = false
2119  if true && g:bool_false
2120    res = true
2121  endif
2122  assert_equal(false, res)
2123
2124  res = false
2125  if false && false
2126    res = true
2127  endif
2128  assert_equal(false, res)
2129
2130  res = false
2131  if true || false
2132    res = true
2133  endif
2134  assert_equal(true, res)
2135
2136  res = false
2137  if g:bool_true || false
2138    res = true
2139  endif
2140  assert_equal(true, res)
2141
2142  res = false
2143  if true || g:bool_false
2144    res = true
2145  endif
2146  assert_equal(true, res)
2147
2148  res = false
2149  if false || false
2150    res = true
2151  endif
2152  assert_equal(false, res)
2153
2154  # with constant "false" expression may be invalid so long as the syntax is OK
2155  if false | eval 0 | endif
2156  if false | eval burp + 234 | endif
2157  if false | echo burp 234 'asd' | endif
2158  if false
2159    burp
2160  endif
2161enddef
2162
2163def Test_if_const_expr_fails()
2164  CheckDefFailure(['if "aaa" == "bbb'], 'E114:')
2165  CheckDefFailure(["if 'aaa' == 'bbb"], 'E115:')
2166  CheckDefFailure(["if has('aaa'"], 'E110:')
2167  CheckDefFailure(["if has('aaa') ? true false"], 'E109:')
2168enddef
2169
2170def RunNested(i: number): number
2171  var x: number = 0
2172  if i % 2
2173    if 1
2174      # comment
2175    else
2176      # comment
2177    endif
2178    x += 1
2179  else
2180    x += 1000
2181  endif
2182  return x
2183enddef
2184
2185def Test_nested_if()
2186  assert_equal(1, RunNested(1))
2187  assert_equal(1000, RunNested(2))
2188enddef
2189
2190def Test_execute_cmd()
2191  # missing argument is ignored
2192  execute
2193  execute # comment
2194
2195  new
2196  setline(1, 'default')
2197  execute 'setline(1, "execute-string")'
2198  assert_equal('execute-string', getline(1))
2199
2200  execute "setline(1, 'execute-string')"
2201  assert_equal('execute-string', getline(1))
2202
2203  var cmd1 = 'setline(1,'
2204  var cmd2 = '"execute-var")'
2205  execute cmd1 cmd2 # comment
2206  assert_equal('execute-var', getline(1))
2207
2208  execute cmd1 cmd2 '|setline(1, "execute-var-string")'
2209  assert_equal('execute-var-string', getline(1))
2210
2211  var cmd_first = 'call '
2212  var cmd_last = 'setline(1, "execute-var-var")'
2213  execute cmd_first .. cmd_last
2214  assert_equal('execute-var-var', getline(1))
2215  bwipe!
2216
2217  var n = true
2218  execute 'echomsg' (n ? '"true"' : '"no"')
2219  assert_match('^true$', Screenline(&lines))
2220
2221  echomsg [1, 2, 3] {a: 1, b: 2}
2222  assert_match('^\[1, 2, 3\] {''a'': 1, ''b'': 2}$', Screenline(&lines))
2223
2224  CheckDefFailure(['execute xxx'], 'E1001:', 1)
2225  CheckDefExecFailure(['execute "tabnext " .. 8'], 'E475:', 1)
2226  CheckDefFailure(['execute "cmd"# comment'], 'E488:', 1)
2227enddef
2228
2229def Test_execute_cmd_vimscript()
2230  # only checks line continuation
2231  var lines =<< trim END
2232      vim9script
2233      execute 'g:someVar'
2234                .. ' = ' ..
2235                   '28'
2236      assert_equal(28, g:someVar)
2237      unlet g:someVar
2238  END
2239  CheckScriptSuccess(lines)
2240enddef
2241
2242def Test_echo_cmd()
2243  echo 'some' # comment
2244  echon 'thing'
2245  assert_match('^something$', Screenline(&lines))
2246
2247  echo "some" # comment
2248  echon "thing"
2249  assert_match('^something$', Screenline(&lines))
2250
2251  var str1 = 'some'
2252  var str2 = 'more'
2253  echo str1 str2
2254  assert_match('^some more$', Screenline(&lines))
2255
2256  CheckDefFailure(['echo "xxx"# comment'], 'E488:')
2257enddef
2258
2259def Test_echomsg_cmd()
2260  echomsg 'some' 'more' # comment
2261  assert_match('^some more$', Screenline(&lines))
2262  echo 'clear'
2263  :1messages
2264  assert_match('^some more$', Screenline(&lines))
2265
2266  CheckDefFailure(['echomsg "xxx"# comment'], 'E488:')
2267enddef
2268
2269def Test_echomsg_cmd_vimscript()
2270  # only checks line continuation
2271  var lines =<< trim END
2272      vim9script
2273      echomsg 'here'
2274                .. ' is ' ..
2275                   'a message'
2276      assert_match('^here is a message$', Screenline(&lines))
2277  END
2278  CheckScriptSuccess(lines)
2279enddef
2280
2281def Test_echoerr_cmd()
2282  try
2283    echoerr 'something' 'wrong' # comment
2284  catch
2285    assert_match('something wrong', v:exception)
2286  endtry
2287enddef
2288
2289def Test_echoerr_cmd_vimscript()
2290  # only checks line continuation
2291  var lines =<< trim END
2292      vim9script
2293      try
2294        echoerr 'this'
2295                .. ' is ' ..
2296                   'wrong'
2297      catch
2298        assert_match('this is wrong', v:exception)
2299      endtry
2300  END
2301  CheckScriptSuccess(lines)
2302enddef
2303
2304def Test_for_outside_of_function()
2305  var lines =<< trim END
2306    vim9script
2307    new
2308    for var in range(0, 3)
2309      append(line('$'), var)
2310    endfor
2311    assert_equal(['', '0', '1', '2', '3'], getline(1, '$'))
2312    bwipe!
2313
2314    var result = ''
2315    for i in [1, 2, 3]
2316      var loop = ' loop ' .. i
2317      result ..= loop
2318    endfor
2319    assert_equal(' loop 1 loop 2 loop 3', result)
2320  END
2321  writefile(lines, 'Xvim9for.vim')
2322  source Xvim9for.vim
2323  delete('Xvim9for.vim')
2324enddef
2325
2326def Test_for_loop()
2327  var lines =<< trim END
2328      var result = ''
2329      for cnt in range(7)
2330        if cnt == 4
2331          break
2332        endif
2333        if cnt == 2
2334          continue
2335        endif
2336        result ..= cnt .. '_'
2337      endfor
2338      assert_equal('0_1_3_', result)
2339
2340      var concat = ''
2341      for str in eval('["one", "two"]')
2342        concat ..= str
2343      endfor
2344      assert_equal('onetwo', concat)
2345
2346      var total = 0
2347      for nr in
2348          [1, 2, 3]
2349        total += nr
2350      endfor
2351      assert_equal(6, total)
2352
2353      total = 0
2354      for nr
2355        in [1, 2, 3]
2356        total += nr
2357      endfor
2358      assert_equal(6, total)
2359
2360      total = 0
2361      for nr
2362        in
2363        [1, 2, 3]
2364        total += nr
2365      endfor
2366      assert_equal(6, total)
2367
2368      # with type
2369      total = 0
2370      for n: number in [1, 2, 3]
2371        total += n
2372      endfor
2373      assert_equal(6, total)
2374
2375      var chars = ''
2376      for s: string in 'foobar'
2377        chars ..= s
2378      endfor
2379      assert_equal('foobar', chars)
2380
2381      # unpack with type
2382      var res = ''
2383      for [n: number, s: string] in [[1, 'a'], [2, 'b']]
2384        res ..= n .. s
2385      endfor
2386      assert_equal('1a2b', res)
2387
2388      # unpack with one var
2389      var reslist = []
2390      for [x] in [['aaa'], ['bbb']]
2391        reslist->add(x)
2392      endfor
2393      assert_equal(['aaa', 'bbb'], reslist)
2394
2395      # loop over string
2396      res = ''
2397      for c in 'aéc̀d'
2398        res ..= c .. '-'
2399      endfor
2400      assert_equal('a-é-c̀-d-', res)
2401
2402      res = ''
2403      for c in ''
2404        res ..= c .. '-'
2405      endfor
2406      assert_equal('', res)
2407
2408      res = ''
2409      for c in test_null_string()
2410        res ..= c .. '-'
2411      endfor
2412      assert_equal('', res)
2413
2414      var foo: list<dict<any>> = [
2415              {a: 'Cat'}
2416            ]
2417      for dd in foo
2418        dd.counter = 12
2419      endfor
2420      assert_equal([{a: 'Cat', counter: 12}], foo)
2421  END
2422  CheckDefAndScriptSuccess(lines)
2423enddef
2424
2425def Test_for_loop_fails()
2426  CheckDefFailure(['for '], 'E1097:')
2427  CheckDefFailure(['for x'], 'E1097:')
2428  CheckDefFailure(['for x in'], 'E1097:')
2429  CheckDefFailure(['for # in range(5)'], 'E690:')
2430  CheckDefFailure(['for i In range(5)'], 'E690:')
2431  CheckDefFailure(['var x = 5', 'for x in range(5)'], 'E1017:')
2432  CheckScriptFailure(['def Func(arg: any)', 'for arg in range(5)', 'enddef', 'defcompile'], 'E1006:')
2433  delfunc! g:Func
2434  CheckDefFailure(['for i in xxx'], 'E1001:')
2435  CheckDefFailure(['endfor'], 'E588:')
2436  CheckDefFailure(['for i in range(3)', 'echo 3'], 'E170:')
2437
2438  # wrong type detected at compile time
2439  CheckDefFailure(['for i in {a: 1}', 'echo 3', 'endfor'], 'E1177: For loop on dict not supported')
2440
2441  # wrong type detected at runtime
2442  g:adict = {a: 1}
2443  CheckDefExecFailure(['for i in g:adict', 'echo 3', 'endfor'], 'E1177: For loop on dict not supported')
2444  unlet g:adict
2445
2446  var lines =<< trim END
2447      var d: list<dict<any>> = [{a: 0}]
2448      for e in d
2449        e = {a: 0, b: ''}
2450      endfor
2451  END
2452  CheckDefAndScriptFailure2(lines, 'E1018:', 'E46:', 3)
2453
2454  lines =<< trim END
2455      for nr: number in ['foo']
2456      endfor
2457  END
2458  CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected number but got string', 1)
2459enddef
2460
2461def Test_for_loop_script_var()
2462  # cannot use s:var in a :def function
2463  CheckDefFailure(['for s:var in range(3)', 'echo 3'], 'E1101:')
2464
2465  # can use s:var in Vim9 script, with or without s:
2466  var lines =<< trim END
2467    vim9script
2468    var total = 0
2469    for s:var in [1, 2, 3]
2470      total += s:var
2471    endfor
2472    assert_equal(6, total)
2473
2474    total = 0
2475    for var in [1, 2, 3]
2476      total += var
2477    endfor
2478    assert_equal(6, total)
2479  END
2480enddef
2481
2482def Test_for_loop_unpack()
2483  var lines =<< trim END
2484      var result = []
2485      for [v1, v2] in [[1, 2], [3, 4]]
2486        result->add(v1)
2487        result->add(v2)
2488      endfor
2489      assert_equal([1, 2, 3, 4], result)
2490
2491      result = []
2492      for [v1, v2; v3] in [[1, 2], [3, 4, 5, 6]]
2493        result->add(v1)
2494        result->add(v2)
2495        result->add(v3)
2496      endfor
2497      assert_equal([1, 2, [], 3, 4, [5, 6]], result)
2498
2499      result = []
2500      for [&ts, &sw] in [[1, 2], [3, 4]]
2501        result->add(&ts)
2502        result->add(&sw)
2503      endfor
2504      assert_equal([1, 2, 3, 4], result)
2505
2506      var slist: list<string>
2507      for [$LOOPVAR, @r, v:errmsg] in [['a', 'b', 'c'], ['d', 'e', 'f']]
2508        slist->add($LOOPVAR)
2509        slist->add(@r)
2510        slist->add(v:errmsg)
2511      endfor
2512      assert_equal(['a', 'b', 'c', 'd', 'e', 'f'], slist)
2513
2514      slist = []
2515      for [g:globalvar, b:bufvar, w:winvar, t:tabvar] in [['global', 'buf', 'win', 'tab'], ['1', '2', '3', '4']]
2516        slist->add(g:globalvar)
2517        slist->add(b:bufvar)
2518        slist->add(w:winvar)
2519        slist->add(t:tabvar)
2520      endfor
2521      assert_equal(['global', 'buf', 'win', 'tab', '1', '2', '3', '4'], slist)
2522      unlet! g:globalvar b:bufvar w:winvar t:tabvar
2523
2524      var res = []
2525      for [_, n, _] in [[1, 2, 3], [4, 5, 6]]
2526        res->add(n)
2527      endfor
2528      assert_equal([2, 5], res)
2529  END
2530  CheckDefAndScriptSuccess(lines)
2531
2532  lines =<< trim END
2533      for [v1, v2] in [[1, 2, 3], [3, 4]]
2534        echo v1 v2
2535      endfor
2536  END
2537  CheckDefExecFailure(lines, 'E710:', 1)
2538
2539  lines =<< trim END
2540      for [v1, v2] in [[1], [3, 4]]
2541        echo v1 v2
2542      endfor
2543  END
2544  CheckDefExecFailure(lines, 'E711:', 1)
2545
2546  lines =<< trim END
2547      for [v1, v1] in [[1, 2], [3, 4]]
2548        echo v1
2549      endfor
2550  END
2551  CheckDefExecFailure(lines, 'E1017:', 1)
2552enddef
2553
2554def Test_for_loop_with_try_continue()
2555  var lines =<< trim END
2556      var looped = 0
2557      var cleanup = 0
2558      for i in range(3)
2559        looped += 1
2560        try
2561          eval [][0]
2562        catch
2563          continue
2564        finally
2565          cleanup += 1
2566        endtry
2567      endfor
2568      assert_equal(3, looped)
2569      assert_equal(3, cleanup)
2570  END
2571  CheckDefAndScriptSuccess(lines)
2572enddef
2573
2574def Test_while_loop()
2575  var result = ''
2576  var cnt = 0
2577  while cnt < 555
2578    if cnt == 3
2579      break
2580    endif
2581    cnt += 1
2582    if cnt == 2
2583      continue
2584    endif
2585    result ..= cnt .. '_'
2586  endwhile
2587  assert_equal('1_3_', result)
2588
2589  var s = ''
2590  while s == 'x' # {comment}
2591  endwhile
2592enddef
2593
2594def Test_while_loop_fails()
2595  CheckDefFailure(['while xxx'], 'E1001:')
2596  CheckDefFailure(['endwhile'], 'E588:')
2597  CheckDefFailure(['continue'], 'E586:')
2598  CheckDefFailure(['if true', 'continue'], 'E586:')
2599  CheckDefFailure(['break'], 'E587:')
2600  CheckDefFailure(['if true', 'break'], 'E587:')
2601  CheckDefFailure(['while 1', 'echo 3'], 'E170:')
2602
2603  var lines =<< trim END
2604      var s = ''
2605      while s = ''
2606      endwhile
2607  END
2608  CheckDefFailure(lines, 'E488:')
2609enddef
2610
2611def Test_interrupt_loop()
2612  var caught = false
2613  var x = 0
2614  try
2615    while 1
2616      x += 1
2617      if x == 100
2618        feedkeys("\<C-C>", 'Lt')
2619      endif
2620    endwhile
2621  catch
2622    caught = true
2623    assert_equal(100, x)
2624  endtry
2625  assert_true(caught, 'should have caught an exception')
2626  # consume the CTRL-C
2627  getchar(0)
2628enddef
2629
2630def Test_automatic_line_continuation()
2631  var mylist = [
2632      'one',
2633      'two',
2634      'three',
2635      ] # comment
2636  assert_equal(['one', 'two', 'three'], mylist)
2637
2638  var mydict = {
2639      ['one']: 1,
2640      ['two']: 2,
2641      ['three']:
2642          3,
2643      } # comment
2644  assert_equal({one: 1, two: 2, three: 3}, mydict)
2645  mydict = {
2646      one: 1,  # comment
2647      two:     # comment
2648           2,  # comment
2649      three: 3 # comment
2650      }
2651  assert_equal({one: 1, two: 2, three: 3}, mydict)
2652  mydict = {
2653      one: 1,
2654      two:
2655           2,
2656      three: 3
2657      }
2658  assert_equal({one: 1, two: 2, three: 3}, mydict)
2659
2660  assert_equal(
2661        ['one', 'two', 'three'],
2662        split('one two three')
2663        )
2664enddef
2665
2666def Test_vim9_comment()
2667  CheckScriptSuccess([
2668      'vim9script',
2669      '# something',
2670      '#something',
2671      '#{something',
2672      ])
2673
2674  split Xfile
2675  CheckScriptSuccess([
2676      'vim9script',
2677      'edit #something',
2678      ])
2679  CheckScriptSuccess([
2680      'vim9script',
2681      'edit #{something',
2682      ])
2683  close
2684
2685  CheckScriptFailure([
2686      'vim9script',
2687      ':# something',
2688      ], 'E488:')
2689  CheckScriptFailure([
2690      '# something',
2691      ], 'E488:')
2692  CheckScriptFailure([
2693      ':# something',
2694      ], 'E488:')
2695
2696  { # block start
2697  } # block end
2698  CheckDefFailure([
2699      '{# comment',
2700      ], 'E488:')
2701  CheckDefFailure([
2702      '{',
2703      '}# comment',
2704      ], 'E488:')
2705
2706  echo "yes" # comment
2707  CheckDefFailure([
2708      'echo "yes"# comment',
2709      ], 'E488:')
2710  CheckScriptSuccess([
2711      'vim9script',
2712      'echo "yes" # something',
2713      ])
2714  CheckScriptFailure([
2715      'vim9script',
2716      'echo "yes"# something',
2717      ], 'E121:')
2718  CheckScriptFailure([
2719      'vim9script',
2720      'echo# something',
2721      ], 'E1144:')
2722  CheckScriptFailure([
2723      'echo "yes" # something',
2724      ], 'E121:')
2725
2726  exe "echo" # comment
2727  CheckDefFailure([
2728      'exe "echo"# comment',
2729      ], 'E488:')
2730  CheckScriptSuccess([
2731      'vim9script',
2732      'exe "echo" # something',
2733      ])
2734  CheckScriptFailure([
2735      'vim9script',
2736      'exe "echo"# something',
2737      ], 'E121:')
2738  CheckScriptFailure([
2739      'vim9script',
2740      'exe# something',
2741      ], 'E1144:')
2742  CheckScriptFailure([
2743      'exe "echo" # something',
2744      ], 'E121:')
2745
2746  CheckDefFailure([
2747      'try# comment',
2748      '  echo "yes"',
2749      'catch',
2750      'endtry',
2751      ], 'E1144:')
2752  CheckScriptFailure([
2753      'vim9script',
2754      'try# comment',
2755      'echo "yes"',
2756      ], 'E1144:')
2757  CheckDefFailure([
2758      'try',
2759      '  throw#comment',
2760      'catch',
2761      'endtry',
2762      ], 'E1144:')
2763  CheckDefFailure([
2764      'try',
2765      '  throw "yes"#comment',
2766      'catch',
2767      'endtry',
2768      ], 'E488:')
2769  CheckDefFailure([
2770      'try',
2771      '  echo "yes"',
2772      'catch# comment',
2773      'endtry',
2774      ], 'E1144:')
2775  CheckScriptFailure([
2776      'vim9script',
2777      'try',
2778      '  echo "yes"',
2779      'catch# comment',
2780      'endtry',
2781      ], 'E1144:')
2782  CheckDefFailure([
2783      'try',
2784      '  echo "yes"',
2785      'catch /pat/# comment',
2786      'endtry',
2787      ], 'E488:')
2788  CheckDefFailure([
2789      'try',
2790      'echo "yes"',
2791      'catch',
2792      'endtry# comment',
2793      ], 'E1144:')
2794  CheckScriptFailure([
2795      'vim9script',
2796      'try',
2797      '  echo "yes"',
2798      'catch',
2799      'endtry# comment',
2800      ], 'E1144:')
2801
2802  CheckScriptSuccess([
2803      'vim9script',
2804      'hi # comment',
2805      ])
2806  CheckScriptFailure([
2807      'vim9script',
2808      'hi# comment',
2809      ], 'E1144:')
2810  CheckScriptSuccess([
2811      'vim9script',
2812      'hi Search # comment',
2813      ])
2814  CheckScriptFailure([
2815      'vim9script',
2816      'hi Search# comment',
2817      ], 'E416:')
2818  CheckScriptSuccess([
2819      'vim9script',
2820      'hi link This Search # comment',
2821      ])
2822  CheckScriptFailure([
2823      'vim9script',
2824      'hi link This That# comment',
2825      ], 'E413:')
2826  CheckScriptSuccess([
2827      'vim9script',
2828      'hi clear This # comment',
2829      'hi clear # comment',
2830      ])
2831  # not tested, because it doesn't give an error but a warning:
2832  # hi clear This# comment',
2833  CheckScriptFailure([
2834      'vim9script',
2835      'hi clear# comment',
2836      ], 'E416:')
2837
2838  CheckScriptSuccess([
2839      'vim9script',
2840      'hi Group term=bold',
2841      'match Group /todo/ # comment',
2842      ])
2843  CheckScriptFailure([
2844      'vim9script',
2845      'hi Group term=bold',
2846      'match Group /todo/# comment',
2847      ], 'E488:')
2848  CheckScriptSuccess([
2849      'vim9script',
2850      'match # comment',
2851      ])
2852  CheckScriptFailure([
2853      'vim9script',
2854      'match# comment',
2855      ], 'E1144:')
2856  CheckScriptSuccess([
2857      'vim9script',
2858      'match none # comment',
2859      ])
2860  CheckScriptFailure([
2861      'vim9script',
2862      'match none# comment',
2863      ], 'E475:')
2864
2865  CheckScriptSuccess([
2866      'vim9script',
2867      'menutrans clear # comment',
2868      ])
2869  CheckScriptFailure([
2870      'vim9script',
2871      'menutrans clear# comment text',
2872      ], 'E474:')
2873
2874  CheckScriptSuccess([
2875      'vim9script',
2876      'syntax clear # comment',
2877      ])
2878  CheckScriptFailure([
2879      'vim9script',
2880      'syntax clear# comment text',
2881      ], 'E28:')
2882  CheckScriptSuccess([
2883      'vim9script',
2884      'syntax keyword Word some',
2885      'syntax clear Word # comment',
2886      ])
2887  CheckScriptFailure([
2888      'vim9script',
2889      'syntax keyword Word some',
2890      'syntax clear Word# comment text',
2891      ], 'E28:')
2892
2893  CheckScriptSuccess([
2894      'vim9script',
2895      'syntax list # comment',
2896      ])
2897  CheckScriptFailure([
2898      'vim9script',
2899      'syntax list# comment text',
2900      ], 'E28:')
2901
2902  CheckScriptSuccess([
2903      'vim9script',
2904      'syntax match Word /pat/ oneline # comment',
2905      ])
2906  CheckScriptFailure([
2907      'vim9script',
2908      'syntax match Word /pat/ oneline# comment',
2909      ], 'E475:')
2910
2911  CheckScriptSuccess([
2912      'vim9script',
2913      'syntax keyword Word word # comm[ent',
2914      ])
2915  CheckScriptFailure([
2916      'vim9script',
2917      'syntax keyword Word word# comm[ent',
2918      ], 'E789:')
2919
2920  CheckScriptSuccess([
2921      'vim9script',
2922      'syntax match Word /pat/ # comment',
2923      ])
2924  CheckScriptFailure([
2925      'vim9script',
2926      'syntax match Word /pat/# comment',
2927      ], 'E402:')
2928
2929  CheckScriptSuccess([
2930      'vim9script',
2931      'syntax match Word /pat/ contains=Something # comment',
2932      ])
2933  CheckScriptFailure([
2934      'vim9script',
2935      'syntax match Word /pat/ contains=Something# comment',
2936      ], 'E475:')
2937  CheckScriptFailure([
2938      'vim9script',
2939      'syntax match Word /pat/ contains= # comment',
2940      ], 'E406:')
2941  CheckScriptFailure([
2942      'vim9script',
2943      'syntax match Word /pat/ contains=# comment',
2944      ], 'E475:')
2945
2946  CheckScriptSuccess([
2947      'vim9script',
2948      'syntax region Word start=/pat/ end=/pat/ # comment',
2949      ])
2950  CheckScriptFailure([
2951      'vim9script',
2952      'syntax region Word start=/pat/ end=/pat/# comment',
2953      ], 'E402:')
2954
2955  CheckScriptSuccess([
2956      'vim9script',
2957      'syntax sync # comment',
2958      ])
2959  CheckScriptFailure([
2960      'vim9script',
2961      'syntax sync# comment',
2962      ], 'E404:')
2963  CheckScriptSuccess([
2964      'vim9script',
2965      'syntax sync ccomment # comment',
2966      ])
2967  CheckScriptFailure([
2968      'vim9script',
2969      'syntax sync ccomment# comment',
2970      ], 'E404:')
2971
2972  CheckScriptSuccess([
2973      'vim9script',
2974      'syntax cluster Some contains=Word # comment',
2975      ])
2976  CheckScriptFailure([
2977      'vim9script',
2978      'syntax cluster Some contains=Word# comment',
2979      ], 'E475:')
2980
2981  CheckScriptSuccess([
2982      'vim9script',
2983      'command Echo echo # comment',
2984      'command Echo # comment',
2985      'delcommand Echo',
2986      ])
2987  CheckScriptFailure([
2988      'vim9script',
2989      'command Echo echo# comment',
2990      'Echo',
2991      ], 'E1144:')
2992  delcommand Echo
2993
2994  var curdir = getcwd()
2995  CheckScriptSuccess([
2996      'command Echo cd " comment',
2997      'Echo',
2998      'delcommand Echo',
2999      ])
3000  CheckScriptSuccess([
3001      'vim9script',
3002      'command Echo cd # comment',
3003      'Echo',
3004      'delcommand Echo',
3005      ])
3006  CheckScriptFailure([
3007      'vim9script',
3008      'command Echo cd " comment',
3009      'Echo',
3010      ], 'E344:')
3011  delcommand Echo
3012  chdir(curdir)
3013
3014  CheckScriptFailure([
3015      'vim9script',
3016      'command Echo# comment',
3017      ], 'E182:')
3018  CheckScriptFailure([
3019      'vim9script',
3020      'command Echo echo',
3021      'command Echo# comment',
3022      ], 'E182:')
3023  delcommand Echo
3024
3025  CheckScriptSuccess([
3026      'vim9script',
3027      'function # comment',
3028      ])
3029  CheckScriptFailure([
3030      'vim9script',
3031      'function " comment',
3032      ], 'E129:')
3033  CheckScriptFailure([
3034      'vim9script',
3035      'function# comment',
3036      ], 'E1144:')
3037  CheckScriptSuccess([
3038      'vim9script',
3039      'function CheckScriptSuccess # comment',
3040      ])
3041  CheckScriptFailure([
3042      'vim9script',
3043      'function CheckScriptSuccess# comment',
3044      ], 'E488:')
3045
3046  CheckScriptSuccess([
3047      'vim9script',
3048      'func g:DeleteMeA()',
3049      'endfunc',
3050      'delfunction g:DeleteMeA # comment',
3051      ])
3052  CheckScriptFailure([
3053      'vim9script',
3054      'func g:DeleteMeB()',
3055      'endfunc',
3056      'delfunction g:DeleteMeB# comment',
3057      ], 'E488:')
3058
3059  CheckScriptSuccess([
3060      'vim9script',
3061      'call execute("ls") # comment',
3062      ])
3063  CheckScriptFailure([
3064      'vim9script',
3065      'call execute("ls")# comment',
3066      ], 'E488:')
3067
3068  CheckScriptFailure([
3069      'def Test() " comment',
3070      'enddef',
3071      ], 'E488:')
3072  CheckScriptFailure([
3073      'vim9script',
3074      'def Test() " comment',
3075      'enddef',
3076      ], 'E488:')
3077
3078  CheckScriptSuccess([
3079      'func Test() " comment',
3080      'endfunc',
3081      'delfunc Test',
3082      ])
3083  CheckScriptSuccess([
3084      'vim9script',
3085      'func Test() " comment',
3086      'endfunc',
3087      ])
3088
3089  CheckScriptSuccess([
3090      'def Test() # comment',
3091      'enddef',
3092      ])
3093  CheckScriptFailure([
3094      'func Test() # comment',
3095      'endfunc',
3096      ], 'E488:')
3097
3098  var lines =<< trim END
3099      vim9script
3100      syn region Text
3101      \ start='foo'
3102      #\ comment
3103      \ end='bar'
3104      syn region Text start='foo'
3105      #\ comment
3106      \ end='bar'
3107  END
3108  CheckScriptSuccess(lines)
3109
3110  lines =<< trim END
3111      vim9script
3112      syn region Text
3113      \ start='foo'
3114      "\ comment
3115      \ end='bar'
3116  END
3117  CheckScriptFailure(lines, 'E399:')
3118enddef
3119
3120def Test_vim9_comment_gui()
3121  CheckCanRunGui
3122
3123  CheckScriptFailure([
3124      'vim9script',
3125      'gui#comment'
3126      ], 'E1144:')
3127  CheckScriptFailure([
3128      'vim9script',
3129      'gui -f#comment'
3130      ], 'E499:')
3131enddef
3132
3133def Test_vim9_comment_not_compiled()
3134  au TabEnter *.vim g:entered = 1
3135  au TabEnter *.x g:entered = 2
3136
3137  edit test.vim
3138  doautocmd TabEnter #comment
3139  assert_equal(1, g:entered)
3140
3141  doautocmd TabEnter f.x
3142  assert_equal(2, g:entered)
3143
3144  g:entered = 0
3145  doautocmd TabEnter f.x #comment
3146  assert_equal(2, g:entered)
3147
3148  assert_fails('doautocmd Syntax#comment', 'E216:')
3149
3150  au! TabEnter
3151  unlet g:entered
3152
3153  CheckScriptSuccess([
3154      'vim9script',
3155      'g:var = 123',
3156      'b:var = 456',
3157      'w:var = 777',
3158      't:var = 888',
3159      'unlet g:var w:var # something',
3160      ])
3161
3162  CheckScriptFailure([
3163      'vim9script',
3164      'let var = 123',
3165      ], 'E1126: Cannot use :let in Vim9 script')
3166
3167  CheckScriptFailure([
3168      'vim9script',
3169      'var g:var = 123',
3170      ], 'E1016: Cannot declare a global variable:')
3171
3172  CheckScriptFailure([
3173      'vim9script',
3174      'var b:var = 123',
3175      ], 'E1016: Cannot declare a buffer variable:')
3176
3177  CheckScriptFailure([
3178      'vim9script',
3179      'var w:var = 123',
3180      ], 'E1016: Cannot declare a window variable:')
3181
3182  CheckScriptFailure([
3183      'vim9script',
3184      'var t:var = 123',
3185      ], 'E1016: Cannot declare a tab variable:')
3186
3187  CheckScriptFailure([
3188      'vim9script',
3189      'var v:version = 123',
3190      ], 'E1016: Cannot declare a v: variable:')
3191
3192  CheckScriptFailure([
3193      'vim9script',
3194      'var $VARIABLE = "text"',
3195      ], 'E1016: Cannot declare an environment variable:')
3196
3197  CheckScriptFailure([
3198      'vim9script',
3199      'g:var = 123',
3200      'unlet g:var# comment1',
3201      ], 'E108:')
3202
3203  CheckScriptFailure([
3204      'let g:var = 123',
3205      'unlet g:var # something',
3206      ], 'E488:')
3207
3208  CheckScriptSuccess([
3209      'vim9script',
3210      'if 1 # comment2',
3211      '  echo "yes"',
3212      'elseif 2 #comment',
3213      '  echo "no"',
3214      'endif',
3215      ])
3216
3217  CheckScriptFailure([
3218      'vim9script',
3219      'if 1# comment3',
3220      '  echo "yes"',
3221      'endif',
3222      ], 'E488:')
3223
3224  CheckScriptFailure([
3225      'vim9script',
3226      'if 0 # comment4',
3227      '  echo "yes"',
3228      'elseif 2#comment',
3229      '  echo "no"',
3230      'endif',
3231      ], 'E488:')
3232
3233  CheckScriptSuccess([
3234      'vim9script',
3235      'var v = 1 # comment5',
3236      ])
3237
3238  CheckScriptFailure([
3239      'vim9script',
3240      'var v = 1# comment6',
3241      ], 'E488:')
3242
3243  CheckScriptSuccess([
3244      'vim9script',
3245      'new'
3246      'setline(1, ["# define pat", "last"])',
3247      ':$',
3248      'dsearch /pat/ #comment',
3249      'bwipe!',
3250      ])
3251
3252  CheckScriptFailure([
3253      'vim9script',
3254      'new'
3255      'setline(1, ["# define pat", "last"])',
3256      ':$',
3257      'dsearch /pat/#comment',
3258      'bwipe!',
3259      ], 'E488:')
3260
3261  CheckScriptFailure([
3262      'vim9script',
3263      'func! SomeFunc()',
3264      ], 'E477:')
3265enddef
3266
3267def Test_finish()
3268  var lines =<< trim END
3269    vim9script
3270    g:res = 'one'
3271    if v:false | finish | endif
3272    g:res = 'two'
3273    finish
3274    g:res = 'three'
3275  END
3276  writefile(lines, 'Xfinished')
3277  source Xfinished
3278  assert_equal('two', g:res)
3279
3280  unlet g:res
3281  delete('Xfinished')
3282enddef
3283
3284def Test_forward_declaration()
3285  var lines =<< trim END
3286    vim9script
3287    def GetValue(): string
3288      return theVal
3289    enddef
3290    var theVal = 'something'
3291    g:initVal = GetValue()
3292    theVal = 'else'
3293    g:laterVal = GetValue()
3294  END
3295  writefile(lines, 'Xforward')
3296  source Xforward
3297  assert_equal('something', g:initVal)
3298  assert_equal('else', g:laterVal)
3299
3300  unlet g:initVal
3301  unlet g:laterVal
3302  delete('Xforward')
3303enddef
3304
3305def Test_source_vim9_from_legacy()
3306  var vim9_lines =<< trim END
3307    vim9script
3308    var local = 'local'
3309    g:global = 'global'
3310    export var exported = 'exported'
3311    export def GetText(): string
3312       return 'text'
3313    enddef
3314  END
3315  writefile(vim9_lines, 'Xvim9_script.vim')
3316
3317  var legacy_lines =<< trim END
3318    source Xvim9_script.vim
3319
3320    call assert_false(exists('local'))
3321    call assert_false(exists('exported'))
3322    call assert_false(exists('s:exported'))
3323    call assert_equal('global', global)
3324    call assert_equal('global', g:global)
3325
3326    " imported variable becomes script-local
3327    import exported from './Xvim9_script.vim'
3328    call assert_equal('exported', s:exported)
3329    call assert_false(exists('exported'))
3330
3331    " imported function becomes script-local
3332    import GetText from './Xvim9_script.vim'
3333    call assert_equal('text', s:GetText())
3334    call assert_false(exists('*GetText'))
3335  END
3336  writefile(legacy_lines, 'Xlegacy_script.vim')
3337
3338  source Xlegacy_script.vim
3339  assert_equal('global', g:global)
3340  unlet g:global
3341
3342  delete('Xlegacy_script.vim')
3343  delete('Xvim9_script.vim')
3344enddef
3345
3346def Test_declare_script_in_func()
3347  var lines =<< trim END
3348      vim9script
3349      func Declare()
3350        let s:local = 123
3351      endfunc
3352      Declare()
3353      assert_equal(123, local)
3354
3355      var error: string
3356      try
3357        local = 'asdf'
3358      catch
3359        error = v:exception
3360      endtry
3361      assert_match('E1012: Type mismatch; expected number but got string', error)
3362
3363      lockvar local
3364      try
3365        local = 999
3366      catch
3367        error = v:exception
3368      endtry
3369      assert_match('E741: Value is locked: local', error)
3370  END
3371  CheckScriptSuccess(lines)
3372enddef
3373
3374
3375func Test_vim9script_not_global()
3376  " check that items defined in Vim9 script are script-local, not global
3377  let vim9lines =<< trim END
3378    vim9script
3379    var name = 'local'
3380    func TheFunc()
3381      echo 'local'
3382    endfunc
3383    def DefFunc()
3384      echo 'local'
3385    enddef
3386  END
3387  call writefile(vim9lines, 'Xvim9script.vim')
3388  source Xvim9script.vim
3389  try
3390    echo g:var
3391    assert_report('did not fail')
3392  catch /E121:/
3393    " caught
3394  endtry
3395  try
3396    call TheFunc()
3397    assert_report('did not fail')
3398  catch /E117:/
3399    " caught
3400  endtry
3401  try
3402    call DefFunc()
3403    assert_report('did not fail')
3404  catch /E117:/
3405    " caught
3406  endtry
3407
3408  call delete('Xvim9script.vim')
3409endfunc
3410
3411def Test_vim9_copen()
3412  # this was giving an error for setting w:quickfix_title
3413  copen
3414  quit
3415enddef
3416
3417" test using an auto-loaded function and variable
3418def Test_vim9_autoload()
3419  var lines =<< trim END
3420     vim9script
3421     def some#gettest(): string
3422       return 'test'
3423     enddef
3424     g:some#name = 'name'
3425     g:some#dict = {key: 'value'}
3426
3427     def some#varargs(a1: string, ...l: list<string>): string
3428       return a1 .. l[0] .. l[1]
3429     enddef
3430  END
3431
3432  mkdir('Xdir/autoload', 'p')
3433  writefile(lines, 'Xdir/autoload/some.vim')
3434  var save_rtp = &rtp
3435  exe 'set rtp^=' .. getcwd() .. '/Xdir'
3436
3437  assert_equal('test', g:some#gettest())
3438  assert_equal('name', g:some#name)
3439  assert_equal('value', g:some#dict.key)
3440  g:some#other = 'other'
3441  assert_equal('other', g:some#other)
3442
3443  assert_equal('abc', some#varargs('a', 'b', 'c'))
3444
3445  # upper case script name works
3446  lines =<< trim END
3447     vim9script
3448     def Other#getOther(): string
3449       return 'other'
3450     enddef
3451  END
3452  writefile(lines, 'Xdir/autoload/Other.vim')
3453  assert_equal('other', g:Other#getOther())
3454
3455  delete('Xdir', 'rf')
3456  &rtp = save_rtp
3457enddef
3458
3459" test using a vim9script that is auto-loaded from an autocmd
3460def Test_vim9_aucmd_autoload()
3461  var lines =<< trim END
3462     vim9script
3463     def foo#test()
3464         echomsg getreg('"')
3465     enddef
3466  END
3467
3468  mkdir('Xdir/autoload', 'p')
3469  writefile(lines, 'Xdir/autoload/foo.vim')
3470  var save_rtp = &rtp
3471  exe 'set rtp^=' .. getcwd() .. '/Xdir'
3472  augroup test
3473    autocmd TextYankPost * call foo#test()
3474  augroup END
3475
3476  normal Y
3477
3478  augroup test
3479    autocmd!
3480  augroup END
3481  delete('Xdir', 'rf')
3482  &rtp = save_rtp
3483enddef
3484
3485" This was causing a crash because suppress_errthrow wasn't reset.
3486def Test_vim9_autoload_error()
3487  var lines =<< trim END
3488      vim9script
3489      def crash#func()
3490          try
3491              for x in List()
3492              endfor
3493          catch
3494          endtry
3495          g:ok = true
3496      enddef
3497      fu List()
3498          invalid
3499      endfu
3500      try
3501          alsoinvalid
3502      catch /wontmatch/
3503      endtry
3504  END
3505  call mkdir('Xruntime/autoload', 'p')
3506  call writefile(lines, 'Xruntime/autoload/crash.vim')
3507
3508  # run in a separate Vim to avoid the side effects of assert_fails()
3509  lines =<< trim END
3510    exe 'set rtp^=' .. getcwd() .. '/Xruntime'
3511    call crash#func()
3512    call writefile(['ok'], 'Xdidit')
3513    qall!
3514  END
3515  writefile(lines, 'Xscript')
3516  RunVim([], [], '-S Xscript')
3517  assert_equal(['ok'], readfile('Xdidit'))
3518
3519  delete('Xdidit')
3520  delete('Xscript')
3521  delete('Xruntime', 'rf')
3522
3523  lines =<< trim END
3524    vim9script
3525    var foo#bar = 'asdf'
3526  END
3527  CheckScriptFailure(lines, 'E461: Illegal variable name: foo#bar', 2)
3528enddef
3529
3530def Test_script_var_in_autocmd()
3531  # using a script variable from an autocommand, defined in a :def function in a
3532  # legacy Vim script, cannot check the variable type.
3533  var lines =<< trim END
3534    let s:counter = 1
3535    def s:Func()
3536      au! CursorHold
3537      au CursorHold * s:counter += 1
3538    enddef
3539    call s:Func()
3540    doau CursorHold
3541    call assert_equal(2, s:counter)
3542    au! CursorHold
3543  END
3544  CheckScriptSuccess(lines)
3545enddef
3546
3547def Test_cmdline_win()
3548  # if the Vim syntax highlighting uses Vim9 constructs they can be used from
3549  # the command line window.
3550  mkdir('rtp/syntax', 'p')
3551  var export_lines =<< trim END
3552    vim9script
3553    export var That = 'yes'
3554  END
3555  writefile(export_lines, 'rtp/syntax/Xexport.vim')
3556  var import_lines =<< trim END
3557    vim9script
3558    import That from './Xexport.vim'
3559  END
3560  writefile(import_lines, 'rtp/syntax/vim.vim')
3561  var save_rtp = &rtp
3562  &rtp = getcwd() .. '/rtp' .. ',' .. &rtp
3563  syntax on
3564  augroup CmdWin
3565    autocmd CmdwinEnter * g:got_there = 'yes'
3566  augroup END
3567  # this will open and also close the cmdline window
3568  feedkeys('q:', 'xt')
3569  assert_equal('yes', g:got_there)
3570
3571  augroup CmdWin
3572    au!
3573  augroup END
3574  &rtp = save_rtp
3575  delete('rtp', 'rf')
3576enddef
3577
3578def Test_invalid_sid()
3579  assert_fails('func <SNR>1234_func', 'E123:')
3580
3581  if RunVim([], ['wq! Xdidit'], '+"func <SNR>1_func"')
3582    assert_equal([], readfile('Xdidit'))
3583  endif
3584  delete('Xdidit')
3585enddef
3586
3587def Test_restoring_cpo()
3588  writefile(['vim9script', 'set nocp'], 'Xsourced')
3589  writefile(['call writefile(["done"], "Xdone")', 'quit!'], 'Xclose')
3590  if RunVim([], [], '-u NONE +"set cpo+=a" -S Xsourced -S Xclose')
3591    assert_equal(['done'], readfile('Xdone'))
3592  endif
3593  delete('Xsourced')
3594  delete('Xclose')
3595  delete('Xdone')
3596
3597  writefile(['vim9script'], 'XanotherScript')
3598  set cpo=aABceFsMny>
3599  edit XanotherScript
3600  so %
3601  assert_equal('aABceFsMny>', &cpo)
3602  :1del
3603  w
3604  so %
3605  assert_equal('aABceFsMny>', &cpo)
3606
3607  delete('XanotherScript')
3608  set cpo&vim
3609enddef
3610
3611" Use :function so we can use Check commands
3612func Test_no_redraw_when_restoring_cpo()
3613  CheckScreendump
3614  CheckFeature timers
3615
3616  let lines =<< trim END
3617    vim9script
3618    def script#func()
3619    enddef
3620  END
3621  call mkdir('Xdir/autoload', 'p')
3622  call writefile(lines, 'Xdir/autoload/script.vim')
3623
3624  let lines =<< trim END
3625      vim9script
3626      set cpo+=M
3627      exe 'set rtp^=' .. getcwd() .. '/Xdir'
3628      au CmdlineEnter : ++once timer_start(0, (_) => script#func())
3629      setline(1, 'some text')
3630  END
3631  call writefile(lines, 'XTest_redraw_cpo')
3632  let buf = RunVimInTerminal('-S XTest_redraw_cpo', {'rows': 6})
3633  call term_sendkeys(buf, "V:")
3634  call VerifyScreenDump(buf, 'Test_vim9_no_redraw', {})
3635
3636  " clean up
3637  call term_sendkeys(buf, "\<Esc>u")
3638  call StopVimInTerminal(buf)
3639  call delete('XTest_redraw_cpo')
3640  call delete('Xdir', 'rf')
3641endfunc
3642
3643
3644def Test_unset_any_variable()
3645  var lines =<< trim END
3646    var name: any
3647    assert_equal(0, name)
3648  END
3649  CheckDefAndScriptSuccess(lines)
3650enddef
3651
3652func Test_define_func_at_command_line()
3653  CheckRunVimInTerminal
3654
3655  " call indirectly to avoid compilation error for missing functions
3656  call Run_Test_define_func_at_command_line()
3657endfunc
3658
3659def Run_Test_define_func_at_command_line()
3660  # run in a separate Vim instance to avoid the script context
3661  var lines =<< trim END
3662    func CheckAndQuit()
3663      call assert_fails('call Afunc()', 'E117: Unknown function: Bfunc')
3664      call writefile(['errors: ' .. string(v:errors)], 'Xdidcmd')
3665    endfunc
3666  END
3667  writefile([''], 'Xdidcmd')
3668  writefile(lines, 'XcallFunc')
3669  var buf = RunVimInTerminal('-S XcallFunc', {rows: 6})
3670  # define Afunc() on the command line
3671  term_sendkeys(buf, ":def Afunc()\<CR>Bfunc()\<CR>enddef\<CR>")
3672  term_sendkeys(buf, ":call CheckAndQuit()\<CR>")
3673  WaitForAssert(() => assert_equal(['errors: []'], readfile('Xdidcmd')))
3674
3675  call StopVimInTerminal(buf)
3676  delete('XcallFunc')
3677  delete('Xdidcmd')
3678enddef
3679
3680def Test_script_var_scope()
3681  var lines =<< trim END
3682      vim9script
3683      if true
3684        if true
3685          var one = 'one'
3686          echo one
3687        endif
3688        echo one
3689      endif
3690  END
3691  CheckScriptFailure(lines, 'E121:', 7)
3692
3693  lines =<< trim END
3694      vim9script
3695      if true
3696        if false
3697          var one = 'one'
3698          echo one
3699        else
3700          var one = 'one'
3701          echo one
3702        endif
3703        echo one
3704      endif
3705  END
3706  CheckScriptFailure(lines, 'E121:', 10)
3707
3708  lines =<< trim END
3709      vim9script
3710      while true
3711        var one = 'one'
3712        echo one
3713        break
3714      endwhile
3715      echo one
3716  END
3717  CheckScriptFailure(lines, 'E121:', 7)
3718
3719  lines =<< trim END
3720      vim9script
3721      for i in range(1)
3722        var one = 'one'
3723        echo one
3724      endfor
3725      echo one
3726  END
3727  CheckScriptFailure(lines, 'E121:', 6)
3728
3729  lines =<< trim END
3730      vim9script
3731      {
3732        var one = 'one'
3733        assert_equal('one', one)
3734      }
3735      assert_false(exists('one'))
3736      assert_false(exists('s:one'))
3737  END
3738  CheckScriptSuccess(lines)
3739
3740  lines =<< trim END
3741      vim9script
3742      {
3743        var one = 'one'
3744        echo one
3745      }
3746      echo one
3747  END
3748  CheckScriptFailure(lines, 'E121:', 6)
3749enddef
3750
3751def Test_catch_exception_in_callback()
3752  var lines =<< trim END
3753    vim9script
3754    def Callback(...l: list<any>)
3755      try
3756        var x: string
3757        var y: string
3758        # this error should be caught with CHECKLEN
3759        [x, y] = ['']
3760      catch
3761        g:caught = 'yes'
3762      endtry
3763    enddef
3764    popup_menu('popup', {callback: Callback})
3765    feedkeys("\r", 'xt')
3766  END
3767  CheckScriptSuccess(lines)
3768
3769  unlet g:caught
3770enddef
3771
3772def Test_no_unknown_error_after_error()
3773  if !has('unix') || !has('job')
3774    throw 'Skipped: not unix of missing +job feature'
3775  endif
3776  var lines =<< trim END
3777      vim9script
3778      var source: list<number>
3779      def Out_cb(...l: list<any>)
3780          eval [][0]
3781      enddef
3782      def Exit_cb(...l: list<any>)
3783          sleep 1m
3784          source += l
3785      enddef
3786      var myjob = job_start('echo burp', {out_cb: Out_cb, exit_cb: Exit_cb, mode: 'raw'})
3787      while job_status(myjob) == 'run'
3788        sleep 10m
3789      endwhile
3790      # wait for Exit_cb() to be called
3791      sleep 200m
3792  END
3793  writefile(lines, 'Xdef')
3794  assert_fails('so Xdef', ['E684:', 'E1012:'])
3795  delete('Xdef')
3796enddef
3797
3798def InvokeNormal()
3799  exe "norm! :m+1\r"
3800enddef
3801
3802def Test_invoke_normal_in_visual_mode()
3803  xnoremap <F3> <Cmd>call <SID>InvokeNormal()<CR>
3804  new
3805  setline(1, ['aaa', 'bbb'])
3806  feedkeys("V\<F3>", 'xt')
3807  assert_equal(['bbb', 'aaa'], getline(1, 2))
3808  xunmap <F3>
3809enddef
3810
3811def Test_white_space_after_command()
3812  var lines =<< trim END
3813    exit_cb: Func})
3814  END
3815  CheckDefAndScriptFailure(lines, 'E1144:', 1)
3816
3817  lines =<< trim END
3818    e#
3819  END
3820  CheckDefAndScriptFailure(lines, 'E1144:', 1)
3821enddef
3822
3823def Test_script_var_gone_when_sourced_twice()
3824  var lines =<< trim END
3825      vim9script
3826      if exists('g:guard')
3827        finish
3828      endif
3829      g:guard = 1
3830      var name = 'thename'
3831      def g:GetName(): string
3832        return name
3833      enddef
3834      def g:SetName(arg: string)
3835        name = arg
3836      enddef
3837  END
3838  writefile(lines, 'XscriptTwice.vim')
3839  so XscriptTwice.vim
3840  assert_equal('thename', g:GetName())
3841  g:SetName('newname')
3842  assert_equal('newname', g:GetName())
3843  so XscriptTwice.vim
3844  assert_fails('call g:GetName()', 'E1149:')
3845  assert_fails('call g:SetName("x")', 'E1149:')
3846
3847  delfunc g:GetName
3848  delfunc g:SetName
3849  delete('XscriptTwice.vim')
3850  unlet g:guard
3851enddef
3852
3853def Test_import_gone_when_sourced_twice()
3854  var exportlines =<< trim END
3855      vim9script
3856      if exists('g:guard')
3857        finish
3858      endif
3859      g:guard = 1
3860      export var name = 'someName'
3861  END
3862  writefile(exportlines, 'XexportScript.vim')
3863
3864  var lines =<< trim END
3865      vim9script
3866      import name from './XexportScript.vim'
3867      def g:GetName(): string
3868        return name
3869      enddef
3870  END
3871  writefile(lines, 'XscriptImport.vim')
3872  so XscriptImport.vim
3873  assert_equal('someName', g:GetName())
3874
3875  so XexportScript.vim
3876  assert_fails('call g:GetName()', 'E1149:')
3877
3878  delfunc g:GetName
3879  delete('XexportScript.vim')
3880  delete('XscriptImport.vim')
3881  unlet g:guard
3882enddef
3883
3884def Test_unsupported_commands()
3885  var lines =<< trim END
3886      ka
3887  END
3888  CheckDefFailure(lines, 'E476:')
3889  CheckScriptFailure(['vim9script'] + lines, 'E492:')
3890
3891  lines =<< trim END
3892      :1ka
3893  END
3894  CheckDefFailure(lines, 'E476:')
3895  CheckScriptFailure(['vim9script'] + lines, 'E492:')
3896
3897  lines =<< trim END
3898    t
3899  END
3900  CheckDefFailure(lines, 'E1100:')
3901  CheckScriptFailure(['vim9script'] + lines, 'E1100:')
3902
3903  lines =<< trim END
3904    x
3905  END
3906  CheckDefFailure(lines, 'E1100:')
3907  CheckScriptFailure(['vim9script'] + lines, 'E1100:')
3908
3909  lines =<< trim END
3910    xit
3911  END
3912  CheckDefFailure(lines, 'E1100:')
3913  CheckScriptFailure(['vim9script'] + lines, 'E1100:')
3914enddef
3915
3916def Test_mapping_line_number()
3917  var lines =<< trim END
3918      vim9script
3919      def g:FuncA()
3920          # Some comment
3921          FuncB(0)
3922      enddef
3923          # Some comment
3924      def FuncB(
3925          # Some comment
3926          n: number
3927      )
3928          exe 'nno '
3929              # Some comment
3930              .. '<F3> a'
3931              .. 'b'
3932              .. 'c'
3933      enddef
3934  END
3935  CheckScriptSuccess(lines)
3936  var res = execute('verbose nmap <F3>')
3937  assert_match('No mapping found', res)
3938
3939  g:FuncA()
3940  res = execute('verbose nmap <F3>')
3941  assert_match(' <F3> .* abc.*Last set from .*XScriptSuccess\d\+ line 11', res)
3942
3943  nunmap <F3>
3944  delfunc g:FuncA
3945enddef
3946
3947def Test_option_modifier()
3948  var lines =<< trim END
3949      set hlsearch &  hlsearch  !
3950      call assert_equal(1, &hlsearch)
3951  END
3952  CheckScriptSuccess(lines)
3953
3954  lines =<< trim END
3955      vim9script
3956      set hlsearch &
3957  END
3958  CheckScriptFailure(lines, 'E518:')
3959
3960  lines =<< trim END
3961      vim9script
3962      set hlsearch &  hlsearch  !
3963  END
3964  CheckScriptFailure(lines, 'E518:')
3965enddef
3966
3967" Keep this last, it messes up highlighting.
3968def Test_substitute_cmd()
3969  new
3970  setline(1, 'something')
3971  :substitute(some(other(
3972  assert_equal('otherthing', getline(1))
3973  bwipe!
3974
3975  # also when the context is Vim9 script
3976  var lines =<< trim END
3977    vim9script
3978    new
3979    setline(1, 'something')
3980    :substitute(some(other(
3981    assert_equal('otherthing', getline(1))
3982    bwipe!
3983  END
3984  writefile(lines, 'Xvim9lines')
3985  source Xvim9lines
3986
3987  delete('Xvim9lines')
3988enddef
3989
3990" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker
3991