Skip to content

Commit 3a1153b

Browse files
committed
fix(javascript): fix file type conversion in JavaScriptTestCodeModifier
Ensure correct file type conversion for JSFile in insertMethod function.
1 parent bae6d12 commit 3a1153b

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

javascript/src/main/kotlin/cc/unitmesh/ide/javascript/provider/testing/JavaScriptTestCodeModifier.kt

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,9 +30,10 @@ open class JavaScriptTestCodeModifier : CodeModifier {
3030

3131
override fun insertMethod(sourceFile: VirtualFile, project: Project, code: String): Boolean {
3232
// todo: spike for insert different method type, like named function, arrow function, etc.
33-
val psiElement = (sourceFile as JSFile).lastChild
33+
val jsFile = PsiManager.getInstance(project).findFile(sourceFile) as JSFile
34+
val psiElement = jsFile.lastChild
3435

35-
val element = PsiFileFactory.getInstance(project).createFileFromText(sourceFile.language, "")
36+
val element = PsiFileFactory.getInstance(project).createFileFromText(jsFile.language, "")
3637
val codeElement = JSPsiElementFactory.createJSStatement(code, element)
3738

3839
runReadAction {

0 commit comments

Comments
 (0)