11package org .jooby .internal .jetty ;
22
3+ import static org .easymock .EasyMock .eq ;
34import static org .easymock .EasyMock .expect ;
45import static org .easymock .EasyMock .expectLastCall ;
56import static org .easymock .EasyMock .isA ;
910import java .util .Map ;
1011
1112import javax .inject .Provider ;
13+ import javax .servlet .ServletContext ;
1214
1315import org .eclipse .jetty .server .ConnectionFactory ;
1416import org .eclipse .jetty .server .HttpConfiguration ;
1517import org .eclipse .jetty .server .HttpConnectionFactory ;
1618import org .eclipse .jetty .server .Server ;
1719import org .eclipse .jetty .server .ServerConnector ;
1820import org .eclipse .jetty .server .handler .ContextHandler ;
21+ import org .eclipse .jetty .util .DecoratedObjectFactory ;
1922import org .eclipse .jetty .util .thread .QueuedThreadPool ;
2023import org .eclipse .jetty .util .thread .ThreadPool ;
2124import org .eclipse .jetty .websocket .api .WebSocketBehavior ;
@@ -104,6 +107,8 @@ public class JettyServerTest {
104107 .build ();
105108 ctx .setContextPath ("/" );
106109 ctx .setHandler (isA (JettyHandler .class ));
110+ ctx .setAttribute (eq (DecoratedObjectFactory .ATTR ), isA (DecoratedObjectFactory .class ));
111+ expect (ctx .getServletContext ()).andReturn (unit .get (ContextHandler .Context .class ));
107112
108113 server .setStopAtShutdown (false );
109114 server .setHandler (ctx );
@@ -173,8 +178,8 @@ public class JettyServerTest {
173178
174179 private Block wsFactory = unit -> {
175180 WebSocketServerFactory factory = unit .constructor (WebSocketServerFactory .class )
176- .args (WebSocketPolicy .class )
177- .build (unit .get (WebSocketPolicy .class ));
181+ .args (ServletContext . class , WebSocketPolicy .class )
182+ .build (unit .get (ContextHandler . Context . class ), unit . get ( WebSocketPolicy .class ));
178183
179184 factory .setCreator (isA (WebSocketCreator .class ));
180185
@@ -187,7 +192,7 @@ public class JettyServerTest {
187192 @ Test
188193 public void startStopServer () throws Exception {
189194
190- new MockUnit (HttpHandler .class , Provider .class )
195+ new MockUnit (HttpHandler .class , Provider .class , ContextHandler . Context . class )
191196 .expect (pool )
192197 .expect (server )
193198 .expect (httpConf )
0 commit comments