Skip to content

Commit 3e3babf

Browse files
Merge main into feature/q-lsp
2 parents 847b17a + 991abe2 commit 3e3babf

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed
Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
{
2+
"type" : "bugfix",
3+
"description" : "Amazon Q /test: Q identify active test file and infer source file for test generation."
4+
}

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqCodeTest/CodeWhispererUTGChatManager.kt

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,7 @@ import java.io.ByteArrayInputStream
5959
import java.io.ByteArrayOutputStream
6060
import java.io.File
6161
import java.io.IOException
62+
import java.nio.file.Path
6263
import java.nio.file.Paths
6364
import java.time.Duration
6465
import java.time.Instant
@@ -249,9 +250,10 @@ class CodeWhispererUTGChatManager(val project: Project, private val cs: Coroutin
249250
if (shortAnswer.stopIteration == "true") {
250251
throw CodeTestException("TestGenFailedError: ${shortAnswer.planSummary}", "TestGenFailedError", shortAnswer.planSummary)
251252
}
253+
val fileName = shortAnswer.sourceFilePath?.let { Path.of(it).fileName.toString() } ?: path.fileName.toString()
252254
codeTestChatHelper.updateAnswer(
253255
CodeTestChatMessageContent(
254-
message = generateSummaryMessage(path.fileName.toString()) + shortAnswer.planSummary,
256+
message = generateSummaryMessage(fileName) + shortAnswer.planSummary,
255257
type = ChatMessageType.Answer
256258
),
257259
messageIdOverride = codeTestResponseContext.testSummaryMessageId

0 commit comments

Comments
 (0)