Spring Boot中WebSocket与JUnit测试冲突解决方案

1. 问题现象与背景分析

最近在基于若依（RuoYi-Vue-Plus）框架开发项目时，遇到一个典型的技术冲突：当启用WebSocket功能后，原本正常的JUnit单元测试开始报错。具体表现为测试类初始化时抛出IllegalStateException，提示"Failed to load ApplicationContext"。

这个问题其实反映了Spring Boot应用中WebSocket与测试环境的兼容性问题。RuoYi-Vue-Plus作为主流的企业级快速开发框架，默认整合了WebSocket模块用于实时通信，但测试配置往往没有考虑这种特殊依赖。

2. 根因深度解析

2.1 WebSocket的初始化机制

WebSocket在Spring中的实现依赖于ServerEndpointExporter，这个Bean会在应用启动时注册WebSocket端点。关键问题在于：

• 它需要真实的Servlet容器环境
• 会尝试绑定到配置的端口（默认8080）
• 需要完整的Spring上下文支持

2.2 JUnit测试的特殊性

单元测试默认使用Mock环境：

• 不启动真实Servlet容器
• 使用轻量级的TestContext框架
• 通常禁用非必要的自动配置

当两者相遇时，测试框架无法满足WebSocket的运行时需求，导致上下文加载失败。

3. 解决方案与实施步骤

3.1 方案一：禁用WebSocket自动配置（推荐）

在测试配置类中添加：

java复制@TestConfiguration
@EnableAutoConfiguration(exclude = {WebSocketServletAutoConfiguration.class})
public class TestConfig {
    // 其他测试配置...
}

优势：

• 完全隔离WebSocket影响
• 不污染生产配置
• 执行效率最高

3.2 方案二：Mock WebSocket环境

对于必须测试WebSocket的场景：

java复制@SpringBootTest
@MockBean(ServerEndpointExporter.class)
public class