feat(test): add test code templates

Add test code templates for different test classes in both Kotlin and Java. The templates provide examples of how to write test code for controllers, services, and general test cases.
unit-mesh · Jan 13, 2024 · 2af285d · 2af285d
1 parent 5a2d2cc
commit 2af285d
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 7 deletions.
diff --git a/java/src/main/kotlin/cc/unitmesh/idea/provider/JavaTestContextProvider.kt b/java/src/main/kotlin/cc/unitmesh/idea/provider/JavaTestContextProvider.kt
@@ -47,7 +47,7 @@ open class JavaTestContextProvider : ChatContextProvider {
 
                 val lookup = project.service<TemplatedTestPrompt>().lookup("ControllerTest.java")
                 if (lookup != null) {
-                    testControllerPrompt += "Here is a template as example\n```$language\n$lookup\n```\n"
+                    testControllerPrompt += "\nHere is the Test code template as example\n```$language\n$lookup\n```\n"
                 }
 
                 ChatContextItem(JavaTestContextProvider::class, testControllerPrompt)
@@ -61,7 +61,7 @@ open class JavaTestContextProvider : ChatContextProvider {
 
                 val lookup = project.service<TemplatedTestPrompt>().lookup("ServiceTest.java")
                 if (lookup != null) {
-                    testServicePrompt += "Here is a template as example\n```$language\n$lookup\n```\n"
+                    testServicePrompt += "\nHere is the Test code template as example\n```$language\n$lookup\n```\n"
                 }
 
                 ChatContextItem(JavaTestContextProvider::class, testServicePrompt)
@@ -70,7 +70,7 @@ open class JavaTestContextProvider : ChatContextProvider {
             else -> {
                 val lookup = project.service<TemplatedTestPrompt>().lookup("Test.java")
                 if (lookup != null) {
-                    prompt += "Here is a template as example\n```$language\n$lookup\n```\n"
+                    prompt += "\nHere is the Test code template as example\n```$language\n$lookup\n```\n"
                 }
                 ChatContextItem(JavaTestContextProvider::class, prompt)
             }

diff --git a/kotlin/src/main/kotlin/cc/unitmesh/kotlin/context/KotlinClassContextBuilder.kt b/kotlin/src/main/kotlin/cc/unitmesh/kotlin/context/KotlinClassContextBuilder.kt
@@ -22,7 +22,7 @@ class KotlinClassContextBuilder : ClassContextBuilder {
 
         val text = psiElement.text
         val name = psiElement.name
-        val ktNamedFunctions = Companion.getFunctions(psiElement)
+        val ktNamedFunctions = getFunctions(psiElement)
         val primaryConstructorFields = getPrimaryConstructorFields(psiElement)
         val allFields = ktNamedFunctions + primaryConstructorFields
         val usages =

diff --git a/kotlin/src/main/kotlin/cc/unitmesh/kotlin/provider/KotlinTestContextProvider.kt b/kotlin/src/main/kotlin/cc/unitmesh/kotlin/provider/KotlinTestContextProvider.kt
@@ -39,7 +39,7 @@ class KotlinTestContextProvider : JavaTestContextProvider() {
 
                 val lookup = project.service<TemplatedTestPrompt>().lookup("ControllerTest.kt")
                 if (lookup != null) {
-                    testControllerPrompt += "Here is a template as example\n```$language\n$lookup\n```\n"
+                    testControllerPrompt += "\nHere is the Test code template as example\n```$language\n$lookup\n```\n"
                 }
 
                 ChatContextItem(JavaTestContextProvider::class, testControllerPrompt)
@@ -54,7 +54,7 @@ class KotlinTestContextProvider : JavaTestContextProvider() {
 
                 val lookup = project.service<TemplatedTestPrompt>().lookup("ServiceTest.kt")
                 if (lookup != null) {
-                    testServicePrompt += "Here is a template as example\n```$language\n$lookup\n```\n"
+                    testServicePrompt += "\nHere is the Test code template as example\n```$language\n$lookup\n```\n"
                 }
 
                 ChatContextItem(JavaTestContextProvider::class, testServicePrompt)
@@ -63,7 +63,8 @@ class KotlinTestContextProvider : JavaTestContextProvider() {
             else -> {
                 val lookup = project.service<TemplatedTestPrompt>().lookup("Test.kt")
                 if (lookup != null) {
-                    prompt += "Here is a template as example\n```$language\n$lookup\n```\n"
+                    prompt += "\n" +
+                            "Here is the Test code template as example\n```$language\n$lookup\n```\n"
                 }
 
                 ChatContextItem(JavaTestContextProvider::class, prompt)

diff --git a/prompts/templates/Test.kt b/prompts/templates/Test.kt
@@ -1,3 +1,4 @@
+// Here is the Test template code.
 // for test intellij plugin
 import com.intellij.testFramework.LightPlatformTestCase
 class /*TestClassName*/Test : LightPlatformTestCase() {