JUnit测试中HttpURLConnection受限请求头的替代启用方法咨询

我之前也踩过一模一样的坑——当被测试类的静态初始化器里提前创建了HttpURLConnection实例时，@BeforeTest（或是JUnit 4的@Before、JUnit 5的@BeforeEach）里设置sun.net.http.allowRestrictedHeaders确实会失效，因为静态代码块的执行时机远早于这些测试生命周期方法。下面给你几个靠谱的替代方案：

1. 在测试类的静态代码块中设置属性 #

把系统属性的设置放到测试类的静态代码块里，确保它在被测试类的静态初始化器之前执行。比如：

public class YourRestrictedHeaderTest {
    // 静态块优先执行，早于被测试类的静态初始化逻辑
    static {
        System.setProperty("sun.net.http.allowRestrictedHeaders", "true");
    }

    // 被测试类的静态引用要放在静态块之后，保证属性先生效
    private static YourClassUnderTest testInstance;

    @BeforeTest
    public void setUp() {
        testInstance = new YourClassUnderTest();
    }

    // 测试方法...
}

注意：如果测试类里有被测试类的静态字段，一定要把设置属性的静态块放在所有静态引用的前面，严格保证属性先被设置完成。

2. 用JVM启动参数全局设置 #

这是最省心的方案——直接在测试的JVM启动参数里加上这个系统属性，让它在整个测试进程启动时就生效，完全不用纠结类加载顺序。

• 如果用Maven，在pom.xml的surefire插件里配置：

<build>
    <plugins>
        <plugin>
            <groupId>org.apache.maven.plugins</groupId>
            <artifactId>maven-surefire-plugin</artifactId>
            <configuration>
                <argLine>-Dsun.net.http.allowRestrictedHeaders=true</argLine>
            </configuration>
        </plugin>
    </plugins>
</build>

• 如果用Gradle，在build.gradle里配置：

test {
    jvmArgs '-Dsun.net.http.allowRestrictedHeaders=true'
}

这个方法的优势在于，不管被测试类的静态代码什么时候执行，都能读到正确的属性值，彻底规避时机问题。

3. 用JUnit类级别的规则（JUnit 4） #

如果用JUnit 4，可以通过@ClassRule实现类级别的初始化，确保属性在测试类初始化阶段就被设置：

public class YourRestrictedHeaderTest {
    public static class AllowRestrictedHeadersRule implements TestRule {
        @Override
        public Statement apply(Statement base, Description description) {
            System.setProperty("sun.net.http.allowRestrictedHeaders", "true");
            return base;
        }
    }

    @ClassRule
    public static AllowRestrictedHeadersRule allowRule = new AllowRestrictedHeadersRule();

    // 被测试类实例和测试方法...
}

@ClassRule会在测试类的所有测试方法执行前、类级初始化完成后运行，但如果被测试类的静态初始化器是在测试类加载时就触发的，这个方法可能还是不如静态块或JVM参数可靠，不过大多数常规场景下足够用。

为什么@BeforeTest会失效？ #

简单来说：@BeforeTest（或@Before）是在每个测试方法执行前才运行的，而类的静态初始化器是在类第一次被加载到JVM时就执行了——这个时机远早于任何测试方法的启动，所以当静态代码里创建HttpURLConnection时，allowRestrictedHeaders还没被设置，自然无效。

内容的提问来源于stack exchange，提问作者Martín Coll