在类的构造函数中模拟方法

Question

我是莫基托的新手。假设我有一节这样的课

public class MyClass {
    int a;
    String b;
    public MyClass(int a) {
        this.a = a;
        this.b = draw();
        System.out.println("class is created");
    }

    public String draw() {
        System.out.println("my");
        return "b";
    }

    public void apple() {
        System.out.println("apple");
    }
}

我正在使用Mockito编写一个JUnit测试，我正在使用构造函数创建类的对象。当我安装类时，是否可以模拟Drew（）方法？

Answer 1

用Mockito测试你的课堂的一种方法是，

import org.junit.Before;
import org.junit.Test;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;

import static org.junit.Assert.assertEquals;
import static org.mockito.Mockito.when;

public class TestMyClass {

    @Mock
    private MyClass clazz;

    @Before
    public void setUp() {
        MockitoAnnotations.initMocks(this);
    }

    @Test
    public void testDraw() {
        when(clazz.draw()).thenReturn("My mock hello");
        assertEquals("My mock hello", clazz.draw());
    }
}

Answer 2

你可以使用间谍进行部分模拟（参见此处了解如何实现）。

但和往常一样：当人们开始考虑使用模拟框架的复杂方法时，真正的答案是：后退一步，改进生产代码。构造器应该只做简单的初始化操作。

在你的例子中，一个合理的方法是：不要调用内部方法来计算该字段的值，而是将该值传递给构造函数（换句话说，使用依赖注入）。

你仍然可以这样做：

public MyClass(int a) {
  this(a, draw());
}

MyClass(int a, String b) { 
  this.a = a; ...

现在，您的单元测试可以很高兴地使用接受两个参数的ctor——您对模拟任何东西的需求完全消失了。

Answer 3

不清楚为什么需要这里的Mockito。这是一个没有Mockito的解决方案。

在您的测试中创建一个MyClass的实例，其中包含重写的Draw（）方法：

final MyClass myClass = new MyClass() {
    @Override
    public String draw() {
        return "mock data";
    }
}

// Now test your class as you want