如何在JUnit测试中管理JNDI Context，避免测试间上下文交叉污染？

方案1：测试前后全局重置InitialContext（快速适配现有代码） #

核心逻辑为适配现有代码，通过测试层的隔离逻辑清理全局共享的JNDI状态，避免交叉污染：

• 移除Provider中静态Context的缓存，每次创建上下文前先判断路径是否存在，避免重复创建抛出异常
• 编写JUnit测试扩展/规则，在每个测试执行前后清理全局JNDI相关状态

修改后的DataSource Provider代码：

@Provides
@Singleton
public DataSource getNavDS() throws NamingException {
    System.setProperty(Context.INITIAL_CONTEXT_FACTORY, "org.apache.naming.java.javaURLContextFactory");
    System.setProperty(Context.URL_PKG_PREFIXES, "org.apache.naming");
    
    BasicDataSource ds = new BasicDataSource();
    ds.setUrl("jdbc:h2:mem:myDB;create=true;MODE=MSSQLServer;DATABASE_TO_UPPER=FALSE;");
    
    Context context = new InitialContext();
    // 逐层判断子上下文是否存在，不存在才创建
    createSubcontextIfMissing(context, "java:");
    createSubcontextIfMissing(context, "java:comp");
    createSubcontextIfMissing(context, "java:comp/env");
    createSubcontextIfMissing(context, "java:comp/env/jdbc");
    context.rebind(NAV_DS, ds);                             
    
    return ds;
}

// 辅助方法：避免重复创建上下文抛出异常
private void createSubcontextIfMissing(Context ctx, String path) throws NamingException {
    try {
        ctx.lookup(path);
    } catch (NameNotFoundException e) {
        ctx.createSubcontext(path);
    }
}

JUnit5全局清理扩展示例：

public class JndiCleanupExtension implements BeforeEachCallback, AfterEachCallback {
    @Override
    public void beforeEach(ExtensionContext context) throws Exception {
        // 执行前清空旧的JNDI系统属性
        System.clearProperty(Context.INITIAL_CONTEXT_FACTORY);
        System.clearProperty(Context.URL_PKG_PREFIXES);
    }

    @Override
    public void afterEach(ExtensionContext context) throws Exception {
        // 执行后销毁InitialContext的静态缓存
        new InitialContext().close();
        // 清空所有相关系统属性
        System.clearProperty(Context.INITIAL_CONTEXT_FACTORY);
        System.clearProperty(Context.URL_PKG_PREFIXES);
    }
}

使用时在单元测试、集成测试类上添加@ExtendWith(JndiCleanupExtension.class)注解即可，保证每个测试的JNDI环境独立。

方案2：绕开JNDI依赖，测试环境直接注入DataSource（推荐，根除问题） #

问题根源是测试环境下EntityManager默认走JNDI查找数据源，只要调整测试用的JPA配置，完全可以避开InitialContext的使用，从根源避免全局污染：

1. 在测试资源目录下新建独立的META-INF/persistence.xml，覆盖生产环境的JNDI配置：

<persistence-unit name="testPU" transaction-type="RESOURCE_LOCAL">
    <!-- 移除原有的jta-data-source/non-jta-data-source JNDI配置 -->
    <class>com.yourbusiness.YourEntityClass</class>
    <properties>
        <!-- Hibernate配置示例，直接指定数据源参数，不需要JNDI -->
        <property name="hibernate.connection.driver_class" value="org.h2.Driver"/>
        <property name="hibernate.connection.url" value="jdbc:h2:mem:myDB;create=true;MODE=MSSQLServer;DATABASE_TO_UPPER=FALSE;"/>
        <property name="hibernate.hbm2ddl.auto" value="create-drop"/>
        <!-- 其他JPA通用配置和生产环境保持一致即可 -->
    </properties>
</persistence-unit>

2. 测试用Guice模块直接绑定EntityManager相关组件，不需要操作JNDI上下文：

@Provides
@Singleton
public EntityManagerFactory getEntityManagerFactory() {
    return Persistence.createEntityManagerFactory("testPU");
}

@Provides
public EntityManager getEntityManager(EntityManagerFactory emf) {
    return emf.createEntityManager();
}

@Provides
@Singleton
public DataSource getNavDS() {
    BasicDataSource ds = new BasicDataSource();
    ds.setUrl("jdbc:h2:mem:myDB;create=true;MODE=MSSQLServer;DATABASE_TO_UPPER=FALSE;");
    return ds;
}

方案3：嵌入式Jetty集成测试单独配置私有JNDI #

如果必须保留JNDI的使用方式，集成测试启动Jetty时可以单独配置Jetty私有JNDI上下文，不使用全局InitialContext，和单元测试环境完全隔离：

// 嵌入式Jetty配置示例
Server server = new Server(8080);
// 启用Jetty独立命名支持
Configuration.ClassList classlist = Configuration.ClassList.setServerDefault(server);
classlist.addBefore("org.eclipse.jetty.webapp.JettyWebXmlConfiguration", 
    "org.eclipse.jetty.plus.webapp.EnvConfiguration", 
    "org.eclipse.jetty.plus.webapp.PlusConfiguration");
// 单独引入Jetty私有JNDI配置文件，不修改全局系统属性
EnvConfiguration envConfig = new EnvConfiguration();
envConfig.setJettyEnvXml(Resource.newResource("src/test/resources/jetty-env.xml"));

在jetty-env.xml中配置数据源即可，Jetty会使用自己的上下文，不会和单元测试的全局InitialContext产生交叉影响。

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