diff options
author | LordFokas <lordfokas@gmail.com> | 2015-10-23 01:29:03 +0100 |
---|---|---|
committer | LordFokas <lordfokas@gmail.com> | 2015-10-23 01:29:03 +0100 |
commit | e032d6b74e6df5385ecef58a044e0fc574f49eb0 (patch) | |
tree | 03a9c64cf41cd56980d2c3c00b8dfbc3ee31d1d9 | |
parent | dbb8383a3367aebd46b66252f30d8c6a0cf49de7 (diff) | |
download | nanohttpd-e032d6b74e6df5385ecef58a044e0fc574f49eb0.tar.gz |
attempt to fix the test
-rw-r--r-- | websocket/src/test/java/fi/iki/elonen/WebSocketResponseHandlerTest.java | 28 |
1 files changed, 26 insertions, 2 deletions
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<String, String> 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<String, String>(); this.headers.put("upgrade", "websocket"); |