From e032d6b74e6df5385ecef58a044e0fc574f49eb0 Mon Sep 17 00:00:00 2001 From: LordFokas Date: Fri, 23 Oct 2015 01:29:03 +0100 Subject: attempt to fix the test --- .../iki/elonen/WebSocketResponseHandlerTest.java | 28 ++++++++++++++++++++-- 1 file changed, 26 insertions(+), 2 deletions(-) (limited to 'websocket/src/test') diff --git a/websocket/src/test/java/fi/iki/elonen/WebSocketResponseHandlerTest.java b/websocket/src/test/java/fi/iki/elonen/WebSocketResponseHandlerTest.java index 69aa943..9dd5945 100644 --- a/websocket/src/test/java/fi/iki/elonen/WebSocketResponseHandlerTest.java +++ b/websocket/src/test/java/fi/iki/elonen/WebSocketResponseHandlerTest.java @@ -38,6 +38,7 @@ import static junit.framework.Assert.assertNotNull; import static junit.framework.Assert.assertNull; import static org.mockito.Mockito.when; +import java.io.IOException; import java.util.HashMap; import java.util.Map; @@ -50,6 +51,8 @@ import org.mockito.runners.MockitoJUnitRunner; import fi.iki.elonen.NanoHTTPD.IHTTPSession; import fi.iki.elonen.NanoHTTPD.Response; +import fi.iki.elonen.NanoWebSocketServer.WebSocketFrame; +import fi.iki.elonen.NanoWebSocketServer.WebSocketFrame.CloseCode; @RunWith(MockitoJUnitRunner.class) public class WebSocketResponseHandlerTest { @@ -60,10 +63,31 @@ public class WebSocketResponseHandlerTest { private NanoWebSocketServer nanoWebSocketServer; private Map headers; - + + private static class MockedWSD extends NanoWebSocketServer{ + 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(NanoWebSocketServer.class, Mockito.CALLS_REAL_METHODS); + this.nanoWebSocketServer = Mockito.mock(MockedWSD.class, Mockito.CALLS_REAL_METHODS); this.headers = new HashMap(); this.headers.put("upgrade", "websocket"); -- cgit v1.2.3