123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129
// Copyright 2018-2024 the oak authors. All rights reserved. MIT license.
// deno-lint-ignore-file
import { assertEquals, assertStrictEquals } from "./test_deps.ts";import { assert, errors } from "./deps.ts";import { createMockContext } from "./testing.ts";import {  compose,  isMiddlewareObject,  type Middleware,  type MiddlewareObject,  type Next,} from "./middleware.ts";import { Context } from "./context.ts";
Deno.test({  name: "test compose()",  async fn() {    const callStack: number[] = [];    const mockContext = createMockContext();    const mw1: Middleware = async (context, next) => {      assertStrictEquals(context, mockContext);      assertEquals(typeof next, "function");      callStack.push(1);      await next();    };    const mw2: Middleware = async (context, next) => {      assertStrictEquals(context, mockContext);      assertEquals(typeof next, "function");      callStack.push(2);      await next();    };    await compose([mw1, mw2])(mockContext);    assertEquals(callStack, [1, 2]);  },});
Deno.test({  name: "isMiddlewareObject()",  async fn() {    class MockMiddlewareObject implements MiddlewareObject {      handleRequest(        _context: Context<Record<string, any>, Record<string, any>>,        _next: Next,      ): unknown {        return;      }    }
    assert(isMiddlewareObject(new MockMiddlewareObject()));    assert(isMiddlewareObject({ handleRequest() {} }));    assert(!isMiddlewareObject(function () {}));  },});
Deno.test({  name: "middleware objects are composed correctly",  async fn() {    const callStack: number[] = [];    const mockContext = createMockContext();
    class MockMiddlewareObject implements MiddlewareObject {      #counter = 0;
      async handleRequest(_context: any, next: Next) {        assertEquals(typeof next, "function");        callStack.push(this.#counter++);        await next();      }    }
    const mwo = new MockMiddlewareObject();    const fn = compose([mwo]);
    await fn(mockContext);    await fn(mockContext);    assertEquals(callStack, [0, 1]);  },});
Deno.test({  name: "next() is catchable",  async fn() {    let caught: any;    const mw1: Middleware = async (ctx, next) => {      try {        await next();      } catch (err) {        caught = err;      }    };    const mw2: Middleware = async (ctx) => {      ctx.throw(500);    };    const context = createMockContext();    await compose([mw1, mw2])(context);    assert(caught instanceof errors.InternalServerError);  },});
Deno.test({  name: "composed middleware accepts next middleware",  async fn() {    const callStack: number[] = [];    const mockContext = createMockContext();
    const mw0: Middleware = async (context, next): Promise<void> => {      assertEquals(typeof next, "function");      callStack.push(3);      await next();    };
    const mw1: Middleware = async (context, next) => {      assertEquals(typeof next, "function");      callStack.push(1);      await next();    };    const mw2: Middleware = async (context, next) => {      assertEquals(typeof next, "function");      callStack.push(2);      await next();    };
    await compose([mw1, mw2])(mockContext, mw0 as () => Promise<void>);    assertEquals(callStack, [1, 2, 3]);  },});