我有以下类:
class LinkUserService() {
//** cake pattern **
oauthProvider: OAuthProvider =>
//******************
def isUserLinked(userId: String, service: String) = {
val cred = oauthProvider.loadCredential(userId)
cred != null
}
def linkUserAccount(userId: String, service: String): (String, Option[String]) = {
if (isUserLinked(userId, service)) {
("SERVICE_LINKED", None)
} else {
val authUrl = oauthProvider.newAuthorizationUrl
("SERVICE_NOT_LINKED", Some(authUrl))
}
}
def setLinkAuthToken(userId: String, service:String, token:String):String = {
oauthProvider.createAndStoreCredential(userId, token)
}
}
通常我会在生产中使用此类,如下所示:
val linkService = LinkUserService with GoogleOAuthProvider
在测试方面,我想用一个模拟替换oauthProvider
,这样我的单元测试已经训练了这样的模拟,可以这样响应:oauthProvider.loadCredential("nobody") returns null
。这可能吗? 如果是这样,我将如何设置我的单元测试来执行此操作?
您遇到此问题是因为您没有完全使用蛋糕图案。如果你写类似的东西
trait LinkUserServiceComponent {
this: OAuthProviderComponent =>
val linkUserService = new LinkUserService
class LinkUserService {
// use oauthProvider explicitly
...
}
}
trait GoogleOAuthProviderComponent {
val oauthProvider = new GoogleOAuthProvider
class GoogleOAuthProvider {
...
}
}
然后你使用这样的模拟:
val combinedComponent = new LinkUserServiceComponent with OAuthProviderComponent {
override val oauthProvider = mock(...)
}
然后你的问题就消失了。如果您还制作了这样的通用接口特征(并使其他组件依赖于接口,而不是实现):
trait OAuthProviderComponent {
def oauthProvider: OAuthProvider
trait OAuthProvider {
// Interface declaration
}
}
然后,您还将拥有通用的可重用和可测试的代码。
这与您的建议非常相似,确实是蛋糕图案的本质。
我能想出的唯一解决方案是一种委托模拟特征,如下所示:
trait MockOAuthProvider extends OAuthProvider {
val mockProvider = mock[OAuthProvider]
def loadCredential(userId: String) = mockProvider.loadCredential(userId)
def newAuthorizationUrl() = mockProvider.newAuthorizationUrl
def createAndStoreCredential(userId: String, authToken: String) = mockProvider.createAndStoreCredential(userId, authToken)
}
我把它放在我的规范的顶部,然后当我声明我的 LinkUserService 时,我会像这样混合这个模拟特征:
val linkUserService = new LinkUserService() with MockOAuthProvider
val mockOAuth = linkUserService.mockProvider
最后,在我的单元测试中,我做了这样的事情:
mockOAuth.loadCredential("nobody") returns null
它有效,但是如果我的特质更大,我可以看到成为PITA