feat(unitest): add test support

unit-mesh · Dec 24, 2023 · 1ebea0e · 1ebea0e
1 parent 8591675
commit 1ebea0e
Show file tree

Hide file tree

Showing 3 changed files with 37 additions and 23 deletions.
diff --git a/unit-picker/src/main/kotlin/cc/unitmesh/pick/builder/unittest/ClassTestCodeBuilder.kt b/unit-picker/src/main/kotlin/cc/unitmesh/pick/builder/unittest/ClassTestCodeBuilder.kt
@@ -20,7 +20,7 @@ class ClassTestCodeBuilder(private val context: JobContext) : TestCodeBuilder {
         dataStruct: CodeDataStruct,
         underTestFile: CodeDataStruct,
         relevantClasses: List<CodeDataStruct>,
-    ): List<ClassTestIns> {
+    ): List<BasicTestIns> {
         val generatedCode = dataStruct.toSourceCode()
         if (generatedCode.lines().size > context.insQualityThreshold.maxLineInCode) {
             return emptyList()
@@ -29,7 +29,7 @@ class ClassTestCodeBuilder(private val context: JobContext) : TestCodeBuilder {
         val namingStyle = dataStruct.checkNamingStyle()
 
         return listOf(
-            ClassTestIns(
+            BasicTestIns(
                 lang = context.project.language,
                 underTestCode = underTestFile.Content,
                 generatedCode = generatedCode,
@@ -44,7 +44,7 @@ class ClassTestCodeBuilder(private val context: JobContext) : TestCodeBuilder {
     }
 }
 
-class ClassTestIns(
+class BasicTestIns(
     val lang: SupportedLang,
     val underTestCode: String,
     val generatedCode: String,

diff --git a/unit-picker/src/main/kotlin/cc/unitmesh/pick/builder/unittest/MethodTestCodeBuilder.kt b/unit-picker/src/main/kotlin/cc/unitmesh/pick/builder/unittest/MethodTestCodeBuilder.kt
@@ -12,32 +12,47 @@ class MethodTestCodeBuilder(private val context: JobContext) : TestCodeBuilder {
         dataStruct: CodeDataStruct,
         underTestFile: CodeDataStruct,
         relevantClasses: List<CodeDataStruct>,
-    ): List<ClassTestIns> {
+    ): List<BasicTestIns> {
         val generatedCode = dataStruct.toSourceCode()
         if (generatedCode.lines().size > context.insQualityThreshold.maxLineInCode) {
             return emptyList()
         }
 
         val namingStyle = dataStruct.checkNamingStyle()
+        // test canonicalName map BasicTestIns
+        val results: HashMap<String, List<BasicTestIns>> = hashMapOf()
 
-        dataStruct.Functions.map {
-            it.FunctionCalls.map {
-                // todo
+        val underTestFunctionMap = underTestFile.Functions.map {
+            val canonicalName = underTestFile.Package + "." + underTestFile.NodeName + ":" + it.Name
+            canonicalName to it.Content
+        }.toMap()
+
+        // analysis test code, and find original function content, put to results
+        dataStruct.Functions.map { function ->
+            function.FunctionCalls.map {
+                if (it.NodeName == underTestFile.NodeName) {
+                    val canonicalName = it.Package + "." + it.NodeName + ":" + it.FunctionName
+                    val originalContent = underTestFunctionMap[canonicalName]
+
+                    if (originalContent != null) {
+                        val testIns = BasicTestIns(
+                            lang = context.project.language,
+                            underTestCode = originalContent,
+                            generatedCode = function.Content,
+                            coreFrameworks = context.project.coreFrameworks,
+                            testFrameworks = context.project.testFrameworks,
+                            testType = TestCodeBuilderType.METHOD_UNIT,
+                            specs = listOf(
+                                "Test class should be named `${namingStyle}`."
+                            )
+                        )
+
+                        results[canonicalName] = results[canonicalName]?.plus(testIns) ?: listOf(testIns)
+                    }
+                }
             }
         }
 
-        return listOf(
-            ClassTestIns(
-                lang = context.project.language,
-                underTestCode = underTestFile.Content,
-                generatedCode = generatedCode,
-                coreFrameworks = context.project.coreFrameworks,
-                testFrameworks = context.project.testFrameworks,
-                testType = TestCodeBuilderType.CLASS_UNIT,
-                specs = listOf(
-                    "Test class should be named `${namingStyle}`."
-                )
-            )
-        )
+        return results.values.flatten()
     }
 }
diff --git a/unit-picker/src/main/kotlin/cc/unitmesh/pick/builder/unittest/lang/JavaTestCodeService.kt b/unit-picker/src/main/kotlin/cc/unitmesh/pick/builder/unittest/lang/JavaTestCodeService.kt
@@ -70,11 +70,10 @@ class JavaTestCodeService(val context: JobContext) : UnitTestService {
         val classTestIns = ClassTestCodeBuilder(context)
             .build(dataStruct, underTestFile, relevantClasses)
 
-        MethodTestCodeBuilder(context)
+        val methodTests = MethodTestCodeBuilder(context)
             .build(dataStruct, underTestFile, relevantClasses)
 
-        // todo: add method level support
-        return classTestIns
+        return classTestIns + methodTests
     }
 }