add support for WordSpec

Author: alfonsoristorato

src/main/kotlin/io/kotest/plugin/intellij/styles/WordSpecStyle.kt

@@ -14,6 +14,7 @@ import io.kotest.plugin.intellij.psi.ifMinusOperator
 import io.kotest.plugin.intellij.psi.ifCallExpressionLhsStringOpenQuote
 import io.kotest.plugin.intellij.psi.ifDotExpressionSeparator
 import io.kotest.plugin.intellij.psi.ifOpenQuoteOfLhsArgOfIndexFunction
+import io.kotest.plugin.intellij.psi.isDataTestMethodCall
 import org.jetbrains.kotlin.name.FqName
 import org.jetbrains.kotlin.psi.KtBinaryExpression
 import org.jetbrains.kotlin.psi.KtCallExpression
@@ -32,6 +33,13 @@ object WordSpecStyle : SpecStyle {
       return "\"$name\" should { }"
    }
 
+   override fun getDataTestMethodNames(): Set<String> =
+      setOf(
+         "withData",
+         "withWhens",
+         "withShoulds"
+      )
+
    override fun isTestElement(element: PsiElement): Boolean = test(element) != null
 
    private fun PsiElement.locateParentWhen(): Test? {
@@ -123,7 +131,7 @@ object WordSpecStyle : SpecStyle {
 
    override fun test(element: PsiElement): Test? {
       return when (element) {
-         is KtCallExpression -> element.trySubject()
+         is KtCallExpression -> element.trySubject() ?: element.tryDataTest()
          is KtBinaryExpression -> (element.tryShould() ?: element.tryWhen())
          is KtDotQualifiedExpression -> element.trySubjectWithConfig()
          else -> null
@@ -134,6 +142,21 @@ object WordSpecStyle : SpecStyle {
       return setOf("OPEN_QUOTE")
    }
 
+   /**
+    * For a FunSpec we consider the following scenarios:
+    *
+    * should("test name") { }
+    * xshould("test name") { }
+    * should("test name").config(...) {}
+    * xshould("test name").config(...) {}
+    * when("test name") {}
+    * xwhen("test name") {}
+    * when("test name").config(...) {}
+    * xwhen("test name").config(...) {}
+    * withData(...) { }
+    * withWhens(...) { }
+    * withShoulds(...) { }
+    */
    override fun test(element: LeafPsiElement): Test? {
       val ktcall = element.ifCallExpressionLhsStringOpenQuote()
       if (ktcall != null) return test(ktcall)
@@ -147,6 +170,12 @@ object WordSpecStyle : SpecStyle {
       val ktdot = element.ifDotExpressionSeparator()
       if (ktdot != null) return test(ktdot)
 
+      // try to find Data Test Method by finding lambda openings
+      val dataMethodCall = element.isDataTestMethodCall(ShouldSpecStyle.getDataTestMethodNames())
+      if (dataMethodCall != null) {
+         return ShouldSpecStyle.test(dataMethodCall)
+      }
+
       return null
    }
 }

src/test/kotlin/io/kotest/plugin/intellij/styles/WordSpecStyleTest.kt

@@ -24,19 +24,20 @@ class WordSpecStyleTest : LightJavaCodeInsightFixtureTestCase() {
       )
 
       val gutters = myFixture.findAllGutters()
-      gutters.size shouldBe 10
+      gutters.size shouldBe 11
 
       val expected = listOf(
-         Gutter("Run WordSpecExample", 87, AllIcons.RunConfigurations.TestState.Run_run),
-         Gutter("Run some should context", 137),
-         Gutter("Run some should context test something", 174),
-         Gutter("Run some should context test something with config", 256),
-         Gutter("Run with capital When", 302),
-         Gutter("Run with capital When and capital Should", 335),
-         Gutter("Run with capital When test something", 373),
-         Gutter("Run with capital When test something with config", 428),
-         Gutter("Disabled - with capital When disabled should", 480, AllIcons.RunConfigurations.TestIgnored),
-         Gutter("Disabled - disabled when", 529, AllIcons.RunConfigurations.TestIgnored),
+         Gutter("Run WordSpecExample", 122, AllIcons.RunConfigurations.TestState.Run_run),
+         Gutter("Run some should context", 172),
+         Gutter("Run some should context test something", 209),
+         Gutter("Run some should context test something with config", 291),
+         Gutter("Run with capital When", 337),
+         Gutter("Run with capital When and capital Should", 370),
+         Gutter("Run with capital When test something", 408),
+         Gutter("Run with capital When test something with config", 463),
+         Gutter("Disabled - with capital When disabled should", 515, AllIcons.RunConfigurations.TestIgnored),
+         Gutter("Disabled - disabled when", 564, AllIcons.RunConfigurations.TestIgnored),
+         Gutter("Run All Spec Tests, including data tests", 625),
       )
 
       gutters.size shouldBe expected.size

src/test/resources/wordspec.kt

@@ -1,6 +1,7 @@
 package com.sksamuel.kotest.specs.wordspec
 
 import io.kotest.core.spec.style.WordSpec
+import io.kotest.datatest.withData
 
 class WordSpecExample : WordSpec() {
   init {
@@ -27,5 +28,8 @@ class WordSpecExample : WordSpec() {
 
     }
 
+     withData(1, 2, 3, 4, 5) { value ->
+        // test here
+     }
   }
 }