Question

当我尝试执行一些JUnit mockito测试时遇到了这个问题。

为了便于理解我的问题，我将在下面提供一个例子：

Class A {
   public String test(String para1) {
       //...do whatever stuff
       return para1;
   }
}

Class B {
   public void run() {
       A a = new A();
       String result = a.test("test");
       System.out.println(result);
   }
}

when(mockA.test(anyString()).thenReturn("mockResult");

A mockA = mock(A.class);

//Instead of doing mockA.test(), I do the following:
B b = new B();
b.run();

问题是，如何用“mockA”对象替换B的run（）方法中的“a”对象？这样我可以从b.run（）开始执行代码并且还利用代码执行过程中的模拟对象。

任何帮助将不胜感激！：P

Answer 1

有几种选择，而不是在run内创建A的新实例：

在构造函数中传递A的实例，如

class B {
    private A a;
    B(A a) {
        this.a = a;
    }
    void run() {
        a.test("something");
    }
}

因此，您的测试代码将更改为

B b = new B(mockA);
b.run();

创建setter方法：

class B {
    private A a;
    void setA(A a) {
        this.a = a;
    }
    void run() {
        a.test("something");
    }
}

因此，您的测试代码将更改为

B b = new B();
b.setA(mockA);
b.run();

通常第二种方法是首选。

如何使用mock对象替换内部代码段中的原始对象？

1 个答案: