blob: fa418b2ce289b9480550bb1b8b52dd8d53459dd4 [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 Moolenaaraa1959b2021-04-15 22:13:39 +0200252def Test_skipped_assignment()
253 var lines =<< trim END
254 for x in []
255 var i: number = 1
256 while false
257 i += 1
258 endwhile
259 endfor
260 END
261 CheckDefAndScriptSuccess(lines)
262enddef
263
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100264def Test_assign_unpack()
265 var lines =<< trim END
266 var v1: number
267 var v2: number
268 [v1, v2] = [1, 2]
269 assert_equal(1, v1)
270 assert_equal(2, v2)
Bram Moolenaarf93bbd02021-04-10 22:35:43 +0200271
272 [v1, _, v2, _] = [1, 99, 2, 77]
273 assert_equal(1, v1)
274 assert_equal(2, v2)
275
276 [v1, v2; _] = [1, 2, 3, 4, 5]
277 assert_equal(1, v1)
278 assert_equal(2, v2)
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100279 END
280 CheckDefAndScriptSuccess(lines)
281
282 lines =<< trim END
283 var v1: number
284 var v2: number
285 [v1, v2] =
286 END
287 CheckDefFailure(lines, 'E1097:', 5)
288
289 lines =<< trim END
290 var v1: number
291 var v2: number
292 [v1, v2] = xxx
293 END
294 CheckDefFailure(lines, 'E1001:', 3)
295
296 lines =<< trim END
297 var v1: number
298 var v2: number
299 [v1, v2] = popup_clear()
300 END
301 CheckDefFailure(lines, 'E1031:', 3)
302
303 lines =<< trim END
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100304 [v1, v2] = [1, 2]
305 END
306 CheckDefFailure(lines, 'E1089', 1)
307 CheckScriptFailure(['vim9script'] + lines, 'E1089', 2)
308
309 lines =<< trim END
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100310 var v1: number
311 var v2: number
312 [v1, v2] = ''
313 END
314 CheckDefFailure(lines, 'E1012: Type mismatch; expected list<any> but got string', 3)
Bram Moolenaarf785aa12021-02-11 21:19:34 +0100315
316 lines =<< trim END
317 g:values = [false, 0]
318 var x: bool
319 var y: string
320 [x, y] = g:values
321 END
322 CheckDefExecAndScriptFailure(lines, 'E1163: Variable 2: type mismatch, expected string but got number')
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100323enddef
324
Bram Moolenaar7f764942020-12-02 15:11:18 +0100325def Test_assign_linebreak()
326 var nr: number
327 nr =
328 123
329 assert_equal(123, nr)
330
331 var n2: number
332 [nr, n2] =
333 [12, 34]
334 assert_equal(12, nr)
335 assert_equal(34, n2)
336
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100337 CheckDefFailure(["var x = #"], 'E1097:', 3)
Bram Moolenaar77709b12021-04-03 21:01:01 +0200338
339 var lines =<< trim END
340 var x: list<string> = ['a']
341 var y: list<number> = x
342 ->copy()
343 ->copy()
344 END
345 CheckDefFailure(lines, 'E1012:', 2)
Bram Moolenaar7f764942020-12-02 15:11:18 +0100346enddef
347
Bram Moolenaardc234ca2020-11-28 18:52:33 +0100348def Test_assign_index()
349 # list of list
350 var l1: list<number>
351 l1[0] = 123
352 assert_equal([123], l1)
353
354 var l2: list<list<number>>
355 l2[0] = []
356 l2[0][0] = 123
357 assert_equal([[123]], l2)
358
359 var l3: list<list<list<number>>>
360 l3[0] = []
361 l3[0][0] = []
362 l3[0][0][0] = 123
363 assert_equal([[[123]]], l3)
364
365 var lines =<< trim END
366 var l3: list<list<number>>
367 l3[0] = []
368 l3[0][0] = []
369 END
370 CheckDefFailure(lines, 'E1012: Type mismatch; expected number but got list<unknown>', 3)
371
372 # dict of dict
373 var d1: dict<number>
374 d1.one = 1
375 assert_equal({one: 1}, d1)
376
377 var d2: dict<dict<number>>
378 d2.one = {}
379 d2.one.two = 123
380 assert_equal({one: {two: 123}}, d2)
381
382 var d3: dict<dict<dict<number>>>
383 d3.one = {}
384 d3.one.two = {}
385 d3.one.two.three = 123
386 assert_equal({one: {two: {three: 123}}}, d3)
387
388 lines =<< trim END
389 var d3: dict<dict<number>>
390 d3.one = {}
391 d3.one.two = {}
392 END
393 CheckDefFailure(lines, 'E1012: Type mismatch; expected number but got dict<unknown>', 3)
394
Bram Moolenaaracbae182020-12-13 18:44:43 +0100395 lines =<< trim END
396 var lines: list<string>
397 lines['a'] = 'asdf'
398 END
Bram Moolenaarf30a14d2021-01-17 21:51:24 +0100399 CheckDefFailure(lines, 'E1012:', 2)
Bram Moolenaaracbae182020-12-13 18:44:43 +0100400
401 lines =<< trim END
402 var lines: string
403 lines[9] = 'asdf'
404 END
405 CheckDefFailure(lines, 'E1141:', 2)
406
Bram Moolenaardc234ca2020-11-28 18:52:33 +0100407 # list of dict
408 var ld: list<dict<number>>
409 ld[0] = {}
410 ld[0].one = 123
411 assert_equal([{one: 123}], ld)
412
413 lines =<< trim END
414 var ld: list<dict<number>>
415 ld[0] = []
416 END
417 CheckDefFailure(lines, 'E1012: Type mismatch; expected dict<number> but got list<unknown>', 2)
418
419 # dict of list
420 var dl: dict<list<number>>
421 dl.one = []
422 dl.one[0] = 123
423 assert_equal({one: [123]}, dl)
424
425 lines =<< trim END
426 var dl: dict<list<number>>
427 dl.one = {}
428 END
429 CheckDefFailure(lines, 'E1012: Type mismatch; expected list<number> but got dict<unknown>', 2)
430enddef
431
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200432def Test_extend_list()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200433 var lines =<< trim END
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200434 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200435 var l: list<number>
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200436 l += [123]
437 assert_equal([123], l)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100438 END
439 CheckScriptSuccess(lines)
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200440
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100441 lines =<< trim END
442 vim9script
443 var list: list<string>
444 extend(list, ['x'])
445 assert_equal(['x'], list)
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200446 END
447 CheckScriptSuccess(lines)
Bram Moolenaar1a739232020-10-10 15:37:58 +0200448
449 # appending to NULL list from a function
450 lines =<< trim END
451 vim9script
452 var list: list<string>
453 def Func()
454 list += ['a', 'b']
455 enddef
456 Func()
457 assert_equal(['a', 'b'], list)
458 END
459 CheckScriptSuccess(lines)
Bram Moolenaar3beaf9c2020-12-18 17:23:14 +0100460 lines =<< trim END
461 vim9script
462 var list: list<string>
463 def Func()
464 extend(list, ['x', 'b'])
465 enddef
466 Func()
467 assert_equal(['x', 'b'], list)
468 END
469 CheckScriptSuccess(lines)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100470
471 lines =<< trim END
472 vim9script
473 var l: list<string> = test_null_list()
474 extend(l, ['x'])
475 assert_equal(['x'], l)
476 END
477 CheckScriptSuccess(lines)
478
479 lines =<< trim END
480 vim9script
481 extend(test_null_list(), ['x'])
482 END
483 CheckScriptFailure(lines, 'E1134:', 2)
484enddef
485
486def Test_extend_dict()
487 var lines =<< trim END
488 vim9script
489 var d: dict<number>
Bram Moolenaare0de1712020-12-02 17:36:54 +0100490 extend(d, {a: 1})
491 assert_equal({a: 1}, d)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100492
493 var d2: dict<number>
494 d2['one'] = 1
Bram Moolenaare0de1712020-12-02 17:36:54 +0100495 assert_equal({one: 1}, d2)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100496 END
497 CheckScriptSuccess(lines)
498
499 lines =<< trim END
500 vim9script
501 var d: dict<string> = test_null_dict()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100502 extend(d, {a: 'x'})
503 assert_equal({a: 'x'}, d)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100504 END
505 CheckScriptSuccess(lines)
506
507 lines =<< trim END
508 vim9script
Bram Moolenaare0de1712020-12-02 17:36:54 +0100509 extend(test_null_dict(), {a: 'x'})
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100510 END
511 CheckScriptFailure(lines, 'E1133:', 2)
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200512enddef
513
Bram Moolenaar971caf52020-09-21 22:21:42 +0200514def Test_single_letter_vars()
515 # single letter variables
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200516 var a: number = 123
Bram Moolenaar971caf52020-09-21 22:21:42 +0200517 a = 123
518 assert_equal(123, a)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200519 var b: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200520 b = 123
521 assert_equal(123, b)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200522 var g: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200523 g = 123
524 assert_equal(123, g)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200525 var s: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200526 s = 123
527 assert_equal(123, s)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200528 var t: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200529 t = 123
530 assert_equal(123, t)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200531 var v: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200532 v = 123
533 assert_equal(123, v)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200534 var w: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200535 w = 123
536 assert_equal(123, w)
537enddef
538
539def Test_vim9_single_char_vars()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200540 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200541 vim9script
542
543 # single character variable declarations work
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200544 var a: string
545 var b: number
546 var l: list<any>
547 var s: string
548 var t: number
549 var v: number
550 var w: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200551
552 # script-local variables can be used without s: prefix
553 a = 'script-a'
554 b = 111
555 l = [1, 2, 3]
556 s = 'script-s'
557 t = 222
558 v = 333
559 w = 444
560
561 assert_equal('script-a', a)
562 assert_equal(111, b)
563 assert_equal([1, 2, 3], l)
564 assert_equal('script-s', s)
565 assert_equal(222, t)
566 assert_equal(333, v)
567 assert_equal(444, w)
568 END
569 writefile(lines, 'Xsinglechar')
570 source Xsinglechar
571 delete('Xsinglechar')
572enddef
573
574def Test_assignment_list()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200575 var list1: list<bool> = [false, true, false]
576 var list2: list<number> = [1, 2, 3]
577 var list3: list<string> = ['sdf', 'asdf']
578 var list4: list<any> = ['yes', true, 1234]
579 var list5: list<blob> = [0z01, 0z02]
Bram Moolenaar971caf52020-09-21 22:21:42 +0200580
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200581 var listS: list<string> = []
582 var listN: list<number> = []
Bram Moolenaar971caf52020-09-21 22:21:42 +0200583
584 assert_equal([1, 2, 3], list2)
585 list2[-1] = 99
586 assert_equal([1, 2, 99], list2)
587 list2[-2] = 88
588 assert_equal([1, 88, 99], list2)
589 list2[-3] = 77
590 assert_equal([77, 88, 99], list2)
591 list2 += [100]
592 assert_equal([77, 88, 99, 100], list2)
593
594 list3 += ['end']
595 assert_equal(['sdf', 'asdf', 'end'], list3)
596
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200597 CheckDefExecFailure(['var ll = [1, 2, 3]', 'll[-4] = 6'], 'E684:')
598 CheckDefExecFailure(['var [v1, v2] = [1, 2]'], 'E1092:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200599
600 # type becomes list<any>
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200601 var somelist = rand() > 0 ? [1, 2, 3] : ['a', 'b', 'c']
Bram Moolenaar4f5e3972020-12-21 17:30:50 +0100602
603 var lines =<< trim END
604 var d = {dd: test_null_list()}
605 d.dd[0] = 0
606 END
607 CheckDefExecFailure(lines, 'E1147:', 2)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200608enddef
609
Bram Moolenaarf30a14d2021-01-17 21:51:24 +0100610def Test_assignment_list_any_index()
611 var l: list<number> = [1, 2]
612 for [x, y, _]
613 in [[0, 1, ''], [1, 3, '']]
614 l[x] = l[x] + y
615 endfor
616 assert_equal([2, 5], l)
617enddef
618
Bram Moolenaar971caf52020-09-21 22:21:42 +0200619def Test_assignment_list_vim9script()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200620 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200621 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200622 var v1: number
623 var v2: number
624 var v3: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200625 [v1, v2, v3] = [1, 2, 3]
626 assert_equal([1, 2, 3], [v1, v2, v3])
627 END
628 CheckScriptSuccess(lines)
629enddef
630
631def Test_assignment_dict()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100632 var dict1: dict<bool> = {one: false, two: true}
633 var dict2: dict<number> = {one: 1, two: 2}
634 var dict3: dict<string> = {key: 'value'}
635 var dict4: dict<any> = {one: 1, two: '2'}
636 var dict5: dict<blob> = {one: 0z01, two: 0z02}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200637
638 # overwrite
639 dict3['key'] = 'another'
Bram Moolenaare0de1712020-12-02 17:36:54 +0100640 assert_equal(dict3, {key: 'another'})
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100641 dict3.key = 'yet another'
Bram Moolenaare0de1712020-12-02 17:36:54 +0100642 assert_equal(dict3, {key: 'yet another'})
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100643
Bram Moolenaard24602f2020-12-20 15:20:56 +0100644 # member "any" can also be a dict and assigned to
645 var anydict: dict<any> = {nest: {}, nr: 0}
646 anydict.nest['this'] = 123
647 anydict.nest.that = 456
648 assert_equal({nest: {this: 123, that: 456}, nr: 0}, anydict)
649
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100650 var lines =<< trim END
Bram Moolenaard88dc4d2020-12-20 14:49:42 +0100651 var dd = {}
652 dd.two = 2
653 assert_equal({two: 2}, dd)
654 END
Bram Moolenaar4f5e3972020-12-21 17:30:50 +0100655 CheckDefAndScriptSuccess(lines)
656
657 lines =<< trim END
658 var d = {dd: {}}
659 d.dd[0] = 2
660 d.dd['x'] = 3
661 d.dd.y = 4
662 assert_equal({dd: {0: 2, x: 3, y: 4}}, d)
663 END
664 CheckDefAndScriptSuccess(lines)
Bram Moolenaard88dc4d2020-12-20 14:49:42 +0100665
666 lines =<< trim END
Bram Moolenaare0de1712020-12-02 17:36:54 +0100667 var dd = {one: 1}
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100668 dd.one) = 2
669 END
Bram Moolenaard88dc4d2020-12-20 14:49:42 +0100670 CheckDefFailure(lines, 'E488:', 2)
671
672 lines =<< trim END
673 var dd = {one: 1}
674 var dd.one = 2
675 END
676 CheckDefAndScriptFailure(lines, 'E1017:', 2)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200677
678 # empty key can be used
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200679 var dd = {}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200680 dd[""] = 6
Bram Moolenaare0de1712020-12-02 17:36:54 +0100681 assert_equal({['']: 6}, dd)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200682
683 # type becomes dict<any>
Bram Moolenaare0de1712020-12-02 17:36:54 +0100684 var somedict = rand() > 0 ? {a: 1, b: 2} : {a: 'a', b: 'b'}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200685
686 # assignment to script-local dict
Bram Moolenaarfc74d032020-11-16 22:11:49 +0100687 lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200688 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200689 var test: dict<any> = {}
Bram Moolenaar971caf52020-09-21 22:21:42 +0200690 def FillDict(): dict<any>
691 test['a'] = 43
692 return test
693 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +0100694 assert_equal({a: 43}, FillDict())
Bram Moolenaar971caf52020-09-21 22:21:42 +0200695 END
696 CheckScriptSuccess(lines)
697
698 lines =<< trim END
699 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200700 var test: dict<any>
Bram Moolenaar971caf52020-09-21 22:21:42 +0200701 def FillDict(): dict<any>
702 test['a'] = 43
703 return test
704 enddef
705 FillDict()
Bram Moolenaar3beaf9c2020-12-18 17:23:14 +0100706 assert_equal({a: 43}, test)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200707 END
Bram Moolenaar3beaf9c2020-12-18 17:23:14 +0100708 CheckScriptSuccess(lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200709
710 # assignment to global dict
711 lines =<< trim END
712 vim9script
713 g:test = {}
714 def FillDict(): dict<any>
715 g:test['a'] = 43
716 return g: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)
721
722 # assignment to buffer dict
723 lines =<< trim END
724 vim9script
725 b:test = {}
726 def FillDict(): dict<any>
727 b:test['a'] = 43
728 return b:test
729 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +0100730 assert_equal({a: 43}, FillDict())
Bram Moolenaar971caf52020-09-21 22:21:42 +0200731 END
732 CheckScriptSuccess(lines)
Bram Moolenaar4f5e3972020-12-21 17:30:50 +0100733
734 lines =<< trim END
735 var d = {dd: test_null_dict()}
736 d.dd[0] = 0
737 END
738 CheckDefExecFailure(lines, 'E1103:', 2)
739
740 lines =<< trim END
741 var d = {dd: 'string'}
742 d.dd[0] = 0
743 END
744 CheckDefExecFailure(lines, 'E1148:', 2)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200745enddef
746
747def Test_assignment_local()
748 # Test in a separated file in order not to the current buffer/window/tab is
749 # changed.
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200750 var script_lines: list<string> =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200751 let b:existing = 'yes'
752 let w:existing = 'yes'
753 let t:existing = 'yes'
754
755 def Test_assignment_local_internal()
756 b:newvar = 'new'
757 assert_equal('new', b:newvar)
758 assert_equal('yes', b:existing)
759 b:existing = 'no'
760 assert_equal('no', b:existing)
761 b:existing ..= 'NO'
762 assert_equal('noNO', b:existing)
763
764 w:newvar = 'new'
765 assert_equal('new', w:newvar)
766 assert_equal('yes', w:existing)
767 w:existing = 'no'
768 assert_equal('no', w:existing)
769 w:existing ..= 'NO'
770 assert_equal('noNO', w:existing)
771
772 t:newvar = 'new'
773 assert_equal('new', t:newvar)
774 assert_equal('yes', t:existing)
775 t:existing = 'no'
776 assert_equal('no', t:existing)
777 t:existing ..= 'NO'
778 assert_equal('noNO', t:existing)
779 enddef
780 call Test_assignment_local_internal()
781 END
782 CheckScriptSuccess(script_lines)
783enddef
784
785def Test_assignment_default()
Bram Moolenaar971caf52020-09-21 22:21:42 +0200786 # Test default values.
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200787 var thebool: bool
Bram Moolenaar971caf52020-09-21 22:21:42 +0200788 assert_equal(v:false, thebool)
789
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200790 var thenumber: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200791 assert_equal(0, thenumber)
792
793 if has('float')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200794 var thefloat: float
Bram Moolenaar971caf52020-09-21 22:21:42 +0200795 assert_equal(0.0, thefloat)
796 endif
797
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200798 var thestring: string
Bram Moolenaar971caf52020-09-21 22:21:42 +0200799 assert_equal('', thestring)
800
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200801 var theblob: blob
Bram Moolenaar971caf52020-09-21 22:21:42 +0200802 assert_equal(0z, theblob)
803
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200804 var Thefunc: func
Bram Moolenaar971caf52020-09-21 22:21:42 +0200805 assert_equal(test_null_function(), Thefunc)
806
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200807 var thelist: list<any>
Bram Moolenaar971caf52020-09-21 22:21:42 +0200808 assert_equal([], thelist)
809
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200810 var thedict: dict<any>
Bram Moolenaar971caf52020-09-21 22:21:42 +0200811 assert_equal({}, thedict)
812
813 if has('channel')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200814 var thejob: job
Bram Moolenaar971caf52020-09-21 22:21:42 +0200815 assert_equal(test_null_job(), thejob)
816
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200817 var thechannel: channel
Bram Moolenaar971caf52020-09-21 22:21:42 +0200818 assert_equal(test_null_channel(), thechannel)
819
820 if has('unix') && executable('cat')
821 # check with non-null job and channel, types must match
Bram Moolenaare0de1712020-12-02 17:36:54 +0100822 thejob = job_start("cat ", {})
Bram Moolenaar971caf52020-09-21 22:21:42 +0200823 thechannel = job_getchannel(thejob)
824 job_stop(thejob, 'kill')
825 endif
826 endif
827
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200828 var nr = 1234 | nr = 5678
Bram Moolenaar971caf52020-09-21 22:21:42 +0200829 assert_equal(5678, nr)
830enddef
831
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100832let scriptvar = 'init'
833
Bram Moolenaar971caf52020-09-21 22:21:42 +0200834def Test_assignment_var_list()
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100835 var lines =<< trim END
836 var v1: string
837 var v2: string
838 var vrem: list<string>
839 [v1] = ['aaa']
840 assert_equal('aaa', v1)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200841
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100842 [v1, v2] = ['one', 'two']
843 assert_equal('one', v1)
844 assert_equal('two', v2)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200845
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100846 [v1, v2; vrem] = ['one', 'two']
847 assert_equal('one', v1)
848 assert_equal('two', v2)
849 assert_equal([], vrem)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200850
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100851 [v1, v2; vrem] = ['one', 'two', 'three']
852 assert_equal('one', v1)
853 assert_equal('two', v2)
854 assert_equal(['three'], vrem)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200855
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100856 [&ts, &sw] = [3, 4]
857 assert_equal(3, &ts)
858 assert_equal(4, &sw)
859 set ts=8 sw=4
Bram Moolenaar004d9b02020-11-30 21:40:03 +0100860
861 [@a, @z] = ['aa', 'zz']
862 assert_equal('aa', @a)
863 assert_equal('zz', @z)
864
865 [$SOME_VAR, $OTHER_VAR] = ['some', 'other']
866 assert_equal('some', $SOME_VAR)
867 assert_equal('other', $OTHER_VAR)
868
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100869 [g:globalvar, b:bufvar, w:winvar, t:tabvar, v:errmsg] =
870 ['global', 'buf', 'win', 'tab', 'error']
Bram Moolenaar004d9b02020-11-30 21:40:03 +0100871 assert_equal('global', g:globalvar)
Bram Moolenaar004d9b02020-11-30 21:40:03 +0100872 assert_equal('buf', b:bufvar)
873 assert_equal('win', w:winvar)
874 assert_equal('tab', t:tabvar)
875 assert_equal('error', v:errmsg)
876 unlet g:globalvar
Bram Moolenaarda7c20c2020-11-30 21:12:19 +0100877 END
878 CheckDefAndScriptSuccess(lines)
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100879
880 [g:globalvar, s:scriptvar, b:bufvar] = ['global', 'script', 'buf']
881 assert_equal('global', g:globalvar)
882 assert_equal('script', s:scriptvar)
883 assert_equal('buf', b:bufvar)
884
885 lines =<< trim END
886 vim9script
887 var s:scriptvar = 'init'
888 [g:globalvar, s:scriptvar, w:winvar] = ['global', 'script', 'win']
889 assert_equal('global', g:globalvar)
890 assert_equal('script', s:scriptvar)
891 assert_equal('win', w:winvar)
892 END
893 CheckScriptSuccess(lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200894enddef
895
896def Test_assignment_vim9script()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200897 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200898 vim9script
899 def Func(): list<number>
900 return [1, 2]
901 enddef
Bram Moolenaar98309382020-09-27 21:58:45 +0200902 var name1: number
903 var name2: number
904 [name1, name2] =
Bram Moolenaar971caf52020-09-21 22:21:42 +0200905 Func()
Bram Moolenaar98309382020-09-27 21:58:45 +0200906 assert_equal(1, name1)
907 assert_equal(2, name2)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200908 var ll =
Bram Moolenaar971caf52020-09-21 22:21:42 +0200909 Func()
910 assert_equal([1, 2], ll)
911
912 @/ = 'text'
913 assert_equal('text', @/)
914 @0 = 'zero'
915 assert_equal('zero', @0)
916 @1 = 'one'
917 assert_equal('one', @1)
918 @9 = 'nine'
919 assert_equal('nine', @9)
920 @- = 'minus'
921 assert_equal('minus', @-)
922 if has('clipboard_working')
923 @* = 'star'
924 assert_equal('star', @*)
925 @+ = 'plus'
926 assert_equal('plus', @+)
927 endif
928
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200929 var a: number = 123
Bram Moolenaar971caf52020-09-21 22:21:42 +0200930 assert_equal(123, a)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200931 var s: string = 'yes'
Bram Moolenaar971caf52020-09-21 22:21:42 +0200932 assert_equal('yes', s)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200933 var b: number = 42
Bram Moolenaar971caf52020-09-21 22:21:42 +0200934 assert_equal(42, b)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200935 var w: number = 43
Bram Moolenaar971caf52020-09-21 22:21:42 +0200936 assert_equal(43, w)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200937 var t: number = 44
Bram Moolenaar971caf52020-09-21 22:21:42 +0200938 assert_equal(44, t)
Bram Moolenaar7a66a172020-10-16 19:56:12 +0200939
940 var to_var = 0
941 to_var = 3
942 assert_equal(3, to_var)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200943 END
944 CheckScriptSuccess(lines)
Bram Moolenaar06f9c692020-09-27 21:27:40 +0200945
946 lines =<< trim END
947 vim9script
948 var n: number
949 def Func()
950 n = 'string'
951 enddef
952 defcompile
953 END
954 CheckScriptFailure(lines, 'E1012: Type mismatch; expected number but got string')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200955enddef
956
957def Mess(): string
958 v:foldstart = 123
959 return 'xxx'
960enddef
961
962def Test_assignment_failure()
Bram Moolenaar98309382020-09-27 21:58:45 +0200963 CheckDefFailure(['var name=234'], 'E1004:')
964 CheckDefFailure(['var name =234'], 'E1004:')
965 CheckDefFailure(['var name= 234'], 'E1004:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200966
Bram Moolenaar98309382020-09-27 21:58:45 +0200967 CheckScriptFailure(['vim9script', 'var name=234'], 'E1004:')
968 CheckScriptFailure(['vim9script', 'var name=234'], "before and after '='")
969 CheckScriptFailure(['vim9script', 'var name =234'], 'E1004:')
970 CheckScriptFailure(['vim9script', 'var name= 234'], 'E1004:')
971 CheckScriptFailure(['vim9script', 'var name = 234', 'name+=234'], 'E1004:')
972 CheckScriptFailure(['vim9script', 'var name = 234', 'name+=234'], "before and after '+='")
973 CheckScriptFailure(['vim9script', 'var name = "x"', 'name..="y"'], 'E1004:')
974 CheckScriptFailure(['vim9script', 'var name = "x"', 'name..="y"'], "before and after '..='")
Bram Moolenaar971caf52020-09-21 22:21:42 +0200975
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200976 CheckDefFailure(['var true = 1'], 'E1034:')
977 CheckDefFailure(['var false = 1'], 'E1034:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200978
979 CheckDefFailure(['[a; b; c] = g:list'], 'E452:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200980 CheckDefExecFailure(['var a: number',
Bram Moolenaar971caf52020-09-21 22:21:42 +0200981 '[a] = test_null_list()'], 'E1093:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200982 CheckDefExecFailure(['var a: number',
Bram Moolenaar971caf52020-09-21 22:21:42 +0200983 '[a] = []'], 'E1093:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200984 CheckDefExecFailure(['var x: number',
985 'var y: number',
Bram Moolenaar971caf52020-09-21 22:21:42 +0200986 '[x, y] = [1]'], 'E1093:')
Bram Moolenaar352134b2020-10-17 22:04:08 +0200987 CheckDefExecFailure(['var x: string',
988 'var y: string',
989 '[x, y] = ["x"]'], 'E1093:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200990 CheckDefExecFailure(['var x: number',
991 'var y: number',
992 'var z: list<number>',
Bram Moolenaar971caf52020-09-21 22:21:42 +0200993 '[x, y; z] = [1]'], 'E1093:')
994
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200995 CheckDefFailure(['var somevar'], "E1022:")
996 CheckDefFailure(['var &tabstop = 4'], 'E1052:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200997 CheckDefFailure(['&g:option = 5'], 'E113:')
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200998 CheckScriptFailure(['vim9script', 'var &tabstop = 4'], 'E1052:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200999
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001000 CheckDefFailure(['var $VAR = 5'], 'E1016: Cannot declare an environment variable:')
1001 CheckScriptFailure(['vim9script', 'var $ENV = "xxx"'], 'E1016:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001002
1003 if has('dnd')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001004 CheckDefFailure(['var @~ = 5'], 'E1066:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001005 else
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001006 CheckDefFailure(['var @~ = 5'], 'E354:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001007 CheckDefFailure(['@~ = 5'], 'E354:')
1008 endif
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001009 CheckDefFailure(['var @a = 5'], 'E1066:')
1010 CheckDefFailure(['var @/ = "x"'], 'E1066:')
1011 CheckScriptFailure(['vim9script', 'var @a = "abc"'], 'E1066:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001012
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001013 CheckDefFailure(['var g:var = 5'], 'E1016: Cannot declare a global variable:')
1014 CheckDefFailure(['var w:var = 5'], 'E1016: Cannot declare a window variable:')
1015 CheckDefFailure(['var b:var = 5'], 'E1016: Cannot declare a buffer variable:')
1016 CheckDefFailure(['var t:var = 5'], 'E1016: Cannot declare a tab variable:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001017
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001018 CheckDefFailure(['var anr = 4', 'anr ..= "text"'], 'E1019:')
1019 CheckDefFailure(['var xnr += 4'], 'E1020:', 1)
1020 CheckScriptFailure(['vim9script', 'var xnr += 4'], 'E1020:')
1021 CheckDefFailure(["var xnr = xnr + 1"], 'E1001:', 1)
1022 CheckScriptFailure(['vim9script', 'var xnr = xnr + 4'], 'E121:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001023
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001024 CheckScriptFailure(['vim9script', 'def Func()', 'var dummy = s:notfound', 'enddef', 'defcompile'], 'E1108:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001025
Bram Moolenaar98309382020-09-27 21:58:45 +02001026 CheckDefFailure(['var name: list<string> = [123]'], 'expected list<string> but got list<number>')
1027 CheckDefFailure(['var name: list<number> = ["xx"]'], 'expected list<number> but got list<string>')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001028
Bram Moolenaare0de1712020-12-02 17:36:54 +01001029 CheckDefFailure(['var name: dict<string> = {key: 123}'], 'expected dict<string> but got dict<number>')
1030 CheckDefFailure(['var name: dict<number> = {key: "xx"}'], 'expected dict<number> but got dict<string>')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001031
Bram Moolenaar98309382020-09-27 21:58:45 +02001032 CheckDefFailure(['var name = feedkeys("0")'], 'E1031:')
1033 CheckDefFailure(['var name: number = feedkeys("0")'], 'expected number but got void')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001034
Bram Moolenaar98309382020-09-27 21:58:45 +02001035 CheckDefFailure(['var name: dict <number>'], 'E1068:')
1036 CheckDefFailure(['var name: dict<number'], 'E1009:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001037
1038 assert_fails('s/^/\=Mess()/n', 'E794:')
Bram Moolenaar98309382020-09-27 21:58:45 +02001039 CheckDefFailure(['var name: dict<number'], 'E1009:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001040
1041 CheckDefFailure(['w:foo: number = 10'],
1042 'E488: Trailing characters: : number = 1')
1043 CheckDefFailure(['t:foo: bool = true'],
1044 'E488: Trailing characters: : bool = true')
1045 CheckDefFailure(['b:foo: string = "x"'],
1046 'E488: Trailing characters: : string = "x"')
1047 CheckDefFailure(['g:foo: number = 123'],
1048 'E488: Trailing characters: : number = 123')
1049enddef
1050
1051def Test_assign_list()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001052 var l: list<string> = []
Bram Moolenaar971caf52020-09-21 22:21:42 +02001053 l[0] = 'value'
1054 assert_equal('value', l[0])
1055
1056 l[1] = 'asdf'
1057 assert_equal('value', l[0])
1058 assert_equal('asdf', l[1])
1059 assert_equal('asdf', l[-1])
1060 assert_equal('value', l[-2])
1061
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001062 var nrl: list<number> = []
Bram Moolenaar971caf52020-09-21 22:21:42 +02001063 for i in range(5)
1064 nrl[i] = i
1065 endfor
1066 assert_equal([0, 1, 2, 3, 4], nrl)
Bram Moolenaar334a8b42020-10-19 16:07:42 +02001067
1068 CheckDefFailure(["var l: list<number> = ['', true]"], 'E1012: Type mismatch; expected list<number> but got list<any>', 1)
1069 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 +02001070enddef
1071
1072def Test_assign_dict()
Bram Moolenaarc3689572021-01-01 19:40:02 +01001073 var lines =<< trim END
1074 var d: dict<string> = {}
1075 d['key'] = 'value'
1076 assert_equal('value', d['key'])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001077
Bram Moolenaarc3689572021-01-01 19:40:02 +01001078 d[123] = 'qwerty'
1079 assert_equal('qwerty', d[123])
1080 assert_equal('qwerty', d['123'])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001081
Bram Moolenaarc3689572021-01-01 19:40:02 +01001082 var nrd: dict<number> = {}
1083 for i in range(3)
1084 nrd[i] = i
1085 endfor
1086 assert_equal({0: 0, 1: 1, 2: 2}, nrd)
1087
1088 d.somekey = 'someval'
1089 assert_equal({key: 'value', '123': 'qwerty', somekey: 'someval'}, d)
1090 # unlet d.somekey
1091 # assert_equal({key: 'value', '123': 'qwerty'}, d)
1092 END
1093 CheckDefAndScriptSuccess(lines)
1094
1095 # TODO: move to above once "unlet d.somekey" in :def is implemented
1096 lines =<< trim END
1097 vim9script
1098 var d: dict<string> = {}
1099 d['key'] = 'value'
1100 d.somekey = 'someval'
1101 assert_equal({key: 'value', somekey: 'someval'}, d)
1102 unlet d.somekey
1103 assert_equal({key: 'value'}, d)
1104 END
1105 CheckScriptSuccess(lines)
Bram Moolenaar334a8b42020-10-19 16:07:42 +02001106
Bram Moolenaare0de1712020-12-02 17:36:54 +01001107 CheckDefFailure(["var d: dict<number> = {a: '', b: true}"], 'E1012: Type mismatch; expected dict<number> but got dict<any>', 1)
1108 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 +02001109enddef
1110
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001111def Test_assign_dict_unknown_type()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001112 var lines =<< trim END
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001113 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001114 var mylist = []
Bram Moolenaare0de1712020-12-02 17:36:54 +01001115 mylist += [{one: 'one'}]
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001116 def Func()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001117 var dd = mylist[0]
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001118 assert_equal('one', dd.one)
1119 enddef
1120 Func()
1121 END
1122 CheckScriptSuccess(lines)
1123
Bram Moolenaarce2c5442020-11-28 21:21:17 +01001124 lines =<< trim END
1125 vim9script
1126 var mylist = [[]]
Bram Moolenaare0de1712020-12-02 17:36:54 +01001127 mylist[0] += [{one: 'one'}]
Bram Moolenaarce2c5442020-11-28 21:21:17 +01001128 def Func()
1129 var dd = mylist[0][0]
1130 assert_equal('one', dd.one)
1131 enddef
1132 Func()
1133 END
1134 CheckScriptSuccess(lines)
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001135enddef
1136
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001137def Test_assign_dict_with_op()
1138 var lines =<< trim END
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001139 var ds: dict<string> = {a: 'x'}
1140 ds['a'] ..= 'y'
1141 ds.a ..= 'z'
1142 assert_equal('xyz', ds.a)
1143
1144 var dn: dict<number> = {a: 9}
1145 dn['a'] += 2
1146 assert_equal(11, dn.a)
1147 dn.a += 2
1148 assert_equal(13, dn.a)
1149
1150 dn['a'] -= 3
1151 assert_equal(10, dn.a)
1152 dn.a -= 2
1153 assert_equal(8, dn.a)
1154
1155 dn['a'] *= 2
1156 assert_equal(16, dn.a)
1157 dn.a *= 2
1158 assert_equal(32, dn.a)
1159
1160 dn['a'] /= 3
1161 assert_equal(10, dn.a)
1162 dn.a /= 2
1163 assert_equal(5, dn.a)
1164
1165 dn['a'] %= 3
1166 assert_equal(2, dn.a)
1167 dn.a %= 6
1168 assert_equal(2, dn.a)
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001169
1170 var dd: dict<dict<list<any>>>
1171 dd.a = {}
1172 dd.a.b = [0]
1173 dd.a.b += [1]
1174 assert_equal({a: {b: [0, 1]}}, dd)
Bram Moolenaar0d351a02021-04-05 21:05:26 +02001175
1176 var dab = {a: ['b']}
1177 dab.a[0] ..= 'c'
1178 assert_equal({a: ['bc']}, dab)
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001179 END
Bram Moolenaare42939a2021-04-05 17:11:17 +02001180 CheckDefAndScriptSuccess(lines)
1181enddef
1182
1183def Test_assign_list_with_op()
1184 var lines =<< trim END
1185 var ls: list<string> = ['x']
1186 ls[0] ..= 'y'
1187 assert_equal('xy', ls[0])
1188
1189 var ln: list<number> = [9]
1190 ln[0] += 2
1191 assert_equal(11, ln[0])
1192
1193 ln[0] -= 3
1194 assert_equal(8, ln[0])
1195
1196 ln[0] *= 2
1197 assert_equal(16, ln[0])
1198
1199 ln[0] /= 3
1200 assert_equal(5, ln[0])
1201
1202 ln[0] %= 3
1203 assert_equal(2, ln[0])
1204 END
1205 CheckDefAndScriptSuccess(lines)
1206enddef
1207
1208def Test_assign_with_op_fails()
1209 var lines =<< trim END
1210 var s = 'abc'
1211 s[1] += 'x'
1212 END
1213 CheckDefAndScriptFailure2(lines, 'E1141:', 'E689:', 2)
1214
1215 lines =<< trim END
1216 var s = 'abc'
1217 s[1] ..= 'x'
1218 END
1219 CheckDefAndScriptFailure2(lines, 'E1141:', 'E689:', 2)
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001220
1221 lines =<< trim END
1222 var dd: dict<dict<list<any>>>
1223 dd.a = {}
1224 dd.a.b += [1]
1225 END
1226 CheckDefExecAndScriptFailure(lines, 'E716:', 3)
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001227enddef
1228
Bram Moolenaar971caf52020-09-21 22:21:42 +02001229def Test_assign_lambda()
1230 # check if assign a lambda to a variable which type is func or any.
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001231 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +02001232 vim9script
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001233 var FuncRef = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001234 assert_equal(123, FuncRef())
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001235 var FuncRef_Func: func = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001236 assert_equal(123, FuncRef_Func())
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001237 var FuncRef_Any: any = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001238 assert_equal(123, FuncRef_Any())
Bram Moolenaar06409502021-02-17 17:00:27 +01001239 var FuncRef_Number: func(): number = () => 321
1240 assert_equal(321, FuncRef_Number())
Bram Moolenaar971caf52020-09-21 22:21:42 +02001241 END
1242 CheckScriptSuccess(lines)
Bram Moolenaar328eac22021-01-07 19:23:08 +01001243
1244 lines =<< trim END
1245 var Ref: func(number)
1246 Ref = (j) => !j
1247 END
Bram Moolenaar06409502021-02-17 17:00:27 +01001248 CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected func(number) but got func(any): bool')
Bram Moolenaarc7dac852021-02-17 18:49:11 +01001249
1250 lines =<< trim END
1251 echo filter([1, 2, 3], (_, v: string) => v + 1)
1252 END
1253 CheckDefAndScriptFailure(lines, 'E1051:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001254enddef
1255
Bram Moolenaarc0e29012020-09-27 14:22:48 +02001256def Test_heredoc()
1257 var lines =<< trim END # comment
1258 text
1259 END
1260 assert_equal(['text'], lines)
1261
1262 CheckDefFailure(['var lines =<< trim END X', 'END'], 'E488:')
1263 CheckDefFailure(['var lines =<< trim END " comment', 'END'], 'E488:')
Bram Moolenaar2914a202020-09-27 18:24:03 +02001264
1265 lines =<< trim [END]
1266 def Func()
1267 var&lines =<< trim END
1268 x
1269 x
Bram Moolenaarb5b94802020-12-13 17:50:20 +01001270 enddef
1271 defcompile
1272 [END]
1273 CheckScriptFailure(lines, 'E1145: Missing heredoc end marker: END')
1274 delfunc! g:Func
1275
1276 lines =<< trim [END]
1277 def Func()
1278 var lines =<< trim END
1279 x
1280 x
Bram Moolenaar2914a202020-09-27 18:24:03 +02001281 x
1282 x
1283 x
1284 x
1285 x
1286 x
1287 enddef
1288 call Func()
1289 [END]
Bram Moolenaarb5b94802020-12-13 17:50:20 +01001290 CheckScriptFailure(lines, 'E1145: Missing heredoc end marker: END')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01001291 delfunc! g:Func
Bram Moolenaarc0e29012020-09-27 14:22:48 +02001292enddef
1293
Bram Moolenaar083966f2020-12-08 11:29:40 +01001294def Test_var_func_call()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001295 var lines =<< trim END
1296 vim9script
1297 func GetValue()
1298 if exists('g:count')
1299 let g:count += 1
1300 else
1301 let g:count = 1
1302 endif
1303 return 'this'
1304 endfunc
1305 var val: string = GetValue()
1306 # env var is always a string
1307 var env = $TERM
1308 END
1309 writefile(lines, 'Xfinished')
1310 source Xfinished
1311 # GetValue() is not called during discovery phase
1312 assert_equal(1, g:count)
1313
1314 unlet g:count
1315 delete('Xfinished')
1316enddef
1317
Bram Moolenaar083966f2020-12-08 11:29:40 +01001318def Test_var_missing_type()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001319 var lines =<< trim END
1320 vim9script
1321 var name = g:unknown
1322 END
1323 CheckScriptFailure(lines, 'E121:')
1324
1325 lines =<< trim END
1326 vim9script
1327 var nr: number = 123
1328 var name = nr
1329 END
1330 CheckScriptSuccess(lines)
1331enddef
1332
Bram Moolenaar083966f2020-12-08 11:29:40 +01001333def Test_var_declaration()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001334 var lines =<< trim END
1335 vim9script
1336 var name: string
1337 g:var_uninit = name
1338 name = 'text'
1339 g:var_test = name
1340 # prefixing s: is optional
1341 s:name = 'prefixed'
1342 g:var_prefixed = s:name
1343
Bram Moolenaar7257af42020-12-08 22:08:51 +01001344 const FOO: number = 123
1345 assert_equal(123, FOO)
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001346 const FOOS = 'foos'
1347 assert_equal('foos', FOOS)
1348 final FLIST = [1]
1349 assert_equal([1], FLIST)
1350 FLIST[0] = 11
1351 assert_equal([11], FLIST)
1352
1353 const g:FOO: number = 321
1354 assert_equal(321, g:FOO)
1355 const g:FOOS = 'gfoos'
1356 assert_equal('gfoos', g:FOOS)
1357 final g:FLIST = [2]
1358 assert_equal([2], g:FLIST)
1359 g:FLIST[0] = 22
1360 assert_equal([22], g:FLIST)
1361
Bram Moolenaard877a572021-04-01 19:42:48 +02001362 def SetGlobalConst()
1363 const g:globConst = 123
1364 enddef
1365 SetGlobalConst()
1366 assert_equal(123, g:globConst)
1367 assert_true(islocked('g:globConst'))
1368
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001369 const w:FOO: number = 46
1370 assert_equal(46, w:FOO)
1371 const w:FOOS = 'wfoos'
1372 assert_equal('wfoos', w:FOOS)
1373 final w:FLIST = [3]
1374 assert_equal([3], w:FLIST)
1375 w:FLIST[0] = 33
1376 assert_equal([33], w:FLIST)
Bram Moolenaar7257af42020-12-08 22:08:51 +01001377
Bram Moolenaar55759b52020-09-30 22:59:42 +02001378 var s:other: number
1379 other = 1234
1380 g:other_var = other
1381
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01001382 var xyz: string # comment
1383
Bram Moolenaar55759b52020-09-30 22:59:42 +02001384 # type is inferred
Bram Moolenaar3862ea32021-01-01 21:05:55 +01001385 var s:dict = {['a']: 222}
Bram Moolenaar55759b52020-09-30 22:59:42 +02001386 def GetDictVal(key: any)
1387 g:dict_val = s:dict[key]
1388 enddef
1389 GetDictVal('a')
Bram Moolenaar08251752021-01-11 21:20:18 +01001390
1391 final adict: dict<string> = {}
1392 def ChangeAdict()
1393 adict.foo = 'foo'
1394 enddef
1395 ChangeAdict()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001396 END
1397 CheckScriptSuccess(lines)
1398 assert_equal('', g:var_uninit)
1399 assert_equal('text', g:var_test)
1400 assert_equal('prefixed', g:var_prefixed)
1401 assert_equal(1234, g:other_var)
1402 assert_equal(222, g:dict_val)
1403
1404 unlet g:var_uninit
1405 unlet g:var_test
1406 unlet g:var_prefixed
1407 unlet g:other_var
Bram Moolenaar6e50ec22021-04-03 19:32:44 +02001408 unlet g:globConst
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001409 unlet g:FOO
1410 unlet g:FOOS
1411 unlet g:FLIST
1412 unlet w:FOO
1413 unlet w:FOOS
1414 unlet w:FLIST
Bram Moolenaar55759b52020-09-30 22:59:42 +02001415enddef
1416
Bram Moolenaar083966f2020-12-08 11:29:40 +01001417def Test_var_declaration_fails()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001418 var lines =<< trim END
1419 vim9script
1420 final var: string
1421 END
1422 CheckScriptFailure(lines, 'E1125:')
1423
1424 lines =<< trim END
1425 vim9script
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001426 const g:constvar = 'string'
1427 g:constvar = 'xx'
1428 END
1429 CheckScriptFailure(lines, 'E741:')
1430 unlet g:constvar
1431
1432 lines =<< trim END
1433 vim9script
Bram Moolenaarf5906aa2021-04-02 14:35:15 +02001434 var name = 'one'
1435 lockvar name
1436 def SetLocked()
1437 name = 'two'
1438 enddef
1439 SetLocked()
1440 END
Bram Moolenaardcf29ac2021-04-02 14:44:02 +02001441 CheckScriptFailure(lines, 'E741: Value is locked: name', 1)
1442
1443 lines =<< trim END
1444 let s:legacy = 'one'
1445 lockvar s:legacy
1446 def SetLocked()
1447 s:legacy = 'two'
1448 enddef
1449 call SetLocked()
1450 END
1451 CheckScriptFailure(lines, 'E741: Value is locked: s:legacy', 1)
Bram Moolenaarf5906aa2021-04-02 14:35:15 +02001452
1453 lines =<< trim END
1454 vim9script
Bram Moolenaard877a572021-04-01 19:42:48 +02001455 def SetGlobalConst()
1456 const g:globConst = 123
1457 enddef
1458 SetGlobalConst()
1459 g:globConst = 234
1460 END
Bram Moolenaar6e50ec22021-04-03 19:32:44 +02001461 CheckScriptFailure(lines, 'E741: Value is locked: g:globConst', 6)
Bram Moolenaard877a572021-04-01 19:42:48 +02001462 unlet g:globConst
1463
1464 lines =<< trim END
1465 vim9script
Bram Moolenaar08251752021-01-11 21:20:18 +01001466 const cdict: dict<string> = {}
1467 def Change()
1468 cdict.foo = 'foo'
1469 enddef
1470 defcompile
1471 END
1472 CheckScriptFailure(lines, 'E46:')
1473
1474 lines =<< trim END
1475 vim9script
Bram Moolenaar89b474d2020-12-22 21:19:39 +01001476 final w:finalvar = [9]
1477 w:finalvar = [8]
1478 END
1479 CheckScriptFailure(lines, 'E1122:')
1480 unlet w:finalvar
1481
1482 lines =<< trim END
1483 vim9script
Bram Moolenaar55759b52020-09-30 22:59:42 +02001484 const var: string
1485 END
1486 CheckScriptFailure(lines, 'E1021:')
1487
1488 lines =<< trim END
1489 vim9script
1490 var 9var: string
1491 END
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01001492 CheckScriptFailure(lines, 'E488:')
Bram Moolenaar7257af42020-12-08 22:08:51 +01001493
1494 CheckDefFailure(['var foo.bar = 2'], 'E1087:')
1495 CheckDefFailure(['var foo[3] = 2'], 'E1087:')
1496 CheckDefFailure(['const foo: number'], 'E1021:')
Bram Moolenaar55759b52020-09-30 22:59:42 +02001497enddef
1498
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001499def Test_script_local_in_legacy()
1500 # OK to define script-local later when prefixed with s:
1501 var lines =<< trim END
1502 def SetLater()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001503 s:legvar = 'two'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001504 enddef
1505 defcompile
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001506 let s:legvar = 'one'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001507 call SetLater()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001508 call assert_equal('two', s:legvar)
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001509 END
1510 CheckScriptSuccess(lines)
1511
1512 # OK to leave out s: prefix when script-local already defined
1513 lines =<< trim END
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001514 let s:legvar = 'one'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001515 def SetNoPrefix()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001516 legvar = 'two'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001517 enddef
1518 call SetNoPrefix()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001519 call assert_equal('two', s:legvar)
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001520 END
1521 CheckScriptSuccess(lines)
1522
1523 # Not OK to leave out s: prefix when script-local defined later
1524 lines =<< trim END
1525 def SetLaterNoPrefix()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001526 legvar = 'two'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001527 enddef
1528 defcompile
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02001529 let s:legvar = 'one'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02001530 END
1531 CheckScriptFailure(lines, 'E476:', 1)
1532enddef
1533
Bram Moolenaar083966f2020-12-08 11:29:40 +01001534def Test_var_type_check()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001535 var lines =<< trim END
1536 vim9script
1537 var name: string
1538 name = 1234
1539 END
1540 CheckScriptFailure(lines, 'E1012:')
1541
1542 lines =<< trim END
1543 vim9script
1544 var name:string
1545 END
1546 CheckScriptFailure(lines, 'E1069:')
1547
1548 lines =<< trim END
1549 vim9script
1550 var name: asdf
1551 END
1552 CheckScriptFailure(lines, 'E1010:')
1553
1554 lines =<< trim END
1555 vim9script
1556 var s:l: list<number>
1557 s:l = []
1558 END
1559 CheckScriptSuccess(lines)
1560
1561 lines =<< trim END
1562 vim9script
1563 var s:d: dict<number>
1564 s:d = {}
1565 END
1566 CheckScriptSuccess(lines)
1567enddef
1568
1569let g:dict_number = #{one: 1, two: 2}
1570
Bram Moolenaar083966f2020-12-08 11:29:40 +01001571def Test_var_list_dict_type()
Bram Moolenaar55759b52020-09-30 22:59:42 +02001572 var ll: list<number>
1573 ll = [1, 2, 2, 3, 3, 3]->uniq()
1574 ll->assert_equal([1, 2, 3])
1575
1576 var dd: dict<number>
1577 dd = g:dict_number
1578 dd->assert_equal(g:dict_number)
1579
1580 var lines =<< trim END
1581 var ll: list<number>
1582 ll = [1, 2, 3]->map('"one"')
1583 END
Bram Moolenaar70250fb2021-01-16 19:01:53 +01001584 CheckDefExecFailure(lines, 'E1012: Type mismatch; expected number but got string')
Bram Moolenaar55759b52020-09-30 22:59:42 +02001585enddef
1586
Bram Moolenaar083966f2020-12-08 11:29:40 +01001587def Test_cannot_use_let()
1588 CheckDefAndScriptFailure(['let a = 34'], 'E1126:', 1)
1589enddef
1590
Bram Moolenaar55759b52020-09-30 22:59:42 +02001591def Test_unlet()
1592 g:somevar = 'yes'
1593 assert_true(exists('g:somevar'))
1594 unlet g:somevar
1595 assert_false(exists('g:somevar'))
1596 unlet! g:somevar
1597
1598 # also works for script-local variable in legacy Vim script
1599 s:somevar = 'legacy'
1600 assert_true(exists('s:somevar'))
1601 unlet s:somevar
1602 assert_false(exists('s:somevar'))
1603 unlet! s:somevar
1604
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001605 CheckDefExecFailure([
1606 'var dd = 111',
1607 'unlet dd',
1608 ], 'E1081:', 2)
1609
Bram Moolenaar752fc692021-01-04 21:57:11 +01001610 # dict unlet
1611 var dd = {a: 1, b: 2, c: 3}
1612 unlet dd['a']
1613 unlet dd.c
1614 assert_equal({b: 2}, dd)
1615
1616 # list unlet
1617 var ll = [1, 2, 3, 4]
1618 unlet ll[1]
1619 unlet ll[-1]
1620 assert_equal([1, 3], ll)
1621
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01001622 ll = [1, 2, 3, 4]
1623 unlet ll[0 : 1]
1624 assert_equal([3, 4], ll)
1625
1626 ll = [1, 2, 3, 4]
1627 unlet ll[2 : 8]
1628 assert_equal([1, 2], ll)
1629
1630 ll = [1, 2, 3, 4]
1631 unlet ll[-2 : -1]
1632 assert_equal([1, 2], ll)
1633
1634 CheckDefFailure([
1635 'var ll = [1, 2]',
1636 'll[1 : 2] = 7',
1637 ], 'E1165:', 2)
1638 CheckDefFailure([
1639 'var dd = {a: 1}',
1640 'unlet dd["a" : "a"]',
1641 ], 'E1166:', 2)
1642 CheckDefExecFailure([
1643 'unlet g:adict[0 : 1]',
1644 ], 'E1148:', 1)
1645 CheckDefFailure([
1646 'var ll = [1, 2]',
1647 'unlet ll[0:1]',
1648 ], 'E1004:', 2)
1649 CheckDefFailure([
1650 'var ll = [1, 2]',
1651 'unlet ll[0 :1]',
1652 ], 'E1004:', 2)
1653 CheckDefFailure([
1654 'var ll = [1, 2]',
1655 'unlet ll[0: 1]',
1656 ], 'E1004:', 2)
1657
1658 CheckDefFailure([
1659 'var ll = [1, 2]',
1660 'unlet ll["x" : 1]',
1661 ], 'E1012:', 2)
1662 CheckDefFailure([
1663 'var ll = [1, 2]',
1664 'unlet ll[0 : "x"]',
1665 ], 'E1012:', 2)
1666
Bram Moolenaar752fc692021-01-04 21:57:11 +01001667 # list of dict unlet
1668 var dl = [{a: 1, b: 2}, {c: 3}]
1669 unlet dl[0]['b']
1670 assert_equal([{a: 1}, {c: 3}], dl)
1671
1672 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001673 'var ll = test_null_list()',
1674 'unlet ll[0]',
1675 ], 'E684:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001676 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001677 'var ll = [1]',
1678 'unlet ll[2]',
1679 ], 'E684:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001680 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001681 'var ll = [1]',
1682 'unlet ll[g:astring]',
Bram Moolenaarf30a14d2021-01-17 21:51:24 +01001683 ], 'E1012:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001684 CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001685 'var dd = test_null_dict()',
1686 'unlet dd["a"]',
1687 ], 'E716:', 2)
1688 CheckDefExecFailure([
1689 'var dd = {a: 1}',
1690 'unlet dd["b"]',
1691 ], 'E716:', 2)
1692 CheckDefExecFailure([
1693 'var dd = {a: 1}',
1694 'unlet dd[g:alist]',
1695 ], 'E1105:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01001696
Bram Moolenaar2ef951d2021-01-03 20:55:26 +01001697 # can compile unlet before variable exists
Bram Moolenaar752fc692021-01-04 21:57:11 +01001698 g:someDict = {key: 'val'}
1699 var k = 'key'
1700 unlet g:someDict[k]
1701 assert_equal({}, g:someDict)
1702 unlet g:someDict
1703 assert_false(exists('g:someDict'))
Bram Moolenaar2ef951d2021-01-03 20:55:26 +01001704
Bram Moolenaar55759b52020-09-30 22:59:42 +02001705 CheckScriptFailure([
1706 'vim9script',
1707 'var svar = 123',
1708 'unlet svar',
1709 ], 'E1081:')
1710 CheckScriptFailure([
1711 'vim9script',
1712 'var svar = 123',
1713 'unlet s:svar',
1714 ], 'E1081:')
1715 CheckScriptFailure([
1716 'vim9script',
1717 'var svar = 123',
1718 'def Func()',
1719 ' unlet svar',
1720 'enddef',
1721 'defcompile',
1722 ], 'E1081:')
1723 CheckScriptFailure([
1724 'vim9script',
1725 'var svar = 123',
Bram Moolenaar9aed7292020-12-18 15:38:00 +01001726 'func Func()',
1727 ' unlet s:svar',
1728 'endfunc',
1729 'Func()',
1730 ], 'E1081:')
1731 CheckScriptFailure([
1732 'vim9script',
1733 'var svar = 123',
Bram Moolenaar55759b52020-09-30 22:59:42 +02001734 'def Func()',
1735 ' unlet s:svar',
1736 'enddef',
1737 'defcompile',
1738 ], 'E1081:')
1739
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01001740 writefile(['vim9script', 'export var svar = 1234'], 'XunletExport.vim')
1741 var lines =<< trim END
1742 vim9script
1743 import svar from './XunletExport.vim'
1744 def UnletSvar()
1745 unlet svar
1746 enddef
1747 defcompile
1748 END
1749 CheckScriptFailure(lines, 'E1081:', 1)
1750 delete('XunletExport.vim')
1751
Bram Moolenaar55759b52020-09-30 22:59:42 +02001752 $ENVVAR = 'foobar'
1753 assert_equal('foobar', $ENVVAR)
1754 unlet $ENVVAR
1755 assert_equal('', $ENVVAR)
1756enddef
1757
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01001758def Test_expr_error_no_assign()
1759 var lines =<< trim END
1760 vim9script
1761 var x = invalid
1762 echo x
1763 END
1764 CheckScriptFailureList(lines, ['E121:', 'E121:'])
1765
1766 lines =<< trim END
1767 vim9script
1768 var x = 1 / 0
1769 echo x
1770 END
Bram Moolenaar227c58a2021-04-28 20:40:44 +02001771 CheckScriptFailure(lines, 'E1154:')
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01001772
1773 lines =<< trim END
1774 vim9script
1775 var x = 1 % 0
1776 echo x
1777 END
Bram Moolenaar227c58a2021-04-28 20:40:44 +02001778 CheckScriptFailure(lines, 'E1154:')
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01001779
1780 lines =<< trim END
1781 var x: string 'string'
1782 END
1783 CheckDefAndScriptFailure(lines, 'E488:')
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01001784enddef
1785
1786
Bram Moolenaar17126b12021-01-07 22:03:02 +01001787def Test_assign_command_modifier()
1788 var lines =<< trim END
Bram Moolenaar9e0f8832021-01-09 12:09:22 +01001789 var verbose = 0
1790 verbose = 1
1791 assert_equal(1, verbose)
1792 silent verbose = 2
1793 assert_equal(2, verbose)
1794 silent verbose += 2
1795 assert_equal(4, verbose)
1796 silent verbose -= 1
1797 assert_equal(3, verbose)
Bram Moolenaar17126b12021-01-07 22:03:02 +01001798
Bram Moolenaar9e0f8832021-01-09 12:09:22 +01001799 var topleft = {one: 1}
1800 sandbox topleft.one = 3
1801 assert_equal({one: 3}, topleft)
1802 leftabove topleft[' '] = 4
1803 assert_equal({one: 3, ' ': 4}, topleft)
1804
1805 var x: number
1806 var y: number
1807 silent [x, y] = [1, 2]
1808 assert_equal(1, x)
1809 assert_equal(2, y)
Bram Moolenaar17126b12021-01-07 22:03:02 +01001810 END
1811 CheckDefAndScriptSuccess(lines)
1812enddef
1813
Bram Moolenaar32154662021-03-28 21:14:06 +02001814def Test_script_funcref_case()
1815 var lines =<< trim END
1816 var Len = (s: string): number => len(s) + 1
1817 assert_equal(5, Len('asdf'))
1818 END
1819 CheckDefAndScriptSuccess(lines)
1820
1821 lines =<< trim END
1822 var len = (s: string): number => len(s) + 1
1823 END
1824 CheckDefAndScriptFailure(lines, 'E704:')
1825
1826 lines =<< trim END
1827 vim9script
1828 var s:Len = (s: string): number => len(s) + 2
1829 assert_equal(6, Len('asdf'))
1830 END
1831 CheckScriptSuccess(lines)
1832
1833 lines =<< trim END
1834 vim9script
1835 var s:len = (s: string): number => len(s) + 1
1836 END
1837 CheckScriptFailure(lines, 'E704:')
1838enddef
1839
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001840def Test_inc_dec()
1841 var lines =<< trim END
1842 var nr = 7
1843 ++nr
Bram Moolenaar730bf302021-04-24 20:43:56 +02001844 assert_equal(8, nr)
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001845 --nr
Bram Moolenaar730bf302021-04-24 20:43:56 +02001846 assert_equal(7, nr)
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001847
1848 var ll = [1, 2]
1849 --ll[0]
1850 ++ll[1]
Bram Moolenaar730bf302021-04-24 20:43:56 +02001851 assert_equal([0, 3], ll)
1852
1853 g:count = 1
1854 ++g:count
1855 --g:count
1856 assert_equal(1, g:count)
1857 unlet g:count
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02001858 END
1859 CheckDefAndScriptSuccess(lines)
1860enddef
1861
Bram Moolenaar55759b52020-09-30 22:59:42 +02001862
Bram Moolenaar971caf52020-09-21 22:21:42 +02001863" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker