resolve #871 infer called function by params num

1 files changed, 125 insertions, 0 deletions

diff --git a/test/type_inference/init.lua b/test/type_inference/init.lua
index 5fa70f5f..394ce263 100644
--- a/test/type_inference/init.lua
+++ b/test/type_inference/init.lua
@@ -2835,3 +2835,128 @@ local b
 
 local <?x?> = echo(b)
 ]]
+
+TEST 'boolean' [[
+---@return boolean
+function f()
+end
+
+---@param x integer
+---@return number
+function f(x)
+end
+
+local <?x?> = f()
+]]
+
+TEST 'number' [[
+---@return boolean
+function f()
+end
+
+---@param x integer
+---@return number
+function f(x)
+end
+
+local <?x?> = f(1)
+]]
+
+TEST 'boolean' [[
+---@return boolean
+function f()
+end
+
+---@param x integer
+---@return number
+function f(x)
+end
+
+function r0()
+    return
+end
+
+local <?x?> = f(r0())
+]]
+
+TEST 'number' [[
+---@return boolean
+function f()
+end
+
+---@param x integer
+---@return number
+function f(x)
+end
+
+function r1()
+    return 1
+end
+
+local <?x?> = f(r1())
+]]
+
+TEST 'boolean' [[
+---@return boolean
+function f()
+end
+
+---@param x integer
+---@return number
+function f(x)
+end
+
+---@type fun()
+local r0
+
+local <?x?> = f(r0())
+]]
+
+TEST 'number' [[
+---@return boolean
+function f()
+end
+
+---@param x integer
+---@return number
+function f(x)
+end
+
+---@type fun():integer
+local r1
+
+local <?x?> = f(r1())
+]]
+
+TEST 'boolean' [[
+---@overload fun(number, number):string
+---@overload fun(number):number
+---@return boolean
+local function f() end
+
+local <?n1?> = f()
+local n2 = f(0)
+local n3 = f(0, 0)
+]]
+
+TEST 'number' [[
+---@overload fun(number, number):string
+---@overload fun(number):number
+---@return boolean
+local function f() end
+
+local n1 = f()
+local <?n2?> = f(0)
+local n3 = f(0, 0)
+]]
+
+TEST 'string' [[
+---@overload fun(number, number):string
+---@overload fun(number):number
+---@return boolean
+local function f() end
+
+local n1 = f()
+local n2 = f(0)
+local <?n3?> = f(0, 0)
+]]