123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371
// Copyright 2018-2019 the oak authors. All rights reserved. MIT license.
import { assertEquals, assertStrictEq, test } from "./test_deps.ts";import { Application } from "./application.ts";import { Context } from "./context.ts";import { Status } from "./deps.ts";import { Router, RouterContext, RouteParams } from "./router.ts";import { Request } from "./request.ts";import { Response } from "./response.ts";
function createMockApp<S extends object = { [key: string]: any }>(  state = {} as S): Application<S> {  return {    state  } as any;}
function createMockContext<S extends object = { [key: string]: any }>(  app: Application<S>,  path = "/",  method = "GET") {  return {    app,    request: {      headers: new Headers(),      method,      path,      search: undefined,      searchParams: new URLSearchParams(),      url: path    } as Request,    response: {      status: Status.OK,      body: undefined,      headers: new Headers()    } as Response,    state: app.state  } as Context<S>;}
function createMockNext() {  return async function next() {};}
function setup<S extends object = { [key: string]: any }>(  path = "/",  method = "GET"): {  app: Application<S>;  context: Context<S>;  next: () => Promise<void>;} {  const app = createMockApp<S>();  const context = createMockContext<S>(app, path, method);  const next = createMockNext();  return { app, context, next };}
test(async function emptyRoutes() {  const { context, next } = setup();
  const router = new Router();  const mw = router.routes();  assertEquals(await mw(context, next), undefined);});
test(async function getSingleMatch() {  const { app, context, next } = setup("/", "GET");
  const callStack: number[] = [];  const router = new Router();  router.get("/", context => {    assertStrictEq(context.router, router);    assertStrictEq(context.app, app);    callStack.push(1);  });  const mw = router.routes();  await mw(context, next);  assertEquals(callStack, [1]);});
test(async function matchSingleParam() {  const { context, next } = setup("/foo/bar", "GET");
  const callStack: number[] = [];  const router = new Router();  router.get("/", context => {    callStack.push(1);  });  router.get("/foo", context => {    callStack.push(2);  });  router.get<{ id: string }>("/foo/:id", context => {    callStack.push(3);    assertEquals(context.params.id, "bar");  });  const mw = router.routes();  await mw(context, next);  assertEquals(callStack, [3]);});
test(async function matchWithNext() {  const { context, next } = setup("/foo", "GET");
  const callStack: number[] = [];  const router = new Router();  router.get("/", _context => {    callStack.push(1);  });  router.get("/foo", async (_context, next) => {    callStack.push(2);    await next();  });  router.get("/foo", () => {    callStack.push(3);  });  router.get("/foo", () => {    callStack.push(4);  });  const mw = router.routes();  await mw(context, next);  assertEquals(callStack, [2, 3]);});
test(async function matchDelete() {  const { context, next } = setup("/", "DELETE");
  const callStack: number[] = [];  const router = new Router();  router.all("/", async (_context, next) => {    callStack.push(0);    await next();  });  router.delete("/", () => {    callStack.push(1);  });  router.get("/", () => {    callStack.push(2);  });  router.head("/", () => {    callStack.push(3);  });  router.options("/", () => {    callStack.push(4);  });  router.patch("/", () => {    callStack.push(5);  });  router.post("/", () => {    callStack.push(6);  });  router.put("/", () => {    callStack.push(7);  });  const mw = router.routes();  await mw(context, next);  assertEquals(callStack, [0, 1]);});
test(async function matchGet() {  const { context, next } = setup("/", "GET");
  const callStack: number[] = [];  const router = new Router();  router.all("/", async (_context, next) => {    callStack.push(0);    await next();  });  router.delete("/", () => {    callStack.push(1);  });  router.get("/", () => {    callStack.push(2);  });  router.head("/", () => {    callStack.push(3);  });  router.options("/", () => {    callStack.push(4);  });  router.patch("/", () => {    callStack.push(5);  });  router.post("/", () => {    callStack.push(6);  });  router.put("/", () => {    callStack.push(7);  });  const mw = router.routes();  await mw(context, next);  assertEquals(callStack, [0, 2]);});
test(async function matchHead() {  const { context, next } = setup("/", "HEAD");
  const callStack: number[] = [];  const router = new Router();  router.all("/", async (_context, next) => {    callStack.push(0);    await next();  });  router.delete("/", () => {    callStack.push(1);  });  router.head("/", () => {    callStack.push(3);  });  router.get("/", () => {    callStack.push(2);  });  router.options("/", () => {    callStack.push(4);  });  router.patch("/", () => {    callStack.push(5);  });  router.post("/", () => {    callStack.push(6);  });  router.put("/", () => {    callStack.push(7);  });  const mw = router.routes();  await mw(context, next);  assertEquals(callStack, [0, 3]);});
test(async function matchOptions() {  const { context, next } = setup("/", "OPTIONS");
  const callStack: number[] = [];  const router = new Router();  router.all("/", async (_context, next) => {    callStack.push(0);    await next();  });  router.delete("/", () => {    callStack.push(1);  });  router.get("/", () => {    callStack.push(2);  });  router.head("/", () => {    callStack.push(3);  });  router.options("/", () => {    callStack.push(4);  });  router.patch("/", () => {    callStack.push(5);  });  router.post("/", () => {    callStack.push(6);  });  router.put("/", () => {    callStack.push(7);  });  const mw = router.routes();  await mw(context, next);  assertEquals(callStack, [4]);});
test(async function matchPatch() {  const { context, next } = setup("/", "PATCH");
  const callStack: number[] = [];  const router = new Router();  router.all("/", async (_context, next) => {    callStack.push(0);    await next();  });  router.delete("/", () => {    callStack.push(1);  });  router.get("/", () => {    callStack.push(2);  });  router.head("/", () => {    callStack.push(3);  });  router.options("/", () => {    callStack.push(4);  });  router.patch("/", () => {    callStack.push(5);  });  router.post("/", () => {    callStack.push(6);  });  router.put("/", () => {    callStack.push(7);  });  const mw = router.routes();  await mw(context, next);  assertEquals(callStack, [5]);});
test(async function matchPost() {  const { context, next } = setup("/", "POST");
  const callStack: number[] = [];  const router = new Router();  router.all("/", async (_context, next) => {    callStack.push(0);    await next();  });  router.delete("/", () => {    callStack.push(1);  });  router.get("/", () => {    callStack.push(2);  });  router.head("/", () => {    callStack.push(3);  });  router.options("/", () => {    callStack.push(4);  });  router.patch("/", () => {    callStack.push(5);  });  router.post("/", () => {    callStack.push(6);  });  router.put("/", () => {    callStack.push(7);  });  const mw = router.routes();  await mw(context, next);  assertEquals(callStack, [0, 6]);});
test(async function matchPut() {  const { context, next } = setup("/", "PUT");
  const callStack: number[] = [];  const router = new Router();  router.all("/", async (_context, next) => {    callStack.push(0);    await next();  });  router.delete("/", () => {    callStack.push(1);  });  router.get("/", () => {    callStack.push(2);  });  router.head("/", () => {    callStack.push(3);  });  router.options("/", () => {    callStack.push(4);  });  router.patch("/", () => {    callStack.push(5);  });  router.post("/", () => {    callStack.push(6);  });  router.put("/", () => {    callStack.push(7);  });  const mw = router.routes();  await mw(context, next);  assertEquals(callStack, [0, 7]);});