193: Fix AbstractInlineTest fixture configuration

2019-10-04 20:09:13 +03:00
parent c07833f13a
commit a5d668a093
1 changed files with 84 additions and 0 deletions
@@ -0,0 +1,84 @@
+/*
+ * Copyright 2010-2019 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.idea.refactoring.inline
+
+import com.intellij.codeInsight.TargetElementUtil
+import com.intellij.codeInsight.TargetElementUtil.ELEMENT_NAME_ACCEPTED
+import com.intellij.codeInsight.TargetElementUtil.REFERENCED_ELEMENT_ACCEPTED
+import com.intellij.lang.refactoring.InlineActionHandler
+import com.intellij.openapi.extensions.Extensions
+import com.intellij.openapi.util.io.FileUtil
+import com.intellij.refactoring.BaseRefactoringProcessor
+import com.intellij.refactoring.util.CommonRefactoringUtil
+import com.intellij.testFramework.UsefulTestCase
+import com.intellij.testFramework.fixtures.JavaCodeInsightTestFixture
+import junit.framework.TestCase
+import org.jetbrains.kotlin.idea.test.KotlinLightCodeInsightFixtureTestCase
+import org.jetbrains.kotlin.idea.test.KotlinWithJdkAndRuntimeLightProjectDescriptor
+import org.jetbrains.kotlin.idea.test.configureCompilerOptions
+import org.jetbrains.kotlin.idea.test.rollbackCompilerOptions
+import org.jetbrains.kotlin.idea.util.application.runWriteAction
+import org.jetbrains.kotlin.test.InTextDirectivesUtils
+import org.jetbrains.kotlin.test.KotlinTestUtils
+import java.io.File
+
+abstract class AbstractInlineTest : KotlinLightCodeInsightFixtureTestCase() {
+    val fixture: JavaCodeInsightTestFixture
+        get() = myFixture
+
+    protected fun doTest(path: String) {
+        val mainFile = File(path)
+        val afterFile = File(path + ".after")
+
+        val mainFileName = mainFile.name
+        val mainFileBaseName = FileUtil.getNameWithoutExtension(mainFileName)
+        val extraFiles = mainFile.parentFile.listFiles { _, name ->
+            name != mainFileName && name.startsWith("$mainFileBaseName.") && (name.endsWith(".kt") || name.endsWith(".java"))
+        }
+        val extraFilesToPsi = extraFiles.associateBy { fixture.configureByFile(path.replace(mainFileName, it.name)) }
+        val file = myFixture.configureByFile(fileName())
+
+        val configured = configureCompilerOptions(file.text, project, module)
+
+        try {
+            val afterFileExists = afterFile.exists()
+
+            val targetElement = TargetElementUtil.findTargetElement(myFixture.editor, ELEMENT_NAME_ACCEPTED or REFERENCED_ELEMENT_ACCEPTED)!!
+            val handler = Extensions.getExtensions(InlineActionHandler.EP_NAME).firstOrNull { it.canInlineElement(targetElement) }
+            val expectedErrors = InTextDirectivesUtils.findLinesWithPrefixesRemoved(myFixture.file.text, "// ERROR: ")
+            if (handler != null) {
+                try {
+                    runWriteAction { handler.inlineElement(myFixture.project, myFixture.editor, targetElement) }
+
+                    UsefulTestCase.assertEmpty(expectedErrors)
+                    KotlinTestUtils.assertEqualsToFile(afterFile, file.text)
+                    for ((extraPsiFile, extraFile) in extraFilesToPsi) {
+                        KotlinTestUtils.assertEqualsToFile(File("${extraFile.path}.after"), extraPsiFile.text)
+                    }
+                }
+                catch (e: CommonRefactoringUtil.RefactoringErrorHintException) {
+                    TestCase.assertFalse("Refactoring not available: ${e.message}", afterFileExists)
+                    TestCase.assertEquals("Expected errors", 1, expectedErrors.size)
+                    TestCase.assertEquals("Error message", expectedErrors[0].replace("\\n", "\n"), e.message)
+                }
+                catch (e: BaseRefactoringProcessor.ConflictsInTestsException) {
+                    TestCase.assertFalse("Conflicts: ${e.message}", afterFileExists)
+                    TestCase.assertEquals("Expected errors", 1, expectedErrors.size)
+                    TestCase.assertEquals("Error message", expectedErrors[0].replace("\\n", "\n"), e.message)
+                }
+            }
+            else {
+                TestCase.assertFalse("No refactoring handler available", afterFileExists)
+            }
+        } finally {
+            if (configured) {
+                rollbackCompilerOptions(project, module)
+            }
+        }
+    }
+
+    override fun getProjectDescriptor() = KotlinWithJdkAndRuntimeLightProjectDescriptor.INSTANCE
+}