Mockito绕过静态方法进行测试 [英] Mockito bypass static method for testing

问题描述

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

I need to test handleIn() method using Mockito.

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

However the code need to call this legacy code Util.getContextPDO which is a static method.

请注意，在测试环境中，此Util.getContextPDO始终返回Exception，并且我打算通过始终返回虚拟IPDO来绕过此Util.getContextPDO（）。

Note that in testing environment this Util.getContextPDO is always returns Exception, and I intend to bypass this Util.getContextPDO() by always return a dummy IPDO.

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;
  } 
}

最初我认为这可以通过使用spy（）来实现类MyClass，所以我可以模拟getIPDO（）的返回值。以下是我最初使用spy（）的努力

Initially I thought this achieveable by using spy() of the class "MyClass", so I can mock the return value of getIPDO(). Below is my initial effort using 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（））似乎调用了真正的方法。

However the when(handler2.getIPDO()).thenReturn(pdo); is throwing the Exception that I want to avoid ( because handler2.getIPDO() ) seems to call the real method.

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

Any idea on how to test this part of code?

推荐答案

将我的测试更改为：

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

}

阅读有效的Mockito 。

这篇关于Mockito绕过静态方法进行测试的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！