blob: 428de1c1358765bcf651ddf44a9e3c2820baa1f3 [file] [log] [blame]
Bram Moolenaar971caf52020-09-21 22:21:42 +02001" Test Vim9 assignments
2
3source check.vim
4source vim9.vim
5
6let s:appendToMe = 'xxx'
7let s:addToMe = 111
8let g:existing = 'yes'
9let g:inc_counter = 1
10let $SOME_ENV_VAR = 'some'
11let g:alist = [7]
Bram Moolenaar5b5ae292021-02-20 17:04:02 +010012let g:adict = #{a: 1}
Bram Moolenaar971caf52020-09-21 22:21:42 +020013let g:astring = 'text'
14
15def Test_assignment_bool()
Bram Moolenaar30fd8202020-09-26 15:09:30 +020016 var bool1: bool = true
Bram Moolenaar971caf52020-09-21 22:21:42 +020017 assert_equal(v:true, bool1)
Bram Moolenaar30fd8202020-09-26 15:09:30 +020018 var bool2: bool = false
Bram Moolenaar971caf52020-09-21 22:21:42 +020019 assert_equal(v:false, bool2)
20
Bram Moolenaar30fd8202020-09-26 15:09:30 +020021 var bool3: bool = 0
Bram Moolenaar971caf52020-09-21 22:21:42 +020022 assert_equal(false, bool3)
Bram Moolenaar30fd8202020-09-26 15:09:30 +020023 var bool4: bool = 1
Bram Moolenaar971caf52020-09-21 22:21:42 +020024 assert_equal(true, bool4)
25
Bram Moolenaar2bb26582020-10-03 22:52:39 +020026 var bool5: bool = 1 && true
Bram Moolenaar971caf52020-09-21 22:21:42 +020027 assert_equal(true, bool5)
Bram Moolenaar2bb26582020-10-03 22:52:39 +020028 var bool6: bool = 0 && 1
Bram Moolenaar971caf52020-09-21 22:21:42 +020029 assert_equal(false, bool6)
Bram Moolenaar2bb26582020-10-03 22:52:39 +020030 var bool7: bool = 0 || 1 && true
Bram Moolenaar971caf52020-09-21 22:21:42 +020031 assert_equal(true, bool7)
32
Bram Moolenaar30fd8202020-09-26 15:09:30 +020033 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +020034 vim9script
35 def GetFlag(): bool
Bram Moolenaar30fd8202020-09-26 15:09:30 +020036 var flag: bool = 1
Bram Moolenaar971caf52020-09-21 22:21:42 +020037 return flag
38 enddef
Bram Moolenaar30fd8202020-09-26 15:09:30 +020039 var flag: bool = GetFlag()
Bram Moolenaar971caf52020-09-21 22:21:42 +020040 assert_equal(true, flag)
41 flag = 0
42 assert_equal(false, flag)
43 flag = 1
44 assert_equal(true, flag)
Bram Moolenaar2bb26582020-10-03 22:52:39 +020045 flag = 1 || true
Bram Moolenaar971caf52020-09-21 22:21:42 +020046 assert_equal(true, flag)
Bram Moolenaar2bb26582020-10-03 22:52:39 +020047 flag = 1 && false
Bram Moolenaar971caf52020-09-21 22:21:42 +020048 assert_equal(false, flag)
Bram Moolenaardd1f4262020-12-31 17:41:01 +010049
50 var cp: bool = &cp
51 var fen: bool = &l:fen
Bram Moolenaar971caf52020-09-21 22:21:42 +020052 END
53 CheckScriptSuccess(lines)
Bram Moolenaar30fd8202020-09-26 15:09:30 +020054 CheckDefAndScriptFailure(['var x: bool = 2'], 'E1012:')
55 CheckDefAndScriptFailure(['var x: bool = -1'], 'E1012:')
56 CheckDefAndScriptFailure(['var x: bool = [1]'], 'E1012:')
57 CheckDefAndScriptFailure(['var x: bool = {}'], 'E1012:')
58 CheckDefAndScriptFailure(['var x: bool = "x"'], 'E1012:')
Bram Moolenaar9a562c12021-01-23 13:39:14 +010059
60 CheckDefAndScriptFailure(['var x: bool = "x"', '', 'eval 0'], 'E1012:', 1)
Bram Moolenaar971caf52020-09-21 22:21:42 +020061enddef
62
63def Test_syntax()
Bram Moolenaar98309382020-09-27 21:58:45 +020064 var name = 234
Bram Moolenaar30fd8202020-09-26 15:09:30 +020065 var other: list<string> = ['asdf']
Bram Moolenaar971caf52020-09-21 22:21:42 +020066enddef
67
68def Test_assignment()
Bram Moolenaar30fd8202020-09-26 15:09:30 +020069 CheckDefFailure(['var x:string'], 'E1069:')
70 CheckDefFailure(['var x:string = "x"'], 'E1069:')
71 CheckDefFailure(['var a:string = "x"'], 'E1069:')
Bram Moolenaar2949cfd2020-12-31 21:28:47 +010072 CheckDefFailure(['var lambda = () => "lambda"'], 'E704:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +020073 CheckScriptFailure(['var x = "x"'], 'E1124:')
Bram Moolenaar971caf52020-09-21 22:21:42 +020074
Bram Moolenaar3f327882021-03-17 20:56:38 +010075 # lower case name is OK for a list
76 var lambdaLines =<< trim END
77 var lambdaList: list<func> = [Test_syntax]
78 lambdaList[0] = () => "lambda"
79 END
80 CheckDefAndScriptSuccess(lambdaLines)
81
Bram Moolenaar30fd8202020-09-26 15:09:30 +020082 var nr: number = 1234
83 CheckDefFailure(['var nr: number = "asdf"'], 'E1012:')
Bram Moolenaar971caf52020-09-21 22:21:42 +020084
Bram Moolenaar30fd8202020-09-26 15:09:30 +020085 var a: number = 6 #comment
Bram Moolenaar971caf52020-09-21 22:21:42 +020086 assert_equal(6, a)
87
88 if has('channel')
Bram Moolenaar30fd8202020-09-26 15:09:30 +020089 var chan1: channel
Bram Moolenaar218450a2020-10-17 18:51:52 +020090 assert_equal('fail', ch_status(chan1))
91
Bram Moolenaar30fd8202020-09-26 15:09:30 +020092 var job1: job
Bram Moolenaar218450a2020-10-17 18:51:52 +020093 assert_equal('fail', job_status(job1))
94
Bram Moolenaarfc8aa6d2020-10-12 20:31:26 +020095 # calling job_start() is in test_vim9_fails.vim, it causes leak reports
Bram Moolenaar971caf52020-09-21 22:21:42 +020096 endif
97 if has('float')
Bram Moolenaar30fd8202020-09-26 15:09:30 +020098 var float1: float = 3.4
Bram Moolenaar971caf52020-09-21 22:21:42 +020099 endif
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200100 var Funky1: func
101 var Funky2: func = function('len')
102 var Party2: func = funcref('g:Test_syntax')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200103
104 g:newvar = 'new' #comment
105 assert_equal('new', g:newvar)
106
107 assert_equal('yes', g:existing)
108 g:existing = 'no'
109 assert_equal('no', g:existing)
110
111 v:char = 'abc'
112 assert_equal('abc', v:char)
113
114 $ENVVAR = 'foobar'
115 assert_equal('foobar', $ENVVAR)
116 $ENVVAR = ''
117
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200118 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200119 vim9script
120 $ENVVAR = 'barfoo'
121 assert_equal('barfoo', $ENVVAR)
122 $ENVVAR = ''
123 END
124 CheckScriptSuccess(lines)
125
126 s:appendToMe ..= 'yyy'
127 assert_equal('xxxyyy', s:appendToMe)
128 s:addToMe += 222
129 assert_equal(333, s:addToMe)
130 s:newVar = 'new'
131 assert_equal('new', s:newVar)
132
133 set ts=7
Bram Moolenaardd1f4262020-12-31 17:41:01 +0100134 var ts: number = &ts
135 assert_equal(7, ts)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200136 &ts += 1
137 assert_equal(8, &ts)
138 &ts -= 3
139 assert_equal(5, &ts)
140 &ts *= 2
141 assert_equal(10, &ts)
142 &ts /= 3
143 assert_equal(3, &ts)
144 set ts=10
145 &ts %= 4
146 assert_equal(2, &ts)
147
148 if has('float')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200149 var f100: float = 100.0
Bram Moolenaar971caf52020-09-21 22:21:42 +0200150 f100 /= 5
151 assert_equal(20.0, f100)
152
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200153 var f200: float = 200.0
Bram Moolenaar971caf52020-09-21 22:21:42 +0200154 f200 /= 5.0
155 assert_equal(40.0, f200)
156
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200157 CheckDefFailure(['var nr: number = 200', 'nr /= 5.0'], 'E1012:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200158 endif
159
160 lines =<< trim END
161 &ts = 6
162 &ts += 3
163 assert_equal(9, &ts)
164
165 &l:ts = 6
166 assert_equal(6, &ts)
167 &l:ts += 2
168 assert_equal(8, &ts)
169
170 &g:ts = 6
171 assert_equal(6, &g:ts)
172 &g:ts += 2
173 assert_equal(8, &g:ts)
Bram Moolenaar0ea04402021-01-04 13:37:54 +0100174
175 &number = true
176 assert_equal(true, &number)
177 &number = 0
178 assert_equal(false, &number)
179 &number = 1
180 assert_equal(true, &number)
181 &number = false
182 assert_equal(false, &number)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200183 END
184 CheckDefAndScriptSuccess(lines)
185
186 CheckDefFailure(['&notex += 3'], 'E113:')
187 CheckDefFailure(['&ts ..= "xxx"'], 'E1019:')
188 CheckDefFailure(['&ts = [7]'], 'E1012:')
189 CheckDefExecFailure(['&ts = g:alist'], 'E1012: Type mismatch; expected number but got list<number>')
190 CheckDefFailure(['&ts = "xx"'], 'E1012:')
191 CheckDefExecFailure(['&ts = g:astring'], 'E1012: Type mismatch; expected number but got string')
192 CheckDefFailure(['&path += 3'], 'E1012:')
193 CheckDefExecFailure(['&bs = "asdf"'], 'E474:')
194 # test freeing ISN_STOREOPT
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200195 CheckDefFailure(['&ts = 3', 'var asdf'], 'E1022:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200196 &ts = 8
197
198 lines =<< trim END
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200199 var save_TI = &t_TI
Bram Moolenaar971caf52020-09-21 22:21:42 +0200200 &t_TI = ''
201 assert_equal('', &t_TI)
202 &t_TI = 'xxx'
203 assert_equal('xxx', &t_TI)
204 &t_TI = save_TI
205 END
206 CheckDefAndScriptSuccess(lines)
207
208 CheckDefFailure(['&t_TI = 123'], 'E1012:')
209 CheckScriptFailure(['vim9script', '&t_TI = 123'], 'E928:')
210
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200211 CheckDefFailure(['var s:var = 123'], 'E1101:')
212 CheckDefFailure(['var s:var: number'], 'E1101:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200213
214 lines =<< trim END
215 vim9script
216 def SomeFunc()
217 s:var = 123
218 enddef
219 defcompile
220 END
221 CheckScriptFailure(lines, 'E1089:')
222
223 g:inc_counter += 1
224 assert_equal(2, g:inc_counter)
225
Bram Moolenaar7bf9a072021-08-02 21:55:15 +0200226 if has('float')
227 var f: float
228 f += 1
229 assert_equal(1.0, f)
230 endif
231
Bram Moolenaar971caf52020-09-21 22:21:42 +0200232 $SOME_ENV_VAR ..= 'more'
233 assert_equal('somemore', $SOME_ENV_VAR)
234 CheckDefFailure(['$SOME_ENV_VAR += "more"'], 'E1051:')
235 CheckDefFailure(['$SOME_ENV_VAR += 123'], 'E1012:')
236
Bram Moolenaar971caf52020-09-21 22:21:42 +0200237 v:errmsg = 'none'
238 v:errmsg ..= 'again'
239 assert_equal('noneagain', v:errmsg)
240 CheckDefFailure(['v:errmsg += "more"'], 'E1051:')
241 CheckDefFailure(['v:errmsg += 123'], 'E1012:')
242
Bram Moolenaarfc8aa6d2020-10-12 20:31:26 +0200243 var text =<< trim END
244 some text
245 END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200246enddef
247
Bram Moolenaarf5d52c92021-07-31 22:51:10 +0200248let g:someNumber = 43
249
250def Test_assign_concat()
251 var lines =<< trim END
252 var s = '-'
253 s ..= 99
254 s ..= true
255 s ..= '-'
256 s ..= v:null
257 s ..= g:someNumber
258 assert_equal('-99true-null43', s)
259 END
260 CheckDefAndScriptSuccess(lines)
261
262 lines =<< trim END
263 var s = '-'
264 s ..= [1, 2]
265 END
266 CheckDefAndScriptFailure2(lines, 'E1105: Cannot convert list to string', 'E734: Wrong variable type for .=', 2)
267 lines =<< trim END
268 var s = '-'
269 s ..= {a: 2}
270 END
271 CheckDefAndScriptFailure2(lines, 'E1105: Cannot convert dict to string', 'E734: Wrong variable type for .=', 2)
272enddef
273
Bram Moolenaar13e45d12021-06-26 13:28:35 +0200274def Test_assign_register()
275 var lines =<< trim END
276 @c = 'areg'
277 @c ..= 'add'
278 assert_equal('aregadd', @c)
279
280 @@ = 'some text'
281 assert_equal('some text', getreg('"'))
282 END
283 CheckDefAndScriptSuccess(lines)
284
285 CheckDefFailure(['@a += "more"'], 'E1051:')
286 CheckDefFailure(['@a += 123'], 'E1012:')
287enddef
288
Bram Moolenaard0edaf92021-05-28 21:06:08 +0200289def Test_reserved_name()
290 for name in ['true', 'false', 'null']
291 CheckDefExecAndScriptFailure(['var ' .. name .. ' = 0'], 'E1034:')
292 CheckDefExecAndScriptFailure(['var ' .. name .. ': bool'], 'E1034:')
293 endfor
294enddef
295
Bram Moolenaaraa1959b2021-04-15 22:13:39 +0200296def Test_skipped_assignment()
297 var lines =<< trim END
298 for x in []
299 var i: number = 1
300 while false
301 i += 1
302 endwhile
303 endfor
304 END
305 CheckDefAndScriptSuccess(lines)
306enddef
307
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100308def Test_assign_unpack()
309 var lines =<< trim END
310 var v1: number
311 var v2: number
312 [v1, v2] = [1, 2]
313 assert_equal(1, v1)
314 assert_equal(2, v2)
Bram Moolenaarf93bbd02021-04-10 22:35:43 +0200315
316 [v1, _, v2, _] = [1, 99, 2, 77]
317 assert_equal(1, v1)
318 assert_equal(2, v2)
319
320 [v1, v2; _] = [1, 2, 3, 4, 5]
321 assert_equal(1, v1)
322 assert_equal(2, v2)
Bram Moolenaar035bd1c2021-06-21 19:44:11 +0200323
Bram Moolenaar4d5dfe22021-06-26 13:59:29 +0200324 var reslist = []
325 for text in ['aaa {bbb} ccc', 'ddd {eee} fff']
326 var before: string
327 var middle: string
328 var after: string
329 [_, before, middle, after; _] = text->matchlist('\(.\{-\}\){\(.\{-\}\)}\(.*\)')
330 reslist->add(before)->add(middle)->add(after)
331 endfor
332 assert_equal(['aaa ', 'bbb', ' ccc', 'ddd ', 'eee', ' fff'], reslist)
333
Bram Moolenaar035bd1c2021-06-21 19:44:11 +0200334 var a = 1
335 var b = 3
336 [a, b] += [2, 4]
337 assert_equal(3, a)
338 assert_equal(7, b)
339
340 [a, b] -= [1, 2]
341 assert_equal(2, a)
342 assert_equal(5, b)
343
344 [a, b] *= [3, 2]
345 assert_equal(6, a)
346 assert_equal(10, b)
347
348 [a, b] /= [2, 4]
349 assert_equal(3, a)
350 assert_equal(2, b)
351
352 [a, b] = [17, 15]
353 [a, b] %= [5, 3]
354 assert_equal(2, a)
355 assert_equal(0, b)
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100356 END
357 CheckDefAndScriptSuccess(lines)
358
359 lines =<< trim END
360 var v1: number
361 var v2: number
362 [v1, v2] =
363 END
364 CheckDefFailure(lines, 'E1097:', 5)
365
366 lines =<< trim END
367 var v1: number
368 var v2: number
369 [v1, v2] = xxx
370 END
371 CheckDefFailure(lines, 'E1001:', 3)
372
373 lines =<< trim END
374 var v1: number
375 var v2: number
376 [v1, v2] = popup_clear()
377 END
378 CheckDefFailure(lines, 'E1031:', 3)
379
380 lines =<< trim END
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100381 [v1, v2] = [1, 2]
382 END
383 CheckDefFailure(lines, 'E1089', 1)
384 CheckScriptFailure(['vim9script'] + lines, 'E1089', 2)
385
386 lines =<< trim END
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100387 var v1: number
388 var v2: number
389 [v1, v2] = ''
390 END
391 CheckDefFailure(lines, 'E1012: Type mismatch; expected list<any> but got string', 3)
Bram Moolenaarf785aa12021-02-11 21:19:34 +0100392
393 lines =<< trim END
394 g:values = [false, 0]
395 var x: bool
396 var y: string
397 [x, y] = g:values
398 END
399 CheckDefExecAndScriptFailure(lines, 'E1163: Variable 2: type mismatch, expected string but got number')
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100400enddef
401
Bram Moolenaar7f764942020-12-02 15:11:18 +0100402def Test_assign_linebreak()
403 var nr: number
404 nr =
405 123
406 assert_equal(123, nr)
407
408 var n2: number
409 [nr, n2] =
410 [12, 34]
411 assert_equal(12, nr)
412 assert_equal(34, n2)
413
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100414 CheckDefFailure(["var x = #"], 'E1097:', 3)
Bram Moolenaar77709b12021-04-03 21:01:01 +0200415
416 var lines =<< trim END
417 var x: list<string> = ['a']
418 var y: list<number> = x
419 ->copy()
420 ->copy()
421 END
422 CheckDefFailure(lines, 'E1012:', 2)
Bram Moolenaar6977dba2021-07-04 22:48:12 +0200423
424 lines =<< trim END
425 var x: any
426 x.key = 1
427 + 2
428 + 3
429 + 4
430 + 5
431 END
432 CheckDefExecAndScriptFailure2(lines, 'E1148:', 'E1203:', 2)
Bram Moolenaar7f764942020-12-02 15:11:18 +0100433enddef
434
Bram Moolenaardc234ca2020-11-28 18:52:33 +0100435def Test_assign_index()
436 # list of list
437 var l1: list<number>
438 l1[0] = 123
439 assert_equal([123], l1)
440
441 var l2: list<list<number>>
442 l2[0] = []
443 l2[0][0] = 123
444 assert_equal([[123]], l2)
445
446 var l3: list<list<list<number>>>
447 l3[0] = []
448 l3[0][0] = []
449 l3[0][0][0] = 123
450 assert_equal([[[123]]], l3)
451
452 var lines =<< trim END
453 var l3: list<list<number>>
454 l3[0] = []
455 l3[0][0] = []
456 END
457 CheckDefFailure(lines, 'E1012: Type mismatch; expected number but got list<unknown>', 3)
458
459 # dict of dict
460 var d1: dict<number>
461 d1.one = 1
462 assert_equal({one: 1}, d1)
463
464 var d2: dict<dict<number>>
465 d2.one = {}
466 d2.one.two = 123
467 assert_equal({one: {two: 123}}, d2)
468
469 var d3: dict<dict<dict<number>>>
470 d3.one = {}
471 d3.one.two = {}
472 d3.one.two.three = 123
473 assert_equal({one: {two: {three: 123}}}, d3)
474
Bram Moolenaare97976b2021-08-01 13:17:17 +0200475 # should not read the next line when generating "a.b"
476 var a = {}
477 a.b = {}
478 a.b.c = {}
479 ->copy()
480
Bram Moolenaardc234ca2020-11-28 18:52:33 +0100481 lines =<< trim END
482 var d3: dict<dict<number>>
483 d3.one = {}
484 d3.one.two = {}
485 END
486 CheckDefFailure(lines, 'E1012: Type mismatch; expected number but got dict<unknown>', 3)
487
Bram Moolenaaracbae182020-12-13 18:44:43 +0100488 lines =<< trim END
489 var lines: list<string>
490 lines['a'] = 'asdf'
491 END
Bram Moolenaarf30a14d2021-01-17 21:51:24 +0100492 CheckDefFailure(lines, 'E1012:', 2)
Bram Moolenaaracbae182020-12-13 18:44:43 +0100493
494 lines =<< trim END
495 var lines: string
496 lines[9] = 'asdf'
497 END
498 CheckDefFailure(lines, 'E1141:', 2)
499
Bram Moolenaardc234ca2020-11-28 18:52:33 +0100500 # list of dict
501 var ld: list<dict<number>>
502 ld[0] = {}
503 ld[0].one = 123
504 assert_equal([{one: 123}], ld)
505
506 lines =<< trim END
507 var ld: list<dict<number>>
508 ld[0] = []
509 END
510 CheckDefFailure(lines, 'E1012: Type mismatch; expected dict<number> but got list<unknown>', 2)
511
512 # dict of list
513 var dl: dict<list<number>>
514 dl.one = []
515 dl.one[0] = 123
516 assert_equal({one: [123]}, dl)
517
518 lines =<< trim END
519 var dl: dict<list<number>>
520 dl.one = {}
521 END
522 CheckDefFailure(lines, 'E1012: Type mismatch; expected list<number> but got dict<unknown>', 2)
523enddef
524
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200525def Test_extend_list()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200526 var lines =<< trim END
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200527 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200528 var l: list<number>
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200529 l += [123]
530 assert_equal([123], l)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100531 END
532 CheckScriptSuccess(lines)
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200533
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100534 lines =<< trim END
535 vim9script
536 var list: list<string>
537 extend(list, ['x'])
538 assert_equal(['x'], list)
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200539 END
540 CheckScriptSuccess(lines)
Bram Moolenaar1a739232020-10-10 15:37:58 +0200541
542 # appending to NULL list from a function
543 lines =<< trim END
544 vim9script
545 var list: list<string>
546 def Func()
547 list += ['a', 'b']
548 enddef
549 Func()
550 assert_equal(['a', 'b'], list)
551 END
552 CheckScriptSuccess(lines)
Bram Moolenaar3beaf9c2020-12-18 17:23:14 +0100553 lines =<< trim END
554 vim9script
555 var list: list<string>
556 def Func()
557 extend(list, ['x', 'b'])
558 enddef
559 Func()
560 assert_equal(['x', 'b'], list)
561 END
562 CheckScriptSuccess(lines)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100563
564 lines =<< trim END
565 vim9script
566 var l: list<string> = test_null_list()
567 extend(l, ['x'])
568 assert_equal(['x'], l)
569 END
570 CheckScriptSuccess(lines)
571
572 lines =<< trim END
573 vim9script
574 extend(test_null_list(), ['x'])
575 END
576 CheckScriptFailure(lines, 'E1134:', 2)
577enddef
578
579def Test_extend_dict()
580 var lines =<< trim END
581 vim9script
582 var d: dict<number>
Bram Moolenaare0de1712020-12-02 17:36:54 +0100583 extend(d, {a: 1})
584 assert_equal({a: 1}, d)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100585
586 var d2: dict<number>
587 d2['one'] = 1
Bram Moolenaare0de1712020-12-02 17:36:54 +0100588 assert_equal({one: 1}, d2)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100589 END
590 CheckScriptSuccess(lines)
591
592 lines =<< trim END
593 vim9script
594 var d: dict<string> = test_null_dict()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100595 extend(d, {a: 'x'})
596 assert_equal({a: 'x'}, d)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100597 END
598 CheckScriptSuccess(lines)
599
600 lines =<< trim END
601 vim9script
Bram Moolenaare0de1712020-12-02 17:36:54 +0100602 extend(test_null_dict(), {a: 'x'})
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100603 END
604 CheckScriptFailure(lines, 'E1133:', 2)
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200605enddef
606
Bram Moolenaar971caf52020-09-21 22:21:42 +0200607def Test_single_letter_vars()
608 # single letter variables
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200609 var a: number = 123
Bram Moolenaar971caf52020-09-21 22:21:42 +0200610 a = 123
611 assert_equal(123, a)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200612 var b: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200613 b = 123
614 assert_equal(123, b)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200615 var g: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200616 g = 123
617 assert_equal(123, g)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200618 var s: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200619 s = 123
620 assert_equal(123, s)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200621 var t: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200622 t = 123
623 assert_equal(123, t)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200624 var v: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200625 v = 123
626 assert_equal(123, v)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200627 var w: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200628 w = 123
629 assert_equal(123, w)
630enddef
631
632def Test_vim9_single_char_vars()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200633 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200634 vim9script
635
636 # single character variable declarations work
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200637 var a: string
638 var b: number
639 var l: list<any>
640 var s: string
641 var t: number
642 var v: number
643 var w: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200644
645 # script-local variables can be used without s: prefix
646 a = 'script-a'
647 b = 111
648 l = [1, 2, 3]
649 s = 'script-s'
650 t = 222
651 v = 333
652 w = 444
653
654 assert_equal('script-a', a)
655 assert_equal(111, b)
656 assert_equal([1, 2, 3], l)
657 assert_equal('script-s', s)
658 assert_equal(222, t)
659 assert_equal(333, v)
660 assert_equal(444, w)
661 END
662 writefile(lines, 'Xsinglechar')
663 source Xsinglechar
664 delete('Xsinglechar')
665enddef
666
667def Test_assignment_list()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200668 var list1: list<bool> = [false, true, false]
669 var list2: list<number> = [1, 2, 3]
670 var list3: list<string> = ['sdf', 'asdf']
671 var list4: list<any> = ['yes', true, 1234]
672 var list5: list<blob> = [0z01, 0z02]
Bram Moolenaar971caf52020-09-21 22:21:42 +0200673
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200674 var listS: list<string> = []
675 var listN: list<number> = []
Bram Moolenaar971caf52020-09-21 22:21:42 +0200676
677 assert_equal([1, 2, 3], list2)
678 list2[-1] = 99
679 assert_equal([1, 2, 99], list2)
680 list2[-2] = 88
681 assert_equal([1, 88, 99], list2)
682 list2[-3] = 77
683 assert_equal([77, 88, 99], list2)
684 list2 += [100]
685 assert_equal([77, 88, 99, 100], list2)
686
687 list3 += ['end']
688 assert_equal(['sdf', 'asdf', 'end'], list3)
689
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200690 CheckDefExecFailure(['var ll = [1, 2, 3]', 'll[-4] = 6'], 'E684:')
691 CheckDefExecFailure(['var [v1, v2] = [1, 2]'], 'E1092:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200692
693 # type becomes list<any>
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200694 var somelist = rand() > 0 ? [1, 2, 3] : ['a', 'b', 'c']
Bram Moolenaar4f5e3972020-12-21 17:30:50 +0100695
696 var lines =<< trim END
697 var d = {dd: test_null_list()}
698 d.dd[0] = 0
699 END
700 CheckDefExecFailure(lines, 'E1147:', 2)
Bram Moolenaar7a40ff02021-07-04 15:54:08 +0200701
702 lines =<< trim END
703 def OneArg(x: bool)
704 enddef
705 def TwoArgs(x: bool, y: bool)
706 enddef
707 var fl: list<func(bool, bool, bool)> = [OneArg, TwoArgs]
708 END
709 CheckDefExecAndScriptFailure(lines, 'E1012:', 5)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200710enddef
711
Bram Moolenaar97f227d2021-07-04 20:20:52 +0200712def PartFuncBool(b: bool): string
Bram Moolenaarb7480cd2021-07-04 18:28:15 +0200713 return 'done'
714enddef
715
716def Test_assignment_partial()
Bram Moolenaar97f227d2021-07-04 20:20:52 +0200717 var lines =<< trim END
718 var Partial: func(): string = function(PartFuncBool, [true])
719 assert_equal('done', Partial())
720 END
721 CheckDefAndScriptSuccess(lines)
Bram Moolenaar22f85d02021-07-04 23:29:30 +0200722
723 lines =<< trim END
724 vim9script
725 def Func(b: bool)
726 enddef
727 var Ref: func = function(Func, [true])
728 assert_equal('func()', typename(Ref))
729 Ref()
730 END
731 CheckScriptSuccess(lines)
Bram Moolenaarb7480cd2021-07-04 18:28:15 +0200732enddef
733
Bram Moolenaarf30a14d2021-01-17 21:51:24 +0100734def Test_assignment_list_any_index()
735 var l: list<number> = [1, 2]
736 for [x, y, _]
737 in [[0, 1, ''], [1, 3, '']]
738 l[x] = l[x] + y
739 endfor
740 assert_equal([2, 5], l)
741enddef
742
Bram Moolenaar971caf52020-09-21 22:21:42 +0200743def Test_assignment_list_vim9script()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200744 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200745 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200746 var v1: number
747 var v2: number
748 var v3: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200749 [v1, v2, v3] = [1, 2, 3]
750 assert_equal([1, 2, 3], [v1, v2, v3])
751 END
752 CheckScriptSuccess(lines)
753enddef
754
755def Test_assignment_dict()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100756 var dict1: dict<bool> = {one: false, two: true}
757 var dict2: dict<number> = {one: 1, two: 2}
758 var dict3: dict<string> = {key: 'value'}
759 var dict4: dict<any> = {one: 1, two: '2'}
760 var dict5: dict<blob> = {one: 0z01, two: 0z02}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200761
762 # overwrite
763 dict3['key'] = 'another'
Bram Moolenaare0de1712020-12-02 17:36:54 +0100764 assert_equal(dict3, {key: 'another'})
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100765 dict3.key = 'yet another'
Bram Moolenaare0de1712020-12-02 17:36:54 +0100766 assert_equal(dict3, {key: 'yet another'})
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100767
Bram Moolenaard24602f2020-12-20 15:20:56 +0100768 # member "any" can also be a dict and assigned to
769 var anydict: dict<any> = {nest: {}, nr: 0}
770 anydict.nest['this'] = 123
771 anydict.nest.that = 456
772 assert_equal({nest: {this: 123, that: 456}, nr: 0}, anydict)
773
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100774 var lines =<< trim END
Bram Moolenaard88dc4d2020-12-20 14:49:42 +0100775 var dd = {}
776 dd.two = 2
777 assert_equal({two: 2}, dd)
778 END
Bram Moolenaar4f5e3972020-12-21 17:30:50 +0100779 CheckDefAndScriptSuccess(lines)
780
781 lines =<< trim END
782 var d = {dd: {}}
783 d.dd[0] = 2
784 d.dd['x'] = 3
785 d.dd.y = 4
786 assert_equal({dd: {0: 2, x: 3, y: 4}}, d)
787 END
788 CheckDefAndScriptSuccess(lines)
Bram Moolenaard88dc4d2020-12-20 14:49:42 +0100789
790 lines =<< trim END
Bram Moolenaare0de1712020-12-02 17:36:54 +0100791 var dd = {one: 1}
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100792 dd.one) = 2
793 END
Bram Moolenaard88dc4d2020-12-20 14:49:42 +0100794 CheckDefFailure(lines, 'E488:', 2)
795
796 lines =<< trim END
797 var dd = {one: 1}
798 var dd.one = 2
799 END
800 CheckDefAndScriptFailure(lines, 'E1017:', 2)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200801
802 # empty key can be used
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200803 var dd = {}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200804 dd[""] = 6
Bram Moolenaare0de1712020-12-02 17:36:54 +0100805 assert_equal({['']: 6}, dd)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200806
807 # type becomes dict<any>
Bram Moolenaare0de1712020-12-02 17:36:54 +0100808 var somedict = rand() > 0 ? {a: 1, b: 2} : {a: 'a', b: 'b'}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200809
810 # assignment to script-local dict
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100811 lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200812 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200813 var test: dict<any> = {}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200814 def FillDict(): dict<any>
815 test['a'] = 43
816 return test
817 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +0100818 assert_equal({a: 43}, FillDict())
Bram Moolenaar971caf52020-09-21 22:21:42 +0200819 END
820 CheckScriptSuccess(lines)
821
822 lines =<< trim END
823 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200824 var test: dict<any>
Bram Moolenaar971caf52020-09-21 22:21:42 +0200825 def FillDict(): dict<any>
826 test['a'] = 43
827 return test
828 enddef
829 FillDict()
Bram Moolenaar3beaf9c2020-12-18 17:23:14 +0100830 assert_equal({a: 43}, test)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200831 END
Bram Moolenaar3beaf9c2020-12-18 17:23:14 +0100832 CheckScriptSuccess(lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200833
834 # assignment to global dict
835 lines =<< trim END
836 vim9script
837 g:test = {}
838 def FillDict(): dict<any>
839 g:test['a'] = 43
840 return g:test
841 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +0100842 assert_equal({a: 43}, FillDict())
Bram Moolenaar971caf52020-09-21 22:21:42 +0200843 END
844 CheckScriptSuccess(lines)
845
846 # assignment to buffer dict
847 lines =<< trim END
848 vim9script
849 b:test = {}
850 def FillDict(): dict<any>
851 b:test['a'] = 43
852 return b:test
853 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +0100854 assert_equal({a: 43}, FillDict())
Bram Moolenaar971caf52020-09-21 22:21:42 +0200855 END
856 CheckScriptSuccess(lines)
Bram Moolenaar4f5e3972020-12-21 17:30:50 +0100857
858 lines =<< trim END
859 var d = {dd: test_null_dict()}
860 d.dd[0] = 0
861 END
862 CheckDefExecFailure(lines, 'E1103:', 2)
863
864 lines =<< trim END
865 var d = {dd: 'string'}
866 d.dd[0] = 0
867 END
868 CheckDefExecFailure(lines, 'E1148:', 2)
Bram Moolenaar3a3b10e2021-06-26 15:00:59 +0200869
870 lines =<< trim END
871 var n: any
872 n.key = 5
873 END
874 CheckDefExecAndScriptFailure2(lines, 'E1148:', 'E1203: Dot can only be used on a dictionary: n.key = 5', 2)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200875enddef
876
877def Test_assignment_local()
878 # Test in a separated file in order not to the current buffer/window/tab is
879 # changed.
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200880 var script_lines: list<string> =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200881 let b:existing = 'yes'
882 let w:existing = 'yes'
883 let t:existing = 'yes'
884
885 def Test_assignment_local_internal()
886 b:newvar = 'new'
887 assert_equal('new', b:newvar)
888 assert_equal('yes', b:existing)
889 b:existing = 'no'
890 assert_equal('no', b:existing)
891 b:existing ..= 'NO'
892 assert_equal('noNO', b:existing)
893
894 w:newvar = 'new'
895 assert_equal('new', w:newvar)
896 assert_equal('yes', w:existing)
897 w:existing = 'no'
898 assert_equal('no', w:existing)
899 w:existing ..= 'NO'
900 assert_equal('noNO', w:existing)
901
902 t:newvar = 'new'
903 assert_equal('new', t:newvar)
904 assert_equal('yes', t:existing)
905 t:existing = 'no'
906 assert_equal('no', t:existing)
907 t:existing ..= 'NO'
908 assert_equal('noNO', t:existing)
909 enddef
910 call Test_assignment_local_internal()
911 END
912 CheckScriptSuccess(script_lines)
913enddef
914
915def Test_assignment_default()
Bram Moolenaar971caf52020-09-21 22:21:42 +0200916 # Test default values.
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200917 var thebool: bool
Bram Moolenaar971caf52020-09-21 22:21:42 +0200918 assert_equal(v:false, thebool)
919
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200920 var thenumber: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200921 assert_equal(0, thenumber)
922
923 if has('float')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200924 var thefloat: float
Bram Moolenaar971caf52020-09-21 22:21:42 +0200925 assert_equal(0.0, thefloat)
926 endif
927
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200928 var thestring: string
Bram Moolenaar971caf52020-09-21 22:21:42 +0200929 assert_equal('', thestring)
930
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200931 var theblob: blob
Bram Moolenaar971caf52020-09-21 22:21:42 +0200932 assert_equal(0z, theblob)
933
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200934 var Thefunc: func
Bram Moolenaar971caf52020-09-21 22:21:42 +0200935 assert_equal(test_null_function(), Thefunc)
936
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200937 var thelist: list<any>
Bram Moolenaar971caf52020-09-21 22:21:42 +0200938 assert_equal([], thelist)
939
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200940 var thedict: dict<any>
Bram Moolenaar971caf52020-09-21 22:21:42 +0200941 assert_equal({}, thedict)
942
943 if has('channel')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200944 var thejob: job
Bram Moolenaar971caf52020-09-21 22:21:42 +0200945 assert_equal(test_null_job(), thejob)
946
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200947 var thechannel: channel
Bram Moolenaar971caf52020-09-21 22:21:42 +0200948 assert_equal(test_null_channel(), thechannel)
949
950 if has('unix') && executable('cat')
951 # check with non-null job and channel, types must match
Bram Moolenaare0de1712020-12-02 17:36:54 +0100952 thejob = job_start("cat ", {})
Bram Moolenaar971caf52020-09-21 22:21:42 +0200953 thechannel = job_getchannel(thejob)
954 job_stop(thejob, 'kill')
955 endif
956 endif
957
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200958 var nr = 1234 | nr = 5678
Bram Moolenaar971caf52020-09-21 22:21:42 +0200959 assert_equal(5678, nr)
960enddef
961
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100962let scriptvar = 'init'
963
Bram Moolenaar971caf52020-09-21 22:21:42 +0200964def Test_assignment_var_list()
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100965 var lines =<< trim END
966 var v1: string
967 var v2: string
968 var vrem: list<string>
969 [v1] = ['aaa']
970 assert_equal('aaa', v1)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200971
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100972 [v1, v2] = ['one', 'two']
973 assert_equal('one', v1)
974 assert_equal('two', v2)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200975
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100976 [v1, v2; vrem] = ['one', 'two']
977 assert_equal('one', v1)
978 assert_equal('two', v2)
979 assert_equal([], vrem)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200980
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100981 [v1, v2; vrem] = ['one', 'two', 'three']
982 assert_equal('one', v1)
983 assert_equal('two', v2)
984 assert_equal(['three'], vrem)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200985
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100986 [&ts, &sw] = [3, 4]
987 assert_equal(3, &ts)
988 assert_equal(4, &sw)
989 set ts=8 sw=4
Bram Moolenaar004d9b02020-11-30 21:40:03 +0100990
991 [@a, @z] = ['aa', 'zz']
992 assert_equal('aa', @a)
993 assert_equal('zz', @z)
994
995 [$SOME_VAR, $OTHER_VAR] = ['some', 'other']
996 assert_equal('some', $SOME_VAR)
997 assert_equal('other', $OTHER_VAR)
998
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100999 [g:globalvar, b:bufvar, w:winvar, t:tabvar, v:errmsg] =
1000 ['global', 'buf', 'win', 'tab', 'error']
Bram Moolenaar004d9b02020-11-30 21:40:03 +01001001 assert_equal('global', g:globalvar)
Bram Moolenaar004d9b02020-11-30 21:40:03 +01001002 assert_equal('buf', b:bufvar)
1003 assert_equal('win', w:winvar)
1004 assert_equal('tab', t:tabvar)
1005 assert_equal('error', v:errmsg)
1006 unlet g:globalvar
Bram Moolenaarda7c20c2020-11-30 21:12:19 +01001007 END
1008 CheckDefAndScriptSuccess(lines)
Bram Moolenaar3862ea32021-01-01 21:05:55 +01001009
1010 [g:globalvar, s:scriptvar, b:bufvar] = ['global', 'script', 'buf']
1011 assert_equal('global', g:globalvar)
1012 assert_equal('script', s:scriptvar)
1013 assert_equal('buf', b:bufvar)
1014
1015 lines =<< trim END
1016 vim9script
1017 var s:scriptvar = 'init'
1018 [g:globalvar, s:scriptvar, w:winvar] = ['global', 'script', 'win']
1019 assert_equal('global', g:globalvar)
1020 assert_equal('script', s:scriptvar)
1021 assert_equal('win', w:winvar)
1022 END
1023 CheckScriptSuccess(lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001024enddef
1025
1026def Test_assignment_vim9script()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001027 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +02001028 vim9script
1029 def Func(): list<number>
1030 return [1, 2]
1031 enddef
Bram Moolenaar98309382020-09-27 21:58:45 +02001032 var name1: number
1033 var name2: number
1034 [name1, name2] =
Bram Moolenaar971caf52020-09-21 22:21:42 +02001035 Func()
Bram Moolenaar98309382020-09-27 21:58:45 +02001036 assert_equal(1, name1)
1037 assert_equal(2, name2)
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001038 var ll =
Bram Moolenaar971caf52020-09-21 22:21:42 +02001039 Func()
1040 assert_equal([1, 2], ll)
1041
1042 @/ = 'text'
1043 assert_equal('text', @/)
1044 @0 = 'zero'
1045 assert_equal('zero', @0)
1046 @1 = 'one'
1047 assert_equal('one', @1)
1048 @9 = 'nine'
1049 assert_equal('nine', @9)
1050 @- = 'minus'
1051 assert_equal('minus', @-)
1052 if has('clipboard_working')
1053 @* = 'star'
1054 assert_equal('star', @*)
1055 @+ = 'plus'
1056 assert_equal('plus', @+)
1057 endif
1058
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001059 var a: number = 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001060 assert_equal(123, a)
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001061 var s: string = 'yes'
Bram Moolenaar971caf52020-09-21 22:21:42 +02001062 assert_equal('yes', s)
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001063 var b: number = 42
Bram Moolenaar971caf52020-09-21 22:21:42 +02001064 assert_equal(42, b)
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001065 var w: number = 43
Bram Moolenaar971caf52020-09-21 22:21:42 +02001066 assert_equal(43, w)
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001067 var t: number = 44
Bram Moolenaar971caf52020-09-21 22:21:42 +02001068 assert_equal(44, t)
Bram Moolenaar7a66a172020-10-16 19:56:12 +02001069
1070 var to_var = 0
1071 to_var = 3
1072 assert_equal(3, to_var)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001073 END
1074 CheckScriptSuccess(lines)
Bram Moolenaar06f9c692020-09-27 21:27:40 +02001075
1076 lines =<< trim END
1077 vim9script
1078 var n: number
1079 def Func()
1080 n = 'string'
1081 enddef
1082 defcompile
1083 END
1084 CheckScriptFailure(lines, 'E1012: Type mismatch; expected number but got string')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001085enddef
1086
1087def Mess(): string
1088 v:foldstart = 123
1089 return 'xxx'
1090enddef
1091
1092def Test_assignment_failure()
Bram Moolenaar98309382020-09-27 21:58:45 +02001093 CheckDefFailure(['var name=234'], 'E1004:')
1094 CheckDefFailure(['var name =234'], 'E1004:')
1095 CheckDefFailure(['var name= 234'], 'E1004:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001096
Bram Moolenaar98309382020-09-27 21:58:45 +02001097 CheckScriptFailure(['vim9script', 'var name=234'], 'E1004:')
1098 CheckScriptFailure(['vim9script', 'var name=234'], "before and after '='")
1099 CheckScriptFailure(['vim9script', 'var name =234'], 'E1004:')
1100 CheckScriptFailure(['vim9script', 'var name= 234'], 'E1004:')
1101 CheckScriptFailure(['vim9script', 'var name = 234', 'name+=234'], 'E1004:')
1102 CheckScriptFailure(['vim9script', 'var name = 234', 'name+=234'], "before and after '+='")
1103 CheckScriptFailure(['vim9script', 'var name = "x"', 'name..="y"'], 'E1004:')
1104 CheckScriptFailure(['vim9script', 'var name = "x"', 'name..="y"'], "before and after '..='")
Bram Moolenaar971caf52020-09-21 22:21:42 +02001105
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001106 CheckDefFailure(['var true = 1'], 'E1034:')
1107 CheckDefFailure(['var false = 1'], 'E1034:')
Bram Moolenaar74235772021-06-12 14:53:05 +02001108 CheckDefFailure(['var null = 1'], 'E1034:')
1109 CheckDefFailure(['var this = 1'], 'E1034:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001110
1111 CheckDefFailure(['[a; b; c] = g:list'], 'E452:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001112 CheckDefExecFailure(['var a: number',
Bram Moolenaar971caf52020-09-21 22:21:42 +02001113 '[a] = test_null_list()'], 'E1093:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001114 CheckDefExecFailure(['var a: number',
Bram Moolenaar971caf52020-09-21 22:21:42 +02001115 '[a] = []'], 'E1093:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001116 CheckDefExecFailure(['var x: number',
1117 'var y: number',
Bram Moolenaar971caf52020-09-21 22:21:42 +02001118 '[x, y] = [1]'], 'E1093:')
Bram Moolenaar352134b2020-10-17 22:04:08 +02001119 CheckDefExecFailure(['var x: string',
1120 'var y: string',
1121 '[x, y] = ["x"]'], 'E1093:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001122 CheckDefExecFailure(['var x: number',
1123 'var y: number',
1124 'var z: list<number>',
Bram Moolenaar971caf52020-09-21 22:21:42 +02001125 '[x, y; z] = [1]'], 'E1093:')
1126
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001127 CheckDefFailure(['var somevar'], "E1022:")
1128 CheckDefFailure(['var &tabstop = 4'], 'E1052:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001129 CheckDefFailure(['&g:option = 5'], 'E113:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001130 CheckScriptFailure(['vim9script', 'var &tabstop = 4'], 'E1052:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001131
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001132 CheckDefFailure(['var $VAR = 5'], 'E1016: Cannot declare an environment variable:')
1133 CheckScriptFailure(['vim9script', 'var $ENV = "xxx"'], 'E1016:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001134
1135 if has('dnd')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001136 CheckDefFailure(['var @~ = 5'], 'E1066:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001137 else
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001138 CheckDefFailure(['var @~ = 5'], 'E354:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001139 CheckDefFailure(['@~ = 5'], 'E354:')
1140 endif
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001141 CheckDefFailure(['var @a = 5'], 'E1066:')
1142 CheckDefFailure(['var @/ = "x"'], 'E1066:')
1143 CheckScriptFailure(['vim9script', 'var @a = "abc"'], 'E1066:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001144
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001145 CheckDefFailure(['var g:var = 5'], 'E1016: Cannot declare a global variable:')
1146 CheckDefFailure(['var w:var = 5'], 'E1016: Cannot declare a window variable:')
1147 CheckDefFailure(['var b:var = 5'], 'E1016: Cannot declare a buffer variable:')
1148 CheckDefFailure(['var t:var = 5'], 'E1016: Cannot declare a tab variable:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001149
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001150 CheckDefFailure(['var anr = 4', 'anr ..= "text"'], 'E1019:')
1151 CheckDefFailure(['var xnr += 4'], 'E1020:', 1)
1152 CheckScriptFailure(['vim9script', 'var xnr += 4'], 'E1020:')
1153 CheckDefFailure(["var xnr = xnr + 1"], 'E1001:', 1)
1154 CheckScriptFailure(['vim9script', 'var xnr = xnr + 4'], 'E121:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001155
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001156 CheckScriptFailure(['vim9script', 'def Func()', 'var dummy = s:notfound', 'enddef', 'defcompile'], 'E1108:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001157
Bram Moolenaar98309382020-09-27 21:58:45 +02001158 CheckDefFailure(['var name: list<string> = [123]'], 'expected list<string> but got list<number>')
1159 CheckDefFailure(['var name: list<number> = ["xx"]'], 'expected list<number> but got list<string>')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001160
Bram Moolenaare0de1712020-12-02 17:36:54 +01001161 CheckDefFailure(['var name: dict<string> = {key: 123}'], 'expected dict<string> but got dict<number>')
1162 CheckDefFailure(['var name: dict<number> = {key: "xx"}'], 'expected dict<number> but got dict<string>')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001163
Bram Moolenaar98309382020-09-27 21:58:45 +02001164 CheckDefFailure(['var name = feedkeys("0")'], 'E1031:')
1165 CheckDefFailure(['var name: number = feedkeys("0")'], 'expected number but got void')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001166
Bram Moolenaar98309382020-09-27 21:58:45 +02001167 CheckDefFailure(['var name: dict <number>'], 'E1068:')
1168 CheckDefFailure(['var name: dict<number'], 'E1009:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001169
1170 assert_fails('s/^/\=Mess()/n', 'E794:')
Bram Moolenaar98309382020-09-27 21:58:45 +02001171 CheckDefFailure(['var name: dict<number'], 'E1009:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001172
1173 CheckDefFailure(['w:foo: number = 10'],
1174 'E488: Trailing characters: : number = 1')
1175 CheckDefFailure(['t:foo: bool = true'],
1176 'E488: Trailing characters: : bool = true')
1177 CheckDefFailure(['b:foo: string = "x"'],
1178 'E488: Trailing characters: : string = "x"')
1179 CheckDefFailure(['g:foo: number = 123'],
1180 'E488: Trailing characters: : number = 123')
1181enddef
1182
1183def Test_assign_list()
Bram Moolenaare65081d2021-06-27 15:04:05 +02001184 var lines =<< trim END
1185 var l: list<string> = []
1186 l[0] = 'value'
1187 assert_equal('value', l[0])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001188
Bram Moolenaare65081d2021-06-27 15:04:05 +02001189 l[1] = 'asdf'
1190 assert_equal('value', l[0])
1191 assert_equal('asdf', l[1])
1192 assert_equal('asdf', l[-1])
1193 assert_equal('value', l[-2])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001194
Bram Moolenaare65081d2021-06-27 15:04:05 +02001195 var nrl: list<number> = []
1196 for i in range(5)
1197 nrl[i] = i
1198 endfor
1199 assert_equal([0, 1, 2, 3, 4], nrl)
1200
1201 var ul: list<any>
1202 ul[0] = 1
1203 ul[1] = 2
1204 ul[2] = 3
1205 assert_equal([1, 2, 3], ul)
1206 END
1207 CheckDefAndScriptSuccess(lines)
Bram Moolenaar334a8b42020-10-19 16:07:42 +02001208
1209 CheckDefFailure(["var l: list<number> = ['', true]"], 'E1012: Type mismatch; expected list<number> but got list<any>', 1)
1210 CheckDefFailure(["var l: list<list<number>> = [['', true]]"], 'E1012: Type mismatch; expected list<list<number>> but got list<list<any>>', 1)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001211enddef
1212
1213def Test_assign_dict()
Bram Moolenaarc3689572021-01-01 19:40:02 +01001214 var lines =<< trim END
1215 var d: dict<string> = {}
1216 d['key'] = 'value'
1217 assert_equal('value', d['key'])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001218
Bram Moolenaarc3689572021-01-01 19:40:02 +01001219 d[123] = 'qwerty'
1220 assert_equal('qwerty', d[123])
1221 assert_equal('qwerty', d['123'])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001222
Bram Moolenaarc3689572021-01-01 19:40:02 +01001223 var nrd: dict<number> = {}
1224 for i in range(3)
1225 nrd[i] = i
1226 endfor
1227 assert_equal({0: 0, 1: 1, 2: 2}, nrd)
1228
1229 d.somekey = 'someval'
1230 assert_equal({key: 'value', '123': 'qwerty', somekey: 'someval'}, d)
Bram Moolenaar23e2e112021-08-03 21:16:18 +02001231 unlet d.somekey
1232 assert_equal({key: 'value', '123': 'qwerty'}, d)
Bram Moolenaarc3689572021-01-01 19:40:02 +01001233 END
1234 CheckDefAndScriptSuccess(lines)
1235
Bram Moolenaare0de1712020-12-02 17:36:54 +01001236 CheckDefFailure(["var d: dict<number> = {a: '', b: true}"], 'E1012: Type mismatch; expected dict<number> but got dict<any>', 1)
1237 CheckDefFailure(["var d: dict<dict<number>> = {x: {a: '', b: true}}"], 'E1012: Type mismatch; expected dict<dict<number>> but got dict<dict<any>>', 1)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001238enddef
1239
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001240def Test_assign_dict_unknown_type()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001241 var lines =<< trim END
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001242 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001243 var mylist = []
Bram Moolenaare0de1712020-12-02 17:36:54 +01001244 mylist += [{one: 'one'}]
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001245 def Func()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001246 var dd = mylist[0]
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001247 assert_equal('one', dd.one)
1248 enddef
1249 Func()
1250 END
1251 CheckScriptSuccess(lines)
1252
Bram Moolenaarce2c5442020-11-28 21:21:17 +01001253 lines =<< trim END
1254 vim9script
1255 var mylist = [[]]
Bram Moolenaare0de1712020-12-02 17:36:54 +01001256 mylist[0] += [{one: 'one'}]
Bram Moolenaarce2c5442020-11-28 21:21:17 +01001257 def Func()
1258 var dd = mylist[0][0]
1259 assert_equal('one', dd.one)
1260 enddef
1261 Func()
1262 END
1263 CheckScriptSuccess(lines)
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001264enddef
1265
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001266def Test_assign_dict_with_op()
1267 var lines =<< trim END
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001268 var ds: dict<string> = {a: 'x'}
1269 ds['a'] ..= 'y'
1270 ds.a ..= 'z'
1271 assert_equal('xyz', ds.a)
1272
1273 var dn: dict<number> = {a: 9}
1274 dn['a'] += 2
1275 assert_equal(11, dn.a)
1276 dn.a += 2
1277 assert_equal(13, dn.a)
1278
1279 dn['a'] -= 3
1280 assert_equal(10, dn.a)
1281 dn.a -= 2
1282 assert_equal(8, dn.a)
1283
1284 dn['a'] *= 2
1285 assert_equal(16, dn.a)
1286 dn.a *= 2
1287 assert_equal(32, dn.a)
1288
1289 dn['a'] /= 3
1290 assert_equal(10, dn.a)
1291 dn.a /= 2
1292 assert_equal(5, dn.a)
1293
1294 dn['a'] %= 3
1295 assert_equal(2, dn.a)
1296 dn.a %= 6
1297 assert_equal(2, dn.a)
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001298
1299 var dd: dict<dict<list<any>>>
1300 dd.a = {}
1301 dd.a.b = [0]
1302 dd.a.b += [1]
1303 assert_equal({a: {b: [0, 1]}}, dd)
Bram Moolenaar0d351a02021-04-05 21:05:26 +02001304
1305 var dab = {a: ['b']}
1306 dab.a[0] ..= 'c'
1307 assert_equal({a: ['bc']}, dab)
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001308 END
Bram Moolenaare42939a2021-04-05 17:11:17 +02001309 CheckDefAndScriptSuccess(lines)
1310enddef
1311
1312def Test_assign_list_with_op()
1313 var lines =<< trim END
1314 var ls: list<string> = ['x']
1315 ls[0] ..= 'y'
1316 assert_equal('xy', ls[0])
1317
1318 var ln: list<number> = [9]
1319 ln[0] += 2
1320 assert_equal(11, ln[0])
1321
1322 ln[0] -= 3
1323 assert_equal(8, ln[0])
1324
1325 ln[0] *= 2
1326 assert_equal(16, ln[0])
1327
1328 ln[0] /= 3
1329 assert_equal(5, ln[0])
1330
1331 ln[0] %= 3
1332 assert_equal(2, ln[0])
1333 END
1334 CheckDefAndScriptSuccess(lines)
1335enddef
1336
1337def Test_assign_with_op_fails()
1338 var lines =<< trim END
1339 var s = 'abc'
1340 s[1] += 'x'
1341 END
1342 CheckDefAndScriptFailure2(lines, 'E1141:', 'E689:', 2)
1343
1344 lines =<< trim END
1345 var s = 'abc'
1346 s[1] ..= 'x'
1347 END
1348 CheckDefAndScriptFailure2(lines, 'E1141:', 'E689:', 2)
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001349
1350 lines =<< trim END
1351 var dd: dict<dict<list<any>>>
1352 dd.a = {}
1353 dd.a.b += [1]
1354 END
1355 CheckDefExecAndScriptFailure(lines, 'E716:', 3)
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001356enddef
1357
Bram Moolenaar971caf52020-09-21 22:21:42 +02001358def Test_assign_lambda()
1359 # check if assign a lambda to a variable which type is func or any.
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001360 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +02001361 vim9script
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001362 var FuncRef = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001363 assert_equal(123, FuncRef())
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001364 var FuncRef_Func: func = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001365 assert_equal(123, FuncRef_Func())
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001366 var FuncRef_Any: any = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001367 assert_equal(123, FuncRef_Any())
Bram Moolenaar06409502021-02-17 17:00:27 +01001368 var FuncRef_Number: func(): number = () => 321
1369 assert_equal(321, FuncRef_Number())
Bram Moolenaar971caf52020-09-21 22:21:42 +02001370 END
1371 CheckScriptSuccess(lines)
Bram Moolenaar328eac22021-01-07 19:23:08 +01001372
1373 lines =<< trim END
1374 var Ref: func(number)
1375 Ref = (j) => !j
1376 END
Bram Moolenaar06409502021-02-17 17:00:27 +01001377 CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected func(number) but got func(any): bool')
Bram Moolenaarc7dac852021-02-17 18:49:11 +01001378
1379 lines =<< trim END
1380 echo filter([1, 2, 3], (_, v: string) => v + 1)
1381 END
1382 CheckDefAndScriptFailure(lines, 'E1051:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001383enddef
1384
Bram Moolenaarc0e29012020-09-27 14:22:48 +02001385def Test_heredoc()
1386 var lines =<< trim END # comment
1387 text
1388 END
1389 assert_equal(['text'], lines)
1390
1391 CheckDefFailure(['var lines =<< trim END X', 'END'], 'E488:')
1392 CheckDefFailure(['var lines =<< trim END " comment', 'END'], 'E488:')
Bram Moolenaar2914a202020-09-27 18:24:03 +02001393
1394 lines =<< trim [END]
1395 def Func()
1396 var&lines =<< trim END
1397 x
1398 x
Bram Moolenaarb5b94802020-12-13 17:50:20 +01001399 enddef
1400 defcompile
1401 [END]
1402 CheckScriptFailure(lines, 'E1145: Missing heredoc end marker: END')
1403 delfunc! g:Func
1404
1405 lines =<< trim [END]
1406 def Func()
1407 var lines =<< trim END
1408 x
1409 x
Bram Moolenaar2914a202020-09-27 18:24:03 +02001410 x
1411 x
1412 x
1413 x
1414 x
1415 x
1416 enddef
1417 call Func()
1418 [END]
Bram Moolenaarb5b94802020-12-13 17:50:20 +01001419 CheckScriptFailure(lines, 'E1145: Missing heredoc end marker: END')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001420 delfunc! g:Func
Bram Moolenaar81530e32021-07-28 21:25:49 +02001421
1422 lines =<< trim END
1423 var lines: number =<< trim STOP
1424 aaa
1425 bbb
1426 STOP
1427 END
1428 CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected number but got list<string>', 1)
Bram Moolenaarc0e29012020-09-27 14:22:48 +02001429enddef
1430
Bram Moolenaar083966f2020-12-08 11:29:40 +01001431def Test_var_func_call()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001432 var lines =<< trim END
1433 vim9script
1434 func GetValue()
1435 if exists('g:count')
1436 let g:count += 1
1437 else
1438 let g:count = 1
1439 endif
1440 return 'this'
1441 endfunc
1442 var val: string = GetValue()
1443 # env var is always a string
1444 var env = $TERM
1445 END
1446 writefile(lines, 'Xfinished')
1447 source Xfinished
1448 # GetValue() is not called during discovery phase
1449 assert_equal(1, g:count)
1450
1451 unlet g:count
1452 delete('Xfinished')
1453enddef
1454
Bram Moolenaar083966f2020-12-08 11:29:40 +01001455def Test_var_missing_type()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001456 var lines =<< trim END
1457 vim9script
1458 var name = g:unknown
1459 END
1460 CheckScriptFailure(lines, 'E121:')
1461
1462 lines =<< trim END
1463 vim9script
1464 var nr: number = 123
1465 var name = nr
1466 END
1467 CheckScriptSuccess(lines)
1468enddef
1469
Bram Moolenaar083966f2020-12-08 11:29:40 +01001470def Test_var_declaration()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001471 var lines =<< trim END
1472 vim9script
1473 var name: string
1474 g:var_uninit = name
1475 name = 'text'
1476 g:var_test = name
1477 # prefixing s: is optional
1478 s:name = 'prefixed'
1479 g:var_prefixed = s:name
1480
Bram Moolenaar7257af42020-12-08 22:08:51 +01001481 const FOO: number = 123
1482 assert_equal(123, FOO)
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001483 const FOOS = 'foos'
1484 assert_equal('foos', FOOS)
1485 final FLIST = [1]
1486 assert_equal([1], FLIST)
1487 FLIST[0] = 11
1488 assert_equal([11], FLIST)
1489
1490 const g:FOO: number = 321
1491 assert_equal(321, g:FOO)
1492 const g:FOOS = 'gfoos'
1493 assert_equal('gfoos', g:FOOS)
1494 final g:FLIST = [2]
1495 assert_equal([2], g:FLIST)
1496 g:FLIST[0] = 22
1497 assert_equal([22], g:FLIST)
1498
Bram Moolenaard877a572021-04-01 19:42:48 +02001499 def SetGlobalConst()
1500 const g:globConst = 123
1501 enddef
1502 SetGlobalConst()
1503 assert_equal(123, g:globConst)
1504 assert_true(islocked('g:globConst'))
1505
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001506 const w:FOO: number = 46
1507 assert_equal(46, w:FOO)
1508 const w:FOOS = 'wfoos'
1509 assert_equal('wfoos', w:FOOS)
1510 final w:FLIST = [3]
1511 assert_equal([3], w:FLIST)
1512 w:FLIST[0] = 33
1513 assert_equal([33], w:FLIST)
Bram Moolenaar7257af42020-12-08 22:08:51 +01001514
Bram Moolenaar55759b52020-09-30 22:59:42 +02001515 var s:other: number
1516 other = 1234
1517 g:other_var = other
1518
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01001519 var xyz: string # comment
1520
Bram Moolenaar55759b52020-09-30 22:59:42 +02001521 # type is inferred
Bram Moolenaar3862ea32021-01-01 21:05:55 +01001522 var s:dict = {['a']: 222}
Bram Moolenaar55759b52020-09-30 22:59:42 +02001523 def GetDictVal(key: any)
1524 g:dict_val = s:dict[key]
1525 enddef
1526 GetDictVal('a')
Bram Moolenaar08251752021-01-11 21:20:18 +01001527
1528 final adict: dict<string> = {}
1529 def ChangeAdict()
1530 adict.foo = 'foo'
1531 enddef
1532 ChangeAdict()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001533 END
1534 CheckScriptSuccess(lines)
1535 assert_equal('', g:var_uninit)
1536 assert_equal('text', g:var_test)
1537 assert_equal('prefixed', g:var_prefixed)
1538 assert_equal(1234, g:other_var)
1539 assert_equal(222, g:dict_val)
1540
1541 unlet g:var_uninit
1542 unlet g:var_test
1543 unlet g:var_prefixed
1544 unlet g:other_var
Bram Moolenaar6e50ec22021-04-03 19:32:44 +02001545 unlet g:globConst
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001546 unlet g:FOO
1547 unlet g:FOOS
1548 unlet g:FLIST
1549 unlet w:FOO
1550 unlet w:FOOS
1551 unlet w:FLIST
Bram Moolenaar55759b52020-09-30 22:59:42 +02001552enddef
1553
Bram Moolenaar083966f2020-12-08 11:29:40 +01001554def Test_var_declaration_fails()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001555 var lines =<< trim END
1556 vim9script
1557 final var: string
1558 END
1559 CheckScriptFailure(lines, 'E1125:')
1560
1561 lines =<< trim END
1562 vim9script
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001563 const g:constvar = 'string'
1564 g:constvar = 'xx'
1565 END
1566 CheckScriptFailure(lines, 'E741:')
1567 unlet g:constvar
1568
1569 lines =<< trim END
1570 vim9script
Bram Moolenaarf5906aa2021-04-02 14:35:15 +02001571 var name = 'one'
1572 lockvar name
1573 def SetLocked()
1574 name = 'two'
1575 enddef
1576 SetLocked()
1577 END
Bram Moolenaardcf29ac2021-04-02 14:44:02 +02001578 CheckScriptFailure(lines, 'E741: Value is locked: name', 1)
1579
1580 lines =<< trim END
1581 let s:legacy = 'one'
1582 lockvar s:legacy
1583 def SetLocked()
1584 s:legacy = 'two'
1585 enddef
1586 call SetLocked()
1587 END
1588 CheckScriptFailure(lines, 'E741: Value is locked: s:legacy', 1)
Bram Moolenaarf5906aa2021-04-02 14:35:15 +02001589
1590 lines =<< trim END
1591 vim9script
Bram Moolenaard877a572021-04-01 19:42:48 +02001592 def SetGlobalConst()
1593 const g:globConst = 123
1594 enddef
1595 SetGlobalConst()
1596 g:globConst = 234
1597 END
Bram Moolenaar6e50ec22021-04-03 19:32:44 +02001598 CheckScriptFailure(lines, 'E741: Value is locked: g:globConst', 6)
Bram Moolenaard877a572021-04-01 19:42:48 +02001599 unlet g:globConst
1600
1601 lines =<< trim END
1602 vim9script
Bram Moolenaar08251752021-01-11 21:20:18 +01001603 const cdict: dict<string> = {}
1604 def Change()
1605 cdict.foo = 'foo'
1606 enddef
1607 defcompile
1608 END
1609 CheckScriptFailure(lines, 'E46:')
1610
1611 lines =<< trim END
1612 vim9script
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001613 final w:finalvar = [9]
1614 w:finalvar = [8]
1615 END
1616 CheckScriptFailure(lines, 'E1122:')
1617 unlet w:finalvar
1618
1619 lines =<< trim END
1620 vim9script
Bram Moolenaar55759b52020-09-30 22:59:42 +02001621 const var: string
1622 END
1623 CheckScriptFailure(lines, 'E1021:')
1624
1625 lines =<< trim END
1626 vim9script
1627 var 9var: string
1628 END
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01001629 CheckScriptFailure(lines, 'E488:')
Bram Moolenaar7257af42020-12-08 22:08:51 +01001630
1631 CheckDefFailure(['var foo.bar = 2'], 'E1087:')
1632 CheckDefFailure(['var foo[3] = 2'], 'E1087:')
1633 CheckDefFailure(['const foo: number'], 'E1021:')
Bram Moolenaar55759b52020-09-30 22:59:42 +02001634enddef
1635
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001636def Test_script_local_in_legacy()
1637 # OK to define script-local later when prefixed with s:
1638 var lines =<< trim END
1639 def SetLater()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001640 s:legvar = 'two'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001641 enddef
1642 defcompile
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001643 let s:legvar = 'one'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001644 call SetLater()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001645 call assert_equal('two', s:legvar)
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001646 END
1647 CheckScriptSuccess(lines)
1648
1649 # OK to leave out s: prefix when script-local already defined
1650 lines =<< trim END
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001651 let s:legvar = 'one'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001652 def SetNoPrefix()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001653 legvar = 'two'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001654 enddef
1655 call SetNoPrefix()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001656 call assert_equal('two', s:legvar)
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001657 END
1658 CheckScriptSuccess(lines)
1659
1660 # Not OK to leave out s: prefix when script-local defined later
1661 lines =<< trim END
1662 def SetLaterNoPrefix()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001663 legvar = 'two'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001664 enddef
1665 defcompile
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001666 let s:legvar = 'one'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001667 END
1668 CheckScriptFailure(lines, 'E476:', 1)
Bram Moolenaar47bc9c32021-07-17 21:24:56 +02001669
1670 edit! Xfile
1671 lines =<< trim END
1672 var edit: bool
1673 legacy edit
1674 END
1675 CheckDefAndScriptSuccess(lines)
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001676enddef
1677
Bram Moolenaar083966f2020-12-08 11:29:40 +01001678def Test_var_type_check()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001679 var lines =<< trim END
1680 vim9script
1681 var name: string
1682 name = 1234
1683 END
1684 CheckScriptFailure(lines, 'E1012:')
1685
1686 lines =<< trim END
1687 vim9script
1688 var name:string
1689 END
1690 CheckScriptFailure(lines, 'E1069:')
1691
1692 lines =<< trim END
1693 vim9script
1694 var name: asdf
1695 END
1696 CheckScriptFailure(lines, 'E1010:')
1697
1698 lines =<< trim END
1699 vim9script
1700 var s:l: list<number>
1701 s:l = []
1702 END
1703 CheckScriptSuccess(lines)
1704
1705 lines =<< trim END
1706 vim9script
1707 var s:d: dict<number>
1708 s:d = {}
1709 END
1710 CheckScriptSuccess(lines)
Bram Moolenaar378697a2021-07-15 19:23:18 +02001711
1712 lines =<< trim END
1713 vim9script
1714 var d = {a: 1, b: [2]}
1715 def Func(b: bool)
1716 var l: list<number> = b ? d.b : [3]
1717 enddef
1718 defcompile
1719 END
1720 CheckScriptSuccess(lines)
Bram Moolenaar55759b52020-09-30 22:59:42 +02001721enddef
1722
1723let g:dict_number = #{one: 1, two: 2}
1724
Bram Moolenaar083966f2020-12-08 11:29:40 +01001725def Test_var_list_dict_type()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001726 var ll: list<number>
1727 ll = [1, 2, 2, 3, 3, 3]->uniq()
1728 ll->assert_equal([1, 2, 3])
1729
1730 var dd: dict<number>
1731 dd = g:dict_number
1732 dd->assert_equal(g:dict_number)
1733
1734 var lines =<< trim END
1735 var ll: list<number>
1736 ll = [1, 2, 3]->map('"one"')
1737 END
Bram Moolenaar70250fb2021-01-16 19:01:53 +01001738 CheckDefExecFailure(lines, 'E1012: Type mismatch; expected number but got string')
Bram Moolenaar55759b52020-09-30 22:59:42 +02001739enddef
1740
Bram Moolenaar083966f2020-12-08 11:29:40 +01001741def Test_cannot_use_let()
1742 CheckDefAndScriptFailure(['let a = 34'], 'E1126:', 1)
1743enddef
1744
Bram Moolenaar55759b52020-09-30 22:59:42 +02001745def Test_unlet()
1746 g:somevar = 'yes'
1747 assert_true(exists('g:somevar'))
1748 unlet g:somevar
1749 assert_false(exists('g:somevar'))
1750 unlet! g:somevar
1751
1752 # also works for script-local variable in legacy Vim script
1753 s:somevar = 'legacy'
1754 assert_true(exists('s:somevar'))
1755 unlet s:somevar
1756 assert_false(exists('s:somevar'))
1757 unlet! s:somevar
1758
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001759 CheckDefExecFailure([
1760 'var dd = 111',
1761 'unlet dd',
1762 ], 'E1081:', 2)
1763
Bram Moolenaar752fc692021-01-04 21:57:11 +01001764 # dict unlet
1765 var dd = {a: 1, b: 2, c: 3}
1766 unlet dd['a']
1767 unlet dd.c
1768 assert_equal({b: 2}, dd)
1769
1770 # list unlet
1771 var ll = [1, 2, 3, 4]
1772 unlet ll[1]
1773 unlet ll[-1]
1774 assert_equal([1, 3], ll)
1775
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001776 ll = [1, 2, 3, 4]
1777 unlet ll[0 : 1]
1778 assert_equal([3, 4], ll)
1779
1780 ll = [1, 2, 3, 4]
1781 unlet ll[2 : 8]
1782 assert_equal([1, 2], ll)
1783
1784 ll = [1, 2, 3, 4]
1785 unlet ll[-2 : -1]
1786 assert_equal([1, 2], ll)
1787
1788 CheckDefFailure([
1789 'var ll = [1, 2]',
1790 'll[1 : 2] = 7',
1791 ], 'E1165:', 2)
1792 CheckDefFailure([
1793 'var dd = {a: 1}',
1794 'unlet dd["a" : "a"]',
1795 ], 'E1166:', 2)
1796 CheckDefExecFailure([
1797 'unlet g:adict[0 : 1]',
1798 ], 'E1148:', 1)
1799 CheckDefFailure([
1800 'var ll = [1, 2]',
1801 'unlet ll[0:1]',
1802 ], 'E1004:', 2)
1803 CheckDefFailure([
1804 'var ll = [1, 2]',
1805 'unlet ll[0 :1]',
1806 ], 'E1004:', 2)
1807 CheckDefFailure([
1808 'var ll = [1, 2]',
1809 'unlet ll[0: 1]',
1810 ], 'E1004:', 2)
1811
1812 CheckDefFailure([
1813 'var ll = [1, 2]',
1814 'unlet ll["x" : 1]',
1815 ], 'E1012:', 2)
1816 CheckDefFailure([
1817 'var ll = [1, 2]',
1818 'unlet ll[0 : "x"]',
1819 ], 'E1012:', 2)
1820
Bram Moolenaar752fc692021-01-04 21:57:11 +01001821 # list of dict unlet
1822 var dl = [{a: 1, b: 2}, {c: 3}]
1823 unlet dl[0]['b']
1824 assert_equal([{a: 1}, {c: 3}], dl)
1825
1826 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001827 'var ll = test_null_list()',
1828 'unlet ll[0]',
1829 ], 'E684:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001830 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001831 'var ll = [1]',
1832 'unlet ll[2]',
1833 ], 'E684:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001834 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001835 'var ll = [1]',
1836 'unlet ll[g:astring]',
Bram Moolenaarf30a14d2021-01-17 21:51:24 +01001837 ], 'E1012:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001838 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001839 'var dd = test_null_dict()',
1840 'unlet dd["a"]',
1841 ], 'E716:', 2)
1842 CheckDefExecFailure([
1843 'var dd = {a: 1}',
1844 'unlet dd["b"]',
1845 ], 'E716:', 2)
1846 CheckDefExecFailure([
1847 'var dd = {a: 1}',
1848 'unlet dd[g:alist]',
1849 ], 'E1105:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001850
Bram Moolenaar2ef951d2021-01-03 20:55:26 +01001851 # can compile unlet before variable exists
Bram Moolenaar752fc692021-01-04 21:57:11 +01001852 g:someDict = {key: 'val'}
1853 var k = 'key'
1854 unlet g:someDict[k]
1855 assert_equal({}, g:someDict)
1856 unlet g:someDict
1857 assert_false(exists('g:someDict'))
Bram Moolenaar2ef951d2021-01-03 20:55:26 +01001858
Bram Moolenaar55759b52020-09-30 22:59:42 +02001859 CheckScriptFailure([
1860 'vim9script',
1861 'var svar = 123',
1862 'unlet svar',
1863 ], 'E1081:')
1864 CheckScriptFailure([
1865 'vim9script',
1866 'var svar = 123',
1867 'unlet s:svar',
1868 ], 'E1081:')
1869 CheckScriptFailure([
1870 'vim9script',
1871 'var svar = 123',
1872 'def Func()',
1873 ' unlet svar',
1874 'enddef',
1875 'defcompile',
1876 ], 'E1081:')
1877 CheckScriptFailure([
1878 'vim9script',
1879 'var svar = 123',
Bram Moolenaar9aed7292020-12-18 15:38:00 +01001880 'func Func()',
1881 ' unlet s:svar',
1882 'endfunc',
1883 'Func()',
1884 ], 'E1081:')
1885 CheckScriptFailure([
1886 'vim9script',
1887 'var svar = 123',
Bram Moolenaar55759b52020-09-30 22:59:42 +02001888 'def Func()',
1889 ' unlet s:svar',
1890 'enddef',
1891 'defcompile',
1892 ], 'E1081:')
1893
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001894 writefile(['vim9script', 'export var svar = 1234'], 'XunletExport.vim')
1895 var lines =<< trim END
1896 vim9script
1897 import svar from './XunletExport.vim'
1898 def UnletSvar()
1899 unlet svar
1900 enddef
1901 defcompile
1902 END
1903 CheckScriptFailure(lines, 'E1081:', 1)
1904 delete('XunletExport.vim')
1905
Bram Moolenaar55759b52020-09-30 22:59:42 +02001906 $ENVVAR = 'foobar'
1907 assert_equal('foobar', $ENVVAR)
1908 unlet $ENVVAR
1909 assert_equal('', $ENVVAR)
1910enddef
1911
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01001912def Test_expr_error_no_assign()
1913 var lines =<< trim END
1914 vim9script
1915 var x = invalid
1916 echo x
1917 END
1918 CheckScriptFailureList(lines, ['E121:', 'E121:'])
1919
1920 lines =<< trim END
1921 vim9script
1922 var x = 1 / 0
1923 echo x
1924 END
Bram Moolenaar227c58a2021-04-28 20:40:44 +02001925 CheckScriptFailure(lines, 'E1154:')
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01001926
1927 lines =<< trim END
1928 vim9script
1929 var x = 1 % 0
1930 echo x
1931 END
Bram Moolenaar227c58a2021-04-28 20:40:44 +02001932 CheckScriptFailure(lines, 'E1154:')
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01001933
1934 lines =<< trim END
1935 var x: string 'string'
1936 END
1937 CheckDefAndScriptFailure(lines, 'E488:')
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01001938enddef
1939
1940
Bram Moolenaar17126b12021-01-07 22:03:02 +01001941def Test_assign_command_modifier()
1942 var lines =<< trim END
Bram Moolenaar9e0f8832021-01-09 12:09:22 +01001943 var verbose = 0
1944 verbose = 1
1945 assert_equal(1, verbose)
1946 silent verbose = 2
1947 assert_equal(2, verbose)
1948 silent verbose += 2
1949 assert_equal(4, verbose)
1950 silent verbose -= 1
1951 assert_equal(3, verbose)
Bram Moolenaar17126b12021-01-07 22:03:02 +01001952
Bram Moolenaar9e0f8832021-01-09 12:09:22 +01001953 var topleft = {one: 1}
1954 sandbox topleft.one = 3
1955 assert_equal({one: 3}, topleft)
1956 leftabove topleft[' '] = 4
1957 assert_equal({one: 3, ' ': 4}, topleft)
1958
1959 var x: number
1960 var y: number
1961 silent [x, y] = [1, 2]
1962 assert_equal(1, x)
1963 assert_equal(2, y)
Bram Moolenaar17126b12021-01-07 22:03:02 +01001964 END
1965 CheckDefAndScriptSuccess(lines)
1966enddef
1967
Bram Moolenaar74f4a962021-06-17 21:03:07 +02001968def Test_assign_alt_buf_register()
1969 var lines =<< trim END
1970 edit 'file_b1'
1971 var b1 = bufnr()
1972 edit 'file_b2'
1973 var b2 = bufnr()
1974 assert_equal(b1, bufnr('#'))
1975 @# = b2
1976 assert_equal(b2, bufnr('#'))
1977 END
1978 CheckDefAndScriptSuccess(lines)
1979enddef
1980
Bram Moolenaar32154662021-03-28 21:14:06 +02001981def Test_script_funcref_case()
1982 var lines =<< trim END
1983 var Len = (s: string): number => len(s) + 1
1984 assert_equal(5, Len('asdf'))
1985 END
1986 CheckDefAndScriptSuccess(lines)
1987
1988 lines =<< trim END
1989 var len = (s: string): number => len(s) + 1
1990 END
1991 CheckDefAndScriptFailure(lines, 'E704:')
1992
1993 lines =<< trim END
1994 vim9script
1995 var s:Len = (s: string): number => len(s) + 2
1996 assert_equal(6, Len('asdf'))
1997 END
1998 CheckScriptSuccess(lines)
1999
2000 lines =<< trim END
2001 vim9script
2002 var s:len = (s: string): number => len(s) + 1
2003 END
2004 CheckScriptFailure(lines, 'E704:')
2005enddef
2006
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002007def Test_inc_dec()
2008 var lines =<< trim END
2009 var nr = 7
2010 ++nr
Bram Moolenaar730bf302021-04-24 20:43:56 +02002011 assert_equal(8, nr)
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002012 --nr
Bram Moolenaar730bf302021-04-24 20:43:56 +02002013 assert_equal(7, nr)
Bram Moolenaarf3d30842021-06-25 19:29:30 +02002014 ++nr | ++nr
2015 assert_equal(9, nr)
2016 ++nr # comment
2017 assert_equal(10, nr)
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002018
2019 var ll = [1, 2]
2020 --ll[0]
2021 ++ll[1]
Bram Moolenaar730bf302021-04-24 20:43:56 +02002022 assert_equal([0, 3], ll)
2023
2024 g:count = 1
2025 ++g:count
2026 --g:count
2027 assert_equal(1, g:count)
2028 unlet g:count
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002029 END
2030 CheckDefAndScriptSuccess(lines)
Bram Moolenaar22480d12021-06-25 21:31:09 +02002031
2032 lines =<< trim END
2033 var nr = 7
2034 ++ nr
2035 END
2036 CheckDefAndScriptFailure(lines, "E1202: No white space allowed after '++': ++ nr")
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002037enddef
2038
Bram Moolenaar327d3ee2021-07-28 19:34:14 +02002039def Test_abort_after_error()
2040 # should abort after strpart() fails, not give another type error
2041 var lines =<< trim END
2042 vim9script
2043 var x: string
2044 x = strpart(1, 2)
2045 END
2046 writefile(lines, 'Xtestscript')
2047 var expected = 'E1174: String required for argument 1'
2048 assert_fails('so Xtestscript', [expected, expected], 3)
2049
2050 delete('Xtestscript')
2051enddef
2052
2053
Bram Moolenaar55759b52020-09-30 22:59:42 +02002054
Bram Moolenaar971caf52020-09-21 22:21:42 +02002055" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker