From 464d83996644fdb3d9e6baf51e9808d9e04d97fb Mon Sep 17 00:00:00 2001 From: Andrey Tarbeev Date: Tue, 22 Nov 2022 15:40:50 +0300 Subject: [PATCH] Add a flag to disable imports addition from formatter side --- .../intellij/plugin/generator/CodeGenerationController.kt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/generator/CodeGenerationController.kt b/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/generator/CodeGenerationController.kt index ebee300a47..20ee983d00 100644 --- a/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/generator/CodeGenerationController.kt +++ b/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/generator/CodeGenerationController.kt @@ -37,6 +37,7 @@ import com.intellij.psi.SmartPointerManager import com.intellij.psi.SmartPsiElementPointer import com.intellij.psi.codeStyle.CodeStyleManager import com.intellij.psi.codeStyle.JavaCodeStyleManager +import com.intellij.psi.codeStyle.JavaCodeStyleManager.DO_NOT_ADD_IMPORTS import com.intellij.psi.search.GlobalSearchScopesCore import com.intellij.testIntegration.TestIntegrationUtils import com.siyeh.ig.psiutils.ImportUtils @@ -773,7 +774,7 @@ object CodeGenerationController { val startOffset = range.startOffset val endOffset = range.endOffset val reformatRange = codeStyleManager.reformatRange(file, startOffset, endOffset, false) - JavaCodeStyleManager.getInstance(project).shortenClassReferences(reformatRange) + JavaCodeStyleManager.getInstance(project).shortenClassReferences(reformatRange, DO_NOT_ADD_IMPORTS) } CodegenLanguage.KOTLIN -> ShortenReferences.DEFAULT.process((testClass as KtUltraLightClass).kotlinOrigin.containingKtFile) }