Adds Run icon in gutter next to ExUnit tests

CHANGELOG.md

@@ -221,6 +221,8 @@
 ### Enhancements
 * [#1792](https://github.com/KronicDeth/intellij-elixir/pull/1792) - [@marcindawidziuk](https://github.com/marcindawidziuk)
   * Quick documentation (F1/Ctrl+Q)
+* [#1807](https://github.com/KronicDeth/intellij-elixir/pull/1807) - [@niknetniko](https://github.com/niknetniko)
+  * Run Icons in gutter next to ExUnit tests.  The icon changes to the state of the test: green for passing and red for failing. 
 
 ### Bug Fixes
 * [#1788](https://github.com/KronicDeth/intellij-elixir/pull/1788) - [@KronicDeth](https://github.com/KronicDeth)

resources/META-INF/changelog.html

@@ -6,6 +6,10 @@ <h1>v11.8.0</h1>
     <p>Enhancements</p>
     <ul>
       <li>Quick documentation (F1/Ctrl+Q)</li>
+      <li>
+        Run Icons in gutter next to ExUnit tests.  The icon changes to the state of the test: green for passing and
+        red for failing.
+      </li>
     </ul>
   </li>
   <li>

resources/META-INF/plugin.xml

@@ -174,6 +174,9 @@
     <lang.formatter language="EEx" implementationClass="com.intellij.psi.templateLanguages.SimpleTemplateLanguageFormattingModelBuilder"/>
     <langCodeStyleSettingsProvider implementation="org.elixir_lang.formatter.settings.LanguageCodeStyleSettingsProvider"/>
 
+    <!-- test -->
+    <runLineMarkerContributor language="Elixir" implementationClass="org.elixir_lang.exunit.ExUnitLineMarkerProvider"/>
+
     <lang.parserDefinition language="Elixir" implementationClass="org.elixir_lang.ElixirParserDefinition"/>
     <lang.psiStructureViewFactory language="Elixir" implementationClass="org.elixir_lang.structure_view.Factory"/>
     <lang.quoteHandler language="Elixir" implementationClass="org.elixir_lang.QuoteHandler"/>

src/org/elixir_lang/exunit/TestLineMarkerProvider.kt

@@ -0,0 +1,75 @@
+package org.elixir_lang.exunit
+
+import com.intellij.execution.TestStateStorage
+import com.intellij.execution.lineMarker.RunLineMarkerContributor
+import com.intellij.openapi.roots.ProjectRootManager
+import com.intellij.psi.PsiElement
+import com.intellij.psi.impl.source.tree.LeafPsiElement
+import org.codehaus.plexus.interpolation.os.Os
+import org.elixir_lang.exunit.configuration.SUFFIX
+import org.elixir_lang.exunit.configuration.lineNumber
+import org.elixir_lang.mix.TestFinder
+import org.elixir_lang.psi.ElixirFile
+import org.elixir_lang.psi.ElixirUnmatchedQualifiedParenthesesCall
+import org.elixir_lang.psi.call.Call
+import org.elixir_lang.psi.operation.Match
+
+/**
+ * Provides the gutter "run" icons for tests.
+ */
+class ExUnitLineMarkerProvider: RunLineMarkerContributor() {
+
+    override fun getInfo(element: PsiElement): Info? {
+
+        // Speed things up by only looking at files inside the test directory.
+        if (!isUnderTestSources(element)) {
+            return null
+        }
+
+        if (!TestFinder.isTestElement(element) || element !is LeafPsiElement) {
+            return null
+        }
+
+        val parent = element.parent?.parent
+
+        if (parent !is Call) {
+            return null
+        }
+
+        // Assignment
+        if (parent.parent is Match) {
+            return null
+        }
+
+        val functionElement = parent.firstChild
+
+        if (functionElement != element.parent) {
+            return null
+        }
+
+        val isClass = when (parent.functionName()) {
+            "test" -> false
+            "describe" -> true
+            "defmodule" -> true
+            "doctest" -> false
+            else -> null
+        } ?: return null
+
+        val number = lineNumber(element)
+        var url = "file://${element.containingFile.virtualFile.path}:${number}"
+        if (Os.isFamily(Os.FAMILY_WINDOWS)) {
+            // In Elixir, the drive letter is lowercase for some reason.
+            val location = "file://".length
+            url = url.replaceRange(location, location + 1, url[location].toLowerCase().toString())
+        }
+        val state = TestStateStorage.getInstance(element.project)?.getState(url)
+        return withExecutorActions(getTestStateIcon(state, isClass))
+    }
+}
+
+fun isUnderTestSources(clazz: PsiElement): Boolean {
+    val psiFile = clazz.containingFile
+    val vFile = psiFile.virtualFile ?: return false
+    val isTest = psiFile is ElixirFile && psiFile.virtualFile.path.endsWith(SUFFIX)
+    return ProjectRootManager.getInstance(clazz.project).fileIndex.isInTestSourceContent(vFile) and isTest
+}

src/org/elixir_lang/exunit/configuration/Producer.kt

@@ -28,7 +28,7 @@ class MixExUnitRunConfigurationProducer :
             } == true
 }
 
-private const val SUFFIX = "_test.exs"
+const val SUFFIX = "_test.exs"
 private const val UNKNOWN_LINE = -1
 
 private fun configurationName(file: PsiFileSystemItem,
@@ -71,7 +71,7 @@ private fun isConfigurationFromContextImpl(configuration: Configuration, psiElem
             contextConfiguration.workingDirectory == configuration.workingDirectory
 }
 
-private fun lineNumber(psiElement: PsiElement): Int {
+fun lineNumber(psiElement: PsiElement): Int {
     val containingFile = psiElement.containingFile
     val documentLineNumber = PsiDocumentManager
             .getInstance(containingFile.project)

src/org/elixir_lang/mix/TestFinder.java

@@ -30,11 +30,12 @@ private static Collection<PsiElement> corresponding(@NotNull PsiElement element,
                                                         @NotNull Function<String, String> correspondingName,
                                                         @NotNull Condition<Call> correspondingCallCondition) {
         Call sourceElement = sourceElement(element);
-        Collection<PsiElement> correspondingCollection = new ArrayList<PsiElement>();
+        Collection<PsiElement> correspondingCollection = new ArrayList<>();
 
-        if (sourceElement != null && sourceElement instanceof StubBased) {
+        if (sourceElement instanceof StubBased) {
+            @SuppressWarnings("rawtypes")
             StubBased sourceStubBased = (StubBased) sourceElement;
-            @SuppressWarnings("unchecked") Set<String> canonicalNameSet = sourceStubBased.canonicalNameSet();
+            Set<String> canonicalNameSet = sourceStubBased.canonicalNameSet();
 
             if (!canonicalNameSet.isEmpty()) {
                 Project project = element.getProject();
@@ -122,18 +123,8 @@ public Call findSourceElement(@NotNull PsiElement from) {
     public Collection<PsiElement> findTestsForClass(@NotNull PsiElement element) {
         return corresponding(
                 element,
-                new Function<String, String>() {
-                    @Override
-                    public String fun(@NotNull String canonicalName) {
-                        return canonicalName + TEST_SUFFIX;
-                    }
-                },
-                new Condition<Call>() {
-                    @Override
-                    public boolean value(Call call) {
-                        return Module.Companion.is(call);
-                    }
-                }
+                canonicalName -> canonicalName + TEST_SUFFIX,
+                Module.Companion::is
         );
     }
 
@@ -142,36 +133,36 @@ public boolean value(Call call) {
     public Collection<PsiElement> findClassesForTest(@NotNull PsiElement element) {
         return corresponding(
                 element,
-                new Function<String, String>() {
-                    @Override
-                    public String fun(String canonicalName) {
-                        String correspondingCanonicalName = null;
-
-                        if (canonicalName.endsWith(TEST_SUFFIX)) {
-                            correspondingCanonicalName =
-                                    canonicalName.substring(0, canonicalName.length() - TEST_SUFFIX.length());
-                        }
+                canonicalName -> {
+                    String correspondingCanonicalName = null;
 
-                        return correspondingCanonicalName;
+                    if (canonicalName.endsWith(TEST_SUFFIX)) {
+                        correspondingCanonicalName =
+                                canonicalName.substring(0, canonicalName.length() - TEST_SUFFIX.length());
                     }
+
+                    return correspondingCanonicalName;
                 },
-                new Condition<Call>() {
-                    @Override
-                    public boolean value(Call call) {
-                        return isModular(call);
-                    }
-                }
+                TestFinder::isModular
         );
     }
 
     @Override
     public boolean isTest(@NotNull PsiElement element) {
-        Call sourceElement = findSourceElement(element);
+        return isTestElement(element);
+    }
+
+    /**
+     * Check if the given element is a test or not.
+     */
+    public static boolean isTestElement(@NotNull PsiElement element) {
+        Call sourceElement = sourceElement(element);
         boolean isTest = false;
 
-        if (sourceElement != null && sourceElement instanceof StubBased) {
+        if (sourceElement instanceof StubBased) {
+            @SuppressWarnings("rawtypes")
             StubBased sourceStubBased = (StubBased) sourceElement;
-            @SuppressWarnings("unchecked") Set<String> canonicalNameSet = sourceStubBased.canonicalNameSet();
+            Set<String> canonicalNameSet = sourceStubBased.canonicalNameSet();
 
             if (!canonicalNameSet.isEmpty()) {
                 for (String canonicalName : canonicalNameSet) {