summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorBram Moolenaar <Bram@vim.org>2020-04-05 22:14:54 +0200
committerBram Moolenaar <Bram@vim.org>2020-04-05 22:14:54 +0200
commit8922860afb2cf9e89417c0c1417f1fb4458d3b44 (patch)
treeac4d2e2ae32f60ac80389414e50878b0fa35c080
parent5ba8d3578c835edcfb7e3b132e623c12e62f250b (diff)
downloadvim-git-8922860afb2cf9e89417c0c1417f1fb4458d3b44.tar.gz
patch 8.2.0519: Vim9: return type not properly checkedv8.2.0519
Problem: Vim9: return type not properly checked. Solution: Check type properly, also at runtime.
-rw-r--r--src/testdir/test_vim9_func.vim19
-rw-r--r--src/version.c2
-rw-r--r--src/vim9compile.c6
3 files changed, 24 insertions, 3 deletions
diff --git a/src/testdir/test_vim9_func.vim b/src/testdir/test_vim9_func.vim
index c12216862..74653b13e 100644
--- a/src/testdir/test_vim9_func.vim
+++ b/src/testdir/test_vim9_func.vim
@@ -380,6 +380,10 @@ def FuncOneArgRetNumber(arg: number): number
return arg
enddef
+def FuncOneArgRetAny(arg: any): any
+ return arg
+enddef
+
def Test_func_type()
let Ref1: func()
funcResult = 0
@@ -417,5 +421,20 @@ def Test_func_type_fails()
CheckDefFailure(['let Ref1: func()', 'Ref1 = FuncOneArgRetNumber'], 'E1013: type mismatch, expected func() but got func(number): number')
enddef
+def Test_func_return_type()
+ let nr: number
+ nr = FuncNoArgRetNumber()
+ assert_equal(1234, nr)
+
+ nr = FuncOneArgRetAny(122)
+ assert_equal(122, nr)
+
+ let str: string
+ str = FuncOneArgRetAny('yes')
+ assert_equal('yes', str)
+
+ CheckDefFailure(['let str: string', 'str = FuncNoArgRetNumber()'], 'E1013: type mismatch, expected string but got number')
+enddef
+
" vim: ts=8 sw=2 sts=2 expandtab tw=80 fdm=marker
diff --git a/src/version.c b/src/version.c
index f420990b9..6e9f00616 100644
--- a/src/version.c
+++ b/src/version.c
@@ -739,6 +739,8 @@ static char *(features[]) =
static int included_patches[] =
{ /* Add new patch number below this line */
/**/
+ 519,
+/**/
518,
/**/
517,
diff --git a/src/vim9compile.c b/src/vim9compile.c
index 98936dd66..0dfe5a163 100644
--- a/src/vim9compile.c
+++ b/src/vim9compile.c
@@ -2432,7 +2432,7 @@ check_type(type_T *expected, type_T *actual, int give_msg)
if (ret == FAIL && give_msg)
type_mismatch(expected, actual);
}
- return OK;
+ return ret;
}
/*
@@ -2444,7 +2444,7 @@ check_type(type_T *expected, type_T *actual, int give_msg)
static int
need_type(type_T *actual, type_T *expected, int offset, cctx_T *cctx)
{
- if (check_type(expected, actual, FALSE))
+ if (check_type(expected, actual, FALSE) == OK)
return OK;
if (actual->tt_type != VAR_ANY && actual->tt_type != VAR_UNKNOWN)
{
@@ -4069,7 +4069,7 @@ compile_assignment(char_u *arg, exarg_T *eap, cmdidx_T cmdidx, cctx_T *cctx)
lvar->lv_type = stacktype;
}
}
- else if (check_type(lvar->lv_type, stacktype, TRUE) == FAIL)
+ else if (need_type(stacktype, lvar->lv_type, -1, cctx) == FAIL)
goto theend;
}
else if (*p != '=' && check_type(type, stacktype, TRUE) == FAIL)