如何解决scalatest - 使用 fallbackTo 测试 Future[S] 的方法
前提:当我的 API 响应对 User 对象的请求时,我想尝试使用 case class PartnerView(id: String,vipStatus: Option[Boolean],latestSession: Option[Timestamp]
的属性来丰富它。由于数据库有时可能不可靠,因此我使用 fallbackTo
提供可选值,因此不会在用户 JSON 响应中显示它们。
到目前为止,以下实现似乎有效(通过 Postman 运行请求返回没有可选值的用户 JSON),但我的单元测试会抱怨,好像我有一个未捕获的异常。
服务类:
class Service(repo: Repository) {
def get(id: String): Future[Partner] = {
val account = repo.getAccount(id)
val getLatestSession = repo.getLatestSession(id)
val partnerView = (for {
account <- getAccount
latestStartTime <- getLatestSession.map {
case Some(x) => x.scheduledStartTime
case _ => None
}
} yield PartnerView(partnerId,account.vipStatus,latestStartTime))
.fallbackTo(Future.successful(PartnerView(id,None,None)))
partnerView
}
}
存储库类:
class Repository(database: DatabaseDef,logger: LoggingAdapter) {
def getAccount(id: String): Future[Account] = database.run((...).result.head)
.recover {
case e: Exception =>
logger.error(e,"DB Server went down")
throw e
}
def getLatestSession(id: String): Future[Option[Session]] = database.run((...).result.headOption)
.recover {
case e: Exception =>
logger.error(e,"DB Server went down")
throw e
}
}
单元测试:
class ServiceSpec extends AsyncFlatSpec with AsyncmockFactory with OneInstancePerTest {
val mockRepo = mock[Repository]
val service = new Service(mockRepo)
behavIoUr of "Service"
it should "get an empty PartnerView when the repository get an Exception" in {
(mockRepository.getAccount _)
.expects("partner")
.throwing(new Exception)
service.get("partner")
.map(partnerView => assert(partnerView.id == "partner" && partnerView.vipStatus.isEmpty))
}
}
测试将失败并显示消息
Testing started at 5:15 p.m. ...
java.lang.Exception was thrown.
{stacktrace here}
我期待 Exception
解决方法
通过将模拟设置更改为以下,测试成功运行:
it should "get an empty PartnerView when the repository get an Exception" in {
(mockRepository.getAccount _)
.expects("partner")
.returning(Future.failed(new Exception))
...
}
因为 recover
方法将 Exception
包裹在 Future
中
来源:
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。