fix(icon-provider): Fix wrong icon for some files

JetBrains · Jul 15, 2020 · b976cd8 · b976cd8
1 parent 03ad920
commit b976cd8
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/src/main/kotlin/com/pestphp/pest/PestIconProvider.kt b/src/main/kotlin/com/pestphp/pest/PestIconProvider.kt
@@ -7,8 +7,10 @@ import javax.swing.Icon
 
 class PestIconProvider : IconProvider() {
     override fun getIcon(element: PsiElement, flags: Int): Icon? {
-        return (element as? PsiFile)
-                ?.isPestTestFile()
-                .let { PestIcons.FILE }
+        if (element is PsiFile && element.isPestTestFile()) {
+            return PestIcons.FILE
+        }
+
+        return null
     }
 }
diff --git a/src/test/kotlin/com/pestphp/pest/tests/PestIconProviderTest.kt b/src/test/kotlin/com/pestphp/pest/tests/PestIconProviderTest.kt
@@ -30,8 +30,8 @@ class PestIconProviderTest: PestLightCodeFixture() {
         val file = myFixture.configureByFile("SimpleScript.php")
 
         assertNotEquals(
-            PestIconProvider().getIcon(file, ICON_FLAG_VISIBILITY),
-            PestIcons.FILE
+            PestIcons.FILE,
+            PestIconProvider().getIcon(file, ICON_FLAG_VISIBILITY)
         )
     }
 }