blob: 92ffa0055c2b55c7e6967fce9847b5c25c0b2b96 [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 Moolenaarf30a14d2021-01-17 21:51:24 +0100664def Test_assignment_list_any_index()
665 var l: list<number> = [1, 2]
666 for [x, y, _]
667 in [[0, 1, ''], [1, 3, '']]
668 l[x] = l[x] + y
669 endfor
670 assert_equal([2, 5], l)
671enddef
672
Bram Moolenaar971caf52020-09-21 22:21:42 +0200673def Test_assignment_list_vim9script()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200674 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200675 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200676 var v1: number
677 var v2: number
678 var v3: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200679 [v1, v2, v3] = [1, 2, 3]
680 assert_equal([1, 2, 3], [v1, v2, v3])
681 END
682 CheckScriptSuccess(lines)
683enddef
684
685def Test_assignment_dict()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100686 var dict1: dict<bool> = {one: false, two: true}
687 var dict2: dict<number> = {one: 1, two: 2}
688 var dict3: dict<string> = {key: 'value'}
689 var dict4: dict<any> = {one: 1, two: '2'}
690 var dict5: dict<blob> = {one: 0z01, two: 0z02}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200691
692 # overwrite
693 dict3['key'] = 'another'
Bram Moolenaare0de1712020-12-02 17:36:54 +0100694 assert_equal(dict3, {key: 'another'})
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100695 dict3.key = 'yet another'
Bram Moolenaare0de1712020-12-02 17:36:54 +0100696 assert_equal(dict3, {key: 'yet another'})
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100697
Bram Moolenaard24602f2020-12-20 15:20:56 +0100698 # member "any" can also be a dict and assigned to
699 var anydict: dict<any> = {nest: {}, nr: 0}
700 anydict.nest['this'] = 123
701 anydict.nest.that = 456
702 assert_equal({nest: {this: 123, that: 456}, nr: 0}, anydict)
703
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100704 var lines =<< trim END
Bram Moolenaard88dc4d2020-12-20 14:49:42 +0100705 var dd = {}
706 dd.two = 2
707 assert_equal({two: 2}, dd)
708 END
Bram Moolenaar4f5e3972020-12-21 17:30:50 +0100709 CheckDefAndScriptSuccess(lines)
710
711 lines =<< trim END
712 var d = {dd: {}}
713 d.dd[0] = 2
714 d.dd['x'] = 3
715 d.dd.y = 4
716 assert_equal({dd: {0: 2, x: 3, y: 4}}, d)
717 END
718 CheckDefAndScriptSuccess(lines)
Bram Moolenaard88dc4d2020-12-20 14:49:42 +0100719
720 lines =<< trim END
Bram Moolenaare0de1712020-12-02 17:36:54 +0100721 var dd = {one: 1}
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100722 dd.one) = 2
723 END
Bram Moolenaard88dc4d2020-12-20 14:49:42 +0100724 CheckDefFailure(lines, 'E488:', 2)
725
726 lines =<< trim END
727 var dd = {one: 1}
728 var dd.one = 2
729 END
730 CheckDefAndScriptFailure(lines, 'E1017:', 2)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200731
732 # empty key can be used
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200733 var dd = {}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200734 dd[""] = 6
Bram Moolenaare0de1712020-12-02 17:36:54 +0100735 assert_equal({['']: 6}, dd)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200736
737 # type becomes dict<any>
Bram Moolenaare0de1712020-12-02 17:36:54 +0100738 var somedict = rand() > 0 ? {a: 1, b: 2} : {a: 'a', b: 'b'}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200739
740 # assignment to script-local dict
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100741 lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200742 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200743 var test: dict<any> = {}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200744 def FillDict(): dict<any>
745 test['a'] = 43
746 return test
747 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +0100748 assert_equal({a: 43}, FillDict())
Bram Moolenaar971caf52020-09-21 22:21:42 +0200749 END
750 CheckScriptSuccess(lines)
751
752 lines =<< trim END
753 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200754 var test: dict<any>
Bram Moolenaar971caf52020-09-21 22:21:42 +0200755 def FillDict(): dict<any>
756 test['a'] = 43
757 return test
758 enddef
759 FillDict()
Bram Moolenaar3beaf9c2020-12-18 17:23:14 +0100760 assert_equal({a: 43}, test)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200761 END
Bram Moolenaar3beaf9c2020-12-18 17:23:14 +0100762 CheckScriptSuccess(lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200763
764 # assignment to global dict
765 lines =<< trim END
766 vim9script
767 g:test = {}
768 def FillDict(): dict<any>
769 g:test['a'] = 43
770 return g:test
771 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +0100772 assert_equal({a: 43}, FillDict())
Bram Moolenaar971caf52020-09-21 22:21:42 +0200773 END
774 CheckScriptSuccess(lines)
775
776 # assignment to buffer dict
777 lines =<< trim END
778 vim9script
779 b:test = {}
780 def FillDict(): dict<any>
781 b:test['a'] = 43
782 return b:test
783 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +0100784 assert_equal({a: 43}, FillDict())
Bram Moolenaar971caf52020-09-21 22:21:42 +0200785 END
786 CheckScriptSuccess(lines)
Bram Moolenaar4f5e3972020-12-21 17:30:50 +0100787
788 lines =<< trim END
789 var d = {dd: test_null_dict()}
790 d.dd[0] = 0
791 END
792 CheckDefExecFailure(lines, 'E1103:', 2)
793
794 lines =<< trim END
795 var d = {dd: 'string'}
796 d.dd[0] = 0
797 END
798 CheckDefExecFailure(lines, 'E1148:', 2)
Bram Moolenaar3a3b10e2021-06-26 15:00:59 +0200799
800 lines =<< trim END
801 var n: any
802 n.key = 5
803 END
804 CheckDefExecAndScriptFailure2(lines, 'E1148:', 'E1203: Dot can only be used on a dictionary: n.key = 5', 2)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200805enddef
806
807def Test_assignment_local()
808 # Test in a separated file in order not to the current buffer/window/tab is
809 # changed.
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200810 var script_lines: list<string> =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200811 let b:existing = 'yes'
812 let w:existing = 'yes'
813 let t:existing = 'yes'
814
815 def Test_assignment_local_internal()
816 b:newvar = 'new'
817 assert_equal('new', b:newvar)
818 assert_equal('yes', b:existing)
819 b:existing = 'no'
820 assert_equal('no', b:existing)
821 b:existing ..= 'NO'
822 assert_equal('noNO', b:existing)
823
824 w:newvar = 'new'
825 assert_equal('new', w:newvar)
826 assert_equal('yes', w:existing)
827 w:existing = 'no'
828 assert_equal('no', w:existing)
829 w:existing ..= 'NO'
830 assert_equal('noNO', w:existing)
831
832 t:newvar = 'new'
833 assert_equal('new', t:newvar)
834 assert_equal('yes', t:existing)
835 t:existing = 'no'
836 assert_equal('no', t:existing)
837 t:existing ..= 'NO'
838 assert_equal('noNO', t:existing)
839 enddef
840 call Test_assignment_local_internal()
841 END
842 CheckScriptSuccess(script_lines)
843enddef
844
845def Test_assignment_default()
Bram Moolenaar971caf52020-09-21 22:21:42 +0200846 # Test default values.
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200847 var thebool: bool
Bram Moolenaar971caf52020-09-21 22:21:42 +0200848 assert_equal(v:false, thebool)
849
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200850 var thenumber: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200851 assert_equal(0, thenumber)
852
853 if has('float')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200854 var thefloat: float
Bram Moolenaar971caf52020-09-21 22:21:42 +0200855 assert_equal(0.0, thefloat)
856 endif
857
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200858 var thestring: string
Bram Moolenaar971caf52020-09-21 22:21:42 +0200859 assert_equal('', thestring)
860
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200861 var theblob: blob
Bram Moolenaar971caf52020-09-21 22:21:42 +0200862 assert_equal(0z, theblob)
863
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200864 var Thefunc: func
Bram Moolenaar971caf52020-09-21 22:21:42 +0200865 assert_equal(test_null_function(), Thefunc)
866
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200867 var thelist: list<any>
Bram Moolenaar971caf52020-09-21 22:21:42 +0200868 assert_equal([], thelist)
869
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200870 var thedict: dict<any>
Bram Moolenaar971caf52020-09-21 22:21:42 +0200871 assert_equal({}, thedict)
872
873 if has('channel')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200874 var thejob: job
Bram Moolenaar971caf52020-09-21 22:21:42 +0200875 assert_equal(test_null_job(), thejob)
876
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200877 var thechannel: channel
Bram Moolenaar971caf52020-09-21 22:21:42 +0200878 assert_equal(test_null_channel(), thechannel)
879
880 if has('unix') && executable('cat')
881 # check with non-null job and channel, types must match
Bram Moolenaare0de1712020-12-02 17:36:54 +0100882 thejob = job_start("cat ", {})
Bram Moolenaar971caf52020-09-21 22:21:42 +0200883 thechannel = job_getchannel(thejob)
884 job_stop(thejob, 'kill')
885 endif
886 endif
887
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200888 var nr = 1234 | nr = 5678
Bram Moolenaar971caf52020-09-21 22:21:42 +0200889 assert_equal(5678, nr)
890enddef
891
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100892let scriptvar = 'init'
893
Bram Moolenaar971caf52020-09-21 22:21:42 +0200894def Test_assignment_var_list()
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100895 var lines =<< trim END
896 var v1: string
897 var v2: string
898 var vrem: list<string>
899 [v1] = ['aaa']
900 assert_equal('aaa', v1)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200901
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100902 [v1, v2] = ['one', 'two']
903 assert_equal('one', v1)
904 assert_equal('two', v2)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200905
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100906 [v1, v2; vrem] = ['one', 'two']
907 assert_equal('one', v1)
908 assert_equal('two', v2)
909 assert_equal([], vrem)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200910
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100911 [v1, v2; vrem] = ['one', 'two', 'three']
912 assert_equal('one', v1)
913 assert_equal('two', v2)
914 assert_equal(['three'], vrem)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200915
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100916 [&ts, &sw] = [3, 4]
917 assert_equal(3, &ts)
918 assert_equal(4, &sw)
919 set ts=8 sw=4
Bram Moolenaar004d9b02020-11-30 21:40:03 +0100920
921 [@a, @z] = ['aa', 'zz']
922 assert_equal('aa', @a)
923 assert_equal('zz', @z)
924
925 [$SOME_VAR, $OTHER_VAR] = ['some', 'other']
926 assert_equal('some', $SOME_VAR)
927 assert_equal('other', $OTHER_VAR)
928
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100929 [g:globalvar, b:bufvar, w:winvar, t:tabvar, v:errmsg] =
930 ['global', 'buf', 'win', 'tab', 'error']
Bram Moolenaar004d9b02020-11-30 21:40:03 +0100931 assert_equal('global', g:globalvar)
Bram Moolenaar004d9b02020-11-30 21:40:03 +0100932 assert_equal('buf', b:bufvar)
933 assert_equal('win', w:winvar)
934 assert_equal('tab', t:tabvar)
935 assert_equal('error', v:errmsg)
936 unlet g:globalvar
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100937 END
938 CheckDefAndScriptSuccess(lines)
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100939
940 [g:globalvar, s:scriptvar, b:bufvar] = ['global', 'script', 'buf']
941 assert_equal('global', g:globalvar)
942 assert_equal('script', s:scriptvar)
943 assert_equal('buf', b:bufvar)
944
945 lines =<< trim END
946 vim9script
947 var s:scriptvar = 'init'
948 [g:globalvar, s:scriptvar, w:winvar] = ['global', 'script', 'win']
949 assert_equal('global', g:globalvar)
950 assert_equal('script', s:scriptvar)
951 assert_equal('win', w:winvar)
952 END
953 CheckScriptSuccess(lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200954enddef
955
956def Test_assignment_vim9script()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200957 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200958 vim9script
959 def Func(): list<number>
960 return [1, 2]
961 enddef
Bram Moolenaar98309382020-09-27 21:58:45 +0200962 var name1: number
963 var name2: number
964 [name1, name2] =
Bram Moolenaar971caf52020-09-21 22:21:42 +0200965 Func()
Bram Moolenaar98309382020-09-27 21:58:45 +0200966 assert_equal(1, name1)
967 assert_equal(2, name2)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200968 var ll =
Bram Moolenaar971caf52020-09-21 22:21:42 +0200969 Func()
970 assert_equal([1, 2], ll)
971
972 @/ = 'text'
973 assert_equal('text', @/)
974 @0 = 'zero'
975 assert_equal('zero', @0)
976 @1 = 'one'
977 assert_equal('one', @1)
978 @9 = 'nine'
979 assert_equal('nine', @9)
980 @- = 'minus'
981 assert_equal('minus', @-)
982 if has('clipboard_working')
983 @* = 'star'
984 assert_equal('star', @*)
985 @+ = 'plus'
986 assert_equal('plus', @+)
987 endif
988
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200989 var a: number = 123
Bram Moolenaar971caf52020-09-21 22:21:42 +0200990 assert_equal(123, a)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200991 var s: string = 'yes'
Bram Moolenaar971caf52020-09-21 22:21:42 +0200992 assert_equal('yes', s)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200993 var b: number = 42
Bram Moolenaar971caf52020-09-21 22:21:42 +0200994 assert_equal(42, b)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200995 var w: number = 43
Bram Moolenaar971caf52020-09-21 22:21:42 +0200996 assert_equal(43, w)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200997 var t: number = 44
Bram Moolenaar971caf52020-09-21 22:21:42 +0200998 assert_equal(44, t)
Bram Moolenaar7a66a172020-10-16 19:56:12 +0200999
1000 var to_var = 0
1001 to_var = 3
1002 assert_equal(3, to_var)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001003 END
1004 CheckScriptSuccess(lines)
Bram Moolenaar06f9c692020-09-27 21:27:40 +02001005
1006 lines =<< trim END
1007 vim9script
1008 var n: number
1009 def Func()
1010 n = 'string'
1011 enddef
1012 defcompile
1013 END
1014 CheckScriptFailure(lines, 'E1012: Type mismatch; expected number but got string')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001015enddef
1016
1017def Mess(): string
1018 v:foldstart = 123
1019 return 'xxx'
1020enddef
1021
1022def Test_assignment_failure()
Bram Moolenaar98309382020-09-27 21:58:45 +02001023 CheckDefFailure(['var name=234'], 'E1004:')
1024 CheckDefFailure(['var name =234'], 'E1004:')
1025 CheckDefFailure(['var name= 234'], 'E1004:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001026
Bram Moolenaar98309382020-09-27 21:58:45 +02001027 CheckScriptFailure(['vim9script', 'var name=234'], 'E1004:')
1028 CheckScriptFailure(['vim9script', 'var name=234'], "before and after '='")
1029 CheckScriptFailure(['vim9script', 'var name =234'], 'E1004:')
1030 CheckScriptFailure(['vim9script', 'var name= 234'], 'E1004:')
1031 CheckScriptFailure(['vim9script', 'var name = 234', 'name+=234'], 'E1004:')
1032 CheckScriptFailure(['vim9script', 'var name = 234', 'name+=234'], "before and after '+='")
1033 CheckScriptFailure(['vim9script', 'var name = "x"', 'name..="y"'], 'E1004:')
1034 CheckScriptFailure(['vim9script', 'var name = "x"', 'name..="y"'], "before and after '..='")
Bram Moolenaar971caf52020-09-21 22:21:42 +02001035
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001036 CheckDefFailure(['var true = 1'], 'E1034:')
1037 CheckDefFailure(['var false = 1'], 'E1034:')
Bram Moolenaar74235772021-06-12 14:53:05 +02001038 CheckDefFailure(['var null = 1'], 'E1034:')
1039 CheckDefFailure(['var this = 1'], 'E1034:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001040
1041 CheckDefFailure(['[a; b; c] = g:list'], 'E452:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001042 CheckDefExecFailure(['var a: number',
Bram Moolenaar971caf52020-09-21 22:21:42 +02001043 '[a] = test_null_list()'], 'E1093:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001044 CheckDefExecFailure(['var a: number',
Bram Moolenaar971caf52020-09-21 22:21:42 +02001045 '[a] = []'], 'E1093:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001046 CheckDefExecFailure(['var x: number',
1047 'var y: number',
Bram Moolenaar971caf52020-09-21 22:21:42 +02001048 '[x, y] = [1]'], 'E1093:')
Bram Moolenaar352134b2020-10-17 22:04:08 +02001049 CheckDefExecFailure(['var x: string',
1050 'var y: string',
1051 '[x, y] = ["x"]'], 'E1093:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001052 CheckDefExecFailure(['var x: number',
1053 'var y: number',
1054 'var z: list<number>',
Bram Moolenaar971caf52020-09-21 22:21:42 +02001055 '[x, y; z] = [1]'], 'E1093:')
1056
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001057 CheckDefFailure(['var somevar'], "E1022:")
1058 CheckDefFailure(['var &tabstop = 4'], 'E1052:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001059 CheckDefFailure(['&g:option = 5'], 'E113:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001060 CheckScriptFailure(['vim9script', 'var &tabstop = 4'], 'E1052:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001061
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001062 CheckDefFailure(['var $VAR = 5'], 'E1016: Cannot declare an environment variable:')
1063 CheckScriptFailure(['vim9script', 'var $ENV = "xxx"'], 'E1016:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001064
1065 if has('dnd')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001066 CheckDefFailure(['var @~ = 5'], 'E1066:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001067 else
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001068 CheckDefFailure(['var @~ = 5'], 'E354:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001069 CheckDefFailure(['@~ = 5'], 'E354:')
1070 endif
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001071 CheckDefFailure(['var @a = 5'], 'E1066:')
1072 CheckDefFailure(['var @/ = "x"'], 'E1066:')
1073 CheckScriptFailure(['vim9script', 'var @a = "abc"'], 'E1066:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001074
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001075 CheckDefFailure(['var g:var = 5'], 'E1016: Cannot declare a global variable:')
1076 CheckDefFailure(['var w:var = 5'], 'E1016: Cannot declare a window variable:')
1077 CheckDefFailure(['var b:var = 5'], 'E1016: Cannot declare a buffer variable:')
1078 CheckDefFailure(['var t:var = 5'], 'E1016: Cannot declare a tab variable:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001079
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001080 CheckDefFailure(['var anr = 4', 'anr ..= "text"'], 'E1019:')
1081 CheckDefFailure(['var xnr += 4'], 'E1020:', 1)
1082 CheckScriptFailure(['vim9script', 'var xnr += 4'], 'E1020:')
1083 CheckDefFailure(["var xnr = xnr + 1"], 'E1001:', 1)
1084 CheckScriptFailure(['vim9script', 'var xnr = xnr + 4'], 'E121:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001085
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001086 CheckScriptFailure(['vim9script', 'def Func()', 'var dummy = s:notfound', 'enddef', 'defcompile'], 'E1108:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001087
Bram Moolenaar98309382020-09-27 21:58:45 +02001088 CheckDefFailure(['var name: list<string> = [123]'], 'expected list<string> but got list<number>')
1089 CheckDefFailure(['var name: list<number> = ["xx"]'], 'expected list<number> but got list<string>')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001090
Bram Moolenaare0de1712020-12-02 17:36:54 +01001091 CheckDefFailure(['var name: dict<string> = {key: 123}'], 'expected dict<string> but got dict<number>')
1092 CheckDefFailure(['var name: dict<number> = {key: "xx"}'], 'expected dict<number> but got dict<string>')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001093
Bram Moolenaar98309382020-09-27 21:58:45 +02001094 CheckDefFailure(['var name = feedkeys("0")'], 'E1031:')
1095 CheckDefFailure(['var name: number = feedkeys("0")'], 'expected number but got void')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001096
Bram Moolenaar98309382020-09-27 21:58:45 +02001097 CheckDefFailure(['var name: dict <number>'], 'E1068:')
1098 CheckDefFailure(['var name: dict<number'], 'E1009:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001099
1100 assert_fails('s/^/\=Mess()/n', 'E794:')
Bram Moolenaar98309382020-09-27 21:58:45 +02001101 CheckDefFailure(['var name: dict<number'], 'E1009:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001102
1103 CheckDefFailure(['w:foo: number = 10'],
1104 'E488: Trailing characters: : number = 1')
1105 CheckDefFailure(['t:foo: bool = true'],
1106 'E488: Trailing characters: : bool = true')
1107 CheckDefFailure(['b:foo: string = "x"'],
1108 'E488: Trailing characters: : string = "x"')
1109 CheckDefFailure(['g:foo: number = 123'],
1110 'E488: Trailing characters: : number = 123')
1111enddef
1112
1113def Test_assign_list()
Bram Moolenaare65081d2021-06-27 15:04:05 +02001114 var lines =<< trim END
1115 var l: list<string> = []
1116 l[0] = 'value'
1117 assert_equal('value', l[0])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001118
Bram Moolenaare65081d2021-06-27 15:04:05 +02001119 l[1] = 'asdf'
1120 assert_equal('value', l[0])
1121 assert_equal('asdf', l[1])
1122 assert_equal('asdf', l[-1])
1123 assert_equal('value', l[-2])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001124
Bram Moolenaare65081d2021-06-27 15:04:05 +02001125 var nrl: list<number> = []
1126 for i in range(5)
1127 nrl[i] = i
1128 endfor
1129 assert_equal([0, 1, 2, 3, 4], nrl)
1130
1131 var ul: list<any>
1132 ul[0] = 1
1133 ul[1] = 2
1134 ul[2] = 3
1135 assert_equal([1, 2, 3], ul)
1136 END
1137 CheckDefAndScriptSuccess(lines)
Bram Moolenaar334a8b42020-10-19 16:07:42 +02001138
1139 CheckDefFailure(["var l: list<number> = ['', true]"], 'E1012: Type mismatch; expected list<number> but got list<any>', 1)
1140 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 +02001141enddef
1142
1143def Test_assign_dict()
Bram Moolenaarc3689572021-01-01 19:40:02 +01001144 var lines =<< trim END
1145 var d: dict<string> = {}
1146 d['key'] = 'value'
1147 assert_equal('value', d['key'])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001148
Bram Moolenaarc3689572021-01-01 19:40:02 +01001149 d[123] = 'qwerty'
1150 assert_equal('qwerty', d[123])
1151 assert_equal('qwerty', d['123'])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001152
Bram Moolenaarc3689572021-01-01 19:40:02 +01001153 var nrd: dict<number> = {}
1154 for i in range(3)
1155 nrd[i] = i
1156 endfor
1157 assert_equal({0: 0, 1: 1, 2: 2}, nrd)
1158
1159 d.somekey = 'someval'
1160 assert_equal({key: 'value', '123': 'qwerty', somekey: 'someval'}, d)
1161 # unlet d.somekey
1162 # assert_equal({key: 'value', '123': 'qwerty'}, d)
1163 END
1164 CheckDefAndScriptSuccess(lines)
1165
1166 # TODO: move to above once "unlet d.somekey" in :def is implemented
1167 lines =<< trim END
1168 vim9script
1169 var d: dict<string> = {}
1170 d['key'] = 'value'
1171 d.somekey = 'someval'
1172 assert_equal({key: 'value', somekey: 'someval'}, d)
1173 unlet d.somekey
1174 assert_equal({key: 'value'}, d)
1175 END
1176 CheckScriptSuccess(lines)
Bram Moolenaar334a8b42020-10-19 16:07:42 +02001177
Bram Moolenaare0de1712020-12-02 17:36:54 +01001178 CheckDefFailure(["var d: dict<number> = {a: '', b: true}"], 'E1012: Type mismatch; expected dict<number> but got dict<any>', 1)
1179 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 +02001180enddef
1181
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001182def Test_assign_dict_unknown_type()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001183 var lines =<< trim END
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001184 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001185 var mylist = []
Bram Moolenaare0de1712020-12-02 17:36:54 +01001186 mylist += [{one: 'one'}]
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001187 def Func()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001188 var dd = mylist[0]
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001189 assert_equal('one', dd.one)
1190 enddef
1191 Func()
1192 END
1193 CheckScriptSuccess(lines)
1194
Bram Moolenaarce2c5442020-11-28 21:21:17 +01001195 lines =<< trim END
1196 vim9script
1197 var mylist = [[]]
Bram Moolenaare0de1712020-12-02 17:36:54 +01001198 mylist[0] += [{one: 'one'}]
Bram Moolenaarce2c5442020-11-28 21:21:17 +01001199 def Func()
1200 var dd = mylist[0][0]
1201 assert_equal('one', dd.one)
1202 enddef
1203 Func()
1204 END
1205 CheckScriptSuccess(lines)
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001206enddef
1207
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001208def Test_assign_dict_with_op()
1209 var lines =<< trim END
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001210 var ds: dict<string> = {a: 'x'}
1211 ds['a'] ..= 'y'
1212 ds.a ..= 'z'
1213 assert_equal('xyz', ds.a)
1214
1215 var dn: dict<number> = {a: 9}
1216 dn['a'] += 2
1217 assert_equal(11, dn.a)
1218 dn.a += 2
1219 assert_equal(13, dn.a)
1220
1221 dn['a'] -= 3
1222 assert_equal(10, dn.a)
1223 dn.a -= 2
1224 assert_equal(8, dn.a)
1225
1226 dn['a'] *= 2
1227 assert_equal(16, dn.a)
1228 dn.a *= 2
1229 assert_equal(32, dn.a)
1230
1231 dn['a'] /= 3
1232 assert_equal(10, dn.a)
1233 dn.a /= 2
1234 assert_equal(5, dn.a)
1235
1236 dn['a'] %= 3
1237 assert_equal(2, dn.a)
1238 dn.a %= 6
1239 assert_equal(2, dn.a)
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001240
1241 var dd: dict<dict<list<any>>>
1242 dd.a = {}
1243 dd.a.b = [0]
1244 dd.a.b += [1]
1245 assert_equal({a: {b: [0, 1]}}, dd)
Bram Moolenaar0d351a02021-04-05 21:05:26 +02001246
1247 var dab = {a: ['b']}
1248 dab.a[0] ..= 'c'
1249 assert_equal({a: ['bc']}, dab)
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001250 END
Bram Moolenaare42939a2021-04-05 17:11:17 +02001251 CheckDefAndScriptSuccess(lines)
1252enddef
1253
1254def Test_assign_list_with_op()
1255 var lines =<< trim END
1256 var ls: list<string> = ['x']
1257 ls[0] ..= 'y'
1258 assert_equal('xy', ls[0])
1259
1260 var ln: list<number> = [9]
1261 ln[0] += 2
1262 assert_equal(11, ln[0])
1263
1264 ln[0] -= 3
1265 assert_equal(8, ln[0])
1266
1267 ln[0] *= 2
1268 assert_equal(16, ln[0])
1269
1270 ln[0] /= 3
1271 assert_equal(5, ln[0])
1272
1273 ln[0] %= 3
1274 assert_equal(2, ln[0])
1275 END
1276 CheckDefAndScriptSuccess(lines)
1277enddef
1278
1279def Test_assign_with_op_fails()
1280 var lines =<< trim END
1281 var s = 'abc'
1282 s[1] += 'x'
1283 END
1284 CheckDefAndScriptFailure2(lines, 'E1141:', 'E689:', 2)
1285
1286 lines =<< trim END
1287 var s = 'abc'
1288 s[1] ..= 'x'
1289 END
1290 CheckDefAndScriptFailure2(lines, 'E1141:', 'E689:', 2)
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001291
1292 lines =<< trim END
1293 var dd: dict<dict<list<any>>>
1294 dd.a = {}
1295 dd.a.b += [1]
1296 END
1297 CheckDefExecAndScriptFailure(lines, 'E716:', 3)
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001298enddef
1299
Bram Moolenaar971caf52020-09-21 22:21:42 +02001300def Test_assign_lambda()
1301 # check if assign a lambda to a variable which type is func or any.
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001302 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +02001303 vim9script
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001304 var FuncRef = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001305 assert_equal(123, FuncRef())
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001306 var FuncRef_Func: func = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001307 assert_equal(123, FuncRef_Func())
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001308 var FuncRef_Any: any = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001309 assert_equal(123, FuncRef_Any())
Bram Moolenaar06409502021-02-17 17:00:27 +01001310 var FuncRef_Number: func(): number = () => 321
1311 assert_equal(321, FuncRef_Number())
Bram Moolenaar971caf52020-09-21 22:21:42 +02001312 END
1313 CheckScriptSuccess(lines)
Bram Moolenaar328eac22021-01-07 19:23:08 +01001314
1315 lines =<< trim END
1316 var Ref: func(number)
1317 Ref = (j) => !j
1318 END
Bram Moolenaar06409502021-02-17 17:00:27 +01001319 CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected func(number) but got func(any): bool')
Bram Moolenaarc7dac852021-02-17 18:49:11 +01001320
1321 lines =<< trim END
1322 echo filter([1, 2, 3], (_, v: string) => v + 1)
1323 END
1324 CheckDefAndScriptFailure(lines, 'E1051:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001325enddef
1326
Bram Moolenaarc0e29012020-09-27 14:22:48 +02001327def Test_heredoc()
1328 var lines =<< trim END # comment
1329 text
1330 END
1331 assert_equal(['text'], lines)
1332
1333 CheckDefFailure(['var lines =<< trim END X', 'END'], 'E488:')
1334 CheckDefFailure(['var lines =<< trim END " comment', 'END'], 'E488:')
Bram Moolenaar2914a202020-09-27 18:24:03 +02001335
1336 lines =<< trim [END]
1337 def Func()
1338 var&lines =<< trim END
1339 x
1340 x
Bram Moolenaarb5b94802020-12-13 17:50:20 +01001341 enddef
1342 defcompile
1343 [END]
1344 CheckScriptFailure(lines, 'E1145: Missing heredoc end marker: END')
1345 delfunc! g:Func
1346
1347 lines =<< trim [END]
1348 def Func()
1349 var lines =<< trim END
1350 x
1351 x
Bram Moolenaar2914a202020-09-27 18:24:03 +02001352 x
1353 x
1354 x
1355 x
1356 x
1357 x
1358 enddef
1359 call Func()
1360 [END]
Bram Moolenaarb5b94802020-12-13 17:50:20 +01001361 CheckScriptFailure(lines, 'E1145: Missing heredoc end marker: END')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001362 delfunc! g:Func
Bram Moolenaarc0e29012020-09-27 14:22:48 +02001363enddef
1364
Bram Moolenaar083966f2020-12-08 11:29:40 +01001365def Test_var_func_call()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001366 var lines =<< trim END
1367 vim9script
1368 func GetValue()
1369 if exists('g:count')
1370 let g:count += 1
1371 else
1372 let g:count = 1
1373 endif
1374 return 'this'
1375 endfunc
1376 var val: string = GetValue()
1377 # env var is always a string
1378 var env = $TERM
1379 END
1380 writefile(lines, 'Xfinished')
1381 source Xfinished
1382 # GetValue() is not called during discovery phase
1383 assert_equal(1, g:count)
1384
1385 unlet g:count
1386 delete('Xfinished')
1387enddef
1388
Bram Moolenaar083966f2020-12-08 11:29:40 +01001389def Test_var_missing_type()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001390 var lines =<< trim END
1391 vim9script
1392 var name = g:unknown
1393 END
1394 CheckScriptFailure(lines, 'E121:')
1395
1396 lines =<< trim END
1397 vim9script
1398 var nr: number = 123
1399 var name = nr
1400 END
1401 CheckScriptSuccess(lines)
1402enddef
1403
Bram Moolenaar083966f2020-12-08 11:29:40 +01001404def Test_var_declaration()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001405 var lines =<< trim END
1406 vim9script
1407 var name: string
1408 g:var_uninit = name
1409 name = 'text'
1410 g:var_test = name
1411 # prefixing s: is optional
1412 s:name = 'prefixed'
1413 g:var_prefixed = s:name
1414
Bram Moolenaar7257af42020-12-08 22:08:51 +01001415 const FOO: number = 123
1416 assert_equal(123, FOO)
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001417 const FOOS = 'foos'
1418 assert_equal('foos', FOOS)
1419 final FLIST = [1]
1420 assert_equal([1], FLIST)
1421 FLIST[0] = 11
1422 assert_equal([11], FLIST)
1423
1424 const g:FOO: number = 321
1425 assert_equal(321, g:FOO)
1426 const g:FOOS = 'gfoos'
1427 assert_equal('gfoos', g:FOOS)
1428 final g:FLIST = [2]
1429 assert_equal([2], g:FLIST)
1430 g:FLIST[0] = 22
1431 assert_equal([22], g:FLIST)
1432
Bram Moolenaard877a572021-04-01 19:42:48 +02001433 def SetGlobalConst()
1434 const g:globConst = 123
1435 enddef
1436 SetGlobalConst()
1437 assert_equal(123, g:globConst)
1438 assert_true(islocked('g:globConst'))
1439
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001440 const w:FOO: number = 46
1441 assert_equal(46, w:FOO)
1442 const w:FOOS = 'wfoos'
1443 assert_equal('wfoos', w:FOOS)
1444 final w:FLIST = [3]
1445 assert_equal([3], w:FLIST)
1446 w:FLIST[0] = 33
1447 assert_equal([33], w:FLIST)
Bram Moolenaar7257af42020-12-08 22:08:51 +01001448
Bram Moolenaar55759b52020-09-30 22:59:42 +02001449 var s:other: number
1450 other = 1234
1451 g:other_var = other
1452
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01001453 var xyz: string # comment
1454
Bram Moolenaar55759b52020-09-30 22:59:42 +02001455 # type is inferred
Bram Moolenaar3862ea32021-01-01 21:05:55 +01001456 var s:dict = {['a']: 222}
Bram Moolenaar55759b52020-09-30 22:59:42 +02001457 def GetDictVal(key: any)
1458 g:dict_val = s:dict[key]
1459 enddef
1460 GetDictVal('a')
Bram Moolenaar08251752021-01-11 21:20:18 +01001461
1462 final adict: dict<string> = {}
1463 def ChangeAdict()
1464 adict.foo = 'foo'
1465 enddef
1466 ChangeAdict()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001467 END
1468 CheckScriptSuccess(lines)
1469 assert_equal('', g:var_uninit)
1470 assert_equal('text', g:var_test)
1471 assert_equal('prefixed', g:var_prefixed)
1472 assert_equal(1234, g:other_var)
1473 assert_equal(222, g:dict_val)
1474
1475 unlet g:var_uninit
1476 unlet g:var_test
1477 unlet g:var_prefixed
1478 unlet g:other_var
Bram Moolenaar6e50ec22021-04-03 19:32:44 +02001479 unlet g:globConst
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001480 unlet g:FOO
1481 unlet g:FOOS
1482 unlet g:FLIST
1483 unlet w:FOO
1484 unlet w:FOOS
1485 unlet w:FLIST
Bram Moolenaar55759b52020-09-30 22:59:42 +02001486enddef
1487
Bram Moolenaar083966f2020-12-08 11:29:40 +01001488def Test_var_declaration_fails()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001489 var lines =<< trim END
1490 vim9script
1491 final var: string
1492 END
1493 CheckScriptFailure(lines, 'E1125:')
1494
1495 lines =<< trim END
1496 vim9script
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001497 const g:constvar = 'string'
1498 g:constvar = 'xx'
1499 END
1500 CheckScriptFailure(lines, 'E741:')
1501 unlet g:constvar
1502
1503 lines =<< trim END
1504 vim9script
Bram Moolenaarf5906aa2021-04-02 14:35:15 +02001505 var name = 'one'
1506 lockvar name
1507 def SetLocked()
1508 name = 'two'
1509 enddef
1510 SetLocked()
1511 END
Bram Moolenaardcf29ac2021-04-02 14:44:02 +02001512 CheckScriptFailure(lines, 'E741: Value is locked: name', 1)
1513
1514 lines =<< trim END
1515 let s:legacy = 'one'
1516 lockvar s:legacy
1517 def SetLocked()
1518 s:legacy = 'two'
1519 enddef
1520 call SetLocked()
1521 END
1522 CheckScriptFailure(lines, 'E741: Value is locked: s:legacy', 1)
Bram Moolenaarf5906aa2021-04-02 14:35:15 +02001523
1524 lines =<< trim END
1525 vim9script
Bram Moolenaard877a572021-04-01 19:42:48 +02001526 def SetGlobalConst()
1527 const g:globConst = 123
1528 enddef
1529 SetGlobalConst()
1530 g:globConst = 234
1531 END
Bram Moolenaar6e50ec22021-04-03 19:32:44 +02001532 CheckScriptFailure(lines, 'E741: Value is locked: g:globConst', 6)
Bram Moolenaard877a572021-04-01 19:42:48 +02001533 unlet g:globConst
1534
1535 lines =<< trim END
1536 vim9script
Bram Moolenaar08251752021-01-11 21:20:18 +01001537 const cdict: dict<string> = {}
1538 def Change()
1539 cdict.foo = 'foo'
1540 enddef
1541 defcompile
1542 END
1543 CheckScriptFailure(lines, 'E46:')
1544
1545 lines =<< trim END
1546 vim9script
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001547 final w:finalvar = [9]
1548 w:finalvar = [8]
1549 END
1550 CheckScriptFailure(lines, 'E1122:')
1551 unlet w:finalvar
1552
1553 lines =<< trim END
1554 vim9script
Bram Moolenaar55759b52020-09-30 22:59:42 +02001555 const var: string
1556 END
1557 CheckScriptFailure(lines, 'E1021:')
1558
1559 lines =<< trim END
1560 vim9script
1561 var 9var: string
1562 END
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01001563 CheckScriptFailure(lines, 'E488:')
Bram Moolenaar7257af42020-12-08 22:08:51 +01001564
1565 CheckDefFailure(['var foo.bar = 2'], 'E1087:')
1566 CheckDefFailure(['var foo[3] = 2'], 'E1087:')
1567 CheckDefFailure(['const foo: number'], 'E1021:')
Bram Moolenaar55759b52020-09-30 22:59:42 +02001568enddef
1569
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001570def Test_script_local_in_legacy()
1571 # OK to define script-local later when prefixed with s:
1572 var lines =<< trim END
1573 def SetLater()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001574 s:legvar = 'two'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001575 enddef
1576 defcompile
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001577 let s:legvar = 'one'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001578 call SetLater()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001579 call assert_equal('two', s:legvar)
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001580 END
1581 CheckScriptSuccess(lines)
1582
1583 # OK to leave out s: prefix when script-local already defined
1584 lines =<< trim END
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001585 let s:legvar = 'one'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001586 def SetNoPrefix()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001587 legvar = 'two'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001588 enddef
1589 call SetNoPrefix()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001590 call assert_equal('two', s:legvar)
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001591 END
1592 CheckScriptSuccess(lines)
1593
1594 # Not OK to leave out s: prefix when script-local defined later
1595 lines =<< trim END
1596 def SetLaterNoPrefix()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001597 legvar = 'two'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001598 enddef
1599 defcompile
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001600 let s:legvar = 'one'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001601 END
1602 CheckScriptFailure(lines, 'E476:', 1)
1603enddef
1604
Bram Moolenaar083966f2020-12-08 11:29:40 +01001605def Test_var_type_check()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001606 var lines =<< trim END
1607 vim9script
1608 var name: string
1609 name = 1234
1610 END
1611 CheckScriptFailure(lines, 'E1012:')
1612
1613 lines =<< trim END
1614 vim9script
1615 var name:string
1616 END
1617 CheckScriptFailure(lines, 'E1069:')
1618
1619 lines =<< trim END
1620 vim9script
1621 var name: asdf
1622 END
1623 CheckScriptFailure(lines, 'E1010:')
1624
1625 lines =<< trim END
1626 vim9script
1627 var s:l: list<number>
1628 s:l = []
1629 END
1630 CheckScriptSuccess(lines)
1631
1632 lines =<< trim END
1633 vim9script
1634 var s:d: dict<number>
1635 s:d = {}
1636 END
1637 CheckScriptSuccess(lines)
1638enddef
1639
1640let g:dict_number = #{one: 1, two: 2}
1641
Bram Moolenaar083966f2020-12-08 11:29:40 +01001642def Test_var_list_dict_type()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001643 var ll: list<number>
1644 ll = [1, 2, 2, 3, 3, 3]->uniq()
1645 ll->assert_equal([1, 2, 3])
1646
1647 var dd: dict<number>
1648 dd = g:dict_number
1649 dd->assert_equal(g:dict_number)
1650
1651 var lines =<< trim END
1652 var ll: list<number>
1653 ll = [1, 2, 3]->map('"one"')
1654 END
Bram Moolenaar70250fb2021-01-16 19:01:53 +01001655 CheckDefExecFailure(lines, 'E1012: Type mismatch; expected number but got string')
Bram Moolenaar55759b52020-09-30 22:59:42 +02001656enddef
1657
Bram Moolenaar083966f2020-12-08 11:29:40 +01001658def Test_cannot_use_let()
1659 CheckDefAndScriptFailure(['let a = 34'], 'E1126:', 1)
1660enddef
1661
Bram Moolenaar55759b52020-09-30 22:59:42 +02001662def Test_unlet()
1663 g:somevar = 'yes'
1664 assert_true(exists('g:somevar'))
1665 unlet g:somevar
1666 assert_false(exists('g:somevar'))
1667 unlet! g:somevar
1668
1669 # also works for script-local variable in legacy Vim script
1670 s:somevar = 'legacy'
1671 assert_true(exists('s:somevar'))
1672 unlet s:somevar
1673 assert_false(exists('s:somevar'))
1674 unlet! s:somevar
1675
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001676 CheckDefExecFailure([
1677 'var dd = 111',
1678 'unlet dd',
1679 ], 'E1081:', 2)
1680
Bram Moolenaar752fc692021-01-04 21:57:11 +01001681 # dict unlet
1682 var dd = {a: 1, b: 2, c: 3}
1683 unlet dd['a']
1684 unlet dd.c
1685 assert_equal({b: 2}, dd)
1686
1687 # list unlet
1688 var ll = [1, 2, 3, 4]
1689 unlet ll[1]
1690 unlet ll[-1]
1691 assert_equal([1, 3], ll)
1692
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001693 ll = [1, 2, 3, 4]
1694 unlet ll[0 : 1]
1695 assert_equal([3, 4], ll)
1696
1697 ll = [1, 2, 3, 4]
1698 unlet ll[2 : 8]
1699 assert_equal([1, 2], ll)
1700
1701 ll = [1, 2, 3, 4]
1702 unlet ll[-2 : -1]
1703 assert_equal([1, 2], ll)
1704
1705 CheckDefFailure([
1706 'var ll = [1, 2]',
1707 'll[1 : 2] = 7',
1708 ], 'E1165:', 2)
1709 CheckDefFailure([
1710 'var dd = {a: 1}',
1711 'unlet dd["a" : "a"]',
1712 ], 'E1166:', 2)
1713 CheckDefExecFailure([
1714 'unlet g:adict[0 : 1]',
1715 ], 'E1148:', 1)
1716 CheckDefFailure([
1717 'var ll = [1, 2]',
1718 'unlet ll[0:1]',
1719 ], 'E1004:', 2)
1720 CheckDefFailure([
1721 'var ll = [1, 2]',
1722 'unlet ll[0 :1]',
1723 ], 'E1004:', 2)
1724 CheckDefFailure([
1725 'var ll = [1, 2]',
1726 'unlet ll[0: 1]',
1727 ], 'E1004:', 2)
1728
1729 CheckDefFailure([
1730 'var ll = [1, 2]',
1731 'unlet ll["x" : 1]',
1732 ], 'E1012:', 2)
1733 CheckDefFailure([
1734 'var ll = [1, 2]',
1735 'unlet ll[0 : "x"]',
1736 ], 'E1012:', 2)
1737
Bram Moolenaar752fc692021-01-04 21:57:11 +01001738 # list of dict unlet
1739 var dl = [{a: 1, b: 2}, {c: 3}]
1740 unlet dl[0]['b']
1741 assert_equal([{a: 1}, {c: 3}], dl)
1742
1743 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001744 'var ll = test_null_list()',
1745 'unlet ll[0]',
1746 ], 'E684:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001747 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001748 'var ll = [1]',
1749 'unlet ll[2]',
1750 ], 'E684:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001751 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001752 'var ll = [1]',
1753 'unlet ll[g:astring]',
Bram Moolenaarf30a14d2021-01-17 21:51:24 +01001754 ], 'E1012:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001755 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001756 'var dd = test_null_dict()',
1757 'unlet dd["a"]',
1758 ], 'E716:', 2)
1759 CheckDefExecFailure([
1760 'var dd = {a: 1}',
1761 'unlet dd["b"]',
1762 ], 'E716:', 2)
1763 CheckDefExecFailure([
1764 'var dd = {a: 1}',
1765 'unlet dd[g:alist]',
1766 ], 'E1105:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001767
Bram Moolenaar2ef951d2021-01-03 20:55:26 +01001768 # can compile unlet before variable exists
Bram Moolenaar752fc692021-01-04 21:57:11 +01001769 g:someDict = {key: 'val'}
1770 var k = 'key'
1771 unlet g:someDict[k]
1772 assert_equal({}, g:someDict)
1773 unlet g:someDict
1774 assert_false(exists('g:someDict'))
Bram Moolenaar2ef951d2021-01-03 20:55:26 +01001775
Bram Moolenaar55759b52020-09-30 22:59:42 +02001776 CheckScriptFailure([
1777 'vim9script',
1778 'var svar = 123',
1779 'unlet svar',
1780 ], 'E1081:')
1781 CheckScriptFailure([
1782 'vim9script',
1783 'var svar = 123',
1784 'unlet s:svar',
1785 ], 'E1081:')
1786 CheckScriptFailure([
1787 'vim9script',
1788 'var svar = 123',
1789 'def Func()',
1790 ' unlet svar',
1791 'enddef',
1792 'defcompile',
1793 ], 'E1081:')
1794 CheckScriptFailure([
1795 'vim9script',
1796 'var svar = 123',
Bram Moolenaar9aed7292020-12-18 15:38:00 +01001797 'func Func()',
1798 ' unlet s:svar',
1799 'endfunc',
1800 'Func()',
1801 ], 'E1081:')
1802 CheckScriptFailure([
1803 'vim9script',
1804 'var svar = 123',
Bram Moolenaar55759b52020-09-30 22:59:42 +02001805 'def Func()',
1806 ' unlet s:svar',
1807 'enddef',
1808 'defcompile',
1809 ], 'E1081:')
1810
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001811 writefile(['vim9script', 'export var svar = 1234'], 'XunletExport.vim')
1812 var lines =<< trim END
1813 vim9script
1814 import svar from './XunletExport.vim'
1815 def UnletSvar()
1816 unlet svar
1817 enddef
1818 defcompile
1819 END
1820 CheckScriptFailure(lines, 'E1081:', 1)
1821 delete('XunletExport.vim')
1822
Bram Moolenaar55759b52020-09-30 22:59:42 +02001823 $ENVVAR = 'foobar'
1824 assert_equal('foobar', $ENVVAR)
1825 unlet $ENVVAR
1826 assert_equal('', $ENVVAR)
1827enddef
1828
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01001829def Test_expr_error_no_assign()
1830 var lines =<< trim END
1831 vim9script
1832 var x = invalid
1833 echo x
1834 END
1835 CheckScriptFailureList(lines, ['E121:', 'E121:'])
1836
1837 lines =<< trim END
1838 vim9script
1839 var x = 1 / 0
1840 echo x
1841 END
Bram Moolenaar227c58a2021-04-28 20:40:44 +02001842 CheckScriptFailure(lines, 'E1154:')
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01001843
1844 lines =<< trim END
1845 vim9script
1846 var x = 1 % 0
1847 echo x
1848 END
Bram Moolenaar227c58a2021-04-28 20:40:44 +02001849 CheckScriptFailure(lines, 'E1154:')
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01001850
1851 lines =<< trim END
1852 var x: string 'string'
1853 END
1854 CheckDefAndScriptFailure(lines, 'E488:')
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01001855enddef
1856
1857
Bram Moolenaar17126b12021-01-07 22:03:02 +01001858def Test_assign_command_modifier()
1859 var lines =<< trim END
Bram Moolenaar9e0f8832021-01-09 12:09:22 +01001860 var verbose = 0
1861 verbose = 1
1862 assert_equal(1, verbose)
1863 silent verbose = 2
1864 assert_equal(2, verbose)
1865 silent verbose += 2
1866 assert_equal(4, verbose)
1867 silent verbose -= 1
1868 assert_equal(3, verbose)
Bram Moolenaar17126b12021-01-07 22:03:02 +01001869
Bram Moolenaar9e0f8832021-01-09 12:09:22 +01001870 var topleft = {one: 1}
1871 sandbox topleft.one = 3
1872 assert_equal({one: 3}, topleft)
1873 leftabove topleft[' '] = 4
1874 assert_equal({one: 3, ' ': 4}, topleft)
1875
1876 var x: number
1877 var y: number
1878 silent [x, y] = [1, 2]
1879 assert_equal(1, x)
1880 assert_equal(2, y)
Bram Moolenaar17126b12021-01-07 22:03:02 +01001881 END
1882 CheckDefAndScriptSuccess(lines)
1883enddef
1884
Bram Moolenaar74f4a962021-06-17 21:03:07 +02001885def Test_assign_alt_buf_register()
1886 var lines =<< trim END
1887 edit 'file_b1'
1888 var b1 = bufnr()
1889 edit 'file_b2'
1890 var b2 = bufnr()
1891 assert_equal(b1, bufnr('#'))
1892 @# = b2
1893 assert_equal(b2, bufnr('#'))
1894 END
1895 CheckDefAndScriptSuccess(lines)
1896enddef
1897
Bram Moolenaar32154662021-03-28 21:14:06 +02001898def Test_script_funcref_case()
1899 var lines =<< trim END
1900 var Len = (s: string): number => len(s) + 1
1901 assert_equal(5, Len('asdf'))
1902 END
1903 CheckDefAndScriptSuccess(lines)
1904
1905 lines =<< trim END
1906 var len = (s: string): number => len(s) + 1
1907 END
1908 CheckDefAndScriptFailure(lines, 'E704:')
1909
1910 lines =<< trim END
1911 vim9script
1912 var s:Len = (s: string): number => len(s) + 2
1913 assert_equal(6, Len('asdf'))
1914 END
1915 CheckScriptSuccess(lines)
1916
1917 lines =<< trim END
1918 vim9script
1919 var s:len = (s: string): number => len(s) + 1
1920 END
1921 CheckScriptFailure(lines, 'E704:')
1922enddef
1923
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001924def Test_inc_dec()
1925 var lines =<< trim END
1926 var nr = 7
1927 ++nr
Bram Moolenaar730bf302021-04-24 20:43:56 +02001928 assert_equal(8, nr)
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001929 --nr
Bram Moolenaar730bf302021-04-24 20:43:56 +02001930 assert_equal(7, nr)
Bram Moolenaarf3d30842021-06-25 19:29:30 +02001931 ++nr | ++nr
1932 assert_equal(9, nr)
1933 ++nr # comment
1934 assert_equal(10, nr)
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001935
1936 var ll = [1, 2]
1937 --ll[0]
1938 ++ll[1]
Bram Moolenaar730bf302021-04-24 20:43:56 +02001939 assert_equal([0, 3], ll)
1940
1941 g:count = 1
1942 ++g:count
1943 --g:count
1944 assert_equal(1, g:count)
1945 unlet g:count
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001946 END
1947 CheckDefAndScriptSuccess(lines)
Bram Moolenaar22480d12021-06-25 21:31:09 +02001948
1949 lines =<< trim END
1950 var nr = 7
1951 ++ nr
1952 END
1953 CheckDefAndScriptFailure(lines, "E1202: No white space allowed after '++': ++ nr")
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001954enddef
1955
Bram Moolenaar55759b52020-09-30 22:59:42 +02001956
Bram Moolenaar971caf52020-09-21 22:21:42 +02001957" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker