Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 1 | " Test Vim9 assignments |
| 2 | |
| 3 | source check.vim |
| 4 | source vim9.vim |
| 5 | |
| 6 | let s:appendToMe = 'xxx' |
| 7 | let s:addToMe = 111 |
| 8 | let g:existing = 'yes' |
| 9 | let g:inc_counter = 1 |
| 10 | let $SOME_ENV_VAR = 'some' |
| 11 | let g:alist = [7] |
Bram Moolenaar | 5b5ae29 | 2021-02-20 17:04:02 +0100 | [diff] [blame] | 12 | let g:adict = #{a: 1} |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 13 | let g:astring = 'text' |
| 14 | |
| 15 | def Test_assignment_bool() |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 16 | var bool1: bool = true |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 17 | assert_equal(v:true, bool1) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 18 | var bool2: bool = false |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 19 | assert_equal(v:false, bool2) |
| 20 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 21 | var bool3: bool = 0 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 22 | assert_equal(false, bool3) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 23 | var bool4: bool = 1 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 24 | assert_equal(true, bool4) |
| 25 | |
Bram Moolenaar | 2bb2658 | 2020-10-03 22:52:39 +0200 | [diff] [blame] | 26 | var bool5: bool = 1 && true |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 27 | assert_equal(true, bool5) |
Bram Moolenaar | 2bb2658 | 2020-10-03 22:52:39 +0200 | [diff] [blame] | 28 | var bool6: bool = 0 && 1 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 29 | assert_equal(false, bool6) |
Bram Moolenaar | 2bb2658 | 2020-10-03 22:52:39 +0200 | [diff] [blame] | 30 | var bool7: bool = 0 || 1 && true |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 31 | assert_equal(true, bool7) |
| 32 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 33 | var lines =<< trim END |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 34 | vim9script |
| 35 | def GetFlag(): bool |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 36 | var flag: bool = 1 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 37 | return flag |
| 38 | enddef |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 39 | var flag: bool = GetFlag() |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 40 | assert_equal(true, flag) |
| 41 | flag = 0 |
| 42 | assert_equal(false, flag) |
| 43 | flag = 1 |
| 44 | assert_equal(true, flag) |
Bram Moolenaar | 2bb2658 | 2020-10-03 22:52:39 +0200 | [diff] [blame] | 45 | flag = 1 || true |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 46 | assert_equal(true, flag) |
Bram Moolenaar | 2bb2658 | 2020-10-03 22:52:39 +0200 | [diff] [blame] | 47 | flag = 1 && false |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 48 | assert_equal(false, flag) |
Bram Moolenaar | dd1f426 | 2020-12-31 17:41:01 +0100 | [diff] [blame] | 49 | |
| 50 | var cp: bool = &cp |
| 51 | var fen: bool = &l:fen |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 52 | END |
| 53 | CheckScriptSuccess(lines) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 54 | CheckDefAndScriptFailure(['var x: bool = 2'], 'E1012:') |
| 55 | CheckDefAndScriptFailure(['var x: bool = -1'], 'E1012:') |
| 56 | CheckDefAndScriptFailure(['var x: bool = [1]'], 'E1012:') |
| 57 | CheckDefAndScriptFailure(['var x: bool = {}'], 'E1012:') |
| 58 | CheckDefAndScriptFailure(['var x: bool = "x"'], 'E1012:') |
Bram Moolenaar | 9a562c1 | 2021-01-23 13:39:14 +0100 | [diff] [blame] | 59 | |
| 60 | CheckDefAndScriptFailure(['var x: bool = "x"', '', 'eval 0'], 'E1012:', 1) |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 61 | enddef |
| 62 | |
| 63 | def Test_syntax() |
Bram Moolenaar | 9830938 | 2020-09-27 21:58:45 +0200 | [diff] [blame] | 64 | var name = 234 |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 65 | var other: list<string> = ['asdf'] |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 66 | enddef |
| 67 | |
| 68 | def Test_assignment() |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 69 | CheckDefFailure(['var x:string'], 'E1069:') |
| 70 | CheckDefFailure(['var x:string = "x"'], 'E1069:') |
| 71 | CheckDefFailure(['var a:string = "x"'], 'E1069:') |
Bram Moolenaar | 2949cfd | 2020-12-31 21:28:47 +0100 | [diff] [blame] | 72 | CheckDefFailure(['var lambda = () => "lambda"'], 'E704:') |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 73 | CheckScriptFailure(['var x = "x"'], 'E1124:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 74 | |
Bram Moolenaar | 3f32788 | 2021-03-17 20:56:38 +0100 | [diff] [blame] | 75 | # lower case name is OK for a list |
| 76 | var lambdaLines =<< trim END |
| 77 | var lambdaList: list<func> = [Test_syntax] |
| 78 | lambdaList[0] = () => "lambda" |
| 79 | END |
| 80 | CheckDefAndScriptSuccess(lambdaLines) |
| 81 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 82 | var nr: number = 1234 |
| 83 | CheckDefFailure(['var nr: number = "asdf"'], 'E1012:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 84 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 85 | var a: number = 6 #comment |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 86 | assert_equal(6, a) |
| 87 | |
| 88 | if has('channel') |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 89 | var chan1: channel |
Bram Moolenaar | 218450a | 2020-10-17 18:51:52 +0200 | [diff] [blame] | 90 | assert_equal('fail', ch_status(chan1)) |
| 91 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 92 | var job1: job |
Bram Moolenaar | 218450a | 2020-10-17 18:51:52 +0200 | [diff] [blame] | 93 | assert_equal('fail', job_status(job1)) |
| 94 | |
Bram Moolenaar | fc8aa6d | 2020-10-12 20:31:26 +0200 | [diff] [blame] | 95 | # calling job_start() is in test_vim9_fails.vim, it causes leak reports |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 96 | endif |
| 97 | if has('float') |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 98 | var float1: float = 3.4 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 99 | endif |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 100 | var Funky1: func |
| 101 | var Funky2: func = function('len') |
| 102 | var Party2: func = funcref('g:Test_syntax') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 103 | |
| 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 Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 118 | var lines =<< trim END |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 119 | vim9script |
| 120 | $ENVVAR = 'barfoo' |
| 121 | assert_equal('barfoo', $ENVVAR) |
| 122 | $ENVVAR = '' |
| 123 | END |
| 124 | CheckScriptSuccess(lines) |
| 125 | |
| 126 | s:appendToMe ..= 'yyy' |
| 127 | assert_equal('xxxyyy', s:appendToMe) |
| 128 | s:addToMe += 222 |
| 129 | assert_equal(333, s:addToMe) |
| 130 | s:newVar = 'new' |
| 131 | assert_equal('new', s:newVar) |
| 132 | |
| 133 | set ts=7 |
Bram Moolenaar | dd1f426 | 2020-12-31 17:41:01 +0100 | [diff] [blame] | 134 | var ts: number = &ts |
| 135 | assert_equal(7, ts) |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 136 | &ts += 1 |
| 137 | assert_equal(8, &ts) |
| 138 | &ts -= 3 |
| 139 | assert_equal(5, &ts) |
| 140 | &ts *= 2 |
| 141 | assert_equal(10, &ts) |
| 142 | &ts /= 3 |
| 143 | assert_equal(3, &ts) |
| 144 | set ts=10 |
| 145 | &ts %= 4 |
| 146 | assert_equal(2, &ts) |
| 147 | |
| 148 | if has('float') |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 149 | var f100: float = 100.0 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 150 | f100 /= 5 |
| 151 | assert_equal(20.0, f100) |
| 152 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 153 | var f200: float = 200.0 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 154 | f200 /= 5.0 |
| 155 | assert_equal(40.0, f200) |
| 156 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 157 | CheckDefFailure(['var nr: number = 200', 'nr /= 5.0'], 'E1012:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 158 | endif |
| 159 | |
| 160 | lines =<< trim END |
| 161 | &ts = 6 |
| 162 | &ts += 3 |
| 163 | assert_equal(9, &ts) |
| 164 | |
| 165 | &l:ts = 6 |
| 166 | assert_equal(6, &ts) |
| 167 | &l:ts += 2 |
| 168 | assert_equal(8, &ts) |
| 169 | |
| 170 | &g:ts = 6 |
| 171 | assert_equal(6, &g:ts) |
| 172 | &g:ts += 2 |
| 173 | assert_equal(8, &g:ts) |
Bram Moolenaar | 0ea0440 | 2021-01-04 13:37:54 +0100 | [diff] [blame] | 174 | |
| 175 | &number = true |
| 176 | assert_equal(true, &number) |
| 177 | &number = 0 |
| 178 | assert_equal(false, &number) |
| 179 | &number = 1 |
| 180 | assert_equal(true, &number) |
| 181 | &number = false |
| 182 | assert_equal(false, &number) |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 183 | END |
| 184 | CheckDefAndScriptSuccess(lines) |
| 185 | |
| 186 | CheckDefFailure(['¬ex += 3'], 'E113:') |
| 187 | CheckDefFailure(['&ts ..= "xxx"'], 'E1019:') |
| 188 | CheckDefFailure(['&ts = [7]'], 'E1012:') |
| 189 | CheckDefExecFailure(['&ts = g:alist'], 'E1012: Type mismatch; expected number but got list<number>') |
| 190 | CheckDefFailure(['&ts = "xx"'], 'E1012:') |
| 191 | CheckDefExecFailure(['&ts = g:astring'], 'E1012: Type mismatch; expected number but got string') |
| 192 | CheckDefFailure(['&path += 3'], 'E1012:') |
| 193 | CheckDefExecFailure(['&bs = "asdf"'], 'E474:') |
| 194 | # test freeing ISN_STOREOPT |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 195 | CheckDefFailure(['&ts = 3', 'var asdf'], 'E1022:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 196 | &ts = 8 |
| 197 | |
| 198 | lines =<< trim END |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 199 | var save_TI = &t_TI |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 200 | &t_TI = '' |
| 201 | assert_equal('', &t_TI) |
| 202 | &t_TI = 'xxx' |
| 203 | assert_equal('xxx', &t_TI) |
| 204 | &t_TI = save_TI |
| 205 | END |
| 206 | CheckDefAndScriptSuccess(lines) |
| 207 | |
| 208 | CheckDefFailure(['&t_TI = 123'], 'E1012:') |
| 209 | CheckScriptFailure(['vim9script', '&t_TI = 123'], 'E928:') |
| 210 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 211 | CheckDefFailure(['var s:var = 123'], 'E1101:') |
| 212 | CheckDefFailure(['var s:var: number'], 'E1101:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 213 | |
| 214 | lines =<< trim END |
| 215 | vim9script |
| 216 | def SomeFunc() |
| 217 | s:var = 123 |
| 218 | enddef |
| 219 | defcompile |
| 220 | END |
| 221 | CheckScriptFailure(lines, 'E1089:') |
| 222 | |
| 223 | g:inc_counter += 1 |
| 224 | assert_equal(2, g:inc_counter) |
| 225 | |
| 226 | $SOME_ENV_VAR ..= 'more' |
| 227 | assert_equal('somemore', $SOME_ENV_VAR) |
| 228 | CheckDefFailure(['$SOME_ENV_VAR += "more"'], 'E1051:') |
| 229 | CheckDefFailure(['$SOME_ENV_VAR += 123'], 'E1012:') |
| 230 | |
| 231 | lines =<< trim END |
| 232 | @c = 'areg' |
| 233 | @c ..= 'add' |
| 234 | assert_equal('aregadd', @c) |
| 235 | END |
| 236 | CheckDefAndScriptSuccess(lines) |
| 237 | |
| 238 | CheckDefFailure(['@a += "more"'], 'E1051:') |
| 239 | CheckDefFailure(['@a += 123'], 'E1012:') |
| 240 | |
| 241 | v:errmsg = 'none' |
| 242 | v:errmsg ..= 'again' |
| 243 | assert_equal('noneagain', v:errmsg) |
| 244 | CheckDefFailure(['v:errmsg += "more"'], 'E1051:') |
| 245 | CheckDefFailure(['v:errmsg += 123'], 'E1012:') |
| 246 | |
Bram Moolenaar | fc8aa6d | 2020-10-12 20:31:26 +0200 | [diff] [blame] | 247 | var text =<< trim END |
| 248 | some text |
| 249 | END |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 250 | enddef |
| 251 | |
Bram Moolenaar | 8ff16e0 | 2020-12-07 21:49:52 +0100 | [diff] [blame] | 252 | def Test_assign_unpack() |
| 253 | var lines =<< trim END |
| 254 | var v1: number |
| 255 | var v2: number |
| 256 | [v1, v2] = [1, 2] |
| 257 | assert_equal(1, v1) |
| 258 | assert_equal(2, v2) |
| 259 | END |
| 260 | CheckDefAndScriptSuccess(lines) |
| 261 | |
| 262 | lines =<< trim END |
| 263 | var v1: number |
| 264 | var v2: number |
| 265 | [v1, v2] = |
| 266 | END |
| 267 | CheckDefFailure(lines, 'E1097:', 5) |
| 268 | |
| 269 | lines =<< trim END |
| 270 | var v1: number |
| 271 | var v2: number |
| 272 | [v1, v2] = xxx |
| 273 | END |
| 274 | CheckDefFailure(lines, 'E1001:', 3) |
| 275 | |
| 276 | lines =<< trim END |
| 277 | var v1: number |
| 278 | var v2: number |
| 279 | [v1, v2] = popup_clear() |
| 280 | END |
| 281 | CheckDefFailure(lines, 'E1031:', 3) |
| 282 | |
| 283 | lines =<< trim END |
Bram Moolenaar | 3862ea3 | 2021-01-01 21:05:55 +0100 | [diff] [blame] | 284 | [v1, v2] = [1, 2] |
| 285 | END |
| 286 | CheckDefFailure(lines, 'E1089', 1) |
| 287 | CheckScriptFailure(['vim9script'] + lines, 'E1089', 2) |
| 288 | |
| 289 | lines =<< trim END |
Bram Moolenaar | 8ff16e0 | 2020-12-07 21:49:52 +0100 | [diff] [blame] | 290 | var v1: number |
| 291 | var v2: number |
| 292 | [v1, v2] = '' |
| 293 | END |
| 294 | CheckDefFailure(lines, 'E1012: Type mismatch; expected list<any> but got string', 3) |
Bram Moolenaar | f785aa1 | 2021-02-11 21:19:34 +0100 | [diff] [blame] | 295 | |
| 296 | lines =<< trim END |
| 297 | g:values = [false, 0] |
| 298 | var x: bool |
| 299 | var y: string |
| 300 | [x, y] = g:values |
| 301 | END |
| 302 | CheckDefExecAndScriptFailure(lines, 'E1163: Variable 2: type mismatch, expected string but got number') |
Bram Moolenaar | 8ff16e0 | 2020-12-07 21:49:52 +0100 | [diff] [blame] | 303 | enddef |
| 304 | |
Bram Moolenaar | 7f76494 | 2020-12-02 15:11:18 +0100 | [diff] [blame] | 305 | def Test_assign_linebreak() |
| 306 | var nr: number |
| 307 | nr = |
| 308 | 123 |
| 309 | assert_equal(123, nr) |
| 310 | |
| 311 | var n2: number |
| 312 | [nr, n2] = |
| 313 | [12, 34] |
| 314 | assert_equal(12, nr) |
| 315 | assert_equal(34, n2) |
| 316 | |
Bram Moolenaar | 8ff16e0 | 2020-12-07 21:49:52 +0100 | [diff] [blame] | 317 | CheckDefFailure(["var x = #"], 'E1097:', 3) |
Bram Moolenaar | 77709b1 | 2021-04-03 21:01:01 +0200 | [diff] [blame] | 318 | |
| 319 | var lines =<< trim END |
| 320 | var x: list<string> = ['a'] |
| 321 | var y: list<number> = x |
| 322 | ->copy() |
| 323 | ->copy() |
| 324 | END |
| 325 | CheckDefFailure(lines, 'E1012:', 2) |
Bram Moolenaar | 7f76494 | 2020-12-02 15:11:18 +0100 | [diff] [blame] | 326 | enddef |
| 327 | |
Bram Moolenaar | dc234ca | 2020-11-28 18:52:33 +0100 | [diff] [blame] | 328 | def Test_assign_index() |
| 329 | # list of list |
| 330 | var l1: list<number> |
| 331 | l1[0] = 123 |
| 332 | assert_equal([123], l1) |
| 333 | |
| 334 | var l2: list<list<number>> |
| 335 | l2[0] = [] |
| 336 | l2[0][0] = 123 |
| 337 | assert_equal([[123]], l2) |
| 338 | |
| 339 | var l3: list<list<list<number>>> |
| 340 | l3[0] = [] |
| 341 | l3[0][0] = [] |
| 342 | l3[0][0][0] = 123 |
| 343 | assert_equal([[[123]]], l3) |
| 344 | |
| 345 | var lines =<< trim END |
| 346 | var l3: list<list<number>> |
| 347 | l3[0] = [] |
| 348 | l3[0][0] = [] |
| 349 | END |
| 350 | CheckDefFailure(lines, 'E1012: Type mismatch; expected number but got list<unknown>', 3) |
| 351 | |
| 352 | # dict of dict |
| 353 | var d1: dict<number> |
| 354 | d1.one = 1 |
| 355 | assert_equal({one: 1}, d1) |
| 356 | |
| 357 | var d2: dict<dict<number>> |
| 358 | d2.one = {} |
| 359 | d2.one.two = 123 |
| 360 | assert_equal({one: {two: 123}}, d2) |
| 361 | |
| 362 | var d3: dict<dict<dict<number>>> |
| 363 | d3.one = {} |
| 364 | d3.one.two = {} |
| 365 | d3.one.two.three = 123 |
| 366 | assert_equal({one: {two: {three: 123}}}, d3) |
| 367 | |
| 368 | lines =<< trim END |
| 369 | var d3: dict<dict<number>> |
| 370 | d3.one = {} |
| 371 | d3.one.two = {} |
| 372 | END |
| 373 | CheckDefFailure(lines, 'E1012: Type mismatch; expected number but got dict<unknown>', 3) |
| 374 | |
Bram Moolenaar | acbae18 | 2020-12-13 18:44:43 +0100 | [diff] [blame] | 375 | lines =<< trim END |
| 376 | var lines: list<string> |
| 377 | lines['a'] = 'asdf' |
| 378 | END |
Bram Moolenaar | f30a14d | 2021-01-17 21:51:24 +0100 | [diff] [blame] | 379 | CheckDefFailure(lines, 'E1012:', 2) |
Bram Moolenaar | acbae18 | 2020-12-13 18:44:43 +0100 | [diff] [blame] | 380 | |
| 381 | lines =<< trim END |
| 382 | var lines: string |
| 383 | lines[9] = 'asdf' |
| 384 | END |
| 385 | CheckDefFailure(lines, 'E1141:', 2) |
| 386 | |
Bram Moolenaar | dc234ca | 2020-11-28 18:52:33 +0100 | [diff] [blame] | 387 | # list of dict |
| 388 | var ld: list<dict<number>> |
| 389 | ld[0] = {} |
| 390 | ld[0].one = 123 |
| 391 | assert_equal([{one: 123}], ld) |
| 392 | |
| 393 | lines =<< trim END |
| 394 | var ld: list<dict<number>> |
| 395 | ld[0] = [] |
| 396 | END |
| 397 | CheckDefFailure(lines, 'E1012: Type mismatch; expected dict<number> but got list<unknown>', 2) |
| 398 | |
| 399 | # dict of list |
| 400 | var dl: dict<list<number>> |
| 401 | dl.one = [] |
| 402 | dl.one[0] = 123 |
| 403 | assert_equal({one: [123]}, dl) |
| 404 | |
| 405 | lines =<< trim END |
| 406 | var dl: dict<list<number>> |
| 407 | dl.one = {} |
| 408 | END |
| 409 | CheckDefFailure(lines, 'E1012: Type mismatch; expected list<number> but got dict<unknown>', 2) |
| 410 | enddef |
| 411 | |
Bram Moolenaar | 81ed496 | 2020-09-23 15:56:50 +0200 | [diff] [blame] | 412 | def Test_extend_list() |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 413 | var lines =<< trim END |
Bram Moolenaar | 81ed496 | 2020-09-23 15:56:50 +0200 | [diff] [blame] | 414 | vim9script |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 415 | var l: list<number> |
Bram Moolenaar | 81ed496 | 2020-09-23 15:56:50 +0200 | [diff] [blame] | 416 | l += [123] |
| 417 | assert_equal([123], l) |
Bram Moolenaar | 348be7e | 2020-11-04 11:36:35 +0100 | [diff] [blame] | 418 | END |
| 419 | CheckScriptSuccess(lines) |
Bram Moolenaar | 81ed496 | 2020-09-23 15:56:50 +0200 | [diff] [blame] | 420 | |
Bram Moolenaar | 348be7e | 2020-11-04 11:36:35 +0100 | [diff] [blame] | 421 | lines =<< trim END |
| 422 | vim9script |
| 423 | var list: list<string> |
| 424 | extend(list, ['x']) |
| 425 | assert_equal(['x'], list) |
Bram Moolenaar | 81ed496 | 2020-09-23 15:56:50 +0200 | [diff] [blame] | 426 | END |
| 427 | CheckScriptSuccess(lines) |
Bram Moolenaar | 1a73923 | 2020-10-10 15:37:58 +0200 | [diff] [blame] | 428 | |
| 429 | # appending to NULL list from a function |
| 430 | lines =<< trim END |
| 431 | vim9script |
| 432 | var list: list<string> |
| 433 | def Func() |
| 434 | list += ['a', 'b'] |
| 435 | enddef |
| 436 | Func() |
| 437 | assert_equal(['a', 'b'], list) |
| 438 | END |
| 439 | CheckScriptSuccess(lines) |
Bram Moolenaar | 3beaf9c | 2020-12-18 17:23:14 +0100 | [diff] [blame] | 440 | lines =<< trim END |
| 441 | vim9script |
| 442 | var list: list<string> |
| 443 | def Func() |
| 444 | extend(list, ['x', 'b']) |
| 445 | enddef |
| 446 | Func() |
| 447 | assert_equal(['x', 'b'], list) |
| 448 | END |
| 449 | CheckScriptSuccess(lines) |
Bram Moolenaar | 348be7e | 2020-11-04 11:36:35 +0100 | [diff] [blame] | 450 | |
| 451 | lines =<< trim END |
| 452 | vim9script |
| 453 | var l: list<string> = test_null_list() |
| 454 | extend(l, ['x']) |
| 455 | assert_equal(['x'], l) |
| 456 | END |
| 457 | CheckScriptSuccess(lines) |
| 458 | |
| 459 | lines =<< trim END |
| 460 | vim9script |
| 461 | extend(test_null_list(), ['x']) |
| 462 | END |
| 463 | CheckScriptFailure(lines, 'E1134:', 2) |
| 464 | enddef |
| 465 | |
| 466 | def Test_extend_dict() |
| 467 | var lines =<< trim END |
| 468 | vim9script |
| 469 | var d: dict<number> |
Bram Moolenaar | e0de171 | 2020-12-02 17:36:54 +0100 | [diff] [blame] | 470 | extend(d, {a: 1}) |
| 471 | assert_equal({a: 1}, d) |
Bram Moolenaar | 348be7e | 2020-11-04 11:36:35 +0100 | [diff] [blame] | 472 | |
| 473 | var d2: dict<number> |
| 474 | d2['one'] = 1 |
Bram Moolenaar | e0de171 | 2020-12-02 17:36:54 +0100 | [diff] [blame] | 475 | assert_equal({one: 1}, d2) |
Bram Moolenaar | 348be7e | 2020-11-04 11:36:35 +0100 | [diff] [blame] | 476 | END |
| 477 | CheckScriptSuccess(lines) |
| 478 | |
| 479 | lines =<< trim END |
| 480 | vim9script |
| 481 | var d: dict<string> = test_null_dict() |
Bram Moolenaar | e0de171 | 2020-12-02 17:36:54 +0100 | [diff] [blame] | 482 | extend(d, {a: 'x'}) |
| 483 | assert_equal({a: 'x'}, d) |
Bram Moolenaar | 348be7e | 2020-11-04 11:36:35 +0100 | [diff] [blame] | 484 | END |
| 485 | CheckScriptSuccess(lines) |
| 486 | |
| 487 | lines =<< trim END |
| 488 | vim9script |
Bram Moolenaar | e0de171 | 2020-12-02 17:36:54 +0100 | [diff] [blame] | 489 | extend(test_null_dict(), {a: 'x'}) |
Bram Moolenaar | 348be7e | 2020-11-04 11:36:35 +0100 | [diff] [blame] | 490 | END |
| 491 | CheckScriptFailure(lines, 'E1133:', 2) |
Bram Moolenaar | 81ed496 | 2020-09-23 15:56:50 +0200 | [diff] [blame] | 492 | enddef |
| 493 | |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 494 | def Test_single_letter_vars() |
| 495 | # single letter variables |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 496 | var a: number = 123 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 497 | a = 123 |
| 498 | assert_equal(123, a) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 499 | var b: number |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 500 | b = 123 |
| 501 | assert_equal(123, b) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 502 | var g: number |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 503 | g = 123 |
| 504 | assert_equal(123, g) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 505 | var s: number |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 506 | s = 123 |
| 507 | assert_equal(123, s) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 508 | var t: number |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 509 | t = 123 |
| 510 | assert_equal(123, t) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 511 | var v: number |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 512 | v = 123 |
| 513 | assert_equal(123, v) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 514 | var w: number |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 515 | w = 123 |
| 516 | assert_equal(123, w) |
| 517 | enddef |
| 518 | |
| 519 | def Test_vim9_single_char_vars() |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 520 | var lines =<< trim END |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 521 | vim9script |
| 522 | |
| 523 | # single character variable declarations work |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 524 | var a: string |
| 525 | var b: number |
| 526 | var l: list<any> |
| 527 | var s: string |
| 528 | var t: number |
| 529 | var v: number |
| 530 | var w: number |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 531 | |
| 532 | # script-local variables can be used without s: prefix |
| 533 | a = 'script-a' |
| 534 | b = 111 |
| 535 | l = [1, 2, 3] |
| 536 | s = 'script-s' |
| 537 | t = 222 |
| 538 | v = 333 |
| 539 | w = 444 |
| 540 | |
| 541 | assert_equal('script-a', a) |
| 542 | assert_equal(111, b) |
| 543 | assert_equal([1, 2, 3], l) |
| 544 | assert_equal('script-s', s) |
| 545 | assert_equal(222, t) |
| 546 | assert_equal(333, v) |
| 547 | assert_equal(444, w) |
| 548 | END |
| 549 | writefile(lines, 'Xsinglechar') |
| 550 | source Xsinglechar |
| 551 | delete('Xsinglechar') |
| 552 | enddef |
| 553 | |
| 554 | def Test_assignment_list() |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 555 | var list1: list<bool> = [false, true, false] |
| 556 | var list2: list<number> = [1, 2, 3] |
| 557 | var list3: list<string> = ['sdf', 'asdf'] |
| 558 | var list4: list<any> = ['yes', true, 1234] |
| 559 | var list5: list<blob> = [0z01, 0z02] |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 560 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 561 | var listS: list<string> = [] |
| 562 | var listN: list<number> = [] |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 563 | |
| 564 | assert_equal([1, 2, 3], list2) |
| 565 | list2[-1] = 99 |
| 566 | assert_equal([1, 2, 99], list2) |
| 567 | list2[-2] = 88 |
| 568 | assert_equal([1, 88, 99], list2) |
| 569 | list2[-3] = 77 |
| 570 | assert_equal([77, 88, 99], list2) |
| 571 | list2 += [100] |
| 572 | assert_equal([77, 88, 99, 100], list2) |
| 573 | |
| 574 | list3 += ['end'] |
| 575 | assert_equal(['sdf', 'asdf', 'end'], list3) |
| 576 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 577 | CheckDefExecFailure(['var ll = [1, 2, 3]', 'll[-4] = 6'], 'E684:') |
| 578 | CheckDefExecFailure(['var [v1, v2] = [1, 2]'], 'E1092:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 579 | |
| 580 | # type becomes list<any> |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 581 | var somelist = rand() > 0 ? [1, 2, 3] : ['a', 'b', 'c'] |
Bram Moolenaar | 4f5e397 | 2020-12-21 17:30:50 +0100 | [diff] [blame] | 582 | |
| 583 | var lines =<< trim END |
| 584 | var d = {dd: test_null_list()} |
| 585 | d.dd[0] = 0 |
| 586 | END |
| 587 | CheckDefExecFailure(lines, 'E1147:', 2) |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 588 | enddef |
| 589 | |
Bram Moolenaar | f30a14d | 2021-01-17 21:51:24 +0100 | [diff] [blame] | 590 | def Test_assignment_list_any_index() |
| 591 | var l: list<number> = [1, 2] |
| 592 | for [x, y, _] |
| 593 | in [[0, 1, ''], [1, 3, '']] |
| 594 | l[x] = l[x] + y |
| 595 | endfor |
| 596 | assert_equal([2, 5], l) |
| 597 | enddef |
| 598 | |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 599 | def Test_assignment_list_vim9script() |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 600 | var lines =<< trim END |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 601 | vim9script |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 602 | var v1: number |
| 603 | var v2: number |
| 604 | var v3: number |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 605 | [v1, v2, v3] = [1, 2, 3] |
| 606 | assert_equal([1, 2, 3], [v1, v2, v3]) |
| 607 | END |
| 608 | CheckScriptSuccess(lines) |
| 609 | enddef |
| 610 | |
| 611 | def Test_assignment_dict() |
Bram Moolenaar | e0de171 | 2020-12-02 17:36:54 +0100 | [diff] [blame] | 612 | var dict1: dict<bool> = {one: false, two: true} |
| 613 | var dict2: dict<number> = {one: 1, two: 2} |
| 614 | var dict3: dict<string> = {key: 'value'} |
| 615 | var dict4: dict<any> = {one: 1, two: '2'} |
| 616 | var dict5: dict<blob> = {one: 0z01, two: 0z02} |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 617 | |
| 618 | # overwrite |
| 619 | dict3['key'] = 'another' |
Bram Moolenaar | e0de171 | 2020-12-02 17:36:54 +0100 | [diff] [blame] | 620 | assert_equal(dict3, {key: 'another'}) |
Bram Moolenaar | fc74d03 | 2020-11-16 22:11:49 +0100 | [diff] [blame] | 621 | dict3.key = 'yet another' |
Bram Moolenaar | e0de171 | 2020-12-02 17:36:54 +0100 | [diff] [blame] | 622 | assert_equal(dict3, {key: 'yet another'}) |
Bram Moolenaar | fc74d03 | 2020-11-16 22:11:49 +0100 | [diff] [blame] | 623 | |
Bram Moolenaar | d24602f | 2020-12-20 15:20:56 +0100 | [diff] [blame] | 624 | # member "any" can also be a dict and assigned to |
| 625 | var anydict: dict<any> = {nest: {}, nr: 0} |
| 626 | anydict.nest['this'] = 123 |
| 627 | anydict.nest.that = 456 |
| 628 | assert_equal({nest: {this: 123, that: 456}, nr: 0}, anydict) |
| 629 | |
Bram Moolenaar | fc74d03 | 2020-11-16 22:11:49 +0100 | [diff] [blame] | 630 | var lines =<< trim END |
Bram Moolenaar | d88dc4d | 2020-12-20 14:49:42 +0100 | [diff] [blame] | 631 | var dd = {} |
| 632 | dd.two = 2 |
| 633 | assert_equal({two: 2}, dd) |
| 634 | END |
Bram Moolenaar | 4f5e397 | 2020-12-21 17:30:50 +0100 | [diff] [blame] | 635 | CheckDefAndScriptSuccess(lines) |
| 636 | |
| 637 | lines =<< trim END |
| 638 | var d = {dd: {}} |
| 639 | d.dd[0] = 2 |
| 640 | d.dd['x'] = 3 |
| 641 | d.dd.y = 4 |
| 642 | assert_equal({dd: {0: 2, x: 3, y: 4}}, d) |
| 643 | END |
| 644 | CheckDefAndScriptSuccess(lines) |
Bram Moolenaar | d88dc4d | 2020-12-20 14:49:42 +0100 | [diff] [blame] | 645 | |
| 646 | lines =<< trim END |
Bram Moolenaar | e0de171 | 2020-12-02 17:36:54 +0100 | [diff] [blame] | 647 | var dd = {one: 1} |
Bram Moolenaar | fc74d03 | 2020-11-16 22:11:49 +0100 | [diff] [blame] | 648 | dd.one) = 2 |
| 649 | END |
Bram Moolenaar | d88dc4d | 2020-12-20 14:49:42 +0100 | [diff] [blame] | 650 | CheckDefFailure(lines, 'E488:', 2) |
| 651 | |
| 652 | lines =<< trim END |
| 653 | var dd = {one: 1} |
| 654 | var dd.one = 2 |
| 655 | END |
| 656 | CheckDefAndScriptFailure(lines, 'E1017:', 2) |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 657 | |
| 658 | # empty key can be used |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 659 | var dd = {} |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 660 | dd[""] = 6 |
Bram Moolenaar | e0de171 | 2020-12-02 17:36:54 +0100 | [diff] [blame] | 661 | assert_equal({['']: 6}, dd) |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 662 | |
| 663 | # type becomes dict<any> |
Bram Moolenaar | e0de171 | 2020-12-02 17:36:54 +0100 | [diff] [blame] | 664 | var somedict = rand() > 0 ? {a: 1, b: 2} : {a: 'a', b: 'b'} |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 665 | |
| 666 | # assignment to script-local dict |
Bram Moolenaar | fc74d03 | 2020-11-16 22:11:49 +0100 | [diff] [blame] | 667 | lines =<< trim END |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 668 | vim9script |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 669 | var test: dict<any> = {} |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 670 | def FillDict(): dict<any> |
| 671 | test['a'] = 43 |
| 672 | return test |
| 673 | enddef |
Bram Moolenaar | e0de171 | 2020-12-02 17:36:54 +0100 | [diff] [blame] | 674 | assert_equal({a: 43}, FillDict()) |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 675 | END |
| 676 | CheckScriptSuccess(lines) |
| 677 | |
| 678 | lines =<< trim END |
| 679 | vim9script |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 680 | var test: dict<any> |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 681 | def FillDict(): dict<any> |
| 682 | test['a'] = 43 |
| 683 | return test |
| 684 | enddef |
| 685 | FillDict() |
Bram Moolenaar | 3beaf9c | 2020-12-18 17:23:14 +0100 | [diff] [blame] | 686 | assert_equal({a: 43}, test) |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 687 | END |
Bram Moolenaar | 3beaf9c | 2020-12-18 17:23:14 +0100 | [diff] [blame] | 688 | CheckScriptSuccess(lines) |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 689 | |
| 690 | # assignment to global dict |
| 691 | lines =<< trim END |
| 692 | vim9script |
| 693 | g:test = {} |
| 694 | def FillDict(): dict<any> |
| 695 | g:test['a'] = 43 |
| 696 | return g:test |
| 697 | enddef |
Bram Moolenaar | e0de171 | 2020-12-02 17:36:54 +0100 | [diff] [blame] | 698 | assert_equal({a: 43}, FillDict()) |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 699 | END |
| 700 | CheckScriptSuccess(lines) |
| 701 | |
| 702 | # assignment to buffer dict |
| 703 | lines =<< trim END |
| 704 | vim9script |
| 705 | b:test = {} |
| 706 | def FillDict(): dict<any> |
| 707 | b:test['a'] = 43 |
| 708 | return b:test |
| 709 | enddef |
Bram Moolenaar | e0de171 | 2020-12-02 17:36:54 +0100 | [diff] [blame] | 710 | assert_equal({a: 43}, FillDict()) |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 711 | END |
| 712 | CheckScriptSuccess(lines) |
Bram Moolenaar | 4f5e397 | 2020-12-21 17:30:50 +0100 | [diff] [blame] | 713 | |
| 714 | lines =<< trim END |
| 715 | var d = {dd: test_null_dict()} |
| 716 | d.dd[0] = 0 |
| 717 | END |
| 718 | CheckDefExecFailure(lines, 'E1103:', 2) |
| 719 | |
| 720 | lines =<< trim END |
| 721 | var d = {dd: 'string'} |
| 722 | d.dd[0] = 0 |
| 723 | END |
| 724 | CheckDefExecFailure(lines, 'E1148:', 2) |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 725 | enddef |
| 726 | |
| 727 | def Test_assignment_local() |
| 728 | # Test in a separated file in order not to the current buffer/window/tab is |
| 729 | # changed. |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 730 | var script_lines: list<string> =<< trim END |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 731 | let b:existing = 'yes' |
| 732 | let w:existing = 'yes' |
| 733 | let t:existing = 'yes' |
| 734 | |
| 735 | def Test_assignment_local_internal() |
| 736 | b:newvar = 'new' |
| 737 | assert_equal('new', b:newvar) |
| 738 | assert_equal('yes', b:existing) |
| 739 | b:existing = 'no' |
| 740 | assert_equal('no', b:existing) |
| 741 | b:existing ..= 'NO' |
| 742 | assert_equal('noNO', b:existing) |
| 743 | |
| 744 | w:newvar = 'new' |
| 745 | assert_equal('new', w:newvar) |
| 746 | assert_equal('yes', w:existing) |
| 747 | w:existing = 'no' |
| 748 | assert_equal('no', w:existing) |
| 749 | w:existing ..= 'NO' |
| 750 | assert_equal('noNO', w:existing) |
| 751 | |
| 752 | t:newvar = 'new' |
| 753 | assert_equal('new', t:newvar) |
| 754 | assert_equal('yes', t:existing) |
| 755 | t:existing = 'no' |
| 756 | assert_equal('no', t:existing) |
| 757 | t:existing ..= 'NO' |
| 758 | assert_equal('noNO', t:existing) |
| 759 | enddef |
| 760 | call Test_assignment_local_internal() |
| 761 | END |
| 762 | CheckScriptSuccess(script_lines) |
| 763 | enddef |
| 764 | |
| 765 | def Test_assignment_default() |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 766 | # Test default values. |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 767 | var thebool: bool |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 768 | assert_equal(v:false, thebool) |
| 769 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 770 | var thenumber: number |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 771 | assert_equal(0, thenumber) |
| 772 | |
| 773 | if has('float') |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 774 | var thefloat: float |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 775 | assert_equal(0.0, thefloat) |
| 776 | endif |
| 777 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 778 | var thestring: string |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 779 | assert_equal('', thestring) |
| 780 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 781 | var theblob: blob |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 782 | assert_equal(0z, theblob) |
| 783 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 784 | var Thefunc: func |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 785 | assert_equal(test_null_function(), Thefunc) |
| 786 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 787 | var thelist: list<any> |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 788 | assert_equal([], thelist) |
| 789 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 790 | var thedict: dict<any> |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 791 | assert_equal({}, thedict) |
| 792 | |
| 793 | if has('channel') |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 794 | var thejob: job |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 795 | assert_equal(test_null_job(), thejob) |
| 796 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 797 | var thechannel: channel |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 798 | assert_equal(test_null_channel(), thechannel) |
| 799 | |
| 800 | if has('unix') && executable('cat') |
| 801 | # check with non-null job and channel, types must match |
Bram Moolenaar | e0de171 | 2020-12-02 17:36:54 +0100 | [diff] [blame] | 802 | thejob = job_start("cat ", {}) |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 803 | thechannel = job_getchannel(thejob) |
| 804 | job_stop(thejob, 'kill') |
| 805 | endif |
| 806 | endif |
| 807 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 808 | var nr = 1234 | nr = 5678 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 809 | assert_equal(5678, nr) |
| 810 | enddef |
| 811 | |
Bram Moolenaar | 3862ea3 | 2021-01-01 21:05:55 +0100 | [diff] [blame] | 812 | let scriptvar = 'init' |
| 813 | |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 814 | def Test_assignment_var_list() |
Bram Moolenaar | da7c20c | 2020-11-30 21:12:19 +0100 | [diff] [blame] | 815 | var lines =<< trim END |
| 816 | var v1: string |
| 817 | var v2: string |
| 818 | var vrem: list<string> |
| 819 | [v1] = ['aaa'] |
| 820 | assert_equal('aaa', v1) |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 821 | |
Bram Moolenaar | da7c20c | 2020-11-30 21:12:19 +0100 | [diff] [blame] | 822 | [v1, v2] = ['one', 'two'] |
| 823 | assert_equal('one', v1) |
| 824 | assert_equal('two', v2) |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 825 | |
Bram Moolenaar | da7c20c | 2020-11-30 21:12:19 +0100 | [diff] [blame] | 826 | [v1, v2; vrem] = ['one', 'two'] |
| 827 | assert_equal('one', v1) |
| 828 | assert_equal('two', v2) |
| 829 | assert_equal([], vrem) |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 830 | |
Bram Moolenaar | da7c20c | 2020-11-30 21:12:19 +0100 | [diff] [blame] | 831 | [v1, v2; vrem] = ['one', 'two', 'three'] |
| 832 | assert_equal('one', v1) |
| 833 | assert_equal('two', v2) |
| 834 | assert_equal(['three'], vrem) |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 835 | |
Bram Moolenaar | da7c20c | 2020-11-30 21:12:19 +0100 | [diff] [blame] | 836 | [&ts, &sw] = [3, 4] |
| 837 | assert_equal(3, &ts) |
| 838 | assert_equal(4, &sw) |
| 839 | set ts=8 sw=4 |
Bram Moolenaar | 004d9b0 | 2020-11-30 21:40:03 +0100 | [diff] [blame] | 840 | |
| 841 | [@a, @z] = ['aa', 'zz'] |
| 842 | assert_equal('aa', @a) |
| 843 | assert_equal('zz', @z) |
| 844 | |
| 845 | [$SOME_VAR, $OTHER_VAR] = ['some', 'other'] |
| 846 | assert_equal('some', $SOME_VAR) |
| 847 | assert_equal('other', $OTHER_VAR) |
| 848 | |
Bram Moolenaar | 3862ea3 | 2021-01-01 21:05:55 +0100 | [diff] [blame] | 849 | [g:globalvar, b:bufvar, w:winvar, t:tabvar, v:errmsg] = |
| 850 | ['global', 'buf', 'win', 'tab', 'error'] |
Bram Moolenaar | 004d9b0 | 2020-11-30 21:40:03 +0100 | [diff] [blame] | 851 | assert_equal('global', g:globalvar) |
Bram Moolenaar | 004d9b0 | 2020-11-30 21:40:03 +0100 | [diff] [blame] | 852 | assert_equal('buf', b:bufvar) |
| 853 | assert_equal('win', w:winvar) |
| 854 | assert_equal('tab', t:tabvar) |
| 855 | assert_equal('error', v:errmsg) |
| 856 | unlet g:globalvar |
Bram Moolenaar | da7c20c | 2020-11-30 21:12:19 +0100 | [diff] [blame] | 857 | END |
| 858 | CheckDefAndScriptSuccess(lines) |
Bram Moolenaar | 3862ea3 | 2021-01-01 21:05:55 +0100 | [diff] [blame] | 859 | |
| 860 | [g:globalvar, s:scriptvar, b:bufvar] = ['global', 'script', 'buf'] |
| 861 | assert_equal('global', g:globalvar) |
| 862 | assert_equal('script', s:scriptvar) |
| 863 | assert_equal('buf', b:bufvar) |
| 864 | |
| 865 | lines =<< trim END |
| 866 | vim9script |
| 867 | var s:scriptvar = 'init' |
| 868 | [g:globalvar, s:scriptvar, w:winvar] = ['global', 'script', 'win'] |
| 869 | assert_equal('global', g:globalvar) |
| 870 | assert_equal('script', s:scriptvar) |
| 871 | assert_equal('win', w:winvar) |
| 872 | END |
| 873 | CheckScriptSuccess(lines) |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 874 | enddef |
| 875 | |
| 876 | def Test_assignment_vim9script() |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 877 | var lines =<< trim END |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 878 | vim9script |
| 879 | def Func(): list<number> |
| 880 | return [1, 2] |
| 881 | enddef |
Bram Moolenaar | 9830938 | 2020-09-27 21:58:45 +0200 | [diff] [blame] | 882 | var name1: number |
| 883 | var name2: number |
| 884 | [name1, name2] = |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 885 | Func() |
Bram Moolenaar | 9830938 | 2020-09-27 21:58:45 +0200 | [diff] [blame] | 886 | assert_equal(1, name1) |
| 887 | assert_equal(2, name2) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 888 | var ll = |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 889 | Func() |
| 890 | assert_equal([1, 2], ll) |
| 891 | |
| 892 | @/ = 'text' |
| 893 | assert_equal('text', @/) |
| 894 | @0 = 'zero' |
| 895 | assert_equal('zero', @0) |
| 896 | @1 = 'one' |
| 897 | assert_equal('one', @1) |
| 898 | @9 = 'nine' |
| 899 | assert_equal('nine', @9) |
| 900 | @- = 'minus' |
| 901 | assert_equal('minus', @-) |
| 902 | if has('clipboard_working') |
| 903 | @* = 'star' |
| 904 | assert_equal('star', @*) |
| 905 | @+ = 'plus' |
| 906 | assert_equal('plus', @+) |
| 907 | endif |
| 908 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 909 | var a: number = 123 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 910 | assert_equal(123, a) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 911 | var s: string = 'yes' |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 912 | assert_equal('yes', s) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 913 | var b: number = 42 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 914 | assert_equal(42, b) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 915 | var w: number = 43 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 916 | assert_equal(43, w) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 917 | var t: number = 44 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 918 | assert_equal(44, t) |
Bram Moolenaar | 7a66a17 | 2020-10-16 19:56:12 +0200 | [diff] [blame] | 919 | |
| 920 | var to_var = 0 |
| 921 | to_var = 3 |
| 922 | assert_equal(3, to_var) |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 923 | END |
| 924 | CheckScriptSuccess(lines) |
Bram Moolenaar | 06f9c69 | 2020-09-27 21:27:40 +0200 | [diff] [blame] | 925 | |
| 926 | lines =<< trim END |
| 927 | vim9script |
| 928 | var n: number |
| 929 | def Func() |
| 930 | n = 'string' |
| 931 | enddef |
| 932 | defcompile |
| 933 | END |
| 934 | CheckScriptFailure(lines, 'E1012: Type mismatch; expected number but got string') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 935 | enddef |
| 936 | |
| 937 | def Mess(): string |
| 938 | v:foldstart = 123 |
| 939 | return 'xxx' |
| 940 | enddef |
| 941 | |
| 942 | def Test_assignment_failure() |
Bram Moolenaar | 9830938 | 2020-09-27 21:58:45 +0200 | [diff] [blame] | 943 | CheckDefFailure(['var name=234'], 'E1004:') |
| 944 | CheckDefFailure(['var name =234'], 'E1004:') |
| 945 | CheckDefFailure(['var name= 234'], 'E1004:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 946 | |
Bram Moolenaar | 9830938 | 2020-09-27 21:58:45 +0200 | [diff] [blame] | 947 | CheckScriptFailure(['vim9script', 'var name=234'], 'E1004:') |
| 948 | CheckScriptFailure(['vim9script', 'var name=234'], "before and after '='") |
| 949 | CheckScriptFailure(['vim9script', 'var name =234'], 'E1004:') |
| 950 | CheckScriptFailure(['vim9script', 'var name= 234'], 'E1004:') |
| 951 | CheckScriptFailure(['vim9script', 'var name = 234', 'name+=234'], 'E1004:') |
| 952 | CheckScriptFailure(['vim9script', 'var name = 234', 'name+=234'], "before and after '+='") |
| 953 | CheckScriptFailure(['vim9script', 'var name = "x"', 'name..="y"'], 'E1004:') |
| 954 | CheckScriptFailure(['vim9script', 'var name = "x"', 'name..="y"'], "before and after '..='") |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 955 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 956 | CheckDefFailure(['var true = 1'], 'E1034:') |
| 957 | CheckDefFailure(['var false = 1'], 'E1034:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 958 | |
| 959 | CheckDefFailure(['[a; b; c] = g:list'], 'E452:') |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 960 | CheckDefExecFailure(['var a: number', |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 961 | '[a] = test_null_list()'], 'E1093:') |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 962 | CheckDefExecFailure(['var a: number', |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 963 | '[a] = []'], 'E1093:') |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 964 | CheckDefExecFailure(['var x: number', |
| 965 | 'var y: number', |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 966 | '[x, y] = [1]'], 'E1093:') |
Bram Moolenaar | 352134b | 2020-10-17 22:04:08 +0200 | [diff] [blame] | 967 | CheckDefExecFailure(['var x: string', |
| 968 | 'var y: string', |
| 969 | '[x, y] = ["x"]'], 'E1093:') |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 970 | CheckDefExecFailure(['var x: number', |
| 971 | 'var y: number', |
| 972 | 'var z: list<number>', |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 973 | '[x, y; z] = [1]'], 'E1093:') |
| 974 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 975 | CheckDefFailure(['var somevar'], "E1022:") |
| 976 | CheckDefFailure(['var &tabstop = 4'], 'E1052:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 977 | CheckDefFailure(['&g:option = 5'], 'E113:') |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 978 | CheckScriptFailure(['vim9script', 'var &tabstop = 4'], 'E1052:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 979 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 980 | CheckDefFailure(['var $VAR = 5'], 'E1016: Cannot declare an environment variable:') |
| 981 | CheckScriptFailure(['vim9script', 'var $ENV = "xxx"'], 'E1016:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 982 | |
| 983 | if has('dnd') |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 984 | CheckDefFailure(['var @~ = 5'], 'E1066:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 985 | else |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 986 | CheckDefFailure(['var @~ = 5'], 'E354:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 987 | CheckDefFailure(['@~ = 5'], 'E354:') |
| 988 | endif |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 989 | CheckDefFailure(['var @a = 5'], 'E1066:') |
| 990 | CheckDefFailure(['var @/ = "x"'], 'E1066:') |
| 991 | CheckScriptFailure(['vim9script', 'var @a = "abc"'], 'E1066:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 992 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 993 | CheckDefFailure(['var g:var = 5'], 'E1016: Cannot declare a global variable:') |
| 994 | CheckDefFailure(['var w:var = 5'], 'E1016: Cannot declare a window variable:') |
| 995 | CheckDefFailure(['var b:var = 5'], 'E1016: Cannot declare a buffer variable:') |
| 996 | CheckDefFailure(['var t:var = 5'], 'E1016: Cannot declare a tab variable:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 997 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 998 | CheckDefFailure(['var anr = 4', 'anr ..= "text"'], 'E1019:') |
| 999 | CheckDefFailure(['var xnr += 4'], 'E1020:', 1) |
| 1000 | CheckScriptFailure(['vim9script', 'var xnr += 4'], 'E1020:') |
| 1001 | CheckDefFailure(["var xnr = xnr + 1"], 'E1001:', 1) |
| 1002 | CheckScriptFailure(['vim9script', 'var xnr = xnr + 4'], 'E121:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 1003 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 1004 | CheckScriptFailure(['vim9script', 'def Func()', 'var dummy = s:notfound', 'enddef', 'defcompile'], 'E1108:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 1005 | |
Bram Moolenaar | 9830938 | 2020-09-27 21:58:45 +0200 | [diff] [blame] | 1006 | CheckDefFailure(['var name: list<string> = [123]'], 'expected list<string> but got list<number>') |
| 1007 | CheckDefFailure(['var name: list<number> = ["xx"]'], 'expected list<number> but got list<string>') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 1008 | |
Bram Moolenaar | e0de171 | 2020-12-02 17:36:54 +0100 | [diff] [blame] | 1009 | CheckDefFailure(['var name: dict<string> = {key: 123}'], 'expected dict<string> but got dict<number>') |
| 1010 | CheckDefFailure(['var name: dict<number> = {key: "xx"}'], 'expected dict<number> but got dict<string>') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 1011 | |
Bram Moolenaar | 9830938 | 2020-09-27 21:58:45 +0200 | [diff] [blame] | 1012 | CheckDefFailure(['var name = feedkeys("0")'], 'E1031:') |
| 1013 | CheckDefFailure(['var name: number = feedkeys("0")'], 'expected number but got void') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 1014 | |
Bram Moolenaar | 9830938 | 2020-09-27 21:58:45 +0200 | [diff] [blame] | 1015 | CheckDefFailure(['var name: dict <number>'], 'E1068:') |
| 1016 | CheckDefFailure(['var name: dict<number'], 'E1009:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 1017 | |
| 1018 | assert_fails('s/^/\=Mess()/n', 'E794:') |
Bram Moolenaar | 9830938 | 2020-09-27 21:58:45 +0200 | [diff] [blame] | 1019 | CheckDefFailure(['var name: dict<number'], 'E1009:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 1020 | |
| 1021 | CheckDefFailure(['w:foo: number = 10'], |
| 1022 | 'E488: Trailing characters: : number = 1') |
| 1023 | CheckDefFailure(['t:foo: bool = true'], |
| 1024 | 'E488: Trailing characters: : bool = true') |
| 1025 | CheckDefFailure(['b:foo: string = "x"'], |
| 1026 | 'E488: Trailing characters: : string = "x"') |
| 1027 | CheckDefFailure(['g:foo: number = 123'], |
| 1028 | 'E488: Trailing characters: : number = 123') |
| 1029 | enddef |
| 1030 | |
| 1031 | def Test_assign_list() |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 1032 | var l: list<string> = [] |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 1033 | l[0] = 'value' |
| 1034 | assert_equal('value', l[0]) |
| 1035 | |
| 1036 | l[1] = 'asdf' |
| 1037 | assert_equal('value', l[0]) |
| 1038 | assert_equal('asdf', l[1]) |
| 1039 | assert_equal('asdf', l[-1]) |
| 1040 | assert_equal('value', l[-2]) |
| 1041 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 1042 | var nrl: list<number> = [] |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 1043 | for i in range(5) |
| 1044 | nrl[i] = i |
| 1045 | endfor |
| 1046 | assert_equal([0, 1, 2, 3, 4], nrl) |
Bram Moolenaar | 334a8b4 | 2020-10-19 16:07:42 +0200 | [diff] [blame] | 1047 | |
| 1048 | CheckDefFailure(["var l: list<number> = ['', true]"], 'E1012: Type mismatch; expected list<number> but got list<any>', 1) |
| 1049 | CheckDefFailure(["var l: list<list<number>> = [['', true]]"], 'E1012: Type mismatch; expected list<list<number>> but got list<list<any>>', 1) |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 1050 | enddef |
| 1051 | |
| 1052 | def Test_assign_dict() |
Bram Moolenaar | c368957 | 2021-01-01 19:40:02 +0100 | [diff] [blame] | 1053 | var lines =<< trim END |
| 1054 | var d: dict<string> = {} |
| 1055 | d['key'] = 'value' |
| 1056 | assert_equal('value', d['key']) |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 1057 | |
Bram Moolenaar | c368957 | 2021-01-01 19:40:02 +0100 | [diff] [blame] | 1058 | d[123] = 'qwerty' |
| 1059 | assert_equal('qwerty', d[123]) |
| 1060 | assert_equal('qwerty', d['123']) |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 1061 | |
Bram Moolenaar | c368957 | 2021-01-01 19:40:02 +0100 | [diff] [blame] | 1062 | var nrd: dict<number> = {} |
| 1063 | for i in range(3) |
| 1064 | nrd[i] = i |
| 1065 | endfor |
| 1066 | assert_equal({0: 0, 1: 1, 2: 2}, nrd) |
| 1067 | |
| 1068 | d.somekey = 'someval' |
| 1069 | assert_equal({key: 'value', '123': 'qwerty', somekey: 'someval'}, d) |
| 1070 | # unlet d.somekey |
| 1071 | # assert_equal({key: 'value', '123': 'qwerty'}, d) |
| 1072 | END |
| 1073 | CheckDefAndScriptSuccess(lines) |
| 1074 | |
| 1075 | # TODO: move to above once "unlet d.somekey" in :def is implemented |
| 1076 | lines =<< trim END |
| 1077 | vim9script |
| 1078 | var d: dict<string> = {} |
| 1079 | d['key'] = 'value' |
| 1080 | d.somekey = 'someval' |
| 1081 | assert_equal({key: 'value', somekey: 'someval'}, d) |
| 1082 | unlet d.somekey |
| 1083 | assert_equal({key: 'value'}, d) |
| 1084 | END |
| 1085 | CheckScriptSuccess(lines) |
Bram Moolenaar | 334a8b4 | 2020-10-19 16:07:42 +0200 | [diff] [blame] | 1086 | |
Bram Moolenaar | e0de171 | 2020-12-02 17:36:54 +0100 | [diff] [blame] | 1087 | CheckDefFailure(["var d: dict<number> = {a: '', b: true}"], 'E1012: Type mismatch; expected dict<number> but got dict<any>', 1) |
| 1088 | CheckDefFailure(["var d: dict<dict<number>> = {x: {a: '', b: true}}"], 'E1012: Type mismatch; expected dict<dict<number>> but got dict<dict<any>>', 1) |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 1089 | enddef |
| 1090 | |
Bram Moolenaar | 04bdd57 | 2020-09-23 13:25:32 +0200 | [diff] [blame] | 1091 | def Test_assign_dict_unknown_type() |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 1092 | var lines =<< trim END |
Bram Moolenaar | 04bdd57 | 2020-09-23 13:25:32 +0200 | [diff] [blame] | 1093 | vim9script |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 1094 | var mylist = [] |
Bram Moolenaar | e0de171 | 2020-12-02 17:36:54 +0100 | [diff] [blame] | 1095 | mylist += [{one: 'one'}] |
Bram Moolenaar | 04bdd57 | 2020-09-23 13:25:32 +0200 | [diff] [blame] | 1096 | def Func() |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 1097 | var dd = mylist[0] |
Bram Moolenaar | 04bdd57 | 2020-09-23 13:25:32 +0200 | [diff] [blame] | 1098 | assert_equal('one', dd.one) |
| 1099 | enddef |
| 1100 | Func() |
| 1101 | END |
| 1102 | CheckScriptSuccess(lines) |
| 1103 | |
Bram Moolenaar | ce2c544 | 2020-11-28 21:21:17 +0100 | [diff] [blame] | 1104 | lines =<< trim END |
| 1105 | vim9script |
| 1106 | var mylist = [[]] |
Bram Moolenaar | e0de171 | 2020-12-02 17:36:54 +0100 | [diff] [blame] | 1107 | mylist[0] += [{one: 'one'}] |
Bram Moolenaar | ce2c544 | 2020-11-28 21:21:17 +0100 | [diff] [blame] | 1108 | def Func() |
| 1109 | var dd = mylist[0][0] |
| 1110 | assert_equal('one', dd.one) |
| 1111 | enddef |
| 1112 | Func() |
| 1113 | END |
| 1114 | CheckScriptSuccess(lines) |
Bram Moolenaar | 04bdd57 | 2020-09-23 13:25:32 +0200 | [diff] [blame] | 1115 | enddef |
| 1116 | |
Bram Moolenaar | f76ec1e | 2021-03-03 17:58:16 +0100 | [diff] [blame] | 1117 | def Test_assign_dict_with_op() |
| 1118 | var lines =<< trim END |
Bram Moolenaar | f76ec1e | 2021-03-03 17:58:16 +0100 | [diff] [blame] | 1119 | var ds: dict<string> = {a: 'x'} |
| 1120 | ds['a'] ..= 'y' |
| 1121 | ds.a ..= 'z' |
| 1122 | assert_equal('xyz', ds.a) |
| 1123 | |
| 1124 | var dn: dict<number> = {a: 9} |
| 1125 | dn['a'] += 2 |
| 1126 | assert_equal(11, dn.a) |
| 1127 | dn.a += 2 |
| 1128 | assert_equal(13, dn.a) |
| 1129 | |
| 1130 | dn['a'] -= 3 |
| 1131 | assert_equal(10, dn.a) |
| 1132 | dn.a -= 2 |
| 1133 | assert_equal(8, dn.a) |
| 1134 | |
| 1135 | dn['a'] *= 2 |
| 1136 | assert_equal(16, dn.a) |
| 1137 | dn.a *= 2 |
| 1138 | assert_equal(32, dn.a) |
| 1139 | |
| 1140 | dn['a'] /= 3 |
| 1141 | assert_equal(10, dn.a) |
| 1142 | dn.a /= 2 |
| 1143 | assert_equal(5, dn.a) |
| 1144 | |
| 1145 | dn['a'] %= 3 |
| 1146 | assert_equal(2, dn.a) |
| 1147 | dn.a %= 6 |
| 1148 | assert_equal(2, dn.a) |
Bram Moolenaar | b9c0cd8 | 2021-04-05 20:51:00 +0200 | [diff] [blame] | 1149 | |
| 1150 | var dd: dict<dict<list<any>>> |
| 1151 | dd.a = {} |
| 1152 | dd.a.b = [0] |
| 1153 | dd.a.b += [1] |
| 1154 | assert_equal({a: {b: [0, 1]}}, dd) |
Bram Moolenaar | 0d351a0 | 2021-04-05 21:05:26 +0200 | [diff] [blame] | 1155 | |
| 1156 | var dab = {a: ['b']} |
| 1157 | dab.a[0] ..= 'c' |
| 1158 | assert_equal({a: ['bc']}, dab) |
Bram Moolenaar | f76ec1e | 2021-03-03 17:58:16 +0100 | [diff] [blame] | 1159 | END |
Bram Moolenaar | e42939a | 2021-04-05 17:11:17 +0200 | [diff] [blame] | 1160 | CheckDefAndScriptSuccess(lines) |
| 1161 | enddef |
| 1162 | |
| 1163 | def Test_assign_list_with_op() |
| 1164 | var lines =<< trim END |
| 1165 | var ls: list<string> = ['x'] |
| 1166 | ls[0] ..= 'y' |
| 1167 | assert_equal('xy', ls[0]) |
| 1168 | |
| 1169 | var ln: list<number> = [9] |
| 1170 | ln[0] += 2 |
| 1171 | assert_equal(11, ln[0]) |
| 1172 | |
| 1173 | ln[0] -= 3 |
| 1174 | assert_equal(8, ln[0]) |
| 1175 | |
| 1176 | ln[0] *= 2 |
| 1177 | assert_equal(16, ln[0]) |
| 1178 | |
| 1179 | ln[0] /= 3 |
| 1180 | assert_equal(5, ln[0]) |
| 1181 | |
| 1182 | ln[0] %= 3 |
| 1183 | assert_equal(2, ln[0]) |
| 1184 | END |
| 1185 | CheckDefAndScriptSuccess(lines) |
| 1186 | enddef |
| 1187 | |
| 1188 | def Test_assign_with_op_fails() |
| 1189 | var lines =<< trim END |
| 1190 | var s = 'abc' |
| 1191 | s[1] += 'x' |
| 1192 | END |
| 1193 | CheckDefAndScriptFailure2(lines, 'E1141:', 'E689:', 2) |
| 1194 | |
| 1195 | lines =<< trim END |
| 1196 | var s = 'abc' |
| 1197 | s[1] ..= 'x' |
| 1198 | END |
| 1199 | CheckDefAndScriptFailure2(lines, 'E1141:', 'E689:', 2) |
Bram Moolenaar | b9c0cd8 | 2021-04-05 20:51:00 +0200 | [diff] [blame] | 1200 | |
| 1201 | lines =<< trim END |
| 1202 | var dd: dict<dict<list<any>>> |
| 1203 | dd.a = {} |
| 1204 | dd.a.b += [1] |
| 1205 | END |
| 1206 | CheckDefExecAndScriptFailure(lines, 'E716:', 3) |
Bram Moolenaar | f76ec1e | 2021-03-03 17:58:16 +0100 | [diff] [blame] | 1207 | enddef |
| 1208 | |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 1209 | def Test_assign_lambda() |
| 1210 | # check if assign a lambda to a variable which type is func or any. |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 1211 | var lines =<< trim END |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 1212 | vim9script |
Bram Moolenaar | 2949cfd | 2020-12-31 21:28:47 +0100 | [diff] [blame] | 1213 | var FuncRef = () => 123 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 1214 | assert_equal(123, FuncRef()) |
Bram Moolenaar | 2949cfd | 2020-12-31 21:28:47 +0100 | [diff] [blame] | 1215 | var FuncRef_Func: func = () => 123 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 1216 | assert_equal(123, FuncRef_Func()) |
Bram Moolenaar | 2949cfd | 2020-12-31 21:28:47 +0100 | [diff] [blame] | 1217 | var FuncRef_Any: any = () => 123 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 1218 | assert_equal(123, FuncRef_Any()) |
Bram Moolenaar | 0640950 | 2021-02-17 17:00:27 +0100 | [diff] [blame] | 1219 | var FuncRef_Number: func(): number = () => 321 |
| 1220 | assert_equal(321, FuncRef_Number()) |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 1221 | END |
| 1222 | CheckScriptSuccess(lines) |
Bram Moolenaar | 328eac2 | 2021-01-07 19:23:08 +0100 | [diff] [blame] | 1223 | |
| 1224 | lines =<< trim END |
| 1225 | var Ref: func(number) |
| 1226 | Ref = (j) => !j |
| 1227 | END |
Bram Moolenaar | 0640950 | 2021-02-17 17:00:27 +0100 | [diff] [blame] | 1228 | CheckDefAndScriptFailure(lines, 'E1012: Type mismatch; expected func(number) but got func(any): bool') |
Bram Moolenaar | c7dac85 | 2021-02-17 18:49:11 +0100 | [diff] [blame] | 1229 | |
| 1230 | lines =<< trim END |
| 1231 | echo filter([1, 2, 3], (_, v: string) => v + 1) |
| 1232 | END |
| 1233 | CheckDefAndScriptFailure(lines, 'E1051:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 1234 | enddef |
| 1235 | |
Bram Moolenaar | c0e2901 | 2020-09-27 14:22:48 +0200 | [diff] [blame] | 1236 | def Test_heredoc() |
| 1237 | var lines =<< trim END # comment |
| 1238 | text |
| 1239 | END |
| 1240 | assert_equal(['text'], lines) |
| 1241 | |
| 1242 | CheckDefFailure(['var lines =<< trim END X', 'END'], 'E488:') |
| 1243 | CheckDefFailure(['var lines =<< trim END " comment', 'END'], 'E488:') |
Bram Moolenaar | 2914a20 | 2020-09-27 18:24:03 +0200 | [diff] [blame] | 1244 | |
| 1245 | lines =<< trim [END] |
| 1246 | def Func() |
| 1247 | var&lines =<< trim END |
| 1248 | x |
| 1249 | x |
Bram Moolenaar | b5b9480 | 2020-12-13 17:50:20 +0100 | [diff] [blame] | 1250 | enddef |
| 1251 | defcompile |
| 1252 | [END] |
| 1253 | CheckScriptFailure(lines, 'E1145: Missing heredoc end marker: END') |
| 1254 | delfunc! g:Func |
| 1255 | |
| 1256 | lines =<< trim [END] |
| 1257 | def Func() |
| 1258 | var lines =<< trim END |
| 1259 | x |
| 1260 | x |
Bram Moolenaar | 2914a20 | 2020-09-27 18:24:03 +0200 | [diff] [blame] | 1261 | x |
| 1262 | x |
| 1263 | x |
| 1264 | x |
| 1265 | x |
| 1266 | x |
| 1267 | enddef |
| 1268 | call Func() |
| 1269 | [END] |
Bram Moolenaar | b5b9480 | 2020-12-13 17:50:20 +0100 | [diff] [blame] | 1270 | CheckScriptFailure(lines, 'E1145: Missing heredoc end marker: END') |
Bram Moolenaar | 2d870f8 | 2020-12-05 13:41:01 +0100 | [diff] [blame] | 1271 | delfunc! g:Func |
Bram Moolenaar | c0e2901 | 2020-09-27 14:22:48 +0200 | [diff] [blame] | 1272 | enddef |
| 1273 | |
Bram Moolenaar | 083966f | 2020-12-08 11:29:40 +0100 | [diff] [blame] | 1274 | def Test_var_func_call() |
Bram Moolenaar | 55759b5 | 2020-09-30 22:59:42 +0200 | [diff] [blame] | 1275 | var lines =<< trim END |
| 1276 | vim9script |
| 1277 | func GetValue() |
| 1278 | if exists('g:count') |
| 1279 | let g:count += 1 |
| 1280 | else |
| 1281 | let g:count = 1 |
| 1282 | endif |
| 1283 | return 'this' |
| 1284 | endfunc |
| 1285 | var val: string = GetValue() |
| 1286 | # env var is always a string |
| 1287 | var env = $TERM |
| 1288 | END |
| 1289 | writefile(lines, 'Xfinished') |
| 1290 | source Xfinished |
| 1291 | # GetValue() is not called during discovery phase |
| 1292 | assert_equal(1, g:count) |
| 1293 | |
| 1294 | unlet g:count |
| 1295 | delete('Xfinished') |
| 1296 | enddef |
| 1297 | |
Bram Moolenaar | 083966f | 2020-12-08 11:29:40 +0100 | [diff] [blame] | 1298 | def Test_var_missing_type() |
Bram Moolenaar | 55759b5 | 2020-09-30 22:59:42 +0200 | [diff] [blame] | 1299 | var lines =<< trim END |
| 1300 | vim9script |
| 1301 | var name = g:unknown |
| 1302 | END |
| 1303 | CheckScriptFailure(lines, 'E121:') |
| 1304 | |
| 1305 | lines =<< trim END |
| 1306 | vim9script |
| 1307 | var nr: number = 123 |
| 1308 | var name = nr |
| 1309 | END |
| 1310 | CheckScriptSuccess(lines) |
| 1311 | enddef |
| 1312 | |
Bram Moolenaar | 083966f | 2020-12-08 11:29:40 +0100 | [diff] [blame] | 1313 | def Test_var_declaration() |
Bram Moolenaar | 55759b5 | 2020-09-30 22:59:42 +0200 | [diff] [blame] | 1314 | var lines =<< trim END |
| 1315 | vim9script |
| 1316 | var name: string |
| 1317 | g:var_uninit = name |
| 1318 | name = 'text' |
| 1319 | g:var_test = name |
| 1320 | # prefixing s: is optional |
| 1321 | s:name = 'prefixed' |
| 1322 | g:var_prefixed = s:name |
| 1323 | |
Bram Moolenaar | 7257af4 | 2020-12-08 22:08:51 +0100 | [diff] [blame] | 1324 | const FOO: number = 123 |
| 1325 | assert_equal(123, FOO) |
Bram Moolenaar | 89b474d | 2020-12-22 21:19:39 +0100 | [diff] [blame] | 1326 | const FOOS = 'foos' |
| 1327 | assert_equal('foos', FOOS) |
| 1328 | final FLIST = [1] |
| 1329 | assert_equal([1], FLIST) |
| 1330 | FLIST[0] = 11 |
| 1331 | assert_equal([11], FLIST) |
| 1332 | |
| 1333 | const g:FOO: number = 321 |
| 1334 | assert_equal(321, g:FOO) |
| 1335 | const g:FOOS = 'gfoos' |
| 1336 | assert_equal('gfoos', g:FOOS) |
| 1337 | final g:FLIST = [2] |
| 1338 | assert_equal([2], g:FLIST) |
| 1339 | g:FLIST[0] = 22 |
| 1340 | assert_equal([22], g:FLIST) |
| 1341 | |
Bram Moolenaar | d877a57 | 2021-04-01 19:42:48 +0200 | [diff] [blame] | 1342 | def SetGlobalConst() |
| 1343 | const g:globConst = 123 |
| 1344 | enddef |
| 1345 | SetGlobalConst() |
| 1346 | assert_equal(123, g:globConst) |
| 1347 | assert_true(islocked('g:globConst')) |
| 1348 | |
Bram Moolenaar | 89b474d | 2020-12-22 21:19:39 +0100 | [diff] [blame] | 1349 | const w:FOO: number = 46 |
| 1350 | assert_equal(46, w:FOO) |
| 1351 | const w:FOOS = 'wfoos' |
| 1352 | assert_equal('wfoos', w:FOOS) |
| 1353 | final w:FLIST = [3] |
| 1354 | assert_equal([3], w:FLIST) |
| 1355 | w:FLIST[0] = 33 |
| 1356 | assert_equal([33], w:FLIST) |
Bram Moolenaar | 7257af4 | 2020-12-08 22:08:51 +0100 | [diff] [blame] | 1357 | |
Bram Moolenaar | 55759b5 | 2020-09-30 22:59:42 +0200 | [diff] [blame] | 1358 | var s:other: number |
| 1359 | other = 1234 |
| 1360 | g:other_var = other |
| 1361 | |
Bram Moolenaar | ccc25aa | 2021-03-26 21:27:52 +0100 | [diff] [blame] | 1362 | var xyz: string # comment |
| 1363 | |
Bram Moolenaar | 55759b5 | 2020-09-30 22:59:42 +0200 | [diff] [blame] | 1364 | # type is inferred |
Bram Moolenaar | 3862ea3 | 2021-01-01 21:05:55 +0100 | [diff] [blame] | 1365 | var s:dict = {['a']: 222} |
Bram Moolenaar | 55759b5 | 2020-09-30 22:59:42 +0200 | [diff] [blame] | 1366 | def GetDictVal(key: any) |
| 1367 | g:dict_val = s:dict[key] |
| 1368 | enddef |
| 1369 | GetDictVal('a') |
Bram Moolenaar | 0825175 | 2021-01-11 21:20:18 +0100 | [diff] [blame] | 1370 | |
| 1371 | final adict: dict<string> = {} |
| 1372 | def ChangeAdict() |
| 1373 | adict.foo = 'foo' |
| 1374 | enddef |
| 1375 | ChangeAdict() |
Bram Moolenaar | 55759b5 | 2020-09-30 22:59:42 +0200 | [diff] [blame] | 1376 | END |
| 1377 | CheckScriptSuccess(lines) |
| 1378 | assert_equal('', g:var_uninit) |
| 1379 | assert_equal('text', g:var_test) |
| 1380 | assert_equal('prefixed', g:var_prefixed) |
| 1381 | assert_equal(1234, g:other_var) |
| 1382 | assert_equal(222, g:dict_val) |
| 1383 | |
| 1384 | unlet g:var_uninit |
| 1385 | unlet g:var_test |
| 1386 | unlet g:var_prefixed |
| 1387 | unlet g:other_var |
Bram Moolenaar | 6e50ec2 | 2021-04-03 19:32:44 +0200 | [diff] [blame] | 1388 | unlet g:globConst |
Bram Moolenaar | 89b474d | 2020-12-22 21:19:39 +0100 | [diff] [blame] | 1389 | unlet g:FOO |
| 1390 | unlet g:FOOS |
| 1391 | unlet g:FLIST |
| 1392 | unlet w:FOO |
| 1393 | unlet w:FOOS |
| 1394 | unlet w:FLIST |
Bram Moolenaar | 55759b5 | 2020-09-30 22:59:42 +0200 | [diff] [blame] | 1395 | enddef |
| 1396 | |
Bram Moolenaar | 083966f | 2020-12-08 11:29:40 +0100 | [diff] [blame] | 1397 | def Test_var_declaration_fails() |
Bram Moolenaar | 55759b5 | 2020-09-30 22:59:42 +0200 | [diff] [blame] | 1398 | var lines =<< trim END |
| 1399 | vim9script |
| 1400 | final var: string |
| 1401 | END |
| 1402 | CheckScriptFailure(lines, 'E1125:') |
| 1403 | |
| 1404 | lines =<< trim END |
| 1405 | vim9script |
Bram Moolenaar | 89b474d | 2020-12-22 21:19:39 +0100 | [diff] [blame] | 1406 | const g:constvar = 'string' |
| 1407 | g:constvar = 'xx' |
| 1408 | END |
| 1409 | CheckScriptFailure(lines, 'E741:') |
| 1410 | unlet g:constvar |
| 1411 | |
| 1412 | lines =<< trim END |
| 1413 | vim9script |
Bram Moolenaar | f5906aa | 2021-04-02 14:35:15 +0200 | [diff] [blame] | 1414 | var name = 'one' |
| 1415 | lockvar name |
| 1416 | def SetLocked() |
| 1417 | name = 'two' |
| 1418 | enddef |
| 1419 | SetLocked() |
| 1420 | END |
Bram Moolenaar | dcf29ac | 2021-04-02 14:44:02 +0200 | [diff] [blame] | 1421 | CheckScriptFailure(lines, 'E741: Value is locked: name', 1) |
| 1422 | |
| 1423 | lines =<< trim END |
| 1424 | let s:legacy = 'one' |
| 1425 | lockvar s:legacy |
| 1426 | def SetLocked() |
| 1427 | s:legacy = 'two' |
| 1428 | enddef |
| 1429 | call SetLocked() |
| 1430 | END |
| 1431 | CheckScriptFailure(lines, 'E741: Value is locked: s:legacy', 1) |
Bram Moolenaar | f5906aa | 2021-04-02 14:35:15 +0200 | [diff] [blame] | 1432 | |
| 1433 | lines =<< trim END |
| 1434 | vim9script |
Bram Moolenaar | d877a57 | 2021-04-01 19:42:48 +0200 | [diff] [blame] | 1435 | def SetGlobalConst() |
| 1436 | const g:globConst = 123 |
| 1437 | enddef |
| 1438 | SetGlobalConst() |
| 1439 | g:globConst = 234 |
| 1440 | END |
Bram Moolenaar | 6e50ec2 | 2021-04-03 19:32:44 +0200 | [diff] [blame] | 1441 | CheckScriptFailure(lines, 'E741: Value is locked: g:globConst', 6) |
Bram Moolenaar | d877a57 | 2021-04-01 19:42:48 +0200 | [diff] [blame] | 1442 | unlet g:globConst |
| 1443 | |
| 1444 | lines =<< trim END |
| 1445 | vim9script |
Bram Moolenaar | 0825175 | 2021-01-11 21:20:18 +0100 | [diff] [blame] | 1446 | const cdict: dict<string> = {} |
| 1447 | def Change() |
| 1448 | cdict.foo = 'foo' |
| 1449 | enddef |
| 1450 | defcompile |
| 1451 | END |
| 1452 | CheckScriptFailure(lines, 'E46:') |
| 1453 | |
| 1454 | lines =<< trim END |
| 1455 | vim9script |
Bram Moolenaar | 89b474d | 2020-12-22 21:19:39 +0100 | [diff] [blame] | 1456 | final w:finalvar = [9] |
| 1457 | w:finalvar = [8] |
| 1458 | END |
| 1459 | CheckScriptFailure(lines, 'E1122:') |
| 1460 | unlet w:finalvar |
| 1461 | |
| 1462 | lines =<< trim END |
| 1463 | vim9script |
Bram Moolenaar | 55759b5 | 2020-09-30 22:59:42 +0200 | [diff] [blame] | 1464 | const var: string |
| 1465 | END |
| 1466 | CheckScriptFailure(lines, 'E1021:') |
| 1467 | |
| 1468 | lines =<< trim END |
| 1469 | vim9script |
| 1470 | var 9var: string |
| 1471 | END |
Bram Moolenaar | ccc25aa | 2021-03-26 21:27:52 +0100 | [diff] [blame] | 1472 | CheckScriptFailure(lines, 'E488:') |
Bram Moolenaar | 7257af4 | 2020-12-08 22:08:51 +0100 | [diff] [blame] | 1473 | |
| 1474 | CheckDefFailure(['var foo.bar = 2'], 'E1087:') |
| 1475 | CheckDefFailure(['var foo[3] = 2'], 'E1087:') |
| 1476 | CheckDefFailure(['const foo: number'], 'E1021:') |
Bram Moolenaar | 55759b5 | 2020-09-30 22:59:42 +0200 | [diff] [blame] | 1477 | enddef |
| 1478 | |
Bram Moolenaar | 643ce6c | 2021-04-06 21:17:27 +0200 | [diff] [blame^] | 1479 | def Test_script_local_in_legacy() |
| 1480 | # OK to define script-local later when prefixed with s: |
| 1481 | var lines =<< trim END |
| 1482 | def SetLater() |
| 1483 | s:legacy = 'two' |
| 1484 | enddef |
| 1485 | defcompile |
| 1486 | let s:legacy = 'one' |
| 1487 | call SetLater() |
| 1488 | call assert_equal('two', s:legacy) |
| 1489 | END |
| 1490 | CheckScriptSuccess(lines) |
| 1491 | |
| 1492 | # OK to leave out s: prefix when script-local already defined |
| 1493 | lines =<< trim END |
| 1494 | let s:legacy = 'one' |
| 1495 | def SetNoPrefix() |
| 1496 | legacy = 'two' |
| 1497 | enddef |
| 1498 | call SetNoPrefix() |
| 1499 | call assert_equal('two', s:legacy) |
| 1500 | END |
| 1501 | CheckScriptSuccess(lines) |
| 1502 | |
| 1503 | # Not OK to leave out s: prefix when script-local defined later |
| 1504 | lines =<< trim END |
| 1505 | def SetLaterNoPrefix() |
| 1506 | legacy = 'two' |
| 1507 | enddef |
| 1508 | defcompile |
| 1509 | let s:legacy = 'one' |
| 1510 | END |
| 1511 | CheckScriptFailure(lines, 'E476:', 1) |
| 1512 | enddef |
| 1513 | |
Bram Moolenaar | 083966f | 2020-12-08 11:29:40 +0100 | [diff] [blame] | 1514 | def Test_var_type_check() |
Bram Moolenaar | 55759b5 | 2020-09-30 22:59:42 +0200 | [diff] [blame] | 1515 | var lines =<< trim END |
| 1516 | vim9script |
| 1517 | var name: string |
| 1518 | name = 1234 |
| 1519 | END |
| 1520 | CheckScriptFailure(lines, 'E1012:') |
| 1521 | |
| 1522 | lines =<< trim END |
| 1523 | vim9script |
| 1524 | var name:string |
| 1525 | END |
| 1526 | CheckScriptFailure(lines, 'E1069:') |
| 1527 | |
| 1528 | lines =<< trim END |
| 1529 | vim9script |
| 1530 | var name: asdf |
| 1531 | END |
| 1532 | CheckScriptFailure(lines, 'E1010:') |
| 1533 | |
| 1534 | lines =<< trim END |
| 1535 | vim9script |
| 1536 | var s:l: list<number> |
| 1537 | s:l = [] |
| 1538 | END |
| 1539 | CheckScriptSuccess(lines) |
| 1540 | |
| 1541 | lines =<< trim END |
| 1542 | vim9script |
| 1543 | var s:d: dict<number> |
| 1544 | s:d = {} |
| 1545 | END |
| 1546 | CheckScriptSuccess(lines) |
| 1547 | enddef |
| 1548 | |
| 1549 | let g:dict_number = #{one: 1, two: 2} |
| 1550 | |
Bram Moolenaar | 083966f | 2020-12-08 11:29:40 +0100 | [diff] [blame] | 1551 | def Test_var_list_dict_type() |
Bram Moolenaar | 55759b5 | 2020-09-30 22:59:42 +0200 | [diff] [blame] | 1552 | var ll: list<number> |
| 1553 | ll = [1, 2, 2, 3, 3, 3]->uniq() |
| 1554 | ll->assert_equal([1, 2, 3]) |
| 1555 | |
| 1556 | var dd: dict<number> |
| 1557 | dd = g:dict_number |
| 1558 | dd->assert_equal(g:dict_number) |
| 1559 | |
| 1560 | var lines =<< trim END |
| 1561 | var ll: list<number> |
| 1562 | ll = [1, 2, 3]->map('"one"') |
| 1563 | END |
Bram Moolenaar | 70250fb | 2021-01-16 19:01:53 +0100 | [diff] [blame] | 1564 | CheckDefExecFailure(lines, 'E1012: Type mismatch; expected number but got string') |
Bram Moolenaar | 55759b5 | 2020-09-30 22:59:42 +0200 | [diff] [blame] | 1565 | enddef |
| 1566 | |
Bram Moolenaar | 083966f | 2020-12-08 11:29:40 +0100 | [diff] [blame] | 1567 | def Test_cannot_use_let() |
| 1568 | CheckDefAndScriptFailure(['let a = 34'], 'E1126:', 1) |
| 1569 | enddef |
| 1570 | |
Bram Moolenaar | 55759b5 | 2020-09-30 22:59:42 +0200 | [diff] [blame] | 1571 | def Test_unlet() |
| 1572 | g:somevar = 'yes' |
| 1573 | assert_true(exists('g:somevar')) |
| 1574 | unlet g:somevar |
| 1575 | assert_false(exists('g:somevar')) |
| 1576 | unlet! g:somevar |
| 1577 | |
| 1578 | # also works for script-local variable in legacy Vim script |
| 1579 | s:somevar = 'legacy' |
| 1580 | assert_true(exists('s:somevar')) |
| 1581 | unlet s:somevar |
| 1582 | assert_false(exists('s:somevar')) |
| 1583 | unlet! s:somevar |
| 1584 | |
Bram Moolenaar | 0acbf5a | 2021-01-05 20:58:25 +0100 | [diff] [blame] | 1585 | CheckDefExecFailure([ |
| 1586 | 'var dd = 111', |
| 1587 | 'unlet dd', |
| 1588 | ], 'E1081:', 2) |
| 1589 | |
Bram Moolenaar | 752fc69 | 2021-01-04 21:57:11 +0100 | [diff] [blame] | 1590 | # dict unlet |
| 1591 | var dd = {a: 1, b: 2, c: 3} |
| 1592 | unlet dd['a'] |
| 1593 | unlet dd.c |
| 1594 | assert_equal({b: 2}, dd) |
| 1595 | |
| 1596 | # list unlet |
| 1597 | var ll = [1, 2, 3, 4] |
| 1598 | unlet ll[1] |
| 1599 | unlet ll[-1] |
| 1600 | assert_equal([1, 3], ll) |
| 1601 | |
Bram Moolenaar | 5b5ae29 | 2021-02-20 17:04:02 +0100 | [diff] [blame] | 1602 | ll = [1, 2, 3, 4] |
| 1603 | unlet ll[0 : 1] |
| 1604 | assert_equal([3, 4], ll) |
| 1605 | |
| 1606 | ll = [1, 2, 3, 4] |
| 1607 | unlet ll[2 : 8] |
| 1608 | assert_equal([1, 2], ll) |
| 1609 | |
| 1610 | ll = [1, 2, 3, 4] |
| 1611 | unlet ll[-2 : -1] |
| 1612 | assert_equal([1, 2], ll) |
| 1613 | |
| 1614 | CheckDefFailure([ |
| 1615 | 'var ll = [1, 2]', |
| 1616 | 'll[1 : 2] = 7', |
| 1617 | ], 'E1165:', 2) |
| 1618 | CheckDefFailure([ |
| 1619 | 'var dd = {a: 1}', |
| 1620 | 'unlet dd["a" : "a"]', |
| 1621 | ], 'E1166:', 2) |
| 1622 | CheckDefExecFailure([ |
| 1623 | 'unlet g:adict[0 : 1]', |
| 1624 | ], 'E1148:', 1) |
| 1625 | CheckDefFailure([ |
| 1626 | 'var ll = [1, 2]', |
| 1627 | 'unlet ll[0:1]', |
| 1628 | ], 'E1004:', 2) |
| 1629 | CheckDefFailure([ |
| 1630 | 'var ll = [1, 2]', |
| 1631 | 'unlet ll[0 :1]', |
| 1632 | ], 'E1004:', 2) |
| 1633 | CheckDefFailure([ |
| 1634 | 'var ll = [1, 2]', |
| 1635 | 'unlet ll[0: 1]', |
| 1636 | ], 'E1004:', 2) |
| 1637 | |
| 1638 | CheckDefFailure([ |
| 1639 | 'var ll = [1, 2]', |
| 1640 | 'unlet ll["x" : 1]', |
| 1641 | ], 'E1012:', 2) |
| 1642 | CheckDefFailure([ |
| 1643 | 'var ll = [1, 2]', |
| 1644 | 'unlet ll[0 : "x"]', |
| 1645 | ], 'E1012:', 2) |
| 1646 | |
Bram Moolenaar | 752fc69 | 2021-01-04 21:57:11 +0100 | [diff] [blame] | 1647 | # list of dict unlet |
| 1648 | var dl = [{a: 1, b: 2}, {c: 3}] |
| 1649 | unlet dl[0]['b'] |
| 1650 | assert_equal([{a: 1}, {c: 3}], dl) |
| 1651 | |
| 1652 | CheckDefExecFailure([ |
Bram Moolenaar | 0acbf5a | 2021-01-05 20:58:25 +0100 | [diff] [blame] | 1653 | 'var ll = test_null_list()', |
| 1654 | 'unlet ll[0]', |
| 1655 | ], 'E684:', 2) |
Bram Moolenaar | 752fc69 | 2021-01-04 21:57:11 +0100 | [diff] [blame] | 1656 | CheckDefExecFailure([ |
Bram Moolenaar | 0acbf5a | 2021-01-05 20:58:25 +0100 | [diff] [blame] | 1657 | 'var ll = [1]', |
| 1658 | 'unlet ll[2]', |
| 1659 | ], 'E684:', 2) |
Bram Moolenaar | 752fc69 | 2021-01-04 21:57:11 +0100 | [diff] [blame] | 1660 | CheckDefExecFailure([ |
Bram Moolenaar | 0acbf5a | 2021-01-05 20:58:25 +0100 | [diff] [blame] | 1661 | 'var ll = [1]', |
| 1662 | 'unlet ll[g:astring]', |
Bram Moolenaar | f30a14d | 2021-01-17 21:51:24 +0100 | [diff] [blame] | 1663 | ], 'E1012:', 2) |
Bram Moolenaar | 752fc69 | 2021-01-04 21:57:11 +0100 | [diff] [blame] | 1664 | CheckDefExecFailure([ |
Bram Moolenaar | 0acbf5a | 2021-01-05 20:58:25 +0100 | [diff] [blame] | 1665 | 'var dd = test_null_dict()', |
| 1666 | 'unlet dd["a"]', |
| 1667 | ], 'E716:', 2) |
| 1668 | CheckDefExecFailure([ |
| 1669 | 'var dd = {a: 1}', |
| 1670 | 'unlet dd["b"]', |
| 1671 | ], 'E716:', 2) |
| 1672 | CheckDefExecFailure([ |
| 1673 | 'var dd = {a: 1}', |
| 1674 | 'unlet dd[g:alist]', |
| 1675 | ], 'E1105:', 2) |
Bram Moolenaar | 752fc69 | 2021-01-04 21:57:11 +0100 | [diff] [blame] | 1676 | |
Bram Moolenaar | 2ef951d | 2021-01-03 20:55:26 +0100 | [diff] [blame] | 1677 | # can compile unlet before variable exists |
Bram Moolenaar | 752fc69 | 2021-01-04 21:57:11 +0100 | [diff] [blame] | 1678 | g:someDict = {key: 'val'} |
| 1679 | var k = 'key' |
| 1680 | unlet g:someDict[k] |
| 1681 | assert_equal({}, g:someDict) |
| 1682 | unlet g:someDict |
| 1683 | assert_false(exists('g:someDict')) |
Bram Moolenaar | 2ef951d | 2021-01-03 20:55:26 +0100 | [diff] [blame] | 1684 | |
Bram Moolenaar | 55759b5 | 2020-09-30 22:59:42 +0200 | [diff] [blame] | 1685 | CheckScriptFailure([ |
| 1686 | 'vim9script', |
| 1687 | 'var svar = 123', |
| 1688 | 'unlet svar', |
| 1689 | ], 'E1081:') |
| 1690 | CheckScriptFailure([ |
| 1691 | 'vim9script', |
| 1692 | 'var svar = 123', |
| 1693 | 'unlet s:svar', |
| 1694 | ], 'E1081:') |
| 1695 | CheckScriptFailure([ |
| 1696 | 'vim9script', |
| 1697 | 'var svar = 123', |
| 1698 | 'def Func()', |
| 1699 | ' unlet svar', |
| 1700 | 'enddef', |
| 1701 | 'defcompile', |
| 1702 | ], 'E1081:') |
| 1703 | CheckScriptFailure([ |
| 1704 | 'vim9script', |
| 1705 | 'var svar = 123', |
Bram Moolenaar | 9aed729 | 2020-12-18 15:38:00 +0100 | [diff] [blame] | 1706 | 'func Func()', |
| 1707 | ' unlet s:svar', |
| 1708 | 'endfunc', |
| 1709 | 'Func()', |
| 1710 | ], 'E1081:') |
| 1711 | CheckScriptFailure([ |
| 1712 | 'vim9script', |
| 1713 | 'var svar = 123', |
Bram Moolenaar | 55759b5 | 2020-09-30 22:59:42 +0200 | [diff] [blame] | 1714 | 'def Func()', |
| 1715 | ' unlet s:svar', |
| 1716 | 'enddef', |
| 1717 | 'defcompile', |
| 1718 | ], 'E1081:') |
| 1719 | |
Bram Moolenaar | 0acbf5a | 2021-01-05 20:58:25 +0100 | [diff] [blame] | 1720 | writefile(['vim9script', 'export var svar = 1234'], 'XunletExport.vim') |
| 1721 | var lines =<< trim END |
| 1722 | vim9script |
| 1723 | import svar from './XunletExport.vim' |
| 1724 | def UnletSvar() |
| 1725 | unlet svar |
| 1726 | enddef |
| 1727 | defcompile |
| 1728 | END |
| 1729 | CheckScriptFailure(lines, 'E1081:', 1) |
| 1730 | delete('XunletExport.vim') |
| 1731 | |
Bram Moolenaar | 55759b5 | 2020-09-30 22:59:42 +0200 | [diff] [blame] | 1732 | $ENVVAR = 'foobar' |
| 1733 | assert_equal('foobar', $ENVVAR) |
| 1734 | unlet $ENVVAR |
| 1735 | assert_equal('', $ENVVAR) |
| 1736 | enddef |
| 1737 | |
Bram Moolenaar | c5f59fa | 2021-01-21 12:34:14 +0100 | [diff] [blame] | 1738 | def Test_expr_error_no_assign() |
| 1739 | var lines =<< trim END |
| 1740 | vim9script |
| 1741 | var x = invalid |
| 1742 | echo x |
| 1743 | END |
| 1744 | CheckScriptFailureList(lines, ['E121:', 'E121:']) |
| 1745 | |
| 1746 | lines =<< trim END |
| 1747 | vim9script |
| 1748 | var x = 1 / 0 |
| 1749 | echo x |
| 1750 | END |
| 1751 | CheckScriptFailureList(lines, ['E1154:', 'E121:']) |
| 1752 | |
| 1753 | lines =<< trim END |
| 1754 | vim9script |
| 1755 | var x = 1 % 0 |
| 1756 | echo x |
| 1757 | END |
| 1758 | CheckScriptFailureList(lines, ['E1154:', 'E121:']) |
Bram Moolenaar | ccc25aa | 2021-03-26 21:27:52 +0100 | [diff] [blame] | 1759 | |
| 1760 | lines =<< trim END |
| 1761 | var x: string 'string' |
| 1762 | END |
| 1763 | CheckDefAndScriptFailure(lines, 'E488:') |
Bram Moolenaar | c5f59fa | 2021-01-21 12:34:14 +0100 | [diff] [blame] | 1764 | enddef |
| 1765 | |
| 1766 | |
Bram Moolenaar | 17126b1 | 2021-01-07 22:03:02 +0100 | [diff] [blame] | 1767 | def Test_assign_command_modifier() |
| 1768 | var lines =<< trim END |
Bram Moolenaar | 9e0f883 | 2021-01-09 12:09:22 +0100 | [diff] [blame] | 1769 | var verbose = 0 |
| 1770 | verbose = 1 |
| 1771 | assert_equal(1, verbose) |
| 1772 | silent verbose = 2 |
| 1773 | assert_equal(2, verbose) |
| 1774 | silent verbose += 2 |
| 1775 | assert_equal(4, verbose) |
| 1776 | silent verbose -= 1 |
| 1777 | assert_equal(3, verbose) |
Bram Moolenaar | 17126b1 | 2021-01-07 22:03:02 +0100 | [diff] [blame] | 1778 | |
Bram Moolenaar | 9e0f883 | 2021-01-09 12:09:22 +0100 | [diff] [blame] | 1779 | var topleft = {one: 1} |
| 1780 | sandbox topleft.one = 3 |
| 1781 | assert_equal({one: 3}, topleft) |
| 1782 | leftabove topleft[' '] = 4 |
| 1783 | assert_equal({one: 3, ' ': 4}, topleft) |
| 1784 | |
| 1785 | var x: number |
| 1786 | var y: number |
| 1787 | silent [x, y] = [1, 2] |
| 1788 | assert_equal(1, x) |
| 1789 | assert_equal(2, y) |
Bram Moolenaar | 17126b1 | 2021-01-07 22:03:02 +0100 | [diff] [blame] | 1790 | END |
| 1791 | CheckDefAndScriptSuccess(lines) |
| 1792 | enddef |
| 1793 | |
Bram Moolenaar | 3215466 | 2021-03-28 21:14:06 +0200 | [diff] [blame] | 1794 | def Test_script_funcref_case() |
| 1795 | var lines =<< trim END |
| 1796 | var Len = (s: string): number => len(s) + 1 |
| 1797 | assert_equal(5, Len('asdf')) |
| 1798 | END |
| 1799 | CheckDefAndScriptSuccess(lines) |
| 1800 | |
| 1801 | lines =<< trim END |
| 1802 | var len = (s: string): number => len(s) + 1 |
| 1803 | END |
| 1804 | CheckDefAndScriptFailure(lines, 'E704:') |
| 1805 | |
| 1806 | lines =<< trim END |
| 1807 | vim9script |
| 1808 | var s:Len = (s: string): number => len(s) + 2 |
| 1809 | assert_equal(6, Len('asdf')) |
| 1810 | END |
| 1811 | CheckScriptSuccess(lines) |
| 1812 | |
| 1813 | lines =<< trim END |
| 1814 | vim9script |
| 1815 | var s:len = (s: string): number => len(s) + 1 |
| 1816 | END |
| 1817 | CheckScriptFailure(lines, 'E704:') |
| 1818 | enddef |
| 1819 | |
Bram Moolenaar | 55759b5 | 2020-09-30 22:59:42 +0200 | [diff] [blame] | 1820 | |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 1821 | " vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker |