如何阻止JUnit测试方法建立出站Socket连接？

下面提供几种可靠的实现方案，涵盖原生Java机制和常用测试工具库：

方案一：使用Java原生SecurityManager（无额外依赖） #

通过自定义SecurityManager拦截所有出站连接请求，一旦检测到就抛出异常。适合不想引入第三方库的场景，注意Java 17及以上版本中SecurityManager已被标记为废弃。

实现步骤： #

1. 定义自定义SecurityManager：

public class NoOutboundSocketSecurityManager extends SecurityManager {
    @Override
    public void checkConnect(String host, int port) {
        // 拒绝所有出站连接
        throw new SecurityException("禁止建立出站Socket连接：" + host + ":" + port);
    }
}

2. 在测试类中切换并恢复SecurityManager：

import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;

public class MyUnitTest {
    private static SecurityManager originalSecurityManager;

    @BeforeClass
    public static void setupSecurityManager() {
        originalSecurityManager = System.getSecurityManager();
        System.setSecurityManager(new NoOutboundSocketSecurityManager());
    }

    @AfterClass
    public static void restoreSecurityManager() {
        System.setSecurityManager(originalSecurityManager);
    }

    @Test(expected = SecurityException.class)
    public void testForbiddenSocketConnection() {
        // 这个方法如果尝试建立Socket连接，会触发SecurityException
        new Socket("example.com", 80);
    }
}

方案二：使用PowerMock拦截Socket构造 #

PowerMock可以直接mock Socket类的构造方法，一旦测试代码尝试创建Socket实例就抛出异常，适合已经在使用PowerMock进行测试的项目。

实现步骤： #

1. 引入PowerMock依赖（以Maven为例）：

<dependency>
    <groupId>org.powermock</groupId>
    <artifactId>powermock-module-junit4</artifactId>
    <version>2.0.9</version>
    <scope>test</scope>
</dependency>
<dependency>
    <groupId>org.powermock</groupId>
    <artifactId>powermock-api-mockito2</artifactId>
    <version>2.0.9</version>
    <scope>test</scope>
</dependency>

2. 编写测试类：

import org.junit.Test;
import org.junit.runner.RunWith;
import org.powermock.api.mockito.PowerMockito;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;

import java.net.Socket;

@RunWith(PowerMockRunner.class)
@PrepareForTest(Socket.class)
public class MyUnitTest {

    @Test(expected = RuntimeException.class)
    public void testForbiddenSocketConnection() throws Exception {
        // mock Socket的所有构造方法，调用时抛出异常
        PowerMockito.whenNew(Socket.class).withAnyArguments()
            .thenThrow(new RuntimeException("禁止创建Socket连接"));

        // 尝试创建Socket会触发异常
        new Socket("example.com", 80);
    }
}

方案三：使用System Rules库（简洁易用） #

System Rules是专门用于JUnit 4的测试辅助库，提供了现成的规则来禁用Socket连接，代码非常简洁。

实现步骤： #

1. 引入System Rules依赖（以Maven为例）：

<dependency>
    <groupId>org.junit.contrib.java.lang.system</groupId>
    <artifactId>system-rules</artifactId>
    <version>1.19.0</version>
    <scope>test</scope>
</dependency>

2. 编写测试类：

import org.junit.Rule;
import org.junit.Test;
import org.junit.contrib.java.lang.system.DisableSocket;

import java.net.Socket;

public class MyUnitTest {
    // 启用禁用Socket的规则，默认会抛出IllegalStateException
    @Rule
    public DisableSocket disableSocket = new DisableSocket();

    @Test(expected = IllegalStateException.class)
    public void testForbiddenSocketConnection() throws Exception {
        // 尝试创建Socket会触发异常
        new Socket("example.com", 80);
    }
}

如果使用JUnit 5，可以改用system-stubs库，用法类似，只需调整注解和规则类。

内容的提问来源于stack exchange，提问作者yegor256