• Home
• org.mockito
• mockito-scala_2.13
• 1.17.37
• source code
• ThreadAwareMockHandler.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

org.mockito.internal.handler.ThreadAwareMockHandler.scala Maven / Gradle / Ivy

package org.mockito.internal.handler

import org.mockito.AdditionalAnswers
import org.mockito.invocation.{ Invocation, InvocationContainer, MockHandler }
import org.mockito.mock.MockCreationSettings
import org.scalactic.Prettifier

class ThreadAwareMockHandler[T](settings: MockCreationSettings[T], realImpl: T)(implicit $pt: Prettifier) extends MockHandler[T] {
  private val currentThread    = Thread.currentThread()
  private val mockDelegate     = ScalaMockHandler(settings)
  private val realImplDelegate = AdditionalAnswers.delegatesTo(realImpl)

  override def handle(invocation: Invocation): AnyRef =
    if (Thread.currentThread() == currentThread) mockDelegate.handle(invocation)
    else realImplDelegate.answer(invocation)

  override def getMockSettings: MockCreationSettings[T] = mockDelegate.getMockSettings

  override def getInvocationContainer: InvocationContainer = mockDelegate.getInvocationContainer
}

object ThreadAwareMockHandler {
  def apply[T](settings: MockCreationSettings[T], realImpl: T)(implicit $pt: Prettifier): ThreadAwareMockHandler[T] =
    new ThreadAwareMockHandler(settings, realImpl)($pt)
}