方法被嘲笑但值未正确返回

Question

我正在尝试对我的班级进行单元测试 这看起来像：-

   class A{
       private void method1(){
          B b=new B();
          C c=b.method2();
          if(c!=null){
             ...
             ...
          }
       }
}

我想在 B 类中模拟“method2()”。 我知道我们需要一个 B() 的模拟对象，这样每当我们调用它的方法时，就会发生模拟。 这是我尝试过的

A aMock=Easymock.createNiceMock(A.class);
B bMock=Easymock.createNiceMock(B.class);
Powermock.expectNew(B.class).andReturn(bMock).anyTimes();
Easymock.expect(bMock.method2()).andReturn(new C()).anyTimes();
//have done all the replays and verifies

并使用aMock.method1() 调用它， 现在主要问题是 method2 被嘲笑（即 method2() 的主体没有被执行）但我无法接收 C 的对象作为对此的响应。

我的测试场景是：-

我想测试 A 类的 method1() ，它又调用 B 类的 method2() ，但我不希望它执行它的代码（method2() ），因为它是一个 DB 调用，我也需要method2() 返回的 C 对象，所以我使用模拟来实现这一点。

Answer 1

如果您想测试 A 类的 method1()，那么您不应该创建 A 的模拟。这意味着你不应该这样做A aMock=Easymock.createNiceMock(A.class);

对于您的测试场景，测试将类似于：

import org.easymock.EasyMock;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.powermock.api.easymock.PowerMock;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;

@RunWith(PowerMockRunner.class)
@PrepareForTest(A.class)
public class ATest {

    A testableA = new A();

    @Test
    public void testMethod1() throws Exception {
        B bMock = EasyMock.createNiceMock(B.class);
        PowerMock.expectNew(B.class).andReturn(bMock).anyTimes();
        EasyMock.expect(bMock.method2()).andReturn(new C()).anyTimes();
        PowerMock.replay(B.class);
        EasyMock.replay(bMock);

        testableA.method1();

        //assert and validate your scenario here
    }
}

您可以关注文档here。