blob: fa2654801879a1e9e7eecf530b820d85f0905566 [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)
1231 # unlet d.somekey
1232 # assert_equal({key: 'value', '123': 'qwerty'}, d)
1233 END
1234 CheckDefAndScriptSuccess(lines)
1235
1236 # TODO: move to above once "unlet d.somekey" in :def is implemented
1237 lines =<< trim END
1238 vim9script
1239 var d: dict<string> = {}
1240 d['key'] = 'value'
1241 d.somekey = 'someval'
1242 assert_equal({key: 'value', somekey: 'someval'}, d)
1243 unlet d.somekey
1244 assert_equal({key: 'value'}, d)
1245 END
1246 CheckScriptSuccess(lines)
Bram Moolenaar334a8b42020-10-19 16:07:42 +02001247
Bram Moolenaare0de1712020-12-02 17:36:54 +01001248 CheckDefFailure(["var d: dict<number> = {a: '', b: true}"], 'E1012: Type mismatch; expected dict<number> but got dict<any>', 1)
1249 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 +02001250enddef
1251
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001252def Test_assign_dict_unknown_type()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001253 var lines =<< trim END
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001254 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001255 var mylist = []
Bram Moolenaare0de1712020-12-02 17:36:54 +01001256 mylist += [{one: 'one'}]
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001257 def Func()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001258 var dd = mylist[0]
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001259 assert_equal('one', dd.one)
1260 enddef
1261 Func()
1262 END
1263 CheckScriptSuccess(lines)
1264
Bram Moolenaarce2c5442020-11-28 21:21:17 +01001265 lines =<< trim END
1266 vim9script
1267 var mylist = [[]]
Bram Moolenaare0de1712020-12-02 17:36:54 +01001268 mylist[0] += [{one: 'one'}]
Bram Moolenaarce2c5442020-11-28 21:21:17 +01001269 def Func()
1270 var dd = mylist[0][0]
1271 assert_equal('one', dd.one)
1272 enddef
1273 Func()
1274 END
1275 CheckScriptSuccess(lines)
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001276enddef
1277
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001278def Test_assign_dict_with_op()
1279 var lines =<< trim END
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001280 var ds: dict<string> = {a: 'x'}
1281 ds['a'] ..= 'y'
1282 ds.a ..= 'z'
1283 assert_equal('xyz', ds.a)
1284
1285 var dn: dict<number> = {a: 9}
1286 dn['a'] += 2
1287 assert_equal(11, dn.a)
1288 dn.a += 2
1289 assert_equal(13, dn.a)
1290
1291 dn['a'] -= 3
1292 assert_equal(10, dn.a)
1293 dn.a -= 2
1294 assert_equal(8, dn.a)
1295
1296 dn['a'] *= 2
1297 assert_equal(16, dn.a)
1298 dn.a *= 2
1299 assert_equal(32, dn.a)
1300
1301 dn['a'] /= 3
1302 assert_equal(10, dn.a)
1303 dn.a /= 2
1304 assert_equal(5, dn.a)
1305
1306 dn['a'] %= 3
1307 assert_equal(2, dn.a)
1308 dn.a %= 6
1309 assert_equal(2, dn.a)
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001310
1311 var dd: dict<dict<list<any>>>
1312 dd.a = {}
1313 dd.a.b = [0]
1314 dd.a.b += [1]
1315 assert_equal({a: {b: [0, 1]}}, dd)
Bram Moolenaar0d351a02021-04-05 21:05:26 +02001316
1317 var dab = {a: ['b']}
1318 dab.a[0] ..= 'c'
1319 assert_equal({a: ['bc']}, dab)
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001320 END
Bram Moolenaare42939a2021-04-05 17:11:17 +02001321 CheckDefAndScriptSuccess(lines)
1322enddef
1323
1324def Test_assign_list_with_op()
1325 var lines =<< trim END
1326 var ls: list<string> = ['x']
1327 ls[0] ..= 'y'
1328 assert_equal('xy', ls[0])
1329
1330 var ln: list<number> = [9]
1331 ln[0] += 2
1332 assert_equal(11, ln[0])
1333
1334 ln[0] -= 3
1335 assert_equal(8, ln[0])
1336
1337 ln[0] *= 2
1338 assert_equal(16, ln[0])
1339
1340 ln[0] /= 3
1341 assert_equal(5, ln[0])
1342
1343 ln[0] %= 3
1344 assert_equal(2, ln[0])
1345 END
1346 CheckDefAndScriptSuccess(lines)
1347enddef
1348
1349def Test_assign_with_op_fails()
1350 var lines =<< trim END
1351 var s = 'abc'
1352 s[1] += 'x'
1353 END
1354 CheckDefAndScriptFailure2(lines, 'E1141:', 'E689:', 2)
1355
1356 lines =<< trim END
1357 var s = 'abc'
1358 s[1] ..= 'x'
1359 END
1360 CheckDefAndScriptFailure2(lines, 'E1141:', 'E689:', 2)
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001361
1362 lines =<< trim END
1363 var dd: dict<dict<list<any>>>
1364 dd.a = {}
1365 dd.a.b += [1]
1366 END
1367 CheckDefExecAndScriptFailure(lines, 'E716:', 3)
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001368enddef
1369
Bram Moolenaar971caf52020-09-21 22:21:42 +02001370def Test_assign_lambda()
1371 # check if assign a lambda to a variable which type is func or any.
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001372 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +02001373 vim9script
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001374 var FuncRef = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001375 assert_equal(123, FuncRef())
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001376 var FuncRef_Func: func = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001377 assert_equal(123, FuncRef_Func())
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001378 var FuncRef_Any: any = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001379 assert_equal(123, FuncRef_Any())
Bram Moolenaar06409502021-02-17 17:00:27 +01001380 var FuncRef_Number: func(): number = () => 321
1381 assert_equal(321, FuncRef_Number())
Bram Moolenaar971caf52020-09-21 22:21:42 +02001382 END
1383 CheckScriptSuccess(lines)
Bram Moolenaar328eac22021-01-07 19:23:08 +01001384
1385 lines =<< trim END
1386 var Ref: func(number)
1387 Ref = (j) => !j
1388 END
Bram Moolenaar06409502021-02-17 17:00:27 +01001389 CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected func(number) but got func(any): bool')
Bram Moolenaarc7dac852021-02-17 18:49:11 +01001390
1391 lines =<< trim END
1392 echo filter([1, 2, 3], (_, v: string) => v + 1)
1393 END
1394 CheckDefAndScriptFailure(lines, 'E1051:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001395enddef
1396
Bram Moolenaarc0e29012020-09-27 14:22:48 +02001397def Test_heredoc()
1398 var lines =<< trim END # comment
1399 text
1400 END
1401 assert_equal(['text'], lines)
1402
1403 CheckDefFailure(['var lines =<< trim END X', 'END'], 'E488:')
1404 CheckDefFailure(['var lines =<< trim END " comment', 'END'], 'E488:')
Bram Moolenaar2914a202020-09-27 18:24:03 +02001405
1406 lines =<< trim [END]
1407 def Func()
1408 var&lines =<< trim END
1409 x
1410 x
Bram Moolenaarb5b94802020-12-13 17:50:20 +01001411 enddef
1412 defcompile
1413 [END]
1414 CheckScriptFailure(lines, 'E1145: Missing heredoc end marker: END')
1415 delfunc! g:Func
1416
1417 lines =<< trim [END]
1418 def Func()
1419 var lines =<< trim END
1420 x
1421 x
Bram Moolenaar2914a202020-09-27 18:24:03 +02001422 x
1423 x
1424 x
1425 x
1426 x
1427 x
1428 enddef
1429 call Func()
1430 [END]
Bram Moolenaarb5b94802020-12-13 17:50:20 +01001431 CheckScriptFailure(lines, 'E1145: Missing heredoc end marker: END')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001432 delfunc! g:Func
Bram Moolenaar81530e32021-07-28 21:25:49 +02001433
1434 lines =<< trim END
1435 var lines: number =<< trim STOP
1436 aaa
1437 bbb
1438 STOP
1439 END
1440 CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected number but got list<string>', 1)
Bram Moolenaarc0e29012020-09-27 14:22:48 +02001441enddef
1442
Bram Moolenaar083966f2020-12-08 11:29:40 +01001443def Test_var_func_call()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001444 var lines =<< trim END
1445 vim9script
1446 func GetValue()
1447 if exists('g:count')
1448 let g:count += 1
1449 else
1450 let g:count = 1
1451 endif
1452 return 'this'
1453 endfunc
1454 var val: string = GetValue()
1455 # env var is always a string
1456 var env = $TERM
1457 END
1458 writefile(lines, 'Xfinished')
1459 source Xfinished
1460 # GetValue() is not called during discovery phase
1461 assert_equal(1, g:count)
1462
1463 unlet g:count
1464 delete('Xfinished')
1465enddef
1466
Bram Moolenaar083966f2020-12-08 11:29:40 +01001467def Test_var_missing_type()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001468 var lines =<< trim END
1469 vim9script
1470 var name = g:unknown
1471 END
1472 CheckScriptFailure(lines, 'E121:')
1473
1474 lines =<< trim END
1475 vim9script
1476 var nr: number = 123
1477 var name = nr
1478 END
1479 CheckScriptSuccess(lines)
1480enddef
1481
Bram Moolenaar083966f2020-12-08 11:29:40 +01001482def Test_var_declaration()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001483 var lines =<< trim END
1484 vim9script
1485 var name: string
1486 g:var_uninit = name
1487 name = 'text'
1488 g:var_test = name
1489 # prefixing s: is optional
1490 s:name = 'prefixed'
1491 g:var_prefixed = s:name
1492
Bram Moolenaar7257af42020-12-08 22:08:51 +01001493 const FOO: number = 123
1494 assert_equal(123, FOO)
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001495 const FOOS = 'foos'
1496 assert_equal('foos', FOOS)
1497 final FLIST = [1]
1498 assert_equal([1], FLIST)
1499 FLIST[0] = 11
1500 assert_equal([11], FLIST)
1501
1502 const g:FOO: number = 321
1503 assert_equal(321, g:FOO)
1504 const g:FOOS = 'gfoos'
1505 assert_equal('gfoos', g:FOOS)
1506 final g:FLIST = [2]
1507 assert_equal([2], g:FLIST)
1508 g:FLIST[0] = 22
1509 assert_equal([22], g:FLIST)
1510
Bram Moolenaard877a572021-04-01 19:42:48 +02001511 def SetGlobalConst()
1512 const g:globConst = 123
1513 enddef
1514 SetGlobalConst()
1515 assert_equal(123, g:globConst)
1516 assert_true(islocked('g:globConst'))
1517
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001518 const w:FOO: number = 46
1519 assert_equal(46, w:FOO)
1520 const w:FOOS = 'wfoos'
1521 assert_equal('wfoos', w:FOOS)
1522 final w:FLIST = [3]
1523 assert_equal([3], w:FLIST)
1524 w:FLIST[0] = 33
1525 assert_equal([33], w:FLIST)
Bram Moolenaar7257af42020-12-08 22:08:51 +01001526
Bram Moolenaar55759b52020-09-30 22:59:42 +02001527 var s:other: number
1528 other = 1234
1529 g:other_var = other
1530
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01001531 var xyz: string # comment
1532
Bram Moolenaar55759b52020-09-30 22:59:42 +02001533 # type is inferred
Bram Moolenaar3862ea32021-01-01 21:05:55 +01001534 var s:dict = {['a']: 222}
Bram Moolenaar55759b52020-09-30 22:59:42 +02001535 def GetDictVal(key: any)
1536 g:dict_val = s:dict[key]
1537 enddef
1538 GetDictVal('a')
Bram Moolenaar08251752021-01-11 21:20:18 +01001539
1540 final adict: dict<string> = {}
1541 def ChangeAdict()
1542 adict.foo = 'foo'
1543 enddef
1544 ChangeAdict()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001545 END
1546 CheckScriptSuccess(lines)
1547 assert_equal('', g:var_uninit)
1548 assert_equal('text', g:var_test)
1549 assert_equal('prefixed', g:var_prefixed)
1550 assert_equal(1234, g:other_var)
1551 assert_equal(222, g:dict_val)
1552
1553 unlet g:var_uninit
1554 unlet g:var_test
1555 unlet g:var_prefixed
1556 unlet g:other_var
Bram Moolenaar6e50ec22021-04-03 19:32:44 +02001557 unlet g:globConst
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001558 unlet g:FOO
1559 unlet g:FOOS
1560 unlet g:FLIST
1561 unlet w:FOO
1562 unlet w:FOOS
1563 unlet w:FLIST
Bram Moolenaar55759b52020-09-30 22:59:42 +02001564enddef
1565
Bram Moolenaar083966f2020-12-08 11:29:40 +01001566def Test_var_declaration_fails()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001567 var lines =<< trim END
1568 vim9script
1569 final var: string
1570 END
1571 CheckScriptFailure(lines, 'E1125:')
1572
1573 lines =<< trim END
1574 vim9script
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001575 const g:constvar = 'string'
1576 g:constvar = 'xx'
1577 END
1578 CheckScriptFailure(lines, 'E741:')
1579 unlet g:constvar
1580
1581 lines =<< trim END
1582 vim9script
Bram Moolenaarf5906aa2021-04-02 14:35:15 +02001583 var name = 'one'
1584 lockvar name
1585 def SetLocked()
1586 name = 'two'
1587 enddef
1588 SetLocked()
1589 END
Bram Moolenaardcf29ac2021-04-02 14:44:02 +02001590 CheckScriptFailure(lines, 'E741: Value is locked: name', 1)
1591
1592 lines =<< trim END
1593 let s:legacy = 'one'
1594 lockvar s:legacy
1595 def SetLocked()
1596 s:legacy = 'two'
1597 enddef
1598 call SetLocked()
1599 END
1600 CheckScriptFailure(lines, 'E741: Value is locked: s:legacy', 1)
Bram Moolenaarf5906aa2021-04-02 14:35:15 +02001601
1602 lines =<< trim END
1603 vim9script
Bram Moolenaard877a572021-04-01 19:42:48 +02001604 def SetGlobalConst()
1605 const g:globConst = 123
1606 enddef
1607 SetGlobalConst()
1608 g:globConst = 234
1609 END
Bram Moolenaar6e50ec22021-04-03 19:32:44 +02001610 CheckScriptFailure(lines, 'E741: Value is locked: g:globConst', 6)
Bram Moolenaard877a572021-04-01 19:42:48 +02001611 unlet g:globConst
1612
1613 lines =<< trim END
1614 vim9script
Bram Moolenaar08251752021-01-11 21:20:18 +01001615 const cdict: dict<string> = {}
1616 def Change()
1617 cdict.foo = 'foo'
1618 enddef
1619 defcompile
1620 END
1621 CheckScriptFailure(lines, 'E46:')
1622
1623 lines =<< trim END
1624 vim9script
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001625 final w:finalvar = [9]
1626 w:finalvar = [8]
1627 END
1628 CheckScriptFailure(lines, 'E1122:')
1629 unlet w:finalvar
1630
1631 lines =<< trim END
1632 vim9script
Bram Moolenaar55759b52020-09-30 22:59:42 +02001633 const var: string
1634 END
1635 CheckScriptFailure(lines, 'E1021:')
1636
1637 lines =<< trim END
1638 vim9script
1639 var 9var: string
1640 END
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01001641 CheckScriptFailure(lines, 'E488:')
Bram Moolenaar7257af42020-12-08 22:08:51 +01001642
1643 CheckDefFailure(['var foo.bar = 2'], 'E1087:')
1644 CheckDefFailure(['var foo[3] = 2'], 'E1087:')
1645 CheckDefFailure(['const foo: number'], 'E1021:')
Bram Moolenaar55759b52020-09-30 22:59:42 +02001646enddef
1647
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001648def Test_script_local_in_legacy()
1649 # OK to define script-local later when prefixed with s:
1650 var lines =<< trim END
1651 def SetLater()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001652 s:legvar = 'two'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001653 enddef
1654 defcompile
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001655 let s:legvar = 'one'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001656 call SetLater()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001657 call assert_equal('two', s:legvar)
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001658 END
1659 CheckScriptSuccess(lines)
1660
1661 # OK to leave out s: prefix when script-local already defined
1662 lines =<< trim END
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001663 let s:legvar = 'one'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001664 def SetNoPrefix()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001665 legvar = 'two'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001666 enddef
1667 call SetNoPrefix()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001668 call assert_equal('two', s:legvar)
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001669 END
1670 CheckScriptSuccess(lines)
1671
1672 # Not OK to leave out s: prefix when script-local defined later
1673 lines =<< trim END
1674 def SetLaterNoPrefix()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001675 legvar = 'two'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001676 enddef
1677 defcompile
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001678 let s:legvar = 'one'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001679 END
1680 CheckScriptFailure(lines, 'E476:', 1)
Bram Moolenaar47bc9c32021-07-17 21:24:56 +02001681
1682 edit! Xfile
1683 lines =<< trim END
1684 var edit: bool
1685 legacy edit
1686 END
1687 CheckDefAndScriptSuccess(lines)
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001688enddef
1689
Bram Moolenaar083966f2020-12-08 11:29:40 +01001690def Test_var_type_check()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001691 var lines =<< trim END
1692 vim9script
1693 var name: string
1694 name = 1234
1695 END
1696 CheckScriptFailure(lines, 'E1012:')
1697
1698 lines =<< trim END
1699 vim9script
1700 var name:string
1701 END
1702 CheckScriptFailure(lines, 'E1069:')
1703
1704 lines =<< trim END
1705 vim9script
1706 var name: asdf
1707 END
1708 CheckScriptFailure(lines, 'E1010:')
1709
1710 lines =<< trim END
1711 vim9script
1712 var s:l: list<number>
1713 s:l = []
1714 END
1715 CheckScriptSuccess(lines)
1716
1717 lines =<< trim END
1718 vim9script
1719 var s:d: dict<number>
1720 s:d = {}
1721 END
1722 CheckScriptSuccess(lines)
Bram Moolenaar378697a2021-07-15 19:23:18 +02001723
1724 lines =<< trim END
1725 vim9script
1726 var d = {a: 1, b: [2]}
1727 def Func(b: bool)
1728 var l: list<number> = b ? d.b : [3]
1729 enddef
1730 defcompile
1731 END
1732 CheckScriptSuccess(lines)
Bram Moolenaar55759b52020-09-30 22:59:42 +02001733enddef
1734
1735let g:dict_number = #{one: 1, two: 2}
1736
Bram Moolenaar083966f2020-12-08 11:29:40 +01001737def Test_var_list_dict_type()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001738 var ll: list<number>
1739 ll = [1, 2, 2, 3, 3, 3]->uniq()
1740 ll->assert_equal([1, 2, 3])
1741
1742 var dd: dict<number>
1743 dd = g:dict_number
1744 dd->assert_equal(g:dict_number)
1745
1746 var lines =<< trim END
1747 var ll: list<number>
1748 ll = [1, 2, 3]->map('"one"')
1749 END
Bram Moolenaar70250fb2021-01-16 19:01:53 +01001750 CheckDefExecFailure(lines, 'E1012: Type mismatch; expected number but got string')
Bram Moolenaar55759b52020-09-30 22:59:42 +02001751enddef
1752
Bram Moolenaar083966f2020-12-08 11:29:40 +01001753def Test_cannot_use_let()
1754 CheckDefAndScriptFailure(['let a = 34'], 'E1126:', 1)
1755enddef
1756
Bram Moolenaar55759b52020-09-30 22:59:42 +02001757def Test_unlet()
1758 g:somevar = 'yes'
1759 assert_true(exists('g:somevar'))
1760 unlet g:somevar
1761 assert_false(exists('g:somevar'))
1762 unlet! g:somevar
1763
1764 # also works for script-local variable in legacy Vim script
1765 s:somevar = 'legacy'
1766 assert_true(exists('s:somevar'))
1767 unlet s:somevar
1768 assert_false(exists('s:somevar'))
1769 unlet! s:somevar
1770
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001771 CheckDefExecFailure([
1772 'var dd = 111',
1773 'unlet dd',
1774 ], 'E1081:', 2)
1775
Bram Moolenaar752fc692021-01-04 21:57:11 +01001776 # dict unlet
1777 var dd = {a: 1, b: 2, c: 3}
1778 unlet dd['a']
1779 unlet dd.c
1780 assert_equal({b: 2}, dd)
1781
1782 # list unlet
1783 var ll = [1, 2, 3, 4]
1784 unlet ll[1]
1785 unlet ll[-1]
1786 assert_equal([1, 3], ll)
1787
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001788 ll = [1, 2, 3, 4]
1789 unlet ll[0 : 1]
1790 assert_equal([3, 4], ll)
1791
1792 ll = [1, 2, 3, 4]
1793 unlet ll[2 : 8]
1794 assert_equal([1, 2], ll)
1795
1796 ll = [1, 2, 3, 4]
1797 unlet ll[-2 : -1]
1798 assert_equal([1, 2], ll)
1799
1800 CheckDefFailure([
1801 'var ll = [1, 2]',
1802 'll[1 : 2] = 7',
1803 ], 'E1165:', 2)
1804 CheckDefFailure([
1805 'var dd = {a: 1}',
1806 'unlet dd["a" : "a"]',
1807 ], 'E1166:', 2)
1808 CheckDefExecFailure([
1809 'unlet g:adict[0 : 1]',
1810 ], 'E1148:', 1)
1811 CheckDefFailure([
1812 'var ll = [1, 2]',
1813 'unlet ll[0:1]',
1814 ], 'E1004:', 2)
1815 CheckDefFailure([
1816 'var ll = [1, 2]',
1817 'unlet ll[0 :1]',
1818 ], 'E1004:', 2)
1819 CheckDefFailure([
1820 'var ll = [1, 2]',
1821 'unlet ll[0: 1]',
1822 ], 'E1004:', 2)
1823
1824 CheckDefFailure([
1825 'var ll = [1, 2]',
1826 'unlet ll["x" : 1]',
1827 ], 'E1012:', 2)
1828 CheckDefFailure([
1829 'var ll = [1, 2]',
1830 'unlet ll[0 : "x"]',
1831 ], 'E1012:', 2)
1832
Bram Moolenaar752fc692021-01-04 21:57:11 +01001833 # list of dict unlet
1834 var dl = [{a: 1, b: 2}, {c: 3}]
1835 unlet dl[0]['b']
1836 assert_equal([{a: 1}, {c: 3}], dl)
1837
1838 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001839 'var ll = test_null_list()',
1840 'unlet ll[0]',
1841 ], 'E684:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001842 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001843 'var ll = [1]',
1844 'unlet ll[2]',
1845 ], 'E684:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001846 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001847 'var ll = [1]',
1848 'unlet ll[g:astring]',
Bram Moolenaarf30a14d2021-01-17 21:51:24 +01001849 ], 'E1012:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001850 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001851 'var dd = test_null_dict()',
1852 'unlet dd["a"]',
1853 ], 'E716:', 2)
1854 CheckDefExecFailure([
1855 'var dd = {a: 1}',
1856 'unlet dd["b"]',
1857 ], 'E716:', 2)
1858 CheckDefExecFailure([
1859 'var dd = {a: 1}',
1860 'unlet dd[g:alist]',
1861 ], 'E1105:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001862
Bram Moolenaar2ef951d2021-01-03 20:55:26 +01001863 # can compile unlet before variable exists
Bram Moolenaar752fc692021-01-04 21:57:11 +01001864 g:someDict = {key: 'val'}
1865 var k = 'key'
1866 unlet g:someDict[k]
1867 assert_equal({}, g:someDict)
1868 unlet g:someDict
1869 assert_false(exists('g:someDict'))
Bram Moolenaar2ef951d2021-01-03 20:55:26 +01001870
Bram Moolenaar55759b52020-09-30 22:59:42 +02001871 CheckScriptFailure([
1872 'vim9script',
1873 'var svar = 123',
1874 'unlet svar',
1875 ], 'E1081:')
1876 CheckScriptFailure([
1877 'vim9script',
1878 'var svar = 123',
1879 'unlet s:svar',
1880 ], 'E1081:')
1881 CheckScriptFailure([
1882 'vim9script',
1883 'var svar = 123',
1884 'def Func()',
1885 ' unlet svar',
1886 'enddef',
1887 'defcompile',
1888 ], 'E1081:')
1889 CheckScriptFailure([
1890 'vim9script',
1891 'var svar = 123',
Bram Moolenaar9aed7292020-12-18 15:38:00 +01001892 'func Func()',
1893 ' unlet s:svar',
1894 'endfunc',
1895 'Func()',
1896 ], 'E1081:')
1897 CheckScriptFailure([
1898 'vim9script',
1899 'var svar = 123',
Bram Moolenaar55759b52020-09-30 22:59:42 +02001900 'def Func()',
1901 ' unlet s:svar',
1902 'enddef',
1903 'defcompile',
1904 ], 'E1081:')
1905
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001906 writefile(['vim9script', 'export var svar = 1234'], 'XunletExport.vim')
1907 var lines =<< trim END
1908 vim9script
1909 import svar from './XunletExport.vim'
1910 def UnletSvar()
1911 unlet svar
1912 enddef
1913 defcompile
1914 END
1915 CheckScriptFailure(lines, 'E1081:', 1)
1916 delete('XunletExport.vim')
1917
Bram Moolenaar55759b52020-09-30 22:59:42 +02001918 $ENVVAR = 'foobar'
1919 assert_equal('foobar', $ENVVAR)
1920 unlet $ENVVAR
1921 assert_equal('', $ENVVAR)
1922enddef
1923
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01001924def Test_expr_error_no_assign()
1925 var lines =<< trim END
1926 vim9script
1927 var x = invalid
1928 echo x
1929 END
1930 CheckScriptFailureList(lines, ['E121:', 'E121:'])
1931
1932 lines =<< trim END
1933 vim9script
1934 var x = 1 / 0
1935 echo x
1936 END
Bram Moolenaar227c58a2021-04-28 20:40:44 +02001937 CheckScriptFailure(lines, 'E1154:')
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01001938
1939 lines =<< trim END
1940 vim9script
1941 var x = 1 % 0
1942 echo x
1943 END
Bram Moolenaar227c58a2021-04-28 20:40:44 +02001944 CheckScriptFailure(lines, 'E1154:')
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01001945
1946 lines =<< trim END
1947 var x: string 'string'
1948 END
1949 CheckDefAndScriptFailure(lines, 'E488:')
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01001950enddef
1951
1952
Bram Moolenaar17126b12021-01-07 22:03:02 +01001953def Test_assign_command_modifier()
1954 var lines =<< trim END
Bram Moolenaar9e0f8832021-01-09 12:09:22 +01001955 var verbose = 0
1956 verbose = 1
1957 assert_equal(1, verbose)
1958 silent verbose = 2
1959 assert_equal(2, verbose)
1960 silent verbose += 2
1961 assert_equal(4, verbose)
1962 silent verbose -= 1
1963 assert_equal(3, verbose)
Bram Moolenaar17126b12021-01-07 22:03:02 +01001964
Bram Moolenaar9e0f8832021-01-09 12:09:22 +01001965 var topleft = {one: 1}
1966 sandbox topleft.one = 3
1967 assert_equal({one: 3}, topleft)
1968 leftabove topleft[' '] = 4
1969 assert_equal({one: 3, ' ': 4}, topleft)
1970
1971 var x: number
1972 var y: number
1973 silent [x, y] = [1, 2]
1974 assert_equal(1, x)
1975 assert_equal(2, y)
Bram Moolenaar17126b12021-01-07 22:03:02 +01001976 END
1977 CheckDefAndScriptSuccess(lines)
1978enddef
1979
Bram Moolenaar74f4a962021-06-17 21:03:07 +02001980def Test_assign_alt_buf_register()
1981 var lines =<< trim END
1982 edit 'file_b1'
1983 var b1 = bufnr()
1984 edit 'file_b2'
1985 var b2 = bufnr()
1986 assert_equal(b1, bufnr('#'))
1987 @# = b2
1988 assert_equal(b2, bufnr('#'))
1989 END
1990 CheckDefAndScriptSuccess(lines)
1991enddef
1992
Bram Moolenaar32154662021-03-28 21:14:06 +02001993def Test_script_funcref_case()
1994 var lines =<< trim END
1995 var Len = (s: string): number => len(s) + 1
1996 assert_equal(5, Len('asdf'))
1997 END
1998 CheckDefAndScriptSuccess(lines)
1999
2000 lines =<< trim END
2001 var len = (s: string): number => len(s) + 1
2002 END
2003 CheckDefAndScriptFailure(lines, 'E704:')
2004
2005 lines =<< trim END
2006 vim9script
2007 var s:Len = (s: string): number => len(s) + 2
2008 assert_equal(6, Len('asdf'))
2009 END
2010 CheckScriptSuccess(lines)
2011
2012 lines =<< trim END
2013 vim9script
2014 var s:len = (s: string): number => len(s) + 1
2015 END
2016 CheckScriptFailure(lines, 'E704:')
2017enddef
2018
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002019def Test_inc_dec()
2020 var lines =<< trim END
2021 var nr = 7
2022 ++nr
Bram Moolenaar730bf302021-04-24 20:43:56 +02002023 assert_equal(8, nr)
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002024 --nr
Bram Moolenaar730bf302021-04-24 20:43:56 +02002025 assert_equal(7, nr)
Bram Moolenaarf3d30842021-06-25 19:29:30 +02002026 ++nr | ++nr
2027 assert_equal(9, nr)
2028 ++nr # comment
2029 assert_equal(10, nr)
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002030
2031 var ll = [1, 2]
2032 --ll[0]
2033 ++ll[1]
Bram Moolenaar730bf302021-04-24 20:43:56 +02002034 assert_equal([0, 3], ll)
2035
2036 g:count = 1
2037 ++g:count
2038 --g:count
2039 assert_equal(1, g:count)
2040 unlet g:count
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002041 END
2042 CheckDefAndScriptSuccess(lines)
Bram Moolenaar22480d12021-06-25 21:31:09 +02002043
2044 lines =<< trim END
2045 var nr = 7
2046 ++ nr
2047 END
2048 CheckDefAndScriptFailure(lines, "E1202: No white space allowed after '++': ++ nr")
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002049enddef
2050
Bram Moolenaar327d3ee2021-07-28 19:34:14 +02002051def Test_abort_after_error()
2052 # should abort after strpart() fails, not give another type error
2053 var lines =<< trim END
2054 vim9script
2055 var x: string
2056 x = strpart(1, 2)
2057 END
2058 writefile(lines, 'Xtestscript')
2059 var expected = 'E1174: String required for argument 1'
2060 assert_fails('so Xtestscript', [expected, expected], 3)
2061
2062 delete('Xtestscript')
2063enddef
2064
2065
Bram Moolenaar55759b52020-09-30 22:59:42 +02002066
Bram Moolenaar971caf52020-09-21 22:21:42 +02002067" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker