From f151cf3173d8dc7f4b4f18d111828ddbc44fb018 Mon Sep 17 00:00:00 2001 From: Igal Tabachnik Date: Sat, 24 Feb 2024 16:44:55 +0200 Subject: [PATCH] Consider additional forms of MutableRunnableSpec (for custom implementations) --- .../scala/zio/intellij/testsupport/ZTestFramework.scala | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/main/scala/zio/intellij/testsupport/ZTestFramework.scala b/src/main/scala/zio/intellij/testsupport/ZTestFramework.scala index 8681dc4..4468a25 100644 --- a/src/main/scala/zio/intellij/testsupport/ZTestFramework.scala +++ b/src/main/scala/zio/intellij/testsupport/ZTestFramework.scala @@ -99,13 +99,18 @@ object ZTestFramework { "_root_.zio.test.MutableRunnableSpec.TestBuilder" ) + private[ZTestFramework] val additionalMethodsRegex = Set( + "_root_.zio.test.*.SuiteBuilder", + "_root_.zio.test.*.TestBuilder" + ) + private def expandsToTestMethod(tpe: ScType) = tpe.extractClass.collect { case c: ScClassImpl => val qname = c.qualifiedName val canonical = (if (qname == null || qname == c.name) c.name else "_root_." + qname) + c.typeParamString - testMethodTypes.contains(canonical) + testMethodTypes.contains(canonical) || additionalMethodsRegex.exists(canonical.matches) }.getOrElse(false) }