org.mockito.scalatest.MockitoSessionFixture.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mockito-scala-scalatest_2.12 Show documentation

mockito-scala-scalatest

The newest version!

package org.mockito.scalatest

import org.mockito.{ MockitoScalaSession, Strictness }
import org.scalatest._

private[mockito] trait MockitoSessionFixture extends TestSuite { this: Suite =>

  val strictness: Strictness = Strictness.StrictStubs

  abstract override protected def withFixture(test: NoArgTest): Outcome = {
    val session = MockitoScalaSession(name = s"${test.name} - session", strictness)

    val result =
      try super.withFixture(test)
      catch {
        case t: Throwable =>
          session.finishMocking(Some(t))
          throw t
      }

    session.finishMocking(result.toOption)
    result
  }
}

private[mockito] trait MockitoSessionAsyncFixture extends AsyncTestSuite { this: Suite =>

  val strictness: Strictness = Strictness.StrictStubs

  abstract override def withFixture(test: NoArgAsyncTest): FutureOutcome = {
    val session = MockitoScalaSession(name = s"${test.name} - session", strictness)

    val result =
      try super.withFixture(test)
      catch {
        case t: Throwable =>
          session.finishMocking(Some(t))
          throw t
      }

    result.onOutcomeThen(o => session.finishMocking(o.toOption))
  }
}