193: Fix code fragment tests fixture configuration
This commit is contained in:
Vyacheslav Gerasimov
+154
@@ -0,0 +1,154 @@
|
||||
/*
|
||||
* 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.debugger.evaluate
|
||||
|
||||
import com.intellij.codeInsight.completion.CompletionType
|
||||
import com.intellij.codeInspection.InspectionProfileEntry
|
||||
import com.intellij.openapi.util.io.FileUtil
|
||||
import com.intellij.openapi.util.text.StringUtil
|
||||
import com.intellij.psi.PsiElement
|
||||
import com.intellij.psi.util.PsiTreeUtil
|
||||
import com.intellij.testFramework.fixtures.JavaCodeInsightTestFixture
|
||||
import org.jetbrains.kotlin.checkers.AbstractPsiCheckerTest
|
||||
import org.jetbrains.kotlin.idea.caches.resolve.analyzeWithContent
|
||||
import org.jetbrains.kotlin.idea.caches.resolve.resolveImportReference
|
||||
import org.jetbrains.kotlin.idea.completion.test.AbstractJvmBasicCompletionTest
|
||||
import org.jetbrains.kotlin.idea.completion.test.ExpectedCompletionUtils
|
||||
import org.jetbrains.kotlin.idea.completion.test.handlers.AbstractCompletionHandlerTest
|
||||
import org.jetbrains.kotlin.idea.debugger.getContextElement
|
||||
import org.jetbrains.kotlin.idea.test.KotlinWithJdkAndRuntimeLightProjectDescriptor
|
||||
import org.jetbrains.kotlin.idea.util.ImportInsertHelper
|
||||
import org.jetbrains.kotlin.idea.util.application.executeWriteCommand
|
||||
import org.jetbrains.kotlin.name.FqName
|
||||
import org.jetbrains.kotlin.psi.*
|
||||
import org.jetbrains.kotlin.resolve.BindingContext
|
||||
import org.jetbrains.kotlin.test.InTextDirectivesUtils
|
||||
import org.jetbrains.kotlin.test.KotlinTestUtils
|
||||
import java.io.File
|
||||
import kotlin.test.assertNull
|
||||
import kotlin.test.assertTrue
|
||||
|
||||
abstract class AbstractCodeFragmentHighlightingTest : AbstractPsiCheckerTest() {
|
||||
override fun doTest(filePath: String) {
|
||||
myFixture.configureByCodeFragment(testDataFile())
|
||||
checkHighlighting(filePath)
|
||||
}
|
||||
|
||||
fun doTestWithImport(filePath: String) {
|
||||
myFixture.configureByCodeFragment(testDataFile())
|
||||
|
||||
project.executeWriteCommand("Imports insertion") {
|
||||
val fileText = FileUtil.loadFile(File(filePath), true)
|
||||
val file = myFixture.file as KtFile
|
||||
InTextDirectivesUtils.findListWithPrefixes(fileText, "// IMPORT: ").forEach {
|
||||
val descriptor = file.resolveImportReference(FqName(it)).singleOrNull()
|
||||
?: error("Could not resolve descriptor to import: $it")
|
||||
ImportInsertHelper.getInstance(project).importDescriptor(file, descriptor)
|
||||
}
|
||||
}
|
||||
|
||||
checkHighlighting(filePath)
|
||||
}
|
||||
|
||||
private fun checkHighlighting(filePath: String) {
|
||||
val inspectionName = InTextDirectivesUtils.findStringWithPrefixes(File(filePath).readText(), "// INSPECTION_CLASS: ")
|
||||
if (inspectionName != null) {
|
||||
val inspection = Class.forName(inspectionName).newInstance() as InspectionProfileEntry
|
||||
myFixture.enableInspections(inspection)
|
||||
try {
|
||||
myFixture.checkHighlighting(true, false, false)
|
||||
} finally {
|
||||
myFixture.disableInspections(inspection)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
myFixture.checkHighlighting(true, false, false)
|
||||
}
|
||||
}
|
||||
|
||||
abstract class AbstractCodeFragmentCompletionTest : AbstractJvmBasicCompletionTest() {
|
||||
override fun setUpFixture(testPath: String) {
|
||||
myFixture.configureByCodeFragment(testDataFile())
|
||||
}
|
||||
}
|
||||
|
||||
abstract class AbstractCodeFragmentCompletionHandlerTest : AbstractCompletionHandlerTest(CompletionType.BASIC) {
|
||||
override fun setUpFixture(testPath: String) {
|
||||
myFixture.configureByCodeFragment(testDataFile())
|
||||
}
|
||||
|
||||
override fun doTest(testPath: String) {
|
||||
super.doTest(testPath)
|
||||
|
||||
val fragment = myFixture.file as KtCodeFragment
|
||||
fragment.checkImports(testPath)
|
||||
}
|
||||
}
|
||||
|
||||
abstract class AbstractCodeFragmentAutoImportTest : AbstractPsiCheckerTest() {
|
||||
override fun doTest(filePath: String) {
|
||||
myFixture.configureByCodeFragment(testDataFile())
|
||||
myFixture.doHighlighting()
|
||||
|
||||
val importFix = myFixture.availableIntentions.singleOrNull { it.familyName == "Import" }
|
||||
?: error("No import fix available")
|
||||
importFix.invoke(project, editor, file)
|
||||
|
||||
myFixture.checkResultByFile(filePath + ".after")
|
||||
|
||||
val fragment = myFixture.file as KtCodeFragment
|
||||
fragment.checkImports(testDataPath + File.separator + filePath)
|
||||
|
||||
val fixAfter = myFixture.availableIntentions.firstOrNull { it.familyName == "Import" }
|
||||
assertNull(fixAfter, "No import fix should be available after")
|
||||
}
|
||||
|
||||
override fun getProjectDescriptor() = KotlinWithJdkAndRuntimeLightProjectDescriptor.INSTANCE
|
||||
override fun getTestDataPath() = KotlinTestUtils.getHomeDirectory()
|
||||
}
|
||||
|
||||
private fun KtCodeFragment.checkImports(testPath: String) {
|
||||
val importList = importsAsImportList()
|
||||
val importsText = StringUtil.convertLineSeparators(importList?.text ?: "")
|
||||
val fragmentAfterFile = File(testPath + ".after.imports")
|
||||
|
||||
if (fragmentAfterFile.exists()) {
|
||||
KotlinTestUtils.assertEqualsToFile(fragmentAfterFile, importsText)
|
||||
}
|
||||
else {
|
||||
assertTrue(importsText.isEmpty(), "Unexpected imports found: $importsText" )
|
||||
}
|
||||
}
|
||||
|
||||
private fun JavaCodeInsightTestFixture.configureByCodeFragment(testDataFile: File) {
|
||||
configureByFile(testDataFile.name)
|
||||
|
||||
val elementAt = file?.findElementAt(caretOffset)
|
||||
val file = createCodeFragment(testDataFile.path, elementAt!!)
|
||||
|
||||
val typeStr = InTextDirectivesUtils.findStringWithPrefixes(getFile().text, "// ${ExpectedCompletionUtils.RUNTIME_TYPE} ")
|
||||
if (typeStr != null) {
|
||||
file.putCopyableUserData(KtCodeFragment.RUNTIME_TYPE_EVALUATOR, {
|
||||
val codeFragment = KtPsiFactory(project).createBlockCodeFragment("val xxx: $typeStr", PsiTreeUtil.getParentOfType(elementAt, KtElement::class.java))
|
||||
val context = codeFragment.analyzeWithContent()
|
||||
val typeReference: KtTypeReference = PsiTreeUtil.getChildOfType(codeFragment.getContentElement().firstChild, KtTypeReference::class.java)!!
|
||||
context[BindingContext.TYPE, typeReference]
|
||||
})
|
||||
}
|
||||
|
||||
configureFromExistingVirtualFile(file.virtualFile!!)
|
||||
}
|
||||
|
||||
private fun createCodeFragment(filePath: String, contextElement: PsiElement): KtCodeFragment {
|
||||
val fileForFragment = File(filePath + ".fragment")
|
||||
val codeFragmentText = FileUtil.loadFile(fileForFragment, true).trim()
|
||||
val psiFactory = KtPsiFactory(contextElement.project)
|
||||
if (fileForFragment.readLines().size == 1) {
|
||||
return psiFactory.createExpressionCodeFragment(codeFragmentText, getContextElement(contextElement))
|
||||
}
|
||||
return psiFactory.createBlockCodeFragment(codeFragmentText, getContextElement(contextElement))
|
||||
}
|
||||
+49
@@ -0,0 +1,49 @@
|
||||
/*
|
||||
* 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.debugger.evaluate
|
||||
|
||||
import com.intellij.testFramework.fixtures.LightCodeInsightFixtureTestCase
|
||||
import org.jetbrains.kotlin.idea.debugger.KotlinEditorTextProvider
|
||||
import org.jetbrains.kotlin.idea.test.KotlinLightProjectDescriptor
|
||||
import org.jetbrains.kotlin.idea.test.PluginTestCaseBase
|
||||
import org.jetbrains.kotlin.idea.test.invalidateLibraryCache
|
||||
import org.jetbrains.kotlin.test.InTextDirectivesUtils
|
||||
import org.junit.Assert
|
||||
|
||||
abstract class AbstractSelectExpressionForDebuggerTest : LightCodeInsightFixtureTestCase() {
|
||||
|
||||
override fun setUp() {
|
||||
super.setUp()
|
||||
invalidateLibraryCache(project)
|
||||
}
|
||||
|
||||
fun doTest(path: String) {
|
||||
doTest(path, true)
|
||||
}
|
||||
|
||||
fun doTestWoMethodCalls(path: String) {
|
||||
doTest(path, false)
|
||||
}
|
||||
|
||||
fun doTest(path: String, allowMethodCalls: Boolean) {
|
||||
myFixture.configureByFile(path.substringAfterLast('/'))
|
||||
|
||||
val elementAt = myFixture.file?.findElementAt(myFixture.caretOffset)!!
|
||||
val selectedExpression = KotlinEditorTextProvider.findExpressionInner(elementAt, allowMethodCalls)
|
||||
|
||||
val expected = InTextDirectivesUtils.findStringWithPrefixes(myFixture.file?.text!!, "// EXPECTED: ")
|
||||
val actualResult = if (selectedExpression != null)
|
||||
KotlinEditorTextProvider.getElementInfo(selectedExpression) { it.text }
|
||||
else
|
||||
"null"
|
||||
|
||||
Assert.assertEquals("Another expression should be selected", expected, actualResult)
|
||||
}
|
||||
|
||||
override fun getProjectDescriptor() = KotlinLightProjectDescriptor.INSTANCE
|
||||
|
||||
override fun getTestDataPath() = PluginTestCaseBase.getTestDataPathBase() + "/debugger/selectExpression"
|
||||
}
|
||||
Reference in New Issue
Block a user