aboutsummaryrefslogtreecommitdiff
path: root/websocket/src/test/java/fi/iki/elonen/WebSocketResponseHandlerTest.java
diff options
context:
space:
mode:
Diffstat (limited to 'websocket/src/test/java/fi/iki/elonen/WebSocketResponseHandlerTest.java')
-rw-r--r--websocket/src/test/java/fi/iki/elonen/WebSocketResponseHandlerTest.java58
1 files changed, 37 insertions, 21 deletions
diff --git a/websocket/src/test/java/fi/iki/elonen/WebSocketResponseHandlerTest.java b/websocket/src/test/java/fi/iki/elonen/WebSocketResponseHandlerTest.java
index 243601b..8c083ee 100644
--- a/websocket/src/test/java/fi/iki/elonen/WebSocketResponseHandlerTest.java
+++ b/websocket/src/test/java/fi/iki/elonen/WebSocketResponseHandlerTest.java
@@ -63,28 +63,44 @@ public class WebSocketResponseHandlerTest {
private NanoWSD nanoWebSocketServer;
private Map<String, String> headers;
-
- private static class MockedWSD extends NanoWSD{
- public MockedWSD(int port) {
- super(port);
- }
-
- public MockedWSD(String hostname, int port) {
- super(hostname, port);
- }
-
- @Override
- protected WebSocket openWebSocket(IHTTPSession handshake) {
- return new WebSocket(handshake) { // Dummy websocket inner class.
- @Override protected void onPong(WebSocketFrame pong) {}
- @Override protected void onOpen() {}
- @Override protected void onMessage(WebSocketFrame message) {}
- @Override protected void onException(IOException exception) {}
- @Override protected void onClose(CloseCode code, String reason, boolean initiatedByRemote) {}
- };
- }
+
+ private static class MockedWSD extends NanoWSD {
+
+ public MockedWSD(int port) {
+ super(port);
+ }
+
+ public MockedWSD(String hostname, int port) {
+ super(hostname, port);
+ }
+
+ @Override
+ protected WebSocket openWebSocket(IHTTPSession handshake) {
+ return new WebSocket(handshake) { // Dummy websocket inner class.
+
+ @Override
+ protected void onPong(WebSocketFrame pong) {
+ }
+
+ @Override
+ protected void onOpen() {
+ }
+
+ @Override
+ protected void onMessage(WebSocketFrame message) {
+ }
+
+ @Override
+ protected void onException(IOException exception) {
+ }
+
+ @Override
+ protected void onClose(CloseCode code, String reason, boolean initiatedByRemote) {
+ }
+ };
+ }
}
-
+
@Before
public void setUp() {
this.nanoWebSocketServer = Mockito.mock(MockedWSD.class, Mockito.CALLS_REAL_METHODS);