//
   //  ========================================================================
   //  Copyright (c) 1995-2013 Mort Bay Consulting Pty. Ltd.
   //  ------------------------------------------------------------------------
   //  All rights reserved. This program and the accompanying materials
   //  are made available under the terms of the Eclipse Public License v1.0
   //  and Apache License v2.0 which accompanies this distribution.
   //
   //      The Eclipse Public License is available at
  //      http://www.eclipse.org/legal/epl-v10.html
  //
  //      The Apache License v2.0 is available at
  //      http://www.opensource.org/licenses/apache2.0.php
  //
  //  You may elect to redistribute this code under either of these licenses.
  //  ========================================================================
  //
  
  package org.eclipse.jetty.server.session;
  
  import java.io.IOException;
  
  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 org.eclipse.jetty.client.ContentExchange;
  import org.eclipse.jetty.client.HttpClient;
  import org.eclipse.jetty.http.HttpMethods;
  import org.eclipse.jetty.server.Request;
  import org.eclipse.jetty.server.SessionManager;
  import org.eclipse.jetty.server.session.AbstractTestServer;
  import org.eclipse.jetty.servlet.ServletHolder;
  import org.junit.Test;
  import static org.junit.Assert.assertEquals;
  import static org.junit.Assert.assertTrue;
  
  
  
  /**
   * AbstractSessionExpiryTest
   *
   *
   *
   */
  public abstract class AbstractSessionExpiryTest
  {
      public abstract AbstractTestServer createServer(int port, int max, int scavenge);
  
      public void pause(int scavengePeriod)
      {
          try
          {
              Thread.sleep(scavengePeriod * 2500L);
          }
          catch (InterruptedException e)
          {
              e.printStackTrace();
          }
      }
      
      @Test
      public void testSessionNotExpired() throws Exception
      {
          String contextPath = "";
          String servletMapping = "/server";
          int inactivePeriod = 10;
          int scavengePeriod = 10;
          AbstractTestServer server1 = createServer(0, inactivePeriod, scavengePeriod);
          TestServlet servlet = new TestServlet();
          ServletHolder holder = new ServletHolder(servlet);
          server1.addContext(contextPath).addServlet(holder, servletMapping);
          server1.start();
          int port1 = server1.getPort();
  
          try
          {
              HttpClient client = new HttpClient();
              client.setConnectorType(HttpClient.CONNECTOR_SOCKET);
              client.start();
              String url = "http://localhost:" + port1 + contextPath + servletMapping;
  
              //make a request to set up a session on the server
              ContentExchange exchange1 = new ContentExchange(true);
              exchange1.setMethod(HttpMethods.GET);
              exchange1.setURL(url + "?action=init");
              client.send(exchange1);
              exchange1.waitForDone();
              assertEquals(HttpServletResponse.SC_OK,exchange1.getResponseStatus());
              String sessionCookie = exchange1.getResponseFields().getStringField("Set-Cookie");
              assertTrue(sessionCookie != null);
              // Mangle the cookie, replacing Path with $Path, etc.
              sessionCookie = sessionCookie.replaceFirst("(\\W)(P|p)ath=", "$1\\$Path=");
              
              //now stop the server
              server1.stop();
              
             //start the server again, before the session times out
             server1.start();            
             port1 = server1.getPort();
             url = "http://localhost:" + port1 + contextPath + servletMapping;
             
             //make another request, the session should not have expired
             ContentExchange exchange2 = new ContentExchange(true);
             exchange2.setMethod(HttpMethods.GET);
             exchange2.setURL(url + "?action=notexpired");
             exchange2.getRequestFields().add("Cookie", sessionCookie);
             client.send(exchange2);
             exchange2.waitForDone();
             assertEquals(HttpServletResponse.SC_OK,exchange2.getResponseStatus());
 
         }
         finally
         {
             server1.stop();
         }
     }
 
     @Test
     public void testSessionExpiry() throws Exception
     {
         String contextPath = "";
         String servletMapping = "/server";
         int inactivePeriod = 2;
         int scavengePeriod = 10;
         AbstractTestServer server1 = createServer(0, inactivePeriod, scavengePeriod);
         TestServlet servlet = new TestServlet();
         ServletHolder holder = new ServletHolder(servlet);
         server1.addContext(contextPath).addServlet(holder, servletMapping);
         server1.start();
         int port1 = server1.getPort();
 
         try
         {
             HttpClient client = new HttpClient();
             client.setConnectorType(HttpClient.CONNECTOR_SOCKET);
             client.start();
             String url = "http://localhost:" + port1 + contextPath + servletMapping;
 
             //make a request to set up a session on the server
             ContentExchange exchange1 = new ContentExchange(true);
             exchange1.setMethod(HttpMethods.GET);
             exchange1.setURL(url + "?action=init");
             client.send(exchange1);
             exchange1.waitForDone();
             assertEquals(HttpServletResponse.SC_OK,exchange1.getResponseStatus());
             String sessionCookie = exchange1.getResponseFields().getStringField("Set-Cookie");
             assertTrue(sessionCookie != null);
             // Mangle the cookie, replacing Path with $Path, etc.
             sessionCookie = sessionCookie.replaceFirst("(\\W)(P|p)ath=", "$1\\$Path=");
             
             //now stop the server
             server1.stop();
             
             //and wait until the expiry time has passed
             pause(inactivePeriod);
             
             //restart the server
             server1.start();            
             port1 = server1.getPort();
             url = "http://localhost:" + port1 + contextPath + servletMapping;
             
             //make another request, the session should have expired
             ContentExchange exchange2 = new ContentExchange(true);
             exchange2.setMethod(HttpMethods.GET);
             exchange2.setURL(url + "?action=test");
             exchange2.getRequestFields().add("Cookie", sessionCookie);
             client.send(exchange2);
             exchange2.waitForDone();
             assertEquals(HttpServletResponse.SC_OK,exchange2.getResponseStatus());
         }
         finally
         {
             server1.stop();
         }
     }
 
     public static class TestServlet extends HttpServlet
     {
         public String originalId = null;
         public String testId = null;
         public String checkId = null;
 
         @Override
         protected void doGet(HttpServletRequest request, HttpServletResponse httpServletResponse) throws ServletException, IOException
         {
             String action = request.getParameter("action");
             if ("init".equals(action))
             {
                 HttpSession session = request.getSession(true);
                 session.setAttribute("test", "test");
                 originalId = session.getId();
             }
             else if ("test".equals(action))
             {
                 HttpSession session = request.getSession(true);
                 assertTrue(session != null);
                 assertTrue(!originalId.equals(session.getId()));
             }
             else if ("notexpired".equals(action))
             {
                 HttpSession session = request.getSession(false);
                 assertTrue(session != null);
                 assertTrue(originalId.equals(session.getId()));
             }
            
         }
     }
 }