blob: 4414f55c067fb1389a7c70b65c8626f2f2c408b1 [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
653 v9.CheckDefExecAndScriptFailure(lines, 'E1030: Using a String as a Number: "22"')
654
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')
Bram Moolenaarb7480cd2021-07-04 18:28:15 +02001107enddef
1108
Bram Moolenaarf30a14d2021-01-17 21:51:24 +01001109def Test_assignment_list_any_index()
1110 var l: list<number> = [1, 2]
1111 for [x, y, _]
1112 in [[0, 1, ''], [1, 3, '']]
1113 l[x] = l[x] + y
1114 endfor
1115 assert_equal([2, 5], l)
1116enddef
1117
Bram Moolenaar971caf52020-09-21 22:21:42 +02001118def Test_assignment_list_vim9script()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001119 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +02001120 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001121 var v1: number
1122 var v2: number
1123 var v3: number
Bram Moolenaar971caf52020-09-21 22:21:42 +02001124 [v1, v2, v3] = [1, 2, 3]
1125 assert_equal([1, 2, 3], [v1, v2, v3])
1126 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001127 v9.CheckScriptSuccess(lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001128enddef
1129
1130def Test_assignment_dict()
Bram Moolenaare0de1712020-12-02 17:36:54 +01001131 var dict1: dict<bool> = {one: false, two: true}
1132 var dict2: dict<number> = {one: 1, two: 2}
1133 var dict3: dict<string> = {key: 'value'}
1134 var dict4: dict<any> = {one: 1, two: '2'}
1135 var dict5: dict<blob> = {one: 0z01, two: 0z02}
Bram Moolenaar971caf52020-09-21 22:21:42 +02001136
Bram Moolenaar31d99482022-05-26 22:24:43 +01001137 # check the type is OK
1138 var events: dict<string> = v:event
1139
Bram Moolenaar971caf52020-09-21 22:21:42 +02001140 # overwrite
1141 dict3['key'] = 'another'
Bram Moolenaare0de1712020-12-02 17:36:54 +01001142 assert_equal(dict3, {key: 'another'})
Bram Moolenaarfc74d032020-11-16 22:11:49 +01001143 dict3.key = 'yet another'
Bram Moolenaare0de1712020-12-02 17:36:54 +01001144 assert_equal(dict3, {key: 'yet another'})
Bram Moolenaarfc74d032020-11-16 22:11:49 +01001145
Bram Moolenaard24602f2020-12-20 15:20:56 +01001146 # member "any" can also be a dict and assigned to
1147 var anydict: dict<any> = {nest: {}, nr: 0}
1148 anydict.nest['this'] = 123
1149 anydict.nest.that = 456
1150 assert_equal({nest: {this: 123, that: 456}, nr: 0}, anydict)
1151
Bram Moolenaarfc74d032020-11-16 22:11:49 +01001152 var lines =<< trim END
Bram Moolenaard88dc4d2020-12-20 14:49:42 +01001153 var dd = {}
1154 dd.two = 2
1155 assert_equal({two: 2}, dd)
1156 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001157 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar4f5e3972020-12-21 17:30:50 +01001158
1159 lines =<< trim END
1160 var d = {dd: {}}
1161 d.dd[0] = 2
1162 d.dd['x'] = 3
1163 d.dd.y = 4
1164 assert_equal({dd: {0: 2, x: 3, y: 4}}, d)
1165 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001166 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaard88dc4d2020-12-20 14:49:42 +01001167
1168 lines =<< trim END
Bram Moolenaar2e17fef2022-03-18 19:44:48 +00001169 var key = 'foo'
1170 g:[key] = 'value'
1171 assert_equal('value', g:foo)
1172 unlet g:foo
1173 END
1174 v9.CheckDefAndScriptSuccess(lines)
1175
1176 lines =<< trim END
Bram Moolenaare0de1712020-12-02 17:36:54 +01001177 var dd = {one: 1}
Bram Moolenaarfc74d032020-11-16 22:11:49 +01001178 dd.one) = 2
1179 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001180 v9.CheckDefFailure(lines, 'E488:', 2)
Bram Moolenaard88dc4d2020-12-20 14:49:42 +01001181
1182 lines =<< trim END
1183 var dd = {one: 1}
1184 var dd.one = 2
1185 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001186 v9.CheckDefAndScriptFailure(lines, 'E1017:', 2)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001187
1188 # empty key can be used
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001189 var dd = {}
Bram Moolenaar971caf52020-09-21 22:21:42 +02001190 dd[""] = 6
Bram Moolenaare0de1712020-12-02 17:36:54 +01001191 assert_equal({['']: 6}, dd)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001192
1193 # type becomes dict<any>
Bram Moolenaare0de1712020-12-02 17:36:54 +01001194 var somedict = rand() > 0 ? {a: 1, b: 2} : {a: 'a', b: 'b'}
Bram Moolenaar971caf52020-09-21 22:21:42 +02001195
Bram Moolenaar114dbda2022-01-03 12:28:03 +00001196 # type is dict<any> even though initializer is dict<number>
1197 var anyDict: dict<any> = {a: 0}
1198 assert_equal({a: 0, b: 'x'}, extend(anyDict, {b: 'x'}))
1199
Bram Moolenaar078a4612022-01-04 15:17:03 +00001200 # using global var, which has no declared type
1201 g:myDict = {}
1202 g:myDict->extend({a: 1})
1203 g:myDict->extend({b: 'x'})
1204 assert_equal({a: 1, b: 'x'}, g:myDict)
1205 unlet g:myDict
1206
1207 # using list with declared type gives an error
1208 lines =<< trim END
1209 var d: dict<number>
1210 g:myDict = d
1211 g:myDict->extend({a: 1})
1212 g:myDict->extend({b: 'x'})
1213 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001214 v9.CheckDefExecAndScriptFailure(lines, 'E1013: Argument 2: type mismatch, expected dict<number> but got dict<string>', 4)
Bram Moolenaar078a4612022-01-04 15:17:03 +00001215 unlet g:myDict
1216
Bram Moolenaar971caf52020-09-21 22:21:42 +02001217 # assignment to script-local dict
Bram Moolenaarfc74d032020-11-16 22:11:49 +01001218 lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +02001219 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001220 var test: dict<any> = {}
Bram Moolenaar971caf52020-09-21 22:21:42 +02001221 def FillDict(): dict<any>
1222 test['a'] = 43
1223 return test
1224 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +01001225 assert_equal({a: 43}, FillDict())
Bram Moolenaar971caf52020-09-21 22:21:42 +02001226 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001227 v9.CheckScriptSuccess(lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001228
1229 lines =<< trim END
1230 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001231 var test: dict<any>
Bram Moolenaar971caf52020-09-21 22:21:42 +02001232 def FillDict(): dict<any>
1233 test['a'] = 43
1234 return test
1235 enddef
1236 FillDict()
Bram Moolenaar3beaf9c2020-12-18 17:23:14 +01001237 assert_equal({a: 43}, test)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001238 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001239 v9.CheckScriptSuccess(lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001240
1241 # assignment to global dict
1242 lines =<< trim END
1243 vim9script
1244 g:test = {}
1245 def FillDict(): dict<any>
1246 g:test['a'] = 43
1247 return g:test
1248 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +01001249 assert_equal({a: 43}, FillDict())
Bram Moolenaar971caf52020-09-21 22:21:42 +02001250 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001251 v9.CheckScriptSuccess(lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001252
1253 # assignment to buffer dict
1254 lines =<< trim END
1255 vim9script
1256 b:test = {}
1257 def FillDict(): dict<any>
1258 b:test['a'] = 43
1259 return b:test
1260 enddef
Bram Moolenaare0de1712020-12-02 17:36:54 +01001261 assert_equal({a: 43}, FillDict())
Bram Moolenaar971caf52020-09-21 22:21:42 +02001262 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001263 v9.CheckScriptSuccess(lines)
Bram Moolenaar4f5e3972020-12-21 17:30:50 +01001264
1265 lines =<< trim END
1266 var d = {dd: test_null_dict()}
1267 d.dd[0] = 0
1268 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001269 v9.CheckDefExecFailure(lines, 'E1103:', 2)
Bram Moolenaar4f5e3972020-12-21 17:30:50 +01001270
1271 lines =<< trim END
1272 var d = {dd: 'string'}
1273 d.dd[0] = 0
1274 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001275 v9.CheckDefExecFailure(lines, 'E1148:', 2)
Bram Moolenaar3a3b10e2021-06-26 15:00:59 +02001276
1277 lines =<< trim END
1278 var n: any
1279 n.key = 5
1280 END
Ernie Raelf8da3242023-10-11 21:22:12 +02001281 v9.CheckDefExecAndScriptFailure(lines, ['E1148:', 'E1203: Dot not allowed after a number: n.key = 5'], 2)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001282enddef
1283
1284def Test_assignment_local()
1285 # Test in a separated file in order not to the current buffer/window/tab is
1286 # changed.
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001287 var script_lines: list<string> =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +02001288 let b:existing = 'yes'
1289 let w:existing = 'yes'
1290 let t:existing = 'yes'
1291
1292 def Test_assignment_local_internal()
1293 b:newvar = 'new'
1294 assert_equal('new', b:newvar)
1295 assert_equal('yes', b:existing)
1296 b:existing = 'no'
1297 assert_equal('no', b:existing)
1298 b:existing ..= 'NO'
1299 assert_equal('noNO', b:existing)
1300
1301 w:newvar = 'new'
1302 assert_equal('new', w:newvar)
1303 assert_equal('yes', w:existing)
1304 w:existing = 'no'
1305 assert_equal('no', w:existing)
1306 w:existing ..= 'NO'
1307 assert_equal('noNO', w:existing)
1308
1309 t:newvar = 'new'
1310 assert_equal('new', t:newvar)
1311 assert_equal('yes', t:existing)
1312 t:existing = 'no'
1313 assert_equal('no', t:existing)
1314 t:existing ..= 'NO'
1315 assert_equal('noNO', t:existing)
1316 enddef
1317 call Test_assignment_local_internal()
1318 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001319 v9.CheckScriptSuccess(script_lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001320enddef
1321
1322def Test_assignment_default()
Bram Moolenaar971caf52020-09-21 22:21:42 +02001323 # Test default values.
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001324 var thebool: bool
Bram Moolenaar971caf52020-09-21 22:21:42 +02001325 assert_equal(v:false, thebool)
1326
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001327 var thenumber: number
Bram Moolenaar971caf52020-09-21 22:21:42 +02001328 assert_equal(0, thenumber)
1329
Bram Moolenaar73e28dc2022-09-17 21:08:33 +01001330 var thefloat: float
1331 assert_equal(0.0, thefloat)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001332
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001333 var thestring: string
Bram Moolenaar971caf52020-09-21 22:21:42 +02001334 assert_equal('', thestring)
1335
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001336 var theblob: blob
Bram Moolenaar971caf52020-09-21 22:21:42 +02001337 assert_equal(0z, theblob)
1338
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001339 var Thefunc: func
Bram Moolenaar971caf52020-09-21 22:21:42 +02001340 assert_equal(test_null_function(), Thefunc)
1341
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001342 var thelist: list<any>
Bram Moolenaar971caf52020-09-21 22:21:42 +02001343 assert_equal([], thelist)
1344
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001345 var thedict: dict<any>
Bram Moolenaar971caf52020-09-21 22:21:42 +02001346 assert_equal({}, thedict)
1347
1348 if has('channel')
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001349 var thejob: job
Bram Moolenaar971caf52020-09-21 22:21:42 +02001350 assert_equal(test_null_job(), thejob)
1351
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001352 var thechannel: channel
Bram Moolenaar971caf52020-09-21 22:21:42 +02001353 assert_equal(test_null_channel(), thechannel)
1354
1355 if has('unix') && executable('cat')
1356 # check with non-null job and channel, types must match
Bram Moolenaare0de1712020-12-02 17:36:54 +01001357 thejob = job_start("cat ", {})
Bram Moolenaar971caf52020-09-21 22:21:42 +02001358 thechannel = job_getchannel(thejob)
1359 job_stop(thejob, 'kill')
1360 endif
1361 endif
1362
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001363 var nr = 1234 | nr = 5678
Bram Moolenaar971caf52020-09-21 22:21:42 +02001364 assert_equal(5678, nr)
1365enddef
1366
Bram Moolenaare08be092022-02-17 13:08:26 +00001367def Test_script_var_default()
1368 var lines =<< trim END
1369 vim9script
1370 var l: list<number>
Bram Moolenaar1983f1a2022-02-28 20:55:02 +00001371 var li = [1, 2]
Bram Moolenaare08be092022-02-17 13:08:26 +00001372 var bl: blob
Bram Moolenaar1983f1a2022-02-28 20:55:02 +00001373 var bli = 0z12
Bram Moolenaare08be092022-02-17 13:08:26 +00001374 var d: dict<number>
Bram Moolenaar1983f1a2022-02-28 20:55:02 +00001375 var di = {'a': 1, 'b': 2}
Bram Moolenaare08be092022-02-17 13:08:26 +00001376 def Echo()
1377 assert_equal([], l)
Bram Moolenaar1983f1a2022-02-28 20:55:02 +00001378 assert_equal([1, 2], li)
Bram Moolenaare08be092022-02-17 13:08:26 +00001379 assert_equal(0z, bl)
Bram Moolenaar1983f1a2022-02-28 20:55:02 +00001380 assert_equal(0z12, bli)
Bram Moolenaare08be092022-02-17 13:08:26 +00001381 assert_equal({}, d)
Bram Moolenaar1983f1a2022-02-28 20:55:02 +00001382 assert_equal({'a': 1, 'b': 2}, di)
Bram Moolenaare08be092022-02-17 13:08:26 +00001383 enddef
Bram Moolenaarea5c8982022-02-17 14:42:02 +00001384 Echo()
Bram Moolenaare08be092022-02-17 13:08:26 +00001385 END
1386 v9.CheckScriptSuccess(lines)
1387enddef
1388
Bram Moolenaara749a422022-02-12 19:52:25 +00001389let s:scriptvar = 'init'
Bram Moolenaar3862ea32021-01-01 21:05:55 +01001390
Bram Moolenaar971caf52020-09-21 22:21:42 +02001391def Test_assignment_var_list()
Bram Moolenaarda7c20c2020-11-30 21:12:19 +01001392 var lines =<< trim END
1393 var v1: string
1394 var v2: string
1395 var vrem: list<string>
1396 [v1] = ['aaa']
1397 assert_equal('aaa', v1)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001398
Bram Moolenaarda7c20c2020-11-30 21:12:19 +01001399 [v1, v2] = ['one', 'two']
1400 assert_equal('one', v1)
1401 assert_equal('two', v2)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001402
Bram Moolenaarda7c20c2020-11-30 21:12:19 +01001403 [v1, v2; vrem] = ['one', 'two']
1404 assert_equal('one', v1)
1405 assert_equal('two', v2)
1406 assert_equal([], vrem)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001407
Bram Moolenaarda7c20c2020-11-30 21:12:19 +01001408 [v1, v2; vrem] = ['one', 'two', 'three']
1409 assert_equal('one', v1)
1410 assert_equal('two', v2)
1411 assert_equal(['three'], vrem)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001412
Bram Moolenaarda7c20c2020-11-30 21:12:19 +01001413 [&ts, &sw] = [3, 4]
1414 assert_equal(3, &ts)
1415 assert_equal(4, &sw)
1416 set ts=8 sw=4
Bram Moolenaar004d9b02020-11-30 21:40:03 +01001417
1418 [@a, @z] = ['aa', 'zz']
1419 assert_equal('aa', @a)
1420 assert_equal('zz', @z)
1421
1422 [$SOME_VAR, $OTHER_VAR] = ['some', 'other']
1423 assert_equal('some', $SOME_VAR)
1424 assert_equal('other', $OTHER_VAR)
1425
Bram Moolenaar3862ea32021-01-01 21:05:55 +01001426 [g:globalvar, b:bufvar, w:winvar, t:tabvar, v:errmsg] =
1427 ['global', 'buf', 'win', 'tab', 'error']
Bram Moolenaar004d9b02020-11-30 21:40:03 +01001428 assert_equal('global', g:globalvar)
Bram Moolenaar004d9b02020-11-30 21:40:03 +01001429 assert_equal('buf', b:bufvar)
1430 assert_equal('win', w:winvar)
1431 assert_equal('tab', t:tabvar)
1432 assert_equal('error', v:errmsg)
1433 unlet g:globalvar
Bram Moolenaarda7c20c2020-11-30 21:12:19 +01001434 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001435 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar3862ea32021-01-01 21:05:55 +01001436
Bram Moolenaara749a422022-02-12 19:52:25 +00001437 [g:globalvar, scriptvar, b:bufvar] = ['global', 'script', 'buf']
Bram Moolenaar3862ea32021-01-01 21:05:55 +01001438 assert_equal('global', g:globalvar)
Bram Moolenaara749a422022-02-12 19:52:25 +00001439 assert_equal('script', scriptvar)
Bram Moolenaar3862ea32021-01-01 21:05:55 +01001440 assert_equal('buf', b:bufvar)
1441
1442 lines =<< trim END
1443 vim9script
Bram Moolenaara749a422022-02-12 19:52:25 +00001444 var scriptvar = 'init'
1445 [g:globalvar, scriptvar, w:winvar] = ['global', 'script', 'win']
Bram Moolenaar3862ea32021-01-01 21:05:55 +01001446 assert_equal('global', g:globalvar)
Bram Moolenaara749a422022-02-12 19:52:25 +00001447 assert_equal('script', scriptvar)
Bram Moolenaar3862ea32021-01-01 21:05:55 +01001448 assert_equal('win', w:winvar)
1449 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001450 v9.CheckScriptSuccess(lines)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001451enddef
1452
Bram Moolenaar2d3ac2e2022-02-03 12:34:05 +00001453def Test_assignment_empty_list()
1454 var lines =<< trim END
1455 var l2: list<any> = []
1456 var l: list<string>
1457 l = l2
1458 END
1459 v9.CheckDefAndScriptSuccess(lines)
1460enddef
1461
Bram Moolenaar971caf52020-09-21 22:21:42 +02001462def Test_assignment_vim9script()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001463 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +02001464 vim9script
1465 def Func(): list<number>
1466 return [1, 2]
1467 enddef
Bram Moolenaar98309382020-09-27 21:58:45 +02001468 var name1: number
1469 var name2: number
1470 [name1, name2] =
Bram Moolenaar971caf52020-09-21 22:21:42 +02001471 Func()
Bram Moolenaar98309382020-09-27 21:58:45 +02001472 assert_equal(1, name1)
1473 assert_equal(2, name2)
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001474 var ll =
Bram Moolenaar971caf52020-09-21 22:21:42 +02001475 Func()
1476 assert_equal([1, 2], ll)
1477
1478 @/ = 'text'
1479 assert_equal('text', @/)
1480 @0 = 'zero'
1481 assert_equal('zero', @0)
1482 @1 = 'one'
1483 assert_equal('one', @1)
1484 @9 = 'nine'
1485 assert_equal('nine', @9)
1486 @- = 'minus'
1487 assert_equal('minus', @-)
1488 if has('clipboard_working')
1489 @* = 'star'
1490 assert_equal('star', @*)
1491 @+ = 'plus'
1492 assert_equal('plus', @+)
1493 endif
1494
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001495 var a: number = 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001496 assert_equal(123, a)
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001497 var s: string = 'yes'
Bram Moolenaar971caf52020-09-21 22:21:42 +02001498 assert_equal('yes', s)
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001499 var b: number = 42
Bram Moolenaar971caf52020-09-21 22:21:42 +02001500 assert_equal(42, b)
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001501 var w: number = 43
Bram Moolenaar971caf52020-09-21 22:21:42 +02001502 assert_equal(43, w)
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001503 var t: number = 44
Bram Moolenaar971caf52020-09-21 22:21:42 +02001504 assert_equal(44, t)
Bram Moolenaar7a66a172020-10-16 19:56:12 +02001505
1506 var to_var = 0
1507 to_var = 3
1508 assert_equal(3, to_var)
Bram Moolenaar971caf52020-09-21 22:21:42 +02001509 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001510 v9.CheckScriptSuccess(lines)
Bram Moolenaar06f9c692020-09-27 21:27:40 +02001511
1512 lines =<< trim END
1513 vim9script
1514 var n: number
1515 def Func()
1516 n = 'string'
1517 enddef
1518 defcompile
1519 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001520 v9.CheckScriptFailure(lines, 'E1012: Type mismatch; expected number but got string')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001521enddef
1522
1523def Mess(): string
1524 v:foldstart = 123
1525 return 'xxx'
1526enddef
1527
1528def Test_assignment_failure()
Bram Moolenaar62aec932022-01-29 21:45:34 +00001529 v9.CheckDefFailure(['var name=234'], 'E1004:')
1530 v9.CheckDefFailure(['var name =234'], 'E1004:')
1531 v9.CheckDefFailure(['var name= 234'], 'E1004:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001532
Bram Moolenaar62aec932022-01-29 21:45:34 +00001533 v9.CheckScriptFailure(['vim9script', 'var name=234'], 'E1004:')
1534 v9.CheckScriptFailure(['vim9script', 'var name=234'], "before and after '='")
1535 v9.CheckScriptFailure(['vim9script', 'var name =234'], 'E1004:')
1536 v9.CheckScriptFailure(['vim9script', 'var name= 234'], 'E1004:')
1537 v9.CheckScriptFailure(['vim9script', 'var name = 234', 'name+=234'], 'E1004:')
1538 v9.CheckScriptFailure(['vim9script', 'var name = 234', 'name+=234'], "before and after '+='")
1539 v9.CheckScriptFailure(['vim9script', 'var name = "x"', 'name..="y"'], 'E1004:')
1540 v9.CheckScriptFailure(['vim9script', 'var name = "x"', 'name..="y"'], "before and after '..='")
Bram Moolenaar971caf52020-09-21 22:21:42 +02001541
Bram Moolenaar62aec932022-01-29 21:45:34 +00001542 v9.CheckDefFailure(['var true = 1'], 'E1034:')
1543 v9.CheckDefFailure(['var false = 1'], 'E1034:')
1544 v9.CheckDefFailure(['var null = 1'], 'E1034:')
1545 v9.CheckDefFailure(['var this = 1'], 'E1034:')
Bram Moolenaarce723f32023-06-10 19:00:12 +01001546 v9.CheckDefFailure(['var super = 1'], 'E1034:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001547
Bram Moolenaar8b716f52022-02-15 21:17:56 +00001548 v9.CheckDefFailure(['[a; b; c] = g:list'], 'E1001:')
1549 v9.CheckDefFailure(['var [a; b; c] = g:list'], 'E1080:')
Bram Moolenaar62aec932022-01-29 21:45:34 +00001550 v9.CheckDefExecFailure(['var a: number',
Bram Moolenaar971caf52020-09-21 22:21:42 +02001551 '[a] = test_null_list()'], 'E1093:')
Bram Moolenaar62aec932022-01-29 21:45:34 +00001552 v9.CheckDefExecFailure(['var a: number',
Bram Moolenaar971caf52020-09-21 22:21:42 +02001553 '[a] = []'], 'E1093:')
Bram Moolenaar62aec932022-01-29 21:45:34 +00001554 v9.CheckDefExecFailure(['var x: number',
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001555 'var y: number',
Bram Moolenaar971caf52020-09-21 22:21:42 +02001556 '[x, y] = [1]'], 'E1093:')
Bram Moolenaar62aec932022-01-29 21:45:34 +00001557 v9.CheckDefExecFailure(['var x: string',
Bram Moolenaar352134b2020-10-17 22:04:08 +02001558 'var y: string',
1559 '[x, y] = ["x"]'], 'E1093:')
Bram Moolenaar62aec932022-01-29 21:45:34 +00001560 v9.CheckDefExecFailure(['var x: number',
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001561 'var y: number',
1562 'var z: list<number>',
Bram Moolenaar971caf52020-09-21 22:21:42 +02001563 '[x, y; z] = [1]'], 'E1093:')
1564
Bram Moolenaar62aec932022-01-29 21:45:34 +00001565 v9.CheckDefFailure(['var somevar'], "E1022:")
1566 v9.CheckDefFailure(['var &tabstop = 4'], 'E1052:')
1567 v9.CheckDefFailure(['&g:option = 5'], 'E113:')
1568 v9.CheckScriptFailure(['vim9script', 'var &tabstop = 4'], 'E1052:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001569
Bram Moolenaar62aec932022-01-29 21:45:34 +00001570 v9.CheckDefFailure(['var $VAR = 5'], 'E1016: Cannot declare an environment variable:')
1571 v9.CheckScriptFailure(['vim9script', 'var $ENV = "xxx"'], 'E1016:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001572
1573 if has('dnd')
Bram Moolenaar62aec932022-01-29 21:45:34 +00001574 v9.CheckDefFailure(['var @~ = 5'], 'E1066:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001575 else
Bram Moolenaar62aec932022-01-29 21:45:34 +00001576 v9.CheckDefFailure(['var @~ = 5'], 'E354:')
1577 v9.CheckDefFailure(['@~ = 5'], 'E354:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001578 endif
Bram Moolenaar62aec932022-01-29 21:45:34 +00001579 v9.CheckDefFailure(['var @a = 5'], 'E1066:')
1580 v9.CheckDefFailure(['var @/ = "x"'], 'E1066:')
1581 v9.CheckScriptFailure(['vim9script', 'var @a = "abc"'], 'E1066:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001582
Bram Moolenaar62aec932022-01-29 21:45:34 +00001583 v9.CheckDefFailure(['var g:var = 5'], 'E1016: Cannot declare a global variable:')
1584 v9.CheckDefFailure(['var w:var = 5'], 'E1016: Cannot declare a window variable:')
1585 v9.CheckDefFailure(['var b:var = 5'], 'E1016: Cannot declare a buffer variable:')
1586 v9.CheckDefFailure(['var t:var = 5'], 'E1016: Cannot declare a tab variable:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001587
Bram Moolenaar62aec932022-01-29 21:45:34 +00001588 v9.CheckDefFailure(['var anr = 4', 'anr ..= "text"'], 'E1019:')
1589 v9.CheckDefFailure(['var xnr += 4'], 'E1020:', 1)
1590 v9.CheckScriptFailure(['vim9script', 'var xnr += 4'], 'E1020:')
1591 v9.CheckDefFailure(["var xnr = xnr + 1"], 'E1001:', 1)
1592 v9.CheckScriptFailure(['vim9script', 'var xnr = xnr + 4'], 'E121:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001593
Bram Moolenaara749a422022-02-12 19:52:25 +00001594 v9.CheckScriptFailure(['vim9script', 'def Func()', 'var dummy = notfound', 'enddef', 'defcompile'], 'E1001:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001595
Bram Moolenaar62aec932022-01-29 21:45:34 +00001596 v9.CheckDefFailure(['var name: list<string> = [123]'], 'expected list<string> but got list<number>')
1597 v9.CheckDefFailure(['var name: list<number> = ["xx"]'], 'expected list<number> but got list<string>')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001598
Bram Moolenaar62aec932022-01-29 21:45:34 +00001599 v9.CheckDefFailure(['var name: dict<string> = {key: 123}'], 'expected dict<string> but got dict<number>')
1600 v9.CheckDefFailure(['var name: dict<number> = {key: "xx"}'], 'expected dict<number> but got dict<string>')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001601
Bram Moolenaar62aec932022-01-29 21:45:34 +00001602 v9.CheckDefFailure(['var name = feedkeys("0")'], 'E1031:')
1603 v9.CheckDefFailure(['var name: number = feedkeys("0")'], 'expected number but got void')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001604
Bram Moolenaar62aec932022-01-29 21:45:34 +00001605 v9.CheckDefFailure(['var name: dict <number>'], 'E1068:')
Bram Moolenaar62e0e2e2022-08-20 12:07:58 +01001606 v9.CheckDefFailure(['var name: dict<number'], 'E1009: Missing > after type: <number')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001607
Bram Moolenaar848fadd2022-01-30 15:28:30 +00001608 assert_fails('s/^/\=g:Mess()/n', 'E794:')
Bram Moolenaar62aec932022-01-29 21:45:34 +00001609 v9.CheckDefFailure(['var name: dict<number'], 'E1009:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001610
Bram Moolenaar62aec932022-01-29 21:45:34 +00001611 v9.CheckDefFailure(['w:foo: number = 10'],
Bram Moolenaar9510d222022-09-11 15:14:05 +01001612 'E1016: Cannot declare a window variable: w:foo')
Bram Moolenaar62aec932022-01-29 21:45:34 +00001613 v9.CheckDefFailure(['t:foo: bool = true'],
Bram Moolenaar9510d222022-09-11 15:14:05 +01001614 'E1016: Cannot declare a tab variable: t:foo')
Bram Moolenaar62aec932022-01-29 21:45:34 +00001615 v9.CheckDefFailure(['b:foo: string = "x"'],
Bram Moolenaar9510d222022-09-11 15:14:05 +01001616 'E1016: Cannot declare a buffer variable: b:foo')
Bram Moolenaar62aec932022-01-29 21:45:34 +00001617 v9.CheckDefFailure(['g:foo: number = 123'],
Bram Moolenaar9510d222022-09-11 15:14:05 +01001618 'E1016: Cannot declare a global variable: g:foo')
1619
1620 v9.CheckScriptFailure(['vim9script', 'w:foo: number = 123'],
1621 'E1304: Cannot use type with this variable: w:foo:')
1622 v9.CheckScriptFailure(['vim9script', 't:foo: number = 123'],
1623 'E1304: Cannot use type with this variable: t:foo:')
1624 v9.CheckScriptFailure(['vim9script', 'b:foo: number = 123'],
1625 'E1304: Cannot use type with this variable: b:foo:')
1626 v9.CheckScriptFailure(['vim9script', 'g:foo: number = 123'],
1627 'E1304: Cannot use type with this variable: g:foo:')
1628
1629 v9.CheckScriptFailure(['vim9script', 'const w:FOO: number = 123'],
1630 'E1304: Cannot use type with this variable: w:FOO:')
1631 v9.CheckScriptFailure(['vim9script', 'const t:FOO: number = 123'],
1632 'E1304: Cannot use type with this variable: t:FOO:')
1633 v9.CheckScriptFailure(['vim9script', 'const b:FOO: number = 123'],
1634 'E1304: Cannot use type with this variable: b:FOO:')
1635 v9.CheckScriptFailure(['vim9script', 'const g:FOO: number = 123'],
1636 'E1304: Cannot use type with this variable: g:FOO:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001637enddef
1638
1639def Test_assign_list()
Bram Moolenaare65081d2021-06-27 15:04:05 +02001640 var lines =<< trim END
1641 var l: list<string> = []
1642 l[0] = 'value'
1643 assert_equal('value', l[0])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001644
Bram Moolenaare65081d2021-06-27 15:04:05 +02001645 l[1] = 'asdf'
1646 assert_equal('value', l[0])
1647 assert_equal('asdf', l[1])
1648 assert_equal('asdf', l[-1])
1649 assert_equal('value', l[-2])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001650
Bram Moolenaare65081d2021-06-27 15:04:05 +02001651 var nrl: list<number> = []
1652 for i in range(5)
1653 nrl[i] = i
1654 endfor
1655 assert_equal([0, 1, 2, 3, 4], nrl)
1656
1657 var ul: list<any>
1658 ul[0] = 1
1659 ul[1] = 2
1660 ul[2] = 3
1661 assert_equal([1, 2, 3], ul)
1662 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001663 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar334a8b42020-10-19 16:07:42 +02001664
Bram Moolenaar1983f1a2022-02-28 20:55:02 +00001665 lines =<< trim END
1666 var l = [1, 2]
1667 g:idx = 'x'
1668 l[g:idx : 1] = [0]
1669 echo l
1670 END
Bram Moolenaar2995e5c2022-03-18 21:41:47 +00001671 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 +00001672
1673 lines =<< trim END
1674 var l = [1, 2]
1675 g:idx = 3
1676 l[g:idx : 1] = [0]
1677 echo l
1678 END
Bram Moolenaarec892232022-05-06 17:53:06 +01001679 v9.CheckDefExecAndScriptFailure(lines, 'E684: List index out of range: 3')
Bram Moolenaar1983f1a2022-02-28 20:55:02 +00001680
1681 lines =<< trim END
1682 var l = [1, 2]
1683 g:idx = 'y'
1684 l[1 : g:idx] = [0]
1685 echo l
1686 END
1687 v9.CheckDefExecAndScriptFailure(lines, ['E1012: Type mismatch; expected number but got string', 'E1030: Using a String as a Number: "y"'])
1688
Bram Moolenaar62aec932022-01-29 21:45:34 +00001689 v9.CheckDefFailure(["var l: list<number> = ['', true]"], 'E1012: Type mismatch; expected list<number> but got list<any>', 1)
1690 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 +02001691enddef
1692
1693def Test_assign_dict()
Bram Moolenaarc3689572021-01-01 19:40:02 +01001694 var lines =<< trim END
1695 var d: dict<string> = {}
1696 d['key'] = 'value'
1697 assert_equal('value', d['key'])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001698
Bram Moolenaarc3689572021-01-01 19:40:02 +01001699 d[123] = 'qwerty'
1700 assert_equal('qwerty', d[123])
1701 assert_equal('qwerty', d['123'])
Bram Moolenaar971caf52020-09-21 22:21:42 +02001702
Bram Moolenaarc3689572021-01-01 19:40:02 +01001703 var nrd: dict<number> = {}
1704 for i in range(3)
1705 nrd[i] = i
1706 endfor
1707 assert_equal({0: 0, 1: 1, 2: 2}, nrd)
1708
1709 d.somekey = 'someval'
1710 assert_equal({key: 'value', '123': 'qwerty', somekey: 'someval'}, d)
Bram Moolenaar23e2e112021-08-03 21:16:18 +02001711 unlet d.somekey
1712 assert_equal({key: 'value', '123': 'qwerty'}, d)
Bram Moolenaarc3689572021-01-01 19:40:02 +01001713 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001714 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaarc3689572021-01-01 19:40:02 +01001715
Bram Moolenaar62aec932022-01-29 21:45:34 +00001716 v9.CheckDefFailure(["var d: dict<number> = {a: '', b: true}"], 'E1012: Type mismatch; expected dict<number> but got dict<any>', 1)
1717 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 +00001718 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 +02001719enddef
1720
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001721def Test_assign_dict_unknown_type()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001722 var lines =<< trim END
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001723 vim9script
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001724 var mylist = []
Bram Moolenaare0de1712020-12-02 17:36:54 +01001725 mylist += [{one: 'one'}]
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001726 def Func()
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001727 var dd = mylist[0]
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001728 assert_equal('one', dd.one)
1729 enddef
1730 Func()
1731 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001732 v9.CheckScriptSuccess(lines)
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001733
Bram Moolenaarce2c5442020-11-28 21:21:17 +01001734 lines =<< trim END
1735 vim9script
1736 var mylist = [[]]
Bram Moolenaare0de1712020-12-02 17:36:54 +01001737 mylist[0] += [{one: 'one'}]
Bram Moolenaarce2c5442020-11-28 21:21:17 +01001738 def Func()
1739 var dd = mylist[0][0]
1740 assert_equal('one', dd.one)
1741 enddef
1742 Func()
1743 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001744 v9.CheckScriptSuccess(lines)
Bram Moolenaar04bdd572020-09-23 13:25:32 +02001745enddef
1746
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001747def Test_assign_dict_with_op()
1748 var lines =<< trim END
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001749 var ds: dict<string> = {a: 'x'}
1750 ds['a'] ..= 'y'
1751 ds.a ..= 'z'
1752 assert_equal('xyz', ds.a)
1753
1754 var dn: dict<number> = {a: 9}
1755 dn['a'] += 2
1756 assert_equal(11, dn.a)
1757 dn.a += 2
1758 assert_equal(13, dn.a)
1759
1760 dn['a'] -= 3
1761 assert_equal(10, dn.a)
1762 dn.a -= 2
1763 assert_equal(8, dn.a)
1764
1765 dn['a'] *= 2
1766 assert_equal(16, dn.a)
1767 dn.a *= 2
1768 assert_equal(32, dn.a)
1769
1770 dn['a'] /= 3
1771 assert_equal(10, dn.a)
1772 dn.a /= 2
1773 assert_equal(5, dn.a)
1774
1775 dn['a'] %= 3
1776 assert_equal(2, dn.a)
1777 dn.a %= 6
1778 assert_equal(2, dn.a)
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001779
1780 var dd: dict<dict<list<any>>>
1781 dd.a = {}
1782 dd.a.b = [0]
1783 dd.a.b += [1]
1784 assert_equal({a: {b: [0, 1]}}, dd)
Bram Moolenaar0d351a02021-04-05 21:05:26 +02001785
1786 var dab = {a: ['b']}
1787 dab.a[0] ..= 'c'
1788 assert_equal({a: ['bc']}, dab)
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001789 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001790 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaare42939a2021-04-05 17:11:17 +02001791enddef
1792
1793def Test_assign_list_with_op()
1794 var lines =<< trim END
1795 var ls: list<string> = ['x']
1796 ls[0] ..= 'y'
1797 assert_equal('xy', ls[0])
1798
1799 var ln: list<number> = [9]
1800 ln[0] += 2
1801 assert_equal(11, ln[0])
1802
1803 ln[0] -= 3
1804 assert_equal(8, ln[0])
1805
1806 ln[0] *= 2
1807 assert_equal(16, ln[0])
1808
1809 ln[0] /= 3
1810 assert_equal(5, ln[0])
1811
1812 ln[0] %= 3
1813 assert_equal(2, ln[0])
1814 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001815 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaare42939a2021-04-05 17:11:17 +02001816enddef
1817
1818def Test_assign_with_op_fails()
1819 var lines =<< trim END
1820 var s = 'abc'
1821 s[1] += 'x'
1822 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001823 v9.CheckDefAndScriptFailure(lines, ['E1141:', 'E689:'], 2)
Bram Moolenaare42939a2021-04-05 17:11:17 +02001824
1825 lines =<< trim END
1826 var s = 'abc'
1827 s[1] ..= 'x'
1828 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001829 v9.CheckDefAndScriptFailure(lines, ['E1141:', 'E689:'], 2)
Bram Moolenaarb9c0cd82021-04-05 20:51:00 +02001830
1831 lines =<< trim END
1832 var dd: dict<dict<list<any>>>
1833 dd.a = {}
1834 dd.a.b += [1]
1835 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001836 v9.CheckDefExecAndScriptFailure(lines, 'E716:', 3)
Bram Moolenaarf76ec1e2021-03-03 17:58:16 +01001837enddef
1838
Bram Moolenaar971caf52020-09-21 22:21:42 +02001839def Test_assign_lambda()
1840 # check if assign a lambda to a variable which type is func or any.
Bram Moolenaar30fd8202020-09-26 15:09:30 +02001841 var lines =<< trim END
Bram Moolenaar971caf52020-09-21 22:21:42 +02001842 vim9script
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001843 var FuncRef = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001844 assert_equal(123, FuncRef())
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001845 var FuncRef_Func: func = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001846 assert_equal(123, FuncRef_Func())
Bram Moolenaar2949cfd2020-12-31 21:28:47 +01001847 var FuncRef_Any: any = () => 123
Bram Moolenaar971caf52020-09-21 22:21:42 +02001848 assert_equal(123, FuncRef_Any())
Bram Moolenaar06409502021-02-17 17:00:27 +01001849 var FuncRef_Number: func(): number = () => 321
1850 assert_equal(321, FuncRef_Number())
Bram Moolenaar971caf52020-09-21 22:21:42 +02001851 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001852 v9.CheckScriptSuccess(lines)
Bram Moolenaar328eac22021-01-07 19:23:08 +01001853
1854 lines =<< trim END
1855 var Ref: func(number)
1856 Ref = (j) => !j
1857 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001858 v9.CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected func(number) but got func(any): bool')
Bram Moolenaarc7dac852021-02-17 18:49:11 +01001859
1860 lines =<< trim END
Bram Moolenaar94722c52023-01-28 19:19:03 +00001861 echo filter([1, 2, 3], (_, v: string) => v + 1)
Bram Moolenaarc7dac852021-02-17 18:49:11 +01001862 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00001863 v9.CheckDefAndScriptFailure(lines, 'E1051:')
Bram Moolenaar971caf52020-09-21 22:21:42 +02001864enddef
1865
Ernie Rael96952b22023-10-17 18:15:01 +02001866def Test_assign_funcref_args()
1867 # unspecified arguments match everything, including varargs
1868 var lines =<< trim END
1869 vim9script
1870
1871 var FuncUnknown: func: number
1872
1873 FuncUnknown = (v): number => v
1874 assert_equal(5, FuncUnknown(5))
1875
1876 FuncUnknown = (v1, v2): number => v1 + v2
1877 assert_equal(7, FuncUnknown(3, 4))
1878
1879 FuncUnknown = (...v1): number => v1[0] + v1[1] + len(v1) * 1000
1880 assert_equal(4007, FuncUnknown(3, 4, 5, 6))
1881
1882 FuncUnknown = (v: list<any>): number => v[0] + v[1] + len(v) * 1000
1883 assert_equal(5009, FuncUnknown([4, 5, 6, 7, 8]))
1884 END
1885 v9.CheckScriptSuccess(lines)
1886
1887 # varargs must match
1888 lines =<< trim END
1889 vim9script
1890 var FuncAnyVA: func(...any): number
1891 FuncAnyVA = (v): number => v
1892 END
Ernie Rael3ec6c1f2023-10-21 11:45:38 +02001893 v9.CheckScriptFailure(lines, 'E1180: Variable arguments type must be a list: any')
Ernie Rael96952b22023-10-17 18:15:01 +02001894
1895 # varargs must match
1896 lines =<< trim END
1897 vim9script
1898 var FuncAnyVA: func(...any): number
1899 FuncAnyVA = (v1, v2): number => v1 + v2
1900 END
Ernie Rael3ec6c1f2023-10-21 11:45:38 +02001901 v9.CheckScriptFailure(lines, 'E1180: Variable arguments type must be a list: any')
Ernie Rael96952b22023-10-17 18:15:01 +02001902
1903 # varargs must match
1904 lines =<< trim END
1905 vim9script
1906 var FuncAnyVA: func(...any): number
1907 FuncAnyVA = (v1: list<any>): number => 3
1908 END
Ernie Rael3ec6c1f2023-10-21 11:45:38 +02001909 v9.CheckScriptFailure(lines, 'E1180: Variable arguments type must be a list: any')
Ernie Rael96952b22023-10-17 18:15:01 +02001910enddef
1911
1912def Test_assign_funcref_arg_any()
1913 var lines =<< trim END
1914 vim9script
1915 var FuncAnyVA: func(any): number
1916 FuncAnyVA = (v): number => v
1917 END
1918 # TODO: Verify this should succeed.
1919 v9.CheckScriptSuccess(lines)
1920enddef
1921
Bram Moolenaarc0e29012020-09-27 14:22:48 +02001922def Test_heredoc()
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01001923 # simple heredoc
1924 var lines =<< trim END
zeertzjqa93d9cd2023-05-02 16:25:47 +01001925 var text =<< trim TEXT # comment
1926 abc
1927 TEXT
1928 assert_equal(['abc'], text)
Bram Moolenaarc0e29012020-09-27 14:22:48 +02001929 END
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01001930 v9.CheckDefAndScriptSuccess(lines)
1931
1932 # empty heredoc
1933 lines =<< trim END
zeertzjqa93d9cd2023-05-02 16:25:47 +01001934 var text =<< trim TEXT
1935 TEXT
1936 assert_equal([], text)
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01001937 END
1938 v9.CheckDefAndScriptSuccess(lines)
1939
1940 # heredoc with a single empty line
1941 lines =<< trim END
zeertzjqa93d9cd2023-05-02 16:25:47 +01001942 var text =<< trim TEXT
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01001943
zeertzjqa93d9cd2023-05-02 16:25:47 +01001944 TEXT
1945 assert_equal([''], text)
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01001946 END
1947 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaarc0e29012020-09-27 14:22:48 +02001948
zeertzjqa93d9cd2023-05-02 16:25:47 +01001949 # assign heredoc to variable with type
1950 lines =<< trim END
1951 var text: list<string> =<< trim TEXT
1952 var foo =<< trim FOO
1953 TEXT
1954 assert_equal(['var foo =<< trim FOO'], text)
1955 END
1956 v9.CheckDefAndScriptSuccess(lines)
1957
1958 # extra whitespace before type is allowed
1959 lines =<< trim END
1960 var text: list<string> =<< trim TEXT
1961 var foo =<< trim FOO
1962 TEXT
1963 assert_equal(['var foo =<< trim FOO'], text)
1964 END
1965 v9.CheckDefAndScriptSuccess(lines)
1966
1967 # missing whitespace before type is an error
1968 lines =<< trim END
1969 var text:list<string> =<< trim TEXT
1970 var foo =<< trim FOO
1971 TEXT
1972 assert_equal(['var foo =<< trim FOO'], text)
1973 END
1974 v9.CheckDefAndScriptFailure(lines, 'E1069:')
1975
1976 # assign heredoc to list slice
1977 lines =<< trim END
1978 var text = ['']
1979 text[ : ] =<< trim TEXT
1980 var foo =<< trim FOO
1981 TEXT
1982 assert_equal(['var foo =<< trim FOO'], text)
1983 END
1984 v9.CheckDefAndScriptSuccess(lines)
1985
1986 # assign heredoc to curly braces name in legacy function in Vim9 script
1987 lines =<< trim END
1988 vim9script
1989 func Func()
1990 let foo_3_bar = ['']
1991 let foo_{1 + 2}_bar[ : ] =<< trim TEXT
1992 var foo =<< trim FOO
1993 TEXT
1994 call assert_equal(['var foo =<< trim FOO'], foo_3_bar)
1995 endfunc
1996 Func()
1997 END
1998 v9.CheckScriptSuccess(lines)
1999
Bram Moolenaar62aec932022-01-29 21:45:34 +00002000 v9.CheckDefFailure(['var lines =<< trim END X', 'END'], 'E488:')
2001 v9.CheckDefFailure(['var lines =<< trim END " comment', 'END'], 'E488:')
Bram Moolenaar2914a202020-09-27 18:24:03 +02002002
2003 lines =<< trim [END]
2004 def Func()
2005 var&lines =<< trim END
2006 x
2007 x
Bram Moolenaarb5b94802020-12-13 17:50:20 +01002008 enddef
2009 defcompile
2010 [END]
Bram Moolenaar62aec932022-01-29 21:45:34 +00002011 v9.CheckScriptFailure(lines, 'E1145: Missing heredoc end marker: END')
Bram Moolenaarb5b94802020-12-13 17:50:20 +01002012 delfunc! g:Func
2013
2014 lines =<< trim [END]
2015 def Func()
2016 var lines =<< trim END
2017 x
2018 x
Bram Moolenaar2914a202020-09-27 18:24:03 +02002019 x
2020 x
2021 x
2022 x
2023 x
2024 x
2025 enddef
2026 call Func()
2027 [END]
Bram Moolenaar62aec932022-01-29 21:45:34 +00002028 v9.CheckScriptFailure(lines, 'E1145: Missing heredoc end marker: END')
Bram Moolenaar2d870f82020-12-05 13:41:01 +01002029 delfunc! g:Func
Bram Moolenaar81530e32021-07-28 21:25:49 +02002030
2031 lines =<< trim END
2032 var lines: number =<< trim STOP
2033 aaa
2034 bbb
2035 STOP
2036 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002037 v9.CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected number but got list<string>', 1)
Bram Moolenaard9876422022-10-12 12:58:54 +01002038
2039 lines =<< trim END
2040 var lines=<< STOP
2041 xxx
2042 STOP
2043 END
2044 v9.CheckDefAndScriptFailure(lines, 'E1004: White space required before and after ''=<<'' at "=<< STOP"', 1)
2045 lines =<< trim END
2046 var lines =<<STOP
2047 xxx
2048 STOP
2049 END
2050 v9.CheckDefAndScriptFailure(lines, 'E1004: White space required before and after ''=<<'' at "=<<STOP"', 1)
2051 lines =<< trim END
2052 var lines=<<STOP
2053 xxx
2054 STOP
2055 END
2056 v9.CheckDefAndScriptFailure(lines, 'E1004: White space required before and after ''=<<'' at "=<<STOP"', 1)
Bram Moolenaarc0e29012020-09-27 14:22:48 +02002057enddef
2058
Bram Moolenaar083966f2020-12-08 11:29:40 +01002059def Test_var_func_call()
Bram Moolenaar55759b52020-09-30 22:59:42 +02002060 var lines =<< trim END
2061 vim9script
2062 func GetValue()
2063 if exists('g:count')
2064 let g:count += 1
2065 else
2066 let g:count = 1
2067 endif
2068 return 'this'
2069 endfunc
Bram Moolenaar94722c52023-01-28 19:19:03 +00002070 var val: string = GetValue()
Bram Moolenaar55759b52020-09-30 22:59:42 +02002071 # env var is always a string
2072 var env = $TERM
2073 END
Bram Moolenaar0e9bdad2022-10-15 20:06:33 +01002074 writefile(lines, 'Xfinished', 'D')
Bram Moolenaar55759b52020-09-30 22:59:42 +02002075 source Xfinished
2076 # GetValue() is not called during discovery phase
2077 assert_equal(1, g:count)
2078
2079 unlet g:count
Bram Moolenaar55759b52020-09-30 22:59:42 +02002080enddef
2081
Bram Moolenaar083966f2020-12-08 11:29:40 +01002082def Test_var_missing_type()
Bram Moolenaar55759b52020-09-30 22:59:42 +02002083 var lines =<< trim END
2084 vim9script
2085 var name = g:unknown
2086 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002087 v9.CheckScriptFailure(lines, 'E121:')
Bram Moolenaar55759b52020-09-30 22:59:42 +02002088
2089 lines =<< trim END
2090 vim9script
2091 var nr: number = 123
2092 var name = nr
2093 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002094 v9.CheckScriptSuccess(lines)
Bram Moolenaar55759b52020-09-30 22:59:42 +02002095enddef
2096
Bram Moolenaar083966f2020-12-08 11:29:40 +01002097def Test_var_declaration()
Bram Moolenaar55759b52020-09-30 22:59:42 +02002098 var lines =<< trim END
2099 vim9script
2100 var name: string
2101 g:var_uninit = name
2102 name = 'text'
2103 g:var_test = name
Bram Moolenaara749a422022-02-12 19:52:25 +00002104 # prefixing s: is not allowed
2105 name = 'prefixed'
2106 g:var_prefixed = name
Bram Moolenaar55759b52020-09-30 22:59:42 +02002107
Bram Moolenaar7257af42020-12-08 22:08:51 +01002108 const FOO: number = 123
2109 assert_equal(123, FOO)
Bram Moolenaar89b474d2020-12-22 21:19:39 +01002110 const FOOS = 'foos'
2111 assert_equal('foos', FOOS)
2112 final FLIST = [1]
2113 assert_equal([1], FLIST)
2114 FLIST[0] = 11
2115 assert_equal([11], FLIST)
2116
Bram Moolenaar89b474d2020-12-22 21:19:39 +01002117 const g:FOOS = 'gfoos'
2118 assert_equal('gfoos', g:FOOS)
2119 final g:FLIST = [2]
2120 assert_equal([2], g:FLIST)
2121 g:FLIST[0] = 22
2122 assert_equal([22], g:FLIST)
2123
Bram Moolenaard877a572021-04-01 19:42:48 +02002124 def SetGlobalConst()
2125 const g:globConst = 123
2126 enddef
2127 SetGlobalConst()
2128 assert_equal(123, g:globConst)
2129 assert_true(islocked('g:globConst'))
2130
Bram Moolenaar89b474d2020-12-22 21:19:39 +01002131 const w:FOOS = 'wfoos'
2132 assert_equal('wfoos', w:FOOS)
2133 final w:FLIST = [3]
2134 assert_equal([3], w:FLIST)
2135 w:FLIST[0] = 33
2136 assert_equal([33], w:FLIST)
Bram Moolenaar7257af42020-12-08 22:08:51 +01002137
Bram Moolenaar55759b52020-09-30 22:59:42 +02002138 var s:other: number
2139 other = 1234
2140 g:other_var = other
2141
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01002142 var xyz: string # comment
2143
Bram Moolenaar55759b52020-09-30 22:59:42 +02002144 # type is inferred
Bram Moolenaara749a422022-02-12 19:52:25 +00002145 var dict = {['a']: 222}
Bram Moolenaar55759b52020-09-30 22:59:42 +02002146 def GetDictVal(key: any)
Bram Moolenaara749a422022-02-12 19:52:25 +00002147 g:dict_val = dict[key]
Bram Moolenaar55759b52020-09-30 22:59:42 +02002148 enddef
2149 GetDictVal('a')
Bram Moolenaar08251752021-01-11 21:20:18 +01002150
2151 final adict: dict<string> = {}
2152 def ChangeAdict()
2153 adict.foo = 'foo'
2154 enddef
2155 ChangeAdict()
Bram Moolenaar55759b52020-09-30 22:59:42 +02002156 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002157 v9.CheckScriptSuccess(lines)
Bram Moolenaar55759b52020-09-30 22:59:42 +02002158 assert_equal('', g:var_uninit)
2159 assert_equal('text', g:var_test)
2160 assert_equal('prefixed', g:var_prefixed)
2161 assert_equal(1234, g:other_var)
2162 assert_equal(222, g:dict_val)
2163
2164 unlet g:var_uninit
2165 unlet g:var_test
2166 unlet g:var_prefixed
2167 unlet g:other_var
Bram Moolenaar6e50ec22021-04-03 19:32:44 +02002168 unlet g:globConst
Bram Moolenaar89b474d2020-12-22 21:19:39 +01002169 unlet g:FOOS
2170 unlet g:FLIST
Bram Moolenaar89b474d2020-12-22 21:19:39 +01002171 unlet w:FOOS
2172 unlet w:FLIST
Bram Moolenaar55759b52020-09-30 22:59:42 +02002173enddef
2174
Bram Moolenaar566badc2022-09-18 13:46:08 +01002175def Test_create_list_after_const()
2176 const a = 1
2177 g:ll = []
2178 assert_equal(0, islocked('g:ll'))
2179 unlet g:ll
2180enddef
2181
Bram Moolenaar083966f2020-12-08 11:29:40 +01002182def Test_var_declaration_fails()
Bram Moolenaar55759b52020-09-30 22:59:42 +02002183 var lines =<< trim END
2184 vim9script
2185 final var: string
2186 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002187 v9.CheckScriptFailure(lines, 'E1125:')
Bram Moolenaar55759b52020-09-30 22:59:42 +02002188
2189 lines =<< trim END
2190 vim9script
Bram Moolenaar89b474d2020-12-22 21:19:39 +01002191 const g:constvar = 'string'
2192 g:constvar = 'xx'
2193 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002194 v9.CheckScriptFailure(lines, 'E741:')
Bram Moolenaar89b474d2020-12-22 21:19:39 +01002195 unlet g:constvar
2196
2197 lines =<< trim END
2198 vim9script
Bram Moolenaarf5906aa2021-04-02 14:35:15 +02002199 var name = 'one'
2200 lockvar name
2201 def SetLocked()
2202 name = 'two'
2203 enddef
2204 SetLocked()
2205 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002206 v9.CheckScriptFailure(lines, 'E741: Value is locked: name', 1)
Bram Moolenaardcf29ac2021-04-02 14:44:02 +02002207
2208 lines =<< trim END
2209 let s:legacy = 'one'
2210 lockvar s:legacy
2211 def SetLocked()
2212 s:legacy = 'two'
2213 enddef
2214 call SetLocked()
2215 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002216 v9.CheckScriptFailure(lines, 'E741: Value is locked: s:legacy', 1)
Bram Moolenaarf5906aa2021-04-02 14:35:15 +02002217
2218 lines =<< trim END
2219 vim9script
Bram Moolenaard877a572021-04-01 19:42:48 +02002220 def SetGlobalConst()
2221 const g:globConst = 123
2222 enddef
2223 SetGlobalConst()
2224 g:globConst = 234
2225 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002226 v9.CheckScriptFailure(lines, 'E741: Value is locked: g:globConst', 6)
Bram Moolenaard877a572021-04-01 19:42:48 +02002227 unlet g:globConst
2228
2229 lines =<< trim END
2230 vim9script
Bram Moolenaar08251752021-01-11 21:20:18 +01002231 const cdict: dict<string> = {}
2232 def Change()
2233 cdict.foo = 'foo'
2234 enddef
2235 defcompile
2236 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002237 v9.CheckScriptFailure(lines, 'E46:')
Bram Moolenaar08251752021-01-11 21:20:18 +01002238
2239 lines =<< trim END
2240 vim9script
Bram Moolenaar89b474d2020-12-22 21:19:39 +01002241 final w:finalvar = [9]
2242 w:finalvar = [8]
2243 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002244 v9.CheckScriptFailure(lines, 'E1122:')
Bram Moolenaar89b474d2020-12-22 21:19:39 +01002245 unlet w:finalvar
2246
2247 lines =<< trim END
2248 vim9script
Bram Moolenaar55759b52020-09-30 22:59:42 +02002249 const var: string
2250 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002251 v9.CheckScriptFailure(lines, 'E1021:')
Bram Moolenaar55759b52020-09-30 22:59:42 +02002252
2253 lines =<< trim END
2254 vim9script
2255 var 9var: string
2256 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002257 v9.CheckScriptFailure(lines, 'E488:')
Bram Moolenaar7257af42020-12-08 22:08:51 +01002258
Bram Moolenaar62aec932022-01-29 21:45:34 +00002259 v9.CheckDefFailure(['var foo.bar = 2'], 'E1087:')
2260 v9.CheckDefFailure(['var foo[3] = 2'], 'E1087:')
2261 v9.CheckDefFailure(['const foo: number'], 'E1021:')
Bram Moolenaare1d12112022-03-05 11:37:48 +00002262
2263 lines =<< trim END
2264 va foo = 123
2265 END
2266 v9.CheckDefAndScriptFailure(lines, 'E1065:', 1)
Bram Moolenaar31d99482022-05-26 22:24:43 +01002267
2268 lines =<< trim END
2269 var foo: func(number
2270 END
2271 v9.CheckDefAndScriptFailure(lines, 'E110:', 1)
2272
2273 lines =<< trim END
2274 var foo: func(number): func(
2275 END
2276 v9.CheckDefAndScriptFailure(lines, 'E110:', 1)
2277
2278 for type in ['num_ber',
2279 'anys', 'ani',
2280 'bools', 'boel',
2281 'blobs', 'blub',
2282 'channels', 'channol',
2283 'dicts', 'duct',
2284 'floats', 'floot',
2285 'funcs', 'funk',
2286 'jobs', 'jop',
Bram Moolenaar2984ed32022-08-20 14:51:17 +01002287 'lists', 'last',
Bram Moolenaar31d99482022-05-26 22:24:43 +01002288 'numbers', 'numbar',
2289 'strings', 'strung',
2290 'voids', 'viod']
2291 v9.CheckDefAndScriptFailure([$'var foo: {type}'], 'E1010:', 1)
2292 endfor
Bram Moolenaar55759b52020-09-30 22:59:42 +02002293enddef
2294
Bram Moolenaare88c6b72022-02-15 15:37:11 +00002295def Test_var_declaration_inferred()
2296 # check that type is set on the list so that extend() fails
2297 var lines =<< trim END
2298 vim9script
2299 def GetList(): list<number>
2300 var l = [1, 2, 3]
2301 return l
2302 enddef
2303 echo GetList()->extend(['x'])
2304 END
2305 v9.CheckScriptFailure(lines, 'E1013:', 6)
Bram Moolenaar31d99482022-05-26 22:24:43 +01002306
2307 lines =<< trim END
2308 vim9script
2309 def GetNr(): number
2310 return 5
2311 enddef
2312 def TestOne()
2313 var some = [function('len'), GetNr]
2314 g:res = typename(some)
2315 enddef
2316 TestOne()
2317 assert_equal('list<func(): number>', g:res)
2318
2319 def TestTwo()
2320 var some = [function('len'), GetNr]
2321 g:res = typename(some)
2322 enddef
2323 TestTwo()
2324 assert_equal('list<func(): number>', g:res)
2325 unlet g:res
2326
2327 # FIXME: why is the type different?
2328 var first = [function('len'), GetNr]
2329 assert_equal('list<func(...): number>', typename(first))
2330 var second = [GetNr, function('len')]
2331 assert_equal('list<func(...): number>', typename(second))
2332 END
2333 v9.CheckScriptSuccess(lines)
Bram Moolenaare88c6b72022-02-15 15:37:11 +00002334enddef
2335
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002336def Test_script_local_in_legacy()
Bram Moolenaara749a422022-02-12 19:52:25 +00002337 # OK to define script-local later but before compiling
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002338 var lines =<< trim END
2339 def SetLater()
Bram Moolenaara749a422022-02-12 19:52:25 +00002340 legvar = 'two'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002341 enddef
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02002342 let s:legvar = 'one'
Bram Moolenaara749a422022-02-12 19:52:25 +00002343 defcompile
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002344 call SetLater()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02002345 call assert_equal('two', s:legvar)
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002346 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002347 v9.CheckScriptSuccess(lines)
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002348
2349 # OK to leave out s: prefix when script-local already defined
2350 lines =<< trim END
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02002351 let s:legvar = 'one'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002352 def SetNoPrefix()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02002353 legvar = 'two'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002354 enddef
2355 call SetNoPrefix()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02002356 call assert_equal('two', s:legvar)
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002357 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002358 v9.CheckScriptSuccess(lines)
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002359
Bram Moolenaara749a422022-02-12 19:52:25 +00002360 # Not OK to leave out s: prefix when script-local defined after compiling
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002361 lines =<< trim END
2362 def SetLaterNoPrefix()
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02002363 legvar = 'two'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002364 enddef
2365 defcompile
Bram Moolenaar96cf4ba2021-04-24 14:15:41 +02002366 let s:legvar = 'one'
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002367 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002368 v9.CheckScriptFailure(lines, 'E476:', 1)
Bram Moolenaar47bc9c32021-07-17 21:24:56 +02002369
Bram Moolenaarb18b4962022-09-02 21:55:50 +01002370 edit! Xslfile
Bram Moolenaar47bc9c32021-07-17 21:24:56 +02002371 lines =<< trim END
2372 var edit: bool
2373 legacy edit
2374 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002375 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar643ce6c2021-04-06 21:17:27 +02002376enddef
2377
Bram Moolenaar083966f2020-12-08 11:29:40 +01002378def Test_var_type_check()
Bram Moolenaar55759b52020-09-30 22:59:42 +02002379 var lines =<< trim END
2380 vim9script
2381 var name: string
2382 name = 1234
2383 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002384 v9.CheckScriptFailure(lines, 'E1012:')
Bram Moolenaar55759b52020-09-30 22:59:42 +02002385
2386 lines =<< trim END
2387 vim9script
2388 var name:string
2389 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002390 v9.CheckScriptFailure(lines, 'E1069:')
Bram Moolenaar55759b52020-09-30 22:59:42 +02002391
Bram Moolenaar62aec932022-01-29 21:45:34 +00002392 v9.CheckDefAndScriptFailure(['var n:number = 42'], 'E1069:')
Bram Moolenaar60faf862021-08-23 22:22:45 +02002393
Bram Moolenaar55759b52020-09-30 22:59:42 +02002394 lines =<< trim END
2395 vim9script
2396 var name: asdf
2397 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002398 v9.CheckScriptFailure(lines, 'E1010:')
Bram Moolenaar55759b52020-09-30 22:59:42 +02002399
2400 lines =<< trim END
2401 vim9script
Bram Moolenaara749a422022-02-12 19:52:25 +00002402 var l: list<number>
2403 l = []
Bram Moolenaar55759b52020-09-30 22:59:42 +02002404 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002405 v9.CheckScriptSuccess(lines)
Bram Moolenaar55759b52020-09-30 22:59:42 +02002406
2407 lines =<< trim END
2408 vim9script
Bram Moolenaara749a422022-02-12 19:52:25 +00002409 var d: dict<number>
2410 d = {}
Bram Moolenaar55759b52020-09-30 22:59:42 +02002411 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002412 v9.CheckScriptSuccess(lines)
Bram Moolenaar378697a2021-07-15 19:23:18 +02002413
2414 lines =<< trim END
2415 vim9script
2416 var d = {a: 1, b: [2]}
2417 def Func(b: bool)
2418 var l: list<number> = b ? d.b : [3]
2419 enddef
2420 defcompile
2421 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002422 v9.CheckScriptSuccess(lines)
Bram Moolenaar55759b52020-09-30 22:59:42 +02002423enddef
2424
2425let g:dict_number = #{one: 1, two: 2}
2426
Bram Moolenaar083966f2020-12-08 11:29:40 +01002427def Test_var_list_dict_type()
Bram Moolenaar55759b52020-09-30 22:59:42 +02002428 var ll: list<number>
2429 ll = [1, 2, 2, 3, 3, 3]->uniq()
2430 ll->assert_equal([1, 2, 3])
2431
2432 var dd: dict<number>
2433 dd = g:dict_number
2434 dd->assert_equal(g:dict_number)
2435
2436 var lines =<< trim END
2437 var ll: list<number>
2438 ll = [1, 2, 3]->map('"one"')
2439 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002440 v9.CheckDefExecFailure(lines, 'E1012: Type mismatch; expected list<number> but got list<string>')
Bram Moolenaar55759b52020-09-30 22:59:42 +02002441enddef
2442
Bram Moolenaar083966f2020-12-08 11:29:40 +01002443def Test_cannot_use_let()
Bram Moolenaar62aec932022-01-29 21:45:34 +00002444 v9.CheckDefAndScriptFailure(['let a = 34'], 'E1126:', 1)
Bram Moolenaar083966f2020-12-08 11:29:40 +01002445enddef
2446
Bram Moolenaar55759b52020-09-30 22:59:42 +02002447def Test_unlet()
2448 g:somevar = 'yes'
2449 assert_true(exists('g:somevar'))
2450 unlet g:somevar
2451 assert_false(exists('g:somevar'))
2452 unlet! g:somevar
2453
2454 # also works for script-local variable in legacy Vim script
2455 s:somevar = 'legacy'
2456 assert_true(exists('s:somevar'))
2457 unlet s:somevar
2458 assert_false(exists('s:somevar'))
2459 unlet! s:somevar
2460
Bram Moolenaar83d0cec2022-02-04 21:17:58 +00002461 if 0
2462 unlet g:does_not_exist
2463 endif
2464
2465 v9.CheckDefExecFailure(['unlet v:notfound.key'], 'E1001:')
2466
Bram Moolenaar62aec932022-01-29 21:45:34 +00002467 v9.CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01002468 'var dd = 111',
2469 'unlet dd',
2470 ], 'E1081:', 2)
2471
Bram Moolenaar752fc692021-01-04 21:57:11 +01002472 # dict unlet
Bram Moolenaarea5c8982022-02-17 14:42:02 +00002473 var dd = {a: 1, b: 2, c: 3, 4: 4}
Bram Moolenaar752fc692021-01-04 21:57:11 +01002474 unlet dd['a']
2475 unlet dd.c
Bram Moolenaarea5c8982022-02-17 14:42:02 +00002476 unlet dd[4]
Bram Moolenaar752fc692021-01-04 21:57:11 +01002477 assert_equal({b: 2}, dd)
2478
Bram Moolenaar6b8c7ba2022-03-20 17:46:06 +00002479 # null key works like empty string
2480 dd = {'': 1, x: 9}
2481 unlet dd[null_string]
2482 assert_equal({x: 9}, dd)
2483
Bram Moolenaar752fc692021-01-04 21:57:11 +01002484 # list unlet
2485 var ll = [1, 2, 3, 4]
2486 unlet ll[1]
2487 unlet ll[-1]
2488 assert_equal([1, 3], ll)
2489
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01002490 ll = [1, 2, 3, 4]
2491 unlet ll[0 : 1]
2492 assert_equal([3, 4], ll)
2493
2494 ll = [1, 2, 3, 4]
2495 unlet ll[2 : 8]
2496 assert_equal([1, 2], ll)
2497
2498 ll = [1, 2, 3, 4]
2499 unlet ll[-2 : -1]
2500 assert_equal([1, 2], ll)
2501
Bram Moolenaar0c7f2612022-02-17 19:44:07 +00002502 g:nrdict = {1: 1, 2: 2}
2503 g:idx = 1
2504 unlet g:nrdict[g:idx]
2505 assert_equal({2: 2}, g:nrdict)
2506 unlet g:nrdict
2507 unlet g:idx
2508
Bram Moolenaar62aec932022-01-29 21:45:34 +00002509 v9.CheckDefFailure([
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01002510 'var ll = [1, 2]',
2511 'll[1 : 2] = 7',
Bram Moolenaar4f0884d2021-08-11 21:49:23 +02002512 ], 'E1012: Type mismatch; expected list<number> but got number', 2)
Bram Moolenaar62aec932022-01-29 21:45:34 +00002513 v9.CheckDefFailure([
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01002514 'var dd = {a: 1}',
2515 'unlet dd["a" : "a"]',
2516 ], 'E1166:', 2)
Bram Moolenaar62aec932022-01-29 21:45:34 +00002517 v9.CheckDefExecFailure([
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01002518 'unlet g:adict[0 : 1]',
2519 ], 'E1148:', 1)
Bram Moolenaar62aec932022-01-29 21:45:34 +00002520 v9.CheckDefFailure([
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01002521 'var ll = [1, 2]',
2522 'unlet ll[0:1]',
2523 ], 'E1004:', 2)
Bram Moolenaar62aec932022-01-29 21:45:34 +00002524 v9.CheckDefFailure([
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01002525 'var ll = [1, 2]',
2526 'unlet ll[0 :1]',
2527 ], 'E1004:', 2)
Bram Moolenaar62aec932022-01-29 21:45:34 +00002528 v9.CheckDefFailure([
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01002529 'var ll = [1, 2]',
2530 'unlet ll[0: 1]',
2531 ], 'E1004:', 2)
Bram Moolenaare08be092022-02-17 13:08:26 +00002532
2533 v9.CheckDefExecFailure([
2534 'g:ll = [1, 2]',
2535 'g:idx = "x"',
2536 'unlet g:ll[g:idx]',
2537 ], 'E1029: Expected number but got string', 3)
2538
2539 v9.CheckDefExecFailure([
2540 'g:ll = [1, 2, 3]',
2541 'g:idx = "x"',
2542 'unlet g:ll[g:idx : 2]',
2543 ], 'E1029: Expected number but got string', 3)
2544
2545 v9.CheckDefExecFailure([
2546 'g:ll = [1, 2, 3]',
2547 'g:idx = "x"',
2548 'unlet g:ll[0 : g:idx]',
2549 ], 'E1029: Expected number but got string', 3)
2550
Bram Moolenaar7f2c3412021-11-29 16:01:49 +00002551 # command recognized as assignment when skipping, should not give an error
Bram Moolenaar62aec932022-01-29 21:45:34 +00002552 v9.CheckScriptSuccess([
Bram Moolenaar7f2c3412021-11-29 16:01:49 +00002553 'vim9script',
2554 'for i in []',
2555 " put =''",
2556 'endfor'])
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01002557
Bram Moolenaar62aec932022-01-29 21:45:34 +00002558 v9.CheckDefFailure([
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01002559 'var ll = [1, 2]',
2560 'unlet ll["x" : 1]',
2561 ], 'E1012:', 2)
Bram Moolenaar62aec932022-01-29 21:45:34 +00002562 v9.CheckDefFailure([
Bram Moolenaar5b5ae292021-02-20 17:04:02 +01002563 'var ll = [1, 2]',
2564 'unlet ll[0 : "x"]',
2565 ], 'E1012:', 2)
2566
Bram Moolenaar752fc692021-01-04 21:57:11 +01002567 # list of dict unlet
2568 var dl = [{a: 1, b: 2}, {c: 3}]
2569 unlet dl[0]['b']
2570 assert_equal([{a: 1}, {c: 3}], dl)
2571
Bram Moolenaar62aec932022-01-29 21:45:34 +00002572 v9.CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01002573 'var ll = test_null_list()',
2574 'unlet ll[0]',
2575 ], 'E684:', 2)
Bram Moolenaar62aec932022-01-29 21:45:34 +00002576 v9.CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01002577 'var ll = [1]',
2578 'unlet ll[2]',
2579 ], 'E684:', 2)
Bram Moolenaar62aec932022-01-29 21:45:34 +00002580 v9.CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01002581 'var ll = [1]',
2582 'unlet ll[g:astring]',
Bram Moolenaarf30a14d2021-01-17 21:51:24 +01002583 ], 'E1012:', 2)
Bram Moolenaar62aec932022-01-29 21:45:34 +00002584 v9.CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01002585 'var dd = test_null_dict()',
2586 'unlet dd["a"]',
2587 ], 'E716:', 2)
Bram Moolenaar62aec932022-01-29 21:45:34 +00002588 v9.CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01002589 'var dd = {a: 1}',
2590 'unlet dd["b"]',
2591 ], 'E716:', 2)
Bram Moolenaar62aec932022-01-29 21:45:34 +00002592 v9.CheckDefExecFailure([
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01002593 'var dd = {a: 1}',
2594 'unlet dd[g:alist]',
2595 ], 'E1105:', 2)
Bram Moolenaar752fc692021-01-04 21:57:11 +01002596
Bram Moolenaarea5c8982022-02-17 14:42:02 +00002597 v9.CheckDefExecFailure([
Bram Moolenaar2984ed32022-08-20 14:51:17 +01002598 'g:dd = {"a": 1, 2: 2}',
Bram Moolenaarea5c8982022-02-17 14:42:02 +00002599 'unlet g:dd[0z11]',
2600 ], 'E1029:', 2)
Bram Moolenaar6296d1e2022-02-17 16:30:11 +00002601 v9.CheckDefExecFailure([
Bram Moolenaar2984ed32022-08-20 14:51:17 +01002602 'g:str = "a string"',
Bram Moolenaar6296d1e2022-02-17 16:30:11 +00002603 'unlet g:str[0]',
2604 ], 'E1148: Cannot index a string', 2)
Bram Moolenaarea5c8982022-02-17 14:42:02 +00002605
Bram Moolenaar2ef951d2021-01-03 20:55:26 +01002606 # can compile unlet before variable exists
Bram Moolenaar752fc692021-01-04 21:57:11 +01002607 g:someDict = {key: 'val'}
2608 var k = 'key'
2609 unlet g:someDict[k]
2610 assert_equal({}, g:someDict)
2611 unlet g:someDict
2612 assert_false(exists('g:someDict'))
Bram Moolenaar2ef951d2021-01-03 20:55:26 +01002613
Bram Moolenaar62aec932022-01-29 21:45:34 +00002614 v9.CheckScriptFailure([
Bram Moolenaar55759b52020-09-30 22:59:42 +02002615 'vim9script',
2616 'var svar = 123',
2617 'unlet svar',
2618 ], 'E1081:')
Bram Moolenaar62aec932022-01-29 21:45:34 +00002619 v9.CheckScriptFailure([
Bram Moolenaar55759b52020-09-30 22:59:42 +02002620 'vim9script',
2621 'var svar = 123',
2622 'unlet s:svar',
Bram Moolenaara749a422022-02-12 19:52:25 +00002623 ], 'E1268:')
Bram Moolenaar62aec932022-01-29 21:45:34 +00002624 v9.CheckScriptFailure([
Bram Moolenaar55759b52020-09-30 22:59:42 +02002625 'vim9script',
2626 'var svar = 123',
2627 'def Func()',
2628 ' unlet svar',
2629 'enddef',
2630 'defcompile',
2631 ], 'E1081:')
Bram Moolenaar62aec932022-01-29 21:45:34 +00002632 v9.CheckScriptFailure([
Bram Moolenaar55759b52020-09-30 22:59:42 +02002633 'vim9script',
2634 'var svar = 123',
Bram Moolenaar9aed7292020-12-18 15:38:00 +01002635 'func Func()',
2636 ' unlet s:svar',
2637 'endfunc',
2638 'Func()',
2639 ], 'E1081:')
Bram Moolenaar62aec932022-01-29 21:45:34 +00002640 v9.CheckScriptFailure([
Bram Moolenaar9aed7292020-12-18 15:38:00 +01002641 'vim9script',
2642 'var svar = 123',
Bram Moolenaar55759b52020-09-30 22:59:42 +02002643 'def Func()',
2644 ' unlet s:svar',
2645 'enddef',
2646 'defcompile',
2647 ], 'E1081:')
2648
Bram Moolenaar62aec932022-01-29 21:45:34 +00002649 v9.CheckScriptFailure([
Bram Moolenaar71b76852021-12-17 20:15:38 +00002650 'vim9script',
2651 'def Delcount(dict: dict<any>)',
2652 ' unlet dict.count',
2653 'enddef',
2654 'Delcount(v:)',
2655 ], 'E742:')
2656
Bram Moolenaar62aec932022-01-29 21:45:34 +00002657 v9.CheckScriptFailure([
Bram Moolenaar71b76852021-12-17 20:15:38 +00002658 'vim9script',
2659 'def DelChangedtick(dict: dict<any>)',
2660 ' unlet dict.changedtick',
2661 'enddef',
2662 'DelChangedtick(b:)',
2663 ], 'E795:')
2664
Bram Moolenaar0e9bdad2022-10-15 20:06:33 +01002665 writefile(['vim9script', 'export var svar = 1234'], 'XunletExport.vim', 'D')
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01002666 var lines =<< trim END
2667 vim9script
Bram Moolenaard5f400c2022-01-06 21:10:28 +00002668 import './XunletExport.vim' as exp
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01002669 def UnletSvar()
Bram Moolenaard5f400c2022-01-06 21:10:28 +00002670 unlet exp.svar
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01002671 enddef
2672 defcompile
2673 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002674 v9.CheckScriptFailure(lines, 'E1260:', 1)
Bram Moolenaar0acbf5a2021-01-05 20:58:25 +01002675
Bram Moolenaar55759b52020-09-30 22:59:42 +02002676 $ENVVAR = 'foobar'
2677 assert_equal('foobar', $ENVVAR)
2678 unlet $ENVVAR
2679 assert_equal('', $ENVVAR)
2680enddef
2681
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01002682def Test_expr_error_no_assign()
2683 var lines =<< trim END
2684 vim9script
2685 var x = invalid
2686 echo x
2687 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002688 v9.CheckScriptFailureList(lines, ['E121:', 'E121:'])
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01002689
2690 lines =<< trim END
2691 vim9script
2692 var x = 1 / 0
2693 echo x
2694 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002695 v9.CheckScriptFailure(lines, 'E1154:')
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01002696
2697 lines =<< trim END
2698 vim9script
2699 var x = 1 % 0
2700 echo x
2701 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002702 v9.CheckScriptFailure(lines, 'E1154:')
Bram Moolenaarccc25aa2021-03-26 21:27:52 +01002703
2704 lines =<< trim END
2705 var x: string 'string'
2706 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002707 v9.CheckDefAndScriptFailure(lines, 'E488:')
Bram Moolenaarc5f59fa2021-01-21 12:34:14 +01002708enddef
2709
2710
Bram Moolenaar17126b12021-01-07 22:03:02 +01002711def Test_assign_command_modifier()
2712 var lines =<< trim END
Bram Moolenaar9e0f8832021-01-09 12:09:22 +01002713 var verbose = 0
2714 verbose = 1
2715 assert_equal(1, verbose)
2716 silent verbose = 2
2717 assert_equal(2, verbose)
2718 silent verbose += 2
2719 assert_equal(4, verbose)
2720 silent verbose -= 1
2721 assert_equal(3, verbose)
Bram Moolenaar17126b12021-01-07 22:03:02 +01002722
Bram Moolenaar9e0f8832021-01-09 12:09:22 +01002723 var topleft = {one: 1}
2724 sandbox topleft.one = 3
2725 assert_equal({one: 3}, topleft)
2726 leftabove topleft[' '] = 4
2727 assert_equal({one: 3, ' ': 4}, topleft)
2728
2729 var x: number
2730 var y: number
2731 silent [x, y] = [1, 2]
2732 assert_equal(1, x)
2733 assert_equal(2, y)
Bram Moolenaar17126b12021-01-07 22:03:02 +01002734 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002735 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar17126b12021-01-07 22:03:02 +01002736enddef
2737
Bram Moolenaar74f4a962021-06-17 21:03:07 +02002738def Test_assign_alt_buf_register()
2739 var lines =<< trim END
2740 edit 'file_b1'
2741 var b1 = bufnr()
2742 edit 'file_b2'
2743 var b2 = bufnr()
2744 assert_equal(b1, bufnr('#'))
2745 @# = b2
2746 assert_equal(b2, bufnr('#'))
2747 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002748 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar74f4a962021-06-17 21:03:07 +02002749enddef
2750
Bram Moolenaar32154662021-03-28 21:14:06 +02002751def Test_script_funcref_case()
2752 var lines =<< trim END
2753 var Len = (s: string): number => len(s) + 1
2754 assert_equal(5, Len('asdf'))
2755 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002756 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar32154662021-03-28 21:14:06 +02002757
2758 lines =<< trim END
2759 var len = (s: string): number => len(s) + 1
2760 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002761 v9.CheckDefAndScriptFailure(lines, 'E704:')
Bram Moolenaar32154662021-03-28 21:14:06 +02002762
2763 lines =<< trim END
2764 vim9script
Bram Moolenaara749a422022-02-12 19:52:25 +00002765 var Len = (s: string): number => len(s) + 2
Bram Moolenaar32154662021-03-28 21:14:06 +02002766 assert_equal(6, Len('asdf'))
2767 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002768 v9.CheckScriptSuccess(lines)
Bram Moolenaar32154662021-03-28 21:14:06 +02002769
2770 lines =<< trim END
2771 vim9script
Bram Moolenaara749a422022-02-12 19:52:25 +00002772 var len = (s: string): number => len(s) + 1
Bram Moolenaar32154662021-03-28 21:14:06 +02002773 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002774 v9.CheckScriptFailure(lines, 'E704:')
Bram Moolenaar32154662021-03-28 21:14:06 +02002775enddef
2776
Bram Moolenaar44a89772021-12-18 12:31:33 +00002777def Test_script_funcref_runtime_type_check()
2778 var lines =<< trim END
2779 vim9script
2780 def FuncWithNumberArg(n: number)
2781 enddef
2782 def Test()
2783 var Ref: func(string) = function(FuncWithNumberArg)
2784 enddef
2785 defcompile
2786 END
2787 # OK at compile time
Bram Moolenaar62aec932022-01-29 21:45:34 +00002788 v9.CheckScriptSuccess(lines)
Bram Moolenaar44a89772021-12-18 12:31:33 +00002789
2790 # Type check fails at runtime
Bram Moolenaar62aec932022-01-29 21:45:34 +00002791 v9.CheckScriptFailure(lines + ['Test()'], 'E1012: Type mismatch; expected func(string) but got func(number)')
Bram Moolenaar44a89772021-12-18 12:31:33 +00002792enddef
2793
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002794def Test_inc_dec()
2795 var lines =<< trim END
2796 var nr = 7
2797 ++nr
Bram Moolenaar730bf302021-04-24 20:43:56 +02002798 assert_equal(8, nr)
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002799 --nr
Bram Moolenaar730bf302021-04-24 20:43:56 +02002800 assert_equal(7, nr)
Bram Moolenaarf3d30842021-06-25 19:29:30 +02002801 ++nr | ++nr
2802 assert_equal(9, nr)
2803 ++nr # comment
2804 assert_equal(10, nr)
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002805
2806 var ll = [1, 2]
2807 --ll[0]
2808 ++ll[1]
Bram Moolenaar730bf302021-04-24 20:43:56 +02002809 assert_equal([0, 3], ll)
2810
2811 g:count = 1
2812 ++g:count
2813 --g:count
2814 assert_equal(1, g:count)
2815 unlet g:count
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002816 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002817 v9.CheckDefAndScriptSuccess(lines)
Bram Moolenaar22480d12021-06-25 21:31:09 +02002818
2819 lines =<< trim END
2820 var nr = 7
2821 ++ nr
2822 END
Bram Moolenaar62aec932022-01-29 21:45:34 +00002823 v9.CheckDefAndScriptFailure(lines, "E1202: No white space allowed after '++': ++ nr")
Bram Moolenaarbdc0f1c2021-04-24 19:08:24 +02002824enddef
2825
Bram Moolenaar327d3ee2021-07-28 19:34:14 +02002826def Test_abort_after_error()
2827 # should abort after strpart() fails, not give another type error
2828 var lines =<< trim END
2829 vim9script
2830 var x: string
2831 x = strpart(1, 2)
2832 END
Bram Moolenaar0e9bdad2022-10-15 20:06:33 +01002833 writefile(lines, 'Xtestscript', 'D')
Bram Moolenaar327d3ee2021-07-28 19:34:14 +02002834 var expected = 'E1174: String required for argument 1'
2835 assert_fails('so Xtestscript', [expected, expected], 3)
Bram Moolenaar327d3ee2021-07-28 19:34:14 +02002836enddef
2837
Bram Moolenaarafa048f2022-02-22 20:43:36 +00002838def Test_using_s_var_in_function()
2839 var lines =<< trim END
2840 vim9script
2841 var scriptlevel = 123
2842 def SomeFunc()
2843 echo s:scriptlevel
2844 enddef
2845 SomeFunc()
2846 END
2847 v9.CheckScriptFailure(lines, 'E1268:')
2848
2849 # OK in legacy script
2850 lines =<< trim END
2851 let s:scriptlevel = 123
2852 def s:SomeFunc()
2853 echo s:scriptlevel
2854 enddef
2855 call s:SomeFunc()
2856 END
2857 v9.CheckScriptSuccess(lines)
2858
2859 lines =<< trim END
2860 vim9script
2861 var scriptlevel = 123
2862 def SomeFunc()
2863 s:scriptlevel = 456
2864 enddef
2865 SomeFunc()
2866 END
2867 v9.CheckScriptFailure(lines, 'E1268:')
2868
2869 # OK in legacy script
2870 lines =<< trim END
2871 let s:scriptlevel = 123
2872 def s:SomeFunc()
2873 s:scriptlevel = 456
2874 enddef
2875 call s:SomeFunc()
2876 call assert_equal(456, s:scriptlevel)
2877 END
2878 v9.CheckScriptSuccess(lines)
2879enddef
2880
Yegappan Lakshmananb5a07192023-10-05 20:14:43 +02002881" Test for specifying a type in assignment
2882def Test_type_specification_in_assignment()
2883 # specify type for an existing script local variable without "var"
2884 var lines =<< trim END
2885 vim9script
2886 var n: number = 10
2887 n: number = 20
2888 END
2889 v9.CheckSourceFailure(lines, 'E488: Trailing characters: : number = 20', 3)
2890
2891 # specify type for a non-existing script local variable without "var"
2892 lines =<< trim END
2893 vim9script
2894 MyVar: string = 'abc'
2895 END
2896 v9.CheckSourceFailure(lines, "E492: Not an editor command: MyVar: string = 'abc'", 2)
2897
2898 # specify type for an existing def local variable without "var"
2899 lines =<< trim END
2900 vim9script
2901 def Foo()
2902 var n: number = 10
2903 n: number = 20
2904 enddef
2905 Foo()
2906 END
2907 v9.CheckSourceFailure(lines, 'E488: Trailing characters: : number = 20', 2)
2908
2909 # specify type for a non-existing def local variable without "var"
2910 lines =<< trim END
2911 vim9script
2912 def Foo()
2913 MyVar: string = 'abc'
2914 enddef
2915 Foo()
2916 END
2917 v9.CheckSourceFailure(lines, "E476: Invalid command: MyVar: string = 'abc'", 1)
2918enddef
2919
Yegappan Lakshmananefbfa862022-04-17 12:47:40 +01002920let g:someVar = 'X'
2921
2922" Test for heredoc with Vim expressions.
2923" This messes up highlighting, keep it near the end.
2924def Test_heredoc_expr()
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01002925 var lines =<< trim CODE
2926 var s = "local"
2927 var a1 = "1"
2928 var a2 = "2"
2929 var a3 = "3"
2930 var a4 = ""
2931 var code =<< trim eval END
LemonBoy2eaef102022-05-06 13:14:50 +01002932 var a = {5 + 10}
2933 var b = {min([10, 6])} + {max([4, 6])}
2934 var c = "{s}"
2935 var d = x{a1}x{a2}x{a3}x{a4}
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01002936 END
2937 assert_equal(['var a = 15', 'var b = 6 + 6', 'var c = "local"', 'var d = x1x2x3x'], code)
2938 CODE
2939 v9.CheckDefAndScriptSuccess(lines)
Yegappan Lakshmananefbfa862022-04-17 12:47:40 +01002940
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01002941 lines =<< trim CODE
2942 var code =<< eval trim END
LemonBoy2eaef102022-05-06 13:14:50 +01002943 var s = "{$SOME_ENV_VAR}"
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01002944 END
2945 assert_equal(['var s = "somemore"'], code)
2946 CODE
2947 v9.CheckDefAndScriptSuccess(lines)
Yegappan Lakshmananefbfa862022-04-17 12:47:40 +01002948
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01002949 lines =<< trim CODE
2950 var code =<< eval END
LemonBoy2eaef102022-05-06 13:14:50 +01002951 var s = "{$SOME_ENV_VAR}"
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01002952 END
2953 assert_equal([' var s = "somemore"'], code)
2954 CODE
2955 v9.CheckDefAndScriptSuccess(lines)
Yegappan Lakshmananefbfa862022-04-17 12:47:40 +01002956
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01002957 lines =<< trim CODE
2958 var code =<< eval trim END
LemonBoy2eaef102022-05-06 13:14:50 +01002959 let a = {{abc}}
2960 let b = {g:someVar}
2961 let c = {{
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01002962 END
LemonBoy2eaef102022-05-06 13:14:50 +01002963 assert_equal(['let a = {abc}', 'let b = X', 'let c = {'], code)
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01002964 CODE
2965 v9.CheckDefAndScriptSuccess(lines)
Yegappan Lakshmananefbfa862022-04-17 12:47:40 +01002966
Yegappan Lakshmanan1fc6ea92022-04-21 23:30:15 +01002967 lines =<< trim LINES
Yegappan Lakshmananefbfa862022-04-17 12:47:40 +01002968 var text =<< eval trim END
LemonBoy2eaef102022-05-06 13:14:50 +01002969 let b = {
Yegappan Lakshmananefbfa862022-04-17 12:47:40 +01002970 END
2971 LINES
LemonBoy2eaef102022-05-06 13:14:50 +01002972 v9.CheckDefAndScriptFailure(lines, "E1279: Missing '}'")
Yegappan Lakshmananefbfa862022-04-17 12:47:40 +01002973
2974 lines =<< trim LINES
2975 var text =<< eval trim END
LemonBoy2eaef102022-05-06 13:14:50 +01002976 let b = {abc
Yegappan Lakshmananefbfa862022-04-17 12:47:40 +01002977 END
2978 LINES
LemonBoy2eaef102022-05-06 13:14:50 +01002979 v9.CheckDefAndScriptFailure(lines, "E1279: Missing '}'")
Yegappan Lakshmananefbfa862022-04-17 12:47:40 +01002980
2981 lines =<< trim LINES
2982 var text =<< eval trim END
LemonBoy2eaef102022-05-06 13:14:50 +01002983 let b = {}
Yegappan Lakshmananefbfa862022-04-17 12:47:40 +01002984 END
2985 LINES
LemonBoy2eaef102022-05-06 13:14:50 +01002986 v9.CheckDefAndScriptFailure(lines, 'E15: Invalid expression: "}"')
Yegappan Lakshmananefbfa862022-04-17 12:47:40 +01002987enddef
2988
Yegappan Lakshmananc229a6a2023-10-26 23:05:07 +02002989" Test for assigning to a multi-dimensional list item.
2990def Test_list_item_assign()
2991 var lines =<< trim END
2992 vim9script
2993
2994 def Foo()
Yegappan Lakshmanan67098162023-11-05 10:07:03 +01002995 var l: list<list<string>> = [['x', 'x', 'x'], ['y', 'y', 'y']]
2996 var z: number = 1
Yegappan Lakshmananc229a6a2023-10-26 23:05:07 +02002997
Yegappan Lakshmanan67098162023-11-05 10:07:03 +01002998 [l[1][2], z] = ['a', 20]
2999 assert_equal([['x', 'x', 'x'], ['y', 'y', 'a']], l)
Yegappan Lakshmananc229a6a2023-10-26 23:05:07 +02003000 enddef
3001 Foo()
3002 END
3003 v9.CheckSourceSuccess(lines)
Yegappan Lakshmanan67098162023-11-05 10:07:03 +01003004
3005 lines =<< trim END
3006 vim9script
3007
3008 var l: list<list<string>> = [['x', 'x', 'x'], ['y', 'y', 'y']]
3009 var z: number = 1
3010
3011 [l[1][2], z] = ['a', 20]
3012 assert_equal([['x', 'x', 'x'], ['y', 'y', 'a']], l)
3013 END
3014 v9.CheckSourceSuccess(lines)
3015enddef
3016
3017" Test for assigning to a multi-dimensional dict item.
3018def Test_dict_item_assign()
3019 # This used to fail with the error "E1105: Cannot convert list to string"
3020 # (Github issue #13485)
3021 var lines =<< trim END
3022 vim9script
3023 def F()
3024 var d: dict<dict<number>> = {a: {b: 0}}
3025
3026 for group in keys(d)
3027 d['a']['b'] += 1
3028 endfor
3029 assert_equal({a: {b: 1}}, d)
3030 enddef
3031 F()
3032 END
3033 v9.CheckSourceSuccess(lines)
3034
3035 # This used to crash Vim
3036 lines =<< trim END
3037 vim9script
3038 def F()
3039 var d: dict<dict<number>> = {a: {b: 0}}
3040 d['a']['b'] += 1
3041 assert_equal({a: {b: 1}}, d)
3042 enddef
3043 F()
3044 END
3045 v9.CheckSourceSuccess(lines)
3046
3047 # Assignment at script level
3048 lines =<< trim END
3049 vim9script
3050 var d: dict<dict<number>> = {a: {b: 0}}
3051
3052 for group in keys(d)
3053 d['a']['b'] += 1
3054 endfor
3055 assert_equal({a: {b: 1}}, d)
3056 END
3057 v9.CheckSourceSuccess(lines)
Yegappan Lakshmananc229a6a2023-10-26 23:05:07 +02003058enddef
3059
Ernie Rael9ed53752023-12-11 17:40:46 +01003060def Test_class_assign()
3061 var lines =<< trim END
3062 vim9script
3063 class C
3064 endclass
3065 class D
3066 endclass
Ernie Rael2025af12023-12-12 16:58:00 +01003067 assert_fails('C = D', 'E1405: Class "D" cannot be used as a value')
Ernie Rael9ed53752023-12-11 17:40:46 +01003068 END
3069 v9.CheckSourceSuccess(lines)
3070enddef
3071
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003072" Test for using various types (dict, list, blob, funcref, class) as variable
3073" in assignments with a different type
3074def Test_type_check()
3075 var lines =<< trim END
3076 vim9script
3077 class A
3078 endclass
Ernie Raelb5011082023-12-18 08:34:41 +01003079 type T = number
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003080 var N: number = 1
3081 var S: string = 'abc'
3082 var d: dict<number> = {}
3083 var l: list<number> = []
3084 var b: blob = 0z10
3085 var Fn: func = function('min')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003086 var o: A = A.new()
3087
3088 # Assign a number
3089 assert_fails('d = N', 'E1012: Type mismatch; expected dict<number> but got number')
3090 assert_fails('l = N', 'E1012: Type mismatch; expected list<number> but got number')
3091 assert_fails('b = N', 'E1012: Type mismatch; expected blob but got number')
3092 assert_fails('Fn = N', 'E1012: Type mismatch; expected func(...): unknown but got number')
Ernie Raele75fde62023-12-21 17:18:54 +01003093 assert_fails('A = N', 'E1405: Class "A" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003094 assert_fails('o = N', 'E1012: Type mismatch; expected object<A> but got number')
Ernie Raele75fde62023-12-21 17:18:54 +01003095 assert_fails('T = N', 'E1403: Type alias "T" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003096
3097 # Use a compound operator with different LHS types
3098 assert_fails('d += N', 'E734: Wrong variable type for +=')
3099 assert_fails('l += N', 'E734: Wrong variable type for +=')
3100 assert_fails('b += N', 'E734: Wrong variable type for +=')
3101 assert_fails('Fn += N', 'E734: Wrong variable type for +=')
Ernie Raele75fde62023-12-21 17:18:54 +01003102 assert_fails('A += N', 'E1405: Class "A" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003103 assert_fails('o += N', 'E734: Wrong variable type for +=')
Ernie Raele75fde62023-12-21 17:18:54 +01003104 assert_fails('T += N', 'E1403: Type alias "T" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003105
3106 # Assign to a number variable
3107 assert_fails('N = d', 'E1012: Type mismatch; expected number but got dict<number>')
3108 assert_fails('N = l', 'E1012: Type mismatch; expected number but got list<number>')
3109 assert_fails('N = b', 'E1012: Type mismatch; expected number but got blob')
3110 assert_fails('N = Fn', 'E1012: Type mismatch; expected number but got func([unknown]): number')
Ernie Rael2025af12023-12-12 16:58:00 +01003111 assert_fails('N = A', 'E1405: Class "A" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003112 assert_fails('N = o', 'E1012: Type mismatch; expected number but got object<A>')
Ernie Raelb5011082023-12-18 08:34:41 +01003113 assert_fails('N = T', 'E1403: Type alias "T" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003114
3115 # Use a compound operator with different RHS types
3116 assert_fails('N += d', 'E734: Wrong variable type for +=')
3117 assert_fails('N += l', 'E734: Wrong variable type for +=')
3118 assert_fails('N += b', 'E974: Using a Blob as a Number')
3119 assert_fails('N += Fn', 'E734: Wrong variable type for +=')
Ernie Rael2025af12023-12-12 16:58:00 +01003120 assert_fails('N += A', 'E1405: Class "A" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003121 assert_fails('N += o', 'E1320: Using an Object as a Number')
Ernie Raelb5011082023-12-18 08:34:41 +01003122 assert_fails('N += T', 'E1403: Type alias "T" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003123
3124 # Initialize multiple variables using []
3125 assert_fails('var [X1: number, Y: number] = [1, d]', 'E1012: Type mismatch; expected number but got dict<number>')
3126 assert_fails('var [X2: number, Y: number] = [1, l]', 'E1012: Type mismatch; expected number but got list<number>')
3127 assert_fails('var [X3: number, Y: number] = [1, b]', 'E1012: Type mismatch; expected number but got blob')
3128 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 +01003129 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 +01003130 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 +01003131 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 +01003132
3133 # String concatenation with various LHS types
3134 assert_fails('S ..= d', 'E734: Wrong variable type for .=')
3135 assert_fails('S ..= l', 'E734: Wrong variable type for .=')
3136 assert_fails('S ..= b', 'E976: Using a Blob as a String')
3137 assert_fails('S ..= Fn', 'E734: Wrong variable type for .=')
Ernie Rael2025af12023-12-12 16:58:00 +01003138 assert_fails('S ..= A', 'E1405: Class "A" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003139 assert_fails('S ..= o', 'E1324: Using an Object as a String')
Ernie Raelb5011082023-12-18 08:34:41 +01003140 assert_fails('S ..= T', 'E1403: Type alias "T" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003141
3142 # String concatenation with various RHS types
3143 assert_fails('d ..= S', 'E734: Wrong variable type for .=')
3144 assert_fails('l ..= S', 'E734: Wrong variable type for .=')
3145 assert_fails('b ..= S', 'E734: Wrong variable type for .=')
3146 assert_fails('Fn ..= S', 'E734: Wrong variable type for .=')
Ernie Raele75fde62023-12-21 17:18:54 +01003147 assert_fails('A ..= S', 'E1405: Class "A" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003148 assert_fails('o ..= S', 'E734: Wrong variable type for .=')
Ernie Raele75fde62023-12-21 17:18:54 +01003149 assert_fails('T ..= S', 'E1403: Type alias "T" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003150 END
3151 v9.CheckSourceSuccess(lines)
Yegappan Lakshmanan3ee25962023-12-04 20:31:14 +01003152
3153 if has('channel')
3154 lines =<< trim END
3155 vim9script
3156 var N: number = 1
3157 var S: string = 'abc'
3158 var j: job = test_null_job()
3159 var ch: channel = test_null_channel()
3160 assert_fails('j = N', 'E1012: Type mismatch; expected job but got number')
3161 assert_fails('ch = N', 'E1012: Type mismatch; expected channel but got number')
3162 assert_fails('j += N', 'E734: Wrong variable type for +=')
3163 assert_fails('ch += N', 'E734: Wrong variable type for +=')
3164 assert_fails('N = j', 'E1012: Type mismatch; expected number but got job')
3165 assert_fails('N = ch', 'E1012: Type mismatch; expected number but got channel')
3166 assert_fails('N += j', 'E910: Using a Job as a Number')
3167 assert_fails('N += ch', 'E913: Using a Channel as a Number')
3168 assert_fails('var [X5: number, Y: number] = [1, j]', 'E1012: Type mismatch; expected number but got job')
3169 assert_fails('var [X6: number, Y: number] = [1, ch]', 'E1012: Type mismatch; expected number but got channel')
3170 assert_fails('S ..= j', 'E908: Using an invalid value as a String: job')
3171 assert_fails('S ..= ch', 'E908: Using an invalid value as a String: channel')
3172 assert_fails('j ..= S', 'E734: Wrong variable type for .=')
3173 assert_fails('ch ..= S', 'E734: Wrong variable type for .=')
3174 END
3175 v9.CheckSourceSuccess(lines)
3176 endif
Ernie Rael5e133152023-12-19 22:15:27 +01003177
3178 lines =<< trim END
3179 vim9script
3180 class A
3181 endclass
3182
3183 def F()
3184 A += 3
3185 enddef
3186 F()
3187 END
3188 v9.CheckScriptFailure(lines, 'E1405: Class "A" cannot be used as a value')
3189
3190 lines =<< trim END
3191 vim9script
3192 class A
3193 endclass
3194
3195 var o = A.new()
3196 def F()
3197 o += 4
3198 enddef
3199 F()
3200 END
3201 v9.CheckScriptFailure(lines, 'E1411: Missing dot after object "o"')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003202enddef
3203
3204" Test for checking the argument type of a def function
3205def Test_func_argtype_check()
3206 var lines =<< trim END
3207 vim9script
3208
3209 # Passing different types as argument to a function expecting a number
3210 def IntArg(n: number)
3211 enddef
3212
3213 class A
3214 endclass
3215 var N: number = 1
3216 var S: string = 'abc'
3217 var d: dict<number> = {}
3218 var l: list<number> = []
3219 var b: blob = 0z10
3220 var Fn: func = function('min')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003221 var o: A = A.new()
3222
3223 assert_fails('IntArg(d)', 'E1013: Argument 1: type mismatch, expected number but got dict<number>')
3224 assert_fails('IntArg(l)', 'E1013: Argument 1: type mismatch, expected number but got list<number>')
3225 assert_fails('IntArg(b)', 'E1013: Argument 1: type mismatch, expected number but got blob')
3226 assert_fails('IntArg(Fn)', 'E1013: Argument 1: type mismatch, expected number but got func([unknown]): number')
Yegappan Lakshmanan3ee25962023-12-04 20:31:14 +01003227 if has('channel')
3228 var j: job = test_null_job()
3229 var ch: channel = test_null_channel()
3230 assert_fails('IntArg(j)', 'E1013: Argument 1: type mismatch, expected number but got job')
3231 assert_fails('IntArg(ch)', 'E1013: Argument 1: type mismatch, expected number but got channel')
3232 endif
Ernie Raelb077b582023-12-14 20:11:44 +01003233 assert_fails('IntArg(A)', 'E1405: Class "A" cannot be used as a value')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003234 assert_fails('IntArg(o)', 'E1013: Argument 1: type mismatch, expected number but got object<A>')
3235
3236 # Passing a number to functions accepting different argument types
3237 def DictArg(_: dict<number>)
3238 enddef
3239 assert_fails('DictArg(N)', 'E1013: Argument 1: type mismatch, expected dict<number> but got number')
3240
3241 def ListArg(_: list<number>)
3242 enddef
3243 assert_fails('ListArg(N)', 'E1013: Argument 1: type mismatch, expected list<number> but got number')
3244
3245 def BlobArg(_: blob)
3246 enddef
3247 assert_fails('BlobArg(N)', 'E1013: Argument 1: type mismatch, expected blob but got number')
3248
3249 def FuncArg(Fn_arg: func)
3250 enddef
3251 assert_fails('FuncArg(N)', 'E1013: Argument 1: type mismatch, expected func(...): unknown but got number')
3252
Yegappan Lakshmanan3ee25962023-12-04 20:31:14 +01003253 if has('channel')
3254 def JobArg(_: job)
3255 enddef
3256 assert_fails('JobArg(N)', 'E1013: Argument 1: type mismatch, expected job but got number')
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003257
Yegappan Lakshmanan3ee25962023-12-04 20:31:14 +01003258 def ChannelArg(_: channel)
3259 enddef
3260 assert_fails('ChannelArg(N)', 'E1013: Argument 1: type mismatch, expected channel but got number')
3261 endif
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003262
3263 def ObjectArg(_: A)
3264 enddef
3265 assert_fails('ObjectArg(N)', 'E1013: Argument 1: type mismatch, expected object<A> but got number')
3266 END
3267 v9.CheckSourceSuccess(lines)
3268
3269 # Calling a function expecting a number type with different argument types
3270 # from another function
3271 var pre_lines =<< trim END
3272 vim9script
3273 class A
3274 endclass
3275 def IntArg(n: number)
3276 enddef
3277 def Foo()
3278 END
3279 var post_lines =<< trim END
3280 enddef
3281 defcompile
3282 END
3283 lines = pre_lines + ['var d: dict<number> = {}', 'IntArg(d)'] + post_lines
3284 v9.CheckSourceFailure(lines, 'E1013: Argument 1: type mismatch, expected number but got dict<number>', 2)
3285 lines = pre_lines + ['var l: list<number> = []', 'IntArg(l)'] + post_lines
3286 v9.CheckSourceFailure(lines, 'E1013: Argument 1: type mismatch, expected number but got list<number>', 2)
3287 lines = pre_lines + ['var b: blob = 0z12', 'IntArg(b)'] + post_lines
3288 v9.CheckSourceFailure(lines, 'E1013: Argument 1: type mismatch, expected number but got blob', 2)
3289 lines = pre_lines + ['var Fn: func = function("min")', 'IntArg(Fn)'] + post_lines
3290 v9.CheckSourceFailure(lines, 'E1013: Argument 1: type mismatch, expected number but got func(...): unknown', 2)
Yegappan Lakshmanan3ee25962023-12-04 20:31:14 +01003291 if has('channel')
3292 lines = pre_lines + ['var j: job = test_null_job()', 'IntArg(j)'] + post_lines
3293 v9.CheckSourceFailure(lines, 'E1013: Argument 1: type mismatch, expected number but got job', 2)
3294 lines = pre_lines + ['var ch: channel = test_null_channel()', 'IntArg(ch)'] + post_lines
3295 v9.CheckSourceFailure(lines, 'E1013: Argument 1: type mismatch, expected number but got channel', 2)
3296 endif
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003297 lines = pre_lines + ['IntArg(A)'] + post_lines
Ernie Raelb077b582023-12-14 20:11:44 +01003298 v9.CheckSourceFailure(lines, 'E1405: Class "A" cannot be used as a value', 1)
Yegappan Lakshmanan1fc47c22023-11-08 21:02:48 +01003299 lines = pre_lines + ['var o: A = A.new()', 'IntArg(o)'] + post_lines
3300 v9.CheckSourceFailure(lines, 'E1013: Argument 1: type mismatch, expected number but got object<A>', 2)
3301enddef
3302
3303" Test for checking the return type of a def function
3304def Test_func_rettype_check()
3305 var lines =<< trim END
3306 vim9script
3307 def Fn(): dict<number>
3308 return 10
3309 enddef
3310 defcompile
3311 END
3312 v9.CheckSourceFailure(lines, 'E1012: Type mismatch; expected dict<number> but got number', 1)
3313
3314 lines =<< trim END
3315 vim9script
3316 def Fn(): list<number>
3317 return 10
3318 enddef
3319 defcompile
3320 END
3321 v9.CheckSourceFailure(lines, 'E1012: Type mismatch; expected list<number> but got number', 1)
3322
3323 lines =<< trim END
3324 vim9script
3325 def Fn(): blob
3326 return 10
3327 enddef
3328 defcompile
3329 END
3330 v9.CheckSourceFailure(lines, 'E1012: Type mismatch; expected blob but got number', 1)
3331
3332 lines =<< trim END
3333 vim9script
3334 def Fn(): func
3335 return 10
3336 enddef
3337 defcompile
3338 END
3339 v9.CheckSourceFailure(lines, 'E1012: Type mismatch; expected func(...): unknown but got number', 1)
3340
3341 lines =<< trim END
3342 vim9script
3343 def Fn(): job
3344 return 10
3345 enddef
3346 defcompile
3347 END
3348 v9.CheckSourceFailure(lines, 'E1012: Type mismatch; expected job but got number', 1)
3349
3350 lines =<< trim END
3351 vim9script
3352 def Fn(): channel
3353 return 10
3354 enddef
3355 defcompile
3356 END
3357 v9.CheckSourceFailure(lines, 'E1012: Type mismatch; expected channel but got number', 1)
3358
3359 lines =<< trim END
3360 vim9script
3361 class A
3362 endclass
3363 def Fn(): A
3364 return 10
3365 enddef
3366 defcompile
3367 END
3368 v9.CheckSourceFailure(lines, 'E1012: Type mismatch; expected object<A> but got number', 1)
3369enddef
3370
Yegappan Lakshmanand3351852023-12-08 21:41:23 +01003371" Test for assigning different types of value to a variable of type "any"
3372def Test_assign_to_any()
3373 for [typestr, val] in [
3374 ["'bool'", 'true'],
3375 ["'number'", '100'],
3376 ["'float'", '1.1'],
3377 ["'string'", '"abc"'],
3378 ["'blob'", '0z10'],
3379 ["'list<number>'", '[1, 2, 3]'],
3380 ["'dict<number>'", '{a: 1}'],
3381 ]
3382 var lines =<< trim eval END
3383 vim9script
3384 var x: any = {val}
3385 assert_equal({typestr}, typename(x))
3386 x = [{{a: 1}}, {{b: 2}}]
3387 assert_equal('list<dict<number>>', typename(x))
3388 def Foo(xarg: any, s: string)
3389 assert_equal(s, typename(xarg))
3390 enddef
3391 Foo({val}, {typestr})
3392 END
3393 v9.CheckSourceSuccess(lines)
3394 endfor
3395enddef
3396
Ernie Raelfa831102023-12-14 20:06:39 +01003397def Test_assign_type_to_list_dict()
3398 var lines =<< trim END
3399 vim9script
3400 class C
3401 endclass
3402
3403 var x = [C]
3404 END
3405 v9.CheckScriptFailure(lines, 'E1405: Class "C" cannot be used as a value')
3406
3407 lines =<< trim END
3408 vim9script
3409 class C
3410 endclass
3411 type T = C
3412
3413 def F()
3414 var x = [3, T, C]
3415 enddef
3416 F()
3417 END
3418 v9.CheckScriptFailure(lines, 'E1405: Class "C" cannot be used as a value')
3419
3420 lines =<< trim END
3421 vim9script
3422 type T = number
3423
3424 def F()
3425 var x = [3, T]
3426 enddef
3427 F()
3428 END
3429 v9.CheckScriptFailure(lines, 'E1407: Cannot use a Typealias as a variable or value')
3430
3431 lines =<< trim END
3432 vim9script
3433 class C
3434 endclass
3435
3436 var x = {e: C}
3437 END
3438 v9.CheckScriptFailure(lines, 'E1405: Class "C" cannot be used as a value')
3439
3440 lines =<< trim END
3441 vim9script
3442 class C
3443 endclass
3444
3445 def F()
3446 var x = {e: C}
3447 enddef
3448 F()
3449 END
3450 v9.CheckScriptFailure(lines, 'E1405: Class "C" cannot be used as a value')
3451
3452 lines =<< trim END
3453 vim9script
3454 type T = number
3455
3456 def F()
3457 var x = {e: T}
3458 enddef
3459 F()
3460 END
3461 v9.CheckScriptFailure(lines, 'E1407: Cannot use a Typealias as a variable or value')
3462
3463 lines =<< trim END
3464 vim9script
3465 class C
3466 endclass
3467
3468 def F()
3469 var x = {e: [C]}
3470 enddef
3471 F()
3472 END
3473 v9.CheckScriptFailure(lines, 'E1405: Class "C" cannot be used as a value')
3474
3475 lines =<< trim END
3476 vim9script
3477 type T = number
3478
3479 def F()
3480 var x = {e: [T]}
3481 enddef
3482 F()
3483 END
3484 v9.CheckScriptFailure(lines, 'E1407: Cannot use a Typealias as a variable or value')
3485enddef
3486
Yegappan Lakshmanan1af35632024-02-06 11:03:36 +01003487" Test for modifying a final variable using a compound operator
3488def Test_final_var_modification_with_compound_op()
3489 var lines =<< trim END
3490 vim9script
3491
3492 final i: number = 1000
3493 assert_fails('i += 2', 'E46: Cannot change read-only variable "i"')
3494 assert_fails('i -= 2', 'E46: Cannot change read-only variable "i"')
3495 assert_fails('i *= 2', 'E46: Cannot change read-only variable "i"')
3496 assert_fails('i /= 2', 'E46: Cannot change read-only variable "i"')
3497 assert_fails('i %= 2', 'E46: Cannot change read-only variable "i"')
3498 assert_equal(1000, i)
3499
3500 final f: float = 1000.0
3501 assert_fails('f += 2', 'E46: Cannot change read-only variable "f"')
3502 assert_fails('f -= 2', 'E46: Cannot change read-only variable "f"')
3503 assert_fails('f *= 2', 'E46: Cannot change read-only variable "f"')
3504 assert_fails('f /= 2', 'E46: Cannot change read-only variable "f"')
3505 assert_equal(1000.0, f)
3506
3507 final s: string = 'abc'
3508 assert_fails('s ..= "y"', 'E46: Cannot change read-only variable "s"')
3509 assert_equal('abc', s)
3510 END
3511 v9.CheckScriptSuccess(lines)
3512enddef
3513
3514" Test for modifying a final variable with a List value
3515def Test_final_var_with_list_value()
3516 var lines =<< trim END
3517 vim9script
3518
3519 final listA: list<string> = []
3520 var listB = listA
3521
3522 listB->add('a')
3523 assert_true(listA is listB)
3524 assert_equal(['a'], listA)
3525 assert_equal(['a'], listB)
3526
3527 listB += ['b']
3528 assert_true(listA is listB)
3529 assert_equal(['a', 'b'], listA)
3530 assert_equal(['a', 'b'], listB)
3531
3532 listA->add('c')
3533 assert_true(listA is listB)
3534 assert_equal(['a', 'b', 'c'], listA)
3535 assert_equal(['a', 'b', 'c'], listB)
3536
3537 listA += ['d']
3538 assert_true(listA is listB)
3539 assert_equal(['a', 'b', 'c', 'd'], listA)
3540 assert_equal(['a', 'b', 'c', 'd'], listB)
3541 END
3542 v9.CheckScriptSuccess(lines)
3543enddef
3544
3545" Test for modifying a final variable with a List value using "+=" from a legacy
3546" function.
3547func Test_final_var_with_list_value_legacy()
3548 vim9cmd final g:TestVar = ['a']
3549 vim9cmd g:TestVar += ['b']
3550 call assert_equal(['a', 'b'], g:TestVar)
3551endfunc
3552
3553" Test for modifying a final variable with a Blob value
3554def Test_final_var_with_blob_value()
3555 var lines =<< trim END
3556 vim9script
3557
3558 final blobA: blob = 0z10
3559 var blobB = blobA
3560
3561 blobB->add(32)
3562 assert_true(blobA is blobB)
3563 assert_equal(0z1020, blobA)
3564 assert_equal(0z1020, blobB)
3565
3566 blobB += 0z30
3567 assert_true(blobA is blobB)
3568 assert_equal(0z102030, blobA)
3569 assert_equal(0z102030, blobB)
3570
3571 blobA->add(64)
3572 assert_true(blobA is blobB)
3573 assert_equal(0z10203040, blobA)
3574 assert_equal(0z10203040, blobB)
3575
3576 blobA += 0z50
3577 assert_true(blobA is blobB)
3578 assert_equal(0z1020304050, blobA)
3579 assert_equal(0z1020304050, blobB)
3580 END
3581 v9.CheckScriptSuccess(lines)
3582enddef
3583
Bram Moolenaar971caf52020-09-21 22:21:42 +02003584" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker