honion/test/middleware/pipeline.test.ts

通用洋葱模型中间件

import { Middleware } from "../../src";
import { TestHonion } from "../test-honion";

test("middleware pipeline", async () => {
  const honion = new TestHonion()
    .add(() => new Mdw1())
    .add(() => new Mdw2())
    .add(() => new Mdw3())
    .add(() => new Mdw4());

  const ctx = await honion.run();
  expect(ctx.get("ctx")).toBe("OK");
  expect(ctx.get("mdw1")).toBe("mdw1");
  expect(ctx.get("mdw2")).toBe("mdw2");
  expect(!!ctx.get("mdw2")).toBeTruthy();
  expect(!!ctx.get("mdw3")).toBeFalsy();
  expect(!!ctx.get("mdw4")).toBeFalsy();
});

class Mdw1 extends Middleware {
  async invoke(): Promise<void> {
    this.ctx.set("mdw1", "mdw1");
    await this.next();
  }
}

class Mdw2 extends Middleware {
  async invoke(): Promise<void> {
    this.ctx.set("mdw2", "mdw2");
    await this.next();
  }
}

class Mdw3 extends Middleware {
  async invoke(): Promise<void> {
    this.ctx.set("ctx", "OK");
  }
}

class Mdw4 extends Middleware {
  async invoke(): Promise<void> {
    this.ctx.set("mdw4", "mdw4");
    await this.next();
  }
}