Mockito旁路静态测试方法

java junit mockito

2022-09-03 05:38:49

我需要使用Mockito测试handErIn（）方法。

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

请注意，在测试环境中，此 Util.getContextPDO 始终返回 Exception，我打算通过始终返回虚拟 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（）的返回值。以下是我使用间谍的初步努力（）

@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（））似乎调用了真正的方法。

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

答案 1

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

假设你做了这个界面：

interface IPDOFacade {

    IPDO getContextPDO();
}

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

class IPDOFacadeImpl implements IPDOFacade {

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

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

public class MyClass {

    private final IPDOFacade ipdoFacade;

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

    private IPDO getIPDO() {
        return ipdoFacade.getContextPDO();
    }

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

        someImportantBusinessLogic(pdo);

        return result;
    }
    
    ...

}

在单元测试中，您可以轻松模拟自己的接口，以您喜欢的任何方式对其进行存根，并将其注入到所测试的单元中。

这

避免了将私有方法包设为私有的需要。
通过避免部分模拟，使您的测试更具可读性。
应用控制反转。
将应用程序与特定的第三方库分离。

答案 2

将我的测试更改为：

@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)));

}

阅读《有效莫基托》后解决。