如何在单元测试中模拟缺乏网络连接

Posted 2023-02-16

【中文标题】如何在单元测试中模拟缺乏网络连接

【英文标题】：how to simulate lack of network connectivity in unit testing

【发布时间】：2015-04-08 10:32:24

【问题描述】：

一般问题是如何模拟（作为JUnit 测试用例套件的一部分）缺乏网络连接，因为在某些测试用例中这是一个重要的考虑因素。有没有办法通过 Java API（或通过 JVM 选项）这样做，以便某些测试用例可以在网络断开连接的情况下运行？ （模拟的还是真实的？）。

更具体的情况（如果没有普遍适用的解决方案）是我正在进行大量 XML 文件处理（包括 XSD 验证），并且我需要确保不会通过网络获取任何内容，具体而言，即xsi:schemaLocation 属性值（提示）未使用，所有 XSD 实际上都是从类路径中获取的。我正在使用带有自定义 LSResourceResolver 的 Validator，它可以从类路径加载任何可能需要的 XSD。

我想我可以实现LSResourceResolver 的resolveResource 方法，从而永远不会返回null（因此，永远不会依赖打开到资源的常规URI 连接的默认行为）但我不确定这是否足够，无论如何，如果我可以在模拟孤岛模式下运行我的JUnit 测试（无需手动关闭我机器上的接口），我会更有信心。

更新

接受的答案，即-DsocksProxyHost 方法正好提供了我需要的解决方案，因为JUnit task 可以接受VM 参数（如果fork 设置为true），所以我可以在我的Ant 中有以下内容文件：

<junit printsummary="true" showoutput="true" fork="true" maxmemory="256m">
    <jvmarg value="-DsocksProxyHost=127.0.0.1"/>
    ...

...包裹在contrib:if 中，因此我可以从命令行控制是否在网络连接条件下运行JUnit 测试。

【问题讨论】：

如果您使用 JVM 选项，那将是一个集成测试。对于正确的单元测试，您的网络连接将是一个可以模拟的独立依赖项。听起来您对 java 网络 api 有很强的依赖关系。

取决于您的代码发生了什么，但如果这不合适，我会模拟 socket() 调用（请参阅***.com/questions/3202675/… 进行讨论）或在单独的 VM 中运行测试 -例如，如果您使用的是一个完全独立的微服务，它不会通过在被测代码中模拟来改变

【参考方案1】：

我不建议这样做，但是... 您可以尝试使用安全管理器。默认情况下 JVM 不使用安全管理器，通过设置 -Djava.security.manager 您将激活 默认安全管理器。 AFAIR，安全管理器的默认行为是阻止任何连接（只要权限未在安全策略中明确授予）。但。每次连接被阻止时，您都会收到java.net.NetPermission Exception。您还可能会遇到本地文件访问、反射调用等问题。

另一种可能性是将网络代理设置为不存在的地址，例如-DsocksProxyHost=127.0.0.1。然后任何 TCP 套接字都会尝试使用 SOCKS 服务器并失败。

【参考方案2】：

Sniffy 允许您在 JUnit 测试中使用 block outgoing network connections - 每当您尝试建立新连接时，它都会抛出 ConnectException。

@Rule public SniffyRule sniffyRule = new SniffyRule();

@Test
@DisableSockets
public void testDisableSockets() throws IOException 
    try 
        new Socket("google.com", 22);
        fail("Sniffy should have thrown ConnectException");
     catch (ConnectException e) 
        assertNotNull(e);
    

它还允许您测试应用程序在runtime 中的连接中断时的行为。只需将-javaagent:sniffy.jar=5559 添加到您的JVM 参数并将您的浏览器指向localhost:5559 - 它会打开一个网页，其中包含所有发现的与下游系统的连接以及禁用某些连接的控件。

免责声明：我是 Sniffy 的作者