pact4s.ziotest.RequestResponsePactForger.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of pact4s-zio-test_3 Show documentation
pact4s-zio-test
There is a newer version: 0.15.2
/*
 * Copyright 2021 io.github.jbwheatley
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package pact4s.ziotest

import au.com.dius.pact.consumer.BaseMockServer
import au.com.dius.pact.core.model.RequestResponseInteraction
import pact4s.RequestResponsePactForgerResources
import zio.test.{Spec, TestAspect, TestEnvironment, ZIOSpecDefault, assertTrue}
import zio.{Scope, UIO, ZIO, ZLayer}

import java.util.concurrent.atomic.AtomicBoolean

trait RequestResponsePactForger extends ZIOSpecDefault with RequestResponsePactForgerResources {

  def specName: String = s"Pact: ${pact.getConsumer.getName} - ${pact.getProvider.getName}"

  private val allTestsSucceeded: AtomicBoolean = new AtomicBoolean(false)

  def verify(interaction: RequestResponseInteraction): Spec[BaseMockServer, Nothing] =
    interaction.getDescription match {
      case description => test(s"Missing verification for message: '$description'")(assertTrue(false))
    }

  def mockServer: ZLayer[Scope, Throwable, BaseMockServer] = ZLayer.fromZIO {
    ZIO.acquireRelease(
      for {
        mockServer <- ZIO.attempt(createServer)
        _          <- ZIO.attempt(mockServer.start())
      } yield mockServer
    )(mockServer =>
      for {
        _ <- ZIO.attempt(mockServer.stop()).catchAll(e => ZIO.logError(s"failed to stop mock server: $e"))
        _ <-
          (beforeWritePacts() *> ZIO
            .attempt(verifyResultAndWritePactFiles(mockServer))
            .catchAll(e => ZIO.logError(s"failed to write pacts: $e"))).when(allTestsSucceeded.get())
      } yield ()
    )
  }

  def tests: Seq[Spec[BaseMockServer, Nothing]] = interactions.map(verify)

  override def spec: Spec[TestEnvironment with Scope, Any] =
    (suite(specName)(tests) @@ TestAspect.afterAllSuccess(ZIO.succeed(allTestsSucceeded.set(true))))
      .provideSomeLayerShared[TestEnvironment with Scope](mockServer)

  override private[pact4s] type Effect[A] = ZIO[Any, Throwable, A]

  override def beforeWritePacts(): UIO[Unit] = ZIO.unit
}