14

我需要使用 Mockito 测试 handleIn() 方法。

然而，代码需要调用这个遗留代码 Util.getContextPDO 这是一个静态方法。

请注意，在测试环境中，此 Util.getContextPDO 始终返回异常，我打算通过始终返回虚拟 IPDO 来绕过此 Util.getContextPDO()。

public class MyClass {
  public IPDO getIPDO() 
  {
    return Util.getContextPDO(); // note that Util.getContextPDO() is a static, not mockable.
  }

  public String handleIn(Object input) throws Throwable 
  {
    String result = "";
    IPDO pdo = getIPDO();

    // some important business logic.

    return result;
  } 
}

最初我认为这可以通过使用“MyClass”类的 spy() 来实现，所以我可以模拟 getIPDO() 的返回值。下面是我使用 spy () 的初步尝试

@Test
public void testHandleIn() throws Exception
{
    IPDO pdo = new PDODummy();


    MyClass handler = new MyClass ();
    MyClass handler2 = spy(handler);

    when(handler2.getIPDO()).thenReturn(pdo);
    PDOUtil.setPDO(pdo, LogicalFieldEnum.P_TX_CTGY, "test123");
    IPDO pdoNew = handler2.getIPDO();

    Assert.assertEquals("test123,(PDOUtil.getValueAsString(pdoNew, LogicalFieldEnum.P_TX_CTGY)));

}

然而when(handler2.getIPDO()).thenReturn(pdo); 正在抛出我想避免的异常（因为 handler2.getIPDO() ）似乎调用了真正的方法。

关于如何测试这部分代码的任何想法？

4

3 回答 3

12

将我的测试更改为：

@Test
public void testHandleIn() throws Exception
{
  IPDO pdo = new PDODummy();


  MyClass handler = new MyClass ();
  MyClass handler2 = spy(handler);

  doReturn(pdo ).when( handler2 ).getIPDO();
  PDOUtil.setPDO(pdo, LogicalFieldEnum.P_TX_CTGY, "test123");
  IPDO pdoNew = handler2.getIPDO();

  Assert.assertEquals("test123,(PDOUtil.getValueAsString(pdoNew, LogicalFieldEnum.P_TX_CTGY)));

}

阅读Effective Mockito后解决。

于 2012-12-28T06:41:58.530 回答
12

摆脱对 3rd 方 API 的静态调用的一个好技术是将静态调用隐藏在接口后面。

假设您制作了这个界面：

interface IPDOFacade {

    IPDO getContextPDO();
}

并有一个默认实现，只需调用 3rd 方 API 上的静态方法：

class IPDOFacadeImpl implements IPDOFacade {

    @Override
    public IPDO getContextPDO() {
        return Util.getContextPDO();
    }
}

然后只需将接口的依赖项注入MyClass并使用接口，而不是直接使用第 3 方 API：

public class MyClass {

    private final IPDOFacade ipdoFacade;

    public MyClass(IPDOFacade ipdoFacade) {
        this.ipdoFacade = ipdoFacade;
    }

    public String handleIn(Object input) throws Throwable
    {
        String result = "";
        IPDO pdo = getIPDO();

        someImportantBusinessLogic(pdo);

        return result;
    }

    ...

}

在您的单元测试中，您可以轻松地模拟您自己的界面，以任何您喜欢的方式存根并将其注入到被测单元中。

这

  • 避免了将私有方法包私有化的需要。
  • 通过避免部分模拟使您的测试更具可读性。
  • 应用控制反转。
  • 将您的应用程序与特定的 3rd 方库分离。
于 2012-12-28T10:04:50.820 回答
1 
when(handler2.getIPDO()).thenReturn(pdo);

将实际调用该方法，然后pdo无论如何都返回。

然而：

doReturn(pdo).when(handler2).getIPDO();

将返回 pdo 而不调用该getIPDO()方法。

于 2015-05-15T01:45:21.417 回答