如何通过模拟嵌套方法来测试方法?

How to test a method by mocking nested methods?

我正在尝试测试一个 Object.method,其中包含来自 Trait 的一些嵌套方法以及一些计算。必须模拟这些嵌套方法(它们访问数据库,所以我想模拟它们的响应)。

当我调用真正的 Object.method 时,它应该跳过嵌套方法调用并检索我想要的内容。我试过模拟它们,但测试仍在调用它们。

这是我的示例源代码:

trait MyTrait {
def myMethodToMock(a: String): String
}

object MyObject extends MyTrait {
def myParentMethod(a:String) = {
val b = myMethodToMock(a)
val c = a + b
c
}
}

然后在我的测试中:

val myTraitMock = mock[MyTrait]
when(myTraitMock.myMethodToMock(a)).thenReturn(b)

//Then I call the parent method:
assert(MyObject.myParentMethod(a) equals c)

它抛出 NullPointerException,因为它仍在访问 myMethodToMock

你应该使用组合而不是继承,所以你可以注入一个 MyTrait 的实例,它可以是模拟的也可以是真实的

您的代码无法编译,所以我将猜测您在这里实际尝试做的一些事情...

您正在模拟一个方法,然后在一个完全不相关的实例上调用它。难怪它不起作用。

一个好的经验法则(也是最佳实践)是永远不要模拟 class你实际测试的东西。将您想要模拟和测试的所有内容拆分成一个单独的 class。这也被称为单一责任原则(每个组件应该负责单一的事情)。

 trait MyTrait {
    def myMethodToMock(a: String): String
 }
 object MyTrait extends MyTrait {
    def myMethodtoMock(a: String) = ???
 }

 class MyObject(helper: MyTrait = MyTrait) {
    def myParentMethod(a: String) = a + helper.myMethodToMock(a) 
 }
 object MyObject extends MyObject()

现在,您可以像这样编写测试:

val myTraitMock = mock[MyTrait]
when(myTraitMock.myMethodToMock(any)).thenReturn("b")
new MyObject(myTraitMock).myParentMethod("a") shouldBe "ab"
verify(myTraitMock).myMethodToMock("a")

这里的主要区别是您将 mock 传递给对象的构造函数,这样当它调用该方法时,它将是您存根的方法,而不是默认提供的实现 class。