Add test:

Fix #240
Fix #241

Author: CppCXY

crates/emmylua_code_analysis/src/compilation/test/closure_generic.rs

@@ -16,4 +16,65 @@ mod test {
         "#,
         );
     }
+
+    #[test]
+    fn test_issue_240() {
+        let mut ws = VirtualWorkspace::new();
+
+        ws.def(
+            r#"
+        --- @generic T
+        --- @param fn fun(...:T...)
+        --- @return fun(...:T...)
+        local function wrap(fn)
+            return fn
+        end
+
+        --- @param a integer
+        foo = wrap(function(a)
+        end) -- type unknown
+
+        --- @param a integer
+        bar = wrap(function(a)
+            _ = a
+        end) -- type fun(a: integer)
+        "#,
+        );
+
+        let foo = ws.expr_ty("foo");
+        let foo_desc = ws.humanize_type(foo);
+        assert_eq!(foo_desc, "fun(a: integer)");
+
+        let bar = ws.expr_ty("bar");
+        let bar_desc = ws.humanize_type(bar);
+        assert_eq!(bar_desc, "fun(a: integer)");
+    }
+
+    #[test]
+    fn test_issue_241() {
+        let mut ws = VirtualWorkspace::new();
+
+        ws.def(
+            r#"
+        --- @generic T
+        --- @param fn fun(...:T...)
+        --- @return fun(...:T...)
+        local function wrap(fn) return fn end
+
+        E = {}
+
+        --- @param a integer
+        E.foo = function(a) _ = a end -- type fun(a: integer) - correct
+
+        --- @param a integer
+        E.foo_wrapped = wrap(function(a) _ = a end) -- type fun(a: integer) - correct
+        E.foo_wrapped2_a = wrap(wrap(E.foo))        -- type fun(a: integer) - correct
+        E.foo_wrapped2_b = wrap(E.foo_wrapped)      -- type fun() - wrong
+        "#,
+        );
+
+        let foo_wrapper2_b = ws.expr_ty("E.foo_wrapped2_b");
+        let desc = ws.humanize_type(foo_wrapper2_b);
+        assert_eq!(desc, "fun(a: integer)");
+    }
 }

crates/emmylua_code_analysis/src/test_lib/mod.rs

@@ -5,8 +5,8 @@ use lsp_types::NumberOrString;
 use tokio_util::sync::CancellationToken;
 
 use crate::{
-    check_type_compact, DiagnosticCode, EmmyLuaAnalysis, Emmyrc, FileId, LuaType,
-    VirtualUrlGenerator,
+    check_type_compact, humanize_type, DiagnosticCode, EmmyLuaAnalysis, Emmyrc, FileId, LuaType,
+    RenderLevel, VirtualUrlGenerator,
 };
 
 /// A virtual workspace for testing.
@@ -177,6 +177,12 @@ impl VirtualWorkspace {
         emmyrc.diagnostics.enables = enables;
         self.analysis.diagnostic.update_config(Arc::new(emmyrc));
     }
+
+    pub fn humanize_type(&self, ty: LuaType) -> String {
+        let db = &self.analysis.compilation.get_db();
+        let level = 0;
+        humanize_type(db, &ty, RenderLevel::Brief)
+    }
 }
 
 #[cfg(test)]