fixed the namint of function

JetBrains-Research · Jul 23, 2024 · 3b70693 · 3b70693
1 parent ff73314
commit 3b70693
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/...ain/kotlin/org/jetbrains/research/testspark/core/generation/llm/prompt/PromptGenerator.kt b/...ain/kotlin/org/jetbrains/research/testspark/core/generation/llm/prompt/PromptGenerator.kt
@@ -46,9 +46,10 @@ class PromptGenerator(
         testSamplesCode: String,
         packageName: String,
     ): String {
+        val name = context.cut?.let { "${it.qualifiedName}.${method.name}" } ?: "$packageName.${method.name}"
         val prompt = PromptBuilder(promptTemplates.methodPrompt)
             .insertLanguage(context.promptConfiguration.desiredLanguage)
-            .insertName("$packageName.${method.name}")
+            .insertName(name)
             .insertTestingPlatform(context.promptConfiguration.desiredTestingPlatform)
             .insertMockingFramework(context.promptConfiguration.desiredMockingFramework)
             .insertCodeUnderTest(method.text, context.classesToTest)

diff --git a/src/main/kotlin/org/jetbrains/research/testspark/tools/llm/generation/LLMProcessManager.kt b/src/main/kotlin/org/jetbrains/research/testspark/tools/llm/generation/LLMProcessManager.kt
@@ -144,6 +144,7 @@ class LLMProcessManager(
             jUnitVersion,
             language,
             testBodyPrinter,
+            packageName
         )
         val testsAssembler = TestsAssemblerFactory.createTestsAssembler(
             indicator,