Bram Moolenaar | 00b28d6 | 2022-12-08 15:32:33 +0000 | [diff] [blame] | 1 | " Test Vim9 classes |
| 2 | |
| 3 | source check.vim |
| 4 | import './vim9.vim' as v9 |
| 5 | |
| 6 | def Test_class_basic() |
| 7 | var lines =<< trim END |
| 8 | class NotWorking |
| 9 | endclass |
| 10 | END |
| 11 | v9.CheckScriptFailure(lines, 'E1316:') |
| 12 | |
| 13 | lines =<< trim END |
| 14 | vim9script |
| 15 | class notWorking |
| 16 | endclass |
| 17 | END |
| 18 | v9.CheckScriptFailure(lines, 'E1314:') |
| 19 | |
| 20 | lines =<< trim END |
| 21 | vim9script |
| 22 | class Not@working |
| 23 | endclass |
| 24 | END |
| 25 | v9.CheckScriptFailure(lines, 'E1315:') |
| 26 | |
| 27 | lines =<< trim END |
| 28 | vim9script |
| 29 | abstract noclass Something |
| 30 | endclass |
| 31 | END |
| 32 | v9.CheckScriptFailure(lines, 'E475:') |
| 33 | |
| 34 | lines =<< trim END |
| 35 | vim9script |
| 36 | abstract classy Something |
| 37 | endclass |
| 38 | END |
| 39 | v9.CheckScriptFailure(lines, 'E475:') |
| 40 | |
| 41 | lines =<< trim END |
| 42 | vim9script |
| 43 | class Something |
| 44 | endcl |
| 45 | END |
| 46 | v9.CheckScriptFailure(lines, 'E1065:') |
| 47 | |
| 48 | lines =<< trim END |
| 49 | vim9script |
| 50 | class Something |
| 51 | endclass school's out |
| 52 | END |
| 53 | v9.CheckScriptFailure(lines, 'E488:') |
| 54 | |
| 55 | lines =<< trim END |
| 56 | vim9script |
| 57 | class Something |
| 58 | endclass | echo 'done' |
| 59 | END |
| 60 | v9.CheckScriptFailure(lines, 'E488:') |
| 61 | |
| 62 | lines =<< trim END |
| 63 | vim9script |
| 64 | class Something |
| 65 | this |
| 66 | endclass |
| 67 | END |
| 68 | v9.CheckScriptFailure(lines, 'E1317:') |
| 69 | |
| 70 | lines =<< trim END |
| 71 | vim9script |
| 72 | class Something |
| 73 | this. |
| 74 | endclass |
| 75 | END |
| 76 | v9.CheckScriptFailure(lines, 'E1317:') |
| 77 | |
| 78 | lines =<< trim END |
| 79 | vim9script |
| 80 | class Something |
| 81 | this .count |
| 82 | endclass |
| 83 | END |
| 84 | v9.CheckScriptFailure(lines, 'E1317:') |
| 85 | |
| 86 | lines =<< trim END |
| 87 | vim9script |
| 88 | class Something |
| 89 | this. count |
| 90 | endclass |
| 91 | END |
| 92 | v9.CheckScriptFailure(lines, 'E1317:') |
| 93 | |
| 94 | lines =<< trim END |
| 95 | vim9script |
| 96 | class Something |
| 97 | this.count: number |
| 98 | that.count |
| 99 | endclass |
| 100 | END |
| 101 | v9.CheckScriptFailure(lines, 'E1318: Not a valid command in a class: that.count') |
| 102 | |
| 103 | lines =<< trim END |
| 104 | vim9script |
| 105 | class Something |
| 106 | this.count |
| 107 | endclass |
| 108 | END |
| 109 | v9.CheckScriptFailure(lines, 'E1022:') |
| 110 | |
| 111 | lines =<< trim END |
| 112 | vim9script |
| 113 | class Something |
Bram Moolenaar | f54cedd | 2022-12-23 17:56:27 +0000 | [diff] [blame] | 114 | def new() |
| 115 | this.state = 0 |
| 116 | enddef |
| 117 | endclass |
| 118 | var obj = Something.new() |
| 119 | END |
| 120 | v9.CheckScriptFailure(lines, 'E1089:') |
| 121 | |
| 122 | lines =<< trim END |
| 123 | vim9script |
| 124 | class Something |
Bram Moolenaar | 00b28d6 | 2022-12-08 15:32:33 +0000 | [diff] [blame] | 125 | this.count : number |
| 126 | endclass |
| 127 | END |
| 128 | v9.CheckScriptFailure(lines, 'E1059:') |
| 129 | |
| 130 | lines =<< trim END |
| 131 | vim9script |
| 132 | class Something |
| 133 | this.count:number |
| 134 | endclass |
| 135 | END |
| 136 | v9.CheckScriptFailure(lines, 'E1069:') |
| 137 | |
| 138 | lines =<< trim END |
| 139 | vim9script |
| 140 | |
| 141 | class TextPosition |
| 142 | this.lnum: number |
Bram Moolenaar | 7ce7daf | 2022-12-10 18:42:12 +0000 | [diff] [blame] | 143 | this.col: number |
Bram Moolenaar | ffdaca9 | 2022-12-09 21:41:48 +0000 | [diff] [blame] | 144 | |
Bram Moolenaar | 418b547 | 2022-12-20 13:38:22 +0000 | [diff] [blame] | 145 | # make a nicely formatted string |
Bram Moolenaar | ffdaca9 | 2022-12-09 21:41:48 +0000 | [diff] [blame] | 146 | def ToString(): string |
| 147 | return $'({this.lnum}, {this.col})' |
| 148 | enddef |
Bram Moolenaar | 00b28d6 | 2022-12-08 15:32:33 +0000 | [diff] [blame] | 149 | endclass |
| 150 | |
Bram Moolenaar | d28d7b9 | 2022-12-08 20:42:00 +0000 | [diff] [blame] | 151 | # use the automatically generated new() method |
| 152 | var pos = TextPosition.new(2, 12) |
| 153 | assert_equal(2, pos.lnum) |
| 154 | assert_equal(12, pos.col) |
Bram Moolenaar | ffdaca9 | 2022-12-09 21:41:48 +0000 | [diff] [blame] | 155 | |
| 156 | # call an object method |
| 157 | assert_equal('(2, 12)', pos.ToString()) |
Bram Moolenaar | 00b28d6 | 2022-12-08 15:32:33 +0000 | [diff] [blame] | 158 | END |
| 159 | v9.CheckScriptSuccess(lines) |
| 160 | enddef |
| 161 | |
Bram Moolenaar | 7ce7daf | 2022-12-10 18:42:12 +0000 | [diff] [blame] | 162 | def Test_class_member_initializer() |
| 163 | var lines =<< trim END |
| 164 | vim9script |
| 165 | |
| 166 | class TextPosition |
| 167 | this.lnum: number = 1 |
| 168 | this.col: number = 1 |
| 169 | |
Bram Moolenaar | 418b547 | 2022-12-20 13:38:22 +0000 | [diff] [blame] | 170 | # constructor with only the line number |
Bram Moolenaar | 7ce7daf | 2022-12-10 18:42:12 +0000 | [diff] [blame] | 171 | def new(lnum: number) |
| 172 | this.lnum = lnum |
| 173 | enddef |
| 174 | endclass |
| 175 | |
| 176 | var pos = TextPosition.new(3) |
| 177 | assert_equal(3, pos.lnum) |
| 178 | assert_equal(1, pos.col) |
| 179 | |
| 180 | var instr = execute('disassemble TextPosition.new') |
| 181 | assert_match('new\_s*' .. |
Bram Moolenaar | 3ea8a1b | 2022-12-10 19:03:51 +0000 | [diff] [blame] | 182 | '0 NEW TextPosition size \d\+\_s*' .. |
Bram Moolenaar | 7ce7daf | 2022-12-10 18:42:12 +0000 | [diff] [blame] | 183 | '\d PUSHNR 1\_s*' .. |
| 184 | '\d STORE_THIS 0\_s*' .. |
| 185 | '\d PUSHNR 1\_s*' .. |
| 186 | '\d STORE_THIS 1\_s*' .. |
| 187 | 'this.lnum = lnum\_s*' .. |
| 188 | '\d LOAD arg\[-1]\_s*' .. |
| 189 | '\d PUSHNR 0\_s*' .. |
| 190 | '\d LOAD $0\_s*' .. |
| 191 | '\d\+ STOREINDEX object\_s*' .. |
| 192 | '\d\+ RETURN object.*', |
| 193 | instr) |
| 194 | END |
| 195 | v9.CheckScriptSuccess(lines) |
| 196 | enddef |
| 197 | |
Bram Moolenaar | 65b0d16 | 2022-12-13 18:43:22 +0000 | [diff] [blame] | 198 | def Test_class_default_new() |
| 199 | var lines =<< trim END |
| 200 | vim9script |
| 201 | |
| 202 | class TextPosition |
| 203 | this.lnum: number = 1 |
| 204 | this.col: number = 1 |
| 205 | endclass |
| 206 | |
| 207 | var pos = TextPosition.new() |
| 208 | assert_equal(1, pos.lnum) |
| 209 | assert_equal(1, pos.col) |
| 210 | |
| 211 | pos = TextPosition.new(v:none, v:none) |
| 212 | assert_equal(1, pos.lnum) |
| 213 | assert_equal(1, pos.col) |
| 214 | |
| 215 | pos = TextPosition.new(3, 22) |
| 216 | assert_equal(3, pos.lnum) |
| 217 | assert_equal(22, pos.col) |
| 218 | |
| 219 | pos = TextPosition.new(v:none, 33) |
| 220 | assert_equal(1, pos.lnum) |
| 221 | assert_equal(33, pos.col) |
| 222 | END |
| 223 | v9.CheckScriptSuccess(lines) |
| 224 | |
| 225 | lines =<< trim END |
| 226 | vim9script |
| 227 | class Person |
| 228 | this.name: string |
| 229 | this.age: number = 42 |
| 230 | this.education: string = "unknown" |
| 231 | |
| 232 | def new(this.name, this.age = v:none, this.education = v:none) |
| 233 | enddef |
| 234 | endclass |
| 235 | |
| 236 | var piet = Person.new("Piet") |
| 237 | assert_equal("Piet", piet.name) |
| 238 | assert_equal(42, piet.age) |
| 239 | assert_equal("unknown", piet.education) |
| 240 | |
| 241 | var chris = Person.new("Chris", 4, "none") |
| 242 | assert_equal("Chris", chris.name) |
| 243 | assert_equal(4, chris.age) |
| 244 | assert_equal("none", chris.education) |
| 245 | END |
| 246 | v9.CheckScriptSuccess(lines) |
Bram Moolenaar | 74e1274 | 2022-12-13 21:14:28 +0000 | [diff] [blame] | 247 | |
| 248 | lines =<< trim END |
| 249 | vim9script |
| 250 | class Person |
| 251 | this.name: string |
| 252 | this.age: number = 42 |
| 253 | this.education: string = "unknown" |
| 254 | |
| 255 | def new(this.name, this.age = v:none, this.education = v:none) |
| 256 | enddef |
| 257 | endclass |
| 258 | |
| 259 | var missing = Person.new() |
| 260 | END |
| 261 | v9.CheckScriptFailure(lines, 'E119:') |
Bram Moolenaar | 65b0d16 | 2022-12-13 18:43:22 +0000 | [diff] [blame] | 262 | enddef |
| 263 | |
Bram Moolenaar | 74e1274 | 2022-12-13 21:14:28 +0000 | [diff] [blame] | 264 | def Test_class_object_member_inits() |
| 265 | var lines =<< trim END |
| 266 | vim9script |
| 267 | class TextPosition |
| 268 | this.lnum: number |
| 269 | this.col = 1 |
| 270 | this.addcol: number = 2 |
| 271 | endclass |
| 272 | |
| 273 | var pos = TextPosition.new() |
| 274 | assert_equal(0, pos.lnum) |
| 275 | assert_equal(1, pos.col) |
| 276 | assert_equal(2, pos.addcol) |
| 277 | END |
| 278 | v9.CheckScriptSuccess(lines) |
| 279 | |
| 280 | lines =<< trim END |
| 281 | vim9script |
| 282 | class TextPosition |
| 283 | this.lnum |
| 284 | this.col = 1 |
| 285 | endclass |
| 286 | END |
| 287 | v9.CheckScriptFailure(lines, 'E1022:') |
| 288 | |
| 289 | lines =<< trim END |
| 290 | vim9script |
| 291 | class TextPosition |
| 292 | this.lnum = v:none |
| 293 | this.col = 1 |
| 294 | endclass |
| 295 | END |
| 296 | v9.CheckScriptFailure(lines, 'E1330:') |
| 297 | enddef |
| 298 | |
Bram Moolenaar | 3d473ee | 2022-12-14 20:59:32 +0000 | [diff] [blame] | 299 | def Test_class_object_member_access() |
| 300 | var lines =<< trim END |
| 301 | vim9script |
| 302 | class Triple |
| 303 | this._one = 1 |
| 304 | this.two = 2 |
| 305 | public this.three = 3 |
| 306 | |
| 307 | def GetOne(): number |
| 308 | return this._one |
| 309 | enddef |
| 310 | endclass |
| 311 | |
| 312 | var trip = Triple.new() |
| 313 | assert_equal(1, trip.GetOne()) |
| 314 | assert_equal(2, trip.two) |
| 315 | assert_equal(3, trip.three) |
| 316 | assert_fails('echo trip._one', 'E1333') |
| 317 | |
| 318 | assert_fails('trip._one = 11', 'E1333') |
| 319 | assert_fails('trip.two = 22', 'E1335') |
| 320 | trip.three = 33 |
| 321 | assert_equal(33, trip.three) |
Bram Moolenaar | d505d17 | 2022-12-18 21:42:55 +0000 | [diff] [blame] | 322 | |
| 323 | assert_fails('trip.four = 4', 'E1334') |
| 324 | END |
| 325 | v9.CheckScriptSuccess(lines) |
Bram Moolenaar | 590162c | 2022-12-24 21:24:06 +0000 | [diff] [blame^] | 326 | |
| 327 | lines =<< trim END |
| 328 | vim9script |
| 329 | |
| 330 | class MyCar |
| 331 | this.make: string |
| 332 | |
| 333 | def new(make_arg: string) |
| 334 | this.make = make_arg |
| 335 | enddef |
| 336 | |
| 337 | def GetMake(): string |
| 338 | return $"make = {this.make}" |
| 339 | enddef |
| 340 | endclass |
| 341 | |
| 342 | var c = MyCar.new("abc") |
| 343 | assert_equal('make = abc', c.GetMake()) |
| 344 | |
| 345 | c = MyCar.new("def") |
| 346 | assert_equal('make = def', c.GetMake()) |
| 347 | |
| 348 | var c2 = MyCar.new("123") |
| 349 | assert_equal('make = 123', c2.GetMake()) |
| 350 | END |
| 351 | v9.CheckScriptSuccess(lines) |
Bram Moolenaar | d505d17 | 2022-12-18 21:42:55 +0000 | [diff] [blame] | 352 | enddef |
| 353 | |
| 354 | def Test_class_member_access() |
| 355 | var lines =<< trim END |
| 356 | vim9script |
| 357 | class TextPos |
| 358 | this.lnum = 1 |
| 359 | this.col = 1 |
| 360 | static counter = 0 |
| 361 | |
| 362 | def AddToCounter(nr: number) |
| 363 | counter += nr |
| 364 | enddef |
| 365 | endclass |
| 366 | |
| 367 | assert_equal(0, TextPos.counter) |
| 368 | TextPos.AddToCounter(3) |
| 369 | assert_equal(3, TextPos.counter) |
Bram Moolenaar | f54cedd | 2022-12-23 17:56:27 +0000 | [diff] [blame] | 370 | assert_fails('echo TextPos.noSuchMember', 'E1338:') |
Bram Moolenaar | d505d17 | 2022-12-18 21:42:55 +0000 | [diff] [blame] | 371 | |
Bram Moolenaar | f54cedd | 2022-12-23 17:56:27 +0000 | [diff] [blame] | 372 | assert_fails('TextPos.noSuchMember = 2', 'E1337:') |
Bram Moolenaar | d505d17 | 2022-12-18 21:42:55 +0000 | [diff] [blame] | 373 | assert_fails('TextPos.counter += 5', 'E1335') |
Bram Moolenaar | 3d473ee | 2022-12-14 20:59:32 +0000 | [diff] [blame] | 374 | END |
| 375 | v9.CheckScriptSuccess(lines) |
| 376 | enddef |
| 377 | |
Bram Moolenaar | 91c9d6d | 2022-12-14 17:30:37 +0000 | [diff] [blame] | 378 | def Test_class_object_to_string() |
| 379 | var lines =<< trim END |
| 380 | vim9script |
| 381 | class TextPosition |
| 382 | this.lnum = 1 |
| 383 | this.col = 22 |
| 384 | endclass |
| 385 | |
| 386 | assert_equal("class TextPosition", string(TextPosition)) |
| 387 | |
| 388 | var pos = TextPosition.new() |
| 389 | assert_equal("object of TextPosition {lnum: 1, col: 22}", string(pos)) |
| 390 | END |
| 391 | v9.CheckScriptSuccess(lines) |
| 392 | enddef |
Bram Moolenaar | 74e1274 | 2022-12-13 21:14:28 +0000 | [diff] [blame] | 393 | |
Bram Moolenaar | 00b28d6 | 2022-12-08 15:32:33 +0000 | [diff] [blame] | 394 | |
| 395 | " vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker |