123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134
import * as assert from 'assert';import { hooks, middleware, HookContext, NextFunction } from '../src';
interface Dummy {  sayHi (name: string): Promise<string>;  addOne (number: number): Promise<number>;}
describe('class objectHooks', () => {  let DummyClass: new () => Dummy;
  beforeEach(() => {    DummyClass = class DummyClass implements Dummy {      async sayHi (name: string) {        return `Hi ${name}`;      }
      async addOne (number: number) {        return number + 1;      }    };  });

  it('hooking object on class adds to the prototype', async () => {    hooks(DummyClass, {      sayHi: middleware([        async (ctx: HookContext, next: NextFunction) => {          assert.deepStrictEqual(ctx, new DummyClass.prototype.sayHi.Context({            arguments: ['David'],            method: 'sayHi',            name: 'David',            self: instance          }));
          await next();
          ctx.result += '?';        }      ]).params('name'),
      addOne: middleware([        async (ctx: HookContext, next: NextFunction) => {          ctx.arguments[0] += 1;
          await next();        }      ])    });
    const instance = new DummyClass();
    assert.strictEqual(await instance.sayHi('David'), 'Hi David?');    assert.strictEqual(await instance.addOne(1), 3);  });
  it('works with inheritance', async () => {    const first = async (ctx: HookContext, next: NextFunction) => {      assert.deepStrictEqual(ctx, new (OtherDummy.prototype.sayHi as any).Context({        arguments: [ 'David' ],        method: 'sayHi',        self: instance      }));
      await next();
      ctx.result += '?';    };    const second = async (ctx: HookContext, next: NextFunction) => {      await next();
      ctx.result += '!';    };
    hooks(DummyClass, {      sayHi: middleware([first])    });
    class OtherDummy extends DummyClass {}
    hooks(OtherDummy, {      sayHi: middleware([second])    });
    const instance = new OtherDummy();
    assert.strictEqual(await instance.sayHi('David'), 'Hi David!?');  });
  it('works with multiple context updaters', async () => {    hooks(DummyClass, {      sayHi: middleware([        async (ctx, next) => {          assert.strictEqual(ctx.name, 'Dave');
          ctx.name = 'Changed';
          await next();        }      ]).params('name')    });
    class OtherDummy extends DummyClass {    }
    hooks(OtherDummy, {      sayHi: middleware([        async (ctx, next) => {          assert.strictEqual(ctx.name, 'Changed');          assert.strictEqual(ctx.gna, 42);
          await next();        }      ]).props({ gna: 42 })    });
    const instance = new OtherDummy();
    hooks(instance, {      sayHi: middleware([        async (ctx, next) => {          assert.strictEqual(ctx.name, 'Changed');          assert.strictEqual(ctx.gna, 42);          assert.strictEqual(ctx.app, 'ok');
          await next();        }      ]).props({ app: 'ok' })    });
    assert.strictEqual(await instance.sayHi('Dave'), 'Hi Changed');  });});