From 9cb72f10c5023a3391945a1572f33f37bda4f4fc Mon Sep 17 00:00:00 2001
From: Luni-4 <luni-4@hotmail.it>
Date: Thu, 10 Dec 2020 23:12:32 +0100
Subject: [PATCH] Do not count Rust functions in return statements for lloc

---
 src/metrics/loc.rs | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/src/metrics/loc.rs b/src/metrics/loc.rs
index c3bd17e90..993aeaeab 100644
--- 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!(