feat: support benches

Author: xepozz

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

@@ -4,15 +4,18 @@ object TestoClasses {
     const val TEST_OLD = "\\Testo\\Attribute\\Test"
     const val TEST_NEW = "\\Testo\\Application\\Attribute\\Test"
     const val TEST_INLINE_OLD = "\\Testo\\Sample\\TestInline"
+    const val TEST_INLINE_NEW = "\\Testo\\Inline\\TestInline"
+
     const val DATA_PROVIDER_OLD = "\\Testo\\Sample\\DataProvider"
     const val DATA_SET_OLD = "\\Testo\\Sample\\DataSet"
-    const val TEST_INLINE_NEW = "\\Testo\\Inline\\TestInline"
     const val DATA_PROVIDER_NEW = "\\Testo\\Data\\DataProvider"
     const val DATA_SET_NEW = "\\Testo\\Data\\DataSet"
     const val DATA_UNION = "\\Testo\\Data\\DataUnion"
     const val DATA_CROSS = "\\Testo\\Data\\DataCross"
     const val DATA_ZIP = "\\Testo\\Data\\DataZip"
 
+    const val BENCH_WITH = "\\Testo\\Bench\\BenchWith"
+
     const val ASSERT = "\\Testo\\Assert"
     const val ASSERTION_EXCEPTION = "\\Testo\\Assert\\State\\Assertion\\AssertionException"
     const val EXPECT = "\\Testo\\Expect"

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

@@ -5,12 +5,17 @@ import com.intellij.psi.PsiElement
 import com.intellij.psi.PsiFile
 import com.intellij.psi.util.PsiTreeUtil
 import com.jetbrains.php.lang.psi.PhpFile
-import com.jetbrains.php.lang.psi.elements.Function
 import com.jetbrains.php.lang.psi.elements.Method
+import com.jetbrains.php.lang.psi.elements.Function
 import com.jetbrains.php.lang.psi.elements.PhpAttributesOwner
 import com.jetbrains.php.lang.psi.elements.PhpClass
 
-fun PsiElement.isTestoExecutable() = isTestoFunction() || isTestoMethod()
+fun PsiElement.isTestoExecutable() = isTestoFunction() || isTestoMethod() || isTestoBench()
+
+fun PsiElement.isTestoBench() = when(this) {
+    is Method -> hasAnyAttribute(TestoClasses.BENCH_WITH)
+    else -> false
+}
 
 fun PsiElement.isTestoFunction() = when(this) {
     is Function -> hasAnyAttribute(TestoClasses.TEST_OLD,TestoClasses.TEST_NEW, TestoClasses.TEST_INLINE_NEW, TestoClasses.TEST_INLINE_OLD)
@@ -32,12 +37,12 @@ fun PhpAttributesOwner.hasAttribute(fqn: String) = getAttributes(fqn).isNotEmpty
 fun PhpAttributesOwner.hasAnyAttribute(vararg fqn: String) = attributes.any { it.fqn in fqn }
 
 fun PsiElement.isTestoClass() = when (this) {
-    is PhpClass -> TestoTestDescriptor.isTestClassName(name) || ownMethods.any { it.isTestoMethod() }
+    is PhpClass -> TestoTestDescriptor.isTestClassName(name) || ownMethods.any { it.isTestoMethod() || it.isTestoBench() }
     else -> false
 }
 
 fun PsiFile.isTestoFile() = when (this) {
-    is PhpFile -> TestoTestDescriptor.isTestClassName(name.substringBeforeLast(".")) || (isTestoClassFile() || isTestoFunctionFile())
+    is PhpFile -> TestoTestDescriptor.isTestClassName(name.substringBeforeLast(".")) || (isTestoClassFile() || isTestoFunctionFile() || isTestBenchFile())
     else -> false
 }
 
@@ -47,6 +52,9 @@ fun PhpFile.isTestoClassFile() = PsiTreeUtil.findChildrenOfType(this, PhpClass::
 fun PhpFile.isTestoFunctionFile() = PsiTreeUtil.findChildrenOfType(this, Function::class.java)
     .any { it.isTestoFunction() }
 
+fun PhpFile.isTestBenchFile() = PsiTreeUtil.findChildrenOfType(this, Function::class.java)
+    .any { it.isTestoBench() }
+
 fun <T> Sequence<T>.takeWhileInclusive(predicate: (T) -> Boolean) = sequence {
     with(iterator()) {
         while (hasNext()) {

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

@@ -5,6 +5,7 @@ import com.github.xepozz.testo.index.TestoDataProviderUtils
 import com.github.xepozz.testo.isTestoClass
 import com.github.xepozz.testo.isTestoDataProviderLike
 import com.github.xepozz.testo.isTestoExecutable
+import com.github.xepozz.testo.tests.TestoTestRunLineMarkerProvider.Companion.getLocationHint
 import com.github.xepozz.testo.util.PsiUtil
 import com.intellij.execution.lineMarker.ExecutorAction
 import com.intellij.execution.lineMarker.RunLineMarkerContributor
@@ -18,8 +19,8 @@ import com.jetbrains.php.config.commandLine.PhpCommandLinePathProcessor
 import com.jetbrains.php.lang.lexer.PhpTokenTypes
 import com.jetbrains.php.lang.psi.PhpPsiUtil
 import com.jetbrains.php.lang.psi.elements.ClassReference
-import com.jetbrains.php.lang.psi.elements.Function
 import com.jetbrains.php.lang.psi.elements.Method
+import com.jetbrains.php.lang.psi.elements.Function
 import com.jetbrains.php.lang.psi.elements.PhpAttribute
 import com.jetbrains.php.lang.psi.elements.PhpAttributesOwner
 import com.jetbrains.php.lang.psi.elements.PhpClass
@@ -95,6 +96,7 @@ class TestoTestRunLineMarkerProvider : RunLineMarkerContributor() {
             TestoClasses.DATA_ZIP,
             TestoClasses.TEST_INLINE_OLD,
             TestoClasses.TEST_INLINE_NEW,
+            TestoClasses.BENCH_WITH,
         )
 
         fun getLocationHint(element: Function) = when (element) {

src/main/kotlin/com/github/xepozz/testo/util/PsiUtil.kt

@@ -19,6 +19,7 @@ object PsiUtil {
         TestoClasses.DATA_UNION,
         TestoClasses.DATA_CROSS,
         TestoClasses.DATA_ZIP,
+        TestoClasses.BENCH_WITH,
     )
 
     fun getAttributeOrder(attribute: PhpAttribute, owner: PhpAttributesOwner): Int = owner