scala - 在 Mockito 单元测试中抛出异常时，Future 的 .recover 不会被调用

Question

以下代码返回一个 Future。

val findUserFuture: Future[Option[User]] = userRepo.findOne(userKeys) 

然后我处理 future

findUserFuture.flatMap {....}
.recover{...}

fineOne返回 Future和 Future结束对 getOneById 的调用

def findOne(userKeys:UserKeys):Future[Option[User]] = {
    Future{
      //val loginInfo:LoginInfo = LoginInfo(userKeys.providerID,userKeys.authProvider)
      val userOption:Option[User] = getOneById(userKeys)
      userOption
    }
  }

我想如果 Future 会调用recover返回者 findOne失败，即抛出异常。所以我通过制作 getOneById 来模拟这一点抛出异常。

when(mockUserRepository.findOne(userKeys)).thenReturn(Future(Some(user)))
      when(mockUserRepository.getOneById(userKeys)).thenThrow(classOf[RuntimeException]) //simulating database error

但单元测试不会抛出异常，测试继续使用值 Future(Some(User)) .

我也尝试从 findOne 抛出异常- when(mockUserRepository.findOne(userKeys)).thenThrow(classOf[RuntimeException])但是测试用例停止了
带有以下两个打印件和 .recover的Future不叫

java.lang.RuntimeException was thrown.
java.lang.RuntimeException

Answer 1

此 findUserFuture: Future[Option[User]]或 userRepo.findOne返回 future ，
因此您需要返回 Future.failed在你的模拟中。
例如。

when(mockUserRepository.findOne(userKeys)).thenReturn(Future(Some(user)))
when(mockUserRepository.getOneById(userKeys)).thenReturn(Future.failed(new RuntimeException("network failure"))

在下面找到完整的工作测试来模拟您的用例:

test("mock future test") {
    case class User(name: String)
    case class UserNotFoundException(name: String) extends Exception
    trait UserRepo {
      def findOne(name: String): Future[Option[User]]
    }
    val name      = "bob"
    val dummyUser = User("dummy")

    val userRepo = mock[UserRepo]
    when(userRepo.findOne(name)).thenReturn(Future.failed(new RuntimeException()))

    val userF = userRepo
      .findOne(name)
      .flatMap {
        case Some(user) ⇒ Future.successful(user)
        case None       ⇒ Future.failed(UserNotFoundException(name))
      }
      .recover {
        case NonFatal(_) ⇒ dummyUser
      }

    userF.futureValue shouldBe dummyUser
  }

更新 *
在仔细查看原始帖子后，我发现您 mock 的方式存在小错误。
试试以下:

when(mockUserRepository.findOne(userKeys)).thenCallRealMethod()
when(mockUserRepository.getOneById(userKeys)).thenThrow(classOf[RuntimeException]) 

通知thenCallRealMethod()在这里，早些时候你在 mock findOne以成功的值返回 future ，这意味着原始方法没有被调用，而原始方法没有被调用 getOneById