blob: 7b7b62305be88cd9aceeee7f429b2eb64c98842d [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
231 lines =<< trim END
232 @c = 'areg'
233 @c ..= 'add'
234 assert_equal('aregadd', @c)
235 END
236 CheckDefAndScriptSuccess(lines)
237
238 CheckDefFailure(['@a += "more"'], 'E1051:')
239 CheckDefFailure(['@a += 123'], 'E1012:')
240
241 v:errmsg = 'none'
242 v:errmsg ..= 'again'
243 assert_equal('noneagain', v:errmsg)
244 CheckDefFailure(['v:errmsg += "more"'], 'E1051:')
245 CheckDefFailure(['v:errmsg += 123'], 'E1012:')
246
Bram Moolenaarfc8aa6d2020-10-12 20:31:26 +0200247 var text =<< trim END
248 some text
249 END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200250enddef
251
Bram Moolenaard0edaf92021-05-28 21:06:08 +0200252def Test_reserved_name()
253 for name in ['true', 'false', 'null']
254 CheckDefExecAndScriptFailure(['var ' .. name .. ' = 0'], 'E1034:')
255 CheckDefExecAndScriptFailure(['var ' .. name .. ': bool'], 'E1034:')
256 endfor
257enddef
258
Bram Moolenaaraa1959b2021-04-15 22:13:39 +0200259def Test_skipped_assignment()
260 var lines =<< trim END
261 for x in []
262 var i: number = 1
263 while false
264 i += 1
265 endwhile
266 endfor
267 END
268 CheckDefAndScriptSuccess(lines)
269enddef
270
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100271def Test_assign_unpack()
272 var lines =<< trim END
273 var v1: number
274 var v2: number
275 [v1, v2] = [1, 2]
276 assert_equal(1, v1)
277 assert_equal(2, v2)
Bram Moolenaarf93bbd02021-04-10 22:35:43 +0200278
279 [v1, _, v2, _] = [1, 99, 2, 77]
280 assert_equal(1, v1)
281 assert_equal(2, v2)
282
283 [v1, v2; _] = [1, 2, 3, 4, 5]
284 assert_equal(1, v1)
285 assert_equal(2, v2)
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100286 END
287 CheckDefAndScriptSuccess(lines)
288
289 lines =<< trim END
290 var v1: number
291 var v2: number
292 [v1, v2] =
293 END
294 CheckDefFailure(lines, 'E1097:', 5)
295
296 lines =<< trim END
297 var v1: number
298 var v2: number
299 [v1, v2] = xxx
300 END
301 CheckDefFailure(lines, 'E1001:', 3)
302
303 lines =<< trim END
304 var v1: number
305 var v2: number
306 [v1, v2] = popup_clear()
307 END
308 CheckDefFailure(lines, 'E1031:', 3)
309
310 lines =<< trim END
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100311 [v1, v2] = [1, 2]
312 END
313 CheckDefFailure(lines, 'E1089', 1)
314 CheckScriptFailure(['vim9script'] + lines, 'E1089', 2)
315
316 lines =<< trim END
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100317 var v1: number
318 var v2: number
319 [v1, v2] = ''
320 END
321 CheckDefFailure(lines, 'E1012: Type mismatch; expected list<any> but got string', 3)
Bram Moolenaarf785aa12021-02-11 21:19:34 +0100322
323 lines =<< trim END
324 g:values = [false, 0]
325 var x: bool
326 var y: string
327 [x, y] = g:values
328 END
329 CheckDefExecAndScriptFailure(lines, 'E1163: Variable 2: type mismatch, expected string but got number')
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100330enddef
331
Bram Moolenaar7f764942020-12-02 15:11:18 +0100332def Test_assign_linebreak()
333 var nr: number
334 nr =
335 123
336 assert_equal(123, nr)
337
338 var n2: number
339 [nr, n2] =
340 [12, 34]
341 assert_equal(12, nr)
342 assert_equal(34, n2)
343
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100344 CheckDefFailure(["var x = #"], 'E1097:', 3)
Bram Moolenaar77709b12021-04-03 21:01:01 +0200345
346 var lines =<< trim END
347 var x: list<string> = ['a']
348 var y: list<number> = x
349 ->copy()
350 ->copy()
351 END
352 CheckDefFailure(lines, 'E1012:', 2)
Bram Moolenaar7f764942020-12-02 15:11:18 +0100353enddef
354
Bram Moolenaardc234ca2020-11-28 18:52:33 +0100355def Test_assign_index()
356 # list of list
357 var l1: list<number>
358 l1[0] = 123
359 assert_equal([123], l1)
360
361 var l2: list<list<number>>
362 l2[0] = []
363 l2[0][0] = 123
364 assert_equal([[123]], l2)
365
366 var l3: list<list<list<number>>>
367 l3[0] = []
368 l3[0][0] = []
369 l3[0][0][0] = 123
370 assert_equal([[[123]]], l3)
371
372 var lines =<< trim END
373 var l3: list<list<number>>
374 l3[0] = []
375 l3[0][0] = []
376 END
377 CheckDefFailure(lines, 'E1012: Type mismatch; expected number but got list<unknown>', 3)
378
379 # dict of dict
380 var d1: dict<number>
381 d1.one = 1
382 assert_equal({one: 1}, d1)
383
384 var d2: dict<dict<number>>
385 d2.one = {}
386 d2.one.two = 123
387 assert_equal({one: {two: 123}}, d2)
388
389 var d3: dict<dict<dict<number>>>
390 d3.one = {}
391 d3.one.two = {}
392 d3.one.two.three = 123
393 assert_equal({one: {two: {three: 123}}}, d3)
394
395 lines =<< trim END
396 var d3: dict<dict<number>>
397 d3.one = {}
398 d3.one.two = {}
399 END
400 CheckDefFailure(lines, 'E1012: Type mismatch; expected number but got dict<unknown>', 3)
401
Bram Moolenaaracbae182020-12-13 18:44:43 +0100402 lines =<< trim END
403 var lines: list<string>
404 lines['a'] = 'asdf'
405 END
Bram Moolenaarf30a14d2021-01-17 21:51:24 +0100406 CheckDefFailure(lines, 'E1012:', 2)
Bram Moolenaaracbae182020-12-13 18:44:43 +0100407
408 lines =<< trim END
409 var lines: string
410 lines[9] = 'asdf'
411 END
412 CheckDefFailure(lines, 'E1141:', 2)
413
Bram Moolenaardc234ca2020-11-28 18:52:33 +0100414 # list of dict
415 var ld: list<dict<number>>
416 ld[0] = {}
417 ld[0].one = 123
418 assert_equal([{one: 123}], ld)
419
420 lines =<< trim END
421 var ld: list<dict<number>>
422 ld[0] = []
423 END
424 CheckDefFailure(lines, 'E1012: Type mismatch; expected dict<number> but got list<unknown>', 2)
425
426 # dict of list
427 var dl: dict<list<number>>
428 dl.one = []
429 dl.one[0] = 123
430 assert_equal({one: [123]}, dl)
431
432 lines =<< trim END
433 var dl: dict<list<number>>
434 dl.one = {}
435 END
436 CheckDefFailure(lines, 'E1012: Type mismatch; expected list<number> but got dict<unknown>', 2)
437enddef
438
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200439def Test_extend_list()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200440 var lines =<< trim END
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200441 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200442 var l: list<number>
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200443 l += [123]
444 assert_equal([123], l)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100445 END
446 CheckScriptSuccess(lines)
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200447
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100448 lines =<< trim END
449 vim9script
450 var list: list<string>
451 extend(list, ['x'])
452 assert_equal(['x'], list)
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200453 END
454 CheckScriptSuccess(lines)
Bram Moolenaar1a739232020-10-10 15:37:58 +0200455
456 # appending to NULL list from a function
457 lines =<< trim END
458 vim9script
459 var list: list<string>
460 def Func()
461 list += ['a', 'b']
462 enddef
463 Func()
464 assert_equal(['a', 'b'], list)
465 END
466 CheckScriptSuccess(lines)
Bram Moolenaar3beaf9c2020-12-18 17:23:14 +0100467 lines =<< trim END
468 vim9script
469 var list: list<string>
470 def Func()
471 extend(list, ['x', 'b'])
472 enddef
473 Func()
474 assert_equal(['x', 'b'], list)
475 END
476 CheckScriptSuccess(lines)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100477
478 lines =<< trim END
479 vim9script
480 var l: list<string> = test_null_list()
481 extend(l, ['x'])
482 assert_equal(['x'], l)
483 END
484 CheckScriptSuccess(lines)
485
486 lines =<< trim END
487 vim9script
488 extend(test_null_list(), ['x'])
489 END
490 CheckScriptFailure(lines, 'E1134:', 2)
491enddef
492
493def Test_extend_dict()
494 var lines =<< trim END
495 vim9script
496 var d: dict<number>
Bram Moolenaare0de1712020-12-02 17:36:54 +0100497 extend(d, {a: 1})
498 assert_equal({a: 1}, d)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100499
500 var d2: dict<number>
501 d2['one'] = 1
Bram Moolenaare0de1712020-12-02 17:36:54 +0100502 assert_equal({one: 1}, d2)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100503 END
504 CheckScriptSuccess(lines)
505
506 lines =<< trim END
507 vim9script
508 var d: dict<string> = test_null_dict()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100509 extend(d, {a: 'x'})
510 assert_equal({a: 'x'}, d)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100511 END
512 CheckScriptSuccess(lines)
513
514 lines =<< trim END
515 vim9script
Bram Moolenaare0de1712020-12-02 17:36:54 +0100516 extend(test_null_dict(), {a: 'x'})
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100517 END
518 CheckScriptFailure(lines, 'E1133:', 2)
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200519enddef
520
Bram Moolenaar971caf52020-09-21 22:21:42 +0200521def Test_single_letter_vars()
522 # single letter variables
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200523 var a: number = 123
Bram Moolenaar971caf52020-09-21 22:21:42 +0200524 a = 123
525 assert_equal(123, a)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200526 var b: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200527 b = 123
528 assert_equal(123, b)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200529 var g: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200530 g = 123
531 assert_equal(123, g)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200532 var s: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200533 s = 123
534 assert_equal(123, s)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200535 var t: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200536 t = 123
537 assert_equal(123, t)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200538 var v: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200539 v = 123
540 assert_equal(123, v)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200541 var w: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200542 w = 123
543 assert_equal(123, w)
544enddef
545
546def Test_vim9_single_char_vars()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200547 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200548 vim9script
549
550 # single character variable declarations work
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200551 var a: string
552 var b: number
553 var l: list<any>
554 var s: string
555 var t: number
556 var v: number
557 var w: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200558
559 # script-local variables can be used without s: prefix
560 a = 'script-a'
561 b = 111
562 l = [1, 2, 3]
563 s = 'script-s'
564 t = 222
565 v = 333
566 w = 444
567
568 assert_equal('script-a', a)
569 assert_equal(111, b)
570 assert_equal([1, 2, 3], l)
571 assert_equal('script-s', s)
572 assert_equal(222, t)
573 assert_equal(333, v)
574 assert_equal(444, w)
575 END
576 writefile(lines, 'Xsinglechar')
577 source Xsinglechar
578 delete('Xsinglechar')
579enddef
580
581def Test_assignment_list()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200582 var list1: list<bool> = [false, true, false]
583 var list2: list<number> = [1, 2, 3]
584 var list3: list<string> = ['sdf', 'asdf']
585 var list4: list<any> = ['yes', true, 1234]
586 var list5: list<blob> = [0z01, 0z02]
Bram Moolenaar971caf52020-09-21 22:21:42 +0200587
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200588 var listS: list<string> = []
589 var listN: list<number> = []
Bram Moolenaar971caf52020-09-21 22:21:42 +0200590
591 assert_equal([1, 2, 3], list2)
592 list2[-1] = 99
593 assert_equal([1, 2, 99], list2)
594 list2[-2] = 88
595 assert_equal([1, 88, 99], list2)
596 list2[-3] = 77
597 assert_equal([77, 88, 99], list2)
598 list2 += [100]
599 assert_equal([77, 88, 99, 100], list2)
600
601 list3 += ['end']
602 assert_equal(['sdf', 'asdf', 'end'], list3)
603
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200604 CheckDefExecFailure(['var ll = [1, 2, 3]', 'll[-4] = 6'], 'E684:')
605 CheckDefExecFailure(['var [v1, v2] = [1, 2]'], 'E1092:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200606
607 # type becomes list<any>
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200608 var somelist = rand() > 0 ? [1, 2, 3] : ['a', 'b', 'c']
Bram Moolenaar4f5e3972020-12-21 17:30:50 +0100609
610 var lines =<< trim END
611 var d = {dd: test_null_list()}
612 d.dd[0] = 0
613 END
614 CheckDefExecFailure(lines, 'E1147:', 2)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200615enddef
616
Bram Moolenaarf30a14d2021-01-17 21:51:24 +0100617def Test_assignment_list_any_index()
618 var l: list<number> = [1, 2]
619 for [x, y, _]
620 in [[0, 1, ''], [1, 3, '']]
621 l[x] = l[x] + y
622 endfor
623 assert_equal([2, 5], l)
624enddef
625
Bram Moolenaar971caf52020-09-21 22:21:42 +0200626def Test_assignment_list_vim9script()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200627 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200628 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200629 var v1: number
630 var v2: number
631 var v3: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200632 [v1, v2, v3] = [1, 2, 3]
633 assert_equal([1, 2, 3], [v1, v2, v3])
634 END
635 CheckScriptSuccess(lines)
636enddef
637
638def Test_assignment_dict()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100639 var dict1: dict<bool> = {one: false, two: true}
640 var dict2: dict<number> = {one: 1, two: 2}
641 var dict3: dict<string> = {key: 'value'}
642 var dict4: dict<any> = {one: 1, two: '2'}
643 var dict5: dict<blob> = {one: 0z01, two: 0z02}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200644
645 # overwrite
646 dict3['key'] = 'another'
Bram Moolenaare0de1712020-12-02 17:36:54 +0100647 assert_equal(dict3, {key: 'another'})
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100648 dict3.key = 'yet another'
Bram Moolenaare0de1712020-12-02 17:36:54 +0100649 assert_equal(dict3, {key: 'yet another'})
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100650
Bram Moolenaard24602f2020-12-20 15:20:56 +0100651 # member "any" can also be a dict and assigned to
652 var anydict: dict<any> = {nest: {}, nr: 0}
653 anydict.nest['this'] = 123
654 anydict.nest.that = 456
655 assert_equal({nest: {this: 123, that: 456}, nr: 0}, anydict)
656
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100657 var lines =<< trim END
Bram Moolenaard88dc4d2020-12-20 14:49:42 +0100658 var dd = {}
659 dd.two = 2
660 assert_equal({two: 2}, dd)
661 END
Bram Moolenaar4f5e3972020-12-21 17:30:50 +0100662 CheckDefAndScriptSuccess(lines)
663
664 lines =<< trim END
665 var d = {dd: {}}
666 d.dd[0] = 2
667 d.dd['x'] = 3
668 d.dd.y = 4
669 assert_equal({dd: {0: 2, x: 3, y: 4}}, d)
670 END
671 CheckDefAndScriptSuccess(lines)
Bram Moolenaard88dc4d2020-12-20 14:49:42 +0100672
673 lines =<< trim END
Bram Moolenaare0de1712020-12-02 17:36:54 +0100674 var dd = {one: 1}
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100675 dd.one) = 2
676 END
Bram Moolenaard88dc4d2020-12-20 14:49:42 +0100677 CheckDefFailure(lines, 'E488:', 2)
678
679 lines =<< trim END
680 var dd = {one: 1}
681 var dd.one = 2
682 END
683 CheckDefAndScriptFailure(lines, 'E1017:', 2)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200684
685 # empty key can be used
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200686 var dd = {}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200687 dd[""] = 6
Bram Moolenaare0de1712020-12-02 17:36:54 +0100688 assert_equal({['']: 6}, dd)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200689
690 # type becomes dict<any>
Bram Moolenaare0de1712020-12-02 17:36:54 +0100691 var somedict = rand() > 0 ? {a: 1, b: 2} : {a: 'a', b: 'b'}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200692
693 # assignment to script-local dict
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100694 lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200695 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200696 var test: dict<any> = {}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200697 def FillDict(): dict<any>
698 test['a'] = 43
699 return test
700 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +0100701 assert_equal({a: 43}, FillDict())
Bram Moolenaar971caf52020-09-21 22:21:42 +0200702 END
703 CheckScriptSuccess(lines)
704
705 lines =<< trim END
706 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200707 var test: dict<any>
Bram Moolenaar971caf52020-09-21 22:21:42 +0200708 def FillDict(): dict<any>
709 test['a'] = 43
710 return test
711 enddef
712 FillDict()
Bram Moolenaar3beaf9c2020-12-18 17:23:14 +0100713 assert_equal({a: 43}, test)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200714 END
Bram Moolenaar3beaf9c2020-12-18 17:23:14 +0100715 CheckScriptSuccess(lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200716
717 # assignment to global dict
718 lines =<< trim END
719 vim9script
720 g:test = {}
721 def FillDict(): dict<any>
722 g:test['a'] = 43
723 return g:test
724 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +0100725 assert_equal({a: 43}, FillDict())
Bram Moolenaar971caf52020-09-21 22:21:42 +0200726 END
727 CheckScriptSuccess(lines)
728
729 # assignment to buffer dict
730 lines =<< trim END
731 vim9script
732 b:test = {}
733 def FillDict(): dict<any>
734 b:test['a'] = 43
735 return b:test
736 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +0100737 assert_equal({a: 43}, FillDict())
Bram Moolenaar971caf52020-09-21 22:21:42 +0200738 END
739 CheckScriptSuccess(lines)
Bram Moolenaar4f5e3972020-12-21 17:30:50 +0100740
741 lines =<< trim END
742 var d = {dd: test_null_dict()}
743 d.dd[0] = 0
744 END
745 CheckDefExecFailure(lines, 'E1103:', 2)
746
747 lines =<< trim END
748 var d = {dd: 'string'}
749 d.dd[0] = 0
750 END
751 CheckDefExecFailure(lines, 'E1148:', 2)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200752enddef
753
754def Test_assignment_local()
755 # Test in a separated file in order not to the current buffer/window/tab is
756 # changed.
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200757 var script_lines: list<string> =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200758 let b:existing = 'yes'
759 let w:existing = 'yes'
760 let t:existing = 'yes'
761
762 def Test_assignment_local_internal()
763 b:newvar = 'new'
764 assert_equal('new', b:newvar)
765 assert_equal('yes', b:existing)
766 b:existing = 'no'
767 assert_equal('no', b:existing)
768 b:existing ..= 'NO'
769 assert_equal('noNO', b:existing)
770
771 w:newvar = 'new'
772 assert_equal('new', w:newvar)
773 assert_equal('yes', w:existing)
774 w:existing = 'no'
775 assert_equal('no', w:existing)
776 w:existing ..= 'NO'
777 assert_equal('noNO', w:existing)
778
779 t:newvar = 'new'
780 assert_equal('new', t:newvar)
781 assert_equal('yes', t:existing)
782 t:existing = 'no'
783 assert_equal('no', t:existing)
784 t:existing ..= 'NO'
785 assert_equal('noNO', t:existing)
786 enddef
787 call Test_assignment_local_internal()
788 END
789 CheckScriptSuccess(script_lines)
790enddef
791
792def Test_assignment_default()
Bram Moolenaar971caf52020-09-21 22:21:42 +0200793 # Test default values.
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200794 var thebool: bool
Bram Moolenaar971caf52020-09-21 22:21:42 +0200795 assert_equal(v:false, thebool)
796
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200797 var thenumber: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200798 assert_equal(0, thenumber)
799
800 if has('float')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200801 var thefloat: float
Bram Moolenaar971caf52020-09-21 22:21:42 +0200802 assert_equal(0.0, thefloat)
803 endif
804
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200805 var thestring: string
Bram Moolenaar971caf52020-09-21 22:21:42 +0200806 assert_equal('', thestring)
807
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200808 var theblob: blob
Bram Moolenaar971caf52020-09-21 22:21:42 +0200809 assert_equal(0z, theblob)
810
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200811 var Thefunc: func
Bram Moolenaar971caf52020-09-21 22:21:42 +0200812 assert_equal(test_null_function(), Thefunc)
813
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200814 var thelist: list<any>
Bram Moolenaar971caf52020-09-21 22:21:42 +0200815 assert_equal([], thelist)
816
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200817 var thedict: dict<any>
Bram Moolenaar971caf52020-09-21 22:21:42 +0200818 assert_equal({}, thedict)
819
820 if has('channel')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200821 var thejob: job
Bram Moolenaar971caf52020-09-21 22:21:42 +0200822 assert_equal(test_null_job(), thejob)
823
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200824 var thechannel: channel
Bram Moolenaar971caf52020-09-21 22:21:42 +0200825 assert_equal(test_null_channel(), thechannel)
826
827 if has('unix') && executable('cat')
828 # check with non-null job and channel, types must match
Bram Moolenaare0de1712020-12-02 17:36:54 +0100829 thejob = job_start("cat ", {})
Bram Moolenaar971caf52020-09-21 22:21:42 +0200830 thechannel = job_getchannel(thejob)
831 job_stop(thejob, 'kill')
832 endif
833 endif
834
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200835 var nr = 1234 | nr = 5678
Bram Moolenaar971caf52020-09-21 22:21:42 +0200836 assert_equal(5678, nr)
837enddef
838
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100839let scriptvar = 'init'
840
Bram Moolenaar971caf52020-09-21 22:21:42 +0200841def Test_assignment_var_list()
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100842 var lines =<< trim END
843 var v1: string
844 var v2: string
845 var vrem: list<string>
846 [v1] = ['aaa']
847 assert_equal('aaa', v1)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200848
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100849 [v1, v2] = ['one', 'two']
850 assert_equal('one', v1)
851 assert_equal('two', v2)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200852
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100853 [v1, v2; vrem] = ['one', 'two']
854 assert_equal('one', v1)
855 assert_equal('two', v2)
856 assert_equal([], vrem)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200857
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100858 [v1, v2; vrem] = ['one', 'two', 'three']
859 assert_equal('one', v1)
860 assert_equal('two', v2)
861 assert_equal(['three'], vrem)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200862
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100863 [&ts, &sw] = [3, 4]
864 assert_equal(3, &ts)
865 assert_equal(4, &sw)
866 set ts=8 sw=4
Bram Moolenaar004d9b02020-11-30 21:40:03 +0100867
868 [@a, @z] = ['aa', 'zz']
869 assert_equal('aa', @a)
870 assert_equal('zz', @z)
871
872 [$SOME_VAR, $OTHER_VAR] = ['some', 'other']
873 assert_equal('some', $SOME_VAR)
874 assert_equal('other', $OTHER_VAR)
875
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100876 [g:globalvar, b:bufvar, w:winvar, t:tabvar, v:errmsg] =
877 ['global', 'buf', 'win', 'tab', 'error']
Bram Moolenaar004d9b02020-11-30 21:40:03 +0100878 assert_equal('global', g:globalvar)
Bram Moolenaar004d9b02020-11-30 21:40:03 +0100879 assert_equal('buf', b:bufvar)
880 assert_equal('win', w:winvar)
881 assert_equal('tab', t:tabvar)
882 assert_equal('error', v:errmsg)
883 unlet g:globalvar
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100884 END
885 CheckDefAndScriptSuccess(lines)
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100886
887 [g:globalvar, s:scriptvar, b:bufvar] = ['global', 'script', 'buf']
888 assert_equal('global', g:globalvar)
889 assert_equal('script', s:scriptvar)
890 assert_equal('buf', b:bufvar)
891
892 lines =<< trim END
893 vim9script
894 var s:scriptvar = 'init'
895 [g:globalvar, s:scriptvar, w:winvar] = ['global', 'script', 'win']
896 assert_equal('global', g:globalvar)
897 assert_equal('script', s:scriptvar)
898 assert_equal('win', w:winvar)
899 END
900 CheckScriptSuccess(lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200901enddef
902
903def Test_assignment_vim9script()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200904 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200905 vim9script
906 def Func(): list<number>
907 return [1, 2]
908 enddef
Bram Moolenaar98309382020-09-27 21:58:45 +0200909 var name1: number
910 var name2: number
911 [name1, name2] =
Bram Moolenaar971caf52020-09-21 22:21:42 +0200912 Func()
Bram Moolenaar98309382020-09-27 21:58:45 +0200913 assert_equal(1, name1)
914 assert_equal(2, name2)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200915 var ll =
Bram Moolenaar971caf52020-09-21 22:21:42 +0200916 Func()
917 assert_equal([1, 2], ll)
918
919 @/ = 'text'
920 assert_equal('text', @/)
921 @0 = 'zero'
922 assert_equal('zero', @0)
923 @1 = 'one'
924 assert_equal('one', @1)
925 @9 = 'nine'
926 assert_equal('nine', @9)
927 @- = 'minus'
928 assert_equal('minus', @-)
929 if has('clipboard_working')
930 @* = 'star'
931 assert_equal('star', @*)
932 @+ = 'plus'
933 assert_equal('plus', @+)
934 endif
935
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200936 var a: number = 123
Bram Moolenaar971caf52020-09-21 22:21:42 +0200937 assert_equal(123, a)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200938 var s: string = 'yes'
Bram Moolenaar971caf52020-09-21 22:21:42 +0200939 assert_equal('yes', s)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200940 var b: number = 42
Bram Moolenaar971caf52020-09-21 22:21:42 +0200941 assert_equal(42, b)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200942 var w: number = 43
Bram Moolenaar971caf52020-09-21 22:21:42 +0200943 assert_equal(43, w)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200944 var t: number = 44
Bram Moolenaar971caf52020-09-21 22:21:42 +0200945 assert_equal(44, t)
Bram Moolenaar7a66a172020-10-16 19:56:12 +0200946
947 var to_var = 0
948 to_var = 3
949 assert_equal(3, to_var)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200950 END
951 CheckScriptSuccess(lines)
Bram Moolenaar06f9c692020-09-27 21:27:40 +0200952
953 lines =<< trim END
954 vim9script
955 var n: number
956 def Func()
957 n = 'string'
958 enddef
959 defcompile
960 END
961 CheckScriptFailure(lines, 'E1012: Type mismatch; expected number but got string')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200962enddef
963
964def Mess(): string
965 v:foldstart = 123
966 return 'xxx'
967enddef
968
969def Test_assignment_failure()
Bram Moolenaar98309382020-09-27 21:58:45 +0200970 CheckDefFailure(['var name=234'], 'E1004:')
971 CheckDefFailure(['var name =234'], 'E1004:')
972 CheckDefFailure(['var name= 234'], 'E1004:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200973
Bram Moolenaar98309382020-09-27 21:58:45 +0200974 CheckScriptFailure(['vim9script', 'var name=234'], 'E1004:')
975 CheckScriptFailure(['vim9script', 'var name=234'], "before and after '='")
976 CheckScriptFailure(['vim9script', 'var name =234'], 'E1004:')
977 CheckScriptFailure(['vim9script', 'var name= 234'], 'E1004:')
978 CheckScriptFailure(['vim9script', 'var name = 234', 'name+=234'], 'E1004:')
979 CheckScriptFailure(['vim9script', 'var name = 234', 'name+=234'], "before and after '+='")
980 CheckScriptFailure(['vim9script', 'var name = "x"', 'name..="y"'], 'E1004:')
981 CheckScriptFailure(['vim9script', 'var name = "x"', 'name..="y"'], "before and after '..='")
Bram Moolenaar971caf52020-09-21 22:21:42 +0200982
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200983 CheckDefFailure(['var true = 1'], 'E1034:')
984 CheckDefFailure(['var false = 1'], 'E1034:')
Bram Moolenaar74235772021-06-12 14:53:05 +0200985 CheckDefFailure(['var null = 1'], 'E1034:')
986 CheckDefFailure(['var this = 1'], 'E1034:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200987
988 CheckDefFailure(['[a; b; c] = g:list'], 'E452:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200989 CheckDefExecFailure(['var a: number',
Bram Moolenaar971caf52020-09-21 22:21:42 +0200990 '[a] = test_null_list()'], 'E1093:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200991 CheckDefExecFailure(['var a: number',
Bram Moolenaar971caf52020-09-21 22:21:42 +0200992 '[a] = []'], 'E1093:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200993 CheckDefExecFailure(['var x: number',
994 'var y: number',
Bram Moolenaar971caf52020-09-21 22:21:42 +0200995 '[x, y] = [1]'], 'E1093:')
Bram Moolenaar352134b2020-10-17 22:04:08 +0200996 CheckDefExecFailure(['var x: string',
997 'var y: string',
998 '[x, y] = ["x"]'], 'E1093:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200999 CheckDefExecFailure(['var x: number',
1000 'var y: number',
1001 'var z: list<number>',
Bram Moolenaar971caf52020-09-21 22:21:42 +02001002 '[x, y; z] = [1]'], 'E1093:')
1003
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001004 CheckDefFailure(['var somevar'], "E1022:")
1005 CheckDefFailure(['var &tabstop = 4'], 'E1052:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001006 CheckDefFailure(['&g:option = 5'], 'E113:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001007 CheckScriptFailure(['vim9script', 'var &tabstop = 4'], 'E1052:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001008
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001009 CheckDefFailure(['var $VAR = 5'], 'E1016: Cannot declare an environment variable:')
1010 CheckScriptFailure(['vim9script', 'var $ENV = "xxx"'], 'E1016:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001011
1012 if has('dnd')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001013 CheckDefFailure(['var @~ = 5'], 'E1066:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001014 else
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001015 CheckDefFailure(['var @~ = 5'], 'E354:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001016 CheckDefFailure(['@~ = 5'], 'E354:')
1017 endif
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001018 CheckDefFailure(['var @a = 5'], 'E1066:')
1019 CheckDefFailure(['var @/ = "x"'], 'E1066:')
1020 CheckScriptFailure(['vim9script', 'var @a = "abc"'], 'E1066:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001021
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001022 CheckDefFailure(['var g:var = 5'], 'E1016: Cannot declare a global variable:')
1023 CheckDefFailure(['var w:var = 5'], 'E1016: Cannot declare a window variable:')
1024 CheckDefFailure(['var b:var = 5'], 'E1016: Cannot declare a buffer variable:')
1025 CheckDefFailure(['var t:var = 5'], 'E1016: Cannot declare a tab variable:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001026
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001027 CheckDefFailure(['var anr = 4', 'anr ..= "text"'], 'E1019:')
1028 CheckDefFailure(['var xnr += 4'], 'E1020:', 1)
1029 CheckScriptFailure(['vim9script', 'var xnr += 4'], 'E1020:')
1030 CheckDefFailure(["var xnr = xnr + 1"], 'E1001:', 1)
1031 CheckScriptFailure(['vim9script', 'var xnr = xnr + 4'], 'E121:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001032
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001033 CheckScriptFailure(['vim9script', 'def Func()', 'var dummy = s:notfound', 'enddef', 'defcompile'], 'E1108:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001034
Bram Moolenaar98309382020-09-27 21:58:45 +02001035 CheckDefFailure(['var name: list<string> = [123]'], 'expected list<string> but got list<number>')
1036 CheckDefFailure(['var name: list<number> = ["xx"]'], 'expected list<number> but got list<string>')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001037
Bram Moolenaare0de1712020-12-02 17:36:54 +01001038 CheckDefFailure(['var name: dict<string> = {key: 123}'], 'expected dict<string> but got dict<number>')
1039 CheckDefFailure(['var name: dict<number> = {key: "xx"}'], 'expected dict<number> but got dict<string>')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001040
Bram Moolenaar98309382020-09-27 21:58:45 +02001041 CheckDefFailure(['var name = feedkeys("0")'], 'E1031:')
1042 CheckDefFailure(['var name: number = feedkeys("0")'], 'expected number but got void')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001043
Bram Moolenaar98309382020-09-27 21:58:45 +02001044 CheckDefFailure(['var name: dict <number>'], 'E1068:')
1045 CheckDefFailure(['var name: dict<number'], 'E1009:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001046
1047 assert_fails('s/^/\=Mess()/n', 'E794:')
Bram Moolenaar98309382020-09-27 21:58:45 +02001048 CheckDefFailure(['var name: dict<number'], 'E1009:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001049
1050 CheckDefFailure(['w:foo: number = 10'],
1051 'E488: Trailing characters: : number = 1')
1052 CheckDefFailure(['t:foo: bool = true'],
1053 'E488: Trailing characters: : bool = true')
1054 CheckDefFailure(['b:foo: string = "x"'],
1055 'E488: Trailing characters: : string = "x"')
1056 CheckDefFailure(['g:foo: number = 123'],
1057 'E488: Trailing characters: : number = 123')
1058enddef
1059
1060def Test_assign_list()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001061 var l: list<string> = []
Bram Moolenaar971caf52020-09-21 22:21:42 +02001062 l[0] = 'value'
1063 assert_equal('value', l[0])
1064
1065 l[1] = 'asdf'
1066 assert_equal('value', l[0])
1067 assert_equal('asdf', l[1])
1068 assert_equal('asdf', l[-1])
1069 assert_equal('value', l[-2])
1070
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001071 var nrl: list<number> = []
Bram Moolenaar971caf52020-09-21 22:21:42 +02001072 for i in range(5)
1073 nrl[i] = i
1074 endfor
1075 assert_equal([0, 1, 2, 3, 4], nrl)
Bram Moolenaar334a8b42020-10-19 16:07:42 +02001076
1077 CheckDefFailure(["var l: list<number> = ['', true]"], 'E1012: Type mismatch; expected list<number> but got list<any>', 1)
1078 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 +02001079enddef
1080
1081def Test_assign_dict()
Bram Moolenaarc3689572021-01-01 19:40:02 +01001082 var lines =<< trim END
1083 var d: dict<string> = {}
1084 d['key'] = 'value'
1085 assert_equal('value', d['key'])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001086
Bram Moolenaarc3689572021-01-01 19:40:02 +01001087 d[123] = 'qwerty'
1088 assert_equal('qwerty', d[123])
1089 assert_equal('qwerty', d['123'])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001090
Bram Moolenaarc3689572021-01-01 19:40:02 +01001091 var nrd: dict<number> = {}
1092 for i in range(3)
1093 nrd[i] = i
1094 endfor
1095 assert_equal({0: 0, 1: 1, 2: 2}, nrd)
1096
1097 d.somekey = 'someval'
1098 assert_equal({key: 'value', '123': 'qwerty', somekey: 'someval'}, d)
1099 # unlet d.somekey
1100 # assert_equal({key: 'value', '123': 'qwerty'}, d)
1101 END
1102 CheckDefAndScriptSuccess(lines)
1103
1104 # TODO: move to above once "unlet d.somekey" in :def is implemented
1105 lines =<< trim END
1106 vim9script
1107 var d: dict<string> = {}
1108 d['key'] = 'value'
1109 d.somekey = 'someval'
1110 assert_equal({key: 'value', somekey: 'someval'}, d)
1111 unlet d.somekey
1112 assert_equal({key: 'value'}, d)
1113 END
1114 CheckScriptSuccess(lines)
Bram Moolenaar334a8b42020-10-19 16:07:42 +02001115
Bram Moolenaare0de1712020-12-02 17:36:54 +01001116 CheckDefFailure(["var d: dict<number> = {a: '', b: true}"], 'E1012: Type mismatch; expected dict<number> but got dict<any>', 1)
1117 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 +02001118enddef
1119
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001120def Test_assign_dict_unknown_type()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001121 var lines =<< trim END
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001122 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001123 var mylist = []
Bram Moolenaare0de1712020-12-02 17:36:54 +01001124 mylist += [{one: 'one'}]
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001125 def Func()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001126 var dd = mylist[0]
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001127 assert_equal('one', dd.one)
1128 enddef
1129 Func()
1130 END
1131 CheckScriptSuccess(lines)
1132
Bram Moolenaarce2c5442020-11-28 21:21:17 +01001133 lines =<< trim END
1134 vim9script
1135 var mylist = [[]]
Bram Moolenaare0de1712020-12-02 17:36:54 +01001136 mylist[0] += [{one: 'one'}]
Bram Moolenaarce2c5442020-11-28 21:21:17 +01001137 def Func()
1138 var dd = mylist[0][0]
1139 assert_equal('one', dd.one)
1140 enddef
1141 Func()
1142 END
1143 CheckScriptSuccess(lines)
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001144enddef
1145
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001146def Test_assign_dict_with_op()
1147 var lines =<< trim END
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001148 var ds: dict<string> = {a: 'x'}
1149 ds['a'] ..= 'y'
1150 ds.a ..= 'z'
1151 assert_equal('xyz', ds.a)
1152
1153 var dn: dict<number> = {a: 9}
1154 dn['a'] += 2
1155 assert_equal(11, dn.a)
1156 dn.a += 2
1157 assert_equal(13, dn.a)
1158
1159 dn['a'] -= 3
1160 assert_equal(10, dn.a)
1161 dn.a -= 2
1162 assert_equal(8, dn.a)
1163
1164 dn['a'] *= 2
1165 assert_equal(16, dn.a)
1166 dn.a *= 2
1167 assert_equal(32, dn.a)
1168
1169 dn['a'] /= 3
1170 assert_equal(10, dn.a)
1171 dn.a /= 2
1172 assert_equal(5, dn.a)
1173
1174 dn['a'] %= 3
1175 assert_equal(2, dn.a)
1176 dn.a %= 6
1177 assert_equal(2, dn.a)
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001178
1179 var dd: dict<dict<list<any>>>
1180 dd.a = {}
1181 dd.a.b = [0]
1182 dd.a.b += [1]
1183 assert_equal({a: {b: [0, 1]}}, dd)
Bram Moolenaar0d351a02021-04-05 21:05:26 +02001184
1185 var dab = {a: ['b']}
1186 dab.a[0] ..= 'c'
1187 assert_equal({a: ['bc']}, dab)
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001188 END
Bram Moolenaare42939a2021-04-05 17:11:17 +02001189 CheckDefAndScriptSuccess(lines)
1190enddef
1191
1192def Test_assign_list_with_op()
1193 var lines =<< trim END
1194 var ls: list<string> = ['x']
1195 ls[0] ..= 'y'
1196 assert_equal('xy', ls[0])
1197
1198 var ln: list<number> = [9]
1199 ln[0] += 2
1200 assert_equal(11, ln[0])
1201
1202 ln[0] -= 3
1203 assert_equal(8, ln[0])
1204
1205 ln[0] *= 2
1206 assert_equal(16, ln[0])
1207
1208 ln[0] /= 3
1209 assert_equal(5, ln[0])
1210
1211 ln[0] %= 3
1212 assert_equal(2, ln[0])
1213 END
1214 CheckDefAndScriptSuccess(lines)
1215enddef
1216
1217def Test_assign_with_op_fails()
1218 var lines =<< trim END
1219 var s = 'abc'
1220 s[1] += 'x'
1221 END
1222 CheckDefAndScriptFailure2(lines, 'E1141:', 'E689:', 2)
1223
1224 lines =<< trim END
1225 var s = 'abc'
1226 s[1] ..= 'x'
1227 END
1228 CheckDefAndScriptFailure2(lines, 'E1141:', 'E689:', 2)
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001229
1230 lines =<< trim END
1231 var dd: dict<dict<list<any>>>
1232 dd.a = {}
1233 dd.a.b += [1]
1234 END
1235 CheckDefExecAndScriptFailure(lines, 'E716:', 3)
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001236enddef
1237
Bram Moolenaar971caf52020-09-21 22:21:42 +02001238def Test_assign_lambda()
1239 # check if assign a lambda to a variable which type is func or any.
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001240 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +02001241 vim9script
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001242 var FuncRef = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001243 assert_equal(123, FuncRef())
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001244 var FuncRef_Func: func = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001245 assert_equal(123, FuncRef_Func())
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001246 var FuncRef_Any: any = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001247 assert_equal(123, FuncRef_Any())
Bram Moolenaar06409502021-02-17 17:00:27 +01001248 var FuncRef_Number: func(): number = () => 321
1249 assert_equal(321, FuncRef_Number())
Bram Moolenaar971caf52020-09-21 22:21:42 +02001250 END
1251 CheckScriptSuccess(lines)
Bram Moolenaar328eac22021-01-07 19:23:08 +01001252
1253 lines =<< trim END
1254 var Ref: func(number)
1255 Ref = (j) => !j
1256 END
Bram Moolenaar06409502021-02-17 17:00:27 +01001257 CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected func(number) but got func(any): bool')
Bram Moolenaarc7dac852021-02-17 18:49:11 +01001258
1259 lines =<< trim END
1260 echo filter([1, 2, 3], (_, v: string) => v + 1)
1261 END
1262 CheckDefAndScriptFailure(lines, 'E1051:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001263enddef
1264
Bram Moolenaarc0e29012020-09-27 14:22:48 +02001265def Test_heredoc()
1266 var lines =<< trim END # comment
1267 text
1268 END
1269 assert_equal(['text'], lines)
1270
1271 CheckDefFailure(['var lines =<< trim END X', 'END'], 'E488:')
1272 CheckDefFailure(['var lines =<< trim END " comment', 'END'], 'E488:')
Bram Moolenaar2914a202020-09-27 18:24:03 +02001273
1274 lines =<< trim [END]
1275 def Func()
1276 var&lines =<< trim END
1277 x
1278 x
Bram Moolenaarb5b94802020-12-13 17:50:20 +01001279 enddef
1280 defcompile
1281 [END]
1282 CheckScriptFailure(lines, 'E1145: Missing heredoc end marker: END')
1283 delfunc! g:Func
1284
1285 lines =<< trim [END]
1286 def Func()
1287 var lines =<< trim END
1288 x
1289 x
Bram Moolenaar2914a202020-09-27 18:24:03 +02001290 x
1291 x
1292 x
1293 x
1294 x
1295 x
1296 enddef
1297 call Func()
1298 [END]
Bram Moolenaarb5b94802020-12-13 17:50:20 +01001299 CheckScriptFailure(lines, 'E1145: Missing heredoc end marker: END')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001300 delfunc! g:Func
Bram Moolenaarc0e29012020-09-27 14:22:48 +02001301enddef
1302
Bram Moolenaar083966f2020-12-08 11:29:40 +01001303def Test_var_func_call()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001304 var lines =<< trim END
1305 vim9script
1306 func GetValue()
1307 if exists('g:count')
1308 let g:count += 1
1309 else
1310 let g:count = 1
1311 endif
1312 return 'this'
1313 endfunc
1314 var val: string = GetValue()
1315 # env var is always a string
1316 var env = $TERM
1317 END
1318 writefile(lines, 'Xfinished')
1319 source Xfinished
1320 # GetValue() is not called during discovery phase
1321 assert_equal(1, g:count)
1322
1323 unlet g:count
1324 delete('Xfinished')
1325enddef
1326
Bram Moolenaar083966f2020-12-08 11:29:40 +01001327def Test_var_missing_type()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001328 var lines =<< trim END
1329 vim9script
1330 var name = g:unknown
1331 END
1332 CheckScriptFailure(lines, 'E121:')
1333
1334 lines =<< trim END
1335 vim9script
1336 var nr: number = 123
1337 var name = nr
1338 END
1339 CheckScriptSuccess(lines)
1340enddef
1341
Bram Moolenaar083966f2020-12-08 11:29:40 +01001342def Test_var_declaration()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001343 var lines =<< trim END
1344 vim9script
1345 var name: string
1346 g:var_uninit = name
1347 name = 'text'
1348 g:var_test = name
1349 # prefixing s: is optional
1350 s:name = 'prefixed'
1351 g:var_prefixed = s:name
1352
Bram Moolenaar7257af42020-12-08 22:08:51 +01001353 const FOO: number = 123
1354 assert_equal(123, FOO)
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001355 const FOOS = 'foos'
1356 assert_equal('foos', FOOS)
1357 final FLIST = [1]
1358 assert_equal([1], FLIST)
1359 FLIST[0] = 11
1360 assert_equal([11], FLIST)
1361
1362 const g:FOO: number = 321
1363 assert_equal(321, g:FOO)
1364 const g:FOOS = 'gfoos'
1365 assert_equal('gfoos', g:FOOS)
1366 final g:FLIST = [2]
1367 assert_equal([2], g:FLIST)
1368 g:FLIST[0] = 22
1369 assert_equal([22], g:FLIST)
1370
Bram Moolenaard877a572021-04-01 19:42:48 +02001371 def SetGlobalConst()
1372 const g:globConst = 123
1373 enddef
1374 SetGlobalConst()
1375 assert_equal(123, g:globConst)
1376 assert_true(islocked('g:globConst'))
1377
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001378 const w:FOO: number = 46
1379 assert_equal(46, w:FOO)
1380 const w:FOOS = 'wfoos'
1381 assert_equal('wfoos', w:FOOS)
1382 final w:FLIST = [3]
1383 assert_equal([3], w:FLIST)
1384 w:FLIST[0] = 33
1385 assert_equal([33], w:FLIST)
Bram Moolenaar7257af42020-12-08 22:08:51 +01001386
Bram Moolenaar55759b52020-09-30 22:59:42 +02001387 var s:other: number
1388 other = 1234
1389 g:other_var = other
1390
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01001391 var xyz: string # comment
1392
Bram Moolenaar55759b52020-09-30 22:59:42 +02001393 # type is inferred
Bram Moolenaar3862ea32021-01-01 21:05:55 +01001394 var s:dict = {['a']: 222}
Bram Moolenaar55759b52020-09-30 22:59:42 +02001395 def GetDictVal(key: any)
1396 g:dict_val = s:dict[key]
1397 enddef
1398 GetDictVal('a')
Bram Moolenaar08251752021-01-11 21:20:18 +01001399
1400 final adict: dict<string> = {}
1401 def ChangeAdict()
1402 adict.foo = 'foo'
1403 enddef
1404 ChangeAdict()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001405 END
1406 CheckScriptSuccess(lines)
1407 assert_equal('', g:var_uninit)
1408 assert_equal('text', g:var_test)
1409 assert_equal('prefixed', g:var_prefixed)
1410 assert_equal(1234, g:other_var)
1411 assert_equal(222, g:dict_val)
1412
1413 unlet g:var_uninit
1414 unlet g:var_test
1415 unlet g:var_prefixed
1416 unlet g:other_var
Bram Moolenaar6e50ec22021-04-03 19:32:44 +02001417 unlet g:globConst
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001418 unlet g:FOO
1419 unlet g:FOOS
1420 unlet g:FLIST
1421 unlet w:FOO
1422 unlet w:FOOS
1423 unlet w:FLIST
Bram Moolenaar55759b52020-09-30 22:59:42 +02001424enddef
1425
Bram Moolenaar083966f2020-12-08 11:29:40 +01001426def Test_var_declaration_fails()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001427 var lines =<< trim END
1428 vim9script
1429 final var: string
1430 END
1431 CheckScriptFailure(lines, 'E1125:')
1432
1433 lines =<< trim END
1434 vim9script
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001435 const g:constvar = 'string'
1436 g:constvar = 'xx'
1437 END
1438 CheckScriptFailure(lines, 'E741:')
1439 unlet g:constvar
1440
1441 lines =<< trim END
1442 vim9script
Bram Moolenaarf5906aa2021-04-02 14:35:15 +02001443 var name = 'one'
1444 lockvar name
1445 def SetLocked()
1446 name = 'two'
1447 enddef
1448 SetLocked()
1449 END
Bram Moolenaardcf29ac2021-04-02 14:44:02 +02001450 CheckScriptFailure(lines, 'E741: Value is locked: name', 1)
1451
1452 lines =<< trim END
1453 let s:legacy = 'one'
1454 lockvar s:legacy
1455 def SetLocked()
1456 s:legacy = 'two'
1457 enddef
1458 call SetLocked()
1459 END
1460 CheckScriptFailure(lines, 'E741: Value is locked: s:legacy', 1)
Bram Moolenaarf5906aa2021-04-02 14:35:15 +02001461
1462 lines =<< trim END
1463 vim9script
Bram Moolenaard877a572021-04-01 19:42:48 +02001464 def SetGlobalConst()
1465 const g:globConst = 123
1466 enddef
1467 SetGlobalConst()
1468 g:globConst = 234
1469 END
Bram Moolenaar6e50ec22021-04-03 19:32:44 +02001470 CheckScriptFailure(lines, 'E741: Value is locked: g:globConst', 6)
Bram Moolenaard877a572021-04-01 19:42:48 +02001471 unlet g:globConst
1472
1473 lines =<< trim END
1474 vim9script
Bram Moolenaar08251752021-01-11 21:20:18 +01001475 const cdict: dict<string> = {}
1476 def Change()
1477 cdict.foo = 'foo'
1478 enddef
1479 defcompile
1480 END
1481 CheckScriptFailure(lines, 'E46:')
1482
1483 lines =<< trim END
1484 vim9script
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001485 final w:finalvar = [9]
1486 w:finalvar = [8]
1487 END
1488 CheckScriptFailure(lines, 'E1122:')
1489 unlet w:finalvar
1490
1491 lines =<< trim END
1492 vim9script
Bram Moolenaar55759b52020-09-30 22:59:42 +02001493 const var: string
1494 END
1495 CheckScriptFailure(lines, 'E1021:')
1496
1497 lines =<< trim END
1498 vim9script
1499 var 9var: string
1500 END
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01001501 CheckScriptFailure(lines, 'E488:')
Bram Moolenaar7257af42020-12-08 22:08:51 +01001502
1503 CheckDefFailure(['var foo.bar = 2'], 'E1087:')
1504 CheckDefFailure(['var foo[3] = 2'], 'E1087:')
1505 CheckDefFailure(['const foo: number'], 'E1021:')
Bram Moolenaar55759b52020-09-30 22:59:42 +02001506enddef
1507
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001508def Test_script_local_in_legacy()
1509 # OK to define script-local later when prefixed with s:
1510 var lines =<< trim END
1511 def SetLater()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001512 s:legvar = 'two'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001513 enddef
1514 defcompile
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001515 let s:legvar = 'one'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001516 call SetLater()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001517 call assert_equal('two', s:legvar)
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001518 END
1519 CheckScriptSuccess(lines)
1520
1521 # OK to leave out s: prefix when script-local already defined
1522 lines =<< trim END
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001523 let s:legvar = 'one'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001524 def SetNoPrefix()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001525 legvar = 'two'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001526 enddef
1527 call SetNoPrefix()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001528 call assert_equal('two', s:legvar)
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001529 END
1530 CheckScriptSuccess(lines)
1531
1532 # Not OK to leave out s: prefix when script-local defined later
1533 lines =<< trim END
1534 def SetLaterNoPrefix()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001535 legvar = 'two'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001536 enddef
1537 defcompile
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001538 let s:legvar = 'one'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001539 END
1540 CheckScriptFailure(lines, 'E476:', 1)
1541enddef
1542
Bram Moolenaar083966f2020-12-08 11:29:40 +01001543def Test_var_type_check()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001544 var lines =<< trim END
1545 vim9script
1546 var name: string
1547 name = 1234
1548 END
1549 CheckScriptFailure(lines, 'E1012:')
1550
1551 lines =<< trim END
1552 vim9script
1553 var name:string
1554 END
1555 CheckScriptFailure(lines, 'E1069:')
1556
1557 lines =<< trim END
1558 vim9script
1559 var name: asdf
1560 END
1561 CheckScriptFailure(lines, 'E1010:')
1562
1563 lines =<< trim END
1564 vim9script
1565 var s:l: list<number>
1566 s:l = []
1567 END
1568 CheckScriptSuccess(lines)
1569
1570 lines =<< trim END
1571 vim9script
1572 var s:d: dict<number>
1573 s:d = {}
1574 END
1575 CheckScriptSuccess(lines)
1576enddef
1577
1578let g:dict_number = #{one: 1, two: 2}
1579
Bram Moolenaar083966f2020-12-08 11:29:40 +01001580def Test_var_list_dict_type()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001581 var ll: list<number>
1582 ll = [1, 2, 2, 3, 3, 3]->uniq()
1583 ll->assert_equal([1, 2, 3])
1584
1585 var dd: dict<number>
1586 dd = g:dict_number
1587 dd->assert_equal(g:dict_number)
1588
1589 var lines =<< trim END
1590 var ll: list<number>
1591 ll = [1, 2, 3]->map('"one"')
1592 END
Bram Moolenaar70250fb2021-01-16 19:01:53 +01001593 CheckDefExecFailure(lines, 'E1012: Type mismatch; expected number but got string')
Bram Moolenaar55759b52020-09-30 22:59:42 +02001594enddef
1595
Bram Moolenaar083966f2020-12-08 11:29:40 +01001596def Test_cannot_use_let()
1597 CheckDefAndScriptFailure(['let a = 34'], 'E1126:', 1)
1598enddef
1599
Bram Moolenaar55759b52020-09-30 22:59:42 +02001600def Test_unlet()
1601 g:somevar = 'yes'
1602 assert_true(exists('g:somevar'))
1603 unlet g:somevar
1604 assert_false(exists('g:somevar'))
1605 unlet! g:somevar
1606
1607 # also works for script-local variable in legacy Vim script
1608 s:somevar = 'legacy'
1609 assert_true(exists('s:somevar'))
1610 unlet s:somevar
1611 assert_false(exists('s:somevar'))
1612 unlet! s:somevar
1613
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001614 CheckDefExecFailure([
1615 'var dd = 111',
1616 'unlet dd',
1617 ], 'E1081:', 2)
1618
Bram Moolenaar752fc692021-01-04 21:57:11 +01001619 # dict unlet
1620 var dd = {a: 1, b: 2, c: 3}
1621 unlet dd['a']
1622 unlet dd.c
1623 assert_equal({b: 2}, dd)
1624
1625 # list unlet
1626 var ll = [1, 2, 3, 4]
1627 unlet ll[1]
1628 unlet ll[-1]
1629 assert_equal([1, 3], ll)
1630
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001631 ll = [1, 2, 3, 4]
1632 unlet ll[0 : 1]
1633 assert_equal([3, 4], ll)
1634
1635 ll = [1, 2, 3, 4]
1636 unlet ll[2 : 8]
1637 assert_equal([1, 2], ll)
1638
1639 ll = [1, 2, 3, 4]
1640 unlet ll[-2 : -1]
1641 assert_equal([1, 2], ll)
1642
1643 CheckDefFailure([
1644 'var ll = [1, 2]',
1645 'll[1 : 2] = 7',
1646 ], 'E1165:', 2)
1647 CheckDefFailure([
1648 'var dd = {a: 1}',
1649 'unlet dd["a" : "a"]',
1650 ], 'E1166:', 2)
1651 CheckDefExecFailure([
1652 'unlet g:adict[0 : 1]',
1653 ], 'E1148:', 1)
1654 CheckDefFailure([
1655 'var ll = [1, 2]',
1656 'unlet ll[0:1]',
1657 ], 'E1004:', 2)
1658 CheckDefFailure([
1659 'var ll = [1, 2]',
1660 'unlet ll[0 :1]',
1661 ], 'E1004:', 2)
1662 CheckDefFailure([
1663 'var ll = [1, 2]',
1664 'unlet ll[0: 1]',
1665 ], 'E1004:', 2)
1666
1667 CheckDefFailure([
1668 'var ll = [1, 2]',
1669 'unlet ll["x" : 1]',
1670 ], 'E1012:', 2)
1671 CheckDefFailure([
1672 'var ll = [1, 2]',
1673 'unlet ll[0 : "x"]',
1674 ], 'E1012:', 2)
1675
Bram Moolenaar752fc692021-01-04 21:57:11 +01001676 # list of dict unlet
1677 var dl = [{a: 1, b: 2}, {c: 3}]
1678 unlet dl[0]['b']
1679 assert_equal([{a: 1}, {c: 3}], dl)
1680
1681 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001682 'var ll = test_null_list()',
1683 'unlet ll[0]',
1684 ], 'E684:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001685 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001686 'var ll = [1]',
1687 'unlet ll[2]',
1688 ], 'E684:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001689 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001690 'var ll = [1]',
1691 'unlet ll[g:astring]',
Bram Moolenaarf30a14d2021-01-17 21:51:24 +01001692 ], 'E1012:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001693 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001694 'var dd = test_null_dict()',
1695 'unlet dd["a"]',
1696 ], 'E716:', 2)
1697 CheckDefExecFailure([
1698 'var dd = {a: 1}',
1699 'unlet dd["b"]',
1700 ], 'E716:', 2)
1701 CheckDefExecFailure([
1702 'var dd = {a: 1}',
1703 'unlet dd[g:alist]',
1704 ], 'E1105:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001705
Bram Moolenaar2ef951d2021-01-03 20:55:26 +01001706 # can compile unlet before variable exists
Bram Moolenaar752fc692021-01-04 21:57:11 +01001707 g:someDict = {key: 'val'}
1708 var k = 'key'
1709 unlet g:someDict[k]
1710 assert_equal({}, g:someDict)
1711 unlet g:someDict
1712 assert_false(exists('g:someDict'))
Bram Moolenaar2ef951d2021-01-03 20:55:26 +01001713
Bram Moolenaar55759b52020-09-30 22:59:42 +02001714 CheckScriptFailure([
1715 'vim9script',
1716 'var svar = 123',
1717 'unlet svar',
1718 ], 'E1081:')
1719 CheckScriptFailure([
1720 'vim9script',
1721 'var svar = 123',
1722 'unlet s:svar',
1723 ], 'E1081:')
1724 CheckScriptFailure([
1725 'vim9script',
1726 'var svar = 123',
1727 'def Func()',
1728 ' unlet svar',
1729 'enddef',
1730 'defcompile',
1731 ], 'E1081:')
1732 CheckScriptFailure([
1733 'vim9script',
1734 'var svar = 123',
Bram Moolenaar9aed7292020-12-18 15:38:00 +01001735 'func Func()',
1736 ' unlet s:svar',
1737 'endfunc',
1738 'Func()',
1739 ], 'E1081:')
1740 CheckScriptFailure([
1741 'vim9script',
1742 'var svar = 123',
Bram Moolenaar55759b52020-09-30 22:59:42 +02001743 'def Func()',
1744 ' unlet s:svar',
1745 'enddef',
1746 'defcompile',
1747 ], 'E1081:')
1748
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001749 writefile(['vim9script', 'export var svar = 1234'], 'XunletExport.vim')
1750 var lines =<< trim END
1751 vim9script
1752 import svar from './XunletExport.vim'
1753 def UnletSvar()
1754 unlet svar
1755 enddef
1756 defcompile
1757 END
1758 CheckScriptFailure(lines, 'E1081:', 1)
1759 delete('XunletExport.vim')
1760
Bram Moolenaar55759b52020-09-30 22:59:42 +02001761 $ENVVAR = 'foobar'
1762 assert_equal('foobar', $ENVVAR)
1763 unlet $ENVVAR
1764 assert_equal('', $ENVVAR)
1765enddef
1766
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01001767def Test_expr_error_no_assign()
1768 var lines =<< trim END
1769 vim9script
1770 var x = invalid
1771 echo x
1772 END
1773 CheckScriptFailureList(lines, ['E121:', 'E121:'])
1774
1775 lines =<< trim END
1776 vim9script
1777 var x = 1 / 0
1778 echo x
1779 END
Bram Moolenaar227c58a2021-04-28 20:40:44 +02001780 CheckScriptFailure(lines, 'E1154:')
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01001781
1782 lines =<< trim END
1783 vim9script
1784 var x = 1 % 0
1785 echo x
1786 END
Bram Moolenaar227c58a2021-04-28 20:40:44 +02001787 CheckScriptFailure(lines, 'E1154:')
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01001788
1789 lines =<< trim END
1790 var x: string 'string'
1791 END
1792 CheckDefAndScriptFailure(lines, 'E488:')
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01001793enddef
1794
1795
Bram Moolenaar17126b12021-01-07 22:03:02 +01001796def Test_assign_command_modifier()
1797 var lines =<< trim END
Bram Moolenaar9e0f8832021-01-09 12:09:22 +01001798 var verbose = 0
1799 verbose = 1
1800 assert_equal(1, verbose)
1801 silent verbose = 2
1802 assert_equal(2, verbose)
1803 silent verbose += 2
1804 assert_equal(4, verbose)
1805 silent verbose -= 1
1806 assert_equal(3, verbose)
Bram Moolenaar17126b12021-01-07 22:03:02 +01001807
Bram Moolenaar9e0f8832021-01-09 12:09:22 +01001808 var topleft = {one: 1}
1809 sandbox topleft.one = 3
1810 assert_equal({one: 3}, topleft)
1811 leftabove topleft[' '] = 4
1812 assert_equal({one: 3, ' ': 4}, topleft)
1813
1814 var x: number
1815 var y: number
1816 silent [x, y] = [1, 2]
1817 assert_equal(1, x)
1818 assert_equal(2, y)
Bram Moolenaar17126b12021-01-07 22:03:02 +01001819 END
1820 CheckDefAndScriptSuccess(lines)
1821enddef
1822
Bram Moolenaar32154662021-03-28 21:14:06 +02001823def Test_script_funcref_case()
1824 var lines =<< trim END
1825 var Len = (s: string): number => len(s) + 1
1826 assert_equal(5, Len('asdf'))
1827 END
1828 CheckDefAndScriptSuccess(lines)
1829
1830 lines =<< trim END
1831 var len = (s: string): number => len(s) + 1
1832 END
1833 CheckDefAndScriptFailure(lines, 'E704:')
1834
1835 lines =<< trim END
1836 vim9script
1837 var s:Len = (s: string): number => len(s) + 2
1838 assert_equal(6, Len('asdf'))
1839 END
1840 CheckScriptSuccess(lines)
1841
1842 lines =<< trim END
1843 vim9script
1844 var s:len = (s: string): number => len(s) + 1
1845 END
1846 CheckScriptFailure(lines, 'E704:')
1847enddef
1848
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001849def Test_inc_dec()
1850 var lines =<< trim END
1851 var nr = 7
1852 ++nr
Bram Moolenaar730bf302021-04-24 20:43:56 +02001853 assert_equal(8, nr)
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001854 --nr
Bram Moolenaar730bf302021-04-24 20:43:56 +02001855 assert_equal(7, nr)
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001856
1857 var ll = [1, 2]
1858 --ll[0]
1859 ++ll[1]
Bram Moolenaar730bf302021-04-24 20:43:56 +02001860 assert_equal([0, 3], ll)
1861
1862 g:count = 1
1863 ++g:count
1864 --g:count
1865 assert_equal(1, g:count)
1866 unlet g:count
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001867 END
1868 CheckDefAndScriptSuccess(lines)
1869enddef
1870
Bram Moolenaar55759b52020-09-30 22:59:42 +02001871
Bram Moolenaar971caf52020-09-21 22:21:42 +02001872" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker