Как протестировать Java Ee7 Websocket
я реализовал WebSocket, используя api, предоставляемый Java EE 7. Кроме того, я реализовал клиент, который запрашивает мой WebSocket без каких-либо проблем. Чтобы убедиться, что это работает при выполнении некоторых изменений кода, я хочу реализовать тесты, которые также могут быть запущены на сервере сборки, например, Jenkins CI, а не только локально. Я использую 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<Session> 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/<some-path>/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 и другие функции контейнера приложений, я использую EJBContainerRunner
С apache TomeEE / OpenEJB в версии 7.0.0-M3
который запускает встроенный 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
public void test() throws SocketException {
String answer = requestSynchronous("Java");
assertThat(answer, is(equalTo("Hello Java!")));
}
public String requestSynchronous(String message) throws SocketException {
if (socket == null || socket.isClosed()) {
socket = new WebsocketClient();
}
socket.reset();
socket.sendMessage(message);
String answer = null;
int i = 0;
while (answer == null && i < 10000) {
try {
answer = socket.getAnswer();
Thread.sleep(1);
i++;
} catch (InterruptedException e) {
e.printStackTrace();
}
}
if (answer != null)
return answer;
else throw new SocketException("Connection timeout");
}
}
это работает под управлением TomEE на моей локальной машине, например,tomee-maven-plugin
. Но этот подход не может быть выполнен на сервере сборки. Есть ли более простой способ, чем настройка специальных портов или настройка виртуальной машины для тестирования?
спасибо заранее!
1 ответов
я нашел решение для моей проблемы тестирования без насмешек, используя vm ...
для отправки ответного сообщения клиенту не нужно звонить session.getBasicRemote().sendText("...");
. Вместо этого верните сообщение, которое вы хотите отправить.
на handleMessage
метод в классе WebSocket
изменения:
@OnMessage
public String handleMessage(String message){
return "Hello "+message+"!";
}
и тест упрощает до:
@RunWith(EJBContainerRunner.class)
public class TestWebsocket {
@Inject
private WebSocket socket;
@Test
public void test() throws SocketException {
String answer = socket.handleMessage("Java");
assertThat(answer, is(equalTo("Hello Java!")));
}
}
этот подход не отправляет сообщения за пределы контейнера веб-приложения, который запускается для теста. Этот делает его гораздо легче справиться с тестом.