websocket java 测试_如何测试Java EE7 Websocket

最新推荐文章于 2026-03-11 00:14:51 发布

原创最新推荐文章于 2026-03-11 00:14:51 发布 · 766 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

本文使用Java EE 7的API实现了WebSocket及客户端，为确保代码更改后仍有效，想实现可在构建服务器运行的测试。给出了服务器端点、客户端端点及当前测试代码，当前测试在本地可行，但在构建服务器无法完成，询问有无更简单的测试方法。

我使用

Java EE 7提供的api实现了一个WebSocket.此外,我已经实现了一个请求我的WebSocket没有任何问题的客户端.为了确保在执行某些代码更改时这仍然有效,我想实现也可以在构建服务器上运行的测试,例如,jenkinsCI不仅在当地.我正在使用maven.

这是我的服务器端点：

import javax.enterprise.context.ApplicationScoped;

import javax.websocket.*;

import javax.websocket.server.ServerEndpoint;

import java.io.IOException;

import java.util.Collections;

import java.util.HashSet;

import java.util.Set;

@ApplicationScoped

@ServerEndpoint("/example")

public class WebSocket {

private final Set sessions = Collections.synchronizedSet(new HashSet<>());

@OnOpen

public void open(Session session) {

sessions.add(session);

}

@OnClose

public void close(Session session) {

sessions.remove(session);

}

@OnError

public void onError(Throwable error) {

//TODO do something

}

@OnMessage

public void handleMessage(String message, Session session) throws IOException {

session.getBasicRemote().sendText("Hello "+message+"!");

}

这是我的客户端端点：

import javax.websocket.*;

import java.io.IOException;

import java.net.URI;

@ClientEndpoint

public class WebsocketClient {

private String uri = "ws://localhost:8181//example";

private Session userSession = null;

private String answer;

public WebsocketClient() {

try {

WebSocketContainer container = ContainerProvider

.getWebSocketContainer();

container.connectToServer(this, new URI(uri));

} catch (Exception e) {

throw new RuntimeException(e);

}

public boolean isClosed() {

return userSession == null || !userSession.isOpen();

}

@OnOpen

public void onOpen(Session userSession) {

this.userSession = userSession;

}

@OnClose

public void onClose(Session userSession, CloseReason reason) {

this.userSession = null;

}

@OnMessage

public void onMessage(String message) {

this.answer = message;

}

public String getAnswer() {

return answer;

}

public void sendMessage(String message) {

if (userSession != null && userSession.isOpen())

try {

this.userSession.getBasicRemote().sendText(message);

} catch (IOException e) {

e.printStackTrace();

}

else {

System.out.println("Session closed!");

}

public void reset() {

this.answer = null;

}

我还有一些在使用CDI的构建服务器上测试的其他类.为了能够使用CDI和其他应用程序容器功能,我在版本7.0.0-M3中使用带有apache TomeEE / OpenEJB的EJBContainerRunner启动嵌入式TomEE运行我的测试并在之后关闭它.使用这种方法我没有要调用的URL.这适用于例如JAX-RS REST服务,您可以在测试中调用类方法并检查响应.但是你如何用WebSockets做这样的事情呢？由于缺少Session,我不能简单地调用方法.

我目前的测试看起来像这样：

import org.apache.openejb.junit.jee.EJBContainerRunner;

import org.junit.Test;

import org.junit.runner.RunWith;

import java.net.SocketException;

import static org.hamcrest.MatcherAssert.assertThat;

import static org.hamcrest.Matchers.equalTo;

import static org.hamcrest.Matchers.is;

@RunWith(EJBContainerRunner.class)

public class TestWebsocket {

private WebsocketClient socket;

@Test