blob: 55b850a4b6976daf7c3ce2a2105c1bda28186fed [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
226 $SOME_ENV_VAR ..= 'more'
227 assert_equal('somemore', $SOME_ENV_VAR)
228 CheckDefFailure(['$SOME_ENV_VAR += "more"'], 'E1051:')
229 CheckDefFailure(['$SOME_ENV_VAR += 123'], 'E1012:')
230
Bram Moolenaar971caf52020-09-21 22:21:42 +0200231 v:errmsg = 'none'
232 v:errmsg ..= 'again'
233 assert_equal('noneagain', v:errmsg)
234 CheckDefFailure(['v:errmsg += "more"'], 'E1051:')
235 CheckDefFailure(['v:errmsg += 123'], 'E1012:')
236
Bram Moolenaarfc8aa6d2020-10-12 20:31:26 +0200237 var text =<< trim END
238 some text
239 END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200240enddef
241
Bram Moolenaar13e45d12021-06-26 13:28:35 +0200242def Test_assign_register()
243 var lines =<< trim END
244 @c = 'areg'
245 @c ..= 'add'
246 assert_equal('aregadd', @c)
247
248 @@ = 'some text'
249 assert_equal('some text', getreg('"'))
250 END
251 CheckDefAndScriptSuccess(lines)
252
253 CheckDefFailure(['@a += "more"'], 'E1051:')
254 CheckDefFailure(['@a += 123'], 'E1012:')
255enddef
256
Bram Moolenaard0edaf92021-05-28 21:06:08 +0200257def Test_reserved_name()
258 for name in ['true', 'false', 'null']
259 CheckDefExecAndScriptFailure(['var ' .. name .. ' = 0'], 'E1034:')
260 CheckDefExecAndScriptFailure(['var ' .. name .. ': bool'], 'E1034:')
261 endfor
262enddef
263
Bram Moolenaaraa1959b2021-04-15 22:13:39 +0200264def Test_skipped_assignment()
265 var lines =<< trim END
266 for x in []
267 var i: number = 1
268 while false
269 i += 1
270 endwhile
271 endfor
272 END
273 CheckDefAndScriptSuccess(lines)
274enddef
275
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100276def Test_assign_unpack()
277 var lines =<< trim END
278 var v1: number
279 var v2: number
280 [v1, v2] = [1, 2]
281 assert_equal(1, v1)
282 assert_equal(2, v2)
Bram Moolenaarf93bbd02021-04-10 22:35:43 +0200283
284 [v1, _, v2, _] = [1, 99, 2, 77]
285 assert_equal(1, v1)
286 assert_equal(2, v2)
287
288 [v1, v2; _] = [1, 2, 3, 4, 5]
289 assert_equal(1, v1)
290 assert_equal(2, v2)
Bram Moolenaar035bd1c2021-06-21 19:44:11 +0200291
292 var a = 1
293 var b = 3
294 [a, b] += [2, 4]
295 assert_equal(3, a)
296 assert_equal(7, b)
297
298 [a, b] -= [1, 2]
299 assert_equal(2, a)
300 assert_equal(5, b)
301
302 [a, b] *= [3, 2]
303 assert_equal(6, a)
304 assert_equal(10, b)
305
306 [a, b] /= [2, 4]
307 assert_equal(3, a)
308 assert_equal(2, b)
309
310 [a, b] = [17, 15]
311 [a, b] %= [5, 3]
312 assert_equal(2, a)
313 assert_equal(0, b)
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100314 END
315 CheckDefAndScriptSuccess(lines)
316
317 lines =<< trim END
318 var v1: number
319 var v2: number
320 [v1, v2] =
321 END
322 CheckDefFailure(lines, 'E1097:', 5)
323
324 lines =<< trim END
325 var v1: number
326 var v2: number
327 [v1, v2] = xxx
328 END
329 CheckDefFailure(lines, 'E1001:', 3)
330
331 lines =<< trim END
332 var v1: number
333 var v2: number
334 [v1, v2] = popup_clear()
335 END
336 CheckDefFailure(lines, 'E1031:', 3)
337
338 lines =<< trim END
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100339 [v1, v2] = [1, 2]
340 END
341 CheckDefFailure(lines, 'E1089', 1)
342 CheckScriptFailure(['vim9script'] + lines, 'E1089', 2)
343
344 lines =<< trim END
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100345 var v1: number
346 var v2: number
347 [v1, v2] = ''
348 END
349 CheckDefFailure(lines, 'E1012: Type mismatch; expected list<any> but got string', 3)
Bram Moolenaarf785aa12021-02-11 21:19:34 +0100350
351 lines =<< trim END
352 g:values = [false, 0]
353 var x: bool
354 var y: string
355 [x, y] = g:values
356 END
357 CheckDefExecAndScriptFailure(lines, 'E1163: Variable 2: type mismatch, expected string but got number')
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100358enddef
359
Bram Moolenaar7f764942020-12-02 15:11:18 +0100360def Test_assign_linebreak()
361 var nr: number
362 nr =
363 123
364 assert_equal(123, nr)
365
366 var n2: number
367 [nr, n2] =
368 [12, 34]
369 assert_equal(12, nr)
370 assert_equal(34, n2)
371
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100372 CheckDefFailure(["var x = #"], 'E1097:', 3)
Bram Moolenaar77709b12021-04-03 21:01:01 +0200373
374 var lines =<< trim END
375 var x: list<string> = ['a']
376 var y: list<number> = x
377 ->copy()
378 ->copy()
379 END
380 CheckDefFailure(lines, 'E1012:', 2)
Bram Moolenaar7f764942020-12-02 15:11:18 +0100381enddef
382
Bram Moolenaardc234ca2020-11-28 18:52:33 +0100383def Test_assign_index()
384 # list of list
385 var l1: list<number>
386 l1[0] = 123
387 assert_equal([123], l1)
388
389 var l2: list<list<number>>
390 l2[0] = []
391 l2[0][0] = 123
392 assert_equal([[123]], l2)
393
394 var l3: list<list<list<number>>>
395 l3[0] = []
396 l3[0][0] = []
397 l3[0][0][0] = 123
398 assert_equal([[[123]]], l3)
399
400 var lines =<< trim END
401 var l3: list<list<number>>
402 l3[0] = []
403 l3[0][0] = []
404 END
405 CheckDefFailure(lines, 'E1012: Type mismatch; expected number but got list<unknown>', 3)
406
407 # dict of dict
408 var d1: dict<number>
409 d1.one = 1
410 assert_equal({one: 1}, d1)
411
412 var d2: dict<dict<number>>
413 d2.one = {}
414 d2.one.two = 123
415 assert_equal({one: {two: 123}}, d2)
416
417 var d3: dict<dict<dict<number>>>
418 d3.one = {}
419 d3.one.two = {}
420 d3.one.two.three = 123
421 assert_equal({one: {two: {three: 123}}}, d3)
422
423 lines =<< trim END
424 var d3: dict<dict<number>>
425 d3.one = {}
426 d3.one.two = {}
427 END
428 CheckDefFailure(lines, 'E1012: Type mismatch; expected number but got dict<unknown>', 3)
429
Bram Moolenaaracbae182020-12-13 18:44:43 +0100430 lines =<< trim END
431 var lines: list<string>
432 lines['a'] = 'asdf'
433 END
Bram Moolenaarf30a14d2021-01-17 21:51:24 +0100434 CheckDefFailure(lines, 'E1012:', 2)
Bram Moolenaaracbae182020-12-13 18:44:43 +0100435
436 lines =<< trim END
437 var lines: string
438 lines[9] = 'asdf'
439 END
440 CheckDefFailure(lines, 'E1141:', 2)
441
Bram Moolenaardc234ca2020-11-28 18:52:33 +0100442 # list of dict
443 var ld: list<dict<number>>
444 ld[0] = {}
445 ld[0].one = 123
446 assert_equal([{one: 123}], ld)
447
448 lines =<< trim END
449 var ld: list<dict<number>>
450 ld[0] = []
451 END
452 CheckDefFailure(lines, 'E1012: Type mismatch; expected dict<number> but got list<unknown>', 2)
453
454 # dict of list
455 var dl: dict<list<number>>
456 dl.one = []
457 dl.one[0] = 123
458 assert_equal({one: [123]}, dl)
459
460 lines =<< trim END
461 var dl: dict<list<number>>
462 dl.one = {}
463 END
464 CheckDefFailure(lines, 'E1012: Type mismatch; expected list<number> but got dict<unknown>', 2)
465enddef
466
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200467def Test_extend_list()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200468 var lines =<< trim END
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200469 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200470 var l: list<number>
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200471 l += [123]
472 assert_equal([123], l)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100473 END
474 CheckScriptSuccess(lines)
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200475
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100476 lines =<< trim END
477 vim9script
478 var list: list<string>
479 extend(list, ['x'])
480 assert_equal(['x'], list)
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200481 END
482 CheckScriptSuccess(lines)
Bram Moolenaar1a739232020-10-10 15:37:58 +0200483
484 # appending to NULL list from a function
485 lines =<< trim END
486 vim9script
487 var list: list<string>
488 def Func()
489 list += ['a', 'b']
490 enddef
491 Func()
492 assert_equal(['a', 'b'], list)
493 END
494 CheckScriptSuccess(lines)
Bram Moolenaar3beaf9c2020-12-18 17:23:14 +0100495 lines =<< trim END
496 vim9script
497 var list: list<string>
498 def Func()
499 extend(list, ['x', 'b'])
500 enddef
501 Func()
502 assert_equal(['x', 'b'], list)
503 END
504 CheckScriptSuccess(lines)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100505
506 lines =<< trim END
507 vim9script
508 var l: list<string> = test_null_list()
509 extend(l, ['x'])
510 assert_equal(['x'], l)
511 END
512 CheckScriptSuccess(lines)
513
514 lines =<< trim END
515 vim9script
516 extend(test_null_list(), ['x'])
517 END
518 CheckScriptFailure(lines, 'E1134:', 2)
519enddef
520
521def Test_extend_dict()
522 var lines =<< trim END
523 vim9script
524 var d: dict<number>
Bram Moolenaare0de1712020-12-02 17:36:54 +0100525 extend(d, {a: 1})
526 assert_equal({a: 1}, d)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100527
528 var d2: dict<number>
529 d2['one'] = 1
Bram Moolenaare0de1712020-12-02 17:36:54 +0100530 assert_equal({one: 1}, d2)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100531 END
532 CheckScriptSuccess(lines)
533
534 lines =<< trim END
535 vim9script
536 var d: dict<string> = test_null_dict()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100537 extend(d, {a: 'x'})
538 assert_equal({a: 'x'}, d)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100539 END
540 CheckScriptSuccess(lines)
541
542 lines =<< trim END
543 vim9script
Bram Moolenaare0de1712020-12-02 17:36:54 +0100544 extend(test_null_dict(), {a: 'x'})
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100545 END
546 CheckScriptFailure(lines, 'E1133:', 2)
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200547enddef
548
Bram Moolenaar971caf52020-09-21 22:21:42 +0200549def Test_single_letter_vars()
550 # single letter variables
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200551 var a: number = 123
Bram Moolenaar971caf52020-09-21 22:21:42 +0200552 a = 123
553 assert_equal(123, a)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200554 var b: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200555 b = 123
556 assert_equal(123, b)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200557 var g: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200558 g = 123
559 assert_equal(123, g)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200560 var s: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200561 s = 123
562 assert_equal(123, s)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200563 var t: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200564 t = 123
565 assert_equal(123, t)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200566 var v: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200567 v = 123
568 assert_equal(123, v)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200569 var w: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200570 w = 123
571 assert_equal(123, w)
572enddef
573
574def Test_vim9_single_char_vars()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200575 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200576 vim9script
577
578 # single character variable declarations work
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200579 var a: string
580 var b: number
581 var l: list<any>
582 var s: string
583 var t: number
584 var v: number
585 var w: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200586
587 # script-local variables can be used without s: prefix
588 a = 'script-a'
589 b = 111
590 l = [1, 2, 3]
591 s = 'script-s'
592 t = 222
593 v = 333
594 w = 444
595
596 assert_equal('script-a', a)
597 assert_equal(111, b)
598 assert_equal([1, 2, 3], l)
599 assert_equal('script-s', s)
600 assert_equal(222, t)
601 assert_equal(333, v)
602 assert_equal(444, w)
603 END
604 writefile(lines, 'Xsinglechar')
605 source Xsinglechar
606 delete('Xsinglechar')
607enddef
608
609def Test_assignment_list()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200610 var list1: list<bool> = [false, true, false]
611 var list2: list<number> = [1, 2, 3]
612 var list3: list<string> = ['sdf', 'asdf']
613 var list4: list<any> = ['yes', true, 1234]
614 var list5: list<blob> = [0z01, 0z02]
Bram Moolenaar971caf52020-09-21 22:21:42 +0200615
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200616 var listS: list<string> = []
617 var listN: list<number> = []
Bram Moolenaar971caf52020-09-21 22:21:42 +0200618
619 assert_equal([1, 2, 3], list2)
620 list2[-1] = 99
621 assert_equal([1, 2, 99], list2)
622 list2[-2] = 88
623 assert_equal([1, 88, 99], list2)
624 list2[-3] = 77
625 assert_equal([77, 88, 99], list2)
626 list2 += [100]
627 assert_equal([77, 88, 99, 100], list2)
628
629 list3 += ['end']
630 assert_equal(['sdf', 'asdf', 'end'], list3)
631
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200632 CheckDefExecFailure(['var ll = [1, 2, 3]', 'll[-4] = 6'], 'E684:')
633 CheckDefExecFailure(['var [v1, v2] = [1, 2]'], 'E1092:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200634
635 # type becomes list<any>
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200636 var somelist = rand() > 0 ? [1, 2, 3] : ['a', 'b', 'c']
Bram Moolenaar4f5e3972020-12-21 17:30:50 +0100637
638 var lines =<< trim END
639 var d = {dd: test_null_list()}
640 d.dd[0] = 0
641 END
642 CheckDefExecFailure(lines, 'E1147:', 2)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200643enddef
644
Bram Moolenaarf30a14d2021-01-17 21:51:24 +0100645def Test_assignment_list_any_index()
646 var l: list<number> = [1, 2]
647 for [x, y, _]
648 in [[0, 1, ''], [1, 3, '']]
649 l[x] = l[x] + y
650 endfor
651 assert_equal([2, 5], l)
652enddef
653
Bram Moolenaar971caf52020-09-21 22:21:42 +0200654def Test_assignment_list_vim9script()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200655 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200656 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200657 var v1: number
658 var v2: number
659 var v3: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200660 [v1, v2, v3] = [1, 2, 3]
661 assert_equal([1, 2, 3], [v1, v2, v3])
662 END
663 CheckScriptSuccess(lines)
664enddef
665
666def Test_assignment_dict()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100667 var dict1: dict<bool> = {one: false, two: true}
668 var dict2: dict<number> = {one: 1, two: 2}
669 var dict3: dict<string> = {key: 'value'}
670 var dict4: dict<any> = {one: 1, two: '2'}
671 var dict5: dict<blob> = {one: 0z01, two: 0z02}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200672
673 # overwrite
674 dict3['key'] = 'another'
Bram Moolenaare0de1712020-12-02 17:36:54 +0100675 assert_equal(dict3, {key: 'another'})
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100676 dict3.key = 'yet another'
Bram Moolenaare0de1712020-12-02 17:36:54 +0100677 assert_equal(dict3, {key: 'yet another'})
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100678
Bram Moolenaard24602f2020-12-20 15:20:56 +0100679 # member "any" can also be a dict and assigned to
680 var anydict: dict<any> = {nest: {}, nr: 0}
681 anydict.nest['this'] = 123
682 anydict.nest.that = 456
683 assert_equal({nest: {this: 123, that: 456}, nr: 0}, anydict)
684
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100685 var lines =<< trim END
Bram Moolenaard88dc4d2020-12-20 14:49:42 +0100686 var dd = {}
687 dd.two = 2
688 assert_equal({two: 2}, dd)
689 END
Bram Moolenaar4f5e3972020-12-21 17:30:50 +0100690 CheckDefAndScriptSuccess(lines)
691
692 lines =<< trim END
693 var d = {dd: {}}
694 d.dd[0] = 2
695 d.dd['x'] = 3
696 d.dd.y = 4
697 assert_equal({dd: {0: 2, x: 3, y: 4}}, d)
698 END
699 CheckDefAndScriptSuccess(lines)
Bram Moolenaard88dc4d2020-12-20 14:49:42 +0100700
701 lines =<< trim END
Bram Moolenaare0de1712020-12-02 17:36:54 +0100702 var dd = {one: 1}
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100703 dd.one) = 2
704 END
Bram Moolenaard88dc4d2020-12-20 14:49:42 +0100705 CheckDefFailure(lines, 'E488:', 2)
706
707 lines =<< trim END
708 var dd = {one: 1}
709 var dd.one = 2
710 END
711 CheckDefAndScriptFailure(lines, 'E1017:', 2)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200712
713 # empty key can be used
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200714 var dd = {}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200715 dd[""] = 6
Bram Moolenaare0de1712020-12-02 17:36:54 +0100716 assert_equal({['']: 6}, dd)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200717
718 # type becomes dict<any>
Bram Moolenaare0de1712020-12-02 17:36:54 +0100719 var somedict = rand() > 0 ? {a: 1, b: 2} : {a: 'a', b: 'b'}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200720
721 # assignment to script-local dict
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100722 lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200723 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200724 var test: dict<any> = {}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200725 def FillDict(): dict<any>
726 test['a'] = 43
727 return test
728 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +0100729 assert_equal({a: 43}, FillDict())
Bram Moolenaar971caf52020-09-21 22:21:42 +0200730 END
731 CheckScriptSuccess(lines)
732
733 lines =<< trim END
734 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200735 var test: dict<any>
Bram Moolenaar971caf52020-09-21 22:21:42 +0200736 def FillDict(): dict<any>
737 test['a'] = 43
738 return test
739 enddef
740 FillDict()
Bram Moolenaar3beaf9c2020-12-18 17:23:14 +0100741 assert_equal({a: 43}, test)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200742 END
Bram Moolenaar3beaf9c2020-12-18 17:23:14 +0100743 CheckScriptSuccess(lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200744
745 # assignment to global dict
746 lines =<< trim END
747 vim9script
748 g:test = {}
749 def FillDict(): dict<any>
750 g:test['a'] = 43
751 return g:test
752 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +0100753 assert_equal({a: 43}, FillDict())
Bram Moolenaar971caf52020-09-21 22:21:42 +0200754 END
755 CheckScriptSuccess(lines)
756
757 # assignment to buffer dict
758 lines =<< trim END
759 vim9script
760 b:test = {}
761 def FillDict(): dict<any>
762 b:test['a'] = 43
763 return b:test
764 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +0100765 assert_equal({a: 43}, FillDict())
Bram Moolenaar971caf52020-09-21 22:21:42 +0200766 END
767 CheckScriptSuccess(lines)
Bram Moolenaar4f5e3972020-12-21 17:30:50 +0100768
769 lines =<< trim END
770 var d = {dd: test_null_dict()}
771 d.dd[0] = 0
772 END
773 CheckDefExecFailure(lines, 'E1103:', 2)
774
775 lines =<< trim END
776 var d = {dd: 'string'}
777 d.dd[0] = 0
778 END
779 CheckDefExecFailure(lines, 'E1148:', 2)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200780enddef
781
782def Test_assignment_local()
783 # Test in a separated file in order not to the current buffer/window/tab is
784 # changed.
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200785 var script_lines: list<string> =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200786 let b:existing = 'yes'
787 let w:existing = 'yes'
788 let t:existing = 'yes'
789
790 def Test_assignment_local_internal()
791 b:newvar = 'new'
792 assert_equal('new', b:newvar)
793 assert_equal('yes', b:existing)
794 b:existing = 'no'
795 assert_equal('no', b:existing)
796 b:existing ..= 'NO'
797 assert_equal('noNO', b:existing)
798
799 w:newvar = 'new'
800 assert_equal('new', w:newvar)
801 assert_equal('yes', w:existing)
802 w:existing = 'no'
803 assert_equal('no', w:existing)
804 w:existing ..= 'NO'
805 assert_equal('noNO', w:existing)
806
807 t:newvar = 'new'
808 assert_equal('new', t:newvar)
809 assert_equal('yes', t:existing)
810 t:existing = 'no'
811 assert_equal('no', t:existing)
812 t:existing ..= 'NO'
813 assert_equal('noNO', t:existing)
814 enddef
815 call Test_assignment_local_internal()
816 END
817 CheckScriptSuccess(script_lines)
818enddef
819
820def Test_assignment_default()
Bram Moolenaar971caf52020-09-21 22:21:42 +0200821 # Test default values.
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200822 var thebool: bool
Bram Moolenaar971caf52020-09-21 22:21:42 +0200823 assert_equal(v:false, thebool)
824
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200825 var thenumber: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200826 assert_equal(0, thenumber)
827
828 if has('float')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200829 var thefloat: float
Bram Moolenaar971caf52020-09-21 22:21:42 +0200830 assert_equal(0.0, thefloat)
831 endif
832
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200833 var thestring: string
Bram Moolenaar971caf52020-09-21 22:21:42 +0200834 assert_equal('', thestring)
835
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200836 var theblob: blob
Bram Moolenaar971caf52020-09-21 22:21:42 +0200837 assert_equal(0z, theblob)
838
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200839 var Thefunc: func
Bram Moolenaar971caf52020-09-21 22:21:42 +0200840 assert_equal(test_null_function(), Thefunc)
841
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200842 var thelist: list<any>
Bram Moolenaar971caf52020-09-21 22:21:42 +0200843 assert_equal([], thelist)
844
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200845 var thedict: dict<any>
Bram Moolenaar971caf52020-09-21 22:21:42 +0200846 assert_equal({}, thedict)
847
848 if has('channel')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200849 var thejob: job
Bram Moolenaar971caf52020-09-21 22:21:42 +0200850 assert_equal(test_null_job(), thejob)
851
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200852 var thechannel: channel
Bram Moolenaar971caf52020-09-21 22:21:42 +0200853 assert_equal(test_null_channel(), thechannel)
854
855 if has('unix') && executable('cat')
856 # check with non-null job and channel, types must match
Bram Moolenaare0de1712020-12-02 17:36:54 +0100857 thejob = job_start("cat ", {})
Bram Moolenaar971caf52020-09-21 22:21:42 +0200858 thechannel = job_getchannel(thejob)
859 job_stop(thejob, 'kill')
860 endif
861 endif
862
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200863 var nr = 1234 | nr = 5678
Bram Moolenaar971caf52020-09-21 22:21:42 +0200864 assert_equal(5678, nr)
865enddef
866
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100867let scriptvar = 'init'
868
Bram Moolenaar971caf52020-09-21 22:21:42 +0200869def Test_assignment_var_list()
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100870 var lines =<< trim END
871 var v1: string
872 var v2: string
873 var vrem: list<string>
874 [v1] = ['aaa']
875 assert_equal('aaa', v1)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200876
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100877 [v1, v2] = ['one', 'two']
878 assert_equal('one', v1)
879 assert_equal('two', v2)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200880
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100881 [v1, v2; vrem] = ['one', 'two']
882 assert_equal('one', v1)
883 assert_equal('two', v2)
884 assert_equal([], vrem)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200885
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100886 [v1, v2; vrem] = ['one', 'two', 'three']
887 assert_equal('one', v1)
888 assert_equal('two', v2)
889 assert_equal(['three'], vrem)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200890
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100891 [&ts, &sw] = [3, 4]
892 assert_equal(3, &ts)
893 assert_equal(4, &sw)
894 set ts=8 sw=4
Bram Moolenaar004d9b02020-11-30 21:40:03 +0100895
896 [@a, @z] = ['aa', 'zz']
897 assert_equal('aa', @a)
898 assert_equal('zz', @z)
899
900 [$SOME_VAR, $OTHER_VAR] = ['some', 'other']
901 assert_equal('some', $SOME_VAR)
902 assert_equal('other', $OTHER_VAR)
903
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100904 [g:globalvar, b:bufvar, w:winvar, t:tabvar, v:errmsg] =
905 ['global', 'buf', 'win', 'tab', 'error']
Bram Moolenaar004d9b02020-11-30 21:40:03 +0100906 assert_equal('global', g:globalvar)
Bram Moolenaar004d9b02020-11-30 21:40:03 +0100907 assert_equal('buf', b:bufvar)
908 assert_equal('win', w:winvar)
909 assert_equal('tab', t:tabvar)
910 assert_equal('error', v:errmsg)
911 unlet g:globalvar
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100912 END
913 CheckDefAndScriptSuccess(lines)
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100914
915 [g:globalvar, s:scriptvar, b:bufvar] = ['global', 'script', 'buf']
916 assert_equal('global', g:globalvar)
917 assert_equal('script', s:scriptvar)
918 assert_equal('buf', b:bufvar)
919
920 lines =<< trim END
921 vim9script
922 var s:scriptvar = 'init'
923 [g:globalvar, s:scriptvar, w:winvar] = ['global', 'script', 'win']
924 assert_equal('global', g:globalvar)
925 assert_equal('script', s:scriptvar)
926 assert_equal('win', w:winvar)
927 END
928 CheckScriptSuccess(lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200929enddef
930
931def Test_assignment_vim9script()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200932 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200933 vim9script
934 def Func(): list<number>
935 return [1, 2]
936 enddef
Bram Moolenaar98309382020-09-27 21:58:45 +0200937 var name1: number
938 var name2: number
939 [name1, name2] =
Bram Moolenaar971caf52020-09-21 22:21:42 +0200940 Func()
Bram Moolenaar98309382020-09-27 21:58:45 +0200941 assert_equal(1, name1)
942 assert_equal(2, name2)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200943 var ll =
Bram Moolenaar971caf52020-09-21 22:21:42 +0200944 Func()
945 assert_equal([1, 2], ll)
946
947 @/ = 'text'
948 assert_equal('text', @/)
949 @0 = 'zero'
950 assert_equal('zero', @0)
951 @1 = 'one'
952 assert_equal('one', @1)
953 @9 = 'nine'
954 assert_equal('nine', @9)
955 @- = 'minus'
956 assert_equal('minus', @-)
957 if has('clipboard_working')
958 @* = 'star'
959 assert_equal('star', @*)
960 @+ = 'plus'
961 assert_equal('plus', @+)
962 endif
963
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200964 var a: number = 123
Bram Moolenaar971caf52020-09-21 22:21:42 +0200965 assert_equal(123, a)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200966 var s: string = 'yes'
Bram Moolenaar971caf52020-09-21 22:21:42 +0200967 assert_equal('yes', s)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200968 var b: number = 42
Bram Moolenaar971caf52020-09-21 22:21:42 +0200969 assert_equal(42, b)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200970 var w: number = 43
Bram Moolenaar971caf52020-09-21 22:21:42 +0200971 assert_equal(43, w)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200972 var t: number = 44
Bram Moolenaar971caf52020-09-21 22:21:42 +0200973 assert_equal(44, t)
Bram Moolenaar7a66a172020-10-16 19:56:12 +0200974
975 var to_var = 0
976 to_var = 3
977 assert_equal(3, to_var)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200978 END
979 CheckScriptSuccess(lines)
Bram Moolenaar06f9c692020-09-27 21:27:40 +0200980
981 lines =<< trim END
982 vim9script
983 var n: number
984 def Func()
985 n = 'string'
986 enddef
987 defcompile
988 END
989 CheckScriptFailure(lines, 'E1012: Type mismatch; expected number but got string')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200990enddef
991
992def Mess(): string
993 v:foldstart = 123
994 return 'xxx'
995enddef
996
997def Test_assignment_failure()
Bram Moolenaar98309382020-09-27 21:58:45 +0200998 CheckDefFailure(['var name=234'], 'E1004:')
999 CheckDefFailure(['var name =234'], 'E1004:')
1000 CheckDefFailure(['var name= 234'], 'E1004:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001001
Bram Moolenaar98309382020-09-27 21:58:45 +02001002 CheckScriptFailure(['vim9script', 'var name=234'], 'E1004:')
1003 CheckScriptFailure(['vim9script', 'var name=234'], "before and after '='")
1004 CheckScriptFailure(['vim9script', 'var name =234'], 'E1004:')
1005 CheckScriptFailure(['vim9script', 'var name= 234'], 'E1004:')
1006 CheckScriptFailure(['vim9script', 'var name = 234', 'name+=234'], 'E1004:')
1007 CheckScriptFailure(['vim9script', 'var name = 234', 'name+=234'], "before and after '+='")
1008 CheckScriptFailure(['vim9script', 'var name = "x"', 'name..="y"'], 'E1004:')
1009 CheckScriptFailure(['vim9script', 'var name = "x"', 'name..="y"'], "before and after '..='")
Bram Moolenaar971caf52020-09-21 22:21:42 +02001010
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001011 CheckDefFailure(['var true = 1'], 'E1034:')
1012 CheckDefFailure(['var false = 1'], 'E1034:')
Bram Moolenaar74235772021-06-12 14:53:05 +02001013 CheckDefFailure(['var null = 1'], 'E1034:')
1014 CheckDefFailure(['var this = 1'], 'E1034:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001015
1016 CheckDefFailure(['[a; b; c] = g:list'], 'E452:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001017 CheckDefExecFailure(['var a: number',
Bram Moolenaar971caf52020-09-21 22:21:42 +02001018 '[a] = test_null_list()'], 'E1093:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001019 CheckDefExecFailure(['var a: number',
Bram Moolenaar971caf52020-09-21 22:21:42 +02001020 '[a] = []'], 'E1093:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001021 CheckDefExecFailure(['var x: number',
1022 'var y: number',
Bram Moolenaar971caf52020-09-21 22:21:42 +02001023 '[x, y] = [1]'], 'E1093:')
Bram Moolenaar352134b2020-10-17 22:04:08 +02001024 CheckDefExecFailure(['var x: string',
1025 'var y: string',
1026 '[x, y] = ["x"]'], 'E1093:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001027 CheckDefExecFailure(['var x: number',
1028 'var y: number',
1029 'var z: list<number>',
Bram Moolenaar971caf52020-09-21 22:21:42 +02001030 '[x, y; z] = [1]'], 'E1093:')
1031
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001032 CheckDefFailure(['var somevar'], "E1022:")
1033 CheckDefFailure(['var &tabstop = 4'], 'E1052:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001034 CheckDefFailure(['&g:option = 5'], 'E113:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001035 CheckScriptFailure(['vim9script', 'var &tabstop = 4'], 'E1052:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001036
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001037 CheckDefFailure(['var $VAR = 5'], 'E1016: Cannot declare an environment variable:')
1038 CheckScriptFailure(['vim9script', 'var $ENV = "xxx"'], 'E1016:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001039
1040 if has('dnd')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001041 CheckDefFailure(['var @~ = 5'], 'E1066:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001042 else
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001043 CheckDefFailure(['var @~ = 5'], 'E354:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001044 CheckDefFailure(['@~ = 5'], 'E354:')
1045 endif
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001046 CheckDefFailure(['var @a = 5'], 'E1066:')
1047 CheckDefFailure(['var @/ = "x"'], 'E1066:')
1048 CheckScriptFailure(['vim9script', 'var @a = "abc"'], 'E1066:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001049
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001050 CheckDefFailure(['var g:var = 5'], 'E1016: Cannot declare a global variable:')
1051 CheckDefFailure(['var w:var = 5'], 'E1016: Cannot declare a window variable:')
1052 CheckDefFailure(['var b:var = 5'], 'E1016: Cannot declare a buffer variable:')
1053 CheckDefFailure(['var t:var = 5'], 'E1016: Cannot declare a tab variable:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001054
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001055 CheckDefFailure(['var anr = 4', 'anr ..= "text"'], 'E1019:')
1056 CheckDefFailure(['var xnr += 4'], 'E1020:', 1)
1057 CheckScriptFailure(['vim9script', 'var xnr += 4'], 'E1020:')
1058 CheckDefFailure(["var xnr = xnr + 1"], 'E1001:', 1)
1059 CheckScriptFailure(['vim9script', 'var xnr = xnr + 4'], 'E121:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001060
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001061 CheckScriptFailure(['vim9script', 'def Func()', 'var dummy = s:notfound', 'enddef', 'defcompile'], 'E1108:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001062
Bram Moolenaar98309382020-09-27 21:58:45 +02001063 CheckDefFailure(['var name: list<string> = [123]'], 'expected list<string> but got list<number>')
1064 CheckDefFailure(['var name: list<number> = ["xx"]'], 'expected list<number> but got list<string>')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001065
Bram Moolenaare0de1712020-12-02 17:36:54 +01001066 CheckDefFailure(['var name: dict<string> = {key: 123}'], 'expected dict<string> but got dict<number>')
1067 CheckDefFailure(['var name: dict<number> = {key: "xx"}'], 'expected dict<number> but got dict<string>')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001068
Bram Moolenaar98309382020-09-27 21:58:45 +02001069 CheckDefFailure(['var name = feedkeys("0")'], 'E1031:')
1070 CheckDefFailure(['var name: number = feedkeys("0")'], 'expected number but got void')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001071
Bram Moolenaar98309382020-09-27 21:58:45 +02001072 CheckDefFailure(['var name: dict <number>'], 'E1068:')
1073 CheckDefFailure(['var name: dict<number'], 'E1009:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001074
1075 assert_fails('s/^/\=Mess()/n', 'E794:')
Bram Moolenaar98309382020-09-27 21:58:45 +02001076 CheckDefFailure(['var name: dict<number'], 'E1009:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001077
1078 CheckDefFailure(['w:foo: number = 10'],
1079 'E488: Trailing characters: : number = 1')
1080 CheckDefFailure(['t:foo: bool = true'],
1081 'E488: Trailing characters: : bool = true')
1082 CheckDefFailure(['b:foo: string = "x"'],
1083 'E488: Trailing characters: : string = "x"')
1084 CheckDefFailure(['g:foo: number = 123'],
1085 'E488: Trailing characters: : number = 123')
1086enddef
1087
1088def Test_assign_list()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001089 var l: list<string> = []
Bram Moolenaar971caf52020-09-21 22:21:42 +02001090 l[0] = 'value'
1091 assert_equal('value', l[0])
1092
1093 l[1] = 'asdf'
1094 assert_equal('value', l[0])
1095 assert_equal('asdf', l[1])
1096 assert_equal('asdf', l[-1])
1097 assert_equal('value', l[-2])
1098
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001099 var nrl: list<number> = []
Bram Moolenaar971caf52020-09-21 22:21:42 +02001100 for i in range(5)
1101 nrl[i] = i
1102 endfor
1103 assert_equal([0, 1, 2, 3, 4], nrl)
Bram Moolenaar334a8b42020-10-19 16:07:42 +02001104
1105 CheckDefFailure(["var l: list<number> = ['', true]"], 'E1012: Type mismatch; expected list<number> but got list<any>', 1)
1106 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 +02001107enddef
1108
1109def Test_assign_dict()
Bram Moolenaarc3689572021-01-01 19:40:02 +01001110 var lines =<< trim END
1111 var d: dict<string> = {}
1112 d['key'] = 'value'
1113 assert_equal('value', d['key'])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001114
Bram Moolenaarc3689572021-01-01 19:40:02 +01001115 d[123] = 'qwerty'
1116 assert_equal('qwerty', d[123])
1117 assert_equal('qwerty', d['123'])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001118
Bram Moolenaarc3689572021-01-01 19:40:02 +01001119 var nrd: dict<number> = {}
1120 for i in range(3)
1121 nrd[i] = i
1122 endfor
1123 assert_equal({0: 0, 1: 1, 2: 2}, nrd)
1124
1125 d.somekey = 'someval'
1126 assert_equal({key: 'value', '123': 'qwerty', somekey: 'someval'}, d)
1127 # unlet d.somekey
1128 # assert_equal({key: 'value', '123': 'qwerty'}, d)
1129 END
1130 CheckDefAndScriptSuccess(lines)
1131
1132 # TODO: move to above once "unlet d.somekey" in :def is implemented
1133 lines =<< trim END
1134 vim9script
1135 var d: dict<string> = {}
1136 d['key'] = 'value'
1137 d.somekey = 'someval'
1138 assert_equal({key: 'value', somekey: 'someval'}, d)
1139 unlet d.somekey
1140 assert_equal({key: 'value'}, d)
1141 END
1142 CheckScriptSuccess(lines)
Bram Moolenaar334a8b42020-10-19 16:07:42 +02001143
Bram Moolenaare0de1712020-12-02 17:36:54 +01001144 CheckDefFailure(["var d: dict<number> = {a: '', b: true}"], 'E1012: Type mismatch; expected dict<number> but got dict<any>', 1)
1145 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 +02001146enddef
1147
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001148def Test_assign_dict_unknown_type()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001149 var lines =<< trim END
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001150 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001151 var mylist = []
Bram Moolenaare0de1712020-12-02 17:36:54 +01001152 mylist += [{one: 'one'}]
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001153 def Func()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001154 var dd = mylist[0]
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001155 assert_equal('one', dd.one)
1156 enddef
1157 Func()
1158 END
1159 CheckScriptSuccess(lines)
1160
Bram Moolenaarce2c5442020-11-28 21:21:17 +01001161 lines =<< trim END
1162 vim9script
1163 var mylist = [[]]
Bram Moolenaare0de1712020-12-02 17:36:54 +01001164 mylist[0] += [{one: 'one'}]
Bram Moolenaarce2c5442020-11-28 21:21:17 +01001165 def Func()
1166 var dd = mylist[0][0]
1167 assert_equal('one', dd.one)
1168 enddef
1169 Func()
1170 END
1171 CheckScriptSuccess(lines)
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001172enddef
1173
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001174def Test_assign_dict_with_op()
1175 var lines =<< trim END
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001176 var ds: dict<string> = {a: 'x'}
1177 ds['a'] ..= 'y'
1178 ds.a ..= 'z'
1179 assert_equal('xyz', ds.a)
1180
1181 var dn: dict<number> = {a: 9}
1182 dn['a'] += 2
1183 assert_equal(11, dn.a)
1184 dn.a += 2
1185 assert_equal(13, dn.a)
1186
1187 dn['a'] -= 3
1188 assert_equal(10, dn.a)
1189 dn.a -= 2
1190 assert_equal(8, dn.a)
1191
1192 dn['a'] *= 2
1193 assert_equal(16, dn.a)
1194 dn.a *= 2
1195 assert_equal(32, dn.a)
1196
1197 dn['a'] /= 3
1198 assert_equal(10, dn.a)
1199 dn.a /= 2
1200 assert_equal(5, dn.a)
1201
1202 dn['a'] %= 3
1203 assert_equal(2, dn.a)
1204 dn.a %= 6
1205 assert_equal(2, dn.a)
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001206
1207 var dd: dict<dict<list<any>>>
1208 dd.a = {}
1209 dd.a.b = [0]
1210 dd.a.b += [1]
1211 assert_equal({a: {b: [0, 1]}}, dd)
Bram Moolenaar0d351a02021-04-05 21:05:26 +02001212
1213 var dab = {a: ['b']}
1214 dab.a[0] ..= 'c'
1215 assert_equal({a: ['bc']}, dab)
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001216 END
Bram Moolenaare42939a2021-04-05 17:11:17 +02001217 CheckDefAndScriptSuccess(lines)
1218enddef
1219
1220def Test_assign_list_with_op()
1221 var lines =<< trim END
1222 var ls: list<string> = ['x']
1223 ls[0] ..= 'y'
1224 assert_equal('xy', ls[0])
1225
1226 var ln: list<number> = [9]
1227 ln[0] += 2
1228 assert_equal(11, ln[0])
1229
1230 ln[0] -= 3
1231 assert_equal(8, ln[0])
1232
1233 ln[0] *= 2
1234 assert_equal(16, ln[0])
1235
1236 ln[0] /= 3
1237 assert_equal(5, ln[0])
1238
1239 ln[0] %= 3
1240 assert_equal(2, ln[0])
1241 END
1242 CheckDefAndScriptSuccess(lines)
1243enddef
1244
1245def Test_assign_with_op_fails()
1246 var lines =<< trim END
1247 var s = 'abc'
1248 s[1] += 'x'
1249 END
1250 CheckDefAndScriptFailure2(lines, 'E1141:', 'E689:', 2)
1251
1252 lines =<< trim END
1253 var s = 'abc'
1254 s[1] ..= 'x'
1255 END
1256 CheckDefAndScriptFailure2(lines, 'E1141:', 'E689:', 2)
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001257
1258 lines =<< trim END
1259 var dd: dict<dict<list<any>>>
1260 dd.a = {}
1261 dd.a.b += [1]
1262 END
1263 CheckDefExecAndScriptFailure(lines, 'E716:', 3)
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001264enddef
1265
Bram Moolenaar971caf52020-09-21 22:21:42 +02001266def Test_assign_lambda()
1267 # check if assign a lambda to a variable which type is func or any.
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001268 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +02001269 vim9script
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001270 var FuncRef = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001271 assert_equal(123, FuncRef())
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001272 var FuncRef_Func: func = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001273 assert_equal(123, FuncRef_Func())
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001274 var FuncRef_Any: any = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001275 assert_equal(123, FuncRef_Any())
Bram Moolenaar06409502021-02-17 17:00:27 +01001276 var FuncRef_Number: func(): number = () => 321
1277 assert_equal(321, FuncRef_Number())
Bram Moolenaar971caf52020-09-21 22:21:42 +02001278 END
1279 CheckScriptSuccess(lines)
Bram Moolenaar328eac22021-01-07 19:23:08 +01001280
1281 lines =<< trim END
1282 var Ref: func(number)
1283 Ref = (j) => !j
1284 END
Bram Moolenaar06409502021-02-17 17:00:27 +01001285 CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected func(number) but got func(any): bool')
Bram Moolenaarc7dac852021-02-17 18:49:11 +01001286
1287 lines =<< trim END
1288 echo filter([1, 2, 3], (_, v: string) => v + 1)
1289 END
1290 CheckDefAndScriptFailure(lines, 'E1051:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001291enddef
1292
Bram Moolenaarc0e29012020-09-27 14:22:48 +02001293def Test_heredoc()
1294 var lines =<< trim END # comment
1295 text
1296 END
1297 assert_equal(['text'], lines)
1298
1299 CheckDefFailure(['var lines =<< trim END X', 'END'], 'E488:')
1300 CheckDefFailure(['var lines =<< trim END " comment', 'END'], 'E488:')
Bram Moolenaar2914a202020-09-27 18:24:03 +02001301
1302 lines =<< trim [END]
1303 def Func()
1304 var&lines =<< trim END
1305 x
1306 x
Bram Moolenaarb5b94802020-12-13 17:50:20 +01001307 enddef
1308 defcompile
1309 [END]
1310 CheckScriptFailure(lines, 'E1145: Missing heredoc end marker: END')
1311 delfunc! g:Func
1312
1313 lines =<< trim [END]
1314 def Func()
1315 var lines =<< trim END
1316 x
1317 x
Bram Moolenaar2914a202020-09-27 18:24:03 +02001318 x
1319 x
1320 x
1321 x
1322 x
1323 x
1324 enddef
1325 call Func()
1326 [END]
Bram Moolenaarb5b94802020-12-13 17:50:20 +01001327 CheckScriptFailure(lines, 'E1145: Missing heredoc end marker: END')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001328 delfunc! g:Func
Bram Moolenaarc0e29012020-09-27 14:22:48 +02001329enddef
1330
Bram Moolenaar083966f2020-12-08 11:29:40 +01001331def Test_var_func_call()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001332 var lines =<< trim END
1333 vim9script
1334 func GetValue()
1335 if exists('g:count')
1336 let g:count += 1
1337 else
1338 let g:count = 1
1339 endif
1340 return 'this'
1341 endfunc
1342 var val: string = GetValue()
1343 # env var is always a string
1344 var env = $TERM
1345 END
1346 writefile(lines, 'Xfinished')
1347 source Xfinished
1348 # GetValue() is not called during discovery phase
1349 assert_equal(1, g:count)
1350
1351 unlet g:count
1352 delete('Xfinished')
1353enddef
1354
Bram Moolenaar083966f2020-12-08 11:29:40 +01001355def Test_var_missing_type()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001356 var lines =<< trim END
1357 vim9script
1358 var name = g:unknown
1359 END
1360 CheckScriptFailure(lines, 'E121:')
1361
1362 lines =<< trim END
1363 vim9script
1364 var nr: number = 123
1365 var name = nr
1366 END
1367 CheckScriptSuccess(lines)
1368enddef
1369
Bram Moolenaar083966f2020-12-08 11:29:40 +01001370def Test_var_declaration()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001371 var lines =<< trim END
1372 vim9script
1373 var name: string
1374 g:var_uninit = name
1375 name = 'text'
1376 g:var_test = name
1377 # prefixing s: is optional
1378 s:name = 'prefixed'
1379 g:var_prefixed = s:name
1380
Bram Moolenaar7257af42020-12-08 22:08:51 +01001381 const FOO: number = 123
1382 assert_equal(123, FOO)
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001383 const FOOS = 'foos'
1384 assert_equal('foos', FOOS)
1385 final FLIST = [1]
1386 assert_equal([1], FLIST)
1387 FLIST[0] = 11
1388 assert_equal([11], FLIST)
1389
1390 const g:FOO: number = 321
1391 assert_equal(321, g:FOO)
1392 const g:FOOS = 'gfoos'
1393 assert_equal('gfoos', g:FOOS)
1394 final g:FLIST = [2]
1395 assert_equal([2], g:FLIST)
1396 g:FLIST[0] = 22
1397 assert_equal([22], g:FLIST)
1398
Bram Moolenaard877a572021-04-01 19:42:48 +02001399 def SetGlobalConst()
1400 const g:globConst = 123
1401 enddef
1402 SetGlobalConst()
1403 assert_equal(123, g:globConst)
1404 assert_true(islocked('g:globConst'))
1405
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001406 const w:FOO: number = 46
1407 assert_equal(46, w:FOO)
1408 const w:FOOS = 'wfoos'
1409 assert_equal('wfoos', w:FOOS)
1410 final w:FLIST = [3]
1411 assert_equal([3], w:FLIST)
1412 w:FLIST[0] = 33
1413 assert_equal([33], w:FLIST)
Bram Moolenaar7257af42020-12-08 22:08:51 +01001414
Bram Moolenaar55759b52020-09-30 22:59:42 +02001415 var s:other: number
1416 other = 1234
1417 g:other_var = other
1418
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01001419 var xyz: string # comment
1420
Bram Moolenaar55759b52020-09-30 22:59:42 +02001421 # type is inferred
Bram Moolenaar3862ea32021-01-01 21:05:55 +01001422 var s:dict = {['a']: 222}
Bram Moolenaar55759b52020-09-30 22:59:42 +02001423 def GetDictVal(key: any)
1424 g:dict_val = s:dict[key]
1425 enddef
1426 GetDictVal('a')
Bram Moolenaar08251752021-01-11 21:20:18 +01001427
1428 final adict: dict<string> = {}
1429 def ChangeAdict()
1430 adict.foo = 'foo'
1431 enddef
1432 ChangeAdict()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001433 END
1434 CheckScriptSuccess(lines)
1435 assert_equal('', g:var_uninit)
1436 assert_equal('text', g:var_test)
1437 assert_equal('prefixed', g:var_prefixed)
1438 assert_equal(1234, g:other_var)
1439 assert_equal(222, g:dict_val)
1440
1441 unlet g:var_uninit
1442 unlet g:var_test
1443 unlet g:var_prefixed
1444 unlet g:other_var
Bram Moolenaar6e50ec22021-04-03 19:32:44 +02001445 unlet g:globConst
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001446 unlet g:FOO
1447 unlet g:FOOS
1448 unlet g:FLIST
1449 unlet w:FOO
1450 unlet w:FOOS
1451 unlet w:FLIST
Bram Moolenaar55759b52020-09-30 22:59:42 +02001452enddef
1453
Bram Moolenaar083966f2020-12-08 11:29:40 +01001454def Test_var_declaration_fails()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001455 var lines =<< trim END
1456 vim9script
1457 final var: string
1458 END
1459 CheckScriptFailure(lines, 'E1125:')
1460
1461 lines =<< trim END
1462 vim9script
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001463 const g:constvar = 'string'
1464 g:constvar = 'xx'
1465 END
1466 CheckScriptFailure(lines, 'E741:')
1467 unlet g:constvar
1468
1469 lines =<< trim END
1470 vim9script
Bram Moolenaarf5906aa2021-04-02 14:35:15 +02001471 var name = 'one'
1472 lockvar name
1473 def SetLocked()
1474 name = 'two'
1475 enddef
1476 SetLocked()
1477 END
Bram Moolenaardcf29ac2021-04-02 14:44:02 +02001478 CheckScriptFailure(lines, 'E741: Value is locked: name', 1)
1479
1480 lines =<< trim END
1481 let s:legacy = 'one'
1482 lockvar s:legacy
1483 def SetLocked()
1484 s:legacy = 'two'
1485 enddef
1486 call SetLocked()
1487 END
1488 CheckScriptFailure(lines, 'E741: Value is locked: s:legacy', 1)
Bram Moolenaarf5906aa2021-04-02 14:35:15 +02001489
1490 lines =<< trim END
1491 vim9script
Bram Moolenaard877a572021-04-01 19:42:48 +02001492 def SetGlobalConst()
1493 const g:globConst = 123
1494 enddef
1495 SetGlobalConst()
1496 g:globConst = 234
1497 END
Bram Moolenaar6e50ec22021-04-03 19:32:44 +02001498 CheckScriptFailure(lines, 'E741: Value is locked: g:globConst', 6)
Bram Moolenaard877a572021-04-01 19:42:48 +02001499 unlet g:globConst
1500
1501 lines =<< trim END
1502 vim9script
Bram Moolenaar08251752021-01-11 21:20:18 +01001503 const cdict: dict<string> = {}
1504 def Change()
1505 cdict.foo = 'foo'
1506 enddef
1507 defcompile
1508 END
1509 CheckScriptFailure(lines, 'E46:')
1510
1511 lines =<< trim END
1512 vim9script
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001513 final w:finalvar = [9]
1514 w:finalvar = [8]
1515 END
1516 CheckScriptFailure(lines, 'E1122:')
1517 unlet w:finalvar
1518
1519 lines =<< trim END
1520 vim9script
Bram Moolenaar55759b52020-09-30 22:59:42 +02001521 const var: string
1522 END
1523 CheckScriptFailure(lines, 'E1021:')
1524
1525 lines =<< trim END
1526 vim9script
1527 var 9var: string
1528 END
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01001529 CheckScriptFailure(lines, 'E488:')
Bram Moolenaar7257af42020-12-08 22:08:51 +01001530
1531 CheckDefFailure(['var foo.bar = 2'], 'E1087:')
1532 CheckDefFailure(['var foo[3] = 2'], 'E1087:')
1533 CheckDefFailure(['const foo: number'], 'E1021:')
Bram Moolenaar55759b52020-09-30 22:59:42 +02001534enddef
1535
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001536def Test_script_local_in_legacy()
1537 # OK to define script-local later when prefixed with s:
1538 var lines =<< trim END
1539 def SetLater()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001540 s:legvar = 'two'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001541 enddef
1542 defcompile
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001543 let s:legvar = 'one'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001544 call SetLater()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001545 call assert_equal('two', s:legvar)
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001546 END
1547 CheckScriptSuccess(lines)
1548
1549 # OK to leave out s: prefix when script-local already defined
1550 lines =<< trim END
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001551 let s:legvar = 'one'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001552 def SetNoPrefix()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001553 legvar = 'two'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001554 enddef
1555 call SetNoPrefix()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001556 call assert_equal('two', s:legvar)
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001557 END
1558 CheckScriptSuccess(lines)
1559
1560 # Not OK to leave out s: prefix when script-local defined later
1561 lines =<< trim END
1562 def SetLaterNoPrefix()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001563 legvar = 'two'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001564 enddef
1565 defcompile
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001566 let s:legvar = 'one'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001567 END
1568 CheckScriptFailure(lines, 'E476:', 1)
1569enddef
1570
Bram Moolenaar083966f2020-12-08 11:29:40 +01001571def Test_var_type_check()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001572 var lines =<< trim END
1573 vim9script
1574 var name: string
1575 name = 1234
1576 END
1577 CheckScriptFailure(lines, 'E1012:')
1578
1579 lines =<< trim END
1580 vim9script
1581 var name:string
1582 END
1583 CheckScriptFailure(lines, 'E1069:')
1584
1585 lines =<< trim END
1586 vim9script
1587 var name: asdf
1588 END
1589 CheckScriptFailure(lines, 'E1010:')
1590
1591 lines =<< trim END
1592 vim9script
1593 var s:l: list<number>
1594 s:l = []
1595 END
1596 CheckScriptSuccess(lines)
1597
1598 lines =<< trim END
1599 vim9script
1600 var s:d: dict<number>
1601 s:d = {}
1602 END
1603 CheckScriptSuccess(lines)
1604enddef
1605
1606let g:dict_number = #{one: 1, two: 2}
1607
Bram Moolenaar083966f2020-12-08 11:29:40 +01001608def Test_var_list_dict_type()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001609 var ll: list<number>
1610 ll = [1, 2, 2, 3, 3, 3]->uniq()
1611 ll->assert_equal([1, 2, 3])
1612
1613 var dd: dict<number>
1614 dd = g:dict_number
1615 dd->assert_equal(g:dict_number)
1616
1617 var lines =<< trim END
1618 var ll: list<number>
1619 ll = [1, 2, 3]->map('"one"')
1620 END
Bram Moolenaar70250fb2021-01-16 19:01:53 +01001621 CheckDefExecFailure(lines, 'E1012: Type mismatch; expected number but got string')
Bram Moolenaar55759b52020-09-30 22:59:42 +02001622enddef
1623
Bram Moolenaar083966f2020-12-08 11:29:40 +01001624def Test_cannot_use_let()
1625 CheckDefAndScriptFailure(['let a = 34'], 'E1126:', 1)
1626enddef
1627
Bram Moolenaar55759b52020-09-30 22:59:42 +02001628def Test_unlet()
1629 g:somevar = 'yes'
1630 assert_true(exists('g:somevar'))
1631 unlet g:somevar
1632 assert_false(exists('g:somevar'))
1633 unlet! g:somevar
1634
1635 # also works for script-local variable in legacy Vim script
1636 s:somevar = 'legacy'
1637 assert_true(exists('s:somevar'))
1638 unlet s:somevar
1639 assert_false(exists('s:somevar'))
1640 unlet! s:somevar
1641
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001642 CheckDefExecFailure([
1643 'var dd = 111',
1644 'unlet dd',
1645 ], 'E1081:', 2)
1646
Bram Moolenaar752fc692021-01-04 21:57:11 +01001647 # dict unlet
1648 var dd = {a: 1, b: 2, c: 3}
1649 unlet dd['a']
1650 unlet dd.c
1651 assert_equal({b: 2}, dd)
1652
1653 # list unlet
1654 var ll = [1, 2, 3, 4]
1655 unlet ll[1]
1656 unlet ll[-1]
1657 assert_equal([1, 3], ll)
1658
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001659 ll = [1, 2, 3, 4]
1660 unlet ll[0 : 1]
1661 assert_equal([3, 4], ll)
1662
1663 ll = [1, 2, 3, 4]
1664 unlet ll[2 : 8]
1665 assert_equal([1, 2], ll)
1666
1667 ll = [1, 2, 3, 4]
1668 unlet ll[-2 : -1]
1669 assert_equal([1, 2], ll)
1670
1671 CheckDefFailure([
1672 'var ll = [1, 2]',
1673 'll[1 : 2] = 7',
1674 ], 'E1165:', 2)
1675 CheckDefFailure([
1676 'var dd = {a: 1}',
1677 'unlet dd["a" : "a"]',
1678 ], 'E1166:', 2)
1679 CheckDefExecFailure([
1680 'unlet g:adict[0 : 1]',
1681 ], 'E1148:', 1)
1682 CheckDefFailure([
1683 'var ll = [1, 2]',
1684 'unlet ll[0:1]',
1685 ], 'E1004:', 2)
1686 CheckDefFailure([
1687 'var ll = [1, 2]',
1688 'unlet ll[0 :1]',
1689 ], 'E1004:', 2)
1690 CheckDefFailure([
1691 'var ll = [1, 2]',
1692 'unlet ll[0: 1]',
1693 ], 'E1004:', 2)
1694
1695 CheckDefFailure([
1696 'var ll = [1, 2]',
1697 'unlet ll["x" : 1]',
1698 ], 'E1012:', 2)
1699 CheckDefFailure([
1700 'var ll = [1, 2]',
1701 'unlet ll[0 : "x"]',
1702 ], 'E1012:', 2)
1703
Bram Moolenaar752fc692021-01-04 21:57:11 +01001704 # list of dict unlet
1705 var dl = [{a: 1, b: 2}, {c: 3}]
1706 unlet dl[0]['b']
1707 assert_equal([{a: 1}, {c: 3}], dl)
1708
1709 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001710 'var ll = test_null_list()',
1711 'unlet ll[0]',
1712 ], 'E684:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001713 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001714 'var ll = [1]',
1715 'unlet ll[2]',
1716 ], 'E684:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001717 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001718 'var ll = [1]',
1719 'unlet ll[g:astring]',
Bram Moolenaarf30a14d2021-01-17 21:51:24 +01001720 ], 'E1012:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001721 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001722 'var dd = test_null_dict()',
1723 'unlet dd["a"]',
1724 ], 'E716:', 2)
1725 CheckDefExecFailure([
1726 'var dd = {a: 1}',
1727 'unlet dd["b"]',
1728 ], 'E716:', 2)
1729 CheckDefExecFailure([
1730 'var dd = {a: 1}',
1731 'unlet dd[g:alist]',
1732 ], 'E1105:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001733
Bram Moolenaar2ef951d2021-01-03 20:55:26 +01001734 # can compile unlet before variable exists
Bram Moolenaar752fc692021-01-04 21:57:11 +01001735 g:someDict = {key: 'val'}
1736 var k = 'key'
1737 unlet g:someDict[k]
1738 assert_equal({}, g:someDict)
1739 unlet g:someDict
1740 assert_false(exists('g:someDict'))
Bram Moolenaar2ef951d2021-01-03 20:55:26 +01001741
Bram Moolenaar55759b52020-09-30 22:59:42 +02001742 CheckScriptFailure([
1743 'vim9script',
1744 'var svar = 123',
1745 'unlet svar',
1746 ], 'E1081:')
1747 CheckScriptFailure([
1748 'vim9script',
1749 'var svar = 123',
1750 'unlet s:svar',
1751 ], 'E1081:')
1752 CheckScriptFailure([
1753 'vim9script',
1754 'var svar = 123',
1755 'def Func()',
1756 ' unlet svar',
1757 'enddef',
1758 'defcompile',
1759 ], 'E1081:')
1760 CheckScriptFailure([
1761 'vim9script',
1762 'var svar = 123',
Bram Moolenaar9aed7292020-12-18 15:38:00 +01001763 'func Func()',
1764 ' unlet s:svar',
1765 'endfunc',
1766 'Func()',
1767 ], 'E1081:')
1768 CheckScriptFailure([
1769 'vim9script',
1770 'var svar = 123',
Bram Moolenaar55759b52020-09-30 22:59:42 +02001771 'def Func()',
1772 ' unlet s:svar',
1773 'enddef',
1774 'defcompile',
1775 ], 'E1081:')
1776
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001777 writefile(['vim9script', 'export var svar = 1234'], 'XunletExport.vim')
1778 var lines =<< trim END
1779 vim9script
1780 import svar from './XunletExport.vim'
1781 def UnletSvar()
1782 unlet svar
1783 enddef
1784 defcompile
1785 END
1786 CheckScriptFailure(lines, 'E1081:', 1)
1787 delete('XunletExport.vim')
1788
Bram Moolenaar55759b52020-09-30 22:59:42 +02001789 $ENVVAR = 'foobar'
1790 assert_equal('foobar', $ENVVAR)
1791 unlet $ENVVAR
1792 assert_equal('', $ENVVAR)
1793enddef
1794
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01001795def Test_expr_error_no_assign()
1796 var lines =<< trim END
1797 vim9script
1798 var x = invalid
1799 echo x
1800 END
1801 CheckScriptFailureList(lines, ['E121:', 'E121:'])
1802
1803 lines =<< trim END
1804 vim9script
1805 var x = 1 / 0
1806 echo x
1807 END
Bram Moolenaar227c58a2021-04-28 20:40:44 +02001808 CheckScriptFailure(lines, 'E1154:')
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01001809
1810 lines =<< trim END
1811 vim9script
1812 var x = 1 % 0
1813 echo x
1814 END
Bram Moolenaar227c58a2021-04-28 20:40:44 +02001815 CheckScriptFailure(lines, 'E1154:')
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01001816
1817 lines =<< trim END
1818 var x: string 'string'
1819 END
1820 CheckDefAndScriptFailure(lines, 'E488:')
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01001821enddef
1822
1823
Bram Moolenaar17126b12021-01-07 22:03:02 +01001824def Test_assign_command_modifier()
1825 var lines =<< trim END
Bram Moolenaar9e0f8832021-01-09 12:09:22 +01001826 var verbose = 0
1827 verbose = 1
1828 assert_equal(1, verbose)
1829 silent verbose = 2
1830 assert_equal(2, verbose)
1831 silent verbose += 2
1832 assert_equal(4, verbose)
1833 silent verbose -= 1
1834 assert_equal(3, verbose)
Bram Moolenaar17126b12021-01-07 22:03:02 +01001835
Bram Moolenaar9e0f8832021-01-09 12:09:22 +01001836 var topleft = {one: 1}
1837 sandbox topleft.one = 3
1838 assert_equal({one: 3}, topleft)
1839 leftabove topleft[' '] = 4
1840 assert_equal({one: 3, ' ': 4}, topleft)
1841
1842 var x: number
1843 var y: number
1844 silent [x, y] = [1, 2]
1845 assert_equal(1, x)
1846 assert_equal(2, y)
Bram Moolenaar17126b12021-01-07 22:03:02 +01001847 END
1848 CheckDefAndScriptSuccess(lines)
1849enddef
1850
Bram Moolenaar74f4a962021-06-17 21:03:07 +02001851def Test_assign_alt_buf_register()
1852 var lines =<< trim END
1853 edit 'file_b1'
1854 var b1 = bufnr()
1855 edit 'file_b2'
1856 var b2 = bufnr()
1857 assert_equal(b1, bufnr('#'))
1858 @# = b2
1859 assert_equal(b2, bufnr('#'))
1860 END
1861 CheckDefAndScriptSuccess(lines)
1862enddef
1863
Bram Moolenaar32154662021-03-28 21:14:06 +02001864def Test_script_funcref_case()
1865 var lines =<< trim END
1866 var Len = (s: string): number => len(s) + 1
1867 assert_equal(5, Len('asdf'))
1868 END
1869 CheckDefAndScriptSuccess(lines)
1870
1871 lines =<< trim END
1872 var len = (s: string): number => len(s) + 1
1873 END
1874 CheckDefAndScriptFailure(lines, 'E704:')
1875
1876 lines =<< trim END
1877 vim9script
1878 var s:Len = (s: string): number => len(s) + 2
1879 assert_equal(6, Len('asdf'))
1880 END
1881 CheckScriptSuccess(lines)
1882
1883 lines =<< trim END
1884 vim9script
1885 var s:len = (s: string): number => len(s) + 1
1886 END
1887 CheckScriptFailure(lines, 'E704:')
1888enddef
1889
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001890def Test_inc_dec()
1891 var lines =<< trim END
1892 var nr = 7
1893 ++nr
Bram Moolenaar730bf302021-04-24 20:43:56 +02001894 assert_equal(8, nr)
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001895 --nr
Bram Moolenaar730bf302021-04-24 20:43:56 +02001896 assert_equal(7, nr)
Bram Moolenaarf3d30842021-06-25 19:29:30 +02001897 ++nr | ++nr
1898 assert_equal(9, nr)
1899 ++nr # comment
1900 assert_equal(10, nr)
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001901
1902 var ll = [1, 2]
1903 --ll[0]
1904 ++ll[1]
Bram Moolenaar730bf302021-04-24 20:43:56 +02001905 assert_equal([0, 3], ll)
1906
1907 g:count = 1
1908 ++g:count
1909 --g:count
1910 assert_equal(1, g:count)
1911 unlet g:count
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001912 END
1913 CheckDefAndScriptSuccess(lines)
Bram Moolenaar22480d12021-06-25 21:31:09 +02001914
1915 lines =<< trim END
1916 var nr = 7
1917 ++ nr
1918 END
1919 CheckDefAndScriptFailure(lines, "E1202: No white space allowed after '++': ++ nr")
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001920enddef
1921
Bram Moolenaar55759b52020-09-30 22:59:42 +02001922
Bram Moolenaar971caf52020-09-21 22:21:42 +02001923" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker