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 | |
Bram Moolenaar | ac56408 | 2020-09-27 19:05:33 +0200 | [diff] [blame] | 6 | " TODO: remove later |
| 7 | let v:disallow_let = 1 |
| 8 | |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 9 | let s:appendToMe = 'xxx' |
| 10 | let s:addToMe = 111 |
| 11 | let g:existing = 'yes' |
| 12 | let g:inc_counter = 1 |
| 13 | let $SOME_ENV_VAR = 'some' |
| 14 | let g:alist = [7] |
| 15 | let g:astring = 'text' |
| 16 | |
| 17 | def Test_assignment_bool() |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 18 | var bool1: bool = true |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 19 | assert_equal(v:true, bool1) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 20 | var bool2: bool = false |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 21 | assert_equal(v:false, bool2) |
| 22 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 23 | var bool3: bool = 0 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 24 | assert_equal(false, bool3) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 25 | var bool4: bool = 1 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 26 | assert_equal(true, bool4) |
| 27 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 28 | var bool5: bool = 'yes' && 'no' |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 29 | assert_equal(true, bool5) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 30 | var bool6: bool = [] && 99 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 31 | assert_equal(false, bool6) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 32 | var bool7: bool = [] || #{a: 1} && 99 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 33 | assert_equal(true, bool7) |
| 34 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 35 | var lines =<< trim END |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 36 | vim9script |
| 37 | def GetFlag(): bool |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 38 | var flag: bool = 1 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 39 | return flag |
| 40 | enddef |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 41 | var flag: bool = GetFlag() |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 42 | assert_equal(true, flag) |
| 43 | flag = 0 |
| 44 | assert_equal(false, flag) |
| 45 | flag = 1 |
| 46 | assert_equal(true, flag) |
| 47 | flag = 99 || 123 |
| 48 | assert_equal(true, flag) |
| 49 | flag = 'yes' && [] |
| 50 | assert_equal(false, flag) |
| 51 | END |
| 52 | CheckScriptSuccess(lines) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 53 | CheckDefAndScriptFailure(['var x: bool = 2'], 'E1012:') |
| 54 | CheckDefAndScriptFailure(['var x: bool = -1'], 'E1012:') |
| 55 | CheckDefAndScriptFailure(['var x: bool = [1]'], 'E1012:') |
| 56 | CheckDefAndScriptFailure(['var x: bool = {}'], 'E1012:') |
| 57 | CheckDefAndScriptFailure(['var x: bool = "x"'], 'E1012:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 58 | enddef |
| 59 | |
| 60 | def Test_syntax() |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 61 | var var = 234 |
| 62 | var other: list<string> = ['asdf'] |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 63 | enddef |
| 64 | |
| 65 | def Test_assignment() |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 66 | CheckDefFailure(['var x:string'], 'E1069:') |
| 67 | CheckDefFailure(['var x:string = "x"'], 'E1069:') |
| 68 | CheckDefFailure(['var a:string = "x"'], 'E1069:') |
| 69 | CheckDefFailure(['var lambda = {-> "lambda"}'], 'E704:') |
| 70 | CheckScriptFailure(['var x = "x"'], 'E1124:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 71 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 72 | var nr: number = 1234 |
| 73 | CheckDefFailure(['var nr: number = "asdf"'], 'E1012:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 74 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 75 | var a: number = 6 #comment |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 76 | assert_equal(6, a) |
| 77 | |
| 78 | if has('channel') |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 79 | var chan1: channel |
| 80 | var job1: job |
| 81 | var job2: job = job_start('willfail') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 82 | endif |
| 83 | if has('float') |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 84 | var float1: float = 3.4 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 85 | endif |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 86 | var Funky1: func |
| 87 | var Funky2: func = function('len') |
| 88 | var Party2: func = funcref('g:Test_syntax') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 89 | |
| 90 | g:newvar = 'new' #comment |
| 91 | assert_equal('new', g:newvar) |
| 92 | |
| 93 | assert_equal('yes', g:existing) |
| 94 | g:existing = 'no' |
| 95 | assert_equal('no', g:existing) |
| 96 | |
| 97 | v:char = 'abc' |
| 98 | assert_equal('abc', v:char) |
| 99 | |
| 100 | $ENVVAR = 'foobar' |
| 101 | assert_equal('foobar', $ENVVAR) |
| 102 | $ENVVAR = '' |
| 103 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 104 | var lines =<< trim END |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 105 | vim9script |
| 106 | $ENVVAR = 'barfoo' |
| 107 | assert_equal('barfoo', $ENVVAR) |
| 108 | $ENVVAR = '' |
| 109 | END |
| 110 | CheckScriptSuccess(lines) |
| 111 | |
| 112 | s:appendToMe ..= 'yyy' |
| 113 | assert_equal('xxxyyy', s:appendToMe) |
| 114 | s:addToMe += 222 |
| 115 | assert_equal(333, s:addToMe) |
| 116 | s:newVar = 'new' |
| 117 | assert_equal('new', s:newVar) |
| 118 | |
| 119 | set ts=7 |
| 120 | &ts += 1 |
| 121 | assert_equal(8, &ts) |
| 122 | &ts -= 3 |
| 123 | assert_equal(5, &ts) |
| 124 | &ts *= 2 |
| 125 | assert_equal(10, &ts) |
| 126 | &ts /= 3 |
| 127 | assert_equal(3, &ts) |
| 128 | set ts=10 |
| 129 | &ts %= 4 |
| 130 | assert_equal(2, &ts) |
| 131 | |
| 132 | if has('float') |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 133 | var f100: float = 100.0 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 134 | f100 /= 5 |
| 135 | assert_equal(20.0, f100) |
| 136 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 137 | var f200: float = 200.0 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 138 | f200 /= 5.0 |
| 139 | assert_equal(40.0, f200) |
| 140 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 141 | CheckDefFailure(['var nr: number = 200', 'nr /= 5.0'], 'E1012:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 142 | endif |
| 143 | |
| 144 | lines =<< trim END |
| 145 | &ts = 6 |
| 146 | &ts += 3 |
| 147 | assert_equal(9, &ts) |
| 148 | |
| 149 | &l:ts = 6 |
| 150 | assert_equal(6, &ts) |
| 151 | &l:ts += 2 |
| 152 | assert_equal(8, &ts) |
| 153 | |
| 154 | &g:ts = 6 |
| 155 | assert_equal(6, &g:ts) |
| 156 | &g:ts += 2 |
| 157 | assert_equal(8, &g:ts) |
| 158 | END |
| 159 | CheckDefAndScriptSuccess(lines) |
| 160 | |
| 161 | CheckDefFailure(['¬ex += 3'], 'E113:') |
| 162 | CheckDefFailure(['&ts ..= "xxx"'], 'E1019:') |
| 163 | CheckDefFailure(['&ts = [7]'], 'E1012:') |
| 164 | CheckDefExecFailure(['&ts = g:alist'], 'E1012: Type mismatch; expected number but got list<number>') |
| 165 | CheckDefFailure(['&ts = "xx"'], 'E1012:') |
| 166 | CheckDefExecFailure(['&ts = g:astring'], 'E1012: Type mismatch; expected number but got string') |
| 167 | CheckDefFailure(['&path += 3'], 'E1012:') |
| 168 | CheckDefExecFailure(['&bs = "asdf"'], 'E474:') |
| 169 | # test freeing ISN_STOREOPT |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 170 | CheckDefFailure(['&ts = 3', 'var asdf'], 'E1022:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 171 | &ts = 8 |
| 172 | |
| 173 | lines =<< trim END |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 174 | var save_TI = &t_TI |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 175 | &t_TI = '' |
| 176 | assert_equal('', &t_TI) |
| 177 | &t_TI = 'xxx' |
| 178 | assert_equal('xxx', &t_TI) |
| 179 | &t_TI = save_TI |
| 180 | END |
| 181 | CheckDefAndScriptSuccess(lines) |
| 182 | |
| 183 | CheckDefFailure(['&t_TI = 123'], 'E1012:') |
| 184 | CheckScriptFailure(['vim9script', '&t_TI = 123'], 'E928:') |
| 185 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 186 | CheckDefFailure(['var s:var = 123'], 'E1101:') |
| 187 | CheckDefFailure(['var s:var: number'], 'E1101:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 188 | |
| 189 | lines =<< trim END |
| 190 | vim9script |
| 191 | def SomeFunc() |
| 192 | s:var = 123 |
| 193 | enddef |
| 194 | defcompile |
| 195 | END |
| 196 | CheckScriptFailure(lines, 'E1089:') |
| 197 | |
| 198 | g:inc_counter += 1 |
| 199 | assert_equal(2, g:inc_counter) |
| 200 | |
| 201 | $SOME_ENV_VAR ..= 'more' |
| 202 | assert_equal('somemore', $SOME_ENV_VAR) |
| 203 | CheckDefFailure(['$SOME_ENV_VAR += "more"'], 'E1051:') |
| 204 | CheckDefFailure(['$SOME_ENV_VAR += 123'], 'E1012:') |
| 205 | |
| 206 | lines =<< trim END |
| 207 | @c = 'areg' |
| 208 | @c ..= 'add' |
| 209 | assert_equal('aregadd', @c) |
| 210 | END |
| 211 | CheckDefAndScriptSuccess(lines) |
| 212 | |
| 213 | CheckDefFailure(['@a += "more"'], 'E1051:') |
| 214 | CheckDefFailure(['@a += 123'], 'E1012:') |
| 215 | |
| 216 | v:errmsg = 'none' |
| 217 | v:errmsg ..= 'again' |
| 218 | assert_equal('noneagain', v:errmsg) |
| 219 | CheckDefFailure(['v:errmsg += "more"'], 'E1051:') |
| 220 | CheckDefFailure(['v:errmsg += 123'], 'E1012:') |
| 221 | |
| 222 | # this should not leak |
| 223 | if 0 |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 224 | var text =<< trim END |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 225 | some text |
| 226 | END |
| 227 | endif |
| 228 | enddef |
| 229 | |
Bram Moolenaar | 81ed496 | 2020-09-23 15:56:50 +0200 | [diff] [blame] | 230 | def Test_extend_list() |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 231 | var lines =<< trim END |
Bram Moolenaar | 81ed496 | 2020-09-23 15:56:50 +0200 | [diff] [blame] | 232 | vim9script |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 233 | var l: list<number> |
Bram Moolenaar | 81ed496 | 2020-09-23 15:56:50 +0200 | [diff] [blame] | 234 | l += [123] |
| 235 | assert_equal([123], l) |
| 236 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 237 | var d: dict<number> |
Bram Moolenaar | 81ed496 | 2020-09-23 15:56:50 +0200 | [diff] [blame] | 238 | d['one'] = 1 |
| 239 | assert_equal(#{one: 1}, d) |
| 240 | END |
| 241 | CheckScriptSuccess(lines) |
| 242 | enddef |
| 243 | |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 244 | def Test_single_letter_vars() |
| 245 | # single letter variables |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 246 | var a: number = 123 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 247 | a = 123 |
| 248 | assert_equal(123, a) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 249 | var b: number |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 250 | b = 123 |
| 251 | assert_equal(123, b) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 252 | var g: number |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 253 | g = 123 |
| 254 | assert_equal(123, g) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 255 | var s: number |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 256 | s = 123 |
| 257 | assert_equal(123, s) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 258 | var t: number |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 259 | t = 123 |
| 260 | assert_equal(123, t) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 261 | var v: number |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 262 | v = 123 |
| 263 | assert_equal(123, v) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 264 | var w: number |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 265 | w = 123 |
| 266 | assert_equal(123, w) |
| 267 | enddef |
| 268 | |
| 269 | def Test_vim9_single_char_vars() |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 270 | var lines =<< trim END |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 271 | vim9script |
| 272 | |
| 273 | # single character variable declarations work |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 274 | var a: string |
| 275 | var b: number |
| 276 | var l: list<any> |
| 277 | var s: string |
| 278 | var t: number |
| 279 | var v: number |
| 280 | var w: number |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 281 | |
| 282 | # script-local variables can be used without s: prefix |
| 283 | a = 'script-a' |
| 284 | b = 111 |
| 285 | l = [1, 2, 3] |
| 286 | s = 'script-s' |
| 287 | t = 222 |
| 288 | v = 333 |
| 289 | w = 444 |
| 290 | |
| 291 | assert_equal('script-a', a) |
| 292 | assert_equal(111, b) |
| 293 | assert_equal([1, 2, 3], l) |
| 294 | assert_equal('script-s', s) |
| 295 | assert_equal(222, t) |
| 296 | assert_equal(333, v) |
| 297 | assert_equal(444, w) |
| 298 | END |
| 299 | writefile(lines, 'Xsinglechar') |
| 300 | source Xsinglechar |
| 301 | delete('Xsinglechar') |
| 302 | enddef |
| 303 | |
| 304 | def Test_assignment_list() |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 305 | var list1: list<bool> = [false, true, false] |
| 306 | var list2: list<number> = [1, 2, 3] |
| 307 | var list3: list<string> = ['sdf', 'asdf'] |
| 308 | var list4: list<any> = ['yes', true, 1234] |
| 309 | var list5: list<blob> = [0z01, 0z02] |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 310 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 311 | var listS: list<string> = [] |
| 312 | var listN: list<number> = [] |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 313 | |
| 314 | assert_equal([1, 2, 3], list2) |
| 315 | list2[-1] = 99 |
| 316 | assert_equal([1, 2, 99], list2) |
| 317 | list2[-2] = 88 |
| 318 | assert_equal([1, 88, 99], list2) |
| 319 | list2[-3] = 77 |
| 320 | assert_equal([77, 88, 99], list2) |
| 321 | list2 += [100] |
| 322 | assert_equal([77, 88, 99, 100], list2) |
| 323 | |
| 324 | list3 += ['end'] |
| 325 | assert_equal(['sdf', 'asdf', 'end'], list3) |
| 326 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 327 | CheckDefExecFailure(['var ll = [1, 2, 3]', 'll[-4] = 6'], 'E684:') |
| 328 | CheckDefExecFailure(['var [v1, v2] = [1, 2]'], 'E1092:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 329 | |
| 330 | # type becomes list<any> |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 331 | var somelist = rand() > 0 ? [1, 2, 3] : ['a', 'b', 'c'] |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 332 | enddef |
| 333 | |
| 334 | def Test_assignment_list_vim9script() |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 335 | var lines =<< trim END |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 336 | vim9script |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 337 | var v1: number |
| 338 | var v2: number |
| 339 | var v3: number |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 340 | [v1, v2, v3] = [1, 2, 3] |
| 341 | assert_equal([1, 2, 3], [v1, v2, v3]) |
| 342 | END |
| 343 | CheckScriptSuccess(lines) |
| 344 | enddef |
| 345 | |
| 346 | def Test_assignment_dict() |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 347 | var dict1: dict<bool> = #{one: false, two: true} |
| 348 | var dict2: dict<number> = #{one: 1, two: 2} |
| 349 | var dict3: dict<string> = #{key: 'value'} |
| 350 | var dict4: dict<any> = #{one: 1, two: '2'} |
| 351 | var dict5: dict<blob> = #{one: 0z01, two: 0z02} |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 352 | |
| 353 | # overwrite |
| 354 | dict3['key'] = 'another' |
| 355 | |
| 356 | # empty key can be used |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 357 | var dd = {} |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 358 | dd[""] = 6 |
| 359 | assert_equal({'': 6}, dd) |
| 360 | |
| 361 | # type becomes dict<any> |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 362 | var somedict = rand() > 0 ? #{a: 1, b: 2} : #{a: 'a', b: 'b'} |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 363 | |
| 364 | # assignment to script-local dict |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 365 | var lines =<< trim END |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 366 | vim9script |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 367 | var test: dict<any> = {} |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 368 | def FillDict(): dict<any> |
| 369 | test['a'] = 43 |
| 370 | return test |
| 371 | enddef |
| 372 | assert_equal(#{a: 43}, FillDict()) |
| 373 | END |
| 374 | CheckScriptSuccess(lines) |
| 375 | |
| 376 | lines =<< trim END |
| 377 | vim9script |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 378 | var test: dict<any> |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 379 | def FillDict(): dict<any> |
| 380 | test['a'] = 43 |
| 381 | return test |
| 382 | enddef |
| 383 | FillDict() |
| 384 | END |
| 385 | CheckScriptFailure(lines, 'E1103:') |
| 386 | |
| 387 | # assignment to global dict |
| 388 | lines =<< trim END |
| 389 | vim9script |
| 390 | g:test = {} |
| 391 | def FillDict(): dict<any> |
| 392 | g:test['a'] = 43 |
| 393 | return g:test |
| 394 | enddef |
| 395 | assert_equal(#{a: 43}, FillDict()) |
| 396 | END |
| 397 | CheckScriptSuccess(lines) |
| 398 | |
| 399 | # assignment to buffer dict |
| 400 | lines =<< trim END |
| 401 | vim9script |
| 402 | b:test = {} |
| 403 | def FillDict(): dict<any> |
| 404 | b:test['a'] = 43 |
| 405 | return b:test |
| 406 | enddef |
| 407 | assert_equal(#{a: 43}, FillDict()) |
| 408 | END |
| 409 | CheckScriptSuccess(lines) |
| 410 | enddef |
| 411 | |
| 412 | def Test_assignment_local() |
| 413 | # Test in a separated file in order not to the current buffer/window/tab is |
| 414 | # changed. |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 415 | var script_lines: list<string> =<< trim END |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 416 | let b:existing = 'yes' |
| 417 | let w:existing = 'yes' |
| 418 | let t:existing = 'yes' |
| 419 | |
| 420 | def Test_assignment_local_internal() |
| 421 | b:newvar = 'new' |
| 422 | assert_equal('new', b:newvar) |
| 423 | assert_equal('yes', b:existing) |
| 424 | b:existing = 'no' |
| 425 | assert_equal('no', b:existing) |
| 426 | b:existing ..= 'NO' |
| 427 | assert_equal('noNO', b:existing) |
| 428 | |
| 429 | w:newvar = 'new' |
| 430 | assert_equal('new', w:newvar) |
| 431 | assert_equal('yes', w:existing) |
| 432 | w:existing = 'no' |
| 433 | assert_equal('no', w:existing) |
| 434 | w:existing ..= 'NO' |
| 435 | assert_equal('noNO', w:existing) |
| 436 | |
| 437 | t:newvar = 'new' |
| 438 | assert_equal('new', t:newvar) |
| 439 | assert_equal('yes', t:existing) |
| 440 | t:existing = 'no' |
| 441 | assert_equal('no', t:existing) |
| 442 | t:existing ..= 'NO' |
| 443 | assert_equal('noNO', t:existing) |
| 444 | enddef |
| 445 | call Test_assignment_local_internal() |
| 446 | END |
| 447 | CheckScriptSuccess(script_lines) |
| 448 | enddef |
| 449 | |
| 450 | def Test_assignment_default() |
| 451 | |
| 452 | # Test default values. |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 453 | var thebool: bool |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 454 | assert_equal(v:false, thebool) |
| 455 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 456 | var thenumber: number |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 457 | assert_equal(0, thenumber) |
| 458 | |
| 459 | if has('float') |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 460 | var thefloat: float |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 461 | assert_equal(0.0, thefloat) |
| 462 | endif |
| 463 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 464 | var thestring: string |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 465 | assert_equal('', thestring) |
| 466 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 467 | var theblob: blob |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 468 | assert_equal(0z, theblob) |
| 469 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 470 | var Thefunc: func |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 471 | assert_equal(test_null_function(), Thefunc) |
| 472 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 473 | var thelist: list<any> |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 474 | assert_equal([], thelist) |
| 475 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 476 | var thedict: dict<any> |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 477 | assert_equal({}, thedict) |
| 478 | |
| 479 | if has('channel') |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 480 | var thejob: job |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 481 | assert_equal(test_null_job(), thejob) |
| 482 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 483 | var thechannel: channel |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 484 | assert_equal(test_null_channel(), thechannel) |
| 485 | |
| 486 | if has('unix') && executable('cat') |
| 487 | # check with non-null job and channel, types must match |
| 488 | thejob = job_start("cat ", #{}) |
| 489 | thechannel = job_getchannel(thejob) |
| 490 | job_stop(thejob, 'kill') |
| 491 | endif |
| 492 | endif |
| 493 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 494 | var nr = 1234 | nr = 5678 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 495 | assert_equal(5678, nr) |
| 496 | enddef |
| 497 | |
| 498 | def Test_assignment_var_list() |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 499 | var v1: string |
| 500 | var v2: string |
| 501 | var vrem: list<string> |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 502 | [v1] = ['aaa'] |
| 503 | assert_equal('aaa', v1) |
| 504 | |
| 505 | [v1, v2] = ['one', 'two'] |
| 506 | assert_equal('one', v1) |
| 507 | assert_equal('two', v2) |
| 508 | |
| 509 | [v1, v2; vrem] = ['one', 'two'] |
| 510 | assert_equal('one', v1) |
| 511 | assert_equal('two', v2) |
| 512 | assert_equal([], vrem) |
| 513 | |
| 514 | [v1, v2; vrem] = ['one', 'two', 'three'] |
| 515 | assert_equal('one', v1) |
| 516 | assert_equal('two', v2) |
| 517 | assert_equal(['three'], vrem) |
| 518 | |
| 519 | [&ts, &sw] = [3, 4] |
| 520 | assert_equal(3, &ts) |
| 521 | assert_equal(4, &sw) |
| 522 | set ts=8 sw=4 |
| 523 | enddef |
| 524 | |
| 525 | def Test_assignment_vim9script() |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 526 | var lines =<< trim END |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 527 | vim9script |
| 528 | def Func(): list<number> |
| 529 | return [1, 2] |
| 530 | enddef |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 531 | var var1: number |
| 532 | var var2: number |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 533 | [var1, var2] = |
| 534 | Func() |
| 535 | assert_equal(1, var1) |
| 536 | assert_equal(2, var2) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 537 | var ll = |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 538 | Func() |
| 539 | assert_equal([1, 2], ll) |
| 540 | |
| 541 | @/ = 'text' |
| 542 | assert_equal('text', @/) |
| 543 | @0 = 'zero' |
| 544 | assert_equal('zero', @0) |
| 545 | @1 = 'one' |
| 546 | assert_equal('one', @1) |
| 547 | @9 = 'nine' |
| 548 | assert_equal('nine', @9) |
| 549 | @- = 'minus' |
| 550 | assert_equal('minus', @-) |
| 551 | if has('clipboard_working') |
| 552 | @* = 'star' |
| 553 | assert_equal('star', @*) |
| 554 | @+ = 'plus' |
| 555 | assert_equal('plus', @+) |
| 556 | endif |
| 557 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 558 | var a: number = 123 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 559 | assert_equal(123, a) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 560 | var s: string = 'yes' |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 561 | assert_equal('yes', s) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 562 | var b: number = 42 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 563 | assert_equal(42, b) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 564 | var w: number = 43 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 565 | assert_equal(43, w) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 566 | var t: number = 44 |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 567 | assert_equal(44, t) |
| 568 | END |
| 569 | CheckScriptSuccess(lines) |
Bram Moolenaar | 06f9c69 | 2020-09-27 21:27:40 +0200 | [diff] [blame^] | 570 | |
| 571 | lines =<< trim END |
| 572 | vim9script |
| 573 | var n: number |
| 574 | def Func() |
| 575 | n = 'string' |
| 576 | enddef |
| 577 | defcompile |
| 578 | END |
| 579 | CheckScriptFailure(lines, 'E1012: Type mismatch; expected number but got string') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 580 | enddef |
| 581 | |
| 582 | def Mess(): string |
| 583 | v:foldstart = 123 |
| 584 | return 'xxx' |
| 585 | enddef |
| 586 | |
| 587 | def Test_assignment_failure() |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 588 | CheckDefFailure(['var var=234'], 'E1004:') |
| 589 | CheckDefFailure(['var var =234'], 'E1004:') |
| 590 | CheckDefFailure(['var var= 234'], 'E1004:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 591 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 592 | CheckScriptFailure(['vim9script', 'var var=234'], 'E1004:') |
| 593 | CheckScriptFailure(['vim9script', 'var var=234'], "before and after '='") |
| 594 | CheckScriptFailure(['vim9script', 'var var =234'], 'E1004:') |
| 595 | CheckScriptFailure(['vim9script', 'var var= 234'], 'E1004:') |
| 596 | CheckScriptFailure(['vim9script', 'var var = 234', 'var+=234'], 'E1004:') |
| 597 | CheckScriptFailure(['vim9script', 'var var = 234', 'var+=234'], "before and after '+='") |
| 598 | CheckScriptFailure(['vim9script', 'var var = "x"', 'var..="y"'], 'E1004:') |
| 599 | CheckScriptFailure(['vim9script', 'var var = "x"', 'var..="y"'], "before and after '..='") |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 600 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 601 | CheckDefFailure(['var true = 1'], 'E1034:') |
| 602 | CheckDefFailure(['var false = 1'], 'E1034:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 603 | |
| 604 | CheckDefFailure(['[a; b; c] = g:list'], 'E452:') |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 605 | CheckDefExecFailure(['var a: number', |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 606 | '[a] = test_null_list()'], 'E1093:') |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 607 | CheckDefExecFailure(['var a: number', |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 608 | '[a] = []'], 'E1093:') |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 609 | CheckDefExecFailure(['var x: number', |
| 610 | 'var y: number', |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 611 | '[x, y] = [1]'], 'E1093:') |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 612 | CheckDefExecFailure(['var x: number', |
| 613 | 'var y: number', |
| 614 | 'var z: list<number>', |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 615 | '[x, y; z] = [1]'], 'E1093:') |
| 616 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 617 | CheckDefFailure(['var somevar'], "E1022:") |
| 618 | CheckDefFailure(['var &tabstop = 4'], 'E1052:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 619 | CheckDefFailure(['&g:option = 5'], 'E113:') |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 620 | CheckScriptFailure(['vim9script', 'var &tabstop = 4'], 'E1052:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 621 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 622 | CheckDefFailure(['var $VAR = 5'], 'E1016: Cannot declare an environment variable:') |
| 623 | CheckScriptFailure(['vim9script', 'var $ENV = "xxx"'], 'E1016:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 624 | |
| 625 | if has('dnd') |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 626 | CheckDefFailure(['var @~ = 5'], 'E1066:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 627 | else |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 628 | CheckDefFailure(['var @~ = 5'], 'E354:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 629 | CheckDefFailure(['@~ = 5'], 'E354:') |
| 630 | endif |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 631 | CheckDefFailure(['var @a = 5'], 'E1066:') |
| 632 | CheckDefFailure(['var @/ = "x"'], 'E1066:') |
| 633 | CheckScriptFailure(['vim9script', 'var @a = "abc"'], 'E1066:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 634 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 635 | CheckDefFailure(['var g:var = 5'], 'E1016: Cannot declare a global variable:') |
| 636 | CheckDefFailure(['var w:var = 5'], 'E1016: Cannot declare a window variable:') |
| 637 | CheckDefFailure(['var b:var = 5'], 'E1016: Cannot declare a buffer variable:') |
| 638 | CheckDefFailure(['var t:var = 5'], 'E1016: Cannot declare a tab variable:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 639 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 640 | CheckDefFailure(['var anr = 4', 'anr ..= "text"'], 'E1019:') |
| 641 | CheckDefFailure(['var xnr += 4'], 'E1020:', 1) |
| 642 | CheckScriptFailure(['vim9script', 'var xnr += 4'], 'E1020:') |
| 643 | CheckDefFailure(["var xnr = xnr + 1"], 'E1001:', 1) |
| 644 | CheckScriptFailure(['vim9script', 'var xnr = xnr + 4'], 'E121:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 645 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 646 | CheckScriptFailure(['vim9script', 'def Func()', 'var dummy = s:notfound', 'enddef', 'defcompile'], 'E1108:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 647 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 648 | CheckDefFailure(['var var: list<string> = [123]'], 'expected list<string> but got list<number>') |
| 649 | CheckDefFailure(['var var: list<number> = ["xx"]'], 'expected list<number> but got list<string>') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 650 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 651 | CheckDefFailure(['var var: dict<string> = #{key: 123}'], 'expected dict<string> but got dict<number>') |
| 652 | CheckDefFailure(['var var: dict<number> = #{key: "xx"}'], 'expected dict<number> but got dict<string>') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 653 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 654 | CheckDefFailure(['var var = feedkeys("0")'], 'E1031:') |
| 655 | CheckDefFailure(['var var: number = feedkeys("0")'], 'expected number but got void') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 656 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 657 | CheckDefFailure(['var var: dict <number>'], 'E1068:') |
| 658 | CheckDefFailure(['var var: dict<number'], 'E1009:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 659 | |
| 660 | assert_fails('s/^/\=Mess()/n', 'E794:') |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 661 | CheckDefFailure(['var var: dict<number'], 'E1009:') |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 662 | |
| 663 | CheckDefFailure(['w:foo: number = 10'], |
| 664 | 'E488: Trailing characters: : number = 1') |
| 665 | CheckDefFailure(['t:foo: bool = true'], |
| 666 | 'E488: Trailing characters: : bool = true') |
| 667 | CheckDefFailure(['b:foo: string = "x"'], |
| 668 | 'E488: Trailing characters: : string = "x"') |
| 669 | CheckDefFailure(['g:foo: number = 123'], |
| 670 | 'E488: Trailing characters: : number = 123') |
| 671 | enddef |
| 672 | |
| 673 | def Test_assign_list() |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 674 | var l: list<string> = [] |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 675 | l[0] = 'value' |
| 676 | assert_equal('value', l[0]) |
| 677 | |
| 678 | l[1] = 'asdf' |
| 679 | assert_equal('value', l[0]) |
| 680 | assert_equal('asdf', l[1]) |
| 681 | assert_equal('asdf', l[-1]) |
| 682 | assert_equal('value', l[-2]) |
| 683 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 684 | var nrl: list<number> = [] |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 685 | for i in range(5) |
| 686 | nrl[i] = i |
| 687 | endfor |
| 688 | assert_equal([0, 1, 2, 3, 4], nrl) |
| 689 | enddef |
| 690 | |
| 691 | def Test_assign_dict() |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 692 | var d: dict<string> = {} |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 693 | d['key'] = 'value' |
| 694 | assert_equal('value', d['key']) |
| 695 | |
| 696 | d[123] = 'qwerty' |
| 697 | assert_equal('qwerty', d[123]) |
| 698 | assert_equal('qwerty', d['123']) |
| 699 | |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 700 | var nrd: dict<number> = {} |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 701 | for i in range(3) |
| 702 | nrd[i] = i |
| 703 | endfor |
| 704 | assert_equal({'0': 0, '1': 1, '2': 2}, nrd) |
| 705 | enddef |
| 706 | |
Bram Moolenaar | 04bdd57 | 2020-09-23 13:25:32 +0200 | [diff] [blame] | 707 | def Test_assign_dict_unknown_type() |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 708 | var lines =<< trim END |
Bram Moolenaar | 04bdd57 | 2020-09-23 13:25:32 +0200 | [diff] [blame] | 709 | vim9script |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 710 | var mylist = [] |
Bram Moolenaar | 04bdd57 | 2020-09-23 13:25:32 +0200 | [diff] [blame] | 711 | mylist += [#{one: 'one'}] |
| 712 | def Func() |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 713 | var dd = mylist[0] |
Bram Moolenaar | 04bdd57 | 2020-09-23 13:25:32 +0200 | [diff] [blame] | 714 | assert_equal('one', dd.one) |
| 715 | enddef |
| 716 | Func() |
| 717 | END |
| 718 | CheckScriptSuccess(lines) |
| 719 | |
| 720 | # doesn't work yet |
| 721 | #lines =<< trim END |
| 722 | # vim9script |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 723 | # var mylist = [[]] |
Bram Moolenaar | 04bdd57 | 2020-09-23 13:25:32 +0200 | [diff] [blame] | 724 | # mylist[0] += [#{one: 'one'}] |
| 725 | # def Func() |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 726 | # var dd = mylist[0][0] |
Bram Moolenaar | 04bdd57 | 2020-09-23 13:25:32 +0200 | [diff] [blame] | 727 | # assert_equal('one', dd.one) |
| 728 | # enddef |
| 729 | # Func() |
| 730 | #END |
| 731 | #CheckScriptSuccess(lines) |
| 732 | enddef |
| 733 | |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 734 | def Test_assign_lambda() |
| 735 | # 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] | 736 | var lines =<< trim END |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 737 | vim9script |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 738 | var FuncRef = {->123} |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 739 | assert_equal(123, FuncRef()) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 740 | var FuncRef_Func: func = {->123} |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 741 | assert_equal(123, FuncRef_Func()) |
Bram Moolenaar | 30fd820 | 2020-09-26 15:09:30 +0200 | [diff] [blame] | 742 | var FuncRef_Any: any = {->123} |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 743 | assert_equal(123, FuncRef_Any()) |
| 744 | END |
| 745 | CheckScriptSuccess(lines) |
| 746 | enddef |
| 747 | |
Bram Moolenaar | c0e2901 | 2020-09-27 14:22:48 +0200 | [diff] [blame] | 748 | def Test_heredoc() |
| 749 | var lines =<< trim END # comment |
| 750 | text |
| 751 | END |
| 752 | assert_equal(['text'], lines) |
| 753 | |
| 754 | CheckDefFailure(['var lines =<< trim END X', 'END'], 'E488:') |
| 755 | CheckDefFailure(['var lines =<< trim END " comment', 'END'], 'E488:') |
Bram Moolenaar | 2914a20 | 2020-09-27 18:24:03 +0200 | [diff] [blame] | 756 | |
| 757 | lines =<< trim [END] |
| 758 | def Func() |
| 759 | var&lines =<< trim END |
| 760 | x |
| 761 | x |
| 762 | x |
| 763 | x |
| 764 | x |
| 765 | x |
| 766 | x |
| 767 | x |
| 768 | enddef |
| 769 | call Func() |
| 770 | [END] |
| 771 | CheckScriptFailure(lines, 'E990:') |
Bram Moolenaar | c0e2901 | 2020-09-27 14:22:48 +0200 | [diff] [blame] | 772 | enddef |
| 773 | |
Bram Moolenaar | 971caf5 | 2020-09-21 22:21:42 +0200 | [diff] [blame] | 774 | " vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker |