blob: 7a72edce0030529b9403947467456eeda569eb91 [file] [log] [blame]
Bram Moolenaar971caf52020-09-21 22:21:42 +02001" Test Vim9 assignments
2
3source check.vim
Bram Moolenaar62aec932022-01-29 21:45:34 +00004import './vim9.vim' as v9
Bram Moolenaarc653e4a2022-01-05 10:16:30 +00005source term_util.vim
Bram Moolenaar971caf52020-09-21 22:21:42 +02006
7let s:appendToMe = 'xxx'
8let s:addToMe = 111
Bram Moolenaara749a422022-02-12 19:52:25 +00009let s:newVar = ''
Bram Moolenaar971caf52020-09-21 22:21:42 +020010let g:existing = 'yes'
11let g:inc_counter = 1
12let $SOME_ENV_VAR = 'some'
13let g:alist = [7]
Bram Moolenaar5b5ae292021-02-20 17:04:02 +010014let g:adict = #{a: 1}
Bram Moolenaar971caf52020-09-21 22:21:42 +020015let g:astring = 'text'
16
17def Test_assignment_bool()
Bram Moolenaar30fd8202020-09-26 15:09:30 +020018 var bool1: bool = true
Bram Moolenaar971caf52020-09-21 22:21:42 +020019 assert_equal(v:true, bool1)
Bram Moolenaar30fd8202020-09-26 15:09:30 +020020 var bool2: bool = false
Bram Moolenaar971caf52020-09-21 22:21:42 +020021 assert_equal(v:false, bool2)
22
Bram Moolenaar30fd8202020-09-26 15:09:30 +020023 var bool3: bool = 0
Bram Moolenaar971caf52020-09-21 22:21:42 +020024 assert_equal(false, bool3)
Bram Moolenaar30fd8202020-09-26 15:09:30 +020025 var bool4: bool = 1
Bram Moolenaar971caf52020-09-21 22:21:42 +020026 assert_equal(true, bool4)
27
Bram Moolenaar2bb26582020-10-03 22:52:39 +020028 var bool5: bool = 1 && true
Bram Moolenaar971caf52020-09-21 22:21:42 +020029 assert_equal(true, bool5)
Bram Moolenaar2bb26582020-10-03 22:52:39 +020030 var bool6: bool = 0 && 1
Bram Moolenaar971caf52020-09-21 22:21:42 +020031 assert_equal(false, bool6)
Bram Moolenaar2bb26582020-10-03 22:52:39 +020032 var bool7: bool = 0 || 1 && true
Bram Moolenaar971caf52020-09-21 22:21:42 +020033 assert_equal(true, bool7)
34
Bram Moolenaar30fd8202020-09-26 15:09:30 +020035 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +020036 vim9script
37 def GetFlag(): bool
Bram Moolenaar30fd8202020-09-26 15:09:30 +020038 var flag: bool = 1
Bram Moolenaar971caf52020-09-21 22:21:42 +020039 return flag
40 enddef
Bram Moolenaar30fd8202020-09-26 15:09:30 +020041 var flag: bool = GetFlag()
Bram Moolenaar971caf52020-09-21 22:21:42 +020042 assert_equal(true, flag)
43 flag = 0
44 assert_equal(false, flag)
45 flag = 1
46 assert_equal(true, flag)
Bram Moolenaar2bb26582020-10-03 22:52:39 +020047 flag = 1 || true
Bram Moolenaar971caf52020-09-21 22:21:42 +020048 assert_equal(true, flag)
Bram Moolenaar2bb26582020-10-03 22:52:39 +020049 flag = 1 && false
Bram Moolenaar971caf52020-09-21 22:21:42 +020050 assert_equal(false, flag)
Bram Moolenaardd1f4262020-12-31 17:41:01 +010051
52 var cp: bool = &cp
53 var fen: bool = &l:fen
Bram Moolenaar971caf52020-09-21 22:21:42 +020054 END
Bram Moolenaar62aec932022-01-29 21:45:34 +000055 v9.CheckScriptSuccess(lines)
56 v9.CheckDefAndScriptFailure(['var x: bool = 2'], 'E1012:')
57 v9.CheckDefAndScriptFailure(['var x: bool = -1'], 'E1012:')
58 v9.CheckDefAndScriptFailure(['var x: bool = [1]'], 'E1012:')
59 v9.CheckDefAndScriptFailure(['var x: bool = {}'], 'E1012:')
60 v9.CheckDefAndScriptFailure(['var x: bool = "x"'], 'E1012:')
Bram Moolenaar9a562c12021-01-23 13:39:14 +010061
Bram Moolenaar62aec932022-01-29 21:45:34 +000062 v9.CheckDefAndScriptFailure(['var x: bool = "x"', '', 'eval 0'], 'E1012:', 1)
Bram Moolenaar971caf52020-09-21 22:21:42 +020063enddef
64
65def Test_syntax()
Bram Moolenaar98309382020-09-27 21:58:45 +020066 var name = 234
Bram Moolenaar30fd8202020-09-26 15:09:30 +020067 var other: list<string> = ['asdf']
Bram Moolenaar971caf52020-09-21 22:21:42 +020068enddef
69
70def Test_assignment()
Bram Moolenaar62aec932022-01-29 21:45:34 +000071 v9.CheckDefFailure(['var x:string'], 'E1069:')
72 v9.CheckDefFailure(['var x:string = "x"'], 'E1069:')
73 v9.CheckDefFailure(['var a:string = "x"'], 'E1069:')
74 v9.CheckDefFailure(['var lambda = () => "lambda"'], 'E704:')
75 v9.CheckScriptFailure(['var x = "x"'], 'E1124:')
Bram Moolenaar971caf52020-09-21 22:21:42 +020076
Bram Moolenaar3f327882021-03-17 20:56:38 +010077 # lower case name is OK for a list
78 var lambdaLines =<< trim END
Bram Moolenaar848fadd2022-01-30 15:28:30 +000079 var lambdaList: list<func> = [g:Test_syntax]
Bram Moolenaar3f327882021-03-17 20:56:38 +010080 lambdaList[0] = () => "lambda"
81 END
Bram Moolenaar62aec932022-01-29 21:45:34 +000082 v9.CheckDefAndScriptSuccess(lambdaLines)
Bram Moolenaar3f327882021-03-17 20:56:38 +010083
Bram Moolenaar30fd8202020-09-26 15:09:30 +020084 var nr: number = 1234
Bram Moolenaar62aec932022-01-29 21:45:34 +000085 v9.CheckDefFailure(['var nr: number = "asdf"'], 'E1012:')
Bram Moolenaar971caf52020-09-21 22:21:42 +020086
Bram Moolenaar30fd8202020-09-26 15:09:30 +020087 var a: number = 6 #comment
Bram Moolenaar971caf52020-09-21 22:21:42 +020088 assert_equal(6, a)
89
90 if has('channel')
Bram Moolenaar30fd8202020-09-26 15:09:30 +020091 var chan1: channel
Bram Moolenaar218450a2020-10-17 18:51:52 +020092 assert_equal('fail', ch_status(chan1))
93
Bram Moolenaar30fd8202020-09-26 15:09:30 +020094 var job1: job
Bram Moolenaar218450a2020-10-17 18:51:52 +020095 assert_equal('fail', job_status(job1))
96
Bram Moolenaarfc8aa6d2020-10-12 20:31:26 +020097 # calling job_start() is in test_vim9_fails.vim, it causes leak reports
Bram Moolenaar971caf52020-09-21 22:21:42 +020098 endif
Bram Moolenaar73e28dc2022-09-17 21:08:33 +010099 var float1: float = 3.4
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
Bram Moolenaar62aec932022-01-29 21:45:34 +0000124 v9.CheckScriptSuccess(lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200125
Bram Moolenaara749a422022-02-12 19:52:25 +0000126 appendToMe ..= 'yyy'
127 assert_equal('xxxyyy', appendToMe)
128 addToMe += 222
129 assert_equal(333, addToMe)
130 newVar = 'new'
131 assert_equal('new', newVar)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200132
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
zeertzjq4c7cb372023-06-14 16:39:54 +0100148 assert_fails('&ts /= 0', ['E1154:', 'E1154:'])
149 assert_fails('&ts %= 0', ['E1154:', 'E1154:'])
150 assert_fails('&ts /= []', ['E745:', 'E745:'])
151 assert_fails('&ts %= []', ['E745:', 'E745:'])
152 assert_equal(2, &ts)
153
Bram Moolenaar73e28dc2022-09-17 21:08:33 +0100154 var f100: float = 100.0
155 f100 /= 5
156 assert_equal(20.0, f100)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200157
Bram Moolenaar73e28dc2022-09-17 21:08:33 +0100158 var f200: float = 200.0
159 f200 /= 5.0
160 assert_equal(40.0, f200)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200161
Bram Moolenaar73e28dc2022-09-17 21:08:33 +0100162 v9.CheckDefFailure(['var nr: number = 200', 'nr /= 5.0'], 'E1012:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200163
164 lines =<< trim END
165 &ts = 6
166 &ts += 3
167 assert_equal(9, &ts)
168
169 &l:ts = 6
170 assert_equal(6, &ts)
171 &l:ts += 2
172 assert_equal(8, &ts)
173
174 &g:ts = 6
175 assert_equal(6, &g:ts)
176 &g:ts += 2
177 assert_equal(8, &g:ts)
Bram Moolenaar0ea04402021-01-04 13:37:54 +0100178
179 &number = true
180 assert_equal(true, &number)
181 &number = 0
182 assert_equal(false, &number)
183 &number = 1
184 assert_equal(true, &number)
185 &number = false
186 assert_equal(false, &number)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200187 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000188 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200189
Bram Moolenaar62aec932022-01-29 21:45:34 +0000190 v9.CheckDefFailure(['&notex += 3'], 'E113:')
191 v9.CheckDefFailure(['&ts ..= "xxx"'], 'E1019:')
192 v9.CheckDefFailure(['&ts = [7]'], 'E1012:')
193 v9.CheckDefExecFailure(['&ts = g:alist'], 'E1012: Type mismatch; expected number but got list<number>')
194 v9.CheckDefFailure(['&ts = "xx"'], 'E1012:')
195 v9.CheckDefExecFailure(['&ts = g:astring'], 'E1012: Type mismatch; expected number but got string')
196 v9.CheckDefFailure(['&path += 3'], 'E1012:')
197 v9.CheckDefExecFailure(['&bs = "asdf"'], 'E474:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200198 # test freeing ISN_STOREOPT
Bram Moolenaar62aec932022-01-29 21:45:34 +0000199 v9.CheckDefFailure(['&ts = 3', 'var asdf'], 'E1022:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200200 &ts = 8
201
202 lines =<< trim END
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200203 var save_TI = &t_TI
Bram Moolenaar971caf52020-09-21 22:21:42 +0200204 &t_TI = ''
205 assert_equal('', &t_TI)
206 &t_TI = 'xxx'
207 assert_equal('xxx', &t_TI)
208 &t_TI = save_TI
209 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000210 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +0200211
Bram Moolenaar62aec932022-01-29 21:45:34 +0000212 v9.CheckDefFailure(['&t_TI = 123'], 'E1012:')
213 v9.CheckScriptFailure(['vim9script', '&t_TI = 123'], 'E928:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200214
Bram Moolenaar62aec932022-01-29 21:45:34 +0000215 v9.CheckDefFailure(['var s:var = 123'], 'E1101:')
216 v9.CheckDefFailure(['var s:var: number'], 'E1101:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200217
Bram Moolenaar5cb53b72022-05-26 19:54:05 +0100218 v9.CheckDefAndScriptFailure(['var $VAR: number'], ['E1016:', 'E475:'])
219
Bram Moolenaar971caf52020-09-21 22:21:42 +0200220 lines =<< trim END
221 vim9script
222 def SomeFunc()
223 s:var = 123
224 enddef
225 defcompile
226 END
Bram Moolenaarafa048f2022-02-22 20:43:36 +0000227 v9.CheckScriptFailure(lines, 'E1268:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200228
229 g:inc_counter += 1
230 assert_equal(2, g:inc_counter)
231
Bram Moolenaar73e28dc2022-09-17 21:08:33 +0100232 var f: float
233 f += 1
234 assert_equal(1.0, f)
Bram Moolenaar7bf9a072021-08-02 21:55:15 +0200235
Bram Moolenaar971caf52020-09-21 22:21:42 +0200236 $SOME_ENV_VAR ..= 'more'
237 assert_equal('somemore', $SOME_ENV_VAR)
Bram Moolenaar62aec932022-01-29 21:45:34 +0000238 v9.CheckDefFailure(['$SOME_ENV_VAR += "more"'], 'E1051:')
239 v9.CheckDefFailure(['$SOME_ENV_VAR += 123'], 'E1012:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200240
Bram Moolenaar971caf52020-09-21 22:21:42 +0200241 v:errmsg = 'none'
242 v:errmsg ..= 'again'
243 assert_equal('noneagain', v:errmsg)
Bram Moolenaar62aec932022-01-29 21:45:34 +0000244 v9.CheckDefFailure(['v:errmsg += "more"'], 'E1051:')
245 v9.CheckDefFailure(['v:errmsg += 123'], 'E1012:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200246
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 Moolenaaraf647e72021-08-05 19:01:17 +0200252def Test_float_and_number()
Bram Moolenaar73e28dc2022-09-17 21:08:33 +0100253 var lines =<< trim END
254 var f: float
255 f += 2
256 f -= 1
257 assert_equal(1.0, f)
258 ++f
259 --f
260 assert_equal(1.0, f)
261 END
262 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaaraf647e72021-08-05 19:01:17 +0200263enddef
264
Bram Moolenaarf5d52c92021-07-31 22:51:10 +0200265let g:someNumber = 43
266
267def Test_assign_concat()
268 var lines =<< trim END
269 var s = '-'
270 s ..= 99
271 s ..= true
272 s ..= '-'
273 s ..= v:null
274 s ..= g:someNumber
275 assert_equal('-99true-null43', s)
276 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000277 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaarf5d52c92021-07-31 22:51:10 +0200278
279 lines =<< trim END
280 var s = '-'
281 s ..= [1, 2]
282 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000283 v9.CheckDefAndScriptFailure(lines, ['E1105: Cannot convert list to string', 'E734: Wrong variable type for .='], 2)
Bram Moolenaarf5d52c92021-07-31 22:51:10 +0200284 lines =<< trim END
285 var s = '-'
286 s ..= {a: 2}
287 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000288 v9.CheckDefAndScriptFailure(lines, ['E1105: Cannot convert dict to string', 'E734: Wrong variable type for .='], 2)
Bram Moolenaar22ebd172022-04-01 15:26:58 +0100289
290 lines =<< trim END
291 var ls: list<string> = []
292 ls[-1] ..= 'foo'
293 END
Bram Moolenaarec892232022-05-06 17:53:06 +0100294 v9.CheckDefExecAndScriptFailure(lines, 'E684: List index out of range: -1', 2)
Bram Moolenaarf5d52c92021-07-31 22:51:10 +0200295enddef
296
Bram Moolenaar13e45d12021-06-26 13:28:35 +0200297def Test_assign_register()
298 var lines =<< trim END
299 @c = 'areg'
300 @c ..= 'add'
301 assert_equal('aregadd', @c)
302
303 @@ = 'some text'
304 assert_equal('some text', getreg('"'))
305 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000306 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar13e45d12021-06-26 13:28:35 +0200307
Bram Moolenaar62aec932022-01-29 21:45:34 +0000308 v9.CheckDefFailure(['@a += "more"'], 'E1051:')
309 v9.CheckDefFailure(['@a += 123'], 'E1012:')
Bram Moolenaar13e45d12021-06-26 13:28:35 +0200310enddef
311
Bram Moolenaard0edaf92021-05-28 21:06:08 +0200312def Test_reserved_name()
Bram Moolenaar8acb9cc2022-03-08 13:18:55 +0000313 var more_names = ['null_job', 'null_channel']
314 if !has('job')
315 more_names = []
316 endif
317
318 for name in ['true',
319 'false',
Bram Moolenaarce723f32023-06-10 19:00:12 +0100320 'this',
321 'super',
Bram Moolenaar8acb9cc2022-03-08 13:18:55 +0000322 'null',
323 'null_blob',
324 'null_dict',
325 'null_function',
326 'null_list',
327 'null_partial',
328 'null_string',
329 ] + more_names
Bram Moolenaar62aec932022-01-29 21:45:34 +0000330 v9.CheckDefExecAndScriptFailure(['var ' .. name .. ' = 0'], 'E1034:')
331 v9.CheckDefExecAndScriptFailure(['var ' .. name .. ': bool'], 'E1034:')
Bram Moolenaard0edaf92021-05-28 21:06:08 +0200332 endfor
Bram Moolenaarce723f32023-06-10 19:00:12 +0100333
334 var lines =<< trim END
335 vim9script
336 def Foo(super: bool)
337 echo 'something'
338 enddef
339 defcompile
340 END
341 v9.CheckScriptFailure(lines, 'E1034:')
Bram Moolenaard0edaf92021-05-28 21:06:08 +0200342enddef
343
Bram Moolenaar8acb9cc2022-03-08 13:18:55 +0000344def Test_null_values()
345 var lines =<< trim END
346 var b: blob = null_blob
347 var dn: dict<number> = null_dict
348 var ds: dict<string> = null_dict
349 var ln: list<number> = null_list
350 var ls: list<string> = null_list
351 var Ff: func(string): string = null_function
352 var Fp: func(number): number = null_partial
353 var s: string = null_string
354 if has('job')
355 var j: job = null_job
356 var c: channel = null_channel
357 endif
Bram Moolenaar56b84b12022-03-09 19:46:48 +0000358
359 var d: dict<func> = {a: function('tr'), b: null_function}
Bram Moolenaarbadf04f2022-03-12 21:28:22 +0000360
361 var bl: list<blob> = [0z12, null_blob]
362 var dnl: list<dict<number>> = [{a: 1}, null_dict]
363 var dsl: list<dict<string>> = [{a: 'x'}, null_dict]
364 var lnl: list<list<number>> = [[1], null_list]
365 var lsl: list<list<string>> = [['x'], null_list]
366 def Len(v: string): number
367 return len(v)
368 enddef
369 var Ffl: list<func(string): number> = [Len, null_function]
370 var Fpl: list<func(string): number> = [Len, null_partial]
371 var sl: list<string> = ['x', null_string]
372 if has('job')
373 var jl: list<job> = [null_job]
374 var cl: list<channel> = [null_channel]
375 endif
Bram Moolenaar8acb9cc2022-03-08 13:18:55 +0000376 END
377 v9.CheckDefAndScriptSuccess(lines)
378enddef
379
Bram Moolenaarce93d162023-01-30 21:12:34 +0000380def Test_type_with_extra_white()
381 var lines =<< trim END
382 const x : number = 3
383 END
384 v9.CheckDefExecAndScriptFailure(lines, 'E1059')
385enddef
386
Bram Moolenaar56b84b12022-03-09 19:46:48 +0000387def Test_keep_type_after_assigning_null()
388 var lines =<< trim END
389 var b: blob
390 b = null_blob
391 b = 'text'
392 END
393 v9.CheckDefExecAndScriptFailure(lines, 'E1012: Type mismatch; expected blob but got string')
394
395 lines =<< trim END
396 var l: list<number>
397 l = null_list
398 l = ['text']
399 END
400 v9.CheckDefExecAndScriptFailure(lines, 'E1012: Type mismatch; expected list<number> but got list<string>')
401
402 lines =<< trim END
403 var d: dict<string>
404 d = null_dict
405 d = {a: 1, b: 2}
406 END
407 v9.CheckDefExecAndScriptFailure(lines, 'E1012: Type mismatch; expected dict<string> but got dict<number>')
408enddef
409
Bram Moolenaaraa1959b2021-04-15 22:13:39 +0200410def Test_skipped_assignment()
411 var lines =<< trim END
412 for x in []
413 var i: number = 1
414 while false
415 i += 1
416 endwhile
417 endfor
418 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000419 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaaraa1959b2021-04-15 22:13:39 +0200420enddef
421
Bram Moolenaar7824fc82021-11-26 17:36:51 +0000422def Test_assign_keep_type()
423 var lines =<< trim END
424 vim9script
425 var l: list<number> = [123]
426 l = [123]
427 l->add('string')
428 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000429 v9.CheckScriptFailure(lines, 'E1012:', 4)
Bram Moolenaar7824fc82021-11-26 17:36:51 +0000430enddef
431
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100432def Test_assign_unpack()
433 var lines =<< trim END
434 var v1: number
435 var v2: number
436 [v1, v2] = [1, 2]
437 assert_equal(1, v1)
438 assert_equal(2, v2)
Bram Moolenaarf93bbd02021-04-10 22:35:43 +0200439
440 [v1, _, v2, _] = [1, 99, 2, 77]
441 assert_equal(1, v1)
442 assert_equal(2, v2)
443
444 [v1, v2; _] = [1, 2, 3, 4, 5]
445 assert_equal(1, v1)
446 assert_equal(2, v2)
Bram Moolenaar035bd1c2021-06-21 19:44:11 +0200447
Bram Moolenaarbc510062022-02-14 21:19:04 +0000448 var _x: number
449 [_x, v2] = [6, 7]
450 assert_equal(6, _x)
451 assert_equal(7, v2)
452
Bram Moolenaar4d5dfe22021-06-26 13:59:29 +0200453 var reslist = []
454 for text in ['aaa {bbb} ccc', 'ddd {eee} fff']
455 var before: string
456 var middle: string
457 var after: string
458 [_, before, middle, after; _] = text->matchlist('\(.\{-\}\){\(.\{-\}\)}\(.*\)')
459 reslist->add(before)->add(middle)->add(after)
460 endfor
461 assert_equal(['aaa ', 'bbb', ' ccc', 'ddd ', 'eee', ' fff'], reslist)
462
Bram Moolenaar035bd1c2021-06-21 19:44:11 +0200463 var a = 1
464 var b = 3
465 [a, b] += [2, 4]
466 assert_equal(3, a)
467 assert_equal(7, b)
468
469 [a, b] -= [1, 2]
470 assert_equal(2, a)
471 assert_equal(5, b)
472
473 [a, b] *= [3, 2]
474 assert_equal(6, a)
475 assert_equal(10, b)
476
477 [a, b] /= [2, 4]
478 assert_equal(3, a)
479 assert_equal(2, b)
480
481 [a, b] = [17, 15]
482 [a, b] %= [5, 3]
483 assert_equal(2, a)
484 assert_equal(0, b)
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100485 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000486 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100487
488 lines =<< trim END
489 var v1: number
490 var v2: number
Bram Moolenaardb9ff9a2021-12-01 17:38:01 +0000491 [v1, v2] = [1, 2, 3]
492 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000493 v9.CheckDefFailure(lines, 'E1093: Expected 2 items but got 3', 3)
Bram Moolenaardb9ff9a2021-12-01 17:38:01 +0000494
495 lines =<< trim END
496 var v1: number
497 var v2: number
498 [v1, v2] = [1]
499 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000500 v9.CheckDefFailure(lines, 'E1093: Expected 2 items but got 1', 3)
Bram Moolenaardb9ff9a2021-12-01 17:38:01 +0000501
502 lines =<< trim END
503 var v1: number
504 var v2: number
505 [v1, v2; _] = [1]
506 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000507 v9.CheckDefFailure(lines, 'E1093: Expected 2 items but got 1', 3)
Bram Moolenaardb9ff9a2021-12-01 17:38:01 +0000508
509 lines =<< trim END
510 var v1: number
511 var v2: number
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100512 [v1, v2] =
513 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000514 v9.CheckDefFailure(lines, 'E1097:', 5)
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100515
516 lines =<< trim END
517 var v1: number
518 var v2: number
519 [v1, v2] = xxx
520 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000521 v9.CheckDefFailure(lines, 'E1001:', 3)
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100522
523 lines =<< trim END
524 var v1: number
525 var v2: number
526 [v1, v2] = popup_clear()
527 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000528 v9.CheckDefFailure(lines, 'E1031:', 3)
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100529
530 lines =<< trim END
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100531 [v1, v2] = [1, 2]
532 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000533 v9.CheckDefFailure(lines, 'E1089', 1)
534 v9.CheckScriptFailure(['vim9script'] + lines, 'E1089', 2)
Bram Moolenaar3862ea32021-01-01 21:05:55 +0100535
536 lines =<< trim END
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100537 var v1: number
538 var v2: number
539 [v1, v2] = ''
540 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000541 v9.CheckDefFailure(lines, 'E1012: Type mismatch; expected list<any> but got string', 3)
Bram Moolenaarf785aa12021-02-11 21:19:34 +0100542
543 lines =<< trim END
544 g:values = [false, 0]
545 var x: bool
546 var y: string
547 [x, y] = g:values
548 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000549 v9.CheckDefExecAndScriptFailure(lines, 'E1163: Variable 2: type mismatch, expected string but got number')
Bram Moolenaar4270d8b2021-08-07 16:30:42 +0200550
551 lines =<< trim END
552 var x: number
553 var y: number
554 var z: string
555 [x, y, z] = [1, 2, 3]
556 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000557 v9.CheckDefAndScriptFailure(lines, 'E1163: Variable 3: type mismatch, expected string but got number')
Bram Moolenaar4270d8b2021-08-07 16:30:42 +0200558
559 lines =<< trim END
560 var x: number
561 var y: string
562 var z: string
563 [x, y, z] = [1, '2', 3]
564 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000565 v9.CheckDefExecAndScriptFailure(lines, 'E1163: Variable 3: type mismatch, expected string but got number')
Bram Moolenaar8ff16e02020-12-07 21:49:52 +0100566enddef
567
Bram Moolenaar7f764942020-12-02 15:11:18 +0100568def Test_assign_linebreak()
569 var nr: number
570 nr =
571 123
572 assert_equal(123, nr)
573
574 var n2: number
575 [nr, n2] =
576 [12, 34]
577 assert_equal(12, nr)
578 assert_equal(34, n2)
579
Bram Moolenaar62aec932022-01-29 21:45:34 +0000580 v9.CheckDefFailure(["var x = #"], 'E1097:', 3)
Bram Moolenaar77709b12021-04-03 21:01:01 +0200581
582 var lines =<< trim END
583 var x: list<string> = ['a']
584 var y: list<number> = x
585 ->copy()
586 ->copy()
587 END
Bram Moolenaar381692b2022-02-02 20:01:27 +0000588 v9.CheckDefExecFailure(lines, 'E1012:', 4)
Bram Moolenaar6977dba2021-07-04 22:48:12 +0200589
590 lines =<< trim END
591 var x: any
592 x.key = 1
593 + 2
594 + 3
595 + 4
596 + 5
597 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000598 v9.CheckDefExecAndScriptFailure(lines, ['E1148:', 'E1203:'], 2)
Bram Moolenaar7f764942020-12-02 15:11:18 +0100599enddef
600
Bram Moolenaardc234ca2020-11-28 18:52:33 +0100601def Test_assign_index()
602 # list of list
603 var l1: list<number>
604 l1[0] = 123
605 assert_equal([123], l1)
606
607 var l2: list<list<number>>
608 l2[0] = []
609 l2[0][0] = 123
610 assert_equal([[123]], l2)
611
612 var l3: list<list<list<number>>>
613 l3[0] = []
614 l3[0][0] = []
615 l3[0][0][0] = 123
616 assert_equal([[[123]]], l3)
617
618 var lines =<< trim END
619 var l3: list<list<number>>
620 l3[0] = []
621 l3[0][0] = []
622 END
Yegappan Lakshmanan66897192023-12-05 15:51:50 +0100623 v9.CheckDefFailure(lines, 'E1012: Type mismatch; expected number but got list<any>', 3)
Bram Moolenaardc234ca2020-11-28 18:52:33 +0100624
625 # dict of dict
626 var d1: dict<number>
627 d1.one = 1
628 assert_equal({one: 1}, d1)
629
630 var d2: dict<dict<number>>
631 d2.one = {}
632 d2.one.two = 123
633 assert_equal({one: {two: 123}}, d2)
634
635 var d3: dict<dict<dict<number>>>
636 d3.one = {}
637 d3.one.two = {}
638 d3.one.two.three = 123
639 assert_equal({one: {two: {three: 123}}}, d3)
640
Bram Moolenaare08be092022-02-17 13:08:26 +0000641 # blob
642 var bl: blob = 0z11223344
643 bl[0] = 0x77
644 assert_equal(0z77223344, bl)
645 bl[-2] = 0x66
646 assert_equal(0z77226644, bl)
647
Bram Moolenaar1983f1a2022-02-28 20:55:02 +0000648 lines =<< trim END
649 g:val = '22'
650 var bl = 0z11
651 bl[1] = g:val
652 END
Yegappan Lakshmanan9937d8b2024-05-08 20:24:33 +0200653 v9.CheckDefExecAndScriptFailure(lines, ['E1030: Using a String as a Number: "22"', 'E1012: Type mismatch; expected number but got string'])
Bram Moolenaar1983f1a2022-02-28 20:55:02 +0000654
Bram Moolenaare97976b2021-08-01 13:17:17 +0200655 # should not read the next line when generating "a.b"
656 var a = {}
657 a.b = {}
658 a.b.c = {}
659 ->copy()
660
Bram Moolenaardc234ca2020-11-28 18:52:33 +0100661 lines =<< trim END
662 var d3: dict<dict<number>>
663 d3.one = {}
664 d3.one.two = {}
665 END
Yegappan Lakshmanan66897192023-12-05 15:51:50 +0100666 v9.CheckDefFailure(lines, 'E1012: Type mismatch; expected number but got dict<any>', 3)
Bram Moolenaardc234ca2020-11-28 18:52:33 +0100667
Bram Moolenaaracbae182020-12-13 18:44:43 +0100668 lines =<< trim END
669 var lines: list<string>
670 lines['a'] = 'asdf'
671 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000672 v9.CheckDefFailure(lines, 'E1012:', 2)
Bram Moolenaaracbae182020-12-13 18:44:43 +0100673
674 lines =<< trim END
675 var lines: string
676 lines[9] = 'asdf'
677 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000678 v9.CheckDefFailure(lines, 'E1141:', 2)
Bram Moolenaaracbae182020-12-13 18:44:43 +0100679
Bram Moolenaardc234ca2020-11-28 18:52:33 +0100680 # list of dict
681 var ld: list<dict<number>>
682 ld[0] = {}
683 ld[0].one = 123
684 assert_equal([{one: 123}], ld)
685
686 lines =<< trim END
687 var ld: list<dict<number>>
688 ld[0] = []
689 END
Yegappan Lakshmanan66897192023-12-05 15:51:50 +0100690 v9.CheckDefFailure(lines, 'E1012: Type mismatch; expected dict<number> but got list<any>', 2)
Bram Moolenaardc234ca2020-11-28 18:52:33 +0100691
692 # dict of list
693 var dl: dict<list<number>>
694 dl.one = []
695 dl.one[0] = 123
696 assert_equal({one: [123]}, dl)
697
698 lines =<< trim END
699 var dl: dict<list<number>>
700 dl.one = {}
701 END
Yegappan Lakshmanan66897192023-12-05 15:51:50 +0100702 v9.CheckDefFailure(lines, 'E1012: Type mismatch; expected list<number> but got dict<any>', 2)
Bram Moolenaare08be092022-02-17 13:08:26 +0000703
704 lines =<< trim END
705 g:l = [1, 2]
706 g:l['x'] = 3
707 END
708 v9.CheckDefExecAndScriptFailure(lines, ['E39:', 'E1030:'], 2)
709
710 lines =<< trim END
711 var bl: blob = test_null_blob()
712 bl[1] = 8
713 END
714 v9.CheckDefExecAndScriptFailure(lines, ['E1184:', 'E979:'], 2)
Bram Moolenaarea5c8982022-02-17 14:42:02 +0000715
716 lines =<< trim END
717 g:bl = 'not a blob'
718 g:bl[1 : 2] = 8
719 END
720 v9.CheckDefExecAndScriptFailure(lines, ['E897:', 'E689:'], 2)
Bram Moolenaardc234ca2020-11-28 18:52:33 +0100721enddef
722
Bram Moolenaar38ecd972022-01-15 21:44:44 +0000723def Test_init_in_for_loop()
724 var lines =<< trim END
725 var l: list<number> = []
726 for i in [3, 4]
727 var n: number
728 add(l, n)
729 n = 123
730 endfor
731 assert_equal([0, 0], l)
732 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000733 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar38ecd972022-01-15 21:44:44 +0000734
735 lines =<< trim END
736 var l: list<number> = []
737 for i in [3, 4]
738 var n: number = 0
739 add(l, n)
740 n = 123
741 endfor
742 assert_equal([0, 0], l)
743 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000744 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar38ecd972022-01-15 21:44:44 +0000745
746 lines =<< trim END
747 var l: list<number> = []
748 for i in [3, 4]
749 var n: number = 3
750 add(l, n)
751 n = 123
752 endfor
753 assert_equal([3, 3], l)
754 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000755 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar38ecd972022-01-15 21:44:44 +0000756enddef
757
Bram Moolenaarbc6f2bf2022-05-06 14:29:09 +0100758def Test_redir_is_not_assign()
759 if false
760 redir => res
761 echo var_job
762 redir END
763 endif
764enddef
765
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200766def Test_extend_list()
Yegappan Lakshmananefbfa862022-04-17 12:47:40 +0100767 # using uninitialized list assigns empty list
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200768 var lines =<< trim END
Bram Moolenaar07802042021-09-09 23:01:14 +0200769 var l1: list<number>
770 var l2 = l1
771 assert_true(l1 is l2)
772 l1 += [123]
773 assert_equal([123], l1)
774 assert_true(l1 is l2)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100775 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000776 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200777
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100778 lines =<< trim END
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100779 var list: list<string>
780 extend(list, ['x'])
781 assert_equal(['x'], list)
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200782 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000783 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar1a739232020-10-10 15:37:58 +0200784
Yegappan Lakshmananefbfa862022-04-17 12:47:40 +0100785 # appending to uninitialized list from a function works
Bram Moolenaar1a739232020-10-10 15:37:58 +0200786 lines =<< trim END
787 vim9script
788 var list: list<string>
789 def Func()
790 list += ['a', 'b']
791 enddef
792 Func()
793 assert_equal(['a', 'b'], list)
794 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000795 v9.CheckScriptSuccess(lines)
Bram Moolenaar3beaf9c2020-12-18 17:23:14 +0100796 lines =<< trim END
797 vim9script
798 var list: list<string>
799 def Func()
800 extend(list, ['x', 'b'])
801 enddef
802 Func()
803 assert_equal(['x', 'b'], list)
804 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000805 v9.CheckScriptSuccess(lines)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100806
Bram Moolenaaraa7d0c22022-04-05 21:40:38 +0100807 # initialized to null, with type, does not default to empty list
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100808 lines =<< trim END
809 vim9script
810 var l: list<string> = test_null_list()
811 extend(l, ['x'])
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100812 END
Bram Moolenaaraa7d0c22022-04-05 21:40:38 +0100813 v9.CheckScriptFailure(lines, 'E1134:', 3)
814
815 # initialized to null, without type, does not default to empty list
816 lines =<< trim END
817 vim9script
818 var l = null_list
819 extend(l, ['x'])
820 END
821 v9.CheckScriptFailure(lines, 'E1134:', 3)
822
823 # assigned null, does not default to empty list
824 lines =<< trim END
825 vim9script
826 var l: list<string>
827 l = null_list
828 extend(l, ['x'])
829 END
830 v9.CheckScriptFailure(lines, 'E1134:', 4)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100831
832 lines =<< trim END
833 vim9script
834 extend(test_null_list(), ['x'])
835 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000836 v9.CheckScriptFailure(lines, 'E1134:', 2)
Bram Moolenaar078a4612022-01-04 15:17:03 +0000837
838 # using global var has no declared type
839 g:myList = []
840 g:myList->extend([1])
841 g:myList->extend(['x'])
842 assert_equal([1, 'x'], g:myList)
843 unlet g:myList
844
845 # using declared list gives an error
846 lines =<< trim END
847 var l: list<number>
848 g:myList = l
849 g:myList->extend([1])
850 g:myList->extend(['x'])
851 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000852 v9.CheckDefExecAndScriptFailure(lines, 'E1013: Argument 2: type mismatch, expected list<number> but got list<string>', 4)
Bram Moolenaar078a4612022-01-04 15:17:03 +0000853 unlet g:myList
Bram Moolenaar160afdb2022-02-06 17:17:02 +0000854
855 lines =<< trim END
856 vim9script
857 var lds = [1, 2, 3]
858 def Func()
859 echo lds->extend(['x'])
860 enddef
861 defcompile
862 END
863 v9.CheckScriptFailure(lines, 'E1013:')
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100864enddef
865
866def Test_extend_dict()
867 var lines =<< trim END
868 vim9script
869 var d: dict<number>
Bram Moolenaare0de1712020-12-02 17:36:54 +0100870 extend(d, {a: 1})
871 assert_equal({a: 1}, d)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100872
873 var d2: dict<number>
874 d2['one'] = 1
Bram Moolenaare0de1712020-12-02 17:36:54 +0100875 assert_equal({one: 1}, d2)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100876 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000877 v9.CheckScriptSuccess(lines)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100878
879 lines =<< trim END
880 vim9script
881 var d: dict<string> = test_null_dict()
Bram Moolenaare0de1712020-12-02 17:36:54 +0100882 extend(d, {a: 'x'})
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100883 END
Bram Moolenaaraa7d0c22022-04-05 21:40:38 +0100884 v9.CheckScriptFailure(lines, 'E1133:', 3)
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100885
886 lines =<< trim END
887 vim9script
Bram Moolenaare0de1712020-12-02 17:36:54 +0100888 extend(test_null_dict(), {a: 'x'})
Bram Moolenaar348be7e2020-11-04 11:36:35 +0100889 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000890 v9.CheckScriptFailure(lines, 'E1133:', 2)
Bram Moolenaar81ed4962020-09-23 15:56:50 +0200891enddef
892
Bram Moolenaar971caf52020-09-21 22:21:42 +0200893def Test_single_letter_vars()
894 # single letter variables
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200895 var a: number = 123
Bram Moolenaar971caf52020-09-21 22:21:42 +0200896 a = 123
897 assert_equal(123, a)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200898 var b: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200899 b = 123
900 assert_equal(123, b)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200901 var g: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200902 g = 123
903 assert_equal(123, g)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200904 var s: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200905 s = 123
906 assert_equal(123, s)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200907 var t: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200908 t = 123
909 assert_equal(123, t)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200910 var v: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200911 v = 123
912 assert_equal(123, v)
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200913 var w: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200914 w = 123
915 assert_equal(123, w)
916enddef
917
918def Test_vim9_single_char_vars()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200919 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +0200920 vim9script
921
922 # single character variable declarations work
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200923 var a: string
924 var b: number
925 var l: list<any>
926 var s: string
927 var t: number
928 var v: number
929 var w: number
Bram Moolenaar971caf52020-09-21 22:21:42 +0200930
931 # script-local variables can be used without s: prefix
932 a = 'script-a'
933 b = 111
934 l = [1, 2, 3]
935 s = 'script-s'
936 t = 222
937 v = 333
938 w = 444
939
940 assert_equal('script-a', a)
941 assert_equal(111, b)
942 assert_equal([1, 2, 3], l)
943 assert_equal('script-s', s)
944 assert_equal(222, t)
945 assert_equal(333, v)
946 assert_equal(444, w)
947 END
Bram Moolenaar0e9bdad2022-10-15 20:06:33 +0100948 writefile(lines, 'Xsinglechar', 'D')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200949 source Xsinglechar
Bram Moolenaar971caf52020-09-21 22:21:42 +0200950enddef
951
952def Test_assignment_list()
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200953 var list1: list<bool> = [false, true, false]
954 var list2: list<number> = [1, 2, 3]
955 var list3: list<string> = ['sdf', 'asdf']
956 var list4: list<any> = ['yes', true, 1234]
957 var list5: list<blob> = [0z01, 0z02]
Bram Moolenaar971caf52020-09-21 22:21:42 +0200958
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200959 var listS: list<string> = []
960 var listN: list<number> = []
Bram Moolenaar971caf52020-09-21 22:21:42 +0200961
962 assert_equal([1, 2, 3], list2)
963 list2[-1] = 99
964 assert_equal([1, 2, 99], list2)
965 list2[-2] = 88
966 assert_equal([1, 88, 99], list2)
967 list2[-3] = 77
968 assert_equal([77, 88, 99], list2)
969 list2 += [100]
970 assert_equal([77, 88, 99, 100], list2)
971
972 list3 += ['end']
973 assert_equal(['sdf', 'asdf', 'end'], list3)
974
Bram Moolenaar62aec932022-01-29 21:45:34 +0000975 v9.CheckDefExecFailure(['var ll = [1, 2, 3]', 'll[-4] = 6'], 'E684:')
Bram Moolenaar6296d1e2022-02-17 16:30:11 +0000976 v9.CheckDefExecFailure(['var ll = [1, 2, 3]', 'unlet ll[8 : 9]'], 'E684:')
977 v9.CheckDefExecFailure(['var ll = [1, 2, 3]', 'unlet ll[1 : -9]'], 'E684:')
978 v9.CheckDefExecFailure(['var ll = [1, 2, 3]', 'unlet ll[2 : 1]'], 'E684:')
Bram Moolenaar971caf52020-09-21 22:21:42 +0200979
980 # type becomes list<any>
Bram Moolenaar30fd8202020-09-26 15:09:30 +0200981 var somelist = rand() > 0 ? [1, 2, 3] : ['a', 'b', 'c']
Bram Moolenaar4f5e3972020-12-21 17:30:50 +0100982
Bram Moolenaar114dbda2022-01-03 12:28:03 +0000983 # type is list<any> even though initializer is list<number>
984 var anyList: list<any> = [0]
985 assert_equal([0, 'x'], extend(anyList, ['x']))
986
Bram Moolenaar4f5e3972020-12-21 17:30:50 +0100987 var lines =<< trim END
988 var d = {dd: test_null_list()}
989 d.dd[0] = 0
990 END
Bram Moolenaar62aec932022-01-29 21:45:34 +0000991 v9.CheckDefExecFailure(lines, 'E1147:', 2)
Bram Moolenaar7a40ff02021-07-04 15:54:08 +0200992
993 lines =<< trim END
994 def OneArg(x: bool)
995 enddef
996 def TwoArgs(x: bool, y: bool)
997 enddef
998 var fl: list<func(bool, bool, bool)> = [OneArg, TwoArgs]
999 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001000 v9.CheckDefExecAndScriptFailure(lines, 'E1012:', 5)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001001enddef
1002
Bram Moolenaarab36e6a2021-11-30 16:14:49 +00001003def Test_list_declaration()
1004 var [v1, v2] = [1, 2]
1005 v1 += 3
1006 assert_equal(4, v1)
1007 v2 *= 3
1008 assert_equal(6, v2)
1009
1010 var lines =<< trim END
1011 var [v1, v2] = [1]
1012 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001013 v9.CheckDefExecAndScriptFailure(lines, ['E1093: Expected 2 items but got 1', 'E688:'])
Bram Moolenaarab36e6a2021-11-30 16:14:49 +00001014 lines =<< trim END
1015 var testlist = [1]
1016 var [v1, v2] = testlist
1017 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001018 v9.CheckDefExecAndScriptFailure(lines, ['E1093: Expected 2 items but got 1', 'E688:'])
Bram Moolenaarab36e6a2021-11-30 16:14:49 +00001019 lines =<< trim END
1020 var [v1, v2] = [1, 2, 3]
1021 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001022 v9.CheckDefExecAndScriptFailure(lines, ['E1093: Expected 2 items but got 3', 'E687:'])
Bram Moolenaarab36e6a2021-11-30 16:14:49 +00001023 lines =<< trim END
1024 var testlist = [1, 2, 3]
1025 var [v1, v2] = testlist
1026 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001027 v9.CheckDefExecAndScriptFailure(lines, ['E1093: Expected 2 items but got 3', 'E687:'])
Bram Moolenaarab36e6a2021-11-30 16:14:49 +00001028
1029 var [vnr, vstr] = [123, 'text']
1030 vnr += 3
1031 assert_equal(126, vnr)
1032 vstr ..= 'end'
1033 assert_equal('textend', vstr)
1034
1035 var [vnr2: number, vstr2: string] = [123, 'text']
1036 vnr2 += 3
1037 assert_equal(126, vnr2)
1038 vstr2 ..= 'end'
1039 assert_equal('textend', vstr2)
1040
1041 var [vnr3: number; vlist: list<string>] = [123, 'foo', 'bar']
1042 vnr3 += 5
1043 assert_equal(128, vnr3)
1044 assert_equal(['foo', 'bar'], vlist)
1045
1046 lines =<< trim END
1047 var [vnr2: number, vstr2: number] = [123, 'text']
1048 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001049 v9.CheckDefExecAndScriptFailure(lines, ['E1163: Variable 2: type mismatch, expected number but got string', 'E1012: Type mismatch; expected number but got string'])
Bram Moolenaarab36e6a2021-11-30 16:14:49 +00001050 lines =<< trim END
1051 var testlist = [234, 'text']
1052 var [vnr2: number, vstr2: number] = testlist
1053 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001054 v9.CheckDefExecAndScriptFailure(lines, ['E1163: Variable 2: type mismatch, expected number but got string', 'E1012: Type mismatch; expected number but got string'])
Bram Moolenaarab36e6a2021-11-30 16:14:49 +00001055enddef
1056
Bram Moolenaar97f227d2021-07-04 20:20:52 +02001057def PartFuncBool(b: bool): string
Bram Moolenaarb7480cd2021-07-04 18:28:15 +02001058 return 'done'
1059enddef
1060
1061def Test_assignment_partial()
Bram Moolenaar97f227d2021-07-04 20:20:52 +02001062 var lines =<< trim END
Bram Moolenaar848fadd2022-01-30 15:28:30 +00001063 var Partial: func(): string = function(g:PartFuncBool, [true])
Bram Moolenaar97f227d2021-07-04 20:20:52 +02001064 assert_equal('done', Partial())
1065 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001066 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar22f85d02021-07-04 23:29:30 +02001067
1068 lines =<< trim END
1069 vim9script
1070 def Func(b: bool)
1071 enddef
1072 var Ref: func = function(Func, [true])
1073 assert_equal('func()', typename(Ref))
1074 Ref()
1075 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001076 v9.CheckScriptSuccess(lines)
Bram Moolenaar13789bf2021-12-30 13:29:00 +00001077
1078 lines =<< trim END
1079 vim9script
1080
1081 var nres: any
1082 var sres: any
Bram Moolenaar9a015112021-12-31 14:06:45 +00001083 def Func(nr: number, s = '')
1084 nres = nr
Bram Moolenaar13789bf2021-12-30 13:29:00 +00001085 sres = s
1086 enddef
1087
1088 var n: number
1089 var Ref = function(Func, [n])
1090 Ref('x')
1091 assert_equal(0, nres)
1092 assert_equal('x', sres)
1093 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001094 v9.CheckScriptSuccess(lines)
Bram Moolenaar13789bf2021-12-30 13:29:00 +00001095
1096 lines =<< trim END
1097 vim9script
1098
Bram Moolenaar9a015112021-12-31 14:06:45 +00001099 def Func(nr: number, s = '')
Bram Moolenaar13789bf2021-12-30 13:29:00 +00001100 enddef
1101
1102 var n: number
1103 var Ref = function(Func, [n])
1104 Ref(0)
1105 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001106 v9.CheckScriptFailure(lines, 'E1013: Argument 2: type mismatch, expected string but got number')
Yegappan Lakshmanan7f520212024-04-10 17:18:19 +02001107
1108 lines =<< trim END
1109 var Fn1 = () => {
1110 return 10
1111 }
1112 assert_equal('func(): number', typename(Fn1))
1113 var Fn2 = () => {
1114 return "a"
1115 }
1116 assert_equal('func(): string', typename(Fn2))
1117 var Fn3 = () => {
1118 return {a: [1]}
1119 }
1120 assert_equal('func(): dict<list<number>>', typename(Fn3))
1121 var Fn4 = (...l: list<string>) => {
1122 return []
1123 }
1124 assert_equal('func(...list<string>): list<any>', typename(Fn4))
1125 END
1126 v9.CheckSourceSuccess(['vim9script'] + lines)
1127 v9.CheckSourceSuccess(['def Xfunc()'] + lines + ['enddef', 'defcompile'])
Bram Moolenaarb7480cd2021-07-04 18:28:15 +02001128enddef
1129
Bram Moolenaarf30a14d2021-01-17 21:51:24 +01001130def Test_assignment_list_any_index()
1131 var l: list<number> = [1, 2]
1132 for [x, y, _]
1133 in [[0, 1, ''], [1, 3, '']]
1134 l[x] = l[x] + y
1135 endfor
1136 assert_equal([2, 5], l)
1137enddef
1138
Bram Moolenaar971caf52020-09-21 22:21:42 +02001139def Test_assignment_list_vim9script()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001140 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +02001141 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001142 var v1: number
1143 var v2: number
1144 var v3: number
Bram Moolenaar971caf52020-09-21 22:21:42 +02001145 [v1, v2, v3] = [1, 2, 3]
1146 assert_equal([1, 2, 3], [v1, v2, v3])
1147 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001148 v9.CheckScriptSuccess(lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001149enddef
1150
1151def Test_assignment_dict()
Bram Moolenaare0de1712020-12-02 17:36:54 +01001152 var dict1: dict<bool> = {one: false, two: true}
1153 var dict2: dict<number> = {one: 1, two: 2}
1154 var dict3: dict<string> = {key: 'value'}
1155 var dict4: dict<any> = {one: 1, two: '2'}
1156 var dict5: dict<blob> = {one: 0z01, two: 0z02}
Bram Moolenaar971caf52020-09-21 22:21:42 +02001157
Bram Moolenaar31d99482022-05-26 22:24:43 +01001158 # check the type is OK
1159 var events: dict<string> = v:event
1160
Bram Moolenaar971caf52020-09-21 22:21:42 +02001161 # overwrite
1162 dict3['key'] = 'another'
Bram Moolenaare0de1712020-12-02 17:36:54 +01001163 assert_equal(dict3, {key: 'another'})
Bram Moolenaarfc74d032020-11-16 22:11:49 +01001164 dict3.key = 'yet another'
Bram Moolenaare0de1712020-12-02 17:36:54 +01001165 assert_equal(dict3, {key: 'yet another'})
Bram Moolenaarfc74d032020-11-16 22:11:49 +01001166
Bram Moolenaard24602f2020-12-20 15:20:56 +01001167 # member "any" can also be a dict and assigned to
1168 var anydict: dict<any> = {nest: {}, nr: 0}
1169 anydict.nest['this'] = 123
1170 anydict.nest.that = 456
1171 assert_equal({nest: {this: 123, that: 456}, nr: 0}, anydict)
1172
Bram Moolenaarfc74d032020-11-16 22:11:49 +01001173 var lines =<< trim END
Bram Moolenaard88dc4d2020-12-20 14:49:42 +01001174 var dd = {}
1175 dd.two = 2
1176 assert_equal({two: 2}, dd)
1177 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001178 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar4f5e3972020-12-21 17:30:50 +01001179
1180 lines =<< trim END
1181 var d = {dd: {}}
1182 d.dd[0] = 2
1183 d.dd['x'] = 3
1184 d.dd.y = 4
1185 assert_equal({dd: {0: 2, x: 3, y: 4}}, d)
1186 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001187 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaard88dc4d2020-12-20 14:49:42 +01001188
1189 lines =<< trim END
Bram Moolenaar2e17fef2022-03-18 19:44:48 +00001190 var key = 'foo'
1191 g:[key] = 'value'
1192 assert_equal('value', g:foo)
1193 unlet g:foo
1194 END
1195 v9.CheckDefAndScriptSuccess(lines)
1196
1197 lines =<< trim END
Bram Moolenaare0de1712020-12-02 17:36:54 +01001198 var dd = {one: 1}
Bram Moolenaarfc74d032020-11-16 22:11:49 +01001199 dd.one) = 2
1200 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001201 v9.CheckDefFailure(lines, 'E488:', 2)
Bram Moolenaard88dc4d2020-12-20 14:49:42 +01001202
1203 lines =<< trim END
1204 var dd = {one: 1}
1205 var dd.one = 2
1206 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001207 v9.CheckDefAndScriptFailure(lines, 'E1017:', 2)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001208
1209 # empty key can be used
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001210 var dd = {}
Bram Moolenaar971caf52020-09-21 22:21:42 +02001211 dd[""] = 6
Bram Moolenaare0de1712020-12-02 17:36:54 +01001212 assert_equal({['']: 6}, dd)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001213
1214 # type becomes dict<any>
Bram Moolenaare0de1712020-12-02 17:36:54 +01001215 var somedict = rand() > 0 ? {a: 1, b: 2} : {a: 'a', b: 'b'}
Bram Moolenaar971caf52020-09-21 22:21:42 +02001216
Bram Moolenaar114dbda2022-01-03 12:28:03 +00001217 # type is dict<any> even though initializer is dict<number>
1218 var anyDict: dict<any> = {a: 0}
1219 assert_equal({a: 0, b: 'x'}, extend(anyDict, {b: 'x'}))
1220
Bram Moolenaar078a4612022-01-04 15:17:03 +00001221 # using global var, which has no declared type
1222 g:myDict = {}
1223 g:myDict->extend({a: 1})
1224 g:myDict->extend({b: 'x'})
1225 assert_equal({a: 1, b: 'x'}, g:myDict)
1226 unlet g:myDict
1227
1228 # using list with declared type gives an error
1229 lines =<< trim END
1230 var d: dict<number>
1231 g:myDict = d
1232 g:myDict->extend({a: 1})
1233 g:myDict->extend({b: 'x'})
1234 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001235 v9.CheckDefExecAndScriptFailure(lines, 'E1013: Argument 2: type mismatch, expected dict<number> but got dict<string>', 4)
Bram Moolenaar078a4612022-01-04 15:17:03 +00001236 unlet g:myDict
1237
Bram Moolenaar971caf52020-09-21 22:21:42 +02001238 # assignment to script-local dict
Bram Moolenaarfc74d032020-11-16 22:11:49 +01001239 lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +02001240 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001241 var test: dict<any> = {}
Bram Moolenaar971caf52020-09-21 22:21:42 +02001242 def FillDict(): dict<any>
1243 test['a'] = 43
1244 return test
1245 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +01001246 assert_equal({a: 43}, FillDict())
Bram Moolenaar971caf52020-09-21 22:21:42 +02001247 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001248 v9.CheckScriptSuccess(lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001249
1250 lines =<< trim END
1251 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001252 var test: dict<any>
Bram Moolenaar971caf52020-09-21 22:21:42 +02001253 def FillDict(): dict<any>
1254 test['a'] = 43
1255 return test
1256 enddef
1257 FillDict()
Bram Moolenaar3beaf9c2020-12-18 17:23:14 +01001258 assert_equal({a: 43}, test)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001259 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001260 v9.CheckScriptSuccess(lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001261
1262 # assignment to global dict
1263 lines =<< trim END
1264 vim9script
1265 g:test = {}
1266 def FillDict(): dict<any>
1267 g:test['a'] = 43
1268 return g:test
1269 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +01001270 assert_equal({a: 43}, FillDict())
Bram Moolenaar971caf52020-09-21 22:21:42 +02001271 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001272 v9.CheckScriptSuccess(lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001273
1274 # assignment to buffer dict
1275 lines =<< trim END
1276 vim9script
1277 b:test = {}
1278 def FillDict(): dict<any>
1279 b:test['a'] = 43
1280 return b:test
1281 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +01001282 assert_equal({a: 43}, FillDict())
Bram Moolenaar971caf52020-09-21 22:21:42 +02001283 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001284 v9.CheckScriptSuccess(lines)
Bram Moolenaar4f5e3972020-12-21 17:30:50 +01001285
1286 lines =<< trim END
1287 var d = {dd: test_null_dict()}
1288 d.dd[0] = 0
1289 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001290 v9.CheckDefExecFailure(lines, 'E1103:', 2)
Bram Moolenaar4f5e3972020-12-21 17:30:50 +01001291
1292 lines =<< trim END
1293 var d = {dd: 'string'}
1294 d.dd[0] = 0
1295 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001296 v9.CheckDefExecFailure(lines, 'E1148:', 2)
Bram Moolenaar3a3b10e2021-06-26 15:00:59 +02001297
1298 lines =<< trim END
1299 var n: any
1300 n.key = 5
1301 END
Ernie Raelf8da3242023-10-11 21:22:12 +02001302 v9.CheckDefExecAndScriptFailure(lines, ['E1148:', 'E1203: Dot not allowed after a number: n.key = 5'], 2)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001303enddef
1304
1305def Test_assignment_local()
1306 # Test in a separated file in order not to the current buffer/window/tab is
1307 # changed.
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001308 var script_lines: list<string> =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +02001309 let b:existing = 'yes'
1310 let w:existing = 'yes'
1311 let t:existing = 'yes'
1312
1313 def Test_assignment_local_internal()
1314 b:newvar = 'new'
1315 assert_equal('new', b:newvar)
1316 assert_equal('yes', b:existing)
1317 b:existing = 'no'
1318 assert_equal('no', b:existing)
1319 b:existing ..= 'NO'
1320 assert_equal('noNO', b:existing)
1321
1322 w:newvar = 'new'
1323 assert_equal('new', w:newvar)
1324 assert_equal('yes', w:existing)
1325 w:existing = 'no'
1326 assert_equal('no', w:existing)
1327 w:existing ..= 'NO'
1328 assert_equal('noNO', w:existing)
1329
1330 t:newvar = 'new'
1331 assert_equal('new', t:newvar)
1332 assert_equal('yes', t:existing)
1333 t:existing = 'no'
1334 assert_equal('no', t:existing)
1335 t:existing ..= 'NO'
1336 assert_equal('noNO', t:existing)
1337 enddef
1338 call Test_assignment_local_internal()
1339 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001340 v9.CheckScriptSuccess(script_lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001341enddef
1342
1343def Test_assignment_default()
Bram Moolenaar971caf52020-09-21 22:21:42 +02001344 # Test default values.
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001345 var thebool: bool
Bram Moolenaar971caf52020-09-21 22:21:42 +02001346 assert_equal(v:false, thebool)
1347
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001348 var thenumber: number
Bram Moolenaar971caf52020-09-21 22:21:42 +02001349 assert_equal(0, thenumber)
1350
Bram Moolenaar73e28dc2022-09-17 21:08:33 +01001351 var thefloat: float
1352 assert_equal(0.0, thefloat)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001353
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001354 var thestring: string
Bram Moolenaar971caf52020-09-21 22:21:42 +02001355 assert_equal('', thestring)
1356
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001357 var theblob: blob
Bram Moolenaar971caf52020-09-21 22:21:42 +02001358 assert_equal(0z, theblob)
1359
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001360 var Thefunc: func
Bram Moolenaar971caf52020-09-21 22:21:42 +02001361 assert_equal(test_null_function(), Thefunc)
1362
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001363 var thelist: list<any>
Bram Moolenaar971caf52020-09-21 22:21:42 +02001364 assert_equal([], thelist)
1365
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001366 var thedict: dict<any>
Bram Moolenaar971caf52020-09-21 22:21:42 +02001367 assert_equal({}, thedict)
1368
1369 if has('channel')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001370 var thejob: job
Bram Moolenaar971caf52020-09-21 22:21:42 +02001371 assert_equal(test_null_job(), thejob)
1372
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001373 var thechannel: channel
Bram Moolenaar971caf52020-09-21 22:21:42 +02001374 assert_equal(test_null_channel(), thechannel)
1375
1376 if has('unix') && executable('cat')
1377 # check with non-null job and channel, types must match
Bram Moolenaare0de1712020-12-02 17:36:54 +01001378 thejob = job_start("cat ", {})
Bram Moolenaar971caf52020-09-21 22:21:42 +02001379 thechannel = job_getchannel(thejob)
1380 job_stop(thejob, 'kill')
1381 endif
1382 endif
1383
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001384 var nr = 1234 | nr = 5678
Bram Moolenaar971caf52020-09-21 22:21:42 +02001385 assert_equal(5678, nr)
1386enddef
1387
Bram Moolenaare08be092022-02-17 13:08:26 +00001388def Test_script_var_default()
1389 var lines =<< trim END
1390 vim9script
1391 var l: list<number>
Bram Moolenaar1983f1a2022-02-28 20:55:02 +00001392 var li = [1, 2]
Bram Moolenaare08be092022-02-17 13:08:26 +00001393 var bl: blob
Bram Moolenaar1983f1a2022-02-28 20:55:02 +00001394 var bli = 0z12
Bram Moolenaare08be092022-02-17 13:08:26 +00001395 var d: dict<number>
Bram Moolenaar1983f1a2022-02-28 20:55:02 +00001396 var di = {'a': 1, 'b': 2}
Bram Moolenaare08be092022-02-17 13:08:26 +00001397 def Echo()
1398 assert_equal([], l)
Bram Moolenaar1983f1a2022-02-28 20:55:02 +00001399 assert_equal([1, 2], li)
Bram Moolenaare08be092022-02-17 13:08:26 +00001400 assert_equal(0z, bl)
Bram Moolenaar1983f1a2022-02-28 20:55:02 +00001401 assert_equal(0z12, bli)
Bram Moolenaare08be092022-02-17 13:08:26 +00001402 assert_equal({}, d)
Bram Moolenaar1983f1a2022-02-28 20:55:02 +00001403 assert_equal({'a': 1, 'b': 2}, di)
Bram Moolenaare08be092022-02-17 13:08:26 +00001404 enddef
Bram Moolenaarea5c8982022-02-17 14:42:02 +00001405 Echo()
Bram Moolenaare08be092022-02-17 13:08:26 +00001406 END
1407 v9.CheckScriptSuccess(lines)
1408enddef
1409
Bram Moolenaara749a422022-02-12 19:52:25 +00001410let s:scriptvar = 'init'
Bram Moolenaar3862ea32021-01-01 21:05:55 +01001411
Bram Moolenaar971caf52020-09-21 22:21:42 +02001412def Test_assignment_var_list()
Bram Moolenaarda7c20c2020-11-30 21:12:19 +01001413 var lines =<< trim END
1414 var v1: string
1415 var v2: string
1416 var vrem: list<string>
1417 [v1] = ['aaa']
1418 assert_equal('aaa', v1)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001419
Bram Moolenaarda7c20c2020-11-30 21:12:19 +01001420 [v1, v2] = ['one', 'two']
1421 assert_equal('one', v1)
1422 assert_equal('two', v2)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001423
Bram Moolenaarda7c20c2020-11-30 21:12:19 +01001424 [v1, v2; vrem] = ['one', 'two']
1425 assert_equal('one', v1)
1426 assert_equal('two', v2)
1427 assert_equal([], vrem)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001428
Bram Moolenaarda7c20c2020-11-30 21:12:19 +01001429 [v1, v2; vrem] = ['one', 'two', 'three']
1430 assert_equal('one', v1)
1431 assert_equal('two', v2)
1432 assert_equal(['three'], vrem)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001433
Bram Moolenaarda7c20c2020-11-30 21:12:19 +01001434 [&ts, &sw] = [3, 4]
1435 assert_equal(3, &ts)
1436 assert_equal(4, &sw)
1437 set ts=8 sw=4
Bram Moolenaar004d9b02020-11-30 21:40:03 +01001438
1439 [@a, @z] = ['aa', 'zz']
1440 assert_equal('aa', @a)
1441 assert_equal('zz', @z)
1442
1443 [$SOME_VAR, $OTHER_VAR] = ['some', 'other']
1444 assert_equal('some', $SOME_VAR)
1445 assert_equal('other', $OTHER_VAR)
1446
Bram Moolenaar3862ea32021-01-01 21:05:55 +01001447 [g:globalvar, b:bufvar, w:winvar, t:tabvar, v:errmsg] =
1448 ['global', 'buf', 'win', 'tab', 'error']
Bram Moolenaar004d9b02020-11-30 21:40:03 +01001449 assert_equal('global', g:globalvar)
Bram Moolenaar004d9b02020-11-30 21:40:03 +01001450 assert_equal('buf', b:bufvar)
1451 assert_equal('win', w:winvar)
1452 assert_equal('tab', t:tabvar)
1453 assert_equal('error', v:errmsg)
1454 unlet g:globalvar
Bram Moolenaarda7c20c2020-11-30 21:12:19 +01001455 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001456 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar3862ea32021-01-01 21:05:55 +01001457
Bram Moolenaara749a422022-02-12 19:52:25 +00001458 [g:globalvar, scriptvar, b:bufvar] = ['global', 'script', 'buf']
Bram Moolenaar3862ea32021-01-01 21:05:55 +01001459 assert_equal('global', g:globalvar)
Bram Moolenaara749a422022-02-12 19:52:25 +00001460 assert_equal('script', scriptvar)
Bram Moolenaar3862ea32021-01-01 21:05:55 +01001461 assert_equal('buf', b:bufvar)
1462
1463 lines =<< trim END
1464 vim9script
Bram Moolenaara749a422022-02-12 19:52:25 +00001465 var scriptvar = 'init'
1466 [g:globalvar, scriptvar, w:winvar] = ['global', 'script', 'win']
Bram Moolenaar3862ea32021-01-01 21:05:55 +01001467 assert_equal('global', g:globalvar)
Bram Moolenaara749a422022-02-12 19:52:25 +00001468 assert_equal('script', scriptvar)
Bram Moolenaar3862ea32021-01-01 21:05:55 +01001469 assert_equal('win', w:winvar)
1470 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001471 v9.CheckScriptSuccess(lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001472enddef
1473
Bram Moolenaar2d3ac2e2022-02-03 12:34:05 +00001474def Test_assignment_empty_list()
1475 var lines =<< trim END
1476 var l2: list<any> = []
1477 var l: list<string>
1478 l = l2
1479 END
1480 v9.CheckDefAndScriptSuccess(lines)
1481enddef
1482
Bram Moolenaar971caf52020-09-21 22:21:42 +02001483def Test_assignment_vim9script()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001484 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +02001485 vim9script
1486 def Func(): list<number>
1487 return [1, 2]
1488 enddef
Bram Moolenaar98309382020-09-27 21:58:45 +02001489 var name1: number
1490 var name2: number
1491 [name1, name2] =
Bram Moolenaar971caf52020-09-21 22:21:42 +02001492 Func()
Bram Moolenaar98309382020-09-27 21:58:45 +02001493 assert_equal(1, name1)
1494 assert_equal(2, name2)
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001495 var ll =
Bram Moolenaar971caf52020-09-21 22:21:42 +02001496 Func()
1497 assert_equal([1, 2], ll)
1498
1499 @/ = 'text'
1500 assert_equal('text', @/)
1501 @0 = 'zero'
1502 assert_equal('zero', @0)
1503 @1 = 'one'
1504 assert_equal('one', @1)
1505 @9 = 'nine'
1506 assert_equal('nine', @9)
1507 @- = 'minus'
1508 assert_equal('minus', @-)
1509 if has('clipboard_working')
1510 @* = 'star'
1511 assert_equal('star', @*)
1512 @+ = 'plus'
1513 assert_equal('plus', @+)
1514 endif
1515
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001516 var a: number = 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001517 assert_equal(123, a)
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001518 var s: string = 'yes'
Bram Moolenaar971caf52020-09-21 22:21:42 +02001519 assert_equal('yes', s)
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001520 var b: number = 42
Bram Moolenaar971caf52020-09-21 22:21:42 +02001521 assert_equal(42, b)
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001522 var w: number = 43
Bram Moolenaar971caf52020-09-21 22:21:42 +02001523 assert_equal(43, w)
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001524 var t: number = 44
Bram Moolenaar971caf52020-09-21 22:21:42 +02001525 assert_equal(44, t)
Bram Moolenaar7a66a172020-10-16 19:56:12 +02001526
1527 var to_var = 0
1528 to_var = 3
1529 assert_equal(3, to_var)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001530 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001531 v9.CheckScriptSuccess(lines)
Bram Moolenaar06f9c692020-09-27 21:27:40 +02001532
1533 lines =<< trim END
1534 vim9script
1535 var n: number
1536 def Func()
1537 n = 'string'
1538 enddef
1539 defcompile
1540 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001541 v9.CheckScriptFailure(lines, 'E1012: Type mismatch; expected number but got string')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001542enddef
1543
1544def Mess(): string
1545 v:foldstart = 123
1546 return 'xxx'
1547enddef
1548
1549def Test_assignment_failure()
Bram Moolenaar62aec932022-01-29 21:45:34 +00001550 v9.CheckDefFailure(['var name=234'], 'E1004:')
1551 v9.CheckDefFailure(['var name =234'], 'E1004:')
1552 v9.CheckDefFailure(['var name= 234'], 'E1004:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001553
Bram Moolenaar62aec932022-01-29 21:45:34 +00001554 v9.CheckScriptFailure(['vim9script', 'var name=234'], 'E1004:')
1555 v9.CheckScriptFailure(['vim9script', 'var name=234'], "before and after '='")
1556 v9.CheckScriptFailure(['vim9script', 'var name =234'], 'E1004:')
1557 v9.CheckScriptFailure(['vim9script', 'var name= 234'], 'E1004:')
1558 v9.CheckScriptFailure(['vim9script', 'var name = 234', 'name+=234'], 'E1004:')
1559 v9.CheckScriptFailure(['vim9script', 'var name = 234', 'name+=234'], "before and after '+='")
1560 v9.CheckScriptFailure(['vim9script', 'var name = "x"', 'name..="y"'], 'E1004:')
1561 v9.CheckScriptFailure(['vim9script', 'var name = "x"', 'name..="y"'], "before and after '..='")
Bram Moolenaar971caf52020-09-21 22:21:42 +02001562
Bram Moolenaar62aec932022-01-29 21:45:34 +00001563 v9.CheckDefFailure(['var true = 1'], 'E1034:')
1564 v9.CheckDefFailure(['var false = 1'], 'E1034:')
1565 v9.CheckDefFailure(['var null = 1'], 'E1034:')
1566 v9.CheckDefFailure(['var this = 1'], 'E1034:')
Bram Moolenaarce723f32023-06-10 19:00:12 +01001567 v9.CheckDefFailure(['var super = 1'], 'E1034:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001568
Bram Moolenaar8b716f52022-02-15 21:17:56 +00001569 v9.CheckDefFailure(['[a; b; c] = g:list'], 'E1001:')
1570 v9.CheckDefFailure(['var [a; b; c] = g:list'], 'E1080:')
Bram Moolenaar62aec932022-01-29 21:45:34 +00001571 v9.CheckDefExecFailure(['var a: number',
Bram Moolenaar971caf52020-09-21 22:21:42 +02001572 '[a] = test_null_list()'], 'E1093:')
Bram Moolenaar62aec932022-01-29 21:45:34 +00001573 v9.CheckDefExecFailure(['var a: number',
Bram Moolenaar971caf52020-09-21 22:21:42 +02001574 '[a] = []'], 'E1093:')
Bram Moolenaar62aec932022-01-29 21:45:34 +00001575 v9.CheckDefExecFailure(['var x: number',
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001576 'var y: number',
Bram Moolenaar971caf52020-09-21 22:21:42 +02001577 '[x, y] = [1]'], 'E1093:')
Bram Moolenaar62aec932022-01-29 21:45:34 +00001578 v9.CheckDefExecFailure(['var x: string',
Bram Moolenaar352134b2020-10-17 22:04:08 +02001579 'var y: string',
1580 '[x, y] = ["x"]'], 'E1093:')
Bram Moolenaar62aec932022-01-29 21:45:34 +00001581 v9.CheckDefExecFailure(['var x: number',
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001582 'var y: number',
1583 'var z: list<number>',
Bram Moolenaar971caf52020-09-21 22:21:42 +02001584 '[x, y; z] = [1]'], 'E1093:')
1585
Bram Moolenaar62aec932022-01-29 21:45:34 +00001586 v9.CheckDefFailure(['var somevar'], "E1022:")
1587 v9.CheckDefFailure(['var &tabstop = 4'], 'E1052:')
1588 v9.CheckDefFailure(['&g:option = 5'], 'E113:')
1589 v9.CheckScriptFailure(['vim9script', 'var &tabstop = 4'], 'E1052:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001590
Bram Moolenaar62aec932022-01-29 21:45:34 +00001591 v9.CheckDefFailure(['var $VAR = 5'], 'E1016: Cannot declare an environment variable:')
1592 v9.CheckScriptFailure(['vim9script', 'var $ENV = "xxx"'], 'E1016:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001593
1594 if has('dnd')
Bram Moolenaar62aec932022-01-29 21:45:34 +00001595 v9.CheckDefFailure(['var @~ = 5'], 'E1066:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001596 else
Bram Moolenaar62aec932022-01-29 21:45:34 +00001597 v9.CheckDefFailure(['var @~ = 5'], 'E354:')
1598 v9.CheckDefFailure(['@~ = 5'], 'E354:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001599 endif
Bram Moolenaar62aec932022-01-29 21:45:34 +00001600 v9.CheckDefFailure(['var @a = 5'], 'E1066:')
1601 v9.CheckDefFailure(['var @/ = "x"'], 'E1066:')
1602 v9.CheckScriptFailure(['vim9script', 'var @a = "abc"'], 'E1066:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001603
Bram Moolenaar62aec932022-01-29 21:45:34 +00001604 v9.CheckDefFailure(['var g:var = 5'], 'E1016: Cannot declare a global variable:')
1605 v9.CheckDefFailure(['var w:var = 5'], 'E1016: Cannot declare a window variable:')
1606 v9.CheckDefFailure(['var b:var = 5'], 'E1016: Cannot declare a buffer variable:')
1607 v9.CheckDefFailure(['var t:var = 5'], 'E1016: Cannot declare a tab variable:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001608
Bram Moolenaar62aec932022-01-29 21:45:34 +00001609 v9.CheckDefFailure(['var anr = 4', 'anr ..= "text"'], 'E1019:')
1610 v9.CheckDefFailure(['var xnr += 4'], 'E1020:', 1)
1611 v9.CheckScriptFailure(['vim9script', 'var xnr += 4'], 'E1020:')
1612 v9.CheckDefFailure(["var xnr = xnr + 1"], 'E1001:', 1)
1613 v9.CheckScriptFailure(['vim9script', 'var xnr = xnr + 4'], 'E121:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001614
Bram Moolenaara749a422022-02-12 19:52:25 +00001615 v9.CheckScriptFailure(['vim9script', 'def Func()', 'var dummy = notfound', 'enddef', 'defcompile'], 'E1001:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001616
Bram Moolenaar62aec932022-01-29 21:45:34 +00001617 v9.CheckDefFailure(['var name: list<string> = [123]'], 'expected list<string> but got list<number>')
1618 v9.CheckDefFailure(['var name: list<number> = ["xx"]'], 'expected list<number> but got list<string>')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001619
Bram Moolenaar62aec932022-01-29 21:45:34 +00001620 v9.CheckDefFailure(['var name: dict<string> = {key: 123}'], 'expected dict<string> but got dict<number>')
1621 v9.CheckDefFailure(['var name: dict<number> = {key: "xx"}'], 'expected dict<number> but got dict<string>')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001622
Bram Moolenaar62aec932022-01-29 21:45:34 +00001623 v9.CheckDefFailure(['var name = feedkeys("0")'], 'E1031:')
1624 v9.CheckDefFailure(['var name: number = feedkeys("0")'], 'expected number but got void')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001625
Bram Moolenaar62aec932022-01-29 21:45:34 +00001626 v9.CheckDefFailure(['var name: dict <number>'], 'E1068:')
Bram Moolenaar62e0e2e2022-08-20 12:07:58 +01001627 v9.CheckDefFailure(['var name: dict<number'], 'E1009: Missing > after type: <number')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001628
Bram Moolenaar848fadd2022-01-30 15:28:30 +00001629 assert_fails('s/^/\=g:Mess()/n', 'E794:')
Bram Moolenaar62aec932022-01-29 21:45:34 +00001630 v9.CheckDefFailure(['var name: dict<number'], 'E1009:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001631
Bram Moolenaar62aec932022-01-29 21:45:34 +00001632 v9.CheckDefFailure(['w:foo: number = 10'],
Bram Moolenaar9510d222022-09-11 15:14:05 +01001633 'E1016: Cannot declare a window variable: w:foo')
Bram Moolenaar62aec932022-01-29 21:45:34 +00001634 v9.CheckDefFailure(['t:foo: bool = true'],
Bram Moolenaar9510d222022-09-11 15:14:05 +01001635 'E1016: Cannot declare a tab variable: t:foo')
Bram Moolenaar62aec932022-01-29 21:45:34 +00001636 v9.CheckDefFailure(['b:foo: string = "x"'],
Bram Moolenaar9510d222022-09-11 15:14:05 +01001637 'E1016: Cannot declare a buffer variable: b:foo')
Bram Moolenaar62aec932022-01-29 21:45:34 +00001638 v9.CheckDefFailure(['g:foo: number = 123'],
Bram Moolenaar9510d222022-09-11 15:14:05 +01001639 'E1016: Cannot declare a global variable: g:foo')
1640
1641 v9.CheckScriptFailure(['vim9script', 'w:foo: number = 123'],
1642 'E1304: Cannot use type with this variable: w:foo:')
1643 v9.CheckScriptFailure(['vim9script', 't:foo: number = 123'],
1644 'E1304: Cannot use type with this variable: t:foo:')
1645 v9.CheckScriptFailure(['vim9script', 'b:foo: number = 123'],
1646 'E1304: Cannot use type with this variable: b:foo:')
1647 v9.CheckScriptFailure(['vim9script', 'g:foo: number = 123'],
1648 'E1304: Cannot use type with this variable: g:foo:')
1649
1650 v9.CheckScriptFailure(['vim9script', 'const w:FOO: number = 123'],
1651 'E1304: Cannot use type with this variable: w:FOO:')
1652 v9.CheckScriptFailure(['vim9script', 'const t:FOO: number = 123'],
1653 'E1304: Cannot use type with this variable: t:FOO:')
1654 v9.CheckScriptFailure(['vim9script', 'const b:FOO: number = 123'],
1655 'E1304: Cannot use type with this variable: b:FOO:')
1656 v9.CheckScriptFailure(['vim9script', 'const g:FOO: number = 123'],
1657 'E1304: Cannot use type with this variable: g:FOO:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001658enddef
1659
1660def Test_assign_list()
Bram Moolenaare65081d2021-06-27 15:04:05 +02001661 var lines =<< trim END
1662 var l: list<string> = []
1663 l[0] = 'value'
1664 assert_equal('value', l[0])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001665
Bram Moolenaare65081d2021-06-27 15:04:05 +02001666 l[1] = 'asdf'
1667 assert_equal('value', l[0])
1668 assert_equal('asdf', l[1])
1669 assert_equal('asdf', l[-1])
1670 assert_equal('value', l[-2])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001671
Bram Moolenaare65081d2021-06-27 15:04:05 +02001672 var nrl: list<number> = []
1673 for i in range(5)
1674 nrl[i] = i
1675 endfor
1676 assert_equal([0, 1, 2, 3, 4], nrl)
1677
1678 var ul: list<any>
1679 ul[0] = 1
1680 ul[1] = 2
1681 ul[2] = 3
1682 assert_equal([1, 2, 3], ul)
1683 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001684 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar334a8b42020-10-19 16:07:42 +02001685
Bram Moolenaar1983f1a2022-02-28 20:55:02 +00001686 lines =<< trim END
1687 var l = [1, 2]
1688 g:idx = 'x'
1689 l[g:idx : 1] = [0]
1690 echo l
1691 END
Bram Moolenaar2995e5c2022-03-18 21:41:47 +00001692 v9.CheckDefExecAndScriptFailure(lines, ['E1012: Type mismatch; expected number but got string', 'E1030: Using a String as a Number: "x"'])
Bram Moolenaar1983f1a2022-02-28 20:55:02 +00001693
1694 lines =<< trim END
1695 var l = [1, 2]
1696 g:idx = 3
1697 l[g:idx : 1] = [0]
1698 echo l
1699 END
Bram Moolenaarec892232022-05-06 17:53:06 +01001700 v9.CheckDefExecAndScriptFailure(lines, 'E684: List index out of range: 3')
Bram Moolenaar1983f1a2022-02-28 20:55:02 +00001701
1702 lines =<< trim END
1703 var l = [1, 2]
1704 g:idx = 'y'
1705 l[1 : g:idx] = [0]
1706 echo l
1707 END
1708 v9.CheckDefExecAndScriptFailure(lines, ['E1012: Type mismatch; expected number but got string', 'E1030: Using a String as a Number: "y"'])
1709
Bram Moolenaar62aec932022-01-29 21:45:34 +00001710 v9.CheckDefFailure(["var l: list<number> = ['', true]"], 'E1012: Type mismatch; expected list<number> but got list<any>', 1)
1711 v9.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 +02001712enddef
1713
1714def Test_assign_dict()
Bram Moolenaarc3689572021-01-01 19:40:02 +01001715 var lines =<< trim END
1716 var d: dict<string> = {}
1717 d['key'] = 'value'
1718 assert_equal('value', d['key'])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001719
Bram Moolenaarc3689572021-01-01 19:40:02 +01001720 d[123] = 'qwerty'
1721 assert_equal('qwerty', d[123])
1722 assert_equal('qwerty', d['123'])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001723
Bram Moolenaarc3689572021-01-01 19:40:02 +01001724 var nrd: dict<number> = {}
1725 for i in range(3)
1726 nrd[i] = i
1727 endfor
1728 assert_equal({0: 0, 1: 1, 2: 2}, nrd)
1729
1730 d.somekey = 'someval'
1731 assert_equal({key: 'value', '123': 'qwerty', somekey: 'someval'}, d)
Bram Moolenaar23e2e112021-08-03 21:16:18 +02001732 unlet d.somekey
1733 assert_equal({key: 'value', '123': 'qwerty'}, d)
Bram Moolenaarc3689572021-01-01 19:40:02 +01001734 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001735 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaarc3689572021-01-01 19:40:02 +01001736
Bram Moolenaar62aec932022-01-29 21:45:34 +00001737 v9.CheckDefFailure(["var d: dict<number> = {a: '', b: true}"], 'E1012: Type mismatch; expected dict<number> but got dict<any>', 1)
1738 v9.CheckDefFailure(["var d: dict<dict<number>> = {x: {a: '', b: true}}"], 'E1012: Type mismatch; expected dict<dict<number>> but got dict<dict<any>>', 1)
Bram Moolenaarbc510062022-02-14 21:19:04 +00001739 v9.CheckDefFailure(["var d = {x: 1}", "d[1 : 2] = {y: 2}"], 'E1165: Cannot use a range with an assignment: d[1 : 2] =', 2)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001740enddef
1741
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001742def Test_assign_dict_unknown_type()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001743 var lines =<< trim END
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001744 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001745 var mylist = []
Bram Moolenaare0de1712020-12-02 17:36:54 +01001746 mylist += [{one: 'one'}]
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001747 def Func()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001748 var dd = mylist[0]
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001749 assert_equal('one', dd.one)
1750 enddef
1751 Func()
1752 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001753 v9.CheckScriptSuccess(lines)
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001754
Bram Moolenaarce2c5442020-11-28 21:21:17 +01001755 lines =<< trim END
1756 vim9script
1757 var mylist = [[]]
Bram Moolenaare0de1712020-12-02 17:36:54 +01001758 mylist[0] += [{one: 'one'}]
Bram Moolenaarce2c5442020-11-28 21:21:17 +01001759 def Func()
1760 var dd = mylist[0][0]
1761 assert_equal('one', dd.one)
1762 enddef
1763 Func()
1764 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001765 v9.CheckScriptSuccess(lines)
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001766enddef
1767
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001768def Test_assign_dict_with_op()
1769 var lines =<< trim END
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001770 var ds: dict<string> = {a: 'x'}
1771 ds['a'] ..= 'y'
1772 ds.a ..= 'z'
1773 assert_equal('xyz', ds.a)
1774
1775 var dn: dict<number> = {a: 9}
1776 dn['a'] += 2
1777 assert_equal(11, dn.a)
1778 dn.a += 2
1779 assert_equal(13, dn.a)
1780
1781 dn['a'] -= 3
1782 assert_equal(10, dn.a)
1783 dn.a -= 2
1784 assert_equal(8, dn.a)
1785
1786 dn['a'] *= 2
1787 assert_equal(16, dn.a)
1788 dn.a *= 2
1789 assert_equal(32, dn.a)
1790
1791 dn['a'] /= 3
1792 assert_equal(10, dn.a)
1793 dn.a /= 2
1794 assert_equal(5, dn.a)
1795
1796 dn['a'] %= 3
1797 assert_equal(2, dn.a)
1798 dn.a %= 6
1799 assert_equal(2, dn.a)
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001800
1801 var dd: dict<dict<list<any>>>
1802 dd.a = {}
1803 dd.a.b = [0]
1804 dd.a.b += [1]
1805 assert_equal({a: {b: [0, 1]}}, dd)
Bram Moolenaar0d351a02021-04-05 21:05:26 +02001806
1807 var dab = {a: ['b']}
1808 dab.a[0] ..= 'c'
1809 assert_equal({a: ['bc']}, dab)
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001810 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001811 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaare42939a2021-04-05 17:11:17 +02001812enddef
1813
1814def Test_assign_list_with_op()
1815 var lines =<< trim END
1816 var ls: list<string> = ['x']
1817 ls[0] ..= 'y'
1818 assert_equal('xy', ls[0])
1819
1820 var ln: list<number> = [9]
1821 ln[0] += 2
1822 assert_equal(11, ln[0])
1823
1824 ln[0] -= 3
1825 assert_equal(8, ln[0])
1826
1827 ln[0] *= 2
1828 assert_equal(16, ln[0])
1829
1830 ln[0] /= 3
1831 assert_equal(5, ln[0])
1832
1833 ln[0] %= 3
1834 assert_equal(2, ln[0])
1835 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001836 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaare42939a2021-04-05 17:11:17 +02001837enddef
1838
1839def Test_assign_with_op_fails()
1840 var lines =<< trim END
1841 var s = 'abc'
1842 s[1] += 'x'
1843 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001844 v9.CheckDefAndScriptFailure(lines, ['E1141:', 'E689:'], 2)
Bram Moolenaare42939a2021-04-05 17:11:17 +02001845
1846 lines =<< trim END
1847 var s = 'abc'
1848 s[1] ..= 'x'
1849 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001850 v9.CheckDefAndScriptFailure(lines, ['E1141:', 'E689:'], 2)
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001851
1852 lines =<< trim END
1853 var dd: dict<dict<list<any>>>
1854 dd.a = {}
1855 dd.a.b += [1]
1856 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001857 v9.CheckDefExecAndScriptFailure(lines, 'E716:', 3)
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001858enddef
1859
Bram Moolenaar971caf52020-09-21 22:21:42 +02001860def Test_assign_lambda()
1861 # check if assign a lambda to a variable which type is func or any.
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001862 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +02001863 vim9script
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001864 var FuncRef = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001865 assert_equal(123, FuncRef())
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001866 var FuncRef_Func: func = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001867 assert_equal(123, FuncRef_Func())
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001868 var FuncRef_Any: any = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001869 assert_equal(123, FuncRef_Any())
Bram Moolenaar06409502021-02-17 17:00:27 +01001870 var FuncRef_Number: func(): number = () => 321
1871 assert_equal(321, FuncRef_Number())
Bram Moolenaar971caf52020-09-21 22:21:42 +02001872 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001873 v9.CheckScriptSuccess(lines)
Bram Moolenaar328eac22021-01-07 19:23:08 +01001874
1875 lines =<< trim END
1876 var Ref: func(number)
1877 Ref = (j) => !j
1878 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001879 v9.CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected func(number) but got func(any): bool')
Bram Moolenaarc7dac852021-02-17 18:49:11 +01001880
1881 lines =<< trim END
Bram Moolenaar94722c52023-01-28 19:19:03 +00001882 echo filter([1, 2, 3], (_, v: string) => v + 1)
Bram Moolenaarc7dac852021-02-17 18:49:11 +01001883 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001884 v9.CheckDefAndScriptFailure(lines, 'E1051:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001885enddef
1886
Ernie Rael96952b22023-10-17 18:15:01 +02001887def Test_assign_funcref_args()
1888 # unspecified arguments match everything, including varargs
1889 var lines =<< trim END
1890 vim9script
1891
1892 var FuncUnknown: func: number
1893
1894 FuncUnknown = (v): number => v
1895 assert_equal(5, FuncUnknown(5))
1896
1897 FuncUnknown = (v1, v2): number => v1 + v2
1898 assert_equal(7, FuncUnknown(3, 4))
1899
1900 FuncUnknown = (...v1): number => v1[0] + v1[1] + len(v1) * 1000
1901 assert_equal(4007, FuncUnknown(3, 4, 5, 6))
1902
1903 FuncUnknown = (v: list<any>): number => v[0] + v[1] + len(v) * 1000
1904 assert_equal(5009, FuncUnknown([4, 5, 6, 7, 8]))
1905 END
1906 v9.CheckScriptSuccess(lines)
1907
1908 # varargs must match
1909 lines =<< trim END
1910 vim9script
1911 var FuncAnyVA: func(...any): number
1912 FuncAnyVA = (v): number => v
1913 END
Ernie Rael3ec6c1f2023-10-21 11:45:38 +02001914 v9.CheckScriptFailure(lines, 'E1180: Variable arguments type must be a list: any')
Ernie Rael96952b22023-10-17 18:15:01 +02001915
1916 # varargs must match
1917 lines =<< trim END
1918 vim9script
1919 var FuncAnyVA: func(...any): number
1920 FuncAnyVA = (v1, v2): number => v1 + v2
1921 END
Ernie Rael3ec6c1f2023-10-21 11:45:38 +02001922 v9.CheckScriptFailure(lines, 'E1180: Variable arguments type must be a list: any')
Ernie Rael96952b22023-10-17 18:15:01 +02001923
1924 # varargs must match
1925 lines =<< trim END
1926 vim9script
1927 var FuncAnyVA: func(...any): number
1928 FuncAnyVA = (v1: list<any>): number => 3
1929 END
Ernie Rael3ec6c1f2023-10-21 11:45:38 +02001930 v9.CheckScriptFailure(lines, 'E1180: Variable arguments type must be a list: any')
Ernie Rael96952b22023-10-17 18:15:01 +02001931enddef
1932
1933def Test_assign_funcref_arg_any()
1934 var lines =<< trim END
1935 vim9script
1936 var FuncAnyVA: func(any): number
1937 FuncAnyVA = (v): number => v
1938 END
1939 # TODO: Verify this should succeed.
1940 v9.CheckScriptSuccess(lines)
1941enddef
1942
Bram Moolenaarc0e29012020-09-27 14:22:48 +02001943def Test_heredoc()
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01001944 # simple heredoc
1945 var lines =<< trim END
zeertzjqa93d9cd2023-05-02 16:25:47 +01001946 var text =<< trim TEXT # comment
1947 abc
1948 TEXT
1949 assert_equal(['abc'], text)
Bram Moolenaarc0e29012020-09-27 14:22:48 +02001950 END
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01001951 v9.CheckDefAndScriptSuccess(lines)
1952
1953 # empty heredoc
1954 lines =<< trim END
zeertzjqa93d9cd2023-05-02 16:25:47 +01001955 var text =<< trim TEXT
1956 TEXT
1957 assert_equal([], text)
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01001958 END
1959 v9.CheckDefAndScriptSuccess(lines)
1960
1961 # heredoc with a single empty line
1962 lines =<< trim END
zeertzjqa93d9cd2023-05-02 16:25:47 +01001963 var text =<< trim TEXT
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01001964
zeertzjqa93d9cd2023-05-02 16:25:47 +01001965 TEXT
1966 assert_equal([''], text)
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01001967 END
1968 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaarc0e29012020-09-27 14:22:48 +02001969
zeertzjqa93d9cd2023-05-02 16:25:47 +01001970 # assign heredoc to variable with type
1971 lines =<< trim END
1972 var text: list<string> =<< trim TEXT
1973 var foo =<< trim FOO
1974 TEXT
1975 assert_equal(['var foo =<< trim FOO'], text)
1976 END
1977 v9.CheckDefAndScriptSuccess(lines)
1978
1979 # extra whitespace before type is allowed
1980 lines =<< trim END
1981 var text: list<string> =<< trim TEXT
1982 var foo =<< trim FOO
1983 TEXT
1984 assert_equal(['var foo =<< trim FOO'], text)
1985 END
1986 v9.CheckDefAndScriptSuccess(lines)
1987
1988 # missing whitespace before type is an error
1989 lines =<< trim END
1990 var text:list<string> =<< trim TEXT
1991 var foo =<< trim FOO
1992 TEXT
1993 assert_equal(['var foo =<< trim FOO'], text)
1994 END
1995 v9.CheckDefAndScriptFailure(lines, 'E1069:')
1996
1997 # assign heredoc to list slice
1998 lines =<< trim END
1999 var text = ['']
2000 text[ : ] =<< trim TEXT
2001 var foo =<< trim FOO
2002 TEXT
2003 assert_equal(['var foo =<< trim FOO'], text)
2004 END
2005 v9.CheckDefAndScriptSuccess(lines)
2006
2007 # assign heredoc to curly braces name in legacy function in Vim9 script
2008 lines =<< trim END
2009 vim9script
2010 func Func()
2011 let foo_3_bar = ['']
2012 let foo_{1 + 2}_bar[ : ] =<< trim TEXT
2013 var foo =<< trim FOO
2014 TEXT
2015 call assert_equal(['var foo =<< trim FOO'], foo_3_bar)
2016 endfunc
2017 Func()
2018 END
2019 v9.CheckScriptSuccess(lines)
2020
zeertzjq9a91d2b2024-04-09 21:47:10 +02002021 # commented out heredoc assignment without space after '#'
2022 lines =<< trim END
2023 vim9script
2024 def Func()
2025 #x =<< trim [CODE]
2026 #[CODE]
2027 enddef
2028 Func()
2029 END
2030 v9.CheckScriptSuccess(lines)
2031
zeertzjq1817ccd2024-04-10 17:37:47 +02002032 # heredoc start should not be recognized in string
2033 lines =<< trim END
2034 vim9script
2035 def Func()
2036 new
2037 @" = 'bar'
2038 ['foo', @"]->setline("]=<<"->count('='))
2039 assert_equal(['foo', 'bar'], getline(1, '$'))
2040 bwipe!
2041 enddef
2042 Func()
2043 END
2044 v9.CheckScriptSuccess(lines)
2045
Bram Moolenaar62aec932022-01-29 21:45:34 +00002046 v9.CheckDefFailure(['var lines =<< trim END X', 'END'], 'E488:')
2047 v9.CheckDefFailure(['var lines =<< trim END " comment', 'END'], 'E488:')
Bram Moolenaar2914a202020-09-27 18:24:03 +02002048
2049 lines =<< trim [END]
2050 def Func()
2051 var&lines =<< trim END
2052 x
2053 x
Bram Moolenaarb5b94802020-12-13 17:50:20 +01002054 enddef
2055 defcompile
2056 [END]
Bram Moolenaar62aec932022-01-29 21:45:34 +00002057 v9.CheckScriptFailure(lines, 'E1145: Missing heredoc end marker: END')
Bram Moolenaarb5b94802020-12-13 17:50:20 +01002058 delfunc! g:Func
2059
2060 lines =<< trim [END]
2061 def Func()
2062 var lines =<< trim END
2063 x
2064 x
Bram Moolenaar2914a202020-09-27 18:24:03 +02002065 x
2066 x
2067 x
2068 x
2069 x
2070 x
2071 enddef
2072 call Func()
2073 [END]
Bram Moolenaar62aec932022-01-29 21:45:34 +00002074 v9.CheckScriptFailure(lines, 'E1145: Missing heredoc end marker: END')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01002075 delfunc! g:Func
Bram Moolenaar81530e32021-07-28 21:25:49 +02002076
2077 lines =<< trim END
2078 var lines: number =<< trim STOP
2079 aaa
2080 bbb
2081 STOP
2082 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002083 v9.CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected number but got list<string>', 1)
Bram Moolenaard9876422022-10-12 12:58:54 +01002084
2085 lines =<< trim END
2086 var lines=<< STOP
2087 xxx
2088 STOP
2089 END
2090 v9.CheckDefAndScriptFailure(lines, 'E1004: White space required before and after ''=<<'' at "=<< STOP"', 1)
2091 lines =<< trim END
2092 var lines =<<STOP
2093 xxx
2094 STOP
2095 END
2096 v9.CheckDefAndScriptFailure(lines, 'E1004: White space required before and after ''=<<'' at "=<<STOP"', 1)
2097 lines =<< trim END
2098 var lines=<<STOP
2099 xxx
2100 STOP
2101 END
2102 v9.CheckDefAndScriptFailure(lines, 'E1004: White space required before and after ''=<<'' at "=<<STOP"', 1)
Bram Moolenaarc0e29012020-09-27 14:22:48 +02002103enddef
2104
Bram Moolenaar083966f2020-12-08 11:29:40 +01002105def Test_var_func_call()
Bram Moolenaar55759b52020-09-30 22:59:42 +02002106 var lines =<< trim END
2107 vim9script
2108 func GetValue()
2109 if exists('g:count')
2110 let g:count += 1
2111 else
2112 let g:count = 1
2113 endif
2114 return 'this'
2115 endfunc
Bram Moolenaar94722c52023-01-28 19:19:03 +00002116 var val: string = GetValue()
Bram Moolenaar55759b52020-09-30 22:59:42 +02002117 # env var is always a string
2118 var env = $TERM
2119 END
Bram Moolenaar0e9bdad2022-10-15 20:06:33 +01002120 writefile(lines, 'Xfinished', 'D')
Bram Moolenaar55759b52020-09-30 22:59:42 +02002121 source Xfinished
2122 # GetValue() is not called during discovery phase
2123 assert_equal(1, g:count)
2124
2125 unlet g:count
Bram Moolenaar55759b52020-09-30 22:59:42 +02002126enddef
2127
Bram Moolenaar083966f2020-12-08 11:29:40 +01002128def Test_var_missing_type()
Bram Moolenaar55759b52020-09-30 22:59:42 +02002129 var lines =<< trim END
2130 vim9script
2131 var name = g:unknown
2132 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002133 v9.CheckScriptFailure(lines, 'E121:')
Bram Moolenaar55759b52020-09-30 22:59:42 +02002134
2135 lines =<< trim END
2136 vim9script
2137 var nr: number = 123
2138 var name = nr
2139 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002140 v9.CheckScriptSuccess(lines)
Bram Moolenaar55759b52020-09-30 22:59:42 +02002141enddef
2142
Bram Moolenaar083966f2020-12-08 11:29:40 +01002143def Test_var_declaration()
Bram Moolenaar55759b52020-09-30 22:59:42 +02002144 var lines =<< trim END
2145 vim9script
2146 var name: string
2147 g:var_uninit = name
2148 name = 'text'
2149 g:var_test = name
Bram Moolenaara749a422022-02-12 19:52:25 +00002150 # prefixing s: is not allowed
2151 name = 'prefixed'
2152 g:var_prefixed = name
Bram Moolenaar55759b52020-09-30 22:59:42 +02002153
Bram Moolenaar7257af42020-12-08 22:08:51 +01002154 const FOO: number = 123
2155 assert_equal(123, FOO)
Bram Moolenaar89b474d2020-12-22 21:19:39 +01002156 const FOOS = 'foos'
2157 assert_equal('foos', FOOS)
2158 final FLIST = [1]
2159 assert_equal([1], FLIST)
2160 FLIST[0] = 11
2161 assert_equal([11], FLIST)
2162
Bram Moolenaar89b474d2020-12-22 21:19:39 +01002163 const g:FOOS = 'gfoos'
2164 assert_equal('gfoos', g:FOOS)
2165 final g:FLIST = [2]
2166 assert_equal([2], g:FLIST)
2167 g:FLIST[0] = 22
2168 assert_equal([22], g:FLIST)
2169
Bram Moolenaard877a572021-04-01 19:42:48 +02002170 def SetGlobalConst()
2171 const g:globConst = 123
2172 enddef
2173 SetGlobalConst()
2174 assert_equal(123, g:globConst)
2175 assert_true(islocked('g:globConst'))
2176
Bram Moolenaar89b474d2020-12-22 21:19:39 +01002177 const w:FOOS = 'wfoos'
2178 assert_equal('wfoos', w:FOOS)
2179 final w:FLIST = [3]
2180 assert_equal([3], w:FLIST)
2181 w:FLIST[0] = 33
2182 assert_equal([33], w:FLIST)
Bram Moolenaar7257af42020-12-08 22:08:51 +01002183
Bram Moolenaar55759b52020-09-30 22:59:42 +02002184 var s:other: number
2185 other = 1234
2186 g:other_var = other
2187
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01002188 var xyz: string # comment
2189
Bram Moolenaar55759b52020-09-30 22:59:42 +02002190 # type is inferred
Bram Moolenaara749a422022-02-12 19:52:25 +00002191 var dict = {['a']: 222}
Bram Moolenaar55759b52020-09-30 22:59:42 +02002192 def GetDictVal(key: any)
Bram Moolenaara749a422022-02-12 19:52:25 +00002193 g:dict_val = dict[key]
Bram Moolenaar55759b52020-09-30 22:59:42 +02002194 enddef
2195 GetDictVal('a')
Bram Moolenaar08251752021-01-11 21:20:18 +01002196
2197 final adict: dict<string> = {}
2198 def ChangeAdict()
2199 adict.foo = 'foo'
2200 enddef
2201 ChangeAdict()
Bram Moolenaar55759b52020-09-30 22:59:42 +02002202 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002203 v9.CheckScriptSuccess(lines)
Bram Moolenaar55759b52020-09-30 22:59:42 +02002204 assert_equal('', g:var_uninit)
2205 assert_equal('text', g:var_test)
2206 assert_equal('prefixed', g:var_prefixed)
2207 assert_equal(1234, g:other_var)
2208 assert_equal(222, g:dict_val)
2209
2210 unlet g:var_uninit
2211 unlet g:var_test
2212 unlet g:var_prefixed
2213 unlet g:other_var
Bram Moolenaar6e50ec22021-04-03 19:32:44 +02002214 unlet g:globConst
Bram Moolenaar89b474d2020-12-22 21:19:39 +01002215 unlet g:FOOS
2216 unlet g:FLIST
Bram Moolenaar89b474d2020-12-22 21:19:39 +01002217 unlet w:FOOS
2218 unlet w:FLIST
Bram Moolenaar55759b52020-09-30 22:59:42 +02002219enddef
2220
Bram Moolenaar566badc2022-09-18 13:46:08 +01002221def Test_create_list_after_const()
2222 const a = 1
2223 g:ll = []
2224 assert_equal(0, islocked('g:ll'))
2225 unlet g:ll
2226enddef
2227
Bram Moolenaar083966f2020-12-08 11:29:40 +01002228def Test_var_declaration_fails()
Bram Moolenaar55759b52020-09-30 22:59:42 +02002229 var lines =<< trim END
2230 vim9script
2231 final var: string
2232 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002233 v9.CheckScriptFailure(lines, 'E1125:')
Bram Moolenaar55759b52020-09-30 22:59:42 +02002234
2235 lines =<< trim END
2236 vim9script
Bram Moolenaar89b474d2020-12-22 21:19:39 +01002237 const g:constvar = 'string'
2238 g:constvar = 'xx'
2239 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002240 v9.CheckScriptFailure(lines, 'E741:')
Bram Moolenaar89b474d2020-12-22 21:19:39 +01002241 unlet g:constvar
2242
2243 lines =<< trim END
2244 vim9script
Bram Moolenaarf5906aa2021-04-02 14:35:15 +02002245 var name = 'one'
2246 lockvar name
2247 def SetLocked()
2248 name = 'two'
2249 enddef
2250 SetLocked()
2251 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002252 v9.CheckScriptFailure(lines, 'E741: Value is locked: name', 1)
Bram Moolenaardcf29ac2021-04-02 14:44:02 +02002253
2254 lines =<< trim END
2255 let s:legacy = 'one'
2256 lockvar s:legacy
2257 def SetLocked()
2258 s:legacy = 'two'
2259 enddef
2260 call SetLocked()
2261 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002262 v9.CheckScriptFailure(lines, 'E741: Value is locked: s:legacy', 1)
Bram Moolenaarf5906aa2021-04-02 14:35:15 +02002263
2264 lines =<< trim END
2265 vim9script
Bram Moolenaard877a572021-04-01 19:42:48 +02002266 def SetGlobalConst()
2267 const g:globConst = 123
2268 enddef
2269 SetGlobalConst()
2270 g:globConst = 234
2271 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002272 v9.CheckScriptFailure(lines, 'E741: Value is locked: g:globConst', 6)
Bram Moolenaard877a572021-04-01 19:42:48 +02002273 unlet g:globConst
2274
2275 lines =<< trim END
2276 vim9script
Bram Moolenaar08251752021-01-11 21:20:18 +01002277 const cdict: dict<string> = {}
2278 def Change()
2279 cdict.foo = 'foo'
2280 enddef
2281 defcompile
2282 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002283 v9.CheckScriptFailure(lines, 'E46:')
Bram Moolenaar08251752021-01-11 21:20:18 +01002284
2285 lines =<< trim END
2286 vim9script
Bram Moolenaar89b474d2020-12-22 21:19:39 +01002287 final w:finalvar = [9]
2288 w:finalvar = [8]
2289 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002290 v9.CheckScriptFailure(lines, 'E1122:')
Bram Moolenaar89b474d2020-12-22 21:19:39 +01002291 unlet w:finalvar
2292
2293 lines =<< trim END
2294 vim9script
Bram Moolenaar55759b52020-09-30 22:59:42 +02002295 const var: string
2296 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002297 v9.CheckScriptFailure(lines, 'E1021:')
Bram Moolenaar55759b52020-09-30 22:59:42 +02002298
2299 lines =<< trim END
2300 vim9script
2301 var 9var: string
2302 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002303 v9.CheckScriptFailure(lines, 'E488:')
Bram Moolenaar7257af42020-12-08 22:08:51 +01002304
Bram Moolenaar62aec932022-01-29 21:45:34 +00002305 v9.CheckDefFailure(['var foo.bar = 2'], 'E1087:')
2306 v9.CheckDefFailure(['var foo[3] = 2'], 'E1087:')
2307 v9.CheckDefFailure(['const foo: number'], 'E1021:')
Bram Moolenaare1d12112022-03-05 11:37:48 +00002308
2309 lines =<< trim END
2310 va foo = 123
2311 END
2312 v9.CheckDefAndScriptFailure(lines, 'E1065:', 1)
Bram Moolenaar31d99482022-05-26 22:24:43 +01002313
2314 lines =<< trim END
2315 var foo: func(number
2316 END
2317 v9.CheckDefAndScriptFailure(lines, 'E110:', 1)
2318
2319 lines =<< trim END
2320 var foo: func(number): func(
2321 END
2322 v9.CheckDefAndScriptFailure(lines, 'E110:', 1)
2323
2324 for type in ['num_ber',
2325 'anys', 'ani',
2326 'bools', 'boel',
2327 'blobs', 'blub',
2328 'channels', 'channol',
2329 'dicts', 'duct',
2330 'floats', 'floot',
2331 'funcs', 'funk',
2332 'jobs', 'jop',
Bram Moolenaar2984ed32022-08-20 14:51:17 +01002333 'lists', 'last',
Bram Moolenaar31d99482022-05-26 22:24:43 +01002334 'numbers', 'numbar',
2335 'strings', 'strung',
2336 'voids', 'viod']
2337 v9.CheckDefAndScriptFailure([$'var foo: {type}'], 'E1010:', 1)
2338 endfor
Bram Moolenaar55759b52020-09-30 22:59:42 +02002339enddef
2340
Bram Moolenaare88c6b72022-02-15 15:37:11 +00002341def Test_var_declaration_inferred()
2342 # check that type is set on the list so that extend() fails
2343 var lines =<< trim END
2344 vim9script
2345 def GetList(): list<number>
2346 var l = [1, 2, 3]
2347 return l
2348 enddef
2349 echo GetList()->extend(['x'])
2350 END
2351 v9.CheckScriptFailure(lines, 'E1013:', 6)
Bram Moolenaar31d99482022-05-26 22:24:43 +01002352
2353 lines =<< trim END
2354 vim9script
2355 def GetNr(): number
2356 return 5
2357 enddef
2358 def TestOne()
2359 var some = [function('len'), GetNr]
2360 g:res = typename(some)
2361 enddef
2362 TestOne()
2363 assert_equal('list<func(): number>', g:res)
2364
2365 def TestTwo()
2366 var some = [function('len'), GetNr]
2367 g:res = typename(some)
2368 enddef
2369 TestTwo()
2370 assert_equal('list<func(): number>', g:res)
2371 unlet g:res
2372
2373 # FIXME: why is the type different?
2374 var first = [function('len'), GetNr]
2375 assert_equal('list<func(...): number>', typename(first))
2376 var second = [GetNr, function('len')]
2377 assert_equal('list<func(...): number>', typename(second))
2378 END
2379 v9.CheckScriptSuccess(lines)
Bram Moolenaare88c6b72022-02-15 15:37:11 +00002380enddef
2381
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002382def Test_script_local_in_legacy()
Bram Moolenaara749a422022-02-12 19:52:25 +00002383 # OK to define script-local later but before compiling
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002384 var lines =<< trim END
2385 def SetLater()
Bram Moolenaara749a422022-02-12 19:52:25 +00002386 legvar = 'two'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002387 enddef
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02002388 let s:legvar = 'one'
Bram Moolenaara749a422022-02-12 19:52:25 +00002389 defcompile
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002390 call SetLater()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02002391 call assert_equal('two', s:legvar)
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002392 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002393 v9.CheckScriptSuccess(lines)
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002394
2395 # OK to leave out s: prefix when script-local already defined
2396 lines =<< trim END
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02002397 let s:legvar = 'one'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002398 def SetNoPrefix()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02002399 legvar = 'two'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002400 enddef
2401 call SetNoPrefix()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02002402 call assert_equal('two', s:legvar)
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002403 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002404 v9.CheckScriptSuccess(lines)
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002405
Bram Moolenaara749a422022-02-12 19:52:25 +00002406 # Not OK to leave out s: prefix when script-local defined after compiling
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002407 lines =<< trim END
2408 def SetLaterNoPrefix()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02002409 legvar = 'two'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002410 enddef
2411 defcompile
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02002412 let s:legvar = 'one'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002413 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002414 v9.CheckScriptFailure(lines, 'E476:', 1)
Bram Moolenaar47bc9c32021-07-17 21:24:56 +02002415
Bram Moolenaarb18b4962022-09-02 21:55:50 +01002416 edit! Xslfile
Bram Moolenaar47bc9c32021-07-17 21:24:56 +02002417 lines =<< trim END
2418 var edit: bool
2419 legacy edit
2420 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002421 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002422enddef
2423
Bram Moolenaar083966f2020-12-08 11:29:40 +01002424def Test_var_type_check()
Bram Moolenaar55759b52020-09-30 22:59:42 +02002425 var lines =<< trim END
2426 vim9script
2427 var name: string
2428 name = 1234
2429 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002430 v9.CheckScriptFailure(lines, 'E1012:')
Bram Moolenaar55759b52020-09-30 22:59:42 +02002431
2432 lines =<< trim END
2433 vim9script
2434 var name:string
2435 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002436 v9.CheckScriptFailure(lines, 'E1069:')
Bram Moolenaar55759b52020-09-30 22:59:42 +02002437
Bram Moolenaar62aec932022-01-29 21:45:34 +00002438 v9.CheckDefAndScriptFailure(['var n:number = 42'], 'E1069:')
Bram Moolenaar60faf862021-08-23 22:22:45 +02002439
Bram Moolenaar55759b52020-09-30 22:59:42 +02002440 lines =<< trim END
2441 vim9script
2442 var name: asdf
2443 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002444 v9.CheckScriptFailure(lines, 'E1010:')
Bram Moolenaar55759b52020-09-30 22:59:42 +02002445
2446 lines =<< trim END
2447 vim9script
Bram Moolenaara749a422022-02-12 19:52:25 +00002448 var l: list<number>
2449 l = []
Bram Moolenaar55759b52020-09-30 22:59:42 +02002450 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002451 v9.CheckScriptSuccess(lines)
Bram Moolenaar55759b52020-09-30 22:59:42 +02002452
2453 lines =<< trim END
2454 vim9script
Bram Moolenaara749a422022-02-12 19:52:25 +00002455 var d: dict<number>
2456 d = {}
Bram Moolenaar55759b52020-09-30 22:59:42 +02002457 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002458 v9.CheckScriptSuccess(lines)
Bram Moolenaar378697a2021-07-15 19:23:18 +02002459
2460 lines =<< trim END
2461 vim9script
2462 var d = {a: 1, b: [2]}
2463 def Func(b: bool)
2464 var l: list<number> = b ? d.b : [3]
2465 enddef
2466 defcompile
2467 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002468 v9.CheckScriptSuccess(lines)
Bram Moolenaar55759b52020-09-30 22:59:42 +02002469enddef
2470
2471let g:dict_number = #{one: 1, two: 2}
2472
Bram Moolenaar083966f2020-12-08 11:29:40 +01002473def Test_var_list_dict_type()
Bram Moolenaar55759b52020-09-30 22:59:42 +02002474 var ll: list<number>
2475 ll = [1, 2, 2, 3, 3, 3]->uniq()
2476 ll->assert_equal([1, 2, 3])
2477
2478 var dd: dict<number>
2479 dd = g:dict_number
2480 dd->assert_equal(g:dict_number)
2481
2482 var lines =<< trim END
2483 var ll: list<number>
2484 ll = [1, 2, 3]->map('"one"')
2485 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002486 v9.CheckDefExecFailure(lines, 'E1012: Type mismatch; expected list<number> but got list<string>')
Bram Moolenaar55759b52020-09-30 22:59:42 +02002487enddef
2488
Bram Moolenaar083966f2020-12-08 11:29:40 +01002489def Test_cannot_use_let()
Bram Moolenaar62aec932022-01-29 21:45:34 +00002490 v9.CheckDefAndScriptFailure(['let a = 34'], 'E1126:', 1)
Bram Moolenaar083966f2020-12-08 11:29:40 +01002491enddef
2492
Bram Moolenaar55759b52020-09-30 22:59:42 +02002493def Test_unlet()
2494 g:somevar = 'yes'
2495 assert_true(exists('g:somevar'))
2496 unlet g:somevar
2497 assert_false(exists('g:somevar'))
2498 unlet! g:somevar
2499
2500 # also works for script-local variable in legacy Vim script
2501 s:somevar = 'legacy'
2502 assert_true(exists('s:somevar'))
2503 unlet s:somevar
2504 assert_false(exists('s:somevar'))
2505 unlet! s:somevar
2506
Bram Moolenaar83d0cec2022-02-04 21:17:58 +00002507 if 0
2508 unlet g:does_not_exist
2509 endif
2510
2511 v9.CheckDefExecFailure(['unlet v:notfound.key'], 'E1001:')
2512
Bram Moolenaar62aec932022-01-29 21:45:34 +00002513 v9.CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01002514 'var dd = 111',
2515 'unlet dd',
2516 ], 'E1081:', 2)
2517
Bram Moolenaar752fc692021-01-04 21:57:11 +01002518 # dict unlet
Bram Moolenaarea5c8982022-02-17 14:42:02 +00002519 var dd = {a: 1, b: 2, c: 3, 4: 4}
Bram Moolenaar752fc692021-01-04 21:57:11 +01002520 unlet dd['a']
2521 unlet dd.c
Bram Moolenaarea5c8982022-02-17 14:42:02 +00002522 unlet dd[4]
Bram Moolenaar752fc692021-01-04 21:57:11 +01002523 assert_equal({b: 2}, dd)
2524
Bram Moolenaar6b8c7ba2022-03-20 17:46:06 +00002525 # null key works like empty string
2526 dd = {'': 1, x: 9}
2527 unlet dd[null_string]
2528 assert_equal({x: 9}, dd)
2529
Bram Moolenaar752fc692021-01-04 21:57:11 +01002530 # list unlet
2531 var ll = [1, 2, 3, 4]
2532 unlet ll[1]
2533 unlet ll[-1]
2534 assert_equal([1, 3], ll)
2535
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01002536 ll = [1, 2, 3, 4]
2537 unlet ll[0 : 1]
2538 assert_equal([3, 4], ll)
2539
2540 ll = [1, 2, 3, 4]
2541 unlet ll[2 : 8]
2542 assert_equal([1, 2], ll)
2543
2544 ll = [1, 2, 3, 4]
2545 unlet ll[-2 : -1]
2546 assert_equal([1, 2], ll)
2547
Bram Moolenaar0c7f2612022-02-17 19:44:07 +00002548 g:nrdict = {1: 1, 2: 2}
2549 g:idx = 1
2550 unlet g:nrdict[g:idx]
2551 assert_equal({2: 2}, g:nrdict)
2552 unlet g:nrdict
2553 unlet g:idx
2554
Bram Moolenaar62aec932022-01-29 21:45:34 +00002555 v9.CheckDefFailure([
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01002556 'var ll = [1, 2]',
2557 'll[1 : 2] = 7',
Bram Moolenaar4f0884d2021-08-11 21:49:23 +02002558 ], 'E1012: Type mismatch; expected list<number> but got number', 2)
Bram Moolenaar62aec932022-01-29 21:45:34 +00002559 v9.CheckDefFailure([
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01002560 'var dd = {a: 1}',
2561 'unlet dd["a" : "a"]',
2562 ], 'E1166:', 2)
Bram Moolenaar62aec932022-01-29 21:45:34 +00002563 v9.CheckDefExecFailure([
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01002564 'unlet g:adict[0 : 1]',
2565 ], 'E1148:', 1)
Bram Moolenaar62aec932022-01-29 21:45:34 +00002566 v9.CheckDefFailure([
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01002567 'var ll = [1, 2]',
2568 'unlet ll[0:1]',
2569 ], 'E1004:', 2)
Bram Moolenaar62aec932022-01-29 21:45:34 +00002570 v9.CheckDefFailure([
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01002571 'var ll = [1, 2]',
2572 'unlet ll[0 :1]',
2573 ], 'E1004:', 2)
Bram Moolenaar62aec932022-01-29 21:45:34 +00002574 v9.CheckDefFailure([
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01002575 'var ll = [1, 2]',
2576 'unlet ll[0: 1]',
2577 ], 'E1004:', 2)
Bram Moolenaare08be092022-02-17 13:08:26 +00002578
2579 v9.CheckDefExecFailure([
2580 'g:ll = [1, 2]',
2581 'g:idx = "x"',
2582 'unlet g:ll[g:idx]',
2583 ], 'E1029: Expected number but got string', 3)
2584
2585 v9.CheckDefExecFailure([
2586 'g:ll = [1, 2, 3]',
2587 'g:idx = "x"',
2588 'unlet g:ll[g:idx : 2]',
2589 ], 'E1029: Expected number but got string', 3)
2590
2591 v9.CheckDefExecFailure([
2592 'g:ll = [1, 2, 3]',
2593 'g:idx = "x"',
2594 'unlet g:ll[0 : g:idx]',
2595 ], 'E1029: Expected number but got string', 3)
2596
Bram Moolenaar7f2c3412021-11-29 16:01:49 +00002597 # command recognized as assignment when skipping, should not give an error
Bram Moolenaar62aec932022-01-29 21:45:34 +00002598 v9.CheckScriptSuccess([
Bram Moolenaar7f2c3412021-11-29 16:01:49 +00002599 'vim9script',
2600 'for i in []',
2601 " put =''",
2602 'endfor'])
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01002603
Bram Moolenaar62aec932022-01-29 21:45:34 +00002604 v9.CheckDefFailure([
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01002605 'var ll = [1, 2]',
2606 'unlet ll["x" : 1]',
2607 ], 'E1012:', 2)
Bram Moolenaar62aec932022-01-29 21:45:34 +00002608 v9.CheckDefFailure([
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01002609 'var ll = [1, 2]',
2610 'unlet ll[0 : "x"]',
2611 ], 'E1012:', 2)
2612
Bram Moolenaar752fc692021-01-04 21:57:11 +01002613 # list of dict unlet
2614 var dl = [{a: 1, b: 2}, {c: 3}]
2615 unlet dl[0]['b']
2616 assert_equal([{a: 1}, {c: 3}], dl)
2617
Bram Moolenaar62aec932022-01-29 21:45:34 +00002618 v9.CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01002619 'var ll = test_null_list()',
2620 'unlet ll[0]',
2621 ], 'E684:', 2)
Bram Moolenaar62aec932022-01-29 21:45:34 +00002622 v9.CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01002623 'var ll = [1]',
2624 'unlet ll[2]',
2625 ], 'E684:', 2)
Bram Moolenaar62aec932022-01-29 21:45:34 +00002626 v9.CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01002627 'var ll = [1]',
2628 'unlet ll[g:astring]',
Bram Moolenaarf30a14d2021-01-17 21:51:24 +01002629 ], 'E1012:', 2)
Bram Moolenaar62aec932022-01-29 21:45:34 +00002630 v9.CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01002631 'var dd = test_null_dict()',
2632 'unlet dd["a"]',
2633 ], 'E716:', 2)
Bram Moolenaar62aec932022-01-29 21:45:34 +00002634 v9.CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01002635 'var dd = {a: 1}',
2636 'unlet dd["b"]',
2637 ], 'E716:', 2)
Bram Moolenaar62aec932022-01-29 21:45:34 +00002638 v9.CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01002639 'var dd = {a: 1}',
2640 'unlet dd[g:alist]',
2641 ], 'E1105:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01002642
Bram Moolenaarea5c8982022-02-17 14:42:02 +00002643 v9.CheckDefExecFailure([
Bram Moolenaar2984ed32022-08-20 14:51:17 +01002644 'g:dd = {"a": 1, 2: 2}',
Bram Moolenaarea5c8982022-02-17 14:42:02 +00002645 'unlet g:dd[0z11]',
2646 ], 'E1029:', 2)
Bram Moolenaar6296d1e2022-02-17 16:30:11 +00002647 v9.CheckDefExecFailure([
Bram Moolenaar2984ed32022-08-20 14:51:17 +01002648 'g:str = "a string"',
Bram Moolenaar6296d1e2022-02-17 16:30:11 +00002649 'unlet g:str[0]',
2650 ], 'E1148: Cannot index a string', 2)
Bram Moolenaarea5c8982022-02-17 14:42:02 +00002651
Bram Moolenaar2ef951d2021-01-03 20:55:26 +01002652 # can compile unlet before variable exists
Bram Moolenaar752fc692021-01-04 21:57:11 +01002653 g:someDict = {key: 'val'}
2654 var k = 'key'
2655 unlet g:someDict[k]
2656 assert_equal({}, g:someDict)
2657 unlet g:someDict
2658 assert_false(exists('g:someDict'))
Bram Moolenaar2ef951d2021-01-03 20:55:26 +01002659
Bram Moolenaar62aec932022-01-29 21:45:34 +00002660 v9.CheckScriptFailure([
Bram Moolenaar55759b52020-09-30 22:59:42 +02002661 'vim9script',
2662 'var svar = 123',
2663 'unlet svar',
2664 ], 'E1081:')
Bram Moolenaar62aec932022-01-29 21:45:34 +00002665 v9.CheckScriptFailure([
Bram Moolenaar55759b52020-09-30 22:59:42 +02002666 'vim9script',
2667 'var svar = 123',
2668 'unlet s:svar',
Bram Moolenaara749a422022-02-12 19:52:25 +00002669 ], 'E1268:')
Bram Moolenaar62aec932022-01-29 21:45:34 +00002670 v9.CheckScriptFailure([
Bram Moolenaar55759b52020-09-30 22:59:42 +02002671 'vim9script',
2672 'var svar = 123',
2673 'def Func()',
2674 ' unlet svar',
2675 'enddef',
2676 'defcompile',
2677 ], 'E1081:')
Bram Moolenaar62aec932022-01-29 21:45:34 +00002678 v9.CheckScriptFailure([
Bram Moolenaar55759b52020-09-30 22:59:42 +02002679 'vim9script',
2680 'var svar = 123',
Bram Moolenaar9aed7292020-12-18 15:38:00 +01002681 'func Func()',
2682 ' unlet s:svar',
2683 'endfunc',
2684 'Func()',
2685 ], 'E1081:')
Bram Moolenaar62aec932022-01-29 21:45:34 +00002686 v9.CheckScriptFailure([
Bram Moolenaar9aed7292020-12-18 15:38:00 +01002687 'vim9script',
2688 'var svar = 123',
Bram Moolenaar55759b52020-09-30 22:59:42 +02002689 'def Func()',
2690 ' unlet s:svar',
2691 'enddef',
2692 'defcompile',
2693 ], 'E1081:')
2694
Bram Moolenaar62aec932022-01-29 21:45:34 +00002695 v9.CheckScriptFailure([
Bram Moolenaar71b76852021-12-17 20:15:38 +00002696 'vim9script',
2697 'def Delcount(dict: dict<any>)',
2698 ' unlet dict.count',
2699 'enddef',
2700 'Delcount(v:)',
2701 ], 'E742:')
2702
Bram Moolenaar62aec932022-01-29 21:45:34 +00002703 v9.CheckScriptFailure([
Bram Moolenaar71b76852021-12-17 20:15:38 +00002704 'vim9script',
2705 'def DelChangedtick(dict: dict<any>)',
2706 ' unlet dict.changedtick',
2707 'enddef',
2708 'DelChangedtick(b:)',
2709 ], 'E795:')
2710
Bram Moolenaar0e9bdad2022-10-15 20:06:33 +01002711 writefile(['vim9script', 'export var svar = 1234'], 'XunletExport.vim', 'D')
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01002712 var lines =<< trim END
2713 vim9script
Bram Moolenaard5f400c2022-01-06 21:10:28 +00002714 import './XunletExport.vim' as exp
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01002715 def UnletSvar()
Bram Moolenaard5f400c2022-01-06 21:10:28 +00002716 unlet exp.svar
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01002717 enddef
2718 defcompile
2719 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002720 v9.CheckScriptFailure(lines, 'E1260:', 1)
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01002721
Bram Moolenaar55759b52020-09-30 22:59:42 +02002722 $ENVVAR = 'foobar'
2723 assert_equal('foobar', $ENVVAR)
2724 unlet $ENVVAR
2725 assert_equal('', $ENVVAR)
2726enddef
2727
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01002728def Test_expr_error_no_assign()
2729 var lines =<< trim END
2730 vim9script
2731 var x = invalid
2732 echo x
2733 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002734 v9.CheckScriptFailureList(lines, ['E121:', 'E121:'])
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01002735
2736 lines =<< trim END
2737 vim9script
2738 var x = 1 / 0
2739 echo x
2740 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002741 v9.CheckScriptFailure(lines, 'E1154:')
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01002742
2743 lines =<< trim END
2744 vim9script
2745 var x = 1 % 0
2746 echo x
2747 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002748 v9.CheckScriptFailure(lines, 'E1154:')
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01002749
2750 lines =<< trim END
2751 var x: string 'string'
2752 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002753 v9.CheckDefAndScriptFailure(lines, 'E488:')
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01002754enddef
2755
2756
Bram Moolenaar17126b12021-01-07 22:03:02 +01002757def Test_assign_command_modifier()
2758 var lines =<< trim END
Bram Moolenaar9e0f8832021-01-09 12:09:22 +01002759 var verbose = 0
2760 verbose = 1
2761 assert_equal(1, verbose)
2762 silent verbose = 2
2763 assert_equal(2, verbose)
2764 silent verbose += 2
2765 assert_equal(4, verbose)
2766 silent verbose -= 1
2767 assert_equal(3, verbose)
Bram Moolenaar17126b12021-01-07 22:03:02 +01002768
Bram Moolenaar9e0f8832021-01-09 12:09:22 +01002769 var topleft = {one: 1}
2770 sandbox topleft.one = 3
2771 assert_equal({one: 3}, topleft)
2772 leftabove topleft[' '] = 4
2773 assert_equal({one: 3, ' ': 4}, topleft)
2774
2775 var x: number
2776 var y: number
2777 silent [x, y] = [1, 2]
2778 assert_equal(1, x)
2779 assert_equal(2, y)
Bram Moolenaar17126b12021-01-07 22:03:02 +01002780 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002781 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar17126b12021-01-07 22:03:02 +01002782enddef
2783
Bram Moolenaar74f4a962021-06-17 21:03:07 +02002784def Test_assign_alt_buf_register()
2785 var lines =<< trim END
2786 edit 'file_b1'
2787 var b1 = bufnr()
2788 edit 'file_b2'
2789 var b2 = bufnr()
2790 assert_equal(b1, bufnr('#'))
2791 @# = b2
2792 assert_equal(b2, bufnr('#'))
2793 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002794 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar74f4a962021-06-17 21:03:07 +02002795enddef
2796
Bram Moolenaar32154662021-03-28 21:14:06 +02002797def Test_script_funcref_case()
2798 var lines =<< trim END
2799 var Len = (s: string): number => len(s) + 1
2800 assert_equal(5, Len('asdf'))
2801 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002802 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar32154662021-03-28 21:14:06 +02002803
2804 lines =<< trim END
2805 var len = (s: string): number => len(s) + 1
2806 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002807 v9.CheckDefAndScriptFailure(lines, 'E704:')
Bram Moolenaar32154662021-03-28 21:14:06 +02002808
2809 lines =<< trim END
2810 vim9script
Bram Moolenaara749a422022-02-12 19:52:25 +00002811 var Len = (s: string): number => len(s) + 2
Bram Moolenaar32154662021-03-28 21:14:06 +02002812 assert_equal(6, Len('asdf'))
2813 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002814 v9.CheckScriptSuccess(lines)
Bram Moolenaar32154662021-03-28 21:14:06 +02002815
2816 lines =<< trim END
2817 vim9script
Bram Moolenaara749a422022-02-12 19:52:25 +00002818 var len = (s: string): number => len(s) + 1
Bram Moolenaar32154662021-03-28 21:14:06 +02002819 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002820 v9.CheckScriptFailure(lines, 'E704:')
Bram Moolenaar32154662021-03-28 21:14:06 +02002821enddef
2822
Bram Moolenaar44a89772021-12-18 12:31:33 +00002823def Test_script_funcref_runtime_type_check()
2824 var lines =<< trim END
2825 vim9script
2826 def FuncWithNumberArg(n: number)
2827 enddef
2828 def Test()
2829 var Ref: func(string) = function(FuncWithNumberArg)
2830 enddef
2831 defcompile
2832 END
2833 # OK at compile time
Bram Moolenaar62aec932022-01-29 21:45:34 +00002834 v9.CheckScriptSuccess(lines)
Bram Moolenaar44a89772021-12-18 12:31:33 +00002835
2836 # Type check fails at runtime
Bram Moolenaar62aec932022-01-29 21:45:34 +00002837 v9.CheckScriptFailure(lines + ['Test()'], 'E1012: Type mismatch; expected func(string) but got func(number)')
Bram Moolenaar44a89772021-12-18 12:31:33 +00002838enddef
2839
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002840def Test_inc_dec()
2841 var lines =<< trim END
2842 var nr = 7
2843 ++nr
Bram Moolenaar730bf302021-04-24 20:43:56 +02002844 assert_equal(8, nr)
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002845 --nr
Bram Moolenaar730bf302021-04-24 20:43:56 +02002846 assert_equal(7, nr)
Bram Moolenaarf3d30842021-06-25 19:29:30 +02002847 ++nr | ++nr
2848 assert_equal(9, nr)
2849 ++nr # comment
2850 assert_equal(10, nr)
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002851
2852 var ll = [1, 2]
2853 --ll[0]
2854 ++ll[1]
Bram Moolenaar730bf302021-04-24 20:43:56 +02002855 assert_equal([0, 3], ll)
2856
2857 g:count = 1
2858 ++g:count
2859 --g:count
2860 assert_equal(1, g:count)
2861 unlet g:count
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002862 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002863 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar22480d12021-06-25 21:31:09 +02002864
2865 lines =<< trim END
2866 var nr = 7
2867 ++ nr
2868 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002869 v9.CheckDefAndScriptFailure(lines, "E1202: No white space allowed after '++': ++ nr")
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002870enddef
2871
Bram Moolenaar327d3ee2021-07-28 19:34:14 +02002872def Test_abort_after_error()
2873 # should abort after strpart() fails, not give another type error
2874 var lines =<< trim END
2875 vim9script
2876 var x: string
2877 x = strpart(1, 2)
2878 END
Bram Moolenaar0e9bdad2022-10-15 20:06:33 +01002879 writefile(lines, 'Xtestscript', 'D')
Bram Moolenaar327d3ee2021-07-28 19:34:14 +02002880 var expected = 'E1174: String required for argument 1'
2881 assert_fails('so Xtestscript', [expected, expected], 3)
Bram Moolenaar327d3ee2021-07-28 19:34:14 +02002882enddef
2883
Bram Moolenaarafa048f2022-02-22 20:43:36 +00002884def Test_using_s_var_in_function()
2885 var lines =<< trim END
2886 vim9script
2887 var scriptlevel = 123
2888 def SomeFunc()
2889 echo s:scriptlevel
2890 enddef
2891 SomeFunc()
2892 END
2893 v9.CheckScriptFailure(lines, 'E1268:')
2894
2895 # OK in legacy script
2896 lines =<< trim END
2897 let s:scriptlevel = 123
2898 def s:SomeFunc()
2899 echo s:scriptlevel
2900 enddef
2901 call s:SomeFunc()
2902 END
2903 v9.CheckScriptSuccess(lines)
2904
2905 lines =<< trim END
2906 vim9script
2907 var scriptlevel = 123
2908 def SomeFunc()
2909 s:scriptlevel = 456
2910 enddef
2911 SomeFunc()
2912 END
2913 v9.CheckScriptFailure(lines, 'E1268:')
2914
2915 # OK in legacy script
2916 lines =<< trim END
2917 let s:scriptlevel = 123
2918 def s:SomeFunc()
2919 s:scriptlevel = 456
2920 enddef
2921 call s:SomeFunc()
2922 call assert_equal(456, s:scriptlevel)
2923 END
2924 v9.CheckScriptSuccess(lines)
2925enddef
2926
Yegappan Lakshmananb5a07192023-10-05 20:14:43 +02002927" Test for specifying a type in assignment
2928def Test_type_specification_in_assignment()
2929 # specify type for an existing script local variable without "var"
2930 var lines =<< trim END
2931 vim9script
2932 var n: number = 10
2933 n: number = 20
2934 END
2935 v9.CheckSourceFailure(lines, 'E488: Trailing characters: : number = 20', 3)
2936
2937 # specify type for a non-existing script local variable without "var"
2938 lines =<< trim END
2939 vim9script
2940 MyVar: string = 'abc'
2941 END
2942 v9.CheckSourceFailure(lines, "E492: Not an editor command: MyVar: string = 'abc'", 2)
2943
2944 # specify type for an existing def local variable without "var"
2945 lines =<< trim END
2946 vim9script
2947 def Foo()
2948 var n: number = 10
2949 n: number = 20
2950 enddef
2951 Foo()
2952 END
2953 v9.CheckSourceFailure(lines, 'E488: Trailing characters: : number = 20', 2)
2954
2955 # specify type for a non-existing def local variable without "var"
2956 lines =<< trim END
2957 vim9script
2958 def Foo()
2959 MyVar: string = 'abc'
2960 enddef
2961 Foo()
2962 END
2963 v9.CheckSourceFailure(lines, "E476: Invalid command: MyVar: string = 'abc'", 1)
2964enddef
2965
Yegappan Lakshmananefbfa862022-04-17 12:47:40 +01002966let g:someVar = 'X'
2967
2968" Test for heredoc with Vim expressions.
2969" This messes up highlighting, keep it near the end.
2970def Test_heredoc_expr()
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01002971 var lines =<< trim CODE
2972 var s = "local"
2973 var a1 = "1"
2974 var a2 = "2"
2975 var a3 = "3"
2976 var a4 = ""
2977 var code =<< trim eval END
LemonBoy2eaef102022-05-06 13:14:50 +01002978 var a = {5 + 10}
2979 var b = {min([10, 6])} + {max([4, 6])}
2980 var c = "{s}"
2981 var d = x{a1}x{a2}x{a3}x{a4}
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01002982 END
2983 assert_equal(['var a = 15', 'var b = 6 + 6', 'var c = "local"', 'var d = x1x2x3x'], code)
2984 CODE
2985 v9.CheckDefAndScriptSuccess(lines)
Yegappan Lakshmananefbfa862022-04-17 12:47:40 +01002986
Yegappan Lakshmananf01493c2024-04-14 23:21:02 +02002987 # Evaluate a dictionary
2988 lines =<< trim CODE
2989 var d1 = {'a': 10, 'b': [1, 2]}
2990 var code =<< trim eval END
2991 var d2 = {d1}
2992 END
2993 assert_equal(["var d2 = {'a': 10, 'b': [1, 2]}"], code)
2994 CODE
2995 v9.CheckDefAndScriptSuccess(lines)
2996
Yegappan Lakshmananbce51d92024-04-15 19:19:52 +02002997 # Evaluate an empty dictionary
2998 lines =<< trim CODE
2999 var d1 = {}
3000 var code =<< trim eval END
3001 var d2 = {d1}
3002 END
3003 assert_equal(["var d2 = {}"], code)
3004 CODE
3005 v9.CheckDefAndScriptSuccess(lines)
3006
3007 # Evaluate a null dictionary
3008 lines =<< trim CODE
3009 var d1 = test_null_dict()
3010 var code =<< trim eval END
3011 var d2 = {d1}
3012 END
3013 assert_equal(["var d2 = {}"], code)
3014 CODE
3015 v9.CheckDefAndScriptSuccess(lines)
3016
3017 # Evaluate a List
3018 lines =<< trim CODE
3019 var l1 = ['a', 'b', 'c']
3020 var code =<< trim eval END
3021 var l2 = {l1}
3022 END
3023 assert_equal(["var l2 = ['a', 'b', 'c']"], code)
3024 CODE
3025 v9.CheckDefAndScriptSuccess(lines)
3026
3027 # Evaluate an empty List
3028 lines =<< trim CODE
3029 var l1 = []
3030 var code =<< trim eval END
3031 var l2 = {l1}
3032 END
3033 assert_equal(["var l2 = []"], code)
3034 CODE
3035 v9.CheckDefAndScriptSuccess(lines)
3036
3037 # Evaluate a null List
3038 lines =<< trim CODE
3039 var l1 = test_null_list()
3040 var code =<< trim eval END
3041 var l2 = {l1}
3042 END
3043 assert_equal(["var l2 = []"], code)
3044 CODE
3045 v9.CheckDefAndScriptSuccess(lines)
3046
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01003047 lines =<< trim CODE
3048 var code =<< eval trim END
LemonBoy2eaef102022-05-06 13:14:50 +01003049 var s = "{$SOME_ENV_VAR}"
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01003050 END
3051 assert_equal(['var s = "somemore"'], code)
3052 CODE
3053 v9.CheckDefAndScriptSuccess(lines)
Yegappan Lakshmananefbfa862022-04-17 12:47:40 +01003054
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01003055 lines =<< trim CODE
3056 var code =<< eval END
LemonBoy2eaef102022-05-06 13:14:50 +01003057 var s = "{$SOME_ENV_VAR}"
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01003058 END
3059 assert_equal([' var s = "somemore"'], code)
3060 CODE
3061 v9.CheckDefAndScriptSuccess(lines)
Yegappan Lakshmananefbfa862022-04-17 12:47:40 +01003062
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01003063 lines =<< trim CODE
3064 var code =<< eval trim END
LemonBoy2eaef102022-05-06 13:14:50 +01003065 let a = {{abc}}
3066 let b = {g:someVar}
3067 let c = {{
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01003068 END
LemonBoy2eaef102022-05-06 13:14:50 +01003069 assert_equal(['let a = {abc}', 'let b = X', 'let c = {'], code)
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01003070 CODE
3071 v9.CheckDefAndScriptSuccess(lines)
Yegappan Lakshmananefbfa862022-04-17 12:47:40 +01003072
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01003073 lines =<< trim LINES
Yegappan Lakshmananefbfa862022-04-17 12:47:40 +01003074 var text =<< eval trim END
LemonBoy2eaef102022-05-06 13:14:50 +01003075 let b = {
Yegappan Lakshmananefbfa862022-04-17 12:47:40 +01003076 END
3077 LINES
LemonBoy2eaef102022-05-06 13:14:50 +01003078 v9.CheckDefAndScriptFailure(lines, "E1279: Missing '}'")
Yegappan Lakshmananefbfa862022-04-17 12:47:40 +01003079
3080 lines =<< trim LINES
3081 var text =<< eval trim END
LemonBoy2eaef102022-05-06 13:14:50 +01003082 let b = {abc
Yegappan Lakshmananefbfa862022-04-17 12:47:40 +01003083 END
3084 LINES
LemonBoy2eaef102022-05-06 13:14:50 +01003085 v9.CheckDefAndScriptFailure(lines, "E1279: Missing '}'")
Yegappan Lakshmananefbfa862022-04-17 12:47:40 +01003086
3087 lines =<< trim LINES
3088 var text =<< eval trim END
LemonBoy2eaef102022-05-06 13:14:50 +01003089 let b = {}
Yegappan Lakshmananefbfa862022-04-17 12:47:40 +01003090 END
3091 LINES
LemonBoy2eaef102022-05-06 13:14:50 +01003092 v9.CheckDefAndScriptFailure(lines, 'E15: Invalid expression: "}"')
Yegappan Lakshmananefbfa862022-04-17 12:47:40 +01003093enddef
3094
Yegappan Lakshmananc229a6a2023-10-26 23:05:07 +02003095" Test for assigning to a multi-dimensional list item.
3096def Test_list_item_assign()
3097 var lines =<< trim END
3098 vim9script
3099
3100 def Foo()
Yegappan Lakshmanan67098162023-11-05 10:07:03 +01003101 var l: list<list<string>> = [['x', 'x', 'x'], ['y', 'y', 'y']]
3102 var z: number = 1
Yegappan Lakshmananc229a6a2023-10-26 23:05:07 +02003103
Yegappan Lakshmanan67098162023-11-05 10:07:03 +01003104 [l[1][2], z] = ['a', 20]
3105 assert_equal([['x', 'x', 'x'], ['y', 'y', 'a']], l)
Yegappan Lakshmananc229a6a2023-10-26 23:05:07 +02003106 enddef
3107 Foo()
3108 END
3109 v9.CheckSourceSuccess(lines)
Yegappan Lakshmanan67098162023-11-05 10:07:03 +01003110
3111 lines =<< trim END
3112 vim9script
3113
3114 var l: list<list<string>> = [['x', 'x', 'x'], ['y', 'y', 'y']]
3115 var z: number = 1
3116
3117 [l[1][2], z] = ['a', 20]
3118 assert_equal([['x', 'x', 'x'], ['y', 'y', 'a']], l)
3119 END
3120 v9.CheckSourceSuccess(lines)
3121enddef
3122
3123" Test for assigning to a multi-dimensional dict item.
3124def Test_dict_item_assign()
3125 # This used to fail with the error "E1105: Cannot convert list to string"
3126 # (Github issue #13485)
3127 var lines =<< trim END
3128 vim9script
3129 def F()
3130 var d: dict<dict<number>> = {a: {b: 0}}
3131
3132 for group in keys(d)
3133 d['a']['b'] += 1
3134 endfor
3135 assert_equal({a: {b: 1}}, d)
3136 enddef
3137 F()
3138 END
3139 v9.CheckSourceSuccess(lines)
3140
3141 # This used to crash Vim
3142 lines =<< trim END
3143 vim9script
3144 def F()
3145 var d: dict<dict<number>> = {a: {b: 0}}
3146 d['a']['b'] += 1
3147 assert_equal({a: {b: 1}}, d)
3148 enddef
3149 F()
3150 END
3151 v9.CheckSourceSuccess(lines)
3152
3153 # Assignment at script level
3154 lines =<< trim END
3155 vim9script
3156 var d: dict<dict<number>> = {a: {b: 0}}
3157
3158 for group in keys(d)
3159 d['a']['b'] += 1
3160 endfor
3161 assert_equal({a: {b: 1}}, d)
3162 END
3163 v9.CheckSourceSuccess(lines)
Yegappan Lakshmananc229a6a2023-10-26 23:05:07 +02003164enddef
3165
Ernie Rael9ed53752023-12-11 17:40:46 +01003166def Test_class_assign()
3167 var lines =<< trim END
3168 vim9script
3169 class C
3170 endclass
3171 class D
3172 endclass
Ernie Rael2025af12023-12-12 16:58:00 +01003173 assert_fails('C = D', 'E1405: Class "D" cannot be used as a value')
Ernie Rael9ed53752023-12-11 17:40:46 +01003174 END
3175 v9.CheckSourceSuccess(lines)
3176enddef
3177
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003178" Test for using various types (dict, list, blob, funcref, class) as variable
3179" in assignments with a different type
3180def Test_type_check()
3181 var lines =<< trim END
3182 vim9script
3183 class A
3184 endclass
Ernie Raelb5011082023-12-18 08:34:41 +01003185 type T = number
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003186 var N: number = 1
3187 var S: string = 'abc'
3188 var d: dict<number> = {}
3189 var l: list<number> = []
3190 var b: blob = 0z10
3191 var Fn: func = function('min')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003192 var o: A = A.new()
3193
3194 # Assign a number
3195 assert_fails('d = N', 'E1012: Type mismatch; expected dict<number> but got number')
3196 assert_fails('l = N', 'E1012: Type mismatch; expected list<number> but got number')
3197 assert_fails('b = N', 'E1012: Type mismatch; expected blob but got number')
3198 assert_fails('Fn = N', 'E1012: Type mismatch; expected func(...): unknown but got number')
Ernie Raele75fde62023-12-21 17:18:54 +01003199 assert_fails('A = N', 'E1405: Class "A" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003200 assert_fails('o = N', 'E1012: Type mismatch; expected object<A> but got number')
Ernie Raele75fde62023-12-21 17:18:54 +01003201 assert_fails('T = N', 'E1403: Type alias "T" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003202
3203 # Use a compound operator with different LHS types
3204 assert_fails('d += N', 'E734: Wrong variable type for +=')
3205 assert_fails('l += N', 'E734: Wrong variable type for +=')
3206 assert_fails('b += N', 'E734: Wrong variable type for +=')
3207 assert_fails('Fn += N', 'E734: Wrong variable type for +=')
Ernie Raele75fde62023-12-21 17:18:54 +01003208 assert_fails('A += N', 'E1405: Class "A" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003209 assert_fails('o += N', 'E734: Wrong variable type for +=')
Ernie Raele75fde62023-12-21 17:18:54 +01003210 assert_fails('T += N', 'E1403: Type alias "T" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003211
3212 # Assign to a number variable
3213 assert_fails('N = d', 'E1012: Type mismatch; expected number but got dict<number>')
3214 assert_fails('N = l', 'E1012: Type mismatch; expected number but got list<number>')
3215 assert_fails('N = b', 'E1012: Type mismatch; expected number but got blob')
3216 assert_fails('N = Fn', 'E1012: Type mismatch; expected number but got func([unknown]): number')
Ernie Rael2025af12023-12-12 16:58:00 +01003217 assert_fails('N = A', 'E1405: Class "A" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003218 assert_fails('N = o', 'E1012: Type mismatch; expected number but got object<A>')
Ernie Raelb5011082023-12-18 08:34:41 +01003219 assert_fails('N = T', 'E1403: Type alias "T" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003220
3221 # Use a compound operator with different RHS types
3222 assert_fails('N += d', 'E734: Wrong variable type for +=')
3223 assert_fails('N += l', 'E734: Wrong variable type for +=')
3224 assert_fails('N += b', 'E974: Using a Blob as a Number')
3225 assert_fails('N += Fn', 'E734: Wrong variable type for +=')
Ernie Rael2025af12023-12-12 16:58:00 +01003226 assert_fails('N += A', 'E1405: Class "A" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003227 assert_fails('N += o', 'E1320: Using an Object as a Number')
Ernie Raelb5011082023-12-18 08:34:41 +01003228 assert_fails('N += T', 'E1403: Type alias "T" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003229
3230 # Initialize multiple variables using []
3231 assert_fails('var [X1: number, Y: number] = [1, d]', 'E1012: Type mismatch; expected number but got dict<number>')
3232 assert_fails('var [X2: number, Y: number] = [1, l]', 'E1012: Type mismatch; expected number but got list<number>')
3233 assert_fails('var [X3: number, Y: number] = [1, b]', 'E1012: Type mismatch; expected number but got blob')
3234 assert_fails('var [X4: number, Y: number] = [1, Fn]', 'E1012: Type mismatch; expected number but got func([unknown]): number')
Ernie Rael2025af12023-12-12 16:58:00 +01003235 assert_fails('var [X7: number, Y: number] = [1, A]', 'E1405: Class "A" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003236 assert_fails('var [X8: number, Y: number] = [1, o]', 'E1012: Type mismatch; expected number but got object<A>')
Ernie Raelb5011082023-12-18 08:34:41 +01003237 assert_fails('var [X8: number, Y: number] = [1, T]', 'E1403: Type alias "T" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003238
3239 # String concatenation with various LHS types
3240 assert_fails('S ..= d', 'E734: Wrong variable type for .=')
3241 assert_fails('S ..= l', 'E734: Wrong variable type for .=')
3242 assert_fails('S ..= b', 'E976: Using a Blob as a String')
3243 assert_fails('S ..= Fn', 'E734: Wrong variable type for .=')
Ernie Rael2025af12023-12-12 16:58:00 +01003244 assert_fails('S ..= A', 'E1405: Class "A" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003245 assert_fails('S ..= o', 'E1324: Using an Object as a String')
Ernie Raelb5011082023-12-18 08:34:41 +01003246 assert_fails('S ..= T', 'E1403: Type alias "T" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003247
3248 # String concatenation with various RHS types
3249 assert_fails('d ..= S', 'E734: Wrong variable type for .=')
3250 assert_fails('l ..= S', 'E734: Wrong variable type for .=')
3251 assert_fails('b ..= S', 'E734: Wrong variable type for .=')
3252 assert_fails('Fn ..= S', 'E734: Wrong variable type for .=')
Ernie Raele75fde62023-12-21 17:18:54 +01003253 assert_fails('A ..= S', 'E1405: Class "A" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003254 assert_fails('o ..= S', 'E734: Wrong variable type for .=')
Ernie Raele75fde62023-12-21 17:18:54 +01003255 assert_fails('T ..= S', 'E1403: Type alias "T" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003256 END
3257 v9.CheckSourceSuccess(lines)
Yegappan Lakshmanan3ee25962023-12-04 20:31:14 +01003258
3259 if has('channel')
3260 lines =<< trim END
3261 vim9script
3262 var N: number = 1
3263 var S: string = 'abc'
3264 var j: job = test_null_job()
3265 var ch: channel = test_null_channel()
3266 assert_fails('j = N', 'E1012: Type mismatch; expected job but got number')
3267 assert_fails('ch = N', 'E1012: Type mismatch; expected channel but got number')
3268 assert_fails('j += N', 'E734: Wrong variable type for +=')
3269 assert_fails('ch += N', 'E734: Wrong variable type for +=')
3270 assert_fails('N = j', 'E1012: Type mismatch; expected number but got job')
3271 assert_fails('N = ch', 'E1012: Type mismatch; expected number but got channel')
3272 assert_fails('N += j', 'E910: Using a Job as a Number')
3273 assert_fails('N += ch', 'E913: Using a Channel as a Number')
3274 assert_fails('var [X5: number, Y: number] = [1, j]', 'E1012: Type mismatch; expected number but got job')
3275 assert_fails('var [X6: number, Y: number] = [1, ch]', 'E1012: Type mismatch; expected number but got channel')
3276 assert_fails('S ..= j', 'E908: Using an invalid value as a String: job')
3277 assert_fails('S ..= ch', 'E908: Using an invalid value as a String: channel')
3278 assert_fails('j ..= S', 'E734: Wrong variable type for .=')
3279 assert_fails('ch ..= S', 'E734: Wrong variable type for .=')
3280 END
3281 v9.CheckSourceSuccess(lines)
3282 endif
Ernie Rael5e133152023-12-19 22:15:27 +01003283
3284 lines =<< trim END
3285 vim9script
3286 class A
3287 endclass
3288
3289 def F()
3290 A += 3
3291 enddef
3292 F()
3293 END
3294 v9.CheckScriptFailure(lines, 'E1405: Class "A" cannot be used as a value')
3295
3296 lines =<< trim END
3297 vim9script
3298 class A
3299 endclass
3300
3301 var o = A.new()
3302 def F()
3303 o += 4
3304 enddef
3305 F()
3306 END
3307 v9.CheckScriptFailure(lines, 'E1411: Missing dot after object "o"')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003308enddef
3309
3310" Test for checking the argument type of a def function
3311def Test_func_argtype_check()
3312 var lines =<< trim END
3313 vim9script
3314
3315 # Passing different types as argument to a function expecting a number
3316 def IntArg(n: number)
3317 enddef
3318
3319 class A
3320 endclass
3321 var N: number = 1
3322 var S: string = 'abc'
3323 var d: dict<number> = {}
3324 var l: list<number> = []
3325 var b: blob = 0z10
3326 var Fn: func = function('min')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003327 var o: A = A.new()
3328
3329 assert_fails('IntArg(d)', 'E1013: Argument 1: type mismatch, expected number but got dict<number>')
3330 assert_fails('IntArg(l)', 'E1013: Argument 1: type mismatch, expected number but got list<number>')
3331 assert_fails('IntArg(b)', 'E1013: Argument 1: type mismatch, expected number but got blob')
3332 assert_fails('IntArg(Fn)', 'E1013: Argument 1: type mismatch, expected number but got func([unknown]): number')
Yegappan Lakshmanan3ee25962023-12-04 20:31:14 +01003333 if has('channel')
3334 var j: job = test_null_job()
3335 var ch: channel = test_null_channel()
3336 assert_fails('IntArg(j)', 'E1013: Argument 1: type mismatch, expected number but got job')
3337 assert_fails('IntArg(ch)', 'E1013: Argument 1: type mismatch, expected number but got channel')
3338 endif
Ernie Raelb077b582023-12-14 20:11:44 +01003339 assert_fails('IntArg(A)', 'E1405: Class "A" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003340 assert_fails('IntArg(o)', 'E1013: Argument 1: type mismatch, expected number but got object<A>')
3341
3342 # Passing a number to functions accepting different argument types
3343 def DictArg(_: dict<number>)
3344 enddef
3345 assert_fails('DictArg(N)', 'E1013: Argument 1: type mismatch, expected dict<number> but got number')
3346
3347 def ListArg(_: list<number>)
3348 enddef
3349 assert_fails('ListArg(N)', 'E1013: Argument 1: type mismatch, expected list<number> but got number')
3350
3351 def BlobArg(_: blob)
3352 enddef
3353 assert_fails('BlobArg(N)', 'E1013: Argument 1: type mismatch, expected blob but got number')
3354
3355 def FuncArg(Fn_arg: func)
3356 enddef
3357 assert_fails('FuncArg(N)', 'E1013: Argument 1: type mismatch, expected func(...): unknown but got number')
3358
Yegappan Lakshmanan3ee25962023-12-04 20:31:14 +01003359 if has('channel')
3360 def JobArg(_: job)
3361 enddef
3362 assert_fails('JobArg(N)', 'E1013: Argument 1: type mismatch, expected job but got number')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003363
Yegappan Lakshmanan3ee25962023-12-04 20:31:14 +01003364 def ChannelArg(_: channel)
3365 enddef
3366 assert_fails('ChannelArg(N)', 'E1013: Argument 1: type mismatch, expected channel but got number')
3367 endif
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003368
3369 def ObjectArg(_: A)
3370 enddef
3371 assert_fails('ObjectArg(N)', 'E1013: Argument 1: type mismatch, expected object<A> but got number')
3372 END
3373 v9.CheckSourceSuccess(lines)
3374
3375 # Calling a function expecting a number type with different argument types
3376 # from another function
3377 var pre_lines =<< trim END
3378 vim9script
3379 class A
3380 endclass
3381 def IntArg(n: number)
3382 enddef
3383 def Foo()
3384 END
3385 var post_lines =<< trim END
3386 enddef
3387 defcompile
3388 END
3389 lines = pre_lines + ['var d: dict<number> = {}', 'IntArg(d)'] + post_lines
3390 v9.CheckSourceFailure(lines, 'E1013: Argument 1: type mismatch, expected number but got dict<number>', 2)
3391 lines = pre_lines + ['var l: list<number> = []', 'IntArg(l)'] + post_lines
3392 v9.CheckSourceFailure(lines, 'E1013: Argument 1: type mismatch, expected number but got list<number>', 2)
3393 lines = pre_lines + ['var b: blob = 0z12', 'IntArg(b)'] + post_lines
3394 v9.CheckSourceFailure(lines, 'E1013: Argument 1: type mismatch, expected number but got blob', 2)
3395 lines = pre_lines + ['var Fn: func = function("min")', 'IntArg(Fn)'] + post_lines
3396 v9.CheckSourceFailure(lines, 'E1013: Argument 1: type mismatch, expected number but got func(...): unknown', 2)
Yegappan Lakshmanan3ee25962023-12-04 20:31:14 +01003397 if has('channel')
3398 lines = pre_lines + ['var j: job = test_null_job()', 'IntArg(j)'] + post_lines
3399 v9.CheckSourceFailure(lines, 'E1013: Argument 1: type mismatch, expected number but got job', 2)
3400 lines = pre_lines + ['var ch: channel = test_null_channel()', 'IntArg(ch)'] + post_lines
3401 v9.CheckSourceFailure(lines, 'E1013: Argument 1: type mismatch, expected number but got channel', 2)
3402 endif
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003403 lines = pre_lines + ['IntArg(A)'] + post_lines
Ernie Raelb077b582023-12-14 20:11:44 +01003404 v9.CheckSourceFailure(lines, 'E1405: Class "A" cannot be used as a value', 1)
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003405 lines = pre_lines + ['var o: A = A.new()', 'IntArg(o)'] + post_lines
3406 v9.CheckSourceFailure(lines, 'E1013: Argument 1: type mismatch, expected number but got object<A>', 2)
3407enddef
3408
3409" Test for checking the return type of a def function
3410def Test_func_rettype_check()
3411 var lines =<< trim END
3412 vim9script
3413 def Fn(): dict<number>
3414 return 10
3415 enddef
3416 defcompile
3417 END
3418 v9.CheckSourceFailure(lines, 'E1012: Type mismatch; expected dict<number> but got number', 1)
3419
3420 lines =<< trim END
3421 vim9script
3422 def Fn(): list<number>
3423 return 10
3424 enddef
3425 defcompile
3426 END
3427 v9.CheckSourceFailure(lines, 'E1012: Type mismatch; expected list<number> but got number', 1)
3428
3429 lines =<< trim END
3430 vim9script
3431 def Fn(): blob
3432 return 10
3433 enddef
3434 defcompile
3435 END
3436 v9.CheckSourceFailure(lines, 'E1012: Type mismatch; expected blob but got number', 1)
3437
3438 lines =<< trim END
3439 vim9script
3440 def Fn(): func
3441 return 10
3442 enddef
3443 defcompile
3444 END
3445 v9.CheckSourceFailure(lines, 'E1012: Type mismatch; expected func(...): unknown but got number', 1)
3446
3447 lines =<< trim END
3448 vim9script
3449 def Fn(): job
3450 return 10
3451 enddef
3452 defcompile
3453 END
3454 v9.CheckSourceFailure(lines, 'E1012: Type mismatch; expected job but got number', 1)
3455
3456 lines =<< trim END
3457 vim9script
3458 def Fn(): channel
3459 return 10
3460 enddef
3461 defcompile
3462 END
3463 v9.CheckSourceFailure(lines, 'E1012: Type mismatch; expected channel but got number', 1)
3464
3465 lines =<< trim END
3466 vim9script
3467 class A
3468 endclass
3469 def Fn(): A
3470 return 10
3471 enddef
3472 defcompile
3473 END
3474 v9.CheckSourceFailure(lines, 'E1012: Type mismatch; expected object<A> but got number', 1)
3475enddef
3476
Yegappan Lakshmanand3351852023-12-08 21:41:23 +01003477" Test for assigning different types of value to a variable of type "any"
3478def Test_assign_to_any()
3479 for [typestr, val] in [
3480 ["'bool'", 'true'],
3481 ["'number'", '100'],
3482 ["'float'", '1.1'],
3483 ["'string'", '"abc"'],
3484 ["'blob'", '0z10'],
3485 ["'list<number>'", '[1, 2, 3]'],
3486 ["'dict<number>'", '{a: 1}'],
3487 ]
3488 var lines =<< trim eval END
3489 vim9script
3490 var x: any = {val}
3491 assert_equal({typestr}, typename(x))
3492 x = [{{a: 1}}, {{b: 2}}]
3493 assert_equal('list<dict<number>>', typename(x))
3494 def Foo(xarg: any, s: string)
3495 assert_equal(s, typename(xarg))
3496 enddef
3497 Foo({val}, {typestr})
3498 END
3499 v9.CheckSourceSuccess(lines)
3500 endfor
3501enddef
3502
Ernie Raelfa831102023-12-14 20:06:39 +01003503def Test_assign_type_to_list_dict()
3504 var lines =<< trim END
3505 vim9script
3506 class C
3507 endclass
3508
3509 var x = [C]
3510 END
3511 v9.CheckScriptFailure(lines, 'E1405: Class "C" cannot be used as a value')
3512
3513 lines =<< trim END
3514 vim9script
3515 class C
3516 endclass
3517 type T = C
3518
3519 def F()
3520 var x = [3, T, C]
3521 enddef
3522 F()
3523 END
3524 v9.CheckScriptFailure(lines, 'E1405: Class "C" cannot be used as a value')
3525
3526 lines =<< trim END
3527 vim9script
3528 type T = number
3529
3530 def F()
3531 var x = [3, T]
3532 enddef
3533 F()
3534 END
3535 v9.CheckScriptFailure(lines, 'E1407: Cannot use a Typealias as a variable or value')
3536
3537 lines =<< trim END
3538 vim9script
3539 class C
3540 endclass
3541
3542 var x = {e: C}
3543 END
3544 v9.CheckScriptFailure(lines, 'E1405: Class "C" cannot be used as a value')
3545
3546 lines =<< trim END
3547 vim9script
3548 class C
3549 endclass
3550
3551 def F()
3552 var x = {e: C}
3553 enddef
3554 F()
3555 END
3556 v9.CheckScriptFailure(lines, 'E1405: Class "C" cannot be used as a value')
3557
3558 lines =<< trim END
3559 vim9script
3560 type T = number
3561
3562 def F()
3563 var x = {e: T}
3564 enddef
3565 F()
3566 END
3567 v9.CheckScriptFailure(lines, 'E1407: Cannot use a Typealias as a variable or value')
3568
3569 lines =<< trim END
3570 vim9script
3571 class C
3572 endclass
3573
3574 def F()
3575 var x = {e: [C]}
3576 enddef
3577 F()
3578 END
3579 v9.CheckScriptFailure(lines, 'E1405: Class "C" cannot be used as a value')
3580
3581 lines =<< trim END
3582 vim9script
3583 type T = number
3584
3585 def F()
3586 var x = {e: [T]}
3587 enddef
3588 F()
3589 END
3590 v9.CheckScriptFailure(lines, 'E1407: Cannot use a Typealias as a variable or value')
3591enddef
3592
Yegappan Lakshmanan1af35632024-02-06 11:03:36 +01003593" Test for modifying a final variable using a compound operator
3594def Test_final_var_modification_with_compound_op()
3595 var lines =<< trim END
3596 vim9script
3597
3598 final i: number = 1000
3599 assert_fails('i += 2', 'E46: Cannot change read-only variable "i"')
3600 assert_fails('i -= 2', 'E46: Cannot change read-only variable "i"')
3601 assert_fails('i *= 2', 'E46: Cannot change read-only variable "i"')
3602 assert_fails('i /= 2', 'E46: Cannot change read-only variable "i"')
3603 assert_fails('i %= 2', 'E46: Cannot change read-only variable "i"')
3604 assert_equal(1000, i)
3605
3606 final f: float = 1000.0
3607 assert_fails('f += 2', 'E46: Cannot change read-only variable "f"')
3608 assert_fails('f -= 2', 'E46: Cannot change read-only variable "f"')
3609 assert_fails('f *= 2', 'E46: Cannot change read-only variable "f"')
3610 assert_fails('f /= 2', 'E46: Cannot change read-only variable "f"')
3611 assert_equal(1000.0, f)
3612
3613 final s: string = 'abc'
3614 assert_fails('s ..= "y"', 'E46: Cannot change read-only variable "s"')
3615 assert_equal('abc', s)
3616 END
3617 v9.CheckScriptSuccess(lines)
3618enddef
3619
3620" Test for modifying a final variable with a List value
3621def Test_final_var_with_list_value()
3622 var lines =<< trim END
3623 vim9script
3624
3625 final listA: list<string> = []
3626 var listB = listA
3627
3628 listB->add('a')
3629 assert_true(listA is listB)
3630 assert_equal(['a'], listA)
3631 assert_equal(['a'], listB)
3632
3633 listB += ['b']
3634 assert_true(listA is listB)
3635 assert_equal(['a', 'b'], listA)
3636 assert_equal(['a', 'b'], listB)
3637
3638 listA->add('c')
3639 assert_true(listA is listB)
3640 assert_equal(['a', 'b', 'c'], listA)
3641 assert_equal(['a', 'b', 'c'], listB)
3642
3643 listA += ['d']
3644 assert_true(listA is listB)
3645 assert_equal(['a', 'b', 'c', 'd'], listA)
3646 assert_equal(['a', 'b', 'c', 'd'], listB)
3647 END
3648 v9.CheckScriptSuccess(lines)
3649enddef
3650
3651" Test for modifying a final variable with a List value using "+=" from a legacy
3652" function.
3653func Test_final_var_with_list_value_legacy()
3654 vim9cmd final g:TestVar = ['a']
3655 vim9cmd g:TestVar += ['b']
3656 call assert_equal(['a', 'b'], g:TestVar)
3657endfunc
3658
3659" Test for modifying a final variable with a Blob value
3660def Test_final_var_with_blob_value()
3661 var lines =<< trim END
3662 vim9script
3663
3664 final blobA: blob = 0z10
3665 var blobB = blobA
3666
3667 blobB->add(32)
3668 assert_true(blobA is blobB)
3669 assert_equal(0z1020, blobA)
3670 assert_equal(0z1020, blobB)
3671
3672 blobB += 0z30
3673 assert_true(blobA is blobB)
3674 assert_equal(0z102030, blobA)
3675 assert_equal(0z102030, blobB)
3676
3677 blobA->add(64)
3678 assert_true(blobA is blobB)
3679 assert_equal(0z10203040, blobA)
3680 assert_equal(0z10203040, blobB)
3681
3682 blobA += 0z50
3683 assert_true(blobA is blobB)
3684 assert_equal(0z1020304050, blobA)
3685 assert_equal(0z1020304050, blobB)
3686 END
3687 v9.CheckScriptSuccess(lines)
3688enddef
3689
Bram Moolenaar971caf52020-09-21 22:21:42 +02003690" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker