test_method.vim (6283B)
1 " Tests for ->method() 2 3 source check.vim 4 5 func Test_list_method() 6 let l = [1, 2, 3] 7 call assert_equal([1, 2, 3, 4], [1, 2, 3]->add(4)) 8 eval l->assert_equal(l) 9 eval l->assert_equal(l, 'wrong') 10 eval l->assert_notequal([3, 2, 1]) 11 eval l->assert_notequal([3, 2, 1], 'wrong') 12 call assert_equal(l, l->copy()) 13 call assert_equal(l, l->deepcopy()) 14 call assert_equal(1, l->count(2)) 15 call assert_false(l->empty()) 16 call assert_true([]->empty()) 17 call assert_equal(579, ['123', '+', '456']->join()->eval()) 18 call assert_equal([1, 2, 3, 4, 5], [1, 2, 3]->extend([4, 5])) 19 call assert_equal([1, 3], [1, 2, 3]->filter('v:val != 2')) 20 call assert_equal(2, l->get(1)) 21 call assert_equal(1, l->index(2)) 22 call assert_equal([0, 1, 2, 3], [1, 2, 3]->insert(0)) 23 call assert_equal('1 2 3', l->join()) 24 call assert_fails('eval l->keys()', 'E1206:') 25 call assert_equal(3, l->len()) 26 call assert_equal([2, 3, 4], [1, 2, 3]->map('v:val + 1')) 27 call assert_equal(3, l->max()) 28 call assert_equal(1, l->min()) 29 call assert_equal(2, [1, 2, 3]->remove(1)) 30 call assert_equal([1, 2, 3, 1, 2, 3], l->repeat(2)) 31 call assert_equal([3, 2, 1], [1, 2, 3]->reverse()) 32 call assert_equal([1, 2, 3, 4], [4, 2, 3, 1]->sort()) 33 call assert_equal('[1, 2, 3]', l->string()) 34 call assert_equal(v:t_list, l->type()) 35 call assert_equal([1, 2, 3], [1, 1, 2, 3, 3]->uniq()) 36 call assert_fails('eval l->values()', 'E1206:') 37 call assert_fails('echo []->len', 'E107:') 38 endfunc 39 40 func Test_dict_method() 41 let d = #{one: 1, two: 2, three: 3} 42 43 call assert_equal(d, d->copy()) 44 call assert_equal(d, d->deepcopy()) 45 call assert_equal(1, d->count(2)) 46 call assert_false(d->empty()) 47 call assert_true({}->empty()) 48 call assert_equal(#{one: 1, two: 2, three: 3, four: 4}, d->extend(#{four: 4})) 49 call assert_equal(#{one: 1, two: 2, three: 3}, d->filter('v:val != 4')) 50 call assert_equal(2, d->get('two')) 51 call assert_fails("let x = d->index(2)", 'E897:') 52 call assert_fails("let x = d->insert(0)", 'E899:') 53 call assert_true(d->has_key('two')) 54 call assert_equal([['one', 1], ['two', 2], ['three', 3]], d->items()) 55 call assert_fails("let x = d->join()", 'E714:') 56 call assert_equal(['one', 'two', 'three'], d->keys()) 57 call assert_equal(3, d->len()) 58 call assert_equal(#{one: 2, two: 3, three: 4}, d->map('v:val + 1')) 59 call assert_equal(#{one: 1, two: 2, three: 3}, d->map('v:val - 1')) 60 call assert_equal(3, d->max()) 61 call assert_equal(1, d->min()) 62 call assert_equal(2, d->remove("two")) 63 let d.two = 2 64 call assert_fails('let x = d->repeat(2)', 'E731:') 65 call assert_fails('let x = d->reverse()', 'E1252:') 66 call assert_fails('let x = d->sort()', 'E686:') 67 call assert_equal("{'one': 1, 'two': 2, 'three': 3}", d->string()) 68 call assert_equal(v:t_dict, d->type()) 69 call assert_fails('let x = d->uniq()', 'E686:') 70 call assert_equal([1, 2, 3], d->values()) 71 endfunc 72 73 func Test_string_method() 74 eval '1 2 3'->split()->assert_equal(['1', '2', '3']) 75 eval '1 2 3'->split()->map({i, v -> str2nr(v)})->assert_equal([1, 2, 3]) 76 eval 'ABC'->str2list()->assert_equal([65, 66, 67]) 77 eval 'ABC'->strlen()->assert_equal(3) 78 eval "a\rb\ec"->strtrans()->assert_equal('a^Mb^[c') 79 eval "aあb"->strwidth()->assert_equal(4) 80 eval 'abc'->substitute('b', 'x', '')->assert_equal('axc') 81 call assert_fails('eval 123->items()', 'E1225:') 82 83 eval 'abc'->printf('the %s arg')->assert_equal('the abc arg') 84 endfunc 85 86 func Test_method_append() 87 new 88 eval ['one', 'two', 'three']->append(1) 89 call assert_equal(['', 'one', 'two', 'three'], getline(1, '$')) 90 91 %del 92 let bnr = bufnr('') 93 wincmd w 94 eval ['one', 'two', 'three']->appendbufline(bnr, 1) 95 call assert_equal(['', 'one', 'two', 'three'], getbufline(bnr, 1, '$')) 96 97 exe 'bwipe! ' .. bnr 98 endfunc 99 100 func Test_method_funcref() 101 func Concat(one, two, three) 102 return a:one .. a:two .. a:three 103 endfunc 104 let FuncRef = function('Concat') 105 eval 'foo'->FuncRef('bar', 'tail')->assert_equal('foobartail') 106 107 " not enough arguments 108 call assert_fails("eval 'foo'->FuncRef('bar')", 'E119:') 109 " too many arguments 110 call assert_fails("eval 'foo'->FuncRef('bar', 'tail', 'four')", 'E118:') 111 112 let Partial = function('Concat', ['two']) 113 eval 'one'->Partial('three')->assert_equal('onetwothree') 114 115 " not enough arguments 116 call assert_fails("eval 'one'->Partial()", 'E119:') 117 " too many arguments 118 call assert_fails("eval 'one'->Partial('three', 'four')", 'E118:') 119 120 delfunc Concat 121 endfunc 122 123 func Test_method_float() 124 CheckFeature float 125 eval 1.234->string()->assert_equal('1.234') 126 eval -1.234->string()->assert_equal('-1.234') 127 endfunc 128 129 func Test_method_syntax() 130 eval [1, 2, 3] ->sort( ) 131 eval [1, 2, 3] 132 \ ->sort( 133 \ ) 134 call assert_fails('eval [1, 2, 3]-> sort()', 'E15:') 135 call assert_fails('eval [1, 2, 3]->sort ()', 'E274:') 136 call assert_fails('eval [1, 2, 3]-> sort ()', 'E15:') 137 138 " Test for using a method name containing a curly brace name 139 let s = 'len' 140 call assert_equal(4, "xxxx"->str{s}()) 141 142 " Test for using a method in an interpolated string 143 call assert_equal('4', $'{"xxxx"->strlen()}') 144 endfunc 145 146 func Test_method_lambda() 147 eval "text"->{x -> x .. " extended"}()->assert_equal('text extended') 148 eval "text"->{x, y -> x .. " extended " .. y}('more')->assert_equal('text extended more') 149 150 call assert_fails('eval "text"->{x -> x .. " extended"} ()', 'E274:') 151 152 " todo: lambda accepts more arguments than it consumes 153 " call assert_fails('eval "text"->{x -> x .. " extended"}("more")', 'E99:') 154 155 " Nvim doesn't include test_refcount(). 156 " let l = [1, 2, 3] 157 " eval l->{x -> x}() 158 " call assert_equal(1, test_refcount(l)) 159 endfunc 160 161 func Test_method_not_supported() 162 call assert_fails('eval 123->changenr()', 'E276:') 163 call assert_fails('echo "abc"->invalidfunc()', 'E117:') 164 " Test for too many or too few arguments to a method 165 call assert_fails('let n="abc"->len(2)', 'E118:') 166 call assert_fails('let n=10->setwinvar()', 'E119:') 167 endfunc 168 169 " Test for passing optional arguments to methods 170 func Test_method_args() 171 let v:errors = [] 172 let n = 10->assert_inrange(1, 5, "Test_assert_inrange") 173 if v:errors[0] !~ 'Test_assert_inrange' 174 call assert_report(v:errors[0]) 175 else 176 " Test passed 177 let v:errors = [] 178 endif 179 endfunc 180 181 " vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker