Don't run heavy tests in _Run all_ action

Fixes #88

Author: demiurg906

src/org/jetbrains/kotlin/test/helper/services/TestDataRunnerService.kt

@@ -22,6 +22,7 @@ import org.jetbrains.kotlin.test.helper.buildRunnerLabel
 import org.jetbrains.kotlin.test.helper.gradle.GradleRunConfig
 import org.jetbrains.kotlin.test.helper.gradle.computeGradleCommandLine
 import org.jetbrains.kotlin.test.helper.gradle.runGradleCommandLine
+import org.jetbrains.kotlin.test.helper.isHeavyTest
 import org.jetbrains.kotlin.test.helper.toFileNamesString
 import javax.swing.ListSelectionModel
 
@@ -45,6 +46,7 @@ class TestDataRunnerService(
 
                 smartReadAction(project) {
                     val testDeclarations = filterAndCollectTestDeclarations(files, project)
+                        .filter { !it.isHeavyTest() }
 
                     val filtered = if (!filterByClass.isNullOrEmpty()) {
                         groupTests(testDeclarations)[filterByClass] ?: testDeclarations
@@ -124,4 +126,4 @@ class TestDataRunnerService(
             .groupBy { it.parentsOfType<PsiClass>().last() }
             .mapKeys { it.key.buildRunnerLabel(testTags) }
     }
-}
+}

src/org/jetbrains/kotlin/test/helper/utils.kt

@@ -1,6 +1,8 @@
 package org.jetbrains.kotlin.test.helper
 
 import com.intellij.psi.PsiClass
+import com.intellij.psi.PsiNameIdentifierOwner
+import com.intellij.psi.util.parentsOfType
 import java.io.IOException
 import java.nio.file.FileVisitResult
 import java.nio.file.Files
@@ -50,10 +52,19 @@ fun PsiClass.buildRunnerLabel(allTags: Map<String, Array<String>>): String {
         } else null
     }.orEmpty()
     return buildString {
+        if (this@buildRunnerLabel.isHeavyTest()) {
+            this.append("{H} ")
+        }
         if (tags.isNotEmpty()) {
             this.append(tags.joinToString(prefix = "[", postfix = "] ", separator = ", "))
         }
         this.append(runnerName)
     }
 }
 
+private const val HEAVY_TEST_ANNOTATION_FQN = "org.jetbrains.kotlin.test.HeavyTest"
+
+fun PsiNameIdentifierOwner.isHeavyTest(): Boolean {
+    return this.parentsOfType<PsiClass>(withSelf = true)
+        .any { it.hasAnnotation(HEAVY_TEST_ANNOTATION_FQN) }
+}