Bram Moolenaar | 1735bc9 | 2016-03-14 23:05:14 +0100 | [diff] [blame] | 1 | " Test binding arguments to a Funcref. |
| 2 | |
| 3 | func MyFunc(arg1, arg2, arg3) |
| 4 | return a:arg1 . '/' . a:arg2 . '/' . a:arg3 |
| 5 | endfunc |
| 6 | |
| 7 | func MySort(up, one, two) |
| 8 | if a:one == a:two |
| 9 | return 0 |
| 10 | endif |
| 11 | if a:up |
Bram Moolenaar | 790500a | 2016-03-15 11:05:45 +0100 | [diff] [blame] | 12 | return a:one > a:two ? 1 : -1 |
Bram Moolenaar | 1735bc9 | 2016-03-14 23:05:14 +0100 | [diff] [blame] | 13 | endif |
Bram Moolenaar | 790500a | 2016-03-15 11:05:45 +0100 | [diff] [blame] | 14 | return a:one < a:two ? 1 : -1 |
Bram Moolenaar | 1735bc9 | 2016-03-14 23:05:14 +0100 | [diff] [blame] | 15 | endfunc |
| 16 | |
| 17 | func Test_partial_args() |
| 18 | let Cb = function('MyFunc', ["foo", "bar"]) |
Bram Moolenaar | 6563903 | 2016-03-16 21:40:30 +0100 | [diff] [blame] | 19 | |
| 20 | call Cb("zzz") |
Bram Moolenaar | 1735bc9 | 2016-03-14 23:05:14 +0100 | [diff] [blame] | 21 | call assert_equal("foo/bar/xxx", Cb("xxx")) |
| 22 | call assert_equal("foo/bar/yyy", call(Cb, ["yyy"])) |
| 23 | |
Bram Moolenaar | 346418c | 2016-03-15 12:36:08 +0100 | [diff] [blame] | 24 | let Cb = function('MyFunc', []) |
| 25 | call assert_equal("a/b/c", Cb("a", "b", "c")) |
| 26 | |
Bram Moolenaar | 1735bc9 | 2016-03-14 23:05:14 +0100 | [diff] [blame] | 27 | let Sort = function('MySort', [1]) |
| 28 | call assert_equal([1, 2, 3], sort([3, 1, 2], Sort)) |
| 29 | let Sort = function('MySort', [0]) |
| 30 | call assert_equal([3, 2, 1], sort([3, 1, 2], Sort)) |
| 31 | endfunc |
| 32 | |
| 33 | func MyDictFunc(arg1, arg2) dict |
| 34 | return self.name . '/' . a:arg1 . '/' . a:arg2 |
| 35 | endfunc |
| 36 | |
| 37 | func Test_partial_dict() |
| 38 | let dict = {'name': 'hello'} |
| 39 | let Cb = function('MyDictFunc', ["foo", "bar"], dict) |
| 40 | call assert_equal("hello/foo/bar", Cb()) |
| 41 | call assert_fails('Cb("xxx")', 'E492:') |
Bram Moolenaar | 346418c | 2016-03-15 12:36:08 +0100 | [diff] [blame] | 42 | |
Bram Moolenaar | 1735bc9 | 2016-03-14 23:05:14 +0100 | [diff] [blame] | 43 | let Cb = function('MyDictFunc', ["foo"], dict) |
| 44 | call assert_equal("hello/foo/xxx", Cb("xxx")) |
| 45 | call assert_fails('Cb()', 'E492:') |
Bram Moolenaar | 346418c | 2016-03-15 12:36:08 +0100 | [diff] [blame] | 46 | |
| 47 | let Cb = function('MyDictFunc', [], dict) |
| 48 | call assert_equal("hello/ttt/xxx", Cb("ttt", "xxx")) |
| 49 | call assert_fails('Cb("yyy")', 'E492:') |
| 50 | |
Bram Moolenaar | 1735bc9 | 2016-03-14 23:05:14 +0100 | [diff] [blame] | 51 | let Cb = function('MyDictFunc', dict) |
| 52 | call assert_equal("hello/xxx/yyy", Cb("xxx", "yyy")) |
Bram Moolenaar | 346418c | 2016-03-15 12:36:08 +0100 | [diff] [blame] | 53 | call assert_fails('Cb("fff")', 'E492:') |
Bram Moolenaar | 6563903 | 2016-03-16 21:40:30 +0100 | [diff] [blame] | 54 | |
| 55 | let dict = {"tr": function('tr', ['hello', 'h', 'H'])} |
| 56 | call assert_equal("Hello", dict.tr()) |
Bram Moolenaar | 1735bc9 | 2016-03-14 23:05:14 +0100 | [diff] [blame] | 57 | endfunc |
Bram Moolenaar | ab1fa39 | 2016-03-15 19:33:34 +0100 | [diff] [blame] | 58 | |
| 59 | func Test_partial_implicit() |
| 60 | let dict = {'name': 'foo'} |
| 61 | func dict.MyFunc(arg) dict |
| 62 | return self.name . '/' . a:arg |
| 63 | endfunc |
| 64 | |
| 65 | call assert_equal('foo/bar', dict.MyFunc('bar')) |
| 66 | |
| 67 | call assert_fails('let func = dict.MyFunc', 'E704:') |
| 68 | let Func = dict.MyFunc |
| 69 | call assert_equal('foo/aaa', Func('aaa')) |
| 70 | |
| 71 | let Func = function(dict.MyFunc, ['bbb']) |
| 72 | call assert_equal('foo/bbb', Func()) |
Bram Moolenaar | ab1fa39 | 2016-03-15 19:33:34 +0100 | [diff] [blame] | 73 | endfunc |
Bram Moolenaar | 7a5c46a | 2016-03-16 20:41:21 +0100 | [diff] [blame] | 74 | |
| 75 | fun InnerCall(funcref) |
| 76 | return a:funcref |
| 77 | endfu |
| 78 | |
| 79 | fun OuterCall() |
| 80 | let opt = { 'func' : function('sin') } |
| 81 | call InnerCall(opt.func) |
| 82 | endfu |
| 83 | |
| 84 | func Test_function_in_dict() |
| 85 | call OuterCall() |
| 86 | endfunc |
| 87 | |
Bram Moolenaar | 6f2e4b3 | 2016-03-16 22:52:12 +0100 | [diff] [blame] | 88 | function! s:cache_clear() dict |
| 89 | return self.name |
| 90 | endfunction |
| 91 | |
| 92 | func Test_script_function_in_dict() |
| 93 | let s:obj = {'name': 'foo'} |
| 94 | let s:obj2 = {'name': 'bar'} |
| 95 | |
| 96 | let s:obj['clear'] = function('s:cache_clear') |
| 97 | |
| 98 | call assert_equal('foo', s:obj.clear()) |
| 99 | let F = s:obj.clear |
| 100 | call assert_equal('foo', F()) |
| 101 | call assert_equal('foo', call(s:obj.clear, [], s:obj)) |
| 102 | call assert_equal('bar', call(s:obj.clear, [], s:obj2)) |
| 103 | |
| 104 | let s:obj2['clear'] = function('s:cache_clear') |
| 105 | call assert_equal('bar', s:obj2.clear()) |
| 106 | let B = s:obj2.clear |
| 107 | call assert_equal('bar', B()) |
| 108 | endfunc |