Skip to content

Commit 968d94b

Browse files
committed
fix: Update line number retrieval in UnitTestsCVProvider
- Replace direct startOffset/endOffset with textRange.startOffset/endOffset - Apply consistent approach for both element and element.parent - Ensure proper line number calculation for unit test generation
1 parent 977011e commit 968d94b

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/main/kotlin/ai/devchat/plugin/hints/UnitTestsCVProvider.kt

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -17,10 +17,10 @@ class UnitTestsCVProvider : ChatCVProviderBase() {
1717
"message" to "/unit_tests " + listOf(
1818
FileDocumentManager.getInstance().getFile(editor.document)!!.path,
1919
(element as? PsiNamedElement)?.name,
20-
editor.document.getLineNumber(element.startOffset),
21-
editor.document.getLineNumber(element.endOffset),
22-
editor.document.getLineNumber(element.parent.startOffset),
23-
editor.document.getLineNumber(element.parent.endOffset),
20+
editor.document.getLineNumber(element.textRange.startOffset),
21+
editor.document.getLineNumber(element.textRange.endOffset),
22+
editor.document.getLineNumber(element.parent.textRange.startOffset),
23+
editor.document.getLineNumber(element.parent.textRange.endOffset),
2424
).joinToString(":::"),
2525
)
2626
)

0 commit comments

Comments
 (0)