fix(java): add simple fix for Java stratLine issue

unit-mesh · Jan 4, 2024 · b9af199 · b9af199
1 parent 290978f
commit b9af199
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/unit-picker/src/main/kotlin/cc/unitmesh/pick/ext/CodeDataStructUtil.kt b/unit-picker/src/main/kotlin/cc/unitmesh/pick/ext/CodeDataStructUtil.kt
@@ -17,6 +17,8 @@ fun CodeContainer.buildSourceCode(codeLines: List<String>) {
                 if (annotationPos.StartLine != 0) {
                     position.StartLine = annotationPos.StartLine
                     position.StartLinePosition = annotationPos.StartLinePosition
+                } else {
+                    position.StartLine -= 2
                 }
             }
 

diff --git a/...cker/src/test/kotlin/cc/unitmesh/pick/builder/unittest/java/JavaTestCodeServiceE2ETest.kt b/...cker/src/test/kotlin/cc/unitmesh/pick/builder/unittest/java/JavaTestCodeServiceE2ETest.kt
@@ -145,7 +145,6 @@ public class CompanyService {
 }
 """
 
-
         val underTestContainer = JavaAnalyser().analysis(underTestCode, "CompanyService.java")
         underTestContainer.buildSourceCode(underTestCode.lines())
-Original file line number
+Diff line change
@@ Expand Up / @@ -145,7 +145,6 @@ public class CompanyService { @@
     }
     """
             val underTestContainer = JavaAnalyser().analysis(underTestCode, "CompanyService.java")
             underTestContainer.buildSourceCode(underTestCode.lines())
@@ Expand Down @@