blob: 958b7b980944fa121d00d788cd6ebf02f6bb3c99 [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 Moolenaar8ff16e02020-12-07 21:49:52 +0100252def Test_assign_unpack()
253 var lines =<< trim END
254 var v1: number
255 var v2: number
256 [v1, v2] = [1, 2]
257 assert_equal(1, v1)
258 assert_equal(2, v2)
Bram Moolenaarf93bbd02021-04-10 22:35:43 +0200259
260 [v1, _, v2, _] = [1, 99, 2, 77]
261 assert_equal(1, v1)
262 assert_equal(2, v2)
263
264 [v1, v2; _] = [1, 2, 3, 4, 5]
265 assert_equal(1, v1)
266 assert_equal(2, v2)
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100267 END
268 CheckDefAndScriptSuccess(lines)
269
270 lines =<< trim END
271 var v1: number
272 var v2: number
273 [v1, v2] =
274 END
275 CheckDefFailure(lines, 'E1097:', 5)
276
277 lines =<< trim END
278 var v1: number
279 var v2: number
280 [v1, v2] = xxx
281 END
282 CheckDefFailure(lines, 'E1001:', 3)
283
284 lines =<< trim END
285 var v1: number
286 var v2: number
287 [v1, v2] = popup_clear()
288 END
289 CheckDefFailure(lines, 'E1031:', 3)
290
291 lines =<< trim END
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100292 [v1, v2] = [1, 2]
293 END
294 CheckDefFailure(lines, 'E1089', 1)
295 CheckScriptFailure(['vim9script'] + lines, 'E1089', 2)
296
297 lines =<< trim END
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100298 var v1: number
299 var v2: number
300 [v1, v2] = ''
301 END
302 CheckDefFailure(lines, 'E1012: Type mismatch; expected list<any> but got string', 3)
Bram Moolenaarf785aa12021-02-11 21:19:34 +0100303
304 lines =<< trim END
305 g:values = [false, 0]
306 var x: bool
307 var y: string
308 [x, y] = g:values
309 END
310 CheckDefExecAndScriptFailure(lines, 'E1163: Variable 2: type mismatch, expected string but got number')
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100311enddef
312
Bram Moolenaar7f764942020-12-02 15:11:18 +0100313def Test_assign_linebreak()
314 var nr: number
315 nr =
316 123
317 assert_equal(123, nr)
318
319 var n2: number
320 [nr, n2] =
321 [12, 34]
322 assert_equal(12, nr)
323 assert_equal(34, n2)
324
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100325 CheckDefFailure(["var x = #"], 'E1097:', 3)
Bram Moolenaar77709b12021-04-03 21:01:01 +0200326
327 var lines =<< trim END
328 var x: list<string> = ['a']
329 var y: list<number> = x
330 ->copy()
331 ->copy()
332 END
333 CheckDefFailure(lines, 'E1012:', 2)
Bram Moolenaar7f764942020-12-02 15:11:18 +0100334enddef
335
Bram Moolenaardc234ca2020-11-28 18:52:33 +0100336def Test_assign_index()
337 # list of list
338 var l1: list<number>
339 l1[0] = 123
340 assert_equal([123], l1)
341
342 var l2: list<list<number>>
343 l2[0] = []
344 l2[0][0] = 123
345 assert_equal([[123]], l2)
346
347 var l3: list<list<list<number>>>
348 l3[0] = []
349 l3[0][0] = []
350 l3[0][0][0] = 123
351 assert_equal([[[123]]], l3)
352
353 var lines =<< trim END
354 var l3: list<list<number>>
355 l3[0] = []
356 l3[0][0] = []
357 END
358 CheckDefFailure(lines, 'E1012: Type mismatch; expected number but got list<unknown>', 3)
359
360 # dict of dict
361 var d1: dict<number>
362 d1.one = 1
363 assert_equal({one: 1}, d1)
364
365 var d2: dict<dict<number>>
366 d2.one = {}
367 d2.one.two = 123
368 assert_equal({one: {two: 123}}, d2)
369
370 var d3: dict<dict<dict<number>>>
371 d3.one = {}
372 d3.one.two = {}
373 d3.one.two.three = 123
374 assert_equal({one: {two: {three: 123}}}, d3)
375
376 lines =<< trim END
377 var d3: dict<dict<number>>
378 d3.one = {}
379 d3.one.two = {}
380 END
381 CheckDefFailure(lines, 'E1012: Type mismatch; expected number but got dict<unknown>', 3)
382
Bram Moolenaaracbae182020-12-13 18:44:43 +0100383 lines =<< trim END
384 var lines: list<string>
385 lines['a'] = 'asdf'
386 END
Bram Moolenaarf30a14d2021-01-17 21:51:24 +0100387 CheckDefFailure(lines, 'E1012:', 2)
Bram Moolenaaracbae182020-12-13 18:44:43 +0100388
389 lines =<< trim END
390 var lines: string
391 lines[9] = 'asdf'
392 END
393 CheckDefFailure(lines, 'E1141:', 2)
394
Bram Moolenaardc234ca2020-11-28 18:52:33 +0100395 # list of dict
396 var ld: list<dict<number>>
397 ld[0] = {}
398 ld[0].one = 123
399 assert_equal([{one: 123}], ld)
400
401 lines =<< trim END
402 var ld: list<dict<number>>
403 ld[0] = []
404 END
405 CheckDefFailure(lines, 'E1012: Type mismatch; expected dict<number> but got list<unknown>', 2)
406
407 # dict of list
408 var dl: dict<list<number>>
409 dl.one = []
410 dl.one[0] = 123
411 assert_equal({one: [123]}, dl)
412
413 lines =<< trim END
414 var dl: dict<list<number>>
415 dl.one = {}
416 END
417 CheckDefFailure(lines, 'E1012: Type mismatch; expected list<number> but got dict<unknown>', 2)
418enddef
419
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200420def Test_extend_list()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200421 var lines =<< trim END
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200422 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200423 var l: list<number>
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200424 l += [123]
425 assert_equal([123], l)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100426 END
427 CheckScriptSuccess(lines)
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200428
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100429 lines =<< trim END
430 vim9script
431 var list: list<string>
432 extend(list, ['x'])
433 assert_equal(['x'], list)
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200434 END
435 CheckScriptSuccess(lines)
Bram Moolenaar1a739232020-10-10 15:37:58 +0200436
437 # appending to NULL list from a function
438 lines =<< trim END
439 vim9script
440 var list: list<string>
441 def Func()
442 list += ['a', 'b']
443 enddef
444 Func()
445 assert_equal(['a', 'b'], list)
446 END
447 CheckScriptSuccess(lines)
Bram Moolenaar3beaf9c2020-12-18 17:23:14 +0100448 lines =<< trim END
449 vim9script
450 var list: list<string>
451 def Func()
452 extend(list, ['x', 'b'])
453 enddef
454 Func()
455 assert_equal(['x', 'b'], list)
456 END
457 CheckScriptSuccess(lines)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100458
459 lines =<< trim END
460 vim9script
461 var l: list<string> = test_null_list()
462 extend(l, ['x'])
463 assert_equal(['x'], l)
464 END
465 CheckScriptSuccess(lines)
466
467 lines =<< trim END
468 vim9script
469 extend(test_null_list(), ['x'])
470 END
471 CheckScriptFailure(lines, 'E1134:', 2)
472enddef
473
474def Test_extend_dict()
475 var lines =<< trim END
476 vim9script
477 var d: dict<number>
Bram Moolenaare0de1712020-12-02 17:36:54 +0100478 extend(d, {a: 1})
479 assert_equal({a: 1}, d)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100480
481 var d2: dict<number>
482 d2['one'] = 1
Bram Moolenaare0de1712020-12-02 17:36:54 +0100483 assert_equal({one: 1}, d2)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100484 END
485 CheckScriptSuccess(lines)
486
487 lines =<< trim END
488 vim9script
489 var d: dict<string> = test_null_dict()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100490 extend(d, {a: 'x'})
491 assert_equal({a: 'x'}, d)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100492 END
493 CheckScriptSuccess(lines)
494
495 lines =<< trim END
496 vim9script
Bram Moolenaare0de1712020-12-02 17:36:54 +0100497 extend(test_null_dict(), {a: 'x'})
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100498 END
499 CheckScriptFailure(lines, 'E1133:', 2)
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200500enddef
501
Bram Moolenaar971caf52020-09-21 22:21:42 +0200502def Test_single_letter_vars()
503 # single letter variables
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200504 var a: number = 123
Bram Moolenaar971caf52020-09-21 22:21:42 +0200505 a = 123
506 assert_equal(123, a)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200507 var b: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200508 b = 123
509 assert_equal(123, b)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200510 var g: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200511 g = 123
512 assert_equal(123, g)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200513 var s: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200514 s = 123
515 assert_equal(123, s)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200516 var t: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200517 t = 123
518 assert_equal(123, t)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200519 var v: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200520 v = 123
521 assert_equal(123, v)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200522 var w: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200523 w = 123
524 assert_equal(123, w)
525enddef
526
527def Test_vim9_single_char_vars()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200528 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200529 vim9script
530
531 # single character variable declarations work
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200532 var a: string
533 var b: number
534 var l: list<any>
535 var s: string
536 var t: number
537 var v: number
538 var w: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200539
540 # script-local variables can be used without s: prefix
541 a = 'script-a'
542 b = 111
543 l = [1, 2, 3]
544 s = 'script-s'
545 t = 222
546 v = 333
547 w = 444
548
549 assert_equal('script-a', a)
550 assert_equal(111, b)
551 assert_equal([1, 2, 3], l)
552 assert_equal('script-s', s)
553 assert_equal(222, t)
554 assert_equal(333, v)
555 assert_equal(444, w)
556 END
557 writefile(lines, 'Xsinglechar')
558 source Xsinglechar
559 delete('Xsinglechar')
560enddef
561
562def Test_assignment_list()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200563 var list1: list<bool> = [false, true, false]
564 var list2: list<number> = [1, 2, 3]
565 var list3: list<string> = ['sdf', 'asdf']
566 var list4: list<any> = ['yes', true, 1234]
567 var list5: list<blob> = [0z01, 0z02]
Bram Moolenaar971caf52020-09-21 22:21:42 +0200568
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200569 var listS: list<string> = []
570 var listN: list<number> = []
Bram Moolenaar971caf52020-09-21 22:21:42 +0200571
572 assert_equal([1, 2, 3], list2)
573 list2[-1] = 99
574 assert_equal([1, 2, 99], list2)
575 list2[-2] = 88
576 assert_equal([1, 88, 99], list2)
577 list2[-3] = 77
578 assert_equal([77, 88, 99], list2)
579 list2 += [100]
580 assert_equal([77, 88, 99, 100], list2)
581
582 list3 += ['end']
583 assert_equal(['sdf', 'asdf', 'end'], list3)
584
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200585 CheckDefExecFailure(['var ll = [1, 2, 3]', 'll[-4] = 6'], 'E684:')
586 CheckDefExecFailure(['var [v1, v2] = [1, 2]'], 'E1092:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200587
588 # type becomes list<any>
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200589 var somelist = rand() > 0 ? [1, 2, 3] : ['a', 'b', 'c']
Bram Moolenaar4f5e3972020-12-21 17:30:50 +0100590
591 var lines =<< trim END
592 var d = {dd: test_null_list()}
593 d.dd[0] = 0
594 END
595 CheckDefExecFailure(lines, 'E1147:', 2)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200596enddef
597
Bram Moolenaarf30a14d2021-01-17 21:51:24 +0100598def Test_assignment_list_any_index()
599 var l: list<number> = [1, 2]
600 for [x, y, _]
601 in [[0, 1, ''], [1, 3, '']]
602 l[x] = l[x] + y
603 endfor
604 assert_equal([2, 5], l)
605enddef
606
Bram Moolenaar971caf52020-09-21 22:21:42 +0200607def Test_assignment_list_vim9script()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200608 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200609 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200610 var v1: number
611 var v2: number
612 var v3: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200613 [v1, v2, v3] = [1, 2, 3]
614 assert_equal([1, 2, 3], [v1, v2, v3])
615 END
616 CheckScriptSuccess(lines)
617enddef
618
619def Test_assignment_dict()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100620 var dict1: dict<bool> = {one: false, two: true}
621 var dict2: dict<number> = {one: 1, two: 2}
622 var dict3: dict<string> = {key: 'value'}
623 var dict4: dict<any> = {one: 1, two: '2'}
624 var dict5: dict<blob> = {one: 0z01, two: 0z02}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200625
626 # overwrite
627 dict3['key'] = 'another'
Bram Moolenaare0de1712020-12-02 17:36:54 +0100628 assert_equal(dict3, {key: 'another'})
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100629 dict3.key = 'yet another'
Bram Moolenaare0de1712020-12-02 17:36:54 +0100630 assert_equal(dict3, {key: 'yet another'})
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100631
Bram Moolenaard24602f2020-12-20 15:20:56 +0100632 # member "any" can also be a dict and assigned to
633 var anydict: dict<any> = {nest: {}, nr: 0}
634 anydict.nest['this'] = 123
635 anydict.nest.that = 456
636 assert_equal({nest: {this: 123, that: 456}, nr: 0}, anydict)
637
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100638 var lines =<< trim END
Bram Moolenaard88dc4d2020-12-20 14:49:42 +0100639 var dd = {}
640 dd.two = 2
641 assert_equal({two: 2}, dd)
642 END
Bram Moolenaar4f5e3972020-12-21 17:30:50 +0100643 CheckDefAndScriptSuccess(lines)
644
645 lines =<< trim END
646 var d = {dd: {}}
647 d.dd[0] = 2
648 d.dd['x'] = 3
649 d.dd.y = 4
650 assert_equal({dd: {0: 2, x: 3, y: 4}}, d)
651 END
652 CheckDefAndScriptSuccess(lines)
Bram Moolenaard88dc4d2020-12-20 14:49:42 +0100653
654 lines =<< trim END
Bram Moolenaare0de1712020-12-02 17:36:54 +0100655 var dd = {one: 1}
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100656 dd.one) = 2
657 END
Bram Moolenaard88dc4d2020-12-20 14:49:42 +0100658 CheckDefFailure(lines, 'E488:', 2)
659
660 lines =<< trim END
661 var dd = {one: 1}
662 var dd.one = 2
663 END
664 CheckDefAndScriptFailure(lines, 'E1017:', 2)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200665
666 # empty key can be used
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200667 var dd = {}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200668 dd[""] = 6
Bram Moolenaare0de1712020-12-02 17:36:54 +0100669 assert_equal({['']: 6}, dd)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200670
671 # type becomes dict<any>
Bram Moolenaare0de1712020-12-02 17:36:54 +0100672 var somedict = rand() > 0 ? {a: 1, b: 2} : {a: 'a', b: 'b'}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200673
674 # assignment to script-local dict
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100675 lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200676 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200677 var test: dict<any> = {}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200678 def FillDict(): dict<any>
679 test['a'] = 43
680 return test
681 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +0100682 assert_equal({a: 43}, FillDict())
Bram Moolenaar971caf52020-09-21 22:21:42 +0200683 END
684 CheckScriptSuccess(lines)
685
686 lines =<< trim END
687 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200688 var test: dict<any>
Bram Moolenaar971caf52020-09-21 22:21:42 +0200689 def FillDict(): dict<any>
690 test['a'] = 43
691 return test
692 enddef
693 FillDict()
Bram Moolenaar3beaf9c2020-12-18 17:23:14 +0100694 assert_equal({a: 43}, test)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200695 END
Bram Moolenaar3beaf9c2020-12-18 17:23:14 +0100696 CheckScriptSuccess(lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200697
698 # assignment to global dict
699 lines =<< trim END
700 vim9script
701 g:test = {}
702 def FillDict(): dict<any>
703 g:test['a'] = 43
704 return g:test
705 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +0100706 assert_equal({a: 43}, FillDict())
Bram Moolenaar971caf52020-09-21 22:21:42 +0200707 END
708 CheckScriptSuccess(lines)
709
710 # assignment to buffer dict
711 lines =<< trim END
712 vim9script
713 b:test = {}
714 def FillDict(): dict<any>
715 b:test['a'] = 43
716 return b:test
717 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +0100718 assert_equal({a: 43}, FillDict())
Bram Moolenaar971caf52020-09-21 22:21:42 +0200719 END
720 CheckScriptSuccess(lines)
Bram Moolenaar4f5e3972020-12-21 17:30:50 +0100721
722 lines =<< trim END
723 var d = {dd: test_null_dict()}
724 d.dd[0] = 0
725 END
726 CheckDefExecFailure(lines, 'E1103:', 2)
727
728 lines =<< trim END
729 var d = {dd: 'string'}
730 d.dd[0] = 0
731 END
732 CheckDefExecFailure(lines, 'E1148:', 2)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200733enddef
734
735def Test_assignment_local()
736 # Test in a separated file in order not to the current buffer/window/tab is
737 # changed.
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200738 var script_lines: list<string> =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200739 let b:existing = 'yes'
740 let w:existing = 'yes'
741 let t:existing = 'yes'
742
743 def Test_assignment_local_internal()
744 b:newvar = 'new'
745 assert_equal('new', b:newvar)
746 assert_equal('yes', b:existing)
747 b:existing = 'no'
748 assert_equal('no', b:existing)
749 b:existing ..= 'NO'
750 assert_equal('noNO', b:existing)
751
752 w:newvar = 'new'
753 assert_equal('new', w:newvar)
754 assert_equal('yes', w:existing)
755 w:existing = 'no'
756 assert_equal('no', w:existing)
757 w:existing ..= 'NO'
758 assert_equal('noNO', w:existing)
759
760 t:newvar = 'new'
761 assert_equal('new', t:newvar)
762 assert_equal('yes', t:existing)
763 t:existing = 'no'
764 assert_equal('no', t:existing)
765 t:existing ..= 'NO'
766 assert_equal('noNO', t:existing)
767 enddef
768 call Test_assignment_local_internal()
769 END
770 CheckScriptSuccess(script_lines)
771enddef
772
773def Test_assignment_default()
Bram Moolenaar971caf52020-09-21 22:21:42 +0200774 # Test default values.
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200775 var thebool: bool
Bram Moolenaar971caf52020-09-21 22:21:42 +0200776 assert_equal(v:false, thebool)
777
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200778 var thenumber: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200779 assert_equal(0, thenumber)
780
781 if has('float')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200782 var thefloat: float
Bram Moolenaar971caf52020-09-21 22:21:42 +0200783 assert_equal(0.0, thefloat)
784 endif
785
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200786 var thestring: string
Bram Moolenaar971caf52020-09-21 22:21:42 +0200787 assert_equal('', thestring)
788
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200789 var theblob: blob
Bram Moolenaar971caf52020-09-21 22:21:42 +0200790 assert_equal(0z, theblob)
791
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200792 var Thefunc: func
Bram Moolenaar971caf52020-09-21 22:21:42 +0200793 assert_equal(test_null_function(), Thefunc)
794
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200795 var thelist: list<any>
Bram Moolenaar971caf52020-09-21 22:21:42 +0200796 assert_equal([], thelist)
797
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200798 var thedict: dict<any>
Bram Moolenaar971caf52020-09-21 22:21:42 +0200799 assert_equal({}, thedict)
800
801 if has('channel')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200802 var thejob: job
Bram Moolenaar971caf52020-09-21 22:21:42 +0200803 assert_equal(test_null_job(), thejob)
804
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200805 var thechannel: channel
Bram Moolenaar971caf52020-09-21 22:21:42 +0200806 assert_equal(test_null_channel(), thechannel)
807
808 if has('unix') && executable('cat')
809 # check with non-null job and channel, types must match
Bram Moolenaare0de1712020-12-02 17:36:54 +0100810 thejob = job_start("cat ", {})
Bram Moolenaar971caf52020-09-21 22:21:42 +0200811 thechannel = job_getchannel(thejob)
812 job_stop(thejob, 'kill')
813 endif
814 endif
815
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200816 var nr = 1234 | nr = 5678
Bram Moolenaar971caf52020-09-21 22:21:42 +0200817 assert_equal(5678, nr)
818enddef
819
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100820let scriptvar = 'init'
821
Bram Moolenaar971caf52020-09-21 22:21:42 +0200822def Test_assignment_var_list()
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100823 var lines =<< trim END
824 var v1: string
825 var v2: string
826 var vrem: list<string>
827 [v1] = ['aaa']
828 assert_equal('aaa', v1)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200829
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100830 [v1, v2] = ['one', 'two']
831 assert_equal('one', v1)
832 assert_equal('two', v2)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200833
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100834 [v1, v2; vrem] = ['one', 'two']
835 assert_equal('one', v1)
836 assert_equal('two', v2)
837 assert_equal([], vrem)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200838
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100839 [v1, v2; vrem] = ['one', 'two', 'three']
840 assert_equal('one', v1)
841 assert_equal('two', v2)
842 assert_equal(['three'], vrem)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200843
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100844 [&ts, &sw] = [3, 4]
845 assert_equal(3, &ts)
846 assert_equal(4, &sw)
847 set ts=8 sw=4
Bram Moolenaar004d9b02020-11-30 21:40:03 +0100848
849 [@a, @z] = ['aa', 'zz']
850 assert_equal('aa', @a)
851 assert_equal('zz', @z)
852
853 [$SOME_VAR, $OTHER_VAR] = ['some', 'other']
854 assert_equal('some', $SOME_VAR)
855 assert_equal('other', $OTHER_VAR)
856
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100857 [g:globalvar, b:bufvar, w:winvar, t:tabvar, v:errmsg] =
858 ['global', 'buf', 'win', 'tab', 'error']
Bram Moolenaar004d9b02020-11-30 21:40:03 +0100859 assert_equal('global', g:globalvar)
Bram Moolenaar004d9b02020-11-30 21:40:03 +0100860 assert_equal('buf', b:bufvar)
861 assert_equal('win', w:winvar)
862 assert_equal('tab', t:tabvar)
863 assert_equal('error', v:errmsg)
864 unlet g:globalvar
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100865 END
866 CheckDefAndScriptSuccess(lines)
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100867
868 [g:globalvar, s:scriptvar, b:bufvar] = ['global', 'script', 'buf']
869 assert_equal('global', g:globalvar)
870 assert_equal('script', s:scriptvar)
871 assert_equal('buf', b:bufvar)
872
873 lines =<< trim END
874 vim9script
875 var s:scriptvar = 'init'
876 [g:globalvar, s:scriptvar, w:winvar] = ['global', 'script', 'win']
877 assert_equal('global', g:globalvar)
878 assert_equal('script', s:scriptvar)
879 assert_equal('win', w:winvar)
880 END
881 CheckScriptSuccess(lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200882enddef
883
884def Test_assignment_vim9script()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200885 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200886 vim9script
887 def Func(): list<number>
888 return [1, 2]
889 enddef
Bram Moolenaar98309382020-09-27 21:58:45 +0200890 var name1: number
891 var name2: number
892 [name1, name2] =
Bram Moolenaar971caf52020-09-21 22:21:42 +0200893 Func()
Bram Moolenaar98309382020-09-27 21:58:45 +0200894 assert_equal(1, name1)
895 assert_equal(2, name2)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200896 var ll =
Bram Moolenaar971caf52020-09-21 22:21:42 +0200897 Func()
898 assert_equal([1, 2], ll)
899
900 @/ = 'text'
901 assert_equal('text', @/)
902 @0 = 'zero'
903 assert_equal('zero', @0)
904 @1 = 'one'
905 assert_equal('one', @1)
906 @9 = 'nine'
907 assert_equal('nine', @9)
908 @- = 'minus'
909 assert_equal('minus', @-)
910 if has('clipboard_working')
911 @* = 'star'
912 assert_equal('star', @*)
913 @+ = 'plus'
914 assert_equal('plus', @+)
915 endif
916
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200917 var a: number = 123
Bram Moolenaar971caf52020-09-21 22:21:42 +0200918 assert_equal(123, a)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200919 var s: string = 'yes'
Bram Moolenaar971caf52020-09-21 22:21:42 +0200920 assert_equal('yes', s)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200921 var b: number = 42
Bram Moolenaar971caf52020-09-21 22:21:42 +0200922 assert_equal(42, b)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200923 var w: number = 43
Bram Moolenaar971caf52020-09-21 22:21:42 +0200924 assert_equal(43, w)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200925 var t: number = 44
Bram Moolenaar971caf52020-09-21 22:21:42 +0200926 assert_equal(44, t)
Bram Moolenaar7a66a172020-10-16 19:56:12 +0200927
928 var to_var = 0
929 to_var = 3
930 assert_equal(3, to_var)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200931 END
932 CheckScriptSuccess(lines)
Bram Moolenaar06f9c692020-09-27 21:27:40 +0200933
934 lines =<< trim END
935 vim9script
936 var n: number
937 def Func()
938 n = 'string'
939 enddef
940 defcompile
941 END
942 CheckScriptFailure(lines, 'E1012: Type mismatch; expected number but got string')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200943enddef
944
945def Mess(): string
946 v:foldstart = 123
947 return 'xxx'
948enddef
949
950def Test_assignment_failure()
Bram Moolenaar98309382020-09-27 21:58:45 +0200951 CheckDefFailure(['var name=234'], 'E1004:')
952 CheckDefFailure(['var name =234'], 'E1004:')
953 CheckDefFailure(['var name= 234'], 'E1004:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200954
Bram Moolenaar98309382020-09-27 21:58:45 +0200955 CheckScriptFailure(['vim9script', 'var name=234'], 'E1004:')
956 CheckScriptFailure(['vim9script', 'var name=234'], "before and after '='")
957 CheckScriptFailure(['vim9script', 'var name =234'], 'E1004:')
958 CheckScriptFailure(['vim9script', 'var name= 234'], 'E1004:')
959 CheckScriptFailure(['vim9script', 'var name = 234', 'name+=234'], 'E1004:')
960 CheckScriptFailure(['vim9script', 'var name = 234', 'name+=234'], "before and after '+='")
961 CheckScriptFailure(['vim9script', 'var name = "x"', 'name..="y"'], 'E1004:')
962 CheckScriptFailure(['vim9script', 'var name = "x"', 'name..="y"'], "before and after '..='")
Bram Moolenaar971caf52020-09-21 22:21:42 +0200963
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200964 CheckDefFailure(['var true = 1'], 'E1034:')
965 CheckDefFailure(['var false = 1'], 'E1034:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200966
967 CheckDefFailure(['[a; b; c] = g:list'], 'E452:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200968 CheckDefExecFailure(['var a: number',
Bram Moolenaar971caf52020-09-21 22:21:42 +0200969 '[a] = test_null_list()'], 'E1093:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200970 CheckDefExecFailure(['var a: number',
Bram Moolenaar971caf52020-09-21 22:21:42 +0200971 '[a] = []'], 'E1093:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200972 CheckDefExecFailure(['var x: number',
973 'var y: number',
Bram Moolenaar971caf52020-09-21 22:21:42 +0200974 '[x, y] = [1]'], 'E1093:')
Bram Moolenaar352134b2020-10-17 22:04:08 +0200975 CheckDefExecFailure(['var x: string',
976 'var y: string',
977 '[x, y] = ["x"]'], 'E1093:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200978 CheckDefExecFailure(['var x: number',
979 'var y: number',
980 'var z: list<number>',
Bram Moolenaar971caf52020-09-21 22:21:42 +0200981 '[x, y; z] = [1]'], 'E1093:')
982
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200983 CheckDefFailure(['var somevar'], "E1022:")
984 CheckDefFailure(['var &tabstop = 4'], 'E1052:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200985 CheckDefFailure(['&g:option = 5'], 'E113:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200986 CheckScriptFailure(['vim9script', 'var &tabstop = 4'], 'E1052:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200987
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200988 CheckDefFailure(['var $VAR = 5'], 'E1016: Cannot declare an environment variable:')
989 CheckScriptFailure(['vim9script', 'var $ENV = "xxx"'], 'E1016:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200990
991 if has('dnd')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200992 CheckDefFailure(['var @~ = 5'], 'E1066:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200993 else
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200994 CheckDefFailure(['var @~ = 5'], 'E354:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200995 CheckDefFailure(['@~ = 5'], 'E354:')
996 endif
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200997 CheckDefFailure(['var @a = 5'], 'E1066:')
998 CheckDefFailure(['var @/ = "x"'], 'E1066:')
999 CheckScriptFailure(['vim9script', 'var @a = "abc"'], 'E1066:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001000
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001001 CheckDefFailure(['var g:var = 5'], 'E1016: Cannot declare a global variable:')
1002 CheckDefFailure(['var w:var = 5'], 'E1016: Cannot declare a window variable:')
1003 CheckDefFailure(['var b:var = 5'], 'E1016: Cannot declare a buffer variable:')
1004 CheckDefFailure(['var t:var = 5'], 'E1016: Cannot declare a tab variable:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001005
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001006 CheckDefFailure(['var anr = 4', 'anr ..= "text"'], 'E1019:')
1007 CheckDefFailure(['var xnr += 4'], 'E1020:', 1)
1008 CheckScriptFailure(['vim9script', 'var xnr += 4'], 'E1020:')
1009 CheckDefFailure(["var xnr = xnr + 1"], 'E1001:', 1)
1010 CheckScriptFailure(['vim9script', 'var xnr = xnr + 4'], 'E121:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001011
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001012 CheckScriptFailure(['vim9script', 'def Func()', 'var dummy = s:notfound', 'enddef', 'defcompile'], 'E1108:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001013
Bram Moolenaar98309382020-09-27 21:58:45 +02001014 CheckDefFailure(['var name: list<string> = [123]'], 'expected list<string> but got list<number>')
1015 CheckDefFailure(['var name: list<number> = ["xx"]'], 'expected list<number> but got list<string>')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001016
Bram Moolenaare0de1712020-12-02 17:36:54 +01001017 CheckDefFailure(['var name: dict<string> = {key: 123}'], 'expected dict<string> but got dict<number>')
1018 CheckDefFailure(['var name: dict<number> = {key: "xx"}'], 'expected dict<number> but got dict<string>')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001019
Bram Moolenaar98309382020-09-27 21:58:45 +02001020 CheckDefFailure(['var name = feedkeys("0")'], 'E1031:')
1021 CheckDefFailure(['var name: number = feedkeys("0")'], 'expected number but got void')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001022
Bram Moolenaar98309382020-09-27 21:58:45 +02001023 CheckDefFailure(['var name: dict <number>'], 'E1068:')
1024 CheckDefFailure(['var name: dict<number'], 'E1009:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001025
1026 assert_fails('s/^/\=Mess()/n', 'E794:')
Bram Moolenaar98309382020-09-27 21:58:45 +02001027 CheckDefFailure(['var name: dict<number'], 'E1009:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001028
1029 CheckDefFailure(['w:foo: number = 10'],
1030 'E488: Trailing characters: : number = 1')
1031 CheckDefFailure(['t:foo: bool = true'],
1032 'E488: Trailing characters: : bool = true')
1033 CheckDefFailure(['b:foo: string = "x"'],
1034 'E488: Trailing characters: : string = "x"')
1035 CheckDefFailure(['g:foo: number = 123'],
1036 'E488: Trailing characters: : number = 123')
1037enddef
1038
1039def Test_assign_list()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001040 var l: list<string> = []
Bram Moolenaar971caf52020-09-21 22:21:42 +02001041 l[0] = 'value'
1042 assert_equal('value', l[0])
1043
1044 l[1] = 'asdf'
1045 assert_equal('value', l[0])
1046 assert_equal('asdf', l[1])
1047 assert_equal('asdf', l[-1])
1048 assert_equal('value', l[-2])
1049
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001050 var nrl: list<number> = []
Bram Moolenaar971caf52020-09-21 22:21:42 +02001051 for i in range(5)
1052 nrl[i] = i
1053 endfor
1054 assert_equal([0, 1, 2, 3, 4], nrl)
Bram Moolenaar334a8b42020-10-19 16:07:42 +02001055
1056 CheckDefFailure(["var l: list<number> = ['', true]"], 'E1012: Type mismatch; expected list<number> but got list<any>', 1)
1057 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 +02001058enddef
1059
1060def Test_assign_dict()
Bram Moolenaarc3689572021-01-01 19:40:02 +01001061 var lines =<< trim END
1062 var d: dict<string> = {}
1063 d['key'] = 'value'
1064 assert_equal('value', d['key'])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001065
Bram Moolenaarc3689572021-01-01 19:40:02 +01001066 d[123] = 'qwerty'
1067 assert_equal('qwerty', d[123])
1068 assert_equal('qwerty', d['123'])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001069
Bram Moolenaarc3689572021-01-01 19:40:02 +01001070 var nrd: dict<number> = {}
1071 for i in range(3)
1072 nrd[i] = i
1073 endfor
1074 assert_equal({0: 0, 1: 1, 2: 2}, nrd)
1075
1076 d.somekey = 'someval'
1077 assert_equal({key: 'value', '123': 'qwerty', somekey: 'someval'}, d)
1078 # unlet d.somekey
1079 # assert_equal({key: 'value', '123': 'qwerty'}, d)
1080 END
1081 CheckDefAndScriptSuccess(lines)
1082
1083 # TODO: move to above once "unlet d.somekey" in :def is implemented
1084 lines =<< trim END
1085 vim9script
1086 var d: dict<string> = {}
1087 d['key'] = 'value'
1088 d.somekey = 'someval'
1089 assert_equal({key: 'value', somekey: 'someval'}, d)
1090 unlet d.somekey
1091 assert_equal({key: 'value'}, d)
1092 END
1093 CheckScriptSuccess(lines)
Bram Moolenaar334a8b42020-10-19 16:07:42 +02001094
Bram Moolenaare0de1712020-12-02 17:36:54 +01001095 CheckDefFailure(["var d: dict<number> = {a: '', b: true}"], 'E1012: Type mismatch; expected dict<number> but got dict<any>', 1)
1096 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 +02001097enddef
1098
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001099def Test_assign_dict_unknown_type()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001100 var lines =<< trim END
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001101 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001102 var mylist = []
Bram Moolenaare0de1712020-12-02 17:36:54 +01001103 mylist += [{one: 'one'}]
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001104 def Func()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001105 var dd = mylist[0]
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001106 assert_equal('one', dd.one)
1107 enddef
1108 Func()
1109 END
1110 CheckScriptSuccess(lines)
1111
Bram Moolenaarce2c5442020-11-28 21:21:17 +01001112 lines =<< trim END
1113 vim9script
1114 var mylist = [[]]
Bram Moolenaare0de1712020-12-02 17:36:54 +01001115 mylist[0] += [{one: 'one'}]
Bram Moolenaarce2c5442020-11-28 21:21:17 +01001116 def Func()
1117 var dd = mylist[0][0]
1118 assert_equal('one', dd.one)
1119 enddef
1120 Func()
1121 END
1122 CheckScriptSuccess(lines)
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001123enddef
1124
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001125def Test_assign_dict_with_op()
1126 var lines =<< trim END
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001127 var ds: dict<string> = {a: 'x'}
1128 ds['a'] ..= 'y'
1129 ds.a ..= 'z'
1130 assert_equal('xyz', ds.a)
1131
1132 var dn: dict<number> = {a: 9}
1133 dn['a'] += 2
1134 assert_equal(11, dn.a)
1135 dn.a += 2
1136 assert_equal(13, dn.a)
1137
1138 dn['a'] -= 3
1139 assert_equal(10, dn.a)
1140 dn.a -= 2
1141 assert_equal(8, dn.a)
1142
1143 dn['a'] *= 2
1144 assert_equal(16, dn.a)
1145 dn.a *= 2
1146 assert_equal(32, dn.a)
1147
1148 dn['a'] /= 3
1149 assert_equal(10, dn.a)
1150 dn.a /= 2
1151 assert_equal(5, dn.a)
1152
1153 dn['a'] %= 3
1154 assert_equal(2, dn.a)
1155 dn.a %= 6
1156 assert_equal(2, dn.a)
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001157
1158 var dd: dict<dict<list<any>>>
1159 dd.a = {}
1160 dd.a.b = [0]
1161 dd.a.b += [1]
1162 assert_equal({a: {b: [0, 1]}}, dd)
Bram Moolenaar0d351a02021-04-05 21:05:26 +02001163
1164 var dab = {a: ['b']}
1165 dab.a[0] ..= 'c'
1166 assert_equal({a: ['bc']}, dab)
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001167 END
Bram Moolenaare42939a2021-04-05 17:11:17 +02001168 CheckDefAndScriptSuccess(lines)
1169enddef
1170
1171def Test_assign_list_with_op()
1172 var lines =<< trim END
1173 var ls: list<string> = ['x']
1174 ls[0] ..= 'y'
1175 assert_equal('xy', ls[0])
1176
1177 var ln: list<number> = [9]
1178 ln[0] += 2
1179 assert_equal(11, ln[0])
1180
1181 ln[0] -= 3
1182 assert_equal(8, ln[0])
1183
1184 ln[0] *= 2
1185 assert_equal(16, ln[0])
1186
1187 ln[0] /= 3
1188 assert_equal(5, ln[0])
1189
1190 ln[0] %= 3
1191 assert_equal(2, ln[0])
1192 END
1193 CheckDefAndScriptSuccess(lines)
1194enddef
1195
1196def Test_assign_with_op_fails()
1197 var lines =<< trim END
1198 var s = 'abc'
1199 s[1] += 'x'
1200 END
1201 CheckDefAndScriptFailure2(lines, 'E1141:', 'E689:', 2)
1202
1203 lines =<< trim END
1204 var s = 'abc'
1205 s[1] ..= 'x'
1206 END
1207 CheckDefAndScriptFailure2(lines, 'E1141:', 'E689:', 2)
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001208
1209 lines =<< trim END
1210 var dd: dict<dict<list<any>>>
1211 dd.a = {}
1212 dd.a.b += [1]
1213 END
1214 CheckDefExecAndScriptFailure(lines, 'E716:', 3)
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001215enddef
1216
Bram Moolenaar971caf52020-09-21 22:21:42 +02001217def Test_assign_lambda()
1218 # check if assign a lambda to a variable which type is func or any.
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001219 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +02001220 vim9script
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001221 var FuncRef = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001222 assert_equal(123, FuncRef())
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001223 var FuncRef_Func: func = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001224 assert_equal(123, FuncRef_Func())
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001225 var FuncRef_Any: any = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001226 assert_equal(123, FuncRef_Any())
Bram Moolenaar06409502021-02-17 17:00:27 +01001227 var FuncRef_Number: func(): number = () => 321
1228 assert_equal(321, FuncRef_Number())
Bram Moolenaar971caf52020-09-21 22:21:42 +02001229 END
1230 CheckScriptSuccess(lines)
Bram Moolenaar328eac22021-01-07 19:23:08 +01001231
1232 lines =<< trim END
1233 var Ref: func(number)
1234 Ref = (j) => !j
1235 END
Bram Moolenaar06409502021-02-17 17:00:27 +01001236 CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected func(number) but got func(any): bool')
Bram Moolenaarc7dac852021-02-17 18:49:11 +01001237
1238 lines =<< trim END
1239 echo filter([1, 2, 3], (_, v: string) => v + 1)
1240 END
1241 CheckDefAndScriptFailure(lines, 'E1051:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001242enddef
1243
Bram Moolenaarc0e29012020-09-27 14:22:48 +02001244def Test_heredoc()
1245 var lines =<< trim END # comment
1246 text
1247 END
1248 assert_equal(['text'], lines)
1249
1250 CheckDefFailure(['var lines =<< trim END X', 'END'], 'E488:')
1251 CheckDefFailure(['var lines =<< trim END " comment', 'END'], 'E488:')
Bram Moolenaar2914a202020-09-27 18:24:03 +02001252
1253 lines =<< trim [END]
1254 def Func()
1255 var&lines =<< trim END
1256 x
1257 x
Bram Moolenaarb5b94802020-12-13 17:50:20 +01001258 enddef
1259 defcompile
1260 [END]
1261 CheckScriptFailure(lines, 'E1145: Missing heredoc end marker: END')
1262 delfunc! g:Func
1263
1264 lines =<< trim [END]
1265 def Func()
1266 var lines =<< trim END
1267 x
1268 x
Bram Moolenaar2914a202020-09-27 18:24:03 +02001269 x
1270 x
1271 x
1272 x
1273 x
1274 x
1275 enddef
1276 call Func()
1277 [END]
Bram Moolenaarb5b94802020-12-13 17:50:20 +01001278 CheckScriptFailure(lines, 'E1145: Missing heredoc end marker: END')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001279 delfunc! g:Func
Bram Moolenaarc0e29012020-09-27 14:22:48 +02001280enddef
1281
Bram Moolenaar083966f2020-12-08 11:29:40 +01001282def Test_var_func_call()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001283 var lines =<< trim END
1284 vim9script
1285 func GetValue()
1286 if exists('g:count')
1287 let g:count += 1
1288 else
1289 let g:count = 1
1290 endif
1291 return 'this'
1292 endfunc
1293 var val: string = GetValue()
1294 # env var is always a string
1295 var env = $TERM
1296 END
1297 writefile(lines, 'Xfinished')
1298 source Xfinished
1299 # GetValue() is not called during discovery phase
1300 assert_equal(1, g:count)
1301
1302 unlet g:count
1303 delete('Xfinished')
1304enddef
1305
Bram Moolenaar083966f2020-12-08 11:29:40 +01001306def Test_var_missing_type()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001307 var lines =<< trim END
1308 vim9script
1309 var name = g:unknown
1310 END
1311 CheckScriptFailure(lines, 'E121:')
1312
1313 lines =<< trim END
1314 vim9script
1315 var nr: number = 123
1316 var name = nr
1317 END
1318 CheckScriptSuccess(lines)
1319enddef
1320
Bram Moolenaar083966f2020-12-08 11:29:40 +01001321def Test_var_declaration()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001322 var lines =<< trim END
1323 vim9script
1324 var name: string
1325 g:var_uninit = name
1326 name = 'text'
1327 g:var_test = name
1328 # prefixing s: is optional
1329 s:name = 'prefixed'
1330 g:var_prefixed = s:name
1331
Bram Moolenaar7257af42020-12-08 22:08:51 +01001332 const FOO: number = 123
1333 assert_equal(123, FOO)
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001334 const FOOS = 'foos'
1335 assert_equal('foos', FOOS)
1336 final FLIST = [1]
1337 assert_equal([1], FLIST)
1338 FLIST[0] = 11
1339 assert_equal([11], FLIST)
1340
1341 const g:FOO: number = 321
1342 assert_equal(321, g:FOO)
1343 const g:FOOS = 'gfoos'
1344 assert_equal('gfoos', g:FOOS)
1345 final g:FLIST = [2]
1346 assert_equal([2], g:FLIST)
1347 g:FLIST[0] = 22
1348 assert_equal([22], g:FLIST)
1349
Bram Moolenaard877a572021-04-01 19:42:48 +02001350 def SetGlobalConst()
1351 const g:globConst = 123
1352 enddef
1353 SetGlobalConst()
1354 assert_equal(123, g:globConst)
1355 assert_true(islocked('g:globConst'))
1356
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001357 const w:FOO: number = 46
1358 assert_equal(46, w:FOO)
1359 const w:FOOS = 'wfoos'
1360 assert_equal('wfoos', w:FOOS)
1361 final w:FLIST = [3]
1362 assert_equal([3], w:FLIST)
1363 w:FLIST[0] = 33
1364 assert_equal([33], w:FLIST)
Bram Moolenaar7257af42020-12-08 22:08:51 +01001365
Bram Moolenaar55759b52020-09-30 22:59:42 +02001366 var s:other: number
1367 other = 1234
1368 g:other_var = other
1369
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01001370 var xyz: string # comment
1371
Bram Moolenaar55759b52020-09-30 22:59:42 +02001372 # type is inferred
Bram Moolenaar3862ea32021-01-01 21:05:55 +01001373 var s:dict = {['a']: 222}
Bram Moolenaar55759b52020-09-30 22:59:42 +02001374 def GetDictVal(key: any)
1375 g:dict_val = s:dict[key]
1376 enddef
1377 GetDictVal('a')
Bram Moolenaar08251752021-01-11 21:20:18 +01001378
1379 final adict: dict<string> = {}
1380 def ChangeAdict()
1381 adict.foo = 'foo'
1382 enddef
1383 ChangeAdict()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001384 END
1385 CheckScriptSuccess(lines)
1386 assert_equal('', g:var_uninit)
1387 assert_equal('text', g:var_test)
1388 assert_equal('prefixed', g:var_prefixed)
1389 assert_equal(1234, g:other_var)
1390 assert_equal(222, g:dict_val)
1391
1392 unlet g:var_uninit
1393 unlet g:var_test
1394 unlet g:var_prefixed
1395 unlet g:other_var
Bram Moolenaar6e50ec22021-04-03 19:32:44 +02001396 unlet g:globConst
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001397 unlet g:FOO
1398 unlet g:FOOS
1399 unlet g:FLIST
1400 unlet w:FOO
1401 unlet w:FOOS
1402 unlet w:FLIST
Bram Moolenaar55759b52020-09-30 22:59:42 +02001403enddef
1404
Bram Moolenaar083966f2020-12-08 11:29:40 +01001405def Test_var_declaration_fails()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001406 var lines =<< trim END
1407 vim9script
1408 final var: string
1409 END
1410 CheckScriptFailure(lines, 'E1125:')
1411
1412 lines =<< trim END
1413 vim9script
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001414 const g:constvar = 'string'
1415 g:constvar = 'xx'
1416 END
1417 CheckScriptFailure(lines, 'E741:')
1418 unlet g:constvar
1419
1420 lines =<< trim END
1421 vim9script
Bram Moolenaarf5906aa2021-04-02 14:35:15 +02001422 var name = 'one'
1423 lockvar name
1424 def SetLocked()
1425 name = 'two'
1426 enddef
1427 SetLocked()
1428 END
Bram Moolenaardcf29ac2021-04-02 14:44:02 +02001429 CheckScriptFailure(lines, 'E741: Value is locked: name', 1)
1430
1431 lines =<< trim END
1432 let s:legacy = 'one'
1433 lockvar s:legacy
1434 def SetLocked()
1435 s:legacy = 'two'
1436 enddef
1437 call SetLocked()
1438 END
1439 CheckScriptFailure(lines, 'E741: Value is locked: s:legacy', 1)
Bram Moolenaarf5906aa2021-04-02 14:35:15 +02001440
1441 lines =<< trim END
1442 vim9script
Bram Moolenaard877a572021-04-01 19:42:48 +02001443 def SetGlobalConst()
1444 const g:globConst = 123
1445 enddef
1446 SetGlobalConst()
1447 g:globConst = 234
1448 END
Bram Moolenaar6e50ec22021-04-03 19:32:44 +02001449 CheckScriptFailure(lines, 'E741: Value is locked: g:globConst', 6)
Bram Moolenaard877a572021-04-01 19:42:48 +02001450 unlet g:globConst
1451
1452 lines =<< trim END
1453 vim9script
Bram Moolenaar08251752021-01-11 21:20:18 +01001454 const cdict: dict<string> = {}
1455 def Change()
1456 cdict.foo = 'foo'
1457 enddef
1458 defcompile
1459 END
1460 CheckScriptFailure(lines, 'E46:')
1461
1462 lines =<< trim END
1463 vim9script
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001464 final w:finalvar = [9]
1465 w:finalvar = [8]
1466 END
1467 CheckScriptFailure(lines, 'E1122:')
1468 unlet w:finalvar
1469
1470 lines =<< trim END
1471 vim9script
Bram Moolenaar55759b52020-09-30 22:59:42 +02001472 const var: string
1473 END
1474 CheckScriptFailure(lines, 'E1021:')
1475
1476 lines =<< trim END
1477 vim9script
1478 var 9var: string
1479 END
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01001480 CheckScriptFailure(lines, 'E488:')
Bram Moolenaar7257af42020-12-08 22:08:51 +01001481
1482 CheckDefFailure(['var foo.bar = 2'], 'E1087:')
1483 CheckDefFailure(['var foo[3] = 2'], 'E1087:')
1484 CheckDefFailure(['const foo: number'], 'E1021:')
Bram Moolenaar55759b52020-09-30 22:59:42 +02001485enddef
1486
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001487def Test_script_local_in_legacy()
1488 # OK to define script-local later when prefixed with s:
1489 var lines =<< trim END
1490 def SetLater()
1491 s:legacy = 'two'
1492 enddef
1493 defcompile
1494 let s:legacy = 'one'
1495 call SetLater()
1496 call assert_equal('two', s:legacy)
1497 END
1498 CheckScriptSuccess(lines)
1499
1500 # OK to leave out s: prefix when script-local already defined
1501 lines =<< trim END
1502 let s:legacy = 'one'
1503 def SetNoPrefix()
1504 legacy = 'two'
1505 enddef
1506 call SetNoPrefix()
1507 call assert_equal('two', s:legacy)
1508 END
1509 CheckScriptSuccess(lines)
1510
1511 # Not OK to leave out s: prefix when script-local defined later
1512 lines =<< trim END
1513 def SetLaterNoPrefix()
1514 legacy = 'two'
1515 enddef
1516 defcompile
1517 let s:legacy = 'one'
1518 END
1519 CheckScriptFailure(lines, 'E476:', 1)
1520enddef
1521
Bram Moolenaar083966f2020-12-08 11:29:40 +01001522def Test_var_type_check()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001523 var lines =<< trim END
1524 vim9script
1525 var name: string
1526 name = 1234
1527 END
1528 CheckScriptFailure(lines, 'E1012:')
1529
1530 lines =<< trim END
1531 vim9script
1532 var name:string
1533 END
1534 CheckScriptFailure(lines, 'E1069:')
1535
1536 lines =<< trim END
1537 vim9script
1538 var name: asdf
1539 END
1540 CheckScriptFailure(lines, 'E1010:')
1541
1542 lines =<< trim END
1543 vim9script
1544 var s:l: list<number>
1545 s:l = []
1546 END
1547 CheckScriptSuccess(lines)
1548
1549 lines =<< trim END
1550 vim9script
1551 var s:d: dict<number>
1552 s:d = {}
1553 END
1554 CheckScriptSuccess(lines)
1555enddef
1556
1557let g:dict_number = #{one: 1, two: 2}
1558
Bram Moolenaar083966f2020-12-08 11:29:40 +01001559def Test_var_list_dict_type()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001560 var ll: list<number>
1561 ll = [1, 2, 2, 3, 3, 3]->uniq()
1562 ll->assert_equal([1, 2, 3])
1563
1564 var dd: dict<number>
1565 dd = g:dict_number
1566 dd->assert_equal(g:dict_number)
1567
1568 var lines =<< trim END
1569 var ll: list<number>
1570 ll = [1, 2, 3]->map('"one"')
1571 END
Bram Moolenaar70250fb2021-01-16 19:01:53 +01001572 CheckDefExecFailure(lines, 'E1012: Type mismatch; expected number but got string')
Bram Moolenaar55759b52020-09-30 22:59:42 +02001573enddef
1574
Bram Moolenaar083966f2020-12-08 11:29:40 +01001575def Test_cannot_use_let()
1576 CheckDefAndScriptFailure(['let a = 34'], 'E1126:', 1)
1577enddef
1578
Bram Moolenaar55759b52020-09-30 22:59:42 +02001579def Test_unlet()
1580 g:somevar = 'yes'
1581 assert_true(exists('g:somevar'))
1582 unlet g:somevar
1583 assert_false(exists('g:somevar'))
1584 unlet! g:somevar
1585
1586 # also works for script-local variable in legacy Vim script
1587 s:somevar = 'legacy'
1588 assert_true(exists('s:somevar'))
1589 unlet s:somevar
1590 assert_false(exists('s:somevar'))
1591 unlet! s:somevar
1592
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001593 CheckDefExecFailure([
1594 'var dd = 111',
1595 'unlet dd',
1596 ], 'E1081:', 2)
1597
Bram Moolenaar752fc692021-01-04 21:57:11 +01001598 # dict unlet
1599 var dd = {a: 1, b: 2, c: 3}
1600 unlet dd['a']
1601 unlet dd.c
1602 assert_equal({b: 2}, dd)
1603
1604 # list unlet
1605 var ll = [1, 2, 3, 4]
1606 unlet ll[1]
1607 unlet ll[-1]
1608 assert_equal([1, 3], ll)
1609
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001610 ll = [1, 2, 3, 4]
1611 unlet ll[0 : 1]
1612 assert_equal([3, 4], ll)
1613
1614 ll = [1, 2, 3, 4]
1615 unlet ll[2 : 8]
1616 assert_equal([1, 2], ll)
1617
1618 ll = [1, 2, 3, 4]
1619 unlet ll[-2 : -1]
1620 assert_equal([1, 2], ll)
1621
1622 CheckDefFailure([
1623 'var ll = [1, 2]',
1624 'll[1 : 2] = 7',
1625 ], 'E1165:', 2)
1626 CheckDefFailure([
1627 'var dd = {a: 1}',
1628 'unlet dd["a" : "a"]',
1629 ], 'E1166:', 2)
1630 CheckDefExecFailure([
1631 'unlet g:adict[0 : 1]',
1632 ], 'E1148:', 1)
1633 CheckDefFailure([
1634 'var ll = [1, 2]',
1635 'unlet ll[0:1]',
1636 ], 'E1004:', 2)
1637 CheckDefFailure([
1638 'var ll = [1, 2]',
1639 'unlet ll[0 :1]',
1640 ], 'E1004:', 2)
1641 CheckDefFailure([
1642 'var ll = [1, 2]',
1643 'unlet ll[0: 1]',
1644 ], 'E1004:', 2)
1645
1646 CheckDefFailure([
1647 'var ll = [1, 2]',
1648 'unlet ll["x" : 1]',
1649 ], 'E1012:', 2)
1650 CheckDefFailure([
1651 'var ll = [1, 2]',
1652 'unlet ll[0 : "x"]',
1653 ], 'E1012:', 2)
1654
Bram Moolenaar752fc692021-01-04 21:57:11 +01001655 # list of dict unlet
1656 var dl = [{a: 1, b: 2}, {c: 3}]
1657 unlet dl[0]['b']
1658 assert_equal([{a: 1}, {c: 3}], dl)
1659
1660 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001661 'var ll = test_null_list()',
1662 'unlet ll[0]',
1663 ], 'E684:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001664 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001665 'var ll = [1]',
1666 'unlet ll[2]',
1667 ], 'E684:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001668 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001669 'var ll = [1]',
1670 'unlet ll[g:astring]',
Bram Moolenaarf30a14d2021-01-17 21:51:24 +01001671 ], 'E1012:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001672 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001673 'var dd = test_null_dict()',
1674 'unlet dd["a"]',
1675 ], 'E716:', 2)
1676 CheckDefExecFailure([
1677 'var dd = {a: 1}',
1678 'unlet dd["b"]',
1679 ], 'E716:', 2)
1680 CheckDefExecFailure([
1681 'var dd = {a: 1}',
1682 'unlet dd[g:alist]',
1683 ], 'E1105:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001684
Bram Moolenaar2ef951d2021-01-03 20:55:26 +01001685 # can compile unlet before variable exists
Bram Moolenaar752fc692021-01-04 21:57:11 +01001686 g:someDict = {key: 'val'}
1687 var k = 'key'
1688 unlet g:someDict[k]
1689 assert_equal({}, g:someDict)
1690 unlet g:someDict
1691 assert_false(exists('g:someDict'))
Bram Moolenaar2ef951d2021-01-03 20:55:26 +01001692
Bram Moolenaar55759b52020-09-30 22:59:42 +02001693 CheckScriptFailure([
1694 'vim9script',
1695 'var svar = 123',
1696 'unlet svar',
1697 ], 'E1081:')
1698 CheckScriptFailure([
1699 'vim9script',
1700 'var svar = 123',
1701 'unlet s:svar',
1702 ], 'E1081:')
1703 CheckScriptFailure([
1704 'vim9script',
1705 'var svar = 123',
1706 'def Func()',
1707 ' unlet svar',
1708 'enddef',
1709 'defcompile',
1710 ], 'E1081:')
1711 CheckScriptFailure([
1712 'vim9script',
1713 'var svar = 123',
Bram Moolenaar9aed7292020-12-18 15:38:00 +01001714 'func Func()',
1715 ' unlet s:svar',
1716 'endfunc',
1717 'Func()',
1718 ], 'E1081:')
1719 CheckScriptFailure([
1720 'vim9script',
1721 'var svar = 123',
Bram Moolenaar55759b52020-09-30 22:59:42 +02001722 'def Func()',
1723 ' unlet s:svar',
1724 'enddef',
1725 'defcompile',
1726 ], 'E1081:')
1727
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001728 writefile(['vim9script', 'export var svar = 1234'], 'XunletExport.vim')
1729 var lines =<< trim END
1730 vim9script
1731 import svar from './XunletExport.vim'
1732 def UnletSvar()
1733 unlet svar
1734 enddef
1735 defcompile
1736 END
1737 CheckScriptFailure(lines, 'E1081:', 1)
1738 delete('XunletExport.vim')
1739
Bram Moolenaar55759b52020-09-30 22:59:42 +02001740 $ENVVAR = 'foobar'
1741 assert_equal('foobar', $ENVVAR)
1742 unlet $ENVVAR
1743 assert_equal('', $ENVVAR)
1744enddef
1745
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01001746def Test_expr_error_no_assign()
1747 var lines =<< trim END
1748 vim9script
1749 var x = invalid
1750 echo x
1751 END
1752 CheckScriptFailureList(lines, ['E121:', 'E121:'])
1753
1754 lines =<< trim END
1755 vim9script
1756 var x = 1 / 0
1757 echo x
1758 END
1759 CheckScriptFailureList(lines, ['E1154:', 'E121:'])
1760
1761 lines =<< trim END
1762 vim9script
1763 var x = 1 % 0
1764 echo x
1765 END
1766 CheckScriptFailureList(lines, ['E1154:', 'E121:'])
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01001767
1768 lines =<< trim END
1769 var x: string 'string'
1770 END
1771 CheckDefAndScriptFailure(lines, 'E488:')
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01001772enddef
1773
1774
Bram Moolenaar17126b12021-01-07 22:03:02 +01001775def Test_assign_command_modifier()
1776 var lines =<< trim END
Bram Moolenaar9e0f8832021-01-09 12:09:22 +01001777 var verbose = 0
1778 verbose = 1
1779 assert_equal(1, verbose)
1780 silent verbose = 2
1781 assert_equal(2, verbose)
1782 silent verbose += 2
1783 assert_equal(4, verbose)
1784 silent verbose -= 1
1785 assert_equal(3, verbose)
Bram Moolenaar17126b12021-01-07 22:03:02 +01001786
Bram Moolenaar9e0f8832021-01-09 12:09:22 +01001787 var topleft = {one: 1}
1788 sandbox topleft.one = 3
1789 assert_equal({one: 3}, topleft)
1790 leftabove topleft[' '] = 4
1791 assert_equal({one: 3, ' ': 4}, topleft)
1792
1793 var x: number
1794 var y: number
1795 silent [x, y] = [1, 2]
1796 assert_equal(1, x)
1797 assert_equal(2, y)
Bram Moolenaar17126b12021-01-07 22:03:02 +01001798 END
1799 CheckDefAndScriptSuccess(lines)
1800enddef
1801
Bram Moolenaar32154662021-03-28 21:14:06 +02001802def Test_script_funcref_case()
1803 var lines =<< trim END
1804 var Len = (s: string): number => len(s) + 1
1805 assert_equal(5, Len('asdf'))
1806 END
1807 CheckDefAndScriptSuccess(lines)
1808
1809 lines =<< trim END
1810 var len = (s: string): number => len(s) + 1
1811 END
1812 CheckDefAndScriptFailure(lines, 'E704:')
1813
1814 lines =<< trim END
1815 vim9script
1816 var s:Len = (s: string): number => len(s) + 2
1817 assert_equal(6, Len('asdf'))
1818 END
1819 CheckScriptSuccess(lines)
1820
1821 lines =<< trim END
1822 vim9script
1823 var s:len = (s: string): number => len(s) + 1
1824 END
1825 CheckScriptFailure(lines, 'E704:')
1826enddef
1827
Bram Moolenaar55759b52020-09-30 22:59:42 +02001828
Bram Moolenaar971caf52020-09-21 22:21:42 +02001829" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker