harness.zio.test.ContractHarnessSpec.scala Maven / Gradle / Ivy

Go to download
package harness.zio.test

import cats.syntax.option.*
import zio.*
import zio.test.*

abstract class ContractHarnessSpec[_R: EnvironmentTag](implName: String, contract: Contract[_R]) extends HarnessSpec[_R] {

  private def modifyCase(spec: TestSpec): Option[TestSpec] =
    spec.caseValue match {
      case Spec.ExecCase(exec, spec)     => modifyCase(spec).map(s => Spec(Spec.ExecCase(exec, s)))
      case Spec.LabeledCase(label, spec) => Spec(Spec.LabeledCase(s"$label ($implName)", spec)).some
      case _                             => None
    }

  override def testSpec: TestSpec = {
    val built = suite(implName)(contract.testSpec)
    modifyCase(built).getOrElse { suite(implName)(built) }
  }

}