blob: 5f5b5d740aa3a00f9c5ac434b182f80d0499f069 [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
Bram Moolenaar4d5dfe22021-06-26 13:59:29 +0200292 var reslist = []
293 for text in ['aaa {bbb} ccc', 'ddd {eee} fff']
294 var before: string
295 var middle: string
296 var after: string
297 [_, before, middle, after; _] = text->matchlist('\(.\{-\}\){\(.\{-\}\)}\(.*\)')
298 reslist->add(before)->add(middle)->add(after)
299 endfor
300 assert_equal(['aaa ', 'bbb', ' ccc', 'ddd ', 'eee', ' fff'], reslist)
301
Bram Moolenaar035bd1c2021-06-21 19:44:11 +0200302 var a = 1
303 var b = 3
304 [a, b] += [2, 4]
305 assert_equal(3, a)
306 assert_equal(7, b)
307
308 [a, b] -= [1, 2]
309 assert_equal(2, a)
310 assert_equal(5, b)
311
312 [a, b] *= [3, 2]
313 assert_equal(6, a)
314 assert_equal(10, b)
315
316 [a, b] /= [2, 4]
317 assert_equal(3, a)
318 assert_equal(2, b)
319
320 [a, b] = [17, 15]
321 [a, b] %= [5, 3]
322 assert_equal(2, a)
323 assert_equal(0, b)
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100324 END
325 CheckDefAndScriptSuccess(lines)
326
327 lines =<< trim END
328 var v1: number
329 var v2: number
330 [v1, v2] =
331 END
332 CheckDefFailure(lines, 'E1097:', 5)
333
334 lines =<< trim END
335 var v1: number
336 var v2: number
337 [v1, v2] = xxx
338 END
339 CheckDefFailure(lines, 'E1001:', 3)
340
341 lines =<< trim END
342 var v1: number
343 var v2: number
344 [v1, v2] = popup_clear()
345 END
346 CheckDefFailure(lines, 'E1031:', 3)
347
348 lines =<< trim END
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100349 [v1, v2] = [1, 2]
350 END
351 CheckDefFailure(lines, 'E1089', 1)
352 CheckScriptFailure(['vim9script'] + lines, 'E1089', 2)
353
354 lines =<< trim END
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100355 var v1: number
356 var v2: number
357 [v1, v2] = ''
358 END
359 CheckDefFailure(lines, 'E1012: Type mismatch; expected list<any> but got string', 3)
Bram Moolenaarf785aa12021-02-11 21:19:34 +0100360
361 lines =<< trim END
362 g:values = [false, 0]
363 var x: bool
364 var y: string
365 [x, y] = g:values
366 END
367 CheckDefExecAndScriptFailure(lines, 'E1163: Variable 2: type mismatch, expected string but got number')
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100368enddef
369
Bram Moolenaar7f764942020-12-02 15:11:18 +0100370def Test_assign_linebreak()
371 var nr: number
372 nr =
373 123
374 assert_equal(123, nr)
375
376 var n2: number
377 [nr, n2] =
378 [12, 34]
379 assert_equal(12, nr)
380 assert_equal(34, n2)
381
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100382 CheckDefFailure(["var x = #"], 'E1097:', 3)
Bram Moolenaar77709b12021-04-03 21:01:01 +0200383
384 var lines =<< trim END
385 var x: list<string> = ['a']
386 var y: list<number> = x
387 ->copy()
388 ->copy()
389 END
390 CheckDefFailure(lines, 'E1012:', 2)
Bram Moolenaar7f764942020-12-02 15:11:18 +0100391enddef
392
Bram Moolenaardc234ca2020-11-28 18:52:33 +0100393def Test_assign_index()
394 # list of list
395 var l1: list<number>
396 l1[0] = 123
397 assert_equal([123], l1)
398
399 var l2: list<list<number>>
400 l2[0] = []
401 l2[0][0] = 123
402 assert_equal([[123]], l2)
403
404 var l3: list<list<list<number>>>
405 l3[0] = []
406 l3[0][0] = []
407 l3[0][0][0] = 123
408 assert_equal([[[123]]], l3)
409
410 var lines =<< trim END
411 var l3: list<list<number>>
412 l3[0] = []
413 l3[0][0] = []
414 END
415 CheckDefFailure(lines, 'E1012: Type mismatch; expected number but got list<unknown>', 3)
416
417 # dict of dict
418 var d1: dict<number>
419 d1.one = 1
420 assert_equal({one: 1}, d1)
421
422 var d2: dict<dict<number>>
423 d2.one = {}
424 d2.one.two = 123
425 assert_equal({one: {two: 123}}, d2)
426
427 var d3: dict<dict<dict<number>>>
428 d3.one = {}
429 d3.one.two = {}
430 d3.one.two.three = 123
431 assert_equal({one: {two: {three: 123}}}, d3)
432
433 lines =<< trim END
434 var d3: dict<dict<number>>
435 d3.one = {}
436 d3.one.two = {}
437 END
438 CheckDefFailure(lines, 'E1012: Type mismatch; expected number but got dict<unknown>', 3)
439
Bram Moolenaaracbae182020-12-13 18:44:43 +0100440 lines =<< trim END
441 var lines: list<string>
442 lines['a'] = 'asdf'
443 END
Bram Moolenaarf30a14d2021-01-17 21:51:24 +0100444 CheckDefFailure(lines, 'E1012:', 2)
Bram Moolenaaracbae182020-12-13 18:44:43 +0100445
446 lines =<< trim END
447 var lines: string
448 lines[9] = 'asdf'
449 END
450 CheckDefFailure(lines, 'E1141:', 2)
451
Bram Moolenaardc234ca2020-11-28 18:52:33 +0100452 # list of dict
453 var ld: list<dict<number>>
454 ld[0] = {}
455 ld[0].one = 123
456 assert_equal([{one: 123}], ld)
457
458 lines =<< trim END
459 var ld: list<dict<number>>
460 ld[0] = []
461 END
462 CheckDefFailure(lines, 'E1012: Type mismatch; expected dict<number> but got list<unknown>', 2)
463
464 # dict of list
465 var dl: dict<list<number>>
466 dl.one = []
467 dl.one[0] = 123
468 assert_equal({one: [123]}, dl)
469
470 lines =<< trim END
471 var dl: dict<list<number>>
472 dl.one = {}
473 END
474 CheckDefFailure(lines, 'E1012: Type mismatch; expected list<number> but got dict<unknown>', 2)
475enddef
476
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200477def Test_extend_list()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200478 var lines =<< trim END
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200479 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200480 var l: list<number>
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200481 l += [123]
482 assert_equal([123], l)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100483 END
484 CheckScriptSuccess(lines)
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200485
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100486 lines =<< trim END
487 vim9script
488 var list: list<string>
489 extend(list, ['x'])
490 assert_equal(['x'], list)
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200491 END
492 CheckScriptSuccess(lines)
Bram Moolenaar1a739232020-10-10 15:37:58 +0200493
494 # appending to NULL list from a function
495 lines =<< trim END
496 vim9script
497 var list: list<string>
498 def Func()
499 list += ['a', 'b']
500 enddef
501 Func()
502 assert_equal(['a', 'b'], list)
503 END
504 CheckScriptSuccess(lines)
Bram Moolenaar3beaf9c2020-12-18 17:23:14 +0100505 lines =<< trim END
506 vim9script
507 var list: list<string>
508 def Func()
509 extend(list, ['x', 'b'])
510 enddef
511 Func()
512 assert_equal(['x', 'b'], list)
513 END
514 CheckScriptSuccess(lines)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100515
516 lines =<< trim END
517 vim9script
518 var l: list<string> = test_null_list()
519 extend(l, ['x'])
520 assert_equal(['x'], l)
521 END
522 CheckScriptSuccess(lines)
523
524 lines =<< trim END
525 vim9script
526 extend(test_null_list(), ['x'])
527 END
528 CheckScriptFailure(lines, 'E1134:', 2)
529enddef
530
531def Test_extend_dict()
532 var lines =<< trim END
533 vim9script
534 var d: dict<number>
Bram Moolenaare0de1712020-12-02 17:36:54 +0100535 extend(d, {a: 1})
536 assert_equal({a: 1}, d)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100537
538 var d2: dict<number>
539 d2['one'] = 1
Bram Moolenaare0de1712020-12-02 17:36:54 +0100540 assert_equal({one: 1}, d2)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100541 END
542 CheckScriptSuccess(lines)
543
544 lines =<< trim END
545 vim9script
546 var d: dict<string> = test_null_dict()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100547 extend(d, {a: 'x'})
548 assert_equal({a: 'x'}, d)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100549 END
550 CheckScriptSuccess(lines)
551
552 lines =<< trim END
553 vim9script
Bram Moolenaare0de1712020-12-02 17:36:54 +0100554 extend(test_null_dict(), {a: 'x'})
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100555 END
556 CheckScriptFailure(lines, 'E1133:', 2)
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200557enddef
558
Bram Moolenaar971caf52020-09-21 22:21:42 +0200559def Test_single_letter_vars()
560 # single letter variables
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200561 var a: number = 123
Bram Moolenaar971caf52020-09-21 22:21:42 +0200562 a = 123
563 assert_equal(123, a)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200564 var b: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200565 b = 123
566 assert_equal(123, b)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200567 var g: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200568 g = 123
569 assert_equal(123, g)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200570 var s: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200571 s = 123
572 assert_equal(123, s)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200573 var t: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200574 t = 123
575 assert_equal(123, t)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200576 var v: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200577 v = 123
578 assert_equal(123, v)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200579 var w: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200580 w = 123
581 assert_equal(123, w)
582enddef
583
584def Test_vim9_single_char_vars()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200585 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200586 vim9script
587
588 # single character variable declarations work
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200589 var a: string
590 var b: number
591 var l: list<any>
592 var s: string
593 var t: number
594 var v: number
595 var w: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200596
597 # script-local variables can be used without s: prefix
598 a = 'script-a'
599 b = 111
600 l = [1, 2, 3]
601 s = 'script-s'
602 t = 222
603 v = 333
604 w = 444
605
606 assert_equal('script-a', a)
607 assert_equal(111, b)
608 assert_equal([1, 2, 3], l)
609 assert_equal('script-s', s)
610 assert_equal(222, t)
611 assert_equal(333, v)
612 assert_equal(444, w)
613 END
614 writefile(lines, 'Xsinglechar')
615 source Xsinglechar
616 delete('Xsinglechar')
617enddef
618
619def Test_assignment_list()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200620 var list1: list<bool> = [false, true, false]
621 var list2: list<number> = [1, 2, 3]
622 var list3: list<string> = ['sdf', 'asdf']
623 var list4: list<any> = ['yes', true, 1234]
624 var list5: list<blob> = [0z01, 0z02]
Bram Moolenaar971caf52020-09-21 22:21:42 +0200625
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200626 var listS: list<string> = []
627 var listN: list<number> = []
Bram Moolenaar971caf52020-09-21 22:21:42 +0200628
629 assert_equal([1, 2, 3], list2)
630 list2[-1] = 99
631 assert_equal([1, 2, 99], list2)
632 list2[-2] = 88
633 assert_equal([1, 88, 99], list2)
634 list2[-3] = 77
635 assert_equal([77, 88, 99], list2)
636 list2 += [100]
637 assert_equal([77, 88, 99, 100], list2)
638
639 list3 += ['end']
640 assert_equal(['sdf', 'asdf', 'end'], list3)
641
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200642 CheckDefExecFailure(['var ll = [1, 2, 3]', 'll[-4] = 6'], 'E684:')
643 CheckDefExecFailure(['var [v1, v2] = [1, 2]'], 'E1092:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200644
645 # type becomes list<any>
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200646 var somelist = rand() > 0 ? [1, 2, 3] : ['a', 'b', 'c']
Bram Moolenaar4f5e3972020-12-21 17:30:50 +0100647
648 var lines =<< trim END
649 var d = {dd: test_null_list()}
650 d.dd[0] = 0
651 END
652 CheckDefExecFailure(lines, 'E1147:', 2)
Bram Moolenaar7a40ff02021-07-04 15:54:08 +0200653
654 lines =<< trim END
655 def OneArg(x: bool)
656 enddef
657 def TwoArgs(x: bool, y: bool)
658 enddef
659 var fl: list<func(bool, bool, bool)> = [OneArg, TwoArgs]
660 END
661 CheckDefExecAndScriptFailure(lines, 'E1012:', 5)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200662enddef
663
Bram Moolenaarb7480cd2021-07-04 18:28:15 +0200664def PartFunc(b: bool): string
665 return 'done'
666enddef
667
668def Test_assignment_partial()
669 var Partial: func(): string = function(PartFunc, [true])
670 assert_equal('done', Partial())
671enddef
672
Bram Moolenaarf30a14d2021-01-17 21:51:24 +0100673def Test_assignment_list_any_index()
674 var l: list<number> = [1, 2]
675 for [x, y, _]
676 in [[0, 1, ''], [1, 3, '']]
677 l[x] = l[x] + y
678 endfor
679 assert_equal([2, 5], l)
680enddef
681
Bram Moolenaar971caf52020-09-21 22:21:42 +0200682def Test_assignment_list_vim9script()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200683 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200684 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200685 var v1: number
686 var v2: number
687 var v3: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200688 [v1, v2, v3] = [1, 2, 3]
689 assert_equal([1, 2, 3], [v1, v2, v3])
690 END
691 CheckScriptSuccess(lines)
692enddef
693
694def Test_assignment_dict()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100695 var dict1: dict<bool> = {one: false, two: true}
696 var dict2: dict<number> = {one: 1, two: 2}
697 var dict3: dict<string> = {key: 'value'}
698 var dict4: dict<any> = {one: 1, two: '2'}
699 var dict5: dict<blob> = {one: 0z01, two: 0z02}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200700
701 # overwrite
702 dict3['key'] = 'another'
Bram Moolenaare0de1712020-12-02 17:36:54 +0100703 assert_equal(dict3, {key: 'another'})
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100704 dict3.key = 'yet another'
Bram Moolenaare0de1712020-12-02 17:36:54 +0100705 assert_equal(dict3, {key: 'yet another'})
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100706
Bram Moolenaard24602f2020-12-20 15:20:56 +0100707 # member "any" can also be a dict and assigned to
708 var anydict: dict<any> = {nest: {}, nr: 0}
709 anydict.nest['this'] = 123
710 anydict.nest.that = 456
711 assert_equal({nest: {this: 123, that: 456}, nr: 0}, anydict)
712
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100713 var lines =<< trim END
Bram Moolenaard88dc4d2020-12-20 14:49:42 +0100714 var dd = {}
715 dd.two = 2
716 assert_equal({two: 2}, dd)
717 END
Bram Moolenaar4f5e3972020-12-21 17:30:50 +0100718 CheckDefAndScriptSuccess(lines)
719
720 lines =<< trim END
721 var d = {dd: {}}
722 d.dd[0] = 2
723 d.dd['x'] = 3
724 d.dd.y = 4
725 assert_equal({dd: {0: 2, x: 3, y: 4}}, d)
726 END
727 CheckDefAndScriptSuccess(lines)
Bram Moolenaard88dc4d2020-12-20 14:49:42 +0100728
729 lines =<< trim END
Bram Moolenaare0de1712020-12-02 17:36:54 +0100730 var dd = {one: 1}
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100731 dd.one) = 2
732 END
Bram Moolenaard88dc4d2020-12-20 14:49:42 +0100733 CheckDefFailure(lines, 'E488:', 2)
734
735 lines =<< trim END
736 var dd = {one: 1}
737 var dd.one = 2
738 END
739 CheckDefAndScriptFailure(lines, 'E1017:', 2)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200740
741 # empty key can be used
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200742 var dd = {}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200743 dd[""] = 6
Bram Moolenaare0de1712020-12-02 17:36:54 +0100744 assert_equal({['']: 6}, dd)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200745
746 # type becomes dict<any>
Bram Moolenaare0de1712020-12-02 17:36:54 +0100747 var somedict = rand() > 0 ? {a: 1, b: 2} : {a: 'a', b: 'b'}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200748
749 # assignment to script-local dict
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100750 lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200751 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200752 var test: dict<any> = {}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200753 def FillDict(): dict<any>
754 test['a'] = 43
755 return test
756 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +0100757 assert_equal({a: 43}, FillDict())
Bram Moolenaar971caf52020-09-21 22:21:42 +0200758 END
759 CheckScriptSuccess(lines)
760
761 lines =<< trim END
762 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200763 var test: dict<any>
Bram Moolenaar971caf52020-09-21 22:21:42 +0200764 def FillDict(): dict<any>
765 test['a'] = 43
766 return test
767 enddef
768 FillDict()
Bram Moolenaar3beaf9c2020-12-18 17:23:14 +0100769 assert_equal({a: 43}, test)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200770 END
Bram Moolenaar3beaf9c2020-12-18 17:23:14 +0100771 CheckScriptSuccess(lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200772
773 # assignment to global dict
774 lines =<< trim END
775 vim9script
776 g:test = {}
777 def FillDict(): dict<any>
778 g:test['a'] = 43
779 return g:test
780 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +0100781 assert_equal({a: 43}, FillDict())
Bram Moolenaar971caf52020-09-21 22:21:42 +0200782 END
783 CheckScriptSuccess(lines)
784
785 # assignment to buffer dict
786 lines =<< trim END
787 vim9script
788 b:test = {}
789 def FillDict(): dict<any>
790 b:test['a'] = 43
791 return b:test
792 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +0100793 assert_equal({a: 43}, FillDict())
Bram Moolenaar971caf52020-09-21 22:21:42 +0200794 END
795 CheckScriptSuccess(lines)
Bram Moolenaar4f5e3972020-12-21 17:30:50 +0100796
797 lines =<< trim END
798 var d = {dd: test_null_dict()}
799 d.dd[0] = 0
800 END
801 CheckDefExecFailure(lines, 'E1103:', 2)
802
803 lines =<< trim END
804 var d = {dd: 'string'}
805 d.dd[0] = 0
806 END
807 CheckDefExecFailure(lines, 'E1148:', 2)
Bram Moolenaar3a3b10e2021-06-26 15:00:59 +0200808
809 lines =<< trim END
810 var n: any
811 n.key = 5
812 END
813 CheckDefExecAndScriptFailure2(lines, 'E1148:', 'E1203: Dot can only be used on a dictionary: n.key = 5', 2)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200814enddef
815
816def Test_assignment_local()
817 # Test in a separated file in order not to the current buffer/window/tab is
818 # changed.
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200819 var script_lines: list<string> =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200820 let b:existing = 'yes'
821 let w:existing = 'yes'
822 let t:existing = 'yes'
823
824 def Test_assignment_local_internal()
825 b:newvar = 'new'
826 assert_equal('new', b:newvar)
827 assert_equal('yes', b:existing)
828 b:existing = 'no'
829 assert_equal('no', b:existing)
830 b:existing ..= 'NO'
831 assert_equal('noNO', b:existing)
832
833 w:newvar = 'new'
834 assert_equal('new', w:newvar)
835 assert_equal('yes', w:existing)
836 w:existing = 'no'
837 assert_equal('no', w:existing)
838 w:existing ..= 'NO'
839 assert_equal('noNO', w:existing)
840
841 t:newvar = 'new'
842 assert_equal('new', t:newvar)
843 assert_equal('yes', t:existing)
844 t:existing = 'no'
845 assert_equal('no', t:existing)
846 t:existing ..= 'NO'
847 assert_equal('noNO', t:existing)
848 enddef
849 call Test_assignment_local_internal()
850 END
851 CheckScriptSuccess(script_lines)
852enddef
853
854def Test_assignment_default()
Bram Moolenaar971caf52020-09-21 22:21:42 +0200855 # Test default values.
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200856 var thebool: bool
Bram Moolenaar971caf52020-09-21 22:21:42 +0200857 assert_equal(v:false, thebool)
858
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200859 var thenumber: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200860 assert_equal(0, thenumber)
861
862 if has('float')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200863 var thefloat: float
Bram Moolenaar971caf52020-09-21 22:21:42 +0200864 assert_equal(0.0, thefloat)
865 endif
866
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200867 var thestring: string
Bram Moolenaar971caf52020-09-21 22:21:42 +0200868 assert_equal('', thestring)
869
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200870 var theblob: blob
Bram Moolenaar971caf52020-09-21 22:21:42 +0200871 assert_equal(0z, theblob)
872
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200873 var Thefunc: func
Bram Moolenaar971caf52020-09-21 22:21:42 +0200874 assert_equal(test_null_function(), Thefunc)
875
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200876 var thelist: list<any>
Bram Moolenaar971caf52020-09-21 22:21:42 +0200877 assert_equal([], thelist)
878
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200879 var thedict: dict<any>
Bram Moolenaar971caf52020-09-21 22:21:42 +0200880 assert_equal({}, thedict)
881
882 if has('channel')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200883 var thejob: job
Bram Moolenaar971caf52020-09-21 22:21:42 +0200884 assert_equal(test_null_job(), thejob)
885
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200886 var thechannel: channel
Bram Moolenaar971caf52020-09-21 22:21:42 +0200887 assert_equal(test_null_channel(), thechannel)
888
889 if has('unix') && executable('cat')
890 # check with non-null job and channel, types must match
Bram Moolenaare0de1712020-12-02 17:36:54 +0100891 thejob = job_start("cat ", {})
Bram Moolenaar971caf52020-09-21 22:21:42 +0200892 thechannel = job_getchannel(thejob)
893 job_stop(thejob, 'kill')
894 endif
895 endif
896
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200897 var nr = 1234 | nr = 5678
Bram Moolenaar971caf52020-09-21 22:21:42 +0200898 assert_equal(5678, nr)
899enddef
900
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100901let scriptvar = 'init'
902
Bram Moolenaar971caf52020-09-21 22:21:42 +0200903def Test_assignment_var_list()
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100904 var lines =<< trim END
905 var v1: string
906 var v2: string
907 var vrem: list<string>
908 [v1] = ['aaa']
909 assert_equal('aaa', v1)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200910
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100911 [v1, v2] = ['one', 'two']
912 assert_equal('one', v1)
913 assert_equal('two', v2)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200914
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100915 [v1, v2; vrem] = ['one', 'two']
916 assert_equal('one', v1)
917 assert_equal('two', v2)
918 assert_equal([], vrem)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200919
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100920 [v1, v2; vrem] = ['one', 'two', 'three']
921 assert_equal('one', v1)
922 assert_equal('two', v2)
923 assert_equal(['three'], vrem)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200924
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100925 [&ts, &sw] = [3, 4]
926 assert_equal(3, &ts)
927 assert_equal(4, &sw)
928 set ts=8 sw=4
Bram Moolenaar004d9b02020-11-30 21:40:03 +0100929
930 [@a, @z] = ['aa', 'zz']
931 assert_equal('aa', @a)
932 assert_equal('zz', @z)
933
934 [$SOME_VAR, $OTHER_VAR] = ['some', 'other']
935 assert_equal('some', $SOME_VAR)
936 assert_equal('other', $OTHER_VAR)
937
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100938 [g:globalvar, b:bufvar, w:winvar, t:tabvar, v:errmsg] =
939 ['global', 'buf', 'win', 'tab', 'error']
Bram Moolenaar004d9b02020-11-30 21:40:03 +0100940 assert_equal('global', g:globalvar)
Bram Moolenaar004d9b02020-11-30 21:40:03 +0100941 assert_equal('buf', b:bufvar)
942 assert_equal('win', w:winvar)
943 assert_equal('tab', t:tabvar)
944 assert_equal('error', v:errmsg)
945 unlet g:globalvar
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100946 END
947 CheckDefAndScriptSuccess(lines)
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100948
949 [g:globalvar, s:scriptvar, b:bufvar] = ['global', 'script', 'buf']
950 assert_equal('global', g:globalvar)
951 assert_equal('script', s:scriptvar)
952 assert_equal('buf', b:bufvar)
953
954 lines =<< trim END
955 vim9script
956 var s:scriptvar = 'init'
957 [g:globalvar, s:scriptvar, w:winvar] = ['global', 'script', 'win']
958 assert_equal('global', g:globalvar)
959 assert_equal('script', s:scriptvar)
960 assert_equal('win', w:winvar)
961 END
962 CheckScriptSuccess(lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200963enddef
964
965def Test_assignment_vim9script()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200966 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200967 vim9script
968 def Func(): list<number>
969 return [1, 2]
970 enddef
Bram Moolenaar98309382020-09-27 21:58:45 +0200971 var name1: number
972 var name2: number
973 [name1, name2] =
Bram Moolenaar971caf52020-09-21 22:21:42 +0200974 Func()
Bram Moolenaar98309382020-09-27 21:58:45 +0200975 assert_equal(1, name1)
976 assert_equal(2, name2)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200977 var ll =
Bram Moolenaar971caf52020-09-21 22:21:42 +0200978 Func()
979 assert_equal([1, 2], ll)
980
981 @/ = 'text'
982 assert_equal('text', @/)
983 @0 = 'zero'
984 assert_equal('zero', @0)
985 @1 = 'one'
986 assert_equal('one', @1)
987 @9 = 'nine'
988 assert_equal('nine', @9)
989 @- = 'minus'
990 assert_equal('minus', @-)
991 if has('clipboard_working')
992 @* = 'star'
993 assert_equal('star', @*)
994 @+ = 'plus'
995 assert_equal('plus', @+)
996 endif
997
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200998 var a: number = 123
Bram Moolenaar971caf52020-09-21 22:21:42 +0200999 assert_equal(123, a)
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001000 var s: string = 'yes'
Bram Moolenaar971caf52020-09-21 22:21:42 +02001001 assert_equal('yes', s)
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001002 var b: number = 42
Bram Moolenaar971caf52020-09-21 22:21:42 +02001003 assert_equal(42, b)
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001004 var w: number = 43
Bram Moolenaar971caf52020-09-21 22:21:42 +02001005 assert_equal(43, w)
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001006 var t: number = 44
Bram Moolenaar971caf52020-09-21 22:21:42 +02001007 assert_equal(44, t)
Bram Moolenaar7a66a172020-10-16 19:56:12 +02001008
1009 var to_var = 0
1010 to_var = 3
1011 assert_equal(3, to_var)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001012 END
1013 CheckScriptSuccess(lines)
Bram Moolenaar06f9c692020-09-27 21:27:40 +02001014
1015 lines =<< trim END
1016 vim9script
1017 var n: number
1018 def Func()
1019 n = 'string'
1020 enddef
1021 defcompile
1022 END
1023 CheckScriptFailure(lines, 'E1012: Type mismatch; expected number but got string')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001024enddef
1025
1026def Mess(): string
1027 v:foldstart = 123
1028 return 'xxx'
1029enddef
1030
1031def Test_assignment_failure()
Bram Moolenaar98309382020-09-27 21:58:45 +02001032 CheckDefFailure(['var name=234'], 'E1004:')
1033 CheckDefFailure(['var name =234'], 'E1004:')
1034 CheckDefFailure(['var name= 234'], 'E1004:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001035
Bram Moolenaar98309382020-09-27 21:58:45 +02001036 CheckScriptFailure(['vim9script', 'var name=234'], 'E1004:')
1037 CheckScriptFailure(['vim9script', 'var name=234'], "before and after '='")
1038 CheckScriptFailure(['vim9script', 'var name =234'], 'E1004:')
1039 CheckScriptFailure(['vim9script', 'var name= 234'], 'E1004:')
1040 CheckScriptFailure(['vim9script', 'var name = 234', 'name+=234'], 'E1004:')
1041 CheckScriptFailure(['vim9script', 'var name = 234', 'name+=234'], "before and after '+='")
1042 CheckScriptFailure(['vim9script', 'var name = "x"', 'name..="y"'], 'E1004:')
1043 CheckScriptFailure(['vim9script', 'var name = "x"', 'name..="y"'], "before and after '..='")
Bram Moolenaar971caf52020-09-21 22:21:42 +02001044
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001045 CheckDefFailure(['var true = 1'], 'E1034:')
1046 CheckDefFailure(['var false = 1'], 'E1034:')
Bram Moolenaar74235772021-06-12 14:53:05 +02001047 CheckDefFailure(['var null = 1'], 'E1034:')
1048 CheckDefFailure(['var this = 1'], 'E1034:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001049
1050 CheckDefFailure(['[a; b; c] = g:list'], 'E452:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001051 CheckDefExecFailure(['var a: number',
Bram Moolenaar971caf52020-09-21 22:21:42 +02001052 '[a] = test_null_list()'], 'E1093:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001053 CheckDefExecFailure(['var a: number',
Bram Moolenaar971caf52020-09-21 22:21:42 +02001054 '[a] = []'], 'E1093:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001055 CheckDefExecFailure(['var x: number',
1056 'var y: number',
Bram Moolenaar971caf52020-09-21 22:21:42 +02001057 '[x, y] = [1]'], 'E1093:')
Bram Moolenaar352134b2020-10-17 22:04:08 +02001058 CheckDefExecFailure(['var x: string',
1059 'var y: string',
1060 '[x, y] = ["x"]'], 'E1093:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001061 CheckDefExecFailure(['var x: number',
1062 'var y: number',
1063 'var z: list<number>',
Bram Moolenaar971caf52020-09-21 22:21:42 +02001064 '[x, y; z] = [1]'], 'E1093:')
1065
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001066 CheckDefFailure(['var somevar'], "E1022:")
1067 CheckDefFailure(['var &tabstop = 4'], 'E1052:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001068 CheckDefFailure(['&g:option = 5'], 'E113:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001069 CheckScriptFailure(['vim9script', 'var &tabstop = 4'], 'E1052:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001070
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001071 CheckDefFailure(['var $VAR = 5'], 'E1016: Cannot declare an environment variable:')
1072 CheckScriptFailure(['vim9script', 'var $ENV = "xxx"'], 'E1016:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001073
1074 if has('dnd')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001075 CheckDefFailure(['var @~ = 5'], 'E1066:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001076 else
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001077 CheckDefFailure(['var @~ = 5'], 'E354:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001078 CheckDefFailure(['@~ = 5'], 'E354:')
1079 endif
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001080 CheckDefFailure(['var @a = 5'], 'E1066:')
1081 CheckDefFailure(['var @/ = "x"'], 'E1066:')
1082 CheckScriptFailure(['vim9script', 'var @a = "abc"'], 'E1066:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001083
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001084 CheckDefFailure(['var g:var = 5'], 'E1016: Cannot declare a global variable:')
1085 CheckDefFailure(['var w:var = 5'], 'E1016: Cannot declare a window variable:')
1086 CheckDefFailure(['var b:var = 5'], 'E1016: Cannot declare a buffer variable:')
1087 CheckDefFailure(['var t:var = 5'], 'E1016: Cannot declare a tab variable:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001088
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001089 CheckDefFailure(['var anr = 4', 'anr ..= "text"'], 'E1019:')
1090 CheckDefFailure(['var xnr += 4'], 'E1020:', 1)
1091 CheckScriptFailure(['vim9script', 'var xnr += 4'], 'E1020:')
1092 CheckDefFailure(["var xnr = xnr + 1"], 'E1001:', 1)
1093 CheckScriptFailure(['vim9script', 'var xnr = xnr + 4'], 'E121:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001094
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001095 CheckScriptFailure(['vim9script', 'def Func()', 'var dummy = s:notfound', 'enddef', 'defcompile'], 'E1108:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001096
Bram Moolenaar98309382020-09-27 21:58:45 +02001097 CheckDefFailure(['var name: list<string> = [123]'], 'expected list<string> but got list<number>')
1098 CheckDefFailure(['var name: list<number> = ["xx"]'], 'expected list<number> but got list<string>')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001099
Bram Moolenaare0de1712020-12-02 17:36:54 +01001100 CheckDefFailure(['var name: dict<string> = {key: 123}'], 'expected dict<string> but got dict<number>')
1101 CheckDefFailure(['var name: dict<number> = {key: "xx"}'], 'expected dict<number> but got dict<string>')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001102
Bram Moolenaar98309382020-09-27 21:58:45 +02001103 CheckDefFailure(['var name = feedkeys("0")'], 'E1031:')
1104 CheckDefFailure(['var name: number = feedkeys("0")'], 'expected number but got void')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001105
Bram Moolenaar98309382020-09-27 21:58:45 +02001106 CheckDefFailure(['var name: dict <number>'], 'E1068:')
1107 CheckDefFailure(['var name: dict<number'], 'E1009:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001108
1109 assert_fails('s/^/\=Mess()/n', 'E794:')
Bram Moolenaar98309382020-09-27 21:58:45 +02001110 CheckDefFailure(['var name: dict<number'], 'E1009:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001111
1112 CheckDefFailure(['w:foo: number = 10'],
1113 'E488: Trailing characters: : number = 1')
1114 CheckDefFailure(['t:foo: bool = true'],
1115 'E488: Trailing characters: : bool = true')
1116 CheckDefFailure(['b:foo: string = "x"'],
1117 'E488: Trailing characters: : string = "x"')
1118 CheckDefFailure(['g:foo: number = 123'],
1119 'E488: Trailing characters: : number = 123')
1120enddef
1121
1122def Test_assign_list()
Bram Moolenaare65081d2021-06-27 15:04:05 +02001123 var lines =<< trim END
1124 var l: list<string> = []
1125 l[0] = 'value'
1126 assert_equal('value', l[0])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001127
Bram Moolenaare65081d2021-06-27 15:04:05 +02001128 l[1] = 'asdf'
1129 assert_equal('value', l[0])
1130 assert_equal('asdf', l[1])
1131 assert_equal('asdf', l[-1])
1132 assert_equal('value', l[-2])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001133
Bram Moolenaare65081d2021-06-27 15:04:05 +02001134 var nrl: list<number> = []
1135 for i in range(5)
1136 nrl[i] = i
1137 endfor
1138 assert_equal([0, 1, 2, 3, 4], nrl)
1139
1140 var ul: list<any>
1141 ul[0] = 1
1142 ul[1] = 2
1143 ul[2] = 3
1144 assert_equal([1, 2, 3], ul)
1145 END
1146 CheckDefAndScriptSuccess(lines)
Bram Moolenaar334a8b42020-10-19 16:07:42 +02001147
1148 CheckDefFailure(["var l: list<number> = ['', true]"], 'E1012: Type mismatch; expected list<number> but got list<any>', 1)
1149 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 +02001150enddef
1151
1152def Test_assign_dict()
Bram Moolenaarc3689572021-01-01 19:40:02 +01001153 var lines =<< trim END
1154 var d: dict<string> = {}
1155 d['key'] = 'value'
1156 assert_equal('value', d['key'])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001157
Bram Moolenaarc3689572021-01-01 19:40:02 +01001158 d[123] = 'qwerty'
1159 assert_equal('qwerty', d[123])
1160 assert_equal('qwerty', d['123'])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001161
Bram Moolenaarc3689572021-01-01 19:40:02 +01001162 var nrd: dict<number> = {}
1163 for i in range(3)
1164 nrd[i] = i
1165 endfor
1166 assert_equal({0: 0, 1: 1, 2: 2}, nrd)
1167
1168 d.somekey = 'someval'
1169 assert_equal({key: 'value', '123': 'qwerty', somekey: 'someval'}, d)
1170 # unlet d.somekey
1171 # assert_equal({key: 'value', '123': 'qwerty'}, d)
1172 END
1173 CheckDefAndScriptSuccess(lines)
1174
1175 # TODO: move to above once "unlet d.somekey" in :def is implemented
1176 lines =<< trim END
1177 vim9script
1178 var d: dict<string> = {}
1179 d['key'] = 'value'
1180 d.somekey = 'someval'
1181 assert_equal({key: 'value', somekey: 'someval'}, d)
1182 unlet d.somekey
1183 assert_equal({key: 'value'}, d)
1184 END
1185 CheckScriptSuccess(lines)
Bram Moolenaar334a8b42020-10-19 16:07:42 +02001186
Bram Moolenaare0de1712020-12-02 17:36:54 +01001187 CheckDefFailure(["var d: dict<number> = {a: '', b: true}"], 'E1012: Type mismatch; expected dict<number> but got dict<any>', 1)
1188 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 +02001189enddef
1190
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001191def Test_assign_dict_unknown_type()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001192 var lines =<< trim END
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001193 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001194 var mylist = []
Bram Moolenaare0de1712020-12-02 17:36:54 +01001195 mylist += [{one: 'one'}]
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001196 def Func()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001197 var dd = mylist[0]
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001198 assert_equal('one', dd.one)
1199 enddef
1200 Func()
1201 END
1202 CheckScriptSuccess(lines)
1203
Bram Moolenaarce2c5442020-11-28 21:21:17 +01001204 lines =<< trim END
1205 vim9script
1206 var mylist = [[]]
Bram Moolenaare0de1712020-12-02 17:36:54 +01001207 mylist[0] += [{one: 'one'}]
Bram Moolenaarce2c5442020-11-28 21:21:17 +01001208 def Func()
1209 var dd = mylist[0][0]
1210 assert_equal('one', dd.one)
1211 enddef
1212 Func()
1213 END
1214 CheckScriptSuccess(lines)
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001215enddef
1216
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001217def Test_assign_dict_with_op()
1218 var lines =<< trim END
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001219 var ds: dict<string> = {a: 'x'}
1220 ds['a'] ..= 'y'
1221 ds.a ..= 'z'
1222 assert_equal('xyz', ds.a)
1223
1224 var dn: dict<number> = {a: 9}
1225 dn['a'] += 2
1226 assert_equal(11, dn.a)
1227 dn.a += 2
1228 assert_equal(13, dn.a)
1229
1230 dn['a'] -= 3
1231 assert_equal(10, dn.a)
1232 dn.a -= 2
1233 assert_equal(8, dn.a)
1234
1235 dn['a'] *= 2
1236 assert_equal(16, dn.a)
1237 dn.a *= 2
1238 assert_equal(32, dn.a)
1239
1240 dn['a'] /= 3
1241 assert_equal(10, dn.a)
1242 dn.a /= 2
1243 assert_equal(5, dn.a)
1244
1245 dn['a'] %= 3
1246 assert_equal(2, dn.a)
1247 dn.a %= 6
1248 assert_equal(2, dn.a)
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001249
1250 var dd: dict<dict<list<any>>>
1251 dd.a = {}
1252 dd.a.b = [0]
1253 dd.a.b += [1]
1254 assert_equal({a: {b: [0, 1]}}, dd)
Bram Moolenaar0d351a02021-04-05 21:05:26 +02001255
1256 var dab = {a: ['b']}
1257 dab.a[0] ..= 'c'
1258 assert_equal({a: ['bc']}, dab)
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001259 END
Bram Moolenaare42939a2021-04-05 17:11:17 +02001260 CheckDefAndScriptSuccess(lines)
1261enddef
1262
1263def Test_assign_list_with_op()
1264 var lines =<< trim END
1265 var ls: list<string> = ['x']
1266 ls[0] ..= 'y'
1267 assert_equal('xy', ls[0])
1268
1269 var ln: list<number> = [9]
1270 ln[0] += 2
1271 assert_equal(11, ln[0])
1272
1273 ln[0] -= 3
1274 assert_equal(8, ln[0])
1275
1276 ln[0] *= 2
1277 assert_equal(16, ln[0])
1278
1279 ln[0] /= 3
1280 assert_equal(5, ln[0])
1281
1282 ln[0] %= 3
1283 assert_equal(2, ln[0])
1284 END
1285 CheckDefAndScriptSuccess(lines)
1286enddef
1287
1288def Test_assign_with_op_fails()
1289 var lines =<< trim END
1290 var s = 'abc'
1291 s[1] += 'x'
1292 END
1293 CheckDefAndScriptFailure2(lines, 'E1141:', 'E689:', 2)
1294
1295 lines =<< trim END
1296 var s = 'abc'
1297 s[1] ..= 'x'
1298 END
1299 CheckDefAndScriptFailure2(lines, 'E1141:', 'E689:', 2)
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001300
1301 lines =<< trim END
1302 var dd: dict<dict<list<any>>>
1303 dd.a = {}
1304 dd.a.b += [1]
1305 END
1306 CheckDefExecAndScriptFailure(lines, 'E716:', 3)
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001307enddef
1308
Bram Moolenaar971caf52020-09-21 22:21:42 +02001309def Test_assign_lambda()
1310 # check if assign a lambda to a variable which type is func or any.
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001311 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +02001312 vim9script
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001313 var FuncRef = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001314 assert_equal(123, FuncRef())
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001315 var FuncRef_Func: func = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001316 assert_equal(123, FuncRef_Func())
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001317 var FuncRef_Any: any = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001318 assert_equal(123, FuncRef_Any())
Bram Moolenaar06409502021-02-17 17:00:27 +01001319 var FuncRef_Number: func(): number = () => 321
1320 assert_equal(321, FuncRef_Number())
Bram Moolenaar971caf52020-09-21 22:21:42 +02001321 END
1322 CheckScriptSuccess(lines)
Bram Moolenaar328eac22021-01-07 19:23:08 +01001323
1324 lines =<< trim END
1325 var Ref: func(number)
1326 Ref = (j) => !j
1327 END
Bram Moolenaar06409502021-02-17 17:00:27 +01001328 CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected func(number) but got func(any): bool')
Bram Moolenaarc7dac852021-02-17 18:49:11 +01001329
1330 lines =<< trim END
1331 echo filter([1, 2, 3], (_, v: string) => v + 1)
1332 END
1333 CheckDefAndScriptFailure(lines, 'E1051:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001334enddef
1335
Bram Moolenaarc0e29012020-09-27 14:22:48 +02001336def Test_heredoc()
1337 var lines =<< trim END # comment
1338 text
1339 END
1340 assert_equal(['text'], lines)
1341
1342 CheckDefFailure(['var lines =<< trim END X', 'END'], 'E488:')
1343 CheckDefFailure(['var lines =<< trim END " comment', 'END'], 'E488:')
Bram Moolenaar2914a202020-09-27 18:24:03 +02001344
1345 lines =<< trim [END]
1346 def Func()
1347 var&lines =<< trim END
1348 x
1349 x
Bram Moolenaarb5b94802020-12-13 17:50:20 +01001350 enddef
1351 defcompile
1352 [END]
1353 CheckScriptFailure(lines, 'E1145: Missing heredoc end marker: END')
1354 delfunc! g:Func
1355
1356 lines =<< trim [END]
1357 def Func()
1358 var lines =<< trim END
1359 x
1360 x
Bram Moolenaar2914a202020-09-27 18:24:03 +02001361 x
1362 x
1363 x
1364 x
1365 x
1366 x
1367 enddef
1368 call Func()
1369 [END]
Bram Moolenaarb5b94802020-12-13 17:50:20 +01001370 CheckScriptFailure(lines, 'E1145: Missing heredoc end marker: END')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001371 delfunc! g:Func
Bram Moolenaarc0e29012020-09-27 14:22:48 +02001372enddef
1373
Bram Moolenaar083966f2020-12-08 11:29:40 +01001374def Test_var_func_call()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001375 var lines =<< trim END
1376 vim9script
1377 func GetValue()
1378 if exists('g:count')
1379 let g:count += 1
1380 else
1381 let g:count = 1
1382 endif
1383 return 'this'
1384 endfunc
1385 var val: string = GetValue()
1386 # env var is always a string
1387 var env = $TERM
1388 END
1389 writefile(lines, 'Xfinished')
1390 source Xfinished
1391 # GetValue() is not called during discovery phase
1392 assert_equal(1, g:count)
1393
1394 unlet g:count
1395 delete('Xfinished')
1396enddef
1397
Bram Moolenaar083966f2020-12-08 11:29:40 +01001398def Test_var_missing_type()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001399 var lines =<< trim END
1400 vim9script
1401 var name = g:unknown
1402 END
1403 CheckScriptFailure(lines, 'E121:')
1404
1405 lines =<< trim END
1406 vim9script
1407 var nr: number = 123
1408 var name = nr
1409 END
1410 CheckScriptSuccess(lines)
1411enddef
1412
Bram Moolenaar083966f2020-12-08 11:29:40 +01001413def Test_var_declaration()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001414 var lines =<< trim END
1415 vim9script
1416 var name: string
1417 g:var_uninit = name
1418 name = 'text'
1419 g:var_test = name
1420 # prefixing s: is optional
1421 s:name = 'prefixed'
1422 g:var_prefixed = s:name
1423
Bram Moolenaar7257af42020-12-08 22:08:51 +01001424 const FOO: number = 123
1425 assert_equal(123, FOO)
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001426 const FOOS = 'foos'
1427 assert_equal('foos', FOOS)
1428 final FLIST = [1]
1429 assert_equal([1], FLIST)
1430 FLIST[0] = 11
1431 assert_equal([11], FLIST)
1432
1433 const g:FOO: number = 321
1434 assert_equal(321, g:FOO)
1435 const g:FOOS = 'gfoos'
1436 assert_equal('gfoos', g:FOOS)
1437 final g:FLIST = [2]
1438 assert_equal([2], g:FLIST)
1439 g:FLIST[0] = 22
1440 assert_equal([22], g:FLIST)
1441
Bram Moolenaard877a572021-04-01 19:42:48 +02001442 def SetGlobalConst()
1443 const g:globConst = 123
1444 enddef
1445 SetGlobalConst()
1446 assert_equal(123, g:globConst)
1447 assert_true(islocked('g:globConst'))
1448
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001449 const w:FOO: number = 46
1450 assert_equal(46, w:FOO)
1451 const w:FOOS = 'wfoos'
1452 assert_equal('wfoos', w:FOOS)
1453 final w:FLIST = [3]
1454 assert_equal([3], w:FLIST)
1455 w:FLIST[0] = 33
1456 assert_equal([33], w:FLIST)
Bram Moolenaar7257af42020-12-08 22:08:51 +01001457
Bram Moolenaar55759b52020-09-30 22:59:42 +02001458 var s:other: number
1459 other = 1234
1460 g:other_var = other
1461
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01001462 var xyz: string # comment
1463
Bram Moolenaar55759b52020-09-30 22:59:42 +02001464 # type is inferred
Bram Moolenaar3862ea32021-01-01 21:05:55 +01001465 var s:dict = {['a']: 222}
Bram Moolenaar55759b52020-09-30 22:59:42 +02001466 def GetDictVal(key: any)
1467 g:dict_val = s:dict[key]
1468 enddef
1469 GetDictVal('a')
Bram Moolenaar08251752021-01-11 21:20:18 +01001470
1471 final adict: dict<string> = {}
1472 def ChangeAdict()
1473 adict.foo = 'foo'
1474 enddef
1475 ChangeAdict()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001476 END
1477 CheckScriptSuccess(lines)
1478 assert_equal('', g:var_uninit)
1479 assert_equal('text', g:var_test)
1480 assert_equal('prefixed', g:var_prefixed)
1481 assert_equal(1234, g:other_var)
1482 assert_equal(222, g:dict_val)
1483
1484 unlet g:var_uninit
1485 unlet g:var_test
1486 unlet g:var_prefixed
1487 unlet g:other_var
Bram Moolenaar6e50ec22021-04-03 19:32:44 +02001488 unlet g:globConst
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001489 unlet g:FOO
1490 unlet g:FOOS
1491 unlet g:FLIST
1492 unlet w:FOO
1493 unlet w:FOOS
1494 unlet w:FLIST
Bram Moolenaar55759b52020-09-30 22:59:42 +02001495enddef
1496
Bram Moolenaar083966f2020-12-08 11:29:40 +01001497def Test_var_declaration_fails()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001498 var lines =<< trim END
1499 vim9script
1500 final var: string
1501 END
1502 CheckScriptFailure(lines, 'E1125:')
1503
1504 lines =<< trim END
1505 vim9script
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001506 const g:constvar = 'string'
1507 g:constvar = 'xx'
1508 END
1509 CheckScriptFailure(lines, 'E741:')
1510 unlet g:constvar
1511
1512 lines =<< trim END
1513 vim9script
Bram Moolenaarf5906aa2021-04-02 14:35:15 +02001514 var name = 'one'
1515 lockvar name
1516 def SetLocked()
1517 name = 'two'
1518 enddef
1519 SetLocked()
1520 END
Bram Moolenaardcf29ac2021-04-02 14:44:02 +02001521 CheckScriptFailure(lines, 'E741: Value is locked: name', 1)
1522
1523 lines =<< trim END
1524 let s:legacy = 'one'
1525 lockvar s:legacy
1526 def SetLocked()
1527 s:legacy = 'two'
1528 enddef
1529 call SetLocked()
1530 END
1531 CheckScriptFailure(lines, 'E741: Value is locked: s:legacy', 1)
Bram Moolenaarf5906aa2021-04-02 14:35:15 +02001532
1533 lines =<< trim END
1534 vim9script
Bram Moolenaard877a572021-04-01 19:42:48 +02001535 def SetGlobalConst()
1536 const g:globConst = 123
1537 enddef
1538 SetGlobalConst()
1539 g:globConst = 234
1540 END
Bram Moolenaar6e50ec22021-04-03 19:32:44 +02001541 CheckScriptFailure(lines, 'E741: Value is locked: g:globConst', 6)
Bram Moolenaard877a572021-04-01 19:42:48 +02001542 unlet g:globConst
1543
1544 lines =<< trim END
1545 vim9script
Bram Moolenaar08251752021-01-11 21:20:18 +01001546 const cdict: dict<string> = {}
1547 def Change()
1548 cdict.foo = 'foo'
1549 enddef
1550 defcompile
1551 END
1552 CheckScriptFailure(lines, 'E46:')
1553
1554 lines =<< trim END
1555 vim9script
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001556 final w:finalvar = [9]
1557 w:finalvar = [8]
1558 END
1559 CheckScriptFailure(lines, 'E1122:')
1560 unlet w:finalvar
1561
1562 lines =<< trim END
1563 vim9script
Bram Moolenaar55759b52020-09-30 22:59:42 +02001564 const var: string
1565 END
1566 CheckScriptFailure(lines, 'E1021:')
1567
1568 lines =<< trim END
1569 vim9script
1570 var 9var: string
1571 END
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01001572 CheckScriptFailure(lines, 'E488:')
Bram Moolenaar7257af42020-12-08 22:08:51 +01001573
1574 CheckDefFailure(['var foo.bar = 2'], 'E1087:')
1575 CheckDefFailure(['var foo[3] = 2'], 'E1087:')
1576 CheckDefFailure(['const foo: number'], 'E1021:')
Bram Moolenaar55759b52020-09-30 22:59:42 +02001577enddef
1578
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001579def Test_script_local_in_legacy()
1580 # OK to define script-local later when prefixed with s:
1581 var lines =<< trim END
1582 def SetLater()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001583 s:legvar = 'two'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001584 enddef
1585 defcompile
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001586 let s:legvar = 'one'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001587 call SetLater()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001588 call assert_equal('two', s:legvar)
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001589 END
1590 CheckScriptSuccess(lines)
1591
1592 # OK to leave out s: prefix when script-local already defined
1593 lines =<< trim END
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001594 let s:legvar = 'one'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001595 def SetNoPrefix()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001596 legvar = 'two'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001597 enddef
1598 call SetNoPrefix()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001599 call assert_equal('two', s:legvar)
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001600 END
1601 CheckScriptSuccess(lines)
1602
1603 # Not OK to leave out s: prefix when script-local defined later
1604 lines =<< trim END
1605 def SetLaterNoPrefix()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001606 legvar = 'two'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001607 enddef
1608 defcompile
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001609 let s:legvar = 'one'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001610 END
1611 CheckScriptFailure(lines, 'E476:', 1)
1612enddef
1613
Bram Moolenaar083966f2020-12-08 11:29:40 +01001614def Test_var_type_check()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001615 var lines =<< trim END
1616 vim9script
1617 var name: string
1618 name = 1234
1619 END
1620 CheckScriptFailure(lines, 'E1012:')
1621
1622 lines =<< trim END
1623 vim9script
1624 var name:string
1625 END
1626 CheckScriptFailure(lines, 'E1069:')
1627
1628 lines =<< trim END
1629 vim9script
1630 var name: asdf
1631 END
1632 CheckScriptFailure(lines, 'E1010:')
1633
1634 lines =<< trim END
1635 vim9script
1636 var s:l: list<number>
1637 s:l = []
1638 END
1639 CheckScriptSuccess(lines)
1640
1641 lines =<< trim END
1642 vim9script
1643 var s:d: dict<number>
1644 s:d = {}
1645 END
1646 CheckScriptSuccess(lines)
1647enddef
1648
1649let g:dict_number = #{one: 1, two: 2}
1650
Bram Moolenaar083966f2020-12-08 11:29:40 +01001651def Test_var_list_dict_type()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001652 var ll: list<number>
1653 ll = [1, 2, 2, 3, 3, 3]->uniq()
1654 ll->assert_equal([1, 2, 3])
1655
1656 var dd: dict<number>
1657 dd = g:dict_number
1658 dd->assert_equal(g:dict_number)
1659
1660 var lines =<< trim END
1661 var ll: list<number>
1662 ll = [1, 2, 3]->map('"one"')
1663 END
Bram Moolenaar70250fb2021-01-16 19:01:53 +01001664 CheckDefExecFailure(lines, 'E1012: Type mismatch; expected number but got string')
Bram Moolenaar55759b52020-09-30 22:59:42 +02001665enddef
1666
Bram Moolenaar083966f2020-12-08 11:29:40 +01001667def Test_cannot_use_let()
1668 CheckDefAndScriptFailure(['let a = 34'], 'E1126:', 1)
1669enddef
1670
Bram Moolenaar55759b52020-09-30 22:59:42 +02001671def Test_unlet()
1672 g:somevar = 'yes'
1673 assert_true(exists('g:somevar'))
1674 unlet g:somevar
1675 assert_false(exists('g:somevar'))
1676 unlet! g:somevar
1677
1678 # also works for script-local variable in legacy Vim script
1679 s:somevar = 'legacy'
1680 assert_true(exists('s:somevar'))
1681 unlet s:somevar
1682 assert_false(exists('s:somevar'))
1683 unlet! s:somevar
1684
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001685 CheckDefExecFailure([
1686 'var dd = 111',
1687 'unlet dd',
1688 ], 'E1081:', 2)
1689
Bram Moolenaar752fc692021-01-04 21:57:11 +01001690 # dict unlet
1691 var dd = {a: 1, b: 2, c: 3}
1692 unlet dd['a']
1693 unlet dd.c
1694 assert_equal({b: 2}, dd)
1695
1696 # list unlet
1697 var ll = [1, 2, 3, 4]
1698 unlet ll[1]
1699 unlet ll[-1]
1700 assert_equal([1, 3], ll)
1701
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001702 ll = [1, 2, 3, 4]
1703 unlet ll[0 : 1]
1704 assert_equal([3, 4], ll)
1705
1706 ll = [1, 2, 3, 4]
1707 unlet ll[2 : 8]
1708 assert_equal([1, 2], ll)
1709
1710 ll = [1, 2, 3, 4]
1711 unlet ll[-2 : -1]
1712 assert_equal([1, 2], ll)
1713
1714 CheckDefFailure([
1715 'var ll = [1, 2]',
1716 'll[1 : 2] = 7',
1717 ], 'E1165:', 2)
1718 CheckDefFailure([
1719 'var dd = {a: 1}',
1720 'unlet dd["a" : "a"]',
1721 ], 'E1166:', 2)
1722 CheckDefExecFailure([
1723 'unlet g:adict[0 : 1]',
1724 ], 'E1148:', 1)
1725 CheckDefFailure([
1726 'var ll = [1, 2]',
1727 'unlet ll[0:1]',
1728 ], 'E1004:', 2)
1729 CheckDefFailure([
1730 'var ll = [1, 2]',
1731 'unlet ll[0 :1]',
1732 ], 'E1004:', 2)
1733 CheckDefFailure([
1734 'var ll = [1, 2]',
1735 'unlet ll[0: 1]',
1736 ], 'E1004:', 2)
1737
1738 CheckDefFailure([
1739 'var ll = [1, 2]',
1740 'unlet ll["x" : 1]',
1741 ], 'E1012:', 2)
1742 CheckDefFailure([
1743 'var ll = [1, 2]',
1744 'unlet ll[0 : "x"]',
1745 ], 'E1012:', 2)
1746
Bram Moolenaar752fc692021-01-04 21:57:11 +01001747 # list of dict unlet
1748 var dl = [{a: 1, b: 2}, {c: 3}]
1749 unlet dl[0]['b']
1750 assert_equal([{a: 1}, {c: 3}], dl)
1751
1752 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001753 'var ll = test_null_list()',
1754 'unlet ll[0]',
1755 ], 'E684:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001756 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001757 'var ll = [1]',
1758 'unlet ll[2]',
1759 ], 'E684:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001760 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001761 'var ll = [1]',
1762 'unlet ll[g:astring]',
Bram Moolenaarf30a14d2021-01-17 21:51:24 +01001763 ], 'E1012:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001764 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001765 'var dd = test_null_dict()',
1766 'unlet dd["a"]',
1767 ], 'E716:', 2)
1768 CheckDefExecFailure([
1769 'var dd = {a: 1}',
1770 'unlet dd["b"]',
1771 ], 'E716:', 2)
1772 CheckDefExecFailure([
1773 'var dd = {a: 1}',
1774 'unlet dd[g:alist]',
1775 ], 'E1105:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001776
Bram Moolenaar2ef951d2021-01-03 20:55:26 +01001777 # can compile unlet before variable exists
Bram Moolenaar752fc692021-01-04 21:57:11 +01001778 g:someDict = {key: 'val'}
1779 var k = 'key'
1780 unlet g:someDict[k]
1781 assert_equal({}, g:someDict)
1782 unlet g:someDict
1783 assert_false(exists('g:someDict'))
Bram Moolenaar2ef951d2021-01-03 20:55:26 +01001784
Bram Moolenaar55759b52020-09-30 22:59:42 +02001785 CheckScriptFailure([
1786 'vim9script',
1787 'var svar = 123',
1788 'unlet svar',
1789 ], 'E1081:')
1790 CheckScriptFailure([
1791 'vim9script',
1792 'var svar = 123',
1793 'unlet s:svar',
1794 ], 'E1081:')
1795 CheckScriptFailure([
1796 'vim9script',
1797 'var svar = 123',
1798 'def Func()',
1799 ' unlet svar',
1800 'enddef',
1801 'defcompile',
1802 ], 'E1081:')
1803 CheckScriptFailure([
1804 'vim9script',
1805 'var svar = 123',
Bram Moolenaar9aed7292020-12-18 15:38:00 +01001806 'func Func()',
1807 ' unlet s:svar',
1808 'endfunc',
1809 'Func()',
1810 ], 'E1081:')
1811 CheckScriptFailure([
1812 'vim9script',
1813 'var svar = 123',
Bram Moolenaar55759b52020-09-30 22:59:42 +02001814 'def Func()',
1815 ' unlet s:svar',
1816 'enddef',
1817 'defcompile',
1818 ], 'E1081:')
1819
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001820 writefile(['vim9script', 'export var svar = 1234'], 'XunletExport.vim')
1821 var lines =<< trim END
1822 vim9script
1823 import svar from './XunletExport.vim'
1824 def UnletSvar()
1825 unlet svar
1826 enddef
1827 defcompile
1828 END
1829 CheckScriptFailure(lines, 'E1081:', 1)
1830 delete('XunletExport.vim')
1831
Bram Moolenaar55759b52020-09-30 22:59:42 +02001832 $ENVVAR = 'foobar'
1833 assert_equal('foobar', $ENVVAR)
1834 unlet $ENVVAR
1835 assert_equal('', $ENVVAR)
1836enddef
1837
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01001838def Test_expr_error_no_assign()
1839 var lines =<< trim END
1840 vim9script
1841 var x = invalid
1842 echo x
1843 END
1844 CheckScriptFailureList(lines, ['E121:', 'E121:'])
1845
1846 lines =<< trim END
1847 vim9script
1848 var x = 1 / 0
1849 echo x
1850 END
Bram Moolenaar227c58a2021-04-28 20:40:44 +02001851 CheckScriptFailure(lines, 'E1154:')
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01001852
1853 lines =<< trim END
1854 vim9script
1855 var x = 1 % 0
1856 echo x
1857 END
Bram Moolenaar227c58a2021-04-28 20:40:44 +02001858 CheckScriptFailure(lines, 'E1154:')
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01001859
1860 lines =<< trim END
1861 var x: string 'string'
1862 END
1863 CheckDefAndScriptFailure(lines, 'E488:')
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01001864enddef
1865
1866
Bram Moolenaar17126b12021-01-07 22:03:02 +01001867def Test_assign_command_modifier()
1868 var lines =<< trim END
Bram Moolenaar9e0f8832021-01-09 12:09:22 +01001869 var verbose = 0
1870 verbose = 1
1871 assert_equal(1, verbose)
1872 silent verbose = 2
1873 assert_equal(2, verbose)
1874 silent verbose += 2
1875 assert_equal(4, verbose)
1876 silent verbose -= 1
1877 assert_equal(3, verbose)
Bram Moolenaar17126b12021-01-07 22:03:02 +01001878
Bram Moolenaar9e0f8832021-01-09 12:09:22 +01001879 var topleft = {one: 1}
1880 sandbox topleft.one = 3
1881 assert_equal({one: 3}, topleft)
1882 leftabove topleft[' '] = 4
1883 assert_equal({one: 3, ' ': 4}, topleft)
1884
1885 var x: number
1886 var y: number
1887 silent [x, y] = [1, 2]
1888 assert_equal(1, x)
1889 assert_equal(2, y)
Bram Moolenaar17126b12021-01-07 22:03:02 +01001890 END
1891 CheckDefAndScriptSuccess(lines)
1892enddef
1893
Bram Moolenaar74f4a962021-06-17 21:03:07 +02001894def Test_assign_alt_buf_register()
1895 var lines =<< trim END
1896 edit 'file_b1'
1897 var b1 = bufnr()
1898 edit 'file_b2'
1899 var b2 = bufnr()
1900 assert_equal(b1, bufnr('#'))
1901 @# = b2
1902 assert_equal(b2, bufnr('#'))
1903 END
1904 CheckDefAndScriptSuccess(lines)
1905enddef
1906
Bram Moolenaar32154662021-03-28 21:14:06 +02001907def Test_script_funcref_case()
1908 var lines =<< trim END
1909 var Len = (s: string): number => len(s) + 1
1910 assert_equal(5, Len('asdf'))
1911 END
1912 CheckDefAndScriptSuccess(lines)
1913
1914 lines =<< trim END
1915 var len = (s: string): number => len(s) + 1
1916 END
1917 CheckDefAndScriptFailure(lines, 'E704:')
1918
1919 lines =<< trim END
1920 vim9script
1921 var s:Len = (s: string): number => len(s) + 2
1922 assert_equal(6, Len('asdf'))
1923 END
1924 CheckScriptSuccess(lines)
1925
1926 lines =<< trim END
1927 vim9script
1928 var s:len = (s: string): number => len(s) + 1
1929 END
1930 CheckScriptFailure(lines, 'E704:')
1931enddef
1932
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001933def Test_inc_dec()
1934 var lines =<< trim END
1935 var nr = 7
1936 ++nr
Bram Moolenaar730bf302021-04-24 20:43:56 +02001937 assert_equal(8, nr)
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001938 --nr
Bram Moolenaar730bf302021-04-24 20:43:56 +02001939 assert_equal(7, nr)
Bram Moolenaarf3d30842021-06-25 19:29:30 +02001940 ++nr | ++nr
1941 assert_equal(9, nr)
1942 ++nr # comment
1943 assert_equal(10, nr)
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001944
1945 var ll = [1, 2]
1946 --ll[0]
1947 ++ll[1]
Bram Moolenaar730bf302021-04-24 20:43:56 +02001948 assert_equal([0, 3], ll)
1949
1950 g:count = 1
1951 ++g:count
1952 --g:count
1953 assert_equal(1, g:count)
1954 unlet g:count
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001955 END
1956 CheckDefAndScriptSuccess(lines)
Bram Moolenaar22480d12021-06-25 21:31:09 +02001957
1958 lines =<< trim END
1959 var nr = 7
1960 ++ nr
1961 END
1962 CheckDefAndScriptFailure(lines, "E1202: No white space allowed after '++': ++ nr")
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001963enddef
1964
Bram Moolenaar55759b52020-09-30 22:59:42 +02001965
Bram Moolenaar971caf52020-09-21 22:21:42 +02001966" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker