test: fix tests for undertest code

unit-mesh · Dec 24, 2023 · fca8755 · fca8755
1 parent ac1e894
commit fca8755
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
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
@@ -10,6 +10,12 @@ fun CodeContainer.buildSourceCode(codeLines: List<String>) {
 
         ds.Content = CodeDataStructUtil.contentByPosition(codeLines, ds.Position)
         ds.Functions.map {
+            // in Chapi, the position of function is not correct, so we need to fix it
+            val position = it.Position
+            if (it.Annotations.isNotEmpty()) {
+                position.StartLine -= 1
+                position.StartLinePosition = 0
+            }
             it.Content = CodeDataStructUtil.contentByPosition(codeLines, it.Position)
         }
     }

diff --git a/...-picker/src/test/kotlin/cc/unitmesh/pick/builder/unittest/lang/JavaTestCodeServiceTest.kt b/...-picker/src/test/kotlin/cc/unitmesh/pick/builder/unittest/lang/JavaTestCodeServiceTest.kt
@@ -189,7 +189,7 @@ public class CompanyService {
         assertEquals(3, result.size)
         val typedTestIns = result[1] as BasicTestIns
         typedTestIns.testType shouldBe TestCodeBuilderType.METHOD_UNIT
-//        typedTestIns.generatedCode shouldContain "@Test"
-//        typedTestIns.underTestCode.length shouldNotBe 0
+        typedTestIns.generatedCode shouldContain "@Test"
+        typedTestIns.underTestCode shouldContain "companyMongoRepository.findAll()"
     }
 }