Do not count Rust functions in return statements for lloc

mozilla · Mar 19, 2021 · 9cb72f1 · 9cb72f1
1 parent 20d1fd8
commit 9cb72f1
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/src/metrics/loc.rs b/src/metrics/loc.rs
@@ -332,7 +332,9 @@ impl Loc for RustCode {
                         | WhileLetExpression
                         | ForExpression
                         | IfExpression
-                        | IfLetExpression,
+                        | IfLetExpression
+                        | ReturnExpression
+                        | AwaitExpression,
                     Block
                 ) == 0
                 {
@@ -675,6 +677,18 @@ mod tests {
         );
     }
 
+    #[test]
+    fn rust_function_in_return_lloc() {
+        check_metrics!(
+            "return foo();
+             await foo();",
+            "foo.rs",
+            RustParser,
+            loc,
+            [(lloc, 2, usize)]
+        );
+    }
+
     #[test]
     fn rust_closure_expression_lloc() {
         check_metrics!(