From d0da11321dce9a58e4c5c542e98e9e375d18aec0 Mon Sep 17 00:00:00 2001 From: Mikhail Glukhikh Date: Fri, 28 Apr 2017 18:22:21 +0300 Subject: [PATCH] Fix of SpringInspectionTestGenerated --- .../kotlin/idea/inspections/InspectionUtils.kt | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/idea/tests/org/jetbrains/kotlin/idea/inspections/InspectionUtils.kt b/idea/tests/org/jetbrains/kotlin/idea/inspections/InspectionUtils.kt index 1b138ce06a0..149cc28fb1d 100644 --- a/idea/tests/org/jetbrains/kotlin/idea/inspections/InspectionUtils.kt +++ b/idea/tests/org/jetbrains/kotlin/idea/inspections/InspectionUtils.kt @@ -24,8 +24,10 @@ import com.intellij.openapi.project.Project import com.intellij.openapi.vfs.VirtualFile import com.intellij.testFramework.InspectionTestUtil import com.intellij.testFramework.createGlobalContextForTool +import com.intellij.util.xml.highlighting.DomElementsInspection import org.jetbrains.kotlin.idea.inspections.gradle.DifferentKotlinGradleVersionInspection import org.jetbrains.kotlin.idea.maven.inspections.DifferentKotlinMavenVersionInspection +import org.jetbrains.plugins.gradle.codeInspection.GradleBaseInspection fun runInspection( inspection: LocalInspectionTool, project: Project, files: List? = null, withTestDir: String? = null @@ -33,9 +35,11 @@ fun runInspection( val wrapper = LocalInspectionToolWrapper(inspection) val tool = wrapper.tool - when (tool) { - is DifferentKotlinMavenVersionInspection -> tool.testVersionMessage = "\$PLUGIN_VERSION" - is DifferentKotlinGradleVersionInspection -> tool.testVersionMessage = "\$PLUGIN_VERSION" + if (tool is DomElementsInspection<*> || tool is GradleBaseInspection) { + when (tool) { + is DifferentKotlinMavenVersionInspection -> tool.testVersionMessage = "\$PLUGIN_VERSION" + is DifferentKotlinGradleVersionInspection -> tool.testVersionMessage = "\$PLUGIN_VERSION" + } } val scope = if (files != null) AnalysisScope(project, files) else AnalysisScope(project)