as31: Fix AbstractImportsTest, add missing project parameter

Settings by default taken from per-project settings manager instead of app settings
2017-10-20 01:35:54 +03:00
parent 9bb3bb2777
commit d58b0798cc
1 changed files with 96 additions and 0 deletions
@@ -0,0 +1,96 @@
+/*
+ * Copyright 2010-2015 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.jetbrains.kotlin
+
+import com.intellij.psi.codeStyle.CodeStyleSettingsManager
+import com.intellij.psi.codeStyle.PackageEntry
+import com.intellij.testFramework.LightProjectDescriptor
+import junit.framework.TestCase
+import org.jetbrains.kotlin.idea.core.formatter.KotlinCodeStyleSettings
+import org.jetbrains.kotlin.idea.test.KotlinLightCodeInsightFixtureTestCase
+import org.jetbrains.kotlin.idea.test.KotlinWithJdkAndRuntimeLightProjectDescriptor
+import org.jetbrains.kotlin.idea.util.application.executeWriteCommand
+import org.jetbrains.kotlin.psi.KtFile
+import org.jetbrains.kotlin.test.InTextDirectivesUtils
+import org.jetbrains.kotlin.test.KotlinTestUtils
+import java.io.File
+
+abstract class AbstractImportsTest : KotlinLightCodeInsightFixtureTestCase() {
+    override fun getTestDataPath() = KotlinTestUtils.getHomeDirectory()
+    override fun getProjectDescriptor(): LightProjectDescriptor = KotlinWithJdkAndRuntimeLightProjectDescriptor.INSTANCE
+
+    protected fun doTest(testPath: String) {
+        val settingManager = CodeStyleSettingsManager.getInstance(project)
+        val tempSettings = settingManager.currentSettings.clone()
+        settingManager.setTemporarySettings(tempSettings)
+
+        val codeStyleSettings = KotlinCodeStyleSettings.getInstance(project)
+
+        try {
+            val fixture = myFixture
+            val dependencySuffixes = listOf(".dependency.kt", ".dependency.java", ".dependency1.kt", ".dependency2.kt")
+            for (suffix in dependencySuffixes) {
+                val dependencyPath = testPath.replace(".kt", suffix)
+                if (File(dependencyPath).exists()) {
+                    fixture.configureByFile(dependencyPath)
+                }
+            }
+
+            fixture.configureByFile(testPath)
+
+            val file = fixture.file as KtFile
+
+            val fileText = file.text
+
+            codeStyleSettings.NAME_COUNT_TO_USE_STAR_IMPORT = InTextDirectivesUtils.getPrefixedInt(fileText, "// NAME_COUNT_TO_USE_STAR_IMPORT:") ?: nameCountToUseStarImportDefault
+            codeStyleSettings.NAME_COUNT_TO_USE_STAR_IMPORT_FOR_MEMBERS = InTextDirectivesUtils.getPrefixedInt(fileText, "// NAME_COUNT_TO_USE_STAR_IMPORT_FOR_MEMBERS:") ?: nameCountToUseStarImportForMembersDefault
+            codeStyleSettings.IMPORT_NESTED_CLASSES = InTextDirectivesUtils.getPrefixedBoolean(fileText, "// IMPORT_NESTED_CLASSES:") ?: false
+
+            InTextDirectivesUtils.findLinesWithPrefixesRemoved(fileText, "// PACKAGE_TO_USE_STAR_IMPORTS:").forEach {
+                codeStyleSettings.PACKAGES_TO_USE_STAR_IMPORTS.addEntry(PackageEntry(false, it.trim(), false))
+            }
+            InTextDirectivesUtils.findLinesWithPrefixesRemoved(fileText, "// PACKAGES_TO_USE_STAR_IMPORTS:").forEach {
+                codeStyleSettings.PACKAGES_TO_USE_STAR_IMPORTS.addEntry(PackageEntry(false, it.trim(), true))
+            }
+
+            val log = project.executeWriteCommand<String?>("") { doTest(file) }
+
+            KotlinTestUtils.assertEqualsToFile(File(testPath + ".after"), myFixture.file.text)
+            if (log != null) {
+                val logFile = File(testPath + ".log")
+                if (log.isNotEmpty()) {
+                    KotlinTestUtils.assertEqualsToFile(logFile, log)
+                }
+                else {
+                    TestCase.assertFalse(logFile.exists())
+                }
+            }
+        }
+        finally {
+            settingManager.dropTemporarySettings()
+        }
+    }
+
+    // returns test log
+    protected abstract fun doTest(file: KtFile): String?
+
+    protected open val nameCountToUseStarImportDefault: Int
+        get() = 1
+
+    protected open val nameCountToUseStarImportForMembersDefault: Int
+        get() = 3
+}