feat(test): update for sample

unit-mesh · Jul 28, 2023 · 636af2b · 636af2b
1 parent b610e6b
commit 636af2b
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/main/kotlin/cc/unitmesh/devti/intentions/WriteTestIntention.kt b/src/main/kotlin/cc/unitmesh/devti/intentions/WriteTestIntention.kt
@@ -88,8 +88,10 @@ class WriteTestIntention : AbstractChatIntention() {
 
                         prompter += "\n```${lang.lowercase()}\n$selectedText\n```\n"
 
-                        if (!testContext.isNewFile) {
-                            prompter += "Start writing test method code here:  \n"
+                        prompter += if (!testContext.isNewFile) {
+                            "Start writing test method code here:  \n"
+                        } else {
+                            "Start with `import` syntax here:  \n"
                         }
 
                         val flow: Flow<String> = ConnectorFactory.getInstance().connector(project).stream(prompter, "")