如何使用 Mockito 模拟 URLConnection 类

编程问答 2022-05-15

问题描述

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

实际方法

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

当我运行此测试并通过它调用实际方法时，URL url 会占用 http://localhost:3000/autocomplete 而不是 Mocked URL 类。然后 uc 也不会占用模拟值并打开连接，即使我也在此处提供了模拟：URLConnection uc = powermockito.mock(URLConnection.class); powermockito.doReturn(uc).when(url).openConnection();。然后 String sb = toStringFromStream(uc.getInputStream()); 这行从实际类执行，我不知道为什么它总是转到异常块。任何人都请提出解决此问题的建议！

解决方法

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

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

如果您仍想确认 I/O 是否有效，请为实际连接到 URL 的那个编写集成测试。

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

httpconnectionjavajavamockitomockitomockitounit-testingurlconnection