使用Mockito模拟内部方法

Question

我是莫基托的新手。我试图通过模拟数据库交互为服务编写jnuit：

我有下面的类(只是实际类的代表)

public class TestService{

    public Response getTestList(String type){
        list = getListbyType(type);

        return response.entity(list);
    }

    private List getListbyType(String type){
          ...
          ..
          TestDAO testdao = new Testdao(sqlconn);
          list = testdao.getListfromDB(type)
          return list;
   }
}

我的测试课就像是

public class TestServiceTest{

    @InjectMocks
    private TestService testService = new TestService();
    @Test
    public void getTestListTest(){

        List testlist = new List();
        tetslist.add("test1");

        TestDAO testdaomock = mock(TestDAO);
        when(testdaomock.getListfromDB("test")).thenreturn(list);

        list = testService.getTestList(test);
    }
}

但是当我运行这个测试时，它仍然调用实际的db调用并从DB中检索值，而不是模拟值，我应该模拟sql连接和非默认构造函数吗？我一无所知。

-- 更新

正如人们所建议的那样，我将DAO实例化移动到我的服务构造函数中，并且还使用了Spy，但是我仍然调用了实际的DB调用而不是模拟调用。

Answer 1

如果您不能更改TestService类(遗留代码)，可以用PowerMockito(https://github . com/jayway/powermock)模拟新实例。它使用自己的junit runner (PowerMockRunner ),允许字节码操作。

以下是您的代码示例：

import static org.mockito.Matchers.anyString;
import static org.mockito.Mockito.when;
import org.mockito.Mock;
import org.powermock.api.mockito.PowerMockito;
import org.powermock.modules.junit4.PowerMockRunner;

@RunWith(PowerMockRunner.class)
public class TestServiceTest {

   @Mock
   TestDAO testDaoMock;

   @Test
   public void test() throws Exception {
    List<String> testlist = new ArrayList<>();
    testlist.add("test1");
    when(testDaoMock.getListfromDB(anyString())).thenReturn(testlist);
    PowerMockito.whenNew(TestDAO.class).withAnyArguments().thenReturn(testDaoMock);
    TestService testService = new TestService();
    Response actualResponse = testService.getTestList("testType");
    assertEquals(expectedResponse, actualResponse);
    }
}

您需要将powermock-api-mockito和powermock-module-junit4添加到您的依赖项中。

Answer 2

您的问题在于以下陈述:

TestDAO testdao = new Testdao(sqlconn);

您从 mock（） 获得的测试道实例不是测试道中使用的实例

为了成功模拟，您需要始终应用依赖关系的反转。这意味着new必须仅显示在您不打算测试的类中，例如简单的工厂或Spring配置文件。

[更新]

如果不能选择适当的IOC，您可以引入一种分配DAO的方法，然后spy（）它。该方法必须是包私有的。这在这里描述。

另一个选项是添加一个包私有构造函数，将＜code＞testdao