如何验证方法是否通过mockito从具有相同类的其他方法调用

时间:2012-01-06 08:57:25

标签: java mockito

我想测试一些在同一个类中调用其他人的方法。它们基本上是相同的方法,但具有不同数量的参数,因为数据库中存在一些默认值。我在这个

上展示
public class A{
    Integer quantity;
    Integer price;        

    A(Integer q, Integer v){
        this quantity = q;
        this.price = p;
    }

    public Float getPriceForOne(){
        return price/quantity;
    }

    public Float getPrice(int quantity){
        return getPriceForOne()*quantity;
    }
}

所以我想测试在调用方法getPrice(int)时是否调用了方法getPriceForOne()。基本上正常调用getPrice(int)并模拟getPriceForOne。

import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
....

public class MyTests {
    A mockedA = createMockA();

    @Test
    public void getPriceTest(){
        A a = new A(3,15);
        ... test logic of method without mock ...

        mockedA.getPrice(2);
        verify(mockedA, times(1)).getPriceForOne();
    }
}

请记住,我有一个更复杂的文件,对其他人来说是一个实用程序,它们必须都在一个文件中。

1 个答案:

答案 0 :(得分:57)

你需要一个间谍,而不是模拟A:

    A a = Mockito.spy(new A(1,1));
    a.getPrice(2);
    verify(a, times(1)).getPriceForOne();