blob: 1cf1eb069981c07362f31049f56670c5dc76bdb9 [file] [log] [blame]
Bram Moolenaard8448622022-01-07 21:39:52 +00001" Test import/export of the Vim9 script language.
Bram Moolenaar160aa862022-01-10 21:29:57 +00002" Also the autoload mechanism.
Bram Moolenaard8448622022-01-07 21:39:52 +00003
4source check.vim
5source term_util.vim
6source vim9.vim
7
8let s:export_script_lines =<< trim END
9 vim9script
10 var name: string = 'bob'
11 def Concat(arg: string): string
12 return name .. arg
13 enddef
14 g:result = Concat('bie')
15 g:localname = name
16
17 export const CONST = 1234
18 export var exported = 9876
19 export var exp_name = 'John'
20 export def Exported(): string
21 return 'Exported'
22 enddef
23 export def ExportedValue(): number
24 return exported
25 enddef
26 export def ExportedInc()
27 exported += 5
28 enddef
29 export final theList = [1]
Bram Moolenaar857c8bb2022-01-15 21:08:19 +000030 export def AddSome(s: string): string
31 return s .. 'some'
32 enddef
33 export var AddRef = AddSome
Bram Moolenaard8448622022-01-07 21:39:52 +000034END
35
36def Undo_export_script_lines()
37 unlet g:result
38 unlet g:localname
39enddef
40
41def Test_vim9_import_export()
42 writefile(s:export_script_lines, 'Xexport.vim')
43 var import_script_lines =<< trim END
44 vim9script
45 var dir = './'
46 var ext = ".vim"
47 import dir .. 'Xexport' .. ext as expo
48
49 g:exported1 = expo.exported
50 expo.exported += 3
51 g:exported2 = expo.exported
52 g:exported3 = expo.ExportedValue()
53
54 expo.ExportedInc()
55 g:exported_i1 = expo.exported
56 g:exported_i2 = expo.ExportedValue()
57
58 expo.exported = 11
59 g:exported_s1 = expo.exported
60 g:exported_s2 = expo.ExportedValue()
61
62 g:imported_func = expo.Exported()
63
64 def GetExported(): string
65 var local_dict = {ref: expo.Exported}
66 return local_dict.ref()
67 enddef
68 g:funcref_result = GetExported()
69
Bram Moolenaar21f0d6c2022-01-20 17:35:49 +000070 def GetName(): string
71 return expo.exp_name .. 'son'
72 enddef
73 g:long_name = GetName()
74
Bram Moolenaard8448622022-01-07 21:39:52 +000075 g:imported_name = expo.exp_name
76 expo.exp_name ..= ' Doe'
Bram Moolenaar47036b62022-01-16 21:18:53 +000077 expo.exp_name = expo.exp_name .. ' Maar'
Bram Moolenaard8448622022-01-07 21:39:52 +000078 g:imported_name_appended = expo.exp_name
79 g:exported_later = expo.exported
80
81 expo.theList->add(2)
82 assert_equal([1, 2], expo.theList)
Bram Moolenaar857c8bb2022-01-15 21:08:19 +000083
84 assert_equal('andthensome', 'andthen'->expo.AddSome())
85 assert_equal('awesome', 'awe'->expo.AddRef())
Bram Moolenaard8448622022-01-07 21:39:52 +000086 END
87 writefile(import_script_lines, 'Ximport.vim')
88 source Ximport.vim
89
90 assert_equal('bobbie', g:result)
91 assert_equal('bob', g:localname)
92 assert_equal(9876, g:exported1)
93 assert_equal(9879, g:exported2)
94 assert_equal(9879, g:exported3)
95
96 assert_equal(9884, g:exported_i1)
97 assert_equal(9884, g:exported_i2)
98
99 assert_equal(11, g:exported_s1)
100 assert_equal(11, g:exported_s2)
101 assert_equal(11, g:exported_later)
102
103 assert_equal('Exported', g:imported_func)
104 assert_equal('Exported', g:funcref_result)
105 assert_equal('John', g:imported_name)
Bram Moolenaar21f0d6c2022-01-20 17:35:49 +0000106 assert_equal('Johnson', g:long_name)
Bram Moolenaar47036b62022-01-16 21:18:53 +0000107 assert_equal('John Doe Maar', g:imported_name_appended)
Bram Moolenaard8448622022-01-07 21:39:52 +0000108 assert_false(exists('g:name'))
109
110 Undo_export_script_lines()
111 unlet g:exported1
112 unlet g:exported2
113 unlet g:exported3
114 unlet g:exported_i1
115 unlet g:exported_i2
116 unlet g:exported_later
117 unlet g:imported_func
Bram Moolenaar21f0d6c2022-01-20 17:35:49 +0000118 unlet g:imported_name g:long_name g:imported_name_appended
Bram Moolenaard8448622022-01-07 21:39:52 +0000119 delete('Ximport.vim')
120
121 # similar, with line breaks
122 var import_line_break_script_lines =<< trim END
123 vim9script
124 import './Xexport.vim'
125 as expo
126 g:exported = expo.exported
127 expo.exported += 7
128 g:exported_added = expo.exported
129 g:imported_func = expo.Exported()
130 END
131 writefile(import_line_break_script_lines, 'Ximport_lbr.vim')
132 source Ximport_lbr.vim
133
134 assert_equal(11, g:exported)
135 assert_equal(18, g:exported_added)
136 assert_equal('Exported', g:imported_func)
137
138 # exported script not sourced again
139 assert_false(exists('g:result'))
140 unlet g:exported
141 unlet g:exported_added
142 unlet g:imported_func
143 delete('Ximport_lbr.vim')
144
145 var line_break_before_dot =<< trim END
146 vim9script
147 import './Xexport.vim' as expo
148 g:exported = expo
149 .exported
150 END
151 writefile(line_break_before_dot, 'Ximport_lbr_before_dot.vim')
152 assert_fails('source Ximport_lbr_before_dot.vim', 'E1060:', '', 3)
153 delete('Ximport_lbr_before_dot.vim')
154
155 var line_break_after_dot =<< trim END
156 vim9script
157 import './Xexport.vim' as expo
158 g:exported = expo.
159 exported
160 END
161 writefile(line_break_after_dot, 'Ximport_lbr_after_dot.vim')
162 assert_fails('source Ximport_lbr_after_dot.vim', 'E1074:', '', 3)
163 delete('Ximport_lbr_after_dot.vim')
164
165 var import_star_as_lines =<< trim END
166 vim9script
167 import './Xexport.vim' as Export
168 def UseExport()
169 g:exported_def = Export.exported
170 enddef
171 g:exported_script = Export.exported
172 assert_equal(1, exists('Export.exported'))
173 assert_equal(0, exists('Export.notexported'))
174 UseExport()
175 END
176 writefile(import_star_as_lines, 'Ximport.vim')
177 source Ximport.vim
178
179 assert_equal(18, g:exported_def)
180 assert_equal(18, g:exported_script)
181 unlet g:exported_def
182 unlet g:exported_script
183
184 var import_star_as_lines_no_dot =<< trim END
185 vim9script
186 import './Xexport.vim' as Export
187 def Func()
188 var dummy = 1
189 var imported = Export + dummy
190 enddef
191 defcompile
192 END
193 writefile(import_star_as_lines_no_dot, 'Ximport.vim')
194 assert_fails('source Ximport.vim', 'E1060:', '', 2, 'Func')
195
196 var import_star_as_lines_dot_space =<< trim END
197 vim9script
198 import './Xexport.vim' as Export
199 def Func()
200 var imported = Export . exported
201 enddef
202 defcompile
203 END
204 writefile(import_star_as_lines_dot_space, 'Ximport.vim')
205 assert_fails('source Ximport.vim', 'E1074:', '', 1, 'Func')
206
Bram Moolenaar7c24dfd2022-01-08 17:03:55 +0000207 writefile(s:export_script_lines, 'Xexport2.vim')
208 var import_as_duplicated =<< trim END
Bram Moolenaard8448622022-01-07 21:39:52 +0000209 vim9script
210 import './Xexport.vim' as expo
Bram Moolenaar7c24dfd2022-01-08 17:03:55 +0000211 import './Xexport2.vim' as expo
Bram Moolenaard8448622022-01-07 21:39:52 +0000212 END
Bram Moolenaar7c24dfd2022-01-08 17:03:55 +0000213 writefile(import_as_duplicated, 'Ximport.vim')
Bram Moolenaard8448622022-01-07 21:39:52 +0000214 assert_fails('source Ximport.vim', 'E1073:', '', 3, 'Ximport.vim')
Bram Moolenaar7c24dfd2022-01-08 17:03:55 +0000215 delete('Xexport2.vim')
Bram Moolenaard8448622022-01-07 21:39:52 +0000216
217 var import_star_as_lines_script_no_dot =<< trim END
218 vim9script
219 import './Xexport.vim' as Export
220 g:imported_script = Export exported
221 END
222 writefile(import_star_as_lines_script_no_dot, 'Ximport.vim')
223 assert_fails('source Ximport.vim', 'E1060: Expected dot after name: Export exported')
224
225 var import_star_as_lines_script_space_after_dot =<< trim END
226 vim9script
227 import './Xexport.vim' as Export
228 g:imported_script = Export. exported
229 END
230 writefile(import_star_as_lines_script_space_after_dot, 'Ximport.vim')
231 assert_fails('source Ximport.vim', 'E1074:')
232
233 var import_star_as_lines_missing_name =<< trim END
234 vim9script
235 import './Xexport.vim' as Export
236 def Func()
237 var imported = Export.
238 enddef
239 defcompile
240 END
241 writefile(import_star_as_lines_missing_name, 'Ximport.vim')
242 assert_fails('source Ximport.vim', 'E1048:', '', 1, 'Func')
243
244 var import_star_as_lbr_lines =<< trim END
245 vim9script
246 import './Xexport.vim'
247 as Export
248 def UseExport()
249 g:exported = Export.exported
250 enddef
251 UseExport()
252 END
253 writefile(import_star_as_lbr_lines, 'Ximport.vim')
254 source Ximport.vim
255 assert_equal(18, g:exported)
256 unlet g:exported
257
258 # try to use something that exists but is not exported
259 var import_not_exported_lines =<< trim END
260 vim9script
261 import './Xexport.vim' as expo
262 echo expo.name
263 END
264 writefile(import_not_exported_lines, 'Ximport.vim')
265 assert_fails('source Ximport.vim', 'E1049:', '', 3, 'Ximport.vim')
266
267 # try to import something that is already defined
268 var import_already_defined =<< trim END
269 vim9script
270 var exported = 'something'
271 import './Xexport.vim' as exported
272 END
273 writefile(import_already_defined, 'Ximport.vim')
274 assert_fails('source Ximport.vim', 'E1054:', '', 3, 'Ximport.vim')
275
276 # try changing an imported const
277 var import_assign_to_const =<< trim END
278 vim9script
279 import './Xexport.vim' as expo
280 def Assign()
281 expo.CONST = 987
282 enddef
283 defcompile
284 END
285 writefile(import_assign_to_const, 'Ximport.vim')
286 assert_fails('source Ximport.vim', 'E46:', '', 1, '_Assign')
287
288 # try changing an imported final
289 var import_assign_to_final =<< trim END
290 vim9script
291 import './Xexport.vim' as expo
292 def Assign()
293 expo.theList = [2]
294 enddef
295 defcompile
296 END
297 writefile(import_assign_to_final, 'Ximport.vim')
298 assert_fails('source Ximport.vim', 'E46:', '', 1, '_Assign')
299
300 var import_no_as_lines =<< trim END
301 vim9script
302 import './Xexport.vim' name
303 END
304 writefile(import_no_as_lines, 'Ximport.vim')
305 assert_fails('source Ximport.vim', 'E488:', '', 2, 'Ximport.vim')
306
307 var import_invalid_string_lines =<< trim END
308 vim9script
309 import Xexport.vim
310 END
311 writefile(import_invalid_string_lines, 'Ximport.vim')
312 assert_fails('source Ximport.vim', 'E121:', '', 2, 'Ximport.vim')
313
314 var import_wrong_name_lines =<< trim END
315 vim9script
316 import './XnoExport.vim'
317 END
318 writefile(import_wrong_name_lines, 'Ximport.vim')
319 assert_fails('source Ximport.vim', 'E1053:', '', 2, 'Ximport.vim')
320
321 var import_redefining_lines =<< trim END
322 vim9script
323 import './Xexport.vim' as exported
324 var exported = 5
325 END
326 writefile(import_redefining_lines, 'Ximport.vim')
327 assert_fails('source Ximport.vim', 'E1213: Redefining imported item "exported"', '', 3)
328
Bram Moolenaar160aa862022-01-10 21:29:57 +0000329 var import_missing_dot_lines =<< trim END
330 vim9script
331 import './Xexport.vim' as expo
332 def Test()
333 expo = 9
334 enddef
335 defcompile
336 END
337 writefile(import_missing_dot_lines, 'Ximport.vim')
338 assert_fails('source Ximport.vim', 'E1258:', '', 1)
339
340 var import_missing_name_lines =<< trim END
341 vim9script
342 import './Xexport.vim' as expo
343 def Test()
344 expo.99 = 9
345 enddef
346 defcompile
347 END
348 writefile(import_missing_name_lines, 'Ximport.vim')
Bram Moolenaar76283822022-01-10 21:39:03 +0000349 assert_fails('source Ximport.vim', 'E1259:', '', 1)
Bram Moolenaar160aa862022-01-10 21:29:57 +0000350
Bram Moolenaard8448622022-01-07 21:39:52 +0000351 var import_assign_wrong_type_lines =<< trim END
352 vim9script
353 import './Xexport.vim' as expo
354 expo.exported = 'xxx'
355 END
356 writefile(import_assign_wrong_type_lines, 'Ximport.vim')
357 assert_fails('source Ximport.vim', 'E1012: Type mismatch; expected number but got string', '', 3)
358
359 var import_assign_const_lines =<< trim END
360 vim9script
361 import './Xexport.vim' as expo
362 expo.CONST = 4321
363 END
364 writefile(import_assign_const_lines, 'Ximport.vim')
365 assert_fails('source Ximport.vim', 'E46: Cannot change read-only variable "CONST"', '', 3)
366
367 delete('Ximport.vim')
368 delete('Ximport3.vim')
369 delete('Xexport.vim')
370
371 # Check that in a Vim9 script 'cpo' is set to the Vim default.
372 # Flags added or removed are also applied to the restored value.
373 set cpo=abcd
374 var lines =<< trim END
375 vim9script
376 g:cpo_in_vim9script = &cpo
377 set cpo+=f
378 set cpo-=c
379 g:cpo_after_vim9script = &cpo
380 END
381 writefile(lines, 'Xvim9_script')
382 source Xvim9_script
383 assert_equal('fabd', &cpo)
384 set cpo&vim
385 assert_equal(&cpo, g:cpo_in_vim9script)
386 var newcpo = substitute(&cpo, 'c', '', '') .. 'f'
387 assert_equal(newcpo, g:cpo_after_vim9script)
388
389 delete('Xvim9_script')
390enddef
391
392def Test_import_funcref()
393 var lines =<< trim END
394 vim9script
395 export def F(): number
396 return 42
397 enddef
398 export const G = F
399 END
400 writefile(lines, 'Xlib.vim')
401
402 lines =<< trim END
403 vim9script
404 import './Xlib.vim' as lib
405 const Foo = lib.G()
406 assert_equal(42, Foo)
407
408 def DoTest()
409 const Goo = lib.G()
410 assert_equal(42, Goo)
411 enddef
412 DoTest()
413 END
414 CheckScriptSuccess(lines)
415
416 delete('Xlib.vim')
417enddef
418
419def Test_import_fails()
420 writefile([], 'Xfoo.vim')
421 var lines =<< trim END
422 import './Xfoo.vim' as foo
423 foo = 'bar'
424 END
425 CheckDefAndScriptFailure(lines, ['E1094:', 'E1236: Cannot use foo itself'])
426 lines =<< trim END
427 vim9script
428 import './Xfoo.vim' as foo
429 var that = foo
430 END
431 CheckScriptFailure(lines, 'E1060: Expected dot after name: foo')
Bram Moolenaardd5893b2022-01-20 21:32:54 +0000432 lines =<< trim END
433 vim9script
434 import './Xfoo.vim' as foo
435 var that: any
436 that += foo
437 END
438 CheckScriptFailure(lines, 'E1060: Expected dot after name: foo')
439 lines =<< trim END
440 vim9script
441 import './Xfoo.vim' as foo
442 foo += 9
443 END
444 CheckScriptFailure(lines, 'E1060: Expected dot after name: foo')
Bram Moolenaard8448622022-01-07 21:39:52 +0000445
446 lines =<< trim END
447 vim9script
448 import './Xfoo.vim' as 9foo
449 END
450 CheckScriptFailure(lines, 'E1047:')
451 lines =<< trim END
452 vim9script
453 import './Xfoo.vim' as the#foo
454 END
455 CheckScriptFailure(lines, 'E1047:')
456 lines =<< trim END
457 vim9script
458 import './Xfoo.vim' as g:foo
459 END
460 CheckScriptFailure(lines, 'E1047:')
461
462 delete('Xfoo.vim')
463
464 lines =<< trim END
465 vim9script
466 def TheFunc()
467 echo 'the func'
468 enddef
469 export var Ref = TheFunc
470 END
471 writefile([], 'Xthat.vim')
Bram Moolenaar7c24dfd2022-01-08 17:03:55 +0000472
Bram Moolenaard8448622022-01-07 21:39:52 +0000473 lines =<< trim END
474 import './Xthat.vim' as That
475 That()
476 END
477 CheckDefAndScriptFailure(lines, ['E1094:', 'E1236: Cannot use That itself'])
Bram Moolenaar7c24dfd2022-01-08 17:03:55 +0000478
479 lines =<< trim END
Bram Moolenaar937610b2022-01-19 17:21:29 +0000480 vim9script
481 import './Xthat.vim' as That
482 def Func()
483 echo That()
484 enddef
485 Func()
486 END
487 CheckScriptFailure(lines, 'E1236: Cannot use That itself')
488
489 lines =<< trim END
Bram Moolenaar7c24dfd2022-01-08 17:03:55 +0000490 import './Xthat.vim' as one
491 import './Xthat.vim' as two
492 END
493 CheckScriptFailure(lines, 'E1262:')
494
495 delete('Xthat.vim')
Bram Moolenaard8448622022-01-07 21:39:52 +0000496
497 mkdir('Ximport')
498
499 writefile(['vim9script'], 'Ximport/.vim')
500 lines =<< trim END
501 vim9script
502 import './Ximport/.vim'
503 END
504 CheckScriptFailure(lines, 'E1261: Cannot import .vim without using "as"')
505 lines =<< trim END
506 vim9script
507 import './Ximport/.vim' as vim
508 END
509 CheckScriptSuccess(lines)
510
511 writefile(['vim9script'], 'Ximport/.vimrc')
512 lines =<< trim END
513 vim9script
514 import './Ximport/.vimrc'
515 END
516 CheckScriptFailure(lines, 'E1257: Imported script must use "as" or end in .vim')
517 lines =<< trim END
518 vim9script
519 import './Ximport/.vimrc' as vimrc
520 END
521 CheckScriptSuccess(lines)
522
523 delete('Ximport', 'rf')
524enddef
525
526func g:Trigger()
527 source Ximport.vim
528 return "echo 'yes'\<CR>"
529endfunc
530
531def Test_import_export_expr_map()
532 # check that :import and :export work when buffer is locked
533 var export_lines =<< trim END
534 vim9script
535 export def That(): string
536 return 'yes'
537 enddef
538 END
539 writefile(export_lines, 'Xexport_that.vim')
540
541 var import_lines =<< trim END
542 vim9script
543 import './Xexport_that.vim' as that
544 assert_equal('yes', that.That())
545 END
546 writefile(import_lines, 'Ximport.vim')
547
548 nnoremap <expr> trigger g:Trigger()
549 feedkeys('trigger', "xt")
550
551 delete('Xexport_that.vim')
552 delete('Ximport.vim')
553 nunmap trigger
554enddef
555
556def Test_import_in_filetype()
557 # check that :import works when the buffer is locked
558 mkdir('ftplugin', 'p')
559 var export_lines =<< trim END
560 vim9script
561 export var That = 'yes'
562 END
563 writefile(export_lines, 'ftplugin/Xexport_ft.vim')
564
565 var import_lines =<< trim END
566 vim9script
567 import './Xexport_ft.vim' as ft
568 assert_equal('yes', ft.That)
569 g:did_load_mytpe = 1
570 END
571 writefile(import_lines, 'ftplugin/qf.vim')
572
573 var save_rtp = &rtp
574 &rtp = getcwd() .. ',' .. &rtp
575
576 filetype plugin on
577 copen
578 assert_equal(1, g:did_load_mytpe)
579
580 quit!
581 delete('Xexport_ft.vim')
582 delete('ftplugin', 'rf')
583 &rtp = save_rtp
584enddef
585
586def Test_use_import_in_mapping()
587 var lines =<< trim END
588 vim9script
589 export def Funcx()
590 g:result = 42
591 enddef
592 END
593 writefile(lines, 'XsomeExport.vim')
594 lines =<< trim END
595 vim9script
596 import './XsomeExport.vim' as some
597 var Funcy = some.Funcx
598 nnoremap <F3> :call <sid>Funcy()<cr>
599 END
600 writefile(lines, 'Xmapscript.vim')
601
602 source Xmapscript.vim
603 feedkeys("\<F3>", "xt")
604 assert_equal(42, g:result)
605
606 unlet g:result
607 delete('XsomeExport.vim')
608 delete('Xmapscript.vim')
609 nunmap <F3>
610enddef
611
Bram Moolenaarf0e7e632022-01-21 13:29:56 +0000612def Test_use_import_in_command_completion()
Bram Moolenaar15d16352022-01-17 20:09:08 +0000613 var lines =<< trim END
614 vim9script
615 export def Complete(..._): list<string>
616 return ['abcd']
617 enddef
618 END
619 writefile(lines, 'Xscript.vim')
620
621 lines =<< trim END
622 vim9script
623 import './Xscript.vim'
624
625 command -nargs=1 -complete=customlist,Xscript.Complete Cmd echo 'ok'
626 feedkeys(":Cmd ab\<Tab>\<C-B>#\<CR>", 'xnt')
627 assert_equal('#Cmd abcd', @:)
628 END
629 CheckScriptSuccess(lines)
630
631 delcommand Cmd
632 delete('Xscript.vim')
633enddef
634
Bram Moolenaarf0e7e632022-01-21 13:29:56 +0000635def Test_use_autoload_import_in_insert_completion()
636 mkdir('Xdir/autoload', 'p')
637 var save_rtp = &rtp
638 exe 'set rtp^=' .. getcwd() .. '/Xdir'
639
640 var lines =<< trim END
641 vim9script
642 export def ThesaurusFunc(findbase: bool, _): any
643 if findbase
644 return 1
645 endif
646 return [
647 'check',
648 'experiment',
649 'test',
650 'verification'
651 ]
652 enddef
653 g:completion_loaded = 'yes'
654 END
655 writefile(lines, 'Xdir/autoload/completion.vim')
656
657 new
658 lines =<< trim END
659 vim9script
660 g:completion_loaded = 'no'
661 import autoload 'completion.vim'
662 set thesaurusfunc=completion.ThesaurusFunc
663 assert_equal('no', g:completion_loaded)
664 feedkeys("i\<C-X>\<C-T>\<C-N>\<Esc>", 'xt')
665 assert_equal('experiment', getline(1))
666 assert_equal('yes', g:completion_loaded)
667 END
668 CheckScriptSuccess(lines)
669
670 set thesaurusfunc=
671 bwipe!
672 delete('Xdir', 'rf')
673 &rtp = save_rtp
674enddef
675
Bram Moolenaare70dd112022-01-21 16:31:11 +0000676def Test_use_autoload_import_in_fold_expression()
677 mkdir('Xdir/autoload', 'p')
678 var save_rtp = &rtp
679 exe 'set rtp^=' .. getcwd() .. '/Xdir'
680
681 var lines =<< trim END
682 vim9script
683 export def Expr(): string
684 return getline(v:lnum) =~ '^#' ? '>1' : '1'
685 enddef
Bram Moolenaar9530b582022-01-22 13:39:08 +0000686 export def Text(): string
687 return 'fold text'
688 enddef
Bram Moolenaare70dd112022-01-21 16:31:11 +0000689 g:fold_loaded = 'yes'
690 END
691 writefile(lines, 'Xdir/autoload/fold.vim')
692
693 lines =<< trim END
694 vim9script
695 import autoload 'fold.vim'
696 &foldexpr = 'fold.Expr()'
Bram Moolenaar9530b582022-01-22 13:39:08 +0000697 &foldtext = 'fold.Text()'
Bram Moolenaare70dd112022-01-21 16:31:11 +0000698 &foldmethod = 'expr'
699 &debug = 'throw'
700 END
701 new
702 setline(1, ['# one', 'text', '# two', 'text'])
703 g:fold_loaded = 'no'
704 CheckScriptSuccess(lines)
705 assert_equal('no', g:fold_loaded)
706 redraw
707 assert_equal('yes', g:fold_loaded)
708
709 # Check that script context of 'foldexpr' is copied to another buffer.
710 edit! otherfile
711 redraw
712
Bram Moolenaar9530b582022-01-22 13:39:08 +0000713 set foldexpr= foldtext& foldmethod& debug=
Bram Moolenaare70dd112022-01-21 16:31:11 +0000714 bwipe!
715 delete('Xdir', 'rf')
716 &rtp = save_rtp
717enddef
718
Bram Moolenaar7b29f6a2022-01-22 17:58:13 +0000719func Test_import_in_diffexpr()
720 CheckExecutable diff
721
722 call Run_Test_import_in_diffexpr()
723endfunc
724
725def Run_Test_import_in_diffexpr()
726 var lines =<< trim END
727 vim9script
728
729 export def DiffExpr()
730 # Prepend some text to check diff type detection
731 writefile(['warning', ' message'], v:fname_out)
732 silent exe '!diff ' .. v:fname_in .. ' '
733 .. v:fname_new .. '>>' .. v:fname_out
734 enddef
735 END
736 writefile(lines, 'Xdiffexpr')
737
738 lines =<< trim END
739 vim9script
740 import './Xdiffexpr' as diff
741
742 set diffexpr=diff.DiffExpr()
743 set diffopt=foldcolumn:0
744 END
745 CheckScriptSuccess(lines)
746
747 enew!
748 call setline(1, ['one', 'two', 'three'])
749 diffthis
750
751 botright vert new
752 call setline(1, ['one', 'two', 'three.'])
753 diffthis
754 # we only check if this does not cause errors
755 redraw
756
757 diffoff!
758 bwipe!
759 bwipe!
760enddef
761
Bram Moolenaar3ba685e2022-01-22 19:17:31 +0000762def Test_import_in_formatexpr()
763 var lines =<< trim END
764 vim9script
765 export def MyFormatExpr(): number
766 g:did_format = 'yes'
767 return 0
768 enddef
769 END
770 writefile(lines, 'Xformatter')
771
772 lines =<< trim END
773 vim9script
774 import './Xformatter' as format
775 set formatexpr=format.MyFormatExpr()
776 END
777 CheckScriptSuccess(lines)
778
779 new
780 setline(1, ['a', 'b', 'c'])
781 normal gqG
782 assert_equal('yes', g:did_format)
783
784 bwipe!
785 delete('Xformatter')
786 unlet g:did_format
787 set formatexpr=
788enddef
789
Bram Moolenaard8448622022-01-07 21:39:52 +0000790def Test_export_fails()
791 CheckScriptFailure(['export var some = 123'], 'E1042:')
792 CheckScriptFailure(['vim9script', 'export var g:some'], 'E1022:')
793 CheckScriptFailure(['vim9script', 'export echo 134'], 'E1043:')
794
795 assert_fails('export something', 'E1043:')
796enddef
797
798func Test_import_fails_without_script()
799 CheckRunVimInTerminal
800
801 " call indirectly to avoid compilation error for missing functions
802 call Run_Test_import_fails_on_command_line()
803endfunc
804
805def Run_Test_import_fails_on_command_line()
806 var export =<< trim END
807 vim9script
808 export def Foo(): number
809 return 0
810 enddef
811 END
812 writefile(export, 'XexportCmd.vim')
813
814 var buf = RunVimInTerminal('-c "import Foo from ''./XexportCmd.vim''"', {
815 rows: 6, wait_for_ruler: 0})
816 WaitForAssert(() => assert_match('^E1094:', term_getline(buf, 5)))
817
818 delete('XexportCmd.vim')
819 StopVimInTerminal(buf)
820enddef
821
822def Test_vim9_reload_noclear()
823 var lines =<< trim END
824 vim9script
825 export var exported = 'thexport'
826
827 export def TheFunc(x = 0)
828 enddef
829 END
830 writefile(lines, 'XExportReload')
831 lines =<< trim END
832 vim9script noclear
833 g:loadCount += 1
834 var s:reloaded = 'init'
835 import './XExportReload' as exp
836
837 def Again(): string
838 return 'again'
839 enddef
840
841 exp.TheFunc()
842
843 if exists('s:loaded') | finish | endif
844 var s:loaded = true
845
846 var s:notReloaded = 'yes'
847 s:reloaded = 'first'
848 def g:Values(): list<string>
849 return [s:reloaded, s:notReloaded, Again(), Once(), exp.exported]
850 enddef
851
852 def Once(): string
853 return 'once'
854 enddef
855 END
856 writefile(lines, 'XReloaded')
857 g:loadCount = 0
858 source XReloaded
859 assert_equal(1, g:loadCount)
860 assert_equal(['first', 'yes', 'again', 'once', 'thexport'], g:Values())
861 source XReloaded
862 assert_equal(2, g:loadCount)
863 assert_equal(['init', 'yes', 'again', 'once', 'thexport'], g:Values())
864 source XReloaded
865 assert_equal(3, g:loadCount)
866 assert_equal(['init', 'yes', 'again', 'once', 'thexport'], g:Values())
867
868 delete('XReloaded')
869 delete('XExportReload')
870 delfunc g:Values
871 unlet g:loadCount
872
873 lines =<< trim END
874 vim9script
875 def Inner()
876 enddef
877 END
878 lines->writefile('XreloadScript.vim')
879 source XreloadScript.vim
880
881 lines =<< trim END
882 vim9script
883 def Outer()
884 def Inner()
885 enddef
886 enddef
887 defcompile
888 END
889 lines->writefile('XreloadScript.vim')
890 source XreloadScript.vim
891
892 delete('XreloadScript.vim')
893enddef
894
895def Test_vim9_reload_import()
896 var lines =<< trim END
897 vim9script
898 const var = ''
899 var valone = 1234
900 def MyFunc(arg: string)
901 valone = 5678
902 enddef
903 END
904 var morelines =<< trim END
905 var valtwo = 222
906 export def GetValtwo(): number
907 return valtwo
908 enddef
909 END
910 writefile(lines + morelines, 'Xreload.vim')
911 source Xreload.vim
912 source Xreload.vim
913 source Xreload.vim
914
915 # cannot declare a var twice
916 lines =<< trim END
917 vim9script
918 var valone = 1234
919 var valone = 5678
920 END
921 writefile(lines, 'Xreload.vim')
922 assert_fails('source Xreload.vim', 'E1041:', '', 3, 'Xreload.vim')
923
924 delete('Xreload.vim')
925 delete('Ximport.vim')
926enddef
927
928" if a script is reloaded with a script-local variable that changed its type, a
929" compiled function using that variable must fail.
930def Test_script_reload_change_type()
931 var lines =<< trim END
932 vim9script noclear
933 var str = 'string'
934 def g:GetStr(): string
935 return str .. 'xxx'
936 enddef
937 END
938 writefile(lines, 'Xreload.vim')
939 source Xreload.vim
940 echo g:GetStr()
941
942 lines =<< trim END
943 vim9script noclear
944 var str = 1234
945 END
946 writefile(lines, 'Xreload.vim')
947 source Xreload.vim
948 assert_fails('echo g:GetStr()', 'E1150:')
949
950 delfunc g:GetStr
951 delete('Xreload.vim')
952enddef
953
954" Define CallFunc so that the test can be compiled
955command CallFunc echo 'nop'
956
957def Test_script_reload_from_function()
958 var lines =<< trim END
959 vim9script
960
961 if exists('g:loaded')
962 finish
963 endif
964 g:loaded = 1
965 delcommand CallFunc
966 command CallFunc Func()
967 def Func()
968 so XreloadFunc.vim
969 g:didTheFunc = 1
970 enddef
971 END
972 writefile(lines, 'XreloadFunc.vim')
973 source XreloadFunc.vim
974 CallFunc
975 assert_equal(1, g:didTheFunc)
976
977 delete('XreloadFunc.vim')
978 delcommand CallFunc
979 unlet g:loaded
980 unlet g:didTheFunc
981enddef
982
983def s:RetSome(): string
984 return 'some'
985enddef
986
987" Not exported function that is referenced needs to be accessed by the
988" script-local name.
989def Test_vim9_funcref()
990 var sortlines =<< trim END
991 vim9script
992 def Compare(i1: number, i2: number): number
993 return i2 - i1
994 enddef
995
996 export def FastSort(): list<number>
997 return range(5)->sort(Compare)
998 enddef
999
1000 export def GetString(arg: string): string
1001 return arg
1002 enddef
1003 END
1004 writefile(sortlines, 'Xsort.vim')
1005
1006 var lines =<< trim END
1007 vim9script
1008 import './Xsort.vim'
1009 def Test()
1010 g:result = Xsort.FastSort()
1011 enddef
1012 Test()
Bram Moolenaar7c24dfd2022-01-08 17:03:55 +00001013 END
1014 writefile(lines, 'Xscript.vim')
1015 source Xscript.vim
1016 assert_equal([4, 3, 2, 1, 0], g:result)
1017 unlet g:result
Bram Moolenaard8448622022-01-07 21:39:52 +00001018
Bram Moolenaar7c24dfd2022-01-08 17:03:55 +00001019 lines =<< trim END
1020 vim9script
Bram Moolenaard8448622022-01-07 21:39:52 +00001021 # using a function imported with "as"
1022 import './Xsort.vim' as anAlias
1023 assert_equal('yes', anAlias.GetString('yes'))
1024
1025 # using the function from a compiled function
1026 def TestMore(): string
1027 var s = s:anAlias.GetString('foo')
1028 return s .. anAlias.GetString('bar')
1029 enddef
1030 assert_equal('foobar', TestMore())
1031
1032 # error when using a function that isn't exported
1033 assert_fails('anAlias.Compare(1, 2)', 'E1049:')
1034 END
1035 writefile(lines, 'Xscript.vim')
1036
Bram Moolenaard8448622022-01-07 21:39:52 +00001037 delete('Xsort.vim')
1038 delete('Xscript.vim')
1039
1040 var Funcref = function('s:RetSome')
1041 assert_equal('some', Funcref())
1042enddef
1043
1044" Check that when searching for "FilterFunc" it finds the import in the
1045" script where FastFilter() is called from, both as a string and as a direct
1046" function reference.
1047def Test_vim9_funcref_other_script()
1048 var filterLines =<< trim END
1049 vim9script
1050 export def FilterFunc(idx: number, val: number): bool
1051 return idx % 2 == 1
1052 enddef
1053 export def FastFilter(): list<number>
1054 return range(10)->filter('FilterFunc(v:key, v:val)')
1055 enddef
1056 export def FastFilterDirect(): list<number>
1057 return range(10)->filter(FilterFunc)
1058 enddef
1059 END
1060 writefile(filterLines, 'Xfilter.vim')
1061
1062 var lines =<< trim END
1063 vim9script
1064 import './Xfilter.vim' as filter
1065 def Test()
1066 var x: list<number> = filter.FastFilter()
1067 enddef
1068 Test()
1069 def TestDirect()
1070 var x: list<number> = filter.FastFilterDirect()
1071 enddef
1072 TestDirect()
1073 END
1074 CheckScriptSuccess(lines)
1075 delete('Xfilter.vim')
1076enddef
1077
1078def Test_import_absolute()
1079 var import_lines = [
1080 'vim9script',
1081 'import "' .. escape(getcwd(), '\') .. '/Xexport_abs.vim" as abs',
1082 'def UseExported()',
1083 ' g:imported_abs = abs.exported',
1084 ' abs.exported = 8888',
1085 ' g:imported_after = abs.exported',
1086 'enddef',
1087 'UseExported()',
1088 'g:import_disassembled = execute("disass UseExported")',
1089 ]
1090 writefile(import_lines, 'Ximport_abs.vim')
1091 writefile(s:export_script_lines, 'Xexport_abs.vim')
1092
1093 source Ximport_abs.vim
1094
1095 assert_equal(9876, g:imported_abs)
1096 assert_equal(8888, g:imported_after)
1097 assert_match('<SNR>\d\+_UseExported\_s*' ..
1098 'g:imported_abs = abs.exported\_s*' ..
1099 '0 LOADSCRIPT exported-2 from .*Xexport_abs.vim\_s*' ..
1100 '1 STOREG g:imported_abs\_s*' ..
1101 'abs.exported = 8888\_s*' ..
1102 '2 PUSHNR 8888\_s*' ..
1103 '3 STORESCRIPT exported-2 in .*Xexport_abs.vim\_s*' ..
1104 'g:imported_after = abs.exported\_s*' ..
1105 '4 LOADSCRIPT exported-2 from .*Xexport_abs.vim\_s*' ..
1106 '5 STOREG g:imported_after',
1107 g:import_disassembled)
1108
1109 Undo_export_script_lines()
1110 unlet g:imported_abs
1111 unlet g:import_disassembled
1112
1113 delete('Ximport_abs.vim')
1114 delete('Xexport_abs.vim')
1115enddef
1116
1117def Test_import_rtp()
1118 var import_lines = [
1119 'vim9script',
1120 'import "Xexport_rtp.vim" as rtp',
1121 'g:imported_rtp = rtp.exported',
1122 ]
1123 writefile(import_lines, 'Ximport_rtp.vim')
1124 mkdir('import', 'p')
1125 writefile(s:export_script_lines, 'import/Xexport_rtp.vim')
1126
1127 var save_rtp = &rtp
1128 &rtp = getcwd()
1129 source Ximport_rtp.vim
1130 &rtp = save_rtp
1131
1132 assert_equal(9876, g:imported_rtp)
1133
1134 Undo_export_script_lines()
1135 unlet g:imported_rtp
1136 delete('Ximport_rtp.vim')
1137 delete('import', 'rf')
1138enddef
1139
1140def Test_import_compile_error()
1141 var export_lines = [
1142 'vim9script',
1143 'export def ExpFunc(): string',
1144 ' return notDefined',
1145 'enddef',
1146 ]
1147 writefile(export_lines, 'Xexported.vim')
1148
1149 var import_lines = [
1150 'vim9script',
1151 'import "./Xexported.vim" as expo',
1152 'def ImpFunc()',
1153 ' echo expo.ExpFunc()',
1154 'enddef',
1155 'defcompile',
1156 ]
1157 writefile(import_lines, 'Ximport.vim')
1158
1159 try
1160 source Ximport.vim
1161 catch /E1001/
1162 # Error should be before the Xexported.vim file.
1163 assert_match('E1001: Variable not found: notDefined', v:exception)
1164 assert_match('function <SNR>\d\+_ImpFunc\[1\]..<SNR>\d\+_ExpFunc, line 1', v:throwpoint)
1165 endtry
1166
1167 delete('Xexported.vim')
1168 delete('Ximport.vim')
1169enddef
1170
1171def Test_func_overrules_import_fails()
1172 var export_lines =<< trim END
1173 vim9script
1174 export def Func()
1175 echo 'imported'
1176 enddef
1177 END
1178 writefile(export_lines, 'XexportedFunc.vim')
1179
1180 var lines =<< trim END
1181 vim9script
1182 import './XexportedFunc.vim' as Func
1183 def Func()
1184 echo 'local to function'
1185 enddef
1186 END
Bram Moolenaar937610b2022-01-19 17:21:29 +00001187 CheckScriptFailure(lines, 'E1213: Redefining imported item "Func"')
Bram Moolenaard8448622022-01-07 21:39:52 +00001188
1189 lines =<< trim END
1190 vim9script
1191 import './XexportedFunc.vim' as Func
1192 def Outer()
1193 def Func()
1194 echo 'local to function'
1195 enddef
1196 enddef
1197 defcompile
1198 END
1199 CheckScriptFailure(lines, 'E1236:')
1200
1201 delete('XexportedFunc.vim')
1202enddef
1203
1204def Test_source_vim9_from_legacy()
1205 var vim9_lines =<< trim END
1206 vim9script
1207 var local = 'local'
1208 g:global = 'global'
1209 export var exported = 'exported'
1210 export def GetText(): string
1211 return 'text'
1212 enddef
1213 END
1214 writefile(vim9_lines, 'Xvim9_script.vim')
1215
1216 var legacy_lines =<< trim END
1217 source Xvim9_script.vim
1218
1219 call assert_false(exists('local'))
1220 call assert_false(exists('exported'))
1221 call assert_false(exists('s:exported'))
1222 call assert_equal('global', global)
1223 call assert_equal('global', g:global)
Bram Moolenaard8448622022-01-07 21:39:52 +00001224 END
1225 writefile(legacy_lines, 'Xlegacy_script.vim')
1226
1227 source Xlegacy_script.vim
1228 assert_equal('global', g:global)
1229 unlet g:global
1230
1231 delete('Xlegacy_script.vim')
1232 delete('Xvim9_script.vim')
1233enddef
1234
Bram Moolenaarc43e6232022-01-13 20:51:56 +00001235def Test_import_vim9_from_legacy()
1236 var vim9_lines =<< trim END
1237 vim9script
1238 var local = 'local'
1239 g:global = 'global'
1240 export var exported = 'exported'
1241 export def GetText(): string
1242 return 'text'
1243 enddef
1244 END
1245 writefile(vim9_lines, 'Xvim9_export.vim')
1246
1247 var legacy_lines =<< trim END
1248 import './Xvim9_export.vim' as vim9
1249
1250 call assert_false(exists('vim9'))
1251 call assert_false(exists('local'))
1252 call assert_false(exists('s:vim9.local'))
1253 call assert_equal('global', global)
1254 call assert_equal('global', g:global)
1255 call assert_false(exists('exported'))
1256 call assert_false(exists('s:exported'))
1257 call assert_false(exists('*GetText'))
1258
1259 " imported symbol is script-local
1260 call assert_equal('exported', s:vim9.exported)
1261 call assert_equal('text', s:vim9.GetText())
1262 END
1263 writefile(legacy_lines, 'Xlegacy_script.vim')
1264
1265 source Xlegacy_script.vim
1266 assert_equal('global', g:global)
1267 unlet g:global
1268
1269 delete('Xlegacy_script.vim')
1270 delete('Xvim9_export.vim')
1271enddef
1272
Bram Moolenaard8448622022-01-07 21:39:52 +00001273def Test_cmdline_win()
1274 # if the Vim syntax highlighting uses Vim9 constructs they can be used from
1275 # the command line window.
1276 mkdir('rtp/syntax', 'p')
1277 var export_lines =<< trim END
1278 vim9script
1279 export var That = 'yes'
1280 END
1281 writefile(export_lines, 'rtp/syntax/Xexport.vim')
1282 var import_lines =<< trim END
1283 vim9script
1284 import './Xexport.vim' as exp
1285 echo exp.That
1286 END
1287 writefile(import_lines, 'rtp/syntax/vim.vim')
1288 var save_rtp = &rtp
1289 &rtp = getcwd() .. '/rtp' .. ',' .. &rtp
1290 syntax on
1291 augroup CmdWin
1292 autocmd CmdwinEnter * g:got_there = 'yes'
1293 augroup END
1294 # this will open and also close the cmdline window
1295 feedkeys('q:', 'xt')
1296 assert_equal('yes', g:got_there)
1297
1298 augroup CmdWin
1299 au!
1300 augroup END
1301 &rtp = save_rtp
1302 delete('rtp', 'rf')
1303enddef
1304
1305def Test_import_gone_when_sourced_twice()
1306 var exportlines =<< trim END
1307 vim9script
1308 if exists('g:guard')
1309 finish
1310 endif
1311 g:guard = 1
1312 export var name = 'someName'
1313 END
1314 writefile(exportlines, 'XexportScript.vim')
1315
1316 var lines =<< trim END
1317 vim9script
1318 import './XexportScript.vim' as expo
1319 def g:GetName(): string
1320 return expo.name
1321 enddef
1322 END
1323 writefile(lines, 'XscriptImport.vim')
1324 so XscriptImport.vim
1325 assert_equal('someName', g:GetName())
1326
1327 so XexportScript.vim
1328 assert_fails('call g:GetName()', 'E1149:')
1329
1330 delfunc g:GetName
1331 delete('XexportScript.vim')
1332 delete('XscriptImport.vim')
1333 unlet g:guard
1334enddef
1335
Bram Moolenaar160aa862022-01-10 21:29:57 +00001336" test using an auto-loaded function and variable
Bram Moolenaar0e3e7ba2022-01-13 20:18:56 +00001337def Test_vim9_autoload_full_name()
Bram Moolenaar160aa862022-01-10 21:29:57 +00001338 var lines =<< trim END
1339 vim9script
1340 def some#gettest(): string
1341 return 'test'
1342 enddef
1343 g:some#name = 'name'
1344 g:some#dict = {key: 'value'}
1345
1346 def some#varargs(a1: string, ...l: list<string>): string
1347 return a1 .. l[0] .. l[1]
1348 enddef
1349 END
1350
1351 mkdir('Xdir/autoload', 'p')
1352 writefile(lines, 'Xdir/autoload/some.vim')
1353 var save_rtp = &rtp
1354 exe 'set rtp^=' .. getcwd() .. '/Xdir'
1355
1356 assert_equal('test', g:some#gettest())
1357 assert_equal('name', g:some#name)
1358 assert_equal('value', g:some#dict.key)
1359 g:some#other = 'other'
1360 assert_equal('other', g:some#other)
1361
1362 assert_equal('abc', some#varargs('a', 'b', 'c'))
1363
1364 # upper case script name works
1365 lines =<< trim END
1366 vim9script
1367 def Other#getOther(): string
1368 return 'other'
1369 enddef
1370 END
1371 writefile(lines, 'Xdir/autoload/Other.vim')
1372 assert_equal('other', g:Other#getOther())
1373
1374 delete('Xdir', 'rf')
1375 &rtp = save_rtp
1376enddef
1377
1378def Test_vim9script_autoload()
1379 mkdir('Xdir/autoload', 'p')
1380 var save_rtp = &rtp
1381 exe 'set rtp^=' .. getcwd() .. '/Xdir'
1382
Bram Moolenaarfd218c82022-01-18 16:26:24 +00001383 # when the path has "/autoload/" prefix is not needed
Bram Moolenaar160aa862022-01-10 21:29:57 +00001384 var lines =<< trim END
Bram Moolenaarfd218c82022-01-18 16:26:24 +00001385 vim9script
Bram Moolenaar17d36cb2022-01-12 11:46:40 +00001386 g:prefixed_loaded += 1
Bram Moolenaar160aa862022-01-10 21:29:57 +00001387
1388 export def Gettest(): string
1389 return 'test'
1390 enddef
1391
Bram Moolenaar0e3e7ba2022-01-13 20:18:56 +00001392 export var name = 'name'
1393
1394 export func GetFunc()
1395 return Gettest() .. 'more' .. s:name
Bram Moolenaar160aa862022-01-10 21:29:57 +00001396 endfunc
1397
Bram Moolenaar0e3e7ba2022-01-13 20:18:56 +00001398 export def GetDef(): string
1399 return Gettest() .. 'more' .. name
1400 enddef
1401
Bram Moolenaar160aa862022-01-10 21:29:57 +00001402 export final fname = 'final'
1403 export const cname = 'const'
1404 END
1405 writefile(lines, 'Xdir/autoload/prefixed.vim')
1406
Bram Moolenaar17d36cb2022-01-12 11:46:40 +00001407 g:prefixed_loaded = 0
1408 g:expected_loaded = 0
Bram Moolenaar160aa862022-01-10 21:29:57 +00001409 lines =<< trim END
1410 vim9script
1411 import autoload 'prefixed.vim'
Bram Moolenaar17d36cb2022-01-12 11:46:40 +00001412 assert_equal(g:expected_loaded, g:prefixed_loaded)
Bram Moolenaar160aa862022-01-10 21:29:57 +00001413 assert_equal('test', prefixed.Gettest())
Bram Moolenaar17d36cb2022-01-12 11:46:40 +00001414 assert_equal(1, g:prefixed_loaded)
Bram Moolenaar160aa862022-01-10 21:29:57 +00001415
Bram Moolenaar0e3e7ba2022-01-13 20:18:56 +00001416 assert_equal('testmorename', prefixed.GetFunc())
1417 assert_equal('testmorename', prefixed.GetDef())
Bram Moolenaar160aa862022-01-10 21:29:57 +00001418 assert_equal('name', prefixed.name)
1419 assert_equal('final', prefixed.fname)
1420 assert_equal('const', prefixed.cname)
1421 END
1422 CheckScriptSuccess(lines)
Bram Moolenaar17d36cb2022-01-12 11:46:40 +00001423 # can source it again, autoload script not loaded again
1424 g:expected_loaded = 1
1425 CheckScriptSuccess(lines)
Bram Moolenaar160aa862022-01-10 21:29:57 +00001426
1427 # can also get the items by autoload name
1428 lines =<< trim END
1429 call assert_equal('test', prefixed#Gettest())
Bram Moolenaar0e3e7ba2022-01-13 20:18:56 +00001430 call assert_equal('testmorename', prefixed#GetFunc())
Bram Moolenaar160aa862022-01-10 21:29:57 +00001431 call assert_equal('name', prefixed#name)
1432 call assert_equal('final', prefixed#fname)
1433 call assert_equal('const', prefixed#cname)
1434 END
1435 CheckScriptSuccess(lines)
1436
Bram Moolenaarf111cdf2022-01-12 12:48:17 +00001437 unlet g:prefixed_loaded
1438 unlet g:expected_loaded
1439 delete('Xdir', 'rf')
1440 &rtp = save_rtp
1441enddef
1442
Bram Moolenaard02dce22022-01-18 17:43:04 +00001443def Test_import_autoload_not_exported()
1444 mkdir('Xdir/autoload', 'p')
1445 var save_rtp = &rtp
1446 exe 'set rtp^=' .. getcwd() .. '/Xdir'
1447
1448 # error when using an item that is not exported from an autoload script
1449 var exportLines =<< trim END
1450 vim9script
1451 var notExported = 123
1452 def NotExport()
1453 echo 'nop'
1454 enddef
1455 END
1456 writefile(exportLines, 'Xdir/autoload/notExport1.vim')
1457
1458 var lines =<< trim END
1459 vim9script
1460 import autoload 'notExport1.vim'
1461 echo notExport1.notFound
1462 END
1463 CheckScriptFailure(lines, 'E1048: Item not found in script: notFound')
1464
1465 lines =<< trim END
1466 vim9script
1467 import autoload 'notExport1.vim'
1468 echo notExport1.notExported
1469 END
1470 CheckScriptFailure(lines, 'E1049: Item not exported in script: notExported')
1471
1472 lines =<< trim END
1473 vim9script
1474 import autoload 'notExport1.vim'
1475 echo notExport1.NotFunc()
1476 END
1477 CheckScriptFailure(lines, 'E1048: Item not found in script: NotFunc')
1478
1479 lines =<< trim END
1480 vim9script
1481 import autoload 'notExport1.vim'
1482 echo notExport1.NotExport()
1483 END
1484 CheckScriptFailure(lines, 'E1049: Item not exported in script: NotExport')
1485
1486 lines =<< trim END
1487 vim9script
1488 import autoload 'notExport1.vim'
1489 echo 'text'->notExport1.NotFunc()
1490 END
1491 CheckScriptFailure(lines, 'E1048: Item not found in script: NotFunc')
1492
1493 lines =<< trim END
1494 vim9script
1495 import autoload 'notExport1.vim'
1496 echo 'text'->notExport1.NotExport()
1497 END
1498 CheckScriptFailure(lines, 'E1049: Item not exported in script: NotExport')
1499
1500 # using a :def function we use a different autoload script every time so that
1501 # the function is compiled without the script loaded
1502 writefile(exportLines, 'Xdir/autoload/notExport2.vim')
1503 lines =<< trim END
1504 vim9script
1505 import autoload 'notExport2.vim'
1506 def Testit()
1507 echo notExport2.notFound
1508 enddef
1509 Testit()
1510 END
1511 CheckScriptFailure(lines, 'E1048: Item not found in script: notExport2#notFound')
1512
1513 writefile(exportLines, 'Xdir/autoload/notExport3.vim')
1514 lines =<< trim END
1515 vim9script
1516 import autoload 'notExport3.vim'
1517 def Testit()
1518 echo notExport3.notExported
1519 enddef
1520 Testit()
1521 END
1522 # don't get E1049 because it is too complicated to figure out
1523 CheckScriptFailure(lines, 'E1048: Item not found in script: notExport3#notExported')
1524
1525 writefile(exportLines, 'Xdir/autoload/notExport4.vim')
1526 lines =<< trim END
1527 vim9script
1528 import autoload 'notExport4.vim'
1529 def Testit()
1530 echo notExport4.NotFunc()
1531 enddef
1532 Testit()
1533 END
1534 CheckScriptFailure(lines, 'E117: Unknown function: notExport4#NotFunc')
1535
1536 writefile(exportLines, 'Xdir/autoload/notExport5.vim')
1537 lines =<< trim END
1538 vim9script
1539 import autoload 'notExport5.vim'
1540 def Testit()
1541 echo notExport5.NotExport()
1542 enddef
1543 Testit()
1544 END
1545 CheckScriptFailure(lines, 'E117: Unknown function: notExport5#NotExport')
1546
1547 writefile(exportLines, 'Xdir/autoload/notExport6.vim')
1548 lines =<< trim END
1549 vim9script
1550 import autoload 'notExport6.vim'
1551 def Testit()
1552 echo 'text'->notExport6.NotFunc()
1553 enddef
1554 Testit()
1555 END
1556 CheckScriptFailure(lines, 'E117: Unknown function: notExport6#NotFunc')
1557
1558 writefile(exportLines, 'Xdir/autoload/notExport7.vim')
1559 lines =<< trim END
1560 vim9script
1561 import autoload 'notExport7.vim'
1562 def Testit()
1563 echo 'text'->notExport7.NotExport()
1564 enddef
1565 Testit()
1566 END
1567 CheckScriptFailure(lines, 'E117: Unknown function: notExport7#NotExport')
1568
1569 delete('Xdir', 'rf')
1570 &rtp = save_rtp
1571enddef
1572
Bram Moolenaarf111cdf2022-01-12 12:48:17 +00001573def Test_vim9script_autoload_call()
1574 mkdir('Xdir/autoload', 'p')
1575 var save_rtp = &rtp
1576 exe 'set rtp^=' .. getcwd() .. '/Xdir'
1577
1578 var lines =<< trim END
Bram Moolenaarfd218c82022-01-18 16:26:24 +00001579 vim9script
Bram Moolenaarf111cdf2022-01-12 12:48:17 +00001580
Bram Moolenaarcbbc48f2022-01-18 12:58:28 +00001581 export def RetArg(arg: string): string
1582 return arg
1583 enddef
1584
Bram Moolenaarf111cdf2022-01-12 12:48:17 +00001585 export def Getother()
1586 g:result = 'other'
1587 enddef
1588 END
Bram Moolenaar5d982692022-01-12 15:15:27 +00001589 writefile(lines, 'Xdir/autoload/another.vim')
Bram Moolenaarf111cdf2022-01-12 12:48:17 +00001590
1591 lines =<< trim END
1592 vim9script
Bram Moolenaar5d982692022-01-12 15:15:27 +00001593 import autoload 'another.vim'
Bram Moolenaarcbbc48f2022-01-18 12:58:28 +00001594
1595 # compile this before 'another.vim' is loaded
1596 def CallAnother()
1597 assert_equal('foo', 'foo'->another.RetArg())
1598 enddef
1599 CallAnother()
1600
Bram Moolenaar5d982692022-01-12 15:15:27 +00001601 call another.Getother()
Bram Moolenaarf111cdf2022-01-12 12:48:17 +00001602 assert_equal('other', g:result)
Bram Moolenaar3d8e25a2022-01-22 11:00:02 +00001603
1604 assert_equal('arg', call('another.RetArg', ['arg']))
Bram Moolenaarf111cdf2022-01-12 12:48:17 +00001605 END
1606 CheckScriptSuccess(lines)
1607
1608 unlet g:result
Bram Moolenaar160aa862022-01-10 21:29:57 +00001609 delete('Xdir', 'rf')
1610 &rtp = save_rtp
1611enddef
1612
Bram Moolenaarb697dc22022-01-22 11:27:29 +00001613def Test_vim9script_noclear_autoload()
1614 mkdir('Xdir/autoload', 'p')
1615 var save_rtp = &rtp
1616 exe 'set rtp^=' .. getcwd() .. '/Xdir'
1617
1618 var lines =<< trim END
1619 vim9script
1620 export def Func(): string
1621 return 'called'
1622 enddef
1623 g:double_loaded = 'yes'
1624 END
1625 writefile(lines, 'Xdir/autoload/double.vim')
1626
1627 lines =<< trim END
1628 vim9script noclear
1629 if exists('g:script_loaded')
1630 finish
1631 endif
1632 g:script_loaded = true
1633
1634 import autoload 'double.vim'
1635 nnoremap <F3> <ScriptCmd>g:result = double.Func()<CR>
1636 END
1637 g:double_loaded = 'no'
1638 writefile(lines, 'Xloaddouble')
1639 source Xloaddouble
1640 assert_equal('no', g:double_loaded)
1641 assert_equal(true, g:script_loaded)
1642 source Xloaddouble
1643 feedkeys("\<F3>", 'xt')
1644 assert_equal('called', g:result)
1645 assert_equal('yes', g:double_loaded)
1646
1647 delete('Xloaddouble')
1648 unlet g:double_loaded
1649 unlet g:script_loaded
1650 unlet g:result
1651 delete('Xdir', 'rf')
1652 &rtp = save_rtp
1653enddef
1654
Bram Moolenaar9c7cae62022-01-20 19:10:25 +00001655def Test_vim9script_autoload_duplicate()
1656 mkdir('Xdir/autoload', 'p')
1657
1658 var lines =<< trim END
1659 vim9script
1660
1661 export def Func()
1662 enddef
1663
1664 def Func()
1665 enddef
1666 END
1667 writefile(lines, 'Xdir/autoload/dupfunc.vim')
1668 assert_fails('source Xdir/autoload/dupfunc.vim', 'E1073:')
1669
1670 lines =<< trim END
1671 vim9script
1672
1673 def Func()
1674 enddef
1675
1676 export def Func()
1677 enddef
1678 END
1679 writefile(lines, 'Xdir/autoload/dup2func.vim')
1680 assert_fails('source Xdir/autoload/dup2func.vim', 'E1073:')
1681
1682 lines =<< trim END
1683 vim9script
1684
1685 def Func()
1686 enddef
1687
1688 export var Func = 'asdf'
1689 END
1690 writefile(lines, 'Xdir/autoload/dup3func.vim')
1691 assert_fails('source Xdir/autoload/dup3func.vim', 'E1041: Redefining script item Func')
1692
1693 lines =<< trim END
1694 vim9script
1695
1696 export var Func = 'asdf'
1697
1698 def Func()
1699 enddef
1700 END
1701 writefile(lines, 'Xdir/autoload/dup4func.vim')
1702 assert_fails('source Xdir/autoload/dup4func.vim', 'E707:')
1703
1704 lines =<< trim END
1705 vim9script
1706
1707 var Func = 'asdf'
1708
1709 export def Func()
1710 enddef
1711 END
1712 writefile(lines, 'Xdir/autoload/dup5func.vim')
1713 assert_fails('source Xdir/autoload/dup5func.vim', 'E707:')
1714
1715 lines =<< trim END
1716 vim9script
1717
1718 export def Func()
1719 enddef
1720
1721 var Func = 'asdf'
1722 END
1723 writefile(lines, 'Xdir/autoload/dup6func.vim')
1724 assert_fails('source Xdir/autoload/dup6func.vim', 'E1041: Redefining script item Func')
1725
1726 delete('Xdir', 'rf')
1727enddef
1728
Bram Moolenaar2017d6f2022-01-20 19:38:46 +00001729def Test_autoload_missing_function_name()
1730 mkdir('Xdir/autoload', 'p')
1731
1732 var lines =<< trim END
1733 vim9script
1734
1735 def loadme#()
1736 enddef
1737 END
1738 writefile(lines, 'Xdir/autoload/loadme.vim')
1739 assert_fails('source Xdir/autoload/loadme.vim', 'E129:')
1740
1741 delete('Xdir', 'rf')
1742enddef
1743
Bram Moolenaar19e69a62022-01-21 20:37:05 +00001744def Test_autoload_name_wring()
1745 var lines =<< trim END
1746 vim9script
1747 def Xscriptname#Func()
1748 enddef
1749 END
1750 writefile(lines, 'Xscriptname.vim')
1751 CheckScriptFailure(lines, 'E1263:')
1752
1753 delete('Xscriptname')
1754enddef
1755
Bram Moolenaard041f422022-01-12 19:54:00 +00001756def Test_import_autoload_postponed()
1757 mkdir('Xdir/autoload', 'p')
1758 var save_rtp = &rtp
1759 exe 'set rtp^=' .. getcwd() .. '/Xdir'
1760
1761 var lines =<< trim END
Bram Moolenaarfd218c82022-01-18 16:26:24 +00001762 vim9script
Bram Moolenaard041f422022-01-12 19:54:00 +00001763
1764 g:loaded_postponed = 'true'
1765 export var variable = 'bla'
1766 export def Function(): string
1767 return 'bla'
1768 enddef
1769 END
1770 writefile(lines, 'Xdir/autoload/postponed.vim')
1771
1772 lines =<< trim END
1773 vim9script
1774
1775 import autoload 'postponed.vim'
1776 def Tryit()
1777 echo postponed.variable
1778 echo postponed.Function()
1779 enddef
1780 defcompile
1781 END
1782 CheckScriptSuccess(lines)
1783 assert_false(exists('g:loaded_postponed'))
1784 CheckScriptSuccess(lines + ['Tryit()'])
1785 assert_equal('true', g:loaded_postponed)
1786
1787 unlet g:loaded_postponed
1788 delete('Xdir', 'rf')
1789 &rtp = save_rtp
1790enddef
1791
Bram Moolenaar3e4fa3d2022-01-13 22:05:09 +00001792def Test_import_autoload_override()
1793 mkdir('Xdir/autoload', 'p')
1794 var save_rtp = &rtp
1795 exe 'set rtp^=' .. getcwd() .. '/Xdir'
1796 test_override('autoload', 1)
1797
1798 var lines =<< trim END
Bram Moolenaarfd218c82022-01-18 16:26:24 +00001799 vim9script
Bram Moolenaar3e4fa3d2022-01-13 22:05:09 +00001800
1801 g:loaded_override = 'true'
1802 export var variable = 'bla'
1803 export def Function(): string
1804 return 'bla'
1805 enddef
1806 END
1807 writefile(lines, 'Xdir/autoload/override.vim')
1808
1809 lines =<< trim END
1810 vim9script
1811
1812 import autoload 'override.vim'
1813 assert_equal('true', g:loaded_override)
1814
1815 def Tryit()
1816 echo override.doesNotExist
1817 enddef
1818 defcompile
1819 END
1820 CheckScriptFailure(lines, 'E1048: Item not found in script: doesNotExist', 1)
1821
1822 test_override('autoload', 0)
1823 unlet g:loaded_override
1824 delete('Xdir', 'rf')
1825 &rtp = save_rtp
1826enddef
1827
Bram Moolenaar19db9e62022-01-11 11:58:19 +00001828def Test_autoload_mapping()
1829 mkdir('Xdir/autoload', 'p')
1830 var save_rtp = &rtp
1831 exe 'set rtp^=' .. getcwd() .. '/Xdir'
1832
1833 var lines =<< trim END
Bram Moolenaarfd218c82022-01-18 16:26:24 +00001834 vim9script
Bram Moolenaar19db9e62022-01-11 11:58:19 +00001835
1836 g:toggle_loaded = 'yes'
1837
1838 export def Toggle(): string
1839 return ":g:toggle_called = 'yes'\<CR>"
1840 enddef
Bram Moolenaare32c3c42022-01-15 18:26:04 +00001841 export def Doit()
1842 g:doit_called = 'yes'
1843 enddef
Bram Moolenaar19db9e62022-01-11 11:58:19 +00001844 END
1845 writefile(lines, 'Xdir/autoload/toggle.vim')
1846
1847 lines =<< trim END
1848 vim9script
1849
1850 import autoload 'toggle.vim'
1851
1852 nnoremap <silent> <expr> tt toggle.Toggle()
Bram Moolenaare32c3c42022-01-15 18:26:04 +00001853 nnoremap <silent> xx <ScriptCmd>toggle.Doit()<CR>
1854 nnoremap <silent> yy <Cmd>toggle.Doit()<CR>
Bram Moolenaar19db9e62022-01-11 11:58:19 +00001855 END
1856 CheckScriptSuccess(lines)
1857 assert_false(exists("g:toggle_loaded"))
1858 assert_false(exists("g:toggle_called"))
Bram Moolenaar6079da72022-01-18 14:16:59 +00001859 assert_match('\d A: \f*[/\\]toggle.vim', execute('scriptnames'))
Bram Moolenaar19db9e62022-01-11 11:58:19 +00001860
1861 feedkeys("tt", 'xt')
1862 assert_equal('yes', g:toggle_loaded)
1863 assert_equal('yes', g:toggle_called)
Bram Moolenaar6079da72022-01-18 14:16:59 +00001864 assert_match('\d: \f*[/\\]toggle.vim', execute('scriptnames'))
Bram Moolenaar19db9e62022-01-11 11:58:19 +00001865
Bram Moolenaare32c3c42022-01-15 18:26:04 +00001866 feedkeys("xx", 'xt')
1867 assert_equal('yes', g:doit_called)
1868
1869 assert_fails('call feedkeys("yy", "xt")', 'E121: Undefined variable: toggle')
1870
Bram Moolenaar19db9e62022-01-11 11:58:19 +00001871 nunmap tt
Bram Moolenaare32c3c42022-01-15 18:26:04 +00001872 nunmap xx
1873 nunmap yy
Bram Moolenaar19db9e62022-01-11 11:58:19 +00001874 unlet g:toggle_loaded
1875 unlet g:toggle_called
1876 delete('Xdir', 'rf')
1877 &rtp = save_rtp
1878enddef
1879
Bram Moolenaar160aa862022-01-10 21:29:57 +00001880def Test_vim9script_autoload_fails()
1881 var lines =<< trim END
1882 vim9script autoload
1883 var n = 0
1884 END
Bram Moolenaarfd218c82022-01-18 16:26:24 +00001885 CheckScriptFailure(lines, 'E475: Invalid argument: autoload')
1886
1887 lines =<< trim END
1888 vim9script noclear noclear
1889 var n = 0
1890 END
1891 CheckScriptFailure(lines, 'E983: Duplicate argument: noclear')
Bram Moolenaar160aa862022-01-10 21:29:57 +00001892enddef
1893
1894def Test_import_autoload_fails()
1895 var lines =<< trim END
1896 vim9script
1897 import autoload autoload 'prefixed.vim'
1898 END
1899 CheckScriptFailure(lines, 'E121: Undefined variable: autoload')
1900
1901 lines =<< trim END
1902 vim9script
Bram Moolenaar1836d612022-01-18 13:14:47 +00001903 import autoload './doesNotExist.vim'
Bram Moolenaar160aa862022-01-10 21:29:57 +00001904 END
1905 CheckScriptFailure(lines, 'E1264:')
Bram Moolenaar1836d612022-01-18 13:14:47 +00001906
1907 lines =<< trim END
1908 vim9script
1909 import autoload '/dir/doesNotExist.vim'
1910 END
1911 CheckScriptFailure(lines, 'E1264:')
1912
1913 lines =<< trim END
1914 vim9script
1915 import autoload 'doesNotExist.vim'
1916 END
1917 CheckScriptFailure(lines, 'E1053: Could not import "doesNotExist.vim"')
Bram Moolenaar160aa862022-01-10 21:29:57 +00001918enddef
1919
1920" test disassembling an auto-loaded function starting with "debug"
1921def Test_vim9_autoload_disass()
1922 mkdir('Xdir/autoload', 'p')
1923 var save_rtp = &rtp
1924 exe 'set rtp^=' .. getcwd() .. '/Xdir'
1925
1926 var lines =<< trim END
1927 vim9script
1928 def debugit#test(): string
1929 return 'debug'
1930 enddef
1931 END
1932 writefile(lines, 'Xdir/autoload/debugit.vim')
1933
1934 lines =<< trim END
1935 vim9script
1936 def profileit#test(): string
1937 return 'profile'
1938 enddef
1939 END
1940 writefile(lines, 'Xdir/autoload/profileit.vim')
1941
1942 lines =<< trim END
1943 vim9script
1944 assert_equal('debug', debugit#test())
1945 disass debugit#test
1946 assert_equal('profile', profileit#test())
1947 disass profileit#test
1948 END
1949 CheckScriptSuccess(lines)
1950
1951 delete('Xdir', 'rf')
1952 &rtp = save_rtp
1953enddef
1954
1955" test using a vim9script that is auto-loaded from an autocmd
1956def Test_vim9_aucmd_autoload()
1957 var lines =<< trim END
1958 vim9script
1959 def foo#test()
1960 echomsg getreg('"')
1961 enddef
1962 END
1963
1964 mkdir('Xdir/autoload', 'p')
1965 writefile(lines, 'Xdir/autoload/foo.vim')
1966 var save_rtp = &rtp
1967 exe 'set rtp^=' .. getcwd() .. '/Xdir'
1968 augroup test
1969 autocmd TextYankPost * call foo#test()
1970 augroup END
1971
1972 normal Y
1973
1974 augroup test
1975 autocmd!
1976 augroup END
1977 delete('Xdir', 'rf')
1978 &rtp = save_rtp
1979enddef
1980
Bram Moolenaar3049fcf2022-01-13 19:25:50 +00001981" test using a autoloaded file that is case sensitive
1982def Test_vim9_autoload_case_sensitive()
1983 var lines =<< trim END
Bram Moolenaarfd218c82022-01-18 16:26:24 +00001984 vim9script
Bram Moolenaar3049fcf2022-01-13 19:25:50 +00001985 export def CaseSensitive(): string
1986 return 'done'
1987 enddef
1988 END
1989
1990 mkdir('Xdir/autoload', 'p')
1991 writefile(lines, 'Xdir/autoload/CaseSensitive.vim')
1992 var save_rtp = &rtp
1993 exe 'set rtp^=' .. getcwd() .. '/Xdir'
1994
1995 lines =<< trim END
1996 vim9script
1997 import autoload 'CaseSensitive.vim'
1998 assert_equal('done', CaseSensitive.CaseSensitive())
1999 END
2000 CheckScriptSuccess(lines)
2001
Bram Moolenaarbfac4092022-01-16 11:12:12 +00002002 if !has('fname_case')
2003 lines =<< trim END
2004 vim9script
2005 import autoload 'CaseSensitive.vim'
2006 import autoload 'casesensitive.vim'
2007 END
2008 CheckScriptFailure(lines, 'E1262:')
2009 endif
2010
Bram Moolenaar3049fcf2022-01-13 19:25:50 +00002011 delete('Xdir', 'rf')
2012 &rtp = save_rtp
2013enddef
2014
Bram Moolenaar160aa862022-01-10 21:29:57 +00002015" This was causing a crash because suppress_errthrow wasn't reset.
2016def Test_vim9_autoload_error()
2017 var lines =<< trim END
2018 vim9script
2019 def crash#func()
2020 try
2021 for x in List()
2022 endfor
2023 catch
2024 endtry
2025 g:ok = true
2026 enddef
2027 fu List()
2028 invalid
2029 endfu
2030 try
2031 alsoinvalid
2032 catch /wontmatch/
2033 endtry
2034 END
2035 call mkdir('Xruntime/autoload', 'p')
2036 call writefile(lines, 'Xruntime/autoload/crash.vim')
2037
2038 # run in a separate Vim to avoid the side effects of assert_fails()
2039 lines =<< trim END
2040 exe 'set rtp^=' .. getcwd() .. '/Xruntime'
2041 call crash#func()
2042 call writefile(['ok'], 'Xdidit')
2043 qall!
2044 END
2045 writefile(lines, 'Xscript')
2046 RunVim([], [], '-S Xscript')
2047 assert_equal(['ok'], readfile('Xdidit'))
2048
2049 delete('Xdidit')
2050 delete('Xscript')
2051 delete('Xruntime', 'rf')
2052
2053 lines =<< trim END
2054 vim9script
2055 var foo#bar = 'asdf'
2056 END
2057 CheckScriptFailure(lines, 'E461: Illegal variable name: foo#bar', 2)
2058enddef
2059
Bram Moolenaard8448622022-01-07 21:39:52 +00002060
2061" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker