Handle expectation with and without query string

Author: ruhan1

core/src/main/java/org/commonjava/test/http/expect/ExpectationServlet.java

@@ -27,13 +27,20 @@
 import java.io.IOException;
 import java.io.InputStream;
 import java.net.MalformedURLException;
-import java.net.URI;
-import java.net.URISyntaxException;
 import java.net.URL;
 import java.util.HashMap;
 import java.util.Map;
 
-@SuppressWarnings( "unused" )
+import javax.servlet.ServletException;
+import javax.servlet.http.HttpServlet;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+
+import org.apache.commons.io.IOUtils;
+import org.commonjava.test.http.common.CommonMethod;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
 public final class ExpectationServlet
         extends HttpServlet
 {
@@ -105,7 +112,7 @@ private String getPath( final String testUrl )
         }
         catch ( final MalformedURLException e )
         {
-            e.printStackTrace();
+            //Do nothing
         }
         return testUrl;
     }
@@ -142,12 +149,30 @@ protected void service( final HttpServletRequest req, final HttpServletResponse
             throws ServletException, IOException
     {
         String wholePath = getWholePath( req );
-        final String key = getAccessKey( req.getMethod(), wholePath );
+        String key = getAccessKey( req.getMethod(), wholePath );
+        accessesByPath.merge(key, 1, Integer::sum);
 
-        logger.info( "Looking up expectation for: {}", key );
+        boolean handled = handle( key, req, resp );
+        if (!handled)
+        {
+            // try simple path
+            String simplePath = getSimplePath(req);
+            if (!simplePath.equals(wholePath))
+            {
+                key = getAccessKey(req.getMethod(), simplePath);
+                handled = handle(key, req, resp);
+            }
+        }
 
-        accessesByPath.merge( key, 1, Integer::sum );
+        if (!handled)
+        {
+            resp.setStatus( 404 );
+        }
+    }
 
+    private boolean handle(String key, HttpServletRequest req, HttpServletResponse resp)
+            throws IOException, ServletException
+    {
         logger.info( "Looking for error: '{}' in:\n{}", key, errors );
         if ( errors.containsKey( key ) )
         {
@@ -157,7 +182,6 @@ protected void service( final HttpServletRequest req, final HttpServletResponse
             if ( error.handler() != null )
             {
                 error.handler().handle( req, resp );
-                return;
             }
             else if ( error.body() != null )
             {
@@ -172,21 +196,19 @@ else if ( error.bodyStream() != null )
             {
                 resp.sendError( error.code() );
             }
-
-            return;
+            return true;
         }
 
         logger.info( "Looking for expectation: '{}'", key );
-        final ContentResponse expectation = expectations.get( key );
-        if ( expectation != null )
+        if ( expectations.containsKey( key ) )
         {
+            final ContentResponse expectation = expectations.get( key );
             logger.info( "Responding via registered expectation: {}", expectation );
 
             if ( expectation.handler() != null )
             {
                 expectation.handler().handle( req, resp );
                 logger.info( "Using handler..." );
-                return;
             }
             else if ( expectation.body() != null )
             {
@@ -207,13 +229,15 @@ else if ( expectation.bodyStream() != null )
                 resp.setStatus( expectation.code() );
                 logger.info( "Set status: {} with no body", expectation.code() );
             }
-
-            return;
+            return true;
         }
 
-        resp.setStatus( 404 );
+        return false;
     }
 
+    /**
+     * Get path with query string.
+     */
     private String getWholePath( HttpServletRequest request )
     {
         String requestURI = request.getRequestURI();
@@ -229,6 +253,11 @@ private String getWholePath( HttpServletRequest request )
         }
     }
 
+    private String getSimplePath( HttpServletRequest request )
+    {
+        return request.getRequestURI();
+    }
+
     public String getAccessKey( final CommonMethod method, final String path )
     {
         return getAccessKey( method.name(), path );