feat: support test classes

Author: xepozz

src/main/kotlin/com/github/xepozz/testo/TestoTestRunLineMarkerProvider.kt

@@ -24,7 +24,15 @@ class TestoTestRunLineMarkerProvider : RunLineMarkerContributor() {
         if (element.nameIdentifier != leaf) return null
 
         return when {
-            element is Function && element.isTesto() -> withExecutorActions(
+            element is Function && element.isTestoMethod() -> withExecutorActions(
+                getTestStateIcon(
+                    getLocationHint(element),
+                    element.project,
+                    false,
+                ),
+            )
+
+            element is PhpClass && element.isTestoClass() -> withExecutorActions(
                 getTestStateIcon(
                     getLocationHint(element),
                     element.project,
@@ -47,13 +55,13 @@ class TestoTestRunLineMarkerProvider : RunLineMarkerContributor() {
                 RUN_TEST_TOOLTIP_PROVIDER
             )
 
-         fun getLocationHint(element: Function) = when (element) {
+        fun getLocationHint(element: Function) = when (element) {
             is Method -> getLocationHint(element.containingClass!!) + "::" + element.name
             else -> getLocationHint(element.containingFile) + "::" + element.fqn
         }
 
-         fun getLocationHint(element: PhpClass) = getLocationHint(element.containingFile) + "::" + element.fqn
-         fun getLocationHint(file: PsiFile) = "${TestoFrameworkType.SCHEMA}://" + getFilePathDeploymentAware(file)
+        fun getLocationHint(element: PhpClass) = getLocationHint(element.containingFile) + "::" + element.fqn
+        fun getLocationHint(file: PsiFile) = "${TestoFrameworkType.SCHEMA}://" + getFilePathDeploymentAware(file)
 
         fun getFilePathDeploymentAware(psiFile: PsiFile): String {
             val localPath = psiFile.virtualFile.path

src/main/kotlin/com/github/xepozz/testo/mixin.kt

@@ -2,14 +2,22 @@ package com.github.xepozz.testo
 
 import com.intellij.psi.PsiElement
 import com.jetbrains.php.lang.psi.elements.Function
+import com.jetbrains.php.lang.psi.elements.PhpClass
 
-fun PsiElement.isTesto(): Boolean {
+fun PsiElement.isTestoMethod(): Boolean {
     return when (this) {
         is Function -> this.getAttributes(TestoClasses.TEST).isNotEmpty()
         else -> false
     }
 }
 
+fun PsiElement.isTestoClass(): Boolean {
+    return when (this) {
+        is PhpClass -> name.endsWith("Test") || methods.any { it.isTestoMethod() }
+        else -> false
+    }
+}
+
 fun <T> Sequence<T>.takeWhileInclusive(predicate: (T) -> Boolean) = sequence {
     with(iterator()) {
         while (hasNext()) {

src/main/kotlin/com/github/xepozz/testo/references/TestFunctionImplicitUsageProvider.kt

@@ -1,13 +1,16 @@
 package com.github.xepozz.testo.references
 
-import com.github.xepozz.testo.isTesto
+import com.github.xepozz.testo.isTestoClass
+import com.github.xepozz.testo.isTestoMethod
 import com.intellij.codeInsight.daemon.ImplicitUsageProvider
 import com.intellij.psi.PsiElement
 
 class TestFunctionImplicitUsageProvider : ImplicitUsageProvider {
-    override fun isImplicitUsage(element: PsiElement) = element.isTesto()
+    override fun isImplicitUsage(element: PsiElement) = element.isTestoMethod() || element.isTestoClass()
 
     override fun isImplicitRead(element: PsiElement) = false
 
     override fun isImplicitWrite(element: PsiElement) = false
+
+    override fun isClassWithCustomizedInitialization(element: PsiElement) = element.isTestoClass()
 }

src/main/kotlin/com/github/xepozz/testo/tests/run/TestoRunConfigurationProducer.kt

@@ -1,6 +1,7 @@
 package com.github.xepozz.testo.tests.run
 
-import com.github.xepozz.testo.isTesto
+import com.github.xepozz.testo.isTestoClass
+import com.github.xepozz.testo.isTestoMethod
 import com.intellij.openapi.project.Project
 import com.intellij.openapi.util.Condition
 import com.intellij.openapi.vfs.VirtualFile
@@ -12,7 +13,6 @@ import com.intellij.util.asSafely
 import com.jetbrains.php.lang.psi.PhpFile
 import com.jetbrains.php.lang.psi.elements.Function
 import com.jetbrains.php.lang.psi.elements.PhpClass
-import com.jetbrains.php.phpunit.PhpUnitTestDescriptor
 import com.jetbrains.php.testFramework.run.PhpTestConfigurationProducer
 
 class TestoRunConfigurationProducer : PhpTestConfigurationProducer<TestoRunConfiguration>(
@@ -34,15 +34,15 @@ class TestoRunConfigurationProducer : PhpTestConfigurationProducer<TestoRunConfi
     override fun getConfigurationFactory() = TestoRunConfigurationFactory(TestoRunConfigurationType.INSTANCE)
 
     companion object Companion {
-        val METHOD = Condition<PsiElement> { it.isTesto() }
+        val METHOD = Condition<PsiElement> { it.isTestoMethod() }
         private val METHOD_NAMER = { element: PsiElement? -> (element as? Function)?.name }
         private val FILE_TO_SCOPE = { file: PsiFile? ->
             println("file to scope: ${file?.virtualFile?.name}")
             file
                 .asSafely<PhpFile>()
                 ?.let { phpFile ->
                     PsiTreeUtil.findChildrenOfType(phpFile, PhpClass::class.java)
-                        .firstOrNull { PhpUnitTestDescriptor.INSTANCE.isTestClassName(it.name) }
+                        .firstOrNull { it.isTestoClass() }
                 }
         }
     }