Close #265

Author: CppCXY

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

@@ -1,6 +1,6 @@
 #[cfg(test)]
 mod test {
-    use crate::VirtualWorkspace;
+    use crate::{DiagnosticCode, VirtualWorkspace};
 
     #[test]
     fn test_flow() {
@@ -26,4 +26,24 @@ mod test {
         let expected = ws.ty("string[]");
         assert_eq!(ty, expected);
     }
+
+    #[test]
+    fn test_issue_265() {
+        let mut ws = VirtualWorkspace::new();
+
+        assert!(ws.check_code_for(
+            DiagnosticCode::ReturnTypeMismatch,
+            r#"
+        local function bar()
+            return ''
+        end
+
+        --- @return integer
+        function foo()
+            return bar() --[[@as integer]]
+        end
+
+        "#,
+        ));
+    }
 }

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

@@ -305,18 +305,20 @@ end
     fn test_issue_266() {
         let mut ws = VirtualWorkspace::new();
 
-        assert!(ws.check_code_for(DiagnosticCode::AssignTypeMismatch, r#"
+        assert!(ws.check_code_for(
+            DiagnosticCode::AssignTypeMismatch,
+            r#"
         --- @return string
         function baz() end
 
         local a
         a = baz() -- a has type nil but should be string    
         d = a    
-        "#));
+        "#
+        ));
 
         let d = ws.expr_ty("d");
         let d_desc = ws.humanize_type(d);
         assert_eq!(d_desc, "string");
     }
-
 }