Updated WebSocketEmulatedHandlerTest for JDK 8 on Ubuntu 14.04.1.

Sanjay Saxena · Sanjay Saxena · commit 91dc145c7320 · 2015-05-26T08:45:14.000-07:00
diff --git a/ws/ws/src/test/java/org/kaazing/gateway/client/impl/wseb/WebSocketEmulatedHandlerTest.java b/ws/ws/src/test/java/org/kaazing/gateway/client/impl/wseb/WebSocketEmulatedHandlerTest.java
@@ -47,6 +47,10 @@ public void testProcessOpen() throws URISyntaxException {
         final UpstreamHandler upstreamHandler = context.mock(UpstreamHandler.class);
         final DownstreamHandler downstreamHandler = context.mock(DownstreamHandler.class);
         final WebSocketHandlerListener listener = context.mock(WebSocketHandlerListener.class);
+        
+        CreateHandlerFactory defaultCreateHandlerFactory = WebSocketEmulatedHandler.createHandlerFactory;
+        UpstreamHandlerFactory defaultUpstreamHandlerFactory = WebSocketEmulatedHandler.upstreamHandlerFactory;
+        DownstreamHandlerFactory defaultDownstreamHandlerFactory = WebSocketEmulatedHandler.downstreamHandlerFactory;
 
         context.checking(new Expectations() {
             {
@@ -119,6 +123,10 @@ public DownstreamHandler createDownstreamHandler() {
 
         handler.processConnect(channel, uri, new String[]{"foo"});
         context.assertIsSatisfied();
+        
+        WebSocketEmulatedHandler.createHandlerFactory = defaultCreateHandlerFactory;
+        WebSocketEmulatedHandler.upstreamHandlerFactory = defaultUpstreamHandlerFactory;
+        WebSocketEmulatedHandler.downstreamHandlerFactory = defaultDownstreamHandlerFactory;
     }
 
     /*
@@ -133,6 +141,10 @@ public void testProcessFailed() throws URISyntaxException {
         final DownstreamHandler downstreamHandler = context.mock(DownstreamHandler.class);
         final WebSocketHandlerListener listener = context.mock(WebSocketHandlerListener.class);
 
+        CreateHandlerFactory defaultCreateHandlerFactory = WebSocketEmulatedHandler.createHandlerFactory;
+        UpstreamHandlerFactory defaultUpstreamHandlerFactory = WebSocketEmulatedHandler.upstreamHandlerFactory;
+        DownstreamHandlerFactory defaultDownstreamHandlerFactory = WebSocketEmulatedHandler.downstreamHandlerFactory;
+
         context.checking(new Expectations() {
 
             {
@@ -191,6 +203,10 @@ public DownstreamHandler createDownstreamHandler() {
 
         handler.processConnect(channel, uri, protocols);
         context.assertIsSatisfied();
+        
+        WebSocketEmulatedHandler.createHandlerFactory = defaultCreateHandlerFactory;
+        WebSocketEmulatedHandler.upstreamHandlerFactory = defaultUpstreamHandlerFactory;
+        WebSocketEmulatedHandler.downstreamHandlerFactory = defaultDownstreamHandlerFactory;
     }
 
 }
