Test fix: spring test for gutter

JetBrains · Dec 22, 2016 · a78694e · a78694e
1 parent 4a1e6d3
commit a78694e
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/...e/tests/org/jetbrains/kotlin/idea/spring/tests/gutter/AbstractSpringClassAnnotatorTest.kt b/...e/tests/org/jetbrains/kotlin/idea/spring/tests/gutter/AbstractSpringClassAnnotatorTest.kt
@@ -63,11 +63,13 @@ abstract class AbstractSpringClassAnnotatorTest : KotlinLightCodeInsightFixtureT
             }
 
             val fileName = config.getString("file")
-            val gutterMark = myFixture.findGutter(fileName)!!
-
             val iconName = config.getString("icon")
             val icon = SpringApiIcons::class.java.getField(iconName)[null]
-            assert(gutterMark.icon == icon)
+
+            val gutterMark = myFixture.findGutter(fileName)!!.let {
+                if (it.icon == icon) it
+                else myFixture.findGuttersAtCaret().first { it.icon == icon }
+            }
 
             val tooltip = config.getString("tooltip")
             assert(gutterMark.tooltipText == tooltip)