Merge pull request #787 from jooby-project/786

Add routes method to chain object fix #786

Author: jknack

coverage-report/src/test/java/org/jooby/issues/Issue786.java

@@ -0,0 +1,43 @@
+package org.jooby.issues;
+
+import static org.junit.Assert.assertEquals;
+
+import java.util.List;
+
+import org.jooby.Route;
+import org.jooby.test.ServerFeature;
+import org.junit.Test;
+
+public class Issue786 extends ServerFeature {
+
+  {
+    use("*", (req, rsp, chain) -> {
+      List<Route> routes = chain.routes();
+      assertEquals(2, routes.size());
+      assertEquals("/r2", routes.get(0).name());
+      assertEquals("/r3", routes.get(1).name());
+      assertEquals("/786/:id", routes.get(routes.size() - 1).pattern());
+      chain.next(req, rsp);
+    }).name("r1");
+
+    use("/786/**", (req, rsp, chain) -> {
+      List<Route> routes = chain.routes();
+      assertEquals(1, routes.size());
+      assertEquals("/r3", routes.get(0).name());
+      assertEquals("/786/:id", routes.get(routes.size() - 1).pattern());
+      chain.next(req, rsp);
+    }).name("r2");
+
+    get("/786/:id", req -> {
+      return req.param("id").value();
+    }).name("r3");
+  }
+
+  @Test
+  public void shouldHaveAccessToRoutePipeline() throws Exception {
+    request()
+        .get("/786/123")
+        .expect("123");
+  }
+
+}

coverage-report/src/test/java/org/jooby/issues/Issue786b.java

@@ -0,0 +1,46 @@
+package org.jooby.issues;
+
+import static org.junit.Assert.assertEquals;
+
+import java.util.List;
+
+import org.jooby.Route;
+import org.jooby.test.ServerFeature;
+import org.junit.Test;
+
+public class Issue786b extends ServerFeature {
+
+  {
+    use("*", (req, rsp, chain) -> {
+      List<Route> routes = chain.routes();
+      assertEquals(2, routes.size());
+      assertEquals("/r2", routes.get(0).name());
+      assertEquals("/r3", routes.get(1).name());
+      assertEquals("/786/:id", routes.get(routes.size() - 1).pattern());
+      chain.next(req, rsp);
+    }).name("r1");
+
+    use("/786/**", (req, rsp, chain) -> {
+      List<Route> routes = chain.routes();
+      assertEquals(1, routes.size());
+      assertEquals("/r3", routes.get(0).name());
+      assertEquals("/786/:id", routes.get(routes.size() - 1).pattern());
+      chain.next(req, rsp);
+    }).name("r2");
+
+    use("/786/:id", (req, rsp, chain) -> {
+      List<Route> routes = chain.routes();
+      assertEquals(0, routes.size());
+
+      rsp.send(req.param("id").value());
+    }).name("r3");
+  }
+
+  @Test
+  public void shouldHaveAccessToRoutePipeline() throws Exception {
+    request()
+        .get("/786/123")
+        .expect("123");
+  }
+
+}

jooby/src/main/java/org/jooby/Route.java

@@ -2159,6 +2159,37 @@ interface Chain {
     default void next(final Request req, final Response rsp) throws Throwable {
       next(null, req, rsp);
     }
+
+    /**
+     * All the pending/next routes from pipeline. Example:
+     *
+     * <pre>{@code
+     *   use("*", (req, rsp, chain) -> {
+     *     List<Route> routes = chain.routes();
+     *     assertEquals(2, routes.size());
+     *     assertEquals("/r2", routes.get(0).name());
+     *     assertEquals("/r3", routes.get(1).name());
+     *     assertEquals("/786/:id", routes.get(routes.size() - 1).pattern());
+     *
+     *     chain.next(req, rsp);
+     *   }).name("r1");
+     *
+     *   use("/786/**", (req, rsp, chain) -> {
+     *     List<Route> routes = chain.routes();
+     *     assertEquals(1, routes.size());
+     *     assertEquals("/r3", routes.get(0).name());
+     *     assertEquals("/786/:id", routes.get(routes.size() - 1).pattern());
+     *     chain.next(req, rsp);
+     *   }).name("r2");
+     *
+     *   get("/786/:id", req -> {
+     *     return req.param("id").value();
+     *   }).name("r3");
+     * }</pre>
+     *
+     * @return Next routes in the pipeline or empty list.
+     */
+    List<Route> routes();
   }
 
   /** Route key. */

jooby/src/main/java/org/jooby/internal/RouteChain.java

@@ -18,7 +18,9 @@
  */
 package org.jooby.internal;
 
+import java.util.Arrays;
 import java.util.HashMap;
+import java.util.List;
 import java.util.Map;
 
 import org.jooby.Request;
@@ -86,6 +88,11 @@ private Route next(final String prefix) {
     return route;
   }
 
+  @Override
+  public List<Route> routes() {
+    return Arrays.asList(routes).subList(i, routes.length - 1);
+  }
+
   private RouteWithFilter get(final Route next) {
     return (RouteWithFilter) Route.Forwarding.unwrap(next);
   }

jooby/src/main/java/org/jooby/test/MockRouter.java

@@ -43,6 +43,7 @@
 import org.jooby.Route.Filter;
 import org.jooby.Status;
 
+import com.google.common.collect.ImmutableList;
 import com.google.common.collect.Lists;
 import com.google.common.reflect.Reflection;
 import com.google.inject.Injector;
@@ -251,7 +252,16 @@
  */
 public class MockRouter {
 
-  private static final Route.Chain NOOP_CHAIN = (req, rsp, next) -> {
+  private static final Route.Chain NOOP_CHAIN = new Route.Chain() {
+
+    @Override
+    public List<Route> routes() {
+      return ImmutableList.of();
+    }
+
+    @Override
+    public void next(final String prefix, final Request req, final Response rsp) throws Throwable {
+    }
   };
 
   private static class MockResponse extends Response.Forwarding {