From aa046640c03c7a63702de1d5cc7c35478cf7d470 Mon Sep 17 00:00:00 2001 From: Sergey Loktev Date: Tue, 7 Mar 2023 17:27:16 +0300 Subject: [PATCH] Fixed non-indexing test files --- .../intellij/plugin/language/js/JsDialogProcessor.kt | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/utbot-intellij-js/src/main/kotlin/org/utbot/intellij/plugin/language/js/JsDialogProcessor.kt b/utbot-intellij-js/src/main/kotlin/org/utbot/intellij/plugin/language/js/JsDialogProcessor.kt index c770587677..3a1f9bbfd1 100644 --- a/utbot-intellij-js/src/main/kotlin/org/utbot/intellij/plugin/language/js/JsDialogProcessor.kt +++ b/utbot-intellij-js/src/main/kotlin/org/utbot/intellij/plugin/language/js/JsDialogProcessor.kt @@ -186,13 +186,16 @@ object JsDialogProcessor { val generatedCode = testGenerator.run() invokeLater { runWriteAction { - val testPsiFile = testDir.findFile(testFileName) ?: PsiFileFactory.getInstance(project) - .createFileFromText(testFileName, JsLanguageAssistant.jsLanguage, generatedCode) + val testPsiFile = testDir.findFile(testFileName) ?: run { + val temp = PsiFileFactory.getInstance(project) + .createFileFromText(testFileName, JsLanguageAssistant.jsLanguage, generatedCode) + testDir.add(temp) + testDir.findFile(testFileName)!! + } val testFileEditor = CodeInsightUtil.positionCursor(project, testPsiFile, testPsiFile) unblockDocument(project, testFileEditor.document) testFileEditor.document.setText(generatedCode) unblockDocument(project, testFileEditor.document) - testDir.findFile(testFileName) ?: testDir.add(testPsiFile) } } }