From 29c5d0bfab56b3cbbb166f3b6c81675e55f768ab Mon Sep 17 00:00:00 2001
From: Anna Kozlova <anna.kozlova@jetbrains.com>
Date: Mon, 6 May 2024 12:12:27 +0200
Subject: [PATCH] [kotlin] setup kotlin plugin mode for multi module tests

GitOrigin-RevId: e3b25f3c6a104182900d407038921af6b6aae4e9
---
 .../safeDelete/AbstractMultiModuleSafeDeleteTest.kt        | 7 +++++++
 .../KotlinMultiModuleChangeSignatureTest.kt                | 7 +++++++
 2 files changed, 14 insertions(+)

diff --git a/plugins/kotlin/idea/tests/test/org/jetbrains/kotlin/idea/refactoring/safeDelete/AbstractMultiModuleSafeDeleteTest.kt b/plugins/kotlin/idea/tests/test/org/jetbrains/kotlin/idea/refactoring/safeDelete/AbstractMultiModuleSafeDeleteTest.kt
index 31391175744c..23b1248118a7 100644
--- a/plugins/kotlin/idea/tests/test/org/jetbrains/kotlin/idea/refactoring/safeDelete/AbstractMultiModuleSafeDeleteTest.kt
+++ b/plugins/kotlin/idea/tests/test/org/jetbrains/kotlin/idea/refactoring/safeDelete/AbstractMultiModuleSafeDeleteTest.kt
@@ -16,10 +16,17 @@ import org.jetbrains.kotlin.idea.refactoring.runRefactoringTest
 import org.jetbrains.kotlin.idea.test.IDEA_TEST_DATA_DIR
 import org.jetbrains.kotlin.idea.test.KotlinMultiFileTestCase
 import org.jetbrains.kotlin.idea.test.KotlinTestUtils
+import org.jetbrains.kotlin.idea.test.util.setUpWithKotlinPlugin
 import org.jetbrains.kotlin.psi.psiUtil.getNonStrictParentOfType
 import java.io.File
 
 abstract class AbstractMultiModuleSafeDeleteTest : KotlinMultiFileTestCase() {
+    override fun setUp() {
+        setUpWithKotlinPlugin(isFirPlugin()) {
+            super.setUp()
+        }
+    }
+
     object SafeDeleteAction : AbstractMultifileRefactoringTest.RefactoringAction {
         override fun runRefactoring(rootDir: VirtualFile, mainFile: PsiFile, elementsAtCaret: List<PsiElement>, config: JsonObject) {
             @Suppress("UNCHECKED_CAST")
diff --git a/plugins/kotlin/refactorings/kotlin.refactorings.tests.k2/test/org/jetbrains/kotlin/idea/k2/refactoring/changeSignature/KotlinMultiModuleChangeSignatureTest.kt b/plugins/kotlin/refactorings/kotlin.refactorings.tests.k2/test/org/jetbrains/kotlin/idea/k2/refactoring/changeSignature/KotlinMultiModuleChangeSignatureTest.kt
index 686038f6dbc4..d4bdc8aa1690 100644
--- a/plugins/kotlin/refactorings/kotlin.refactorings.tests.k2/test/org/jetbrains/kotlin/idea/k2/refactoring/changeSignature/KotlinMultiModuleChangeSignatureTest.kt
+++ b/plugins/kotlin/refactorings/kotlin.refactorings.tests.k2/test/org/jetbrains/kotlin/idea/k2/refactoring/changeSignature/KotlinMultiModuleChangeSignatureTest.kt
@@ -11,6 +11,7 @@ import org.jetbrains.kotlin.idea.refactoring.changeSignature.KotlinValVar
 import org.jetbrains.kotlin.idea.test.IDEA_TEST_DATA_DIR
 import org.jetbrains.kotlin.idea.test.KotlinMultiFileTestCase
 import org.jetbrains.kotlin.idea.test.extractMarkerOffset
+import org.jetbrains.kotlin.idea.test.util.setUpWithKotlinPlugin
 import org.jetbrains.kotlin.psi.KtElement
 import org.jetbrains.kotlin.psi.KtNamedDeclaration
 import org.jetbrains.kotlin.psi.KtPsiFactory
@@ -24,6 +25,12 @@ class KotlinMultiModuleChangeSignatureTest : KotlinMultiFileTestCase() {
         isMultiModule = true
     }
 
+    override fun setUp() {
+        setUpWithKotlinPlugin(true) {
+            super.setUp()
+        }
+    }
+
     override fun getTestRoot(): String = "/refactoring/changeSignatureMultiModule/"
 
     override fun getTestDataDirectory() = IDEA_TEST_DATA_DIR