如何使用 Mockito 模拟 URLConnection class

Question

我正在尝试模拟一种方法，该方法使用 URLConnection class 打开连接，然后获取 stream 输入。 我想有效地模拟 URLConnection class 但我不能这样做。 以下是我的代码：

实际方法

@GetMapping(value = "/search/autocomplete")
    public String autocomplete() {
        try {
            URL url = new URL("http://" + serverHost + ":" + serverPort + "/autocomplete");
            URLConnection uc = url.openConnection();
            String sb = toStringFromStream(uc.getInputStream());
            return sb;
        } catch (Exception e) {
            log.warn(e.getMessage());
        }
        return "";
    }

测试方法：

@Test
    public void autocomplete() throws Exception {
        URL url = PowerMockito.mock(URL.class);
        PowerMockito.whenNew(URL.class).withArguments("http://" + serverHost + ":" + serverPort + 
        "/autocomplete").thenReturn(url);
        ReflectionTestUtils.setField(searchController,"serverHost","localhost");
        ReflectionTestUtils.setField(searchController,"serverPort",3000);
        InputStream anyInputStream = new ByteArrayInputStream("molly\nted".getBytes());
        URLConnection uc = PowerMockito.mock(URLConnection.class);
        PowerMockito.doReturn(uc).when(url).openConnection();
        Mockito.doReturn(anyInputStream).when(uc).getInputStream();
        PowerMockito.mock(Streamutils.class);
        PowerMockito.when(Streamutils.toStringFromStream(anyInputStream)).thenReturn("molly");
        String string = searchController.autocomplete();
        assertEquals("",string);
    }

When I run this test and call the actual method through it, URL url takes up the http://localhost:3000/autocomplete instead of the Mocked URL class. 然后uc也不会占用模拟值并打开连接，即使我在这里也提供了 mocking 给它： URLConnection uc = PowerMockito.mock(URLConnection.class); PowerMockito.doReturn(uc).when(url).openConnection(); URLConnection uc = PowerMockito.mock(URLConnection.class); PowerMockito.doReturn(uc).when(url).openConnection(); String sb = toStringFromStream(uc.getInputStream()); 此行从实际的 class 执行，我不知道为什么它总是进入异常块。 请任何人建议解决此问题！

Answer 1

不是您要求的答案，而是一些避免问题的建议。

我建议不要将低级 I/O 操作作为单元测试的一部分。 将连接逻辑拆分为依赖项，并将单元测试集中在您自己的业务逻辑上。 如果您为这样的低级内容编写单元测试，那么您主要是在测试框架代码，而不是您自己的代码。

如果您仍想确认 I/O 工作正常，请为实际连接到 URL 的 I/O 编写集成测试。

编辑：有关集成测试的有价值的方法，请参阅https://docs.pact.io/