ContextRefreshedEvent 在 Spring 集成测试中过早触发

Question

我想测试一个像Example这样的类，它处理ContextRefreshedEvent并在处理程序方法中连接到服务器：

public class Example {
@EventListener
public void onApplicationEvent(ContextRefreshedEvent event) {
startWebSocketConnection();
}
// ...
}

但是在集成测试中，应用程序上下文是在 Web 套接字服务器启动并运行之前构建的，因此我收到一个异常，指出连接失败(在这种情况下java.net.ConnectException: Connection refused: no further information(。

测试如下所示：

@ExtendWith(SpringExtension.class)
@TestInstance(TestInstance.Lifecycle.PER_CLASS)
@SpringBootTest
public class WebSocketDataSourceTest {
@Autowired 
private Example example;
@Autowired
private WebSocketServer server; // created too late
// ...
}

是否可以以某种方式抑制ContextRefreshedEvent或延迟应用程序上下文的创建，以便 Web 套接字服务器可以在之前启动？还是有其他解决方案？

Answer 1

似乎没有办法抑制 Spring 框架触发的事件或延迟应用程序上下文的创建。所以我想出了以下解决方法：

import org.springframework.core.env.Environment;
public class Example {
private boolean skipNextEvent;
@Autowired
public Example(Environment environment) {
skipNextEvent = environment.acceptsProfiles("test");
}
@EventListener
public void onApplicationEvent(ContextRefreshedEvent event) {
if (skipNextEvent) {
skipNextEvent = false;
return;
}
startWebSocketConnection();
}
// ...
}

测试手动触发事件处理程序。

@ExtendWith(SpringExtension.class)
@TestInstance(TestInstance.Lifecycle.PER_CLASS)
@SpringBootTest
@ActiveProfiles("test") // set profile "test"
public class WebSocketDataSourceTest {
@Autowired 
private Example example;
@Autowired
private WebSocketServer server;
@Test
public void shouldWork() {
// ...
example.onApplicationEvent(null); // trigger manually
// ...
}
}