1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
// Copyright 2000-2021 JetBrains s.r.o. and contributors. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
package org.jetbrains.kotlin.idea.structuralsearch
import com.intellij.openapi.util.text.StringUtil
import com.intellij.structuralsearch.MatchOptions
import com.intellij.structuralsearch.Matcher
import com.intellij.structuralsearch.PatternContext
import com.intellij.structuralsearch.StructuralSearchUtil
import com.intellij.structuralsearch.impl.matcher.CompiledPattern
import com.intellij.structuralsearch.impl.matcher.compiler.PatternCompiler
import com.intellij.structuralsearch.inspection.SSBasedInspection
import com.intellij.structuralsearch.inspection.StructuralSearchProfileActionProvider
import com.intellij.structuralsearch.plugin.ui.SearchConfiguration
import com.intellij.structuralsearch.plugin.ui.UIUtil
import com.intellij.testFramework.LightProjectDescriptor
import com.intellij.testFramework.fixtures.BasePlatformTestCase
import com.intellij.util.SmartList
import org.jetbrains.kotlin.idea.KotlinFileType
import org.jetbrains.kotlin.idea.test.IDEA_TEST_DATA_DIR
import org.jetbrains.kotlin.idea.test.ProjectDescriptorWithStdlibSources
import org.jetbrains.kotlin.idea.test.util.slashedPath
abstract class KotlinSSResourceInspectionTest : BasePlatformTestCase() {
private var myInspection: SSBasedInspection? = null
private val myConfiguration = SearchConfiguration().apply {
name = "SSR"
matchOptions.setFileType(KotlinFileType.INSTANCE)
}
override fun getProjectDescriptor(): LightProjectDescriptor = ProjectDescriptorWithStdlibSources.INSTANCE
override fun setUp() {
super.setUp()
myInspection = SSBasedInspection()
myFixture.enableInspections(myInspection)
}
protected fun doTest(pattern: String, context: PatternContext = KotlinStructuralSearchProfile.DEFAULT_CONTEXT) {
myFixture.configureByFile(getTestName(true) + ".kt")
val options = myConfiguration.matchOptions.apply {
fillSearchCriteria(pattern)
patternContext = context
}
Matcher.validate(project, options)
val message = checkApplicableConstraints(options, PatternCompiler.compilePattern(project, options, true, false))
assertNull("Constraint applicability error: $message\n", message)
StructuralSearchProfileActionProvider.createNewInspection(myConfiguration, project)
myFixture.testHighlighting(true, false, false)
}
override fun getTestDataPath(): String = IDEA_TEST_DATA_DIR.resolve("structuralsearch/$basePath").slashedPath
companion object {
fun checkApplicableConstraints(options: MatchOptions, compiledPattern: CompiledPattern): String? {
val profile = StructuralSearchUtil.getProfileByFileType(options.fileType)!!
for (varName in options.variableConstraintNames) {
val nodes = compiledPattern.getVariableNodes(varName)
val constraint = options.getVariableConstraint(varName)
val usedConstraints: MutableList<String> = SmartList()
if (!StringUtil.isEmpty(constraint.regExp)) usedConstraints.add(UIUtil.TEXT)
if (constraint.isWithinHierarchy) usedConstraints.add(UIUtil.TEXT_HIERARCHY)
if (constraint.minCount == 0) usedConstraints.add(UIUtil.MINIMUM_ZERO)
if (constraint.maxCount > 1) usedConstraints.add(UIUtil.MAXIMUM_UNLIMITED)
if (!StringUtil.isEmpty(constraint.nameOfExprType)) usedConstraints.add(UIUtil.TYPE)
if (!StringUtil.isEmpty(constraint.nameOfFormalArgType)) usedConstraints.add(UIUtil.EXPECTED_TYPE)
if (!StringUtil.isEmpty(constraint.referenceConstraint)) usedConstraints.add(UIUtil.REFERENCE)
usedConstraints.firstOrNull { !profile.isApplicableConstraint(it, nodes, false, constraint.isPartOfSearchResults) }?.let {
return@checkApplicableConstraints "$it not applicable for $varName"
}
}
return null
}
}
}
|
