132 lines
3.6 KiB
TypeScript
132 lines
3.6 KiB
TypeScript
/*
|
|
* Copyright 2025 coze-dev Authors
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
import { type Ctxs, type IPlugin, Program, on, before, after } from '../src';
|
|
|
|
interface Ctx {
|
|
count: number;
|
|
}
|
|
interface Hook extends Ctxs {
|
|
add: Ctx;
|
|
reduce: Ctx;
|
|
acc: { text: string };
|
|
}
|
|
|
|
class AddPlugin implements IPlugin {
|
|
apply(program: Program<Hook>): void {
|
|
program.register(on('add'), ctx => {
|
|
ctx.count += 1;
|
|
return ctx;
|
|
});
|
|
}
|
|
}
|
|
class ReducePlugin implements IPlugin {
|
|
apply(program: Program<Hook>): void {
|
|
program.register(on('reduce'), ctx => {
|
|
ctx.count -= 1;
|
|
return ctx;
|
|
});
|
|
}
|
|
}
|
|
|
|
describe('Program', () => {
|
|
it('should apply plugin', () => {
|
|
const program = Program.create<Hook>([new AddPlugin(), new ReducePlugin()]);
|
|
expect(program.trigger('add', { count: 0 }).count).toEqual(1);
|
|
expect(program.trigger('reduce', { count: 0 }).count).toEqual(-1);
|
|
});
|
|
it('should apply plugin by priority', () => {
|
|
class FirstPlugin implements IPlugin {
|
|
apply(program: Program<Hook>): void {
|
|
program.register(
|
|
on('reduce'),
|
|
ctx => {
|
|
ctx.count = 1;
|
|
return ctx;
|
|
},
|
|
0,
|
|
);
|
|
}
|
|
}
|
|
class SecondPlugin implements IPlugin {
|
|
apply(program: Program<Hook>): void {
|
|
program.register(
|
|
on('reduce'),
|
|
ctx => {
|
|
ctx.count = 2;
|
|
return ctx;
|
|
},
|
|
1,
|
|
);
|
|
}
|
|
}
|
|
const program = Program.create<Hook>([
|
|
new SecondPlugin(),
|
|
new FirstPlugin(),
|
|
]);
|
|
expect(program.trigger('reduce', { count: 0 }).count).toEqual(2);
|
|
});
|
|
it('should apply plugin by order: before -> on -> after', () => {
|
|
class Plugin implements IPlugin {
|
|
apply(program: Program<Hook>): void {
|
|
program.register(before('acc'), ctx => {
|
|
ctx.text += 'before';
|
|
return ctx;
|
|
});
|
|
program.register(on('acc'), ctx => {
|
|
ctx.text += '_on';
|
|
return ctx;
|
|
});
|
|
program.register(after('acc'), ctx => {
|
|
ctx.text += '_after';
|
|
return ctx;
|
|
});
|
|
}
|
|
}
|
|
const program = Program.create<Hook>([new Plugin()]);
|
|
expect(program.trigger('acc', { text: '' }).text).toEqual(
|
|
'before_on_after',
|
|
);
|
|
});
|
|
it('should load plugin be fine', () => {
|
|
class Plugin implements IPlugin {
|
|
apply(program: Program<Hook>): void {
|
|
program.register(before('acc'), ctx => {
|
|
ctx.text += 'before';
|
|
return ctx;
|
|
});
|
|
program.register(on('acc'), ctx => {
|
|
ctx.text += '_on';
|
|
return ctx;
|
|
});
|
|
program.register(after('acc'), ctx => {
|
|
ctx.text += '_after';
|
|
return ctx;
|
|
});
|
|
}
|
|
}
|
|
const program = Program.create<Hook>([]);
|
|
program.loadPlugins([new Plugin()]);
|
|
expect(program.trigger('acc', { text: '' }).text).toEqual(
|
|
'before_on_after',
|
|
);
|
|
});
|
|
it('should throw error if has no register', () => {
|
|
const program = Program.create<Hook>([]);
|
|
expect(() => program.trigger('acc', { text: '' })).toThrowError();
|
|
});
|
|
});
|