diff --git a/src/testdir/test_vim9_func.vim b/src/testdir/test_vim9_func.vim index 3fa5d6abea..ed5972185b 100644 --- a/src/testdir/test_vim9_func.vim +++ b/src/testdir/test_vim9_func.vim @@ -270,6 +270,19 @@ def Test_call_funcref() assert_equal(123, Funcref()) END CheckScriptSuccess(lines) + + lines =<< trim END + vim9script + def RetNumber(): number + return 123 + enddef + def Bar(F: func: number): number + return F() + enddef + let Funcref = function('RetNumber') + assert_equal(123, Bar(Funcref)) + END + CheckScriptSuccess(lines) enddef let SomeFunc = function('len') diff --git a/src/version.c b/src/version.c index 9ac0340a30..0bb6b0a75e 100644 --- a/src/version.c +++ b/src/version.c @@ -754,6 +754,8 @@ static char *(features[]) = static int included_patches[] = { /* Add new patch number below this line */ +/**/ + 1271, /**/ 1270, /**/ diff --git a/src/vim9compile.c b/src/vim9compile.c index 55f08a73ea..999833b522 100644 --- a/src/vim9compile.c +++ b/src/vim9compile.c @@ -599,6 +599,28 @@ check_argtype(type_T *expected, typval_T *actual_tv) member.tt_member = &t_any; actual.tt_member = &member; } + else if (actual_tv->v_type == VAR_FUNC || actual_tv->v_type == VAR_PARTIAL) + { + char_u *name = NULL; + ufunc_T *ufunc = NULL; + + if (actual_tv->v_type == VAR_PARTIAL) + { + if (actual_tv->vval.v_partial->pt_func != NULL) + ufunc = actual_tv->vval.v_partial->pt_func; + else + name = actual_tv->vval.v_partial->pt_name; + } + else + name = actual_tv->vval.v_string; + if (name != NULL) + // TODO: how about a builtin function? + ufunc = find_func(name, FALSE, NULL); + if (ufunc != NULL && ufunc->uf_func_type != NULL) + actual = *ufunc->uf_func_type; + else + actual.tt_member = &t_any; + } else actual.tt_member = &t_any; return check_type(expected, &actual, TRUE);