package org.eclipse.jetty.server.session;
   
   import static org.junit.Assert.*;
   
   import java.io.IOException;
   import java.util.EventListener;
   
   import javax.servlet.ServletException;
   import javax.servlet.http.HttpServlet;
  import javax.servlet.http.HttpServletRequest;
  import javax.servlet.http.HttpServletResponse;
  import javax.servlet.http.HttpSession;
  import javax.servlet.http.HttpSessionActivationListener;
  import javax.servlet.http.HttpSessionEvent;
  import javax.servlet.http.HttpSessionListener;
  
  import org.eclipse.jetty.client.ContentExchange;
  import org.eclipse.jetty.client.HttpClient;
  import org.eclipse.jetty.http.HttpMethods;
  import org.eclipse.jetty.servlet.ServletContextHandler;
  import org.junit.Test;
  
  public abstract class AbstractRemoveSessionTest
  {
      public abstract AbstractTestServer createServer(int port, int max, int scavenge);
      
      
      @Test
      public void testRemoveSession() throws Exception
      {
          String contextPath = "";
          String servletMapping = "/server";
          int scavengePeriod = 3;
          AbstractTestServer server = createServer(0, 1, scavengePeriod);
          ServletContextHandler context = server.addContext(contextPath);
          context.addServlet(TestServlet.class, servletMapping);
          TestEventListener testListener = new TestEventListener();
          context.getSessionHandler().addEventListener(testListener);
          server.start();
          int port = server.getPort();
          try
          {
              HttpClient client = new HttpClient();
              client.setConnectorType(HttpClient.CONNECTOR_SOCKET);
              client.start();
              try
              {
                  ContentExchange exchange = new ContentExchange(true);
                  exchange.setMethod(HttpMethods.GET);
                  exchange.setURL("http://localhost:" + port + contextPath + servletMapping + "?action=create");
                  client.send(exchange);
                  exchange.waitForDone();
                  assertEquals(HttpServletResponse.SC_OK,exchange.getResponseStatus());
                  String sessionCookie = exchange.getResponseFields().getStringField("Set-Cookie");
                  assertTrue(sessionCookie != null);
                  // Mangle the cookie, replacing Path with $Path, etc.
                  sessionCookie = sessionCookie.replaceFirst("(\\W)(P|p)ath=", "$1\\$Path=");
                  //ensure sessionCreated listener is called
                  assertTrue (testListener.isCreated());
  
                  //now delete the session
                  exchange = new ContentExchange(true);
                  exchange.setMethod(HttpMethods.GET);
                  exchange.setURL("http://localhost:" + port + contextPath + servletMapping + "?action=delete");
                  exchange.getRequestFields().add("Cookie", sessionCookie);
                  client.send(exchange);
                  exchange.waitForDone();
                  assertEquals(HttpServletResponse.SC_OK,exchange.getResponseStatus());
                  //ensure sessionDestroyed listener is called
                  assertTrue(testListener.isDestroyed());
                  
                  
                  // The session is not there anymore, but we present an old cookie
                  // The server creates a new session, we must ensure we released all locks
                  exchange = new ContentExchange(true);
                  exchange.setMethod(HttpMethods.GET);
                  exchange.setURL("http://localhost:" + port + contextPath + servletMapping + "?action=check");
                  exchange.getRequestFields().add("Cookie", sessionCookie);
                  client.send(exchange);
                  exchange.waitForDone();
                  assertEquals(HttpServletResponse.SC_OK,exchange.getResponseStatus());
              }
              finally
              {
                  client.stop();
              }
          }
          finally
          {
              server.stop();
          }
  
      }
      public static class TestServlet extends HttpServlet
      {
          protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException
          {
              String action = request.getParameter("action");
              if ("create".equals(action))
             {
                 request.getSession(true);
             }
             else if ("delete".equals(action))
             {
                 HttpSession s = request.getSession(false);
                 assertNotNull(s);
                 s.invalidate();
                 s = request.getSession(false);
                 assertNull(s);
             }
             else
             {
                HttpSession s = request.getSession(false);
                assertNull(s);
             }
         }
     }
     
     public static class TestEventListener implements HttpSessionListener
     {
         boolean wasCreated;
         boolean wasDestroyed;
 
         public void sessionCreated(HttpSessionEvent se)
         {
             wasCreated = true;
         }
 
         public void sessionDestroyed(HttpSessionEvent se)
         {
            wasDestroyed = true;
         }
 
         public boolean isDestroyed()
         {
             return wasDestroyed;
         }
 
 
         public boolean isCreated()
         {
             return wasCreated;
         }
 
     }
     
 }