Fennec/fennec-be
1
0
Fork 0
Code Issues 1 Pull Requests Projects Releases Wiki Activity

test(pipeline-tasks): 更新测试用例。

Browse Source
This commit is contained in:
Ivan 2021-03-08 10:48:46 +08:00
parent f00f75673b
commit ba0ba46a35
1 changed files with 18 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
src/pipeline-tasks/pipeline-tasks.service.spec.ts
View File

@ -155,20 +155,31 @@ describe('PipelineTasksService', () => {
it('pipeline is busy', async () => { it('pipeline is busy', async () => {
let remainTimes = 3; let remainTimes = 3;
const incr = jest let lckValue: string;
.spyOn(redisClient, 'incr') const set = jest
.mockImplementation(() => remainTimes--); .spyOn(redisClient, 'set')
.mockImplementation(async (...args) => {
if (remainTimes-- > 0) {
throw new Error();
} else {
lckValue = args[3] as string;
}
});
const get = jest
.spyOn(redisClient, 'get')
.mockImplementation(async () => lckValue);
const del = jest.spyOn(redisClient, 'del');
const rpop = jest.spyOn(redisClient, 'rpop'); const rpop = jest.spyOn(redisClient, 'rpop');
const decr = jest.spyOn(redisClient, 'decr');
const add = jest.spyOn(taskQueue, 'add'); const add = jest.spyOn(taskQueue, 'add');
await service.doNextTask(getBasePipeline()); await service.doNextTask(getBasePipeline());
expect(rpop).toHaveBeenCalledTimes(1); expect(rpop).toHaveBeenCalledTimes(1);
expect(incr).toHaveBeenCalledTimes(3); expect(set).toHaveBeenCalledTimes(4);
expect(decr).toHaveBeenCalledTimes(3); expect(get).toHaveBeenCalledTimes(1);
expect(del).toHaveBeenCalledTimes(1);
expect(add).toHaveBeenCalledWith(getTask()); expect(add).toHaveBeenCalledWith(getTask());
}); }, 10_000);
it('pipeline always busy and timeout', async () => { it('pipeline always busy and timeout', async () => {
const set = jest const set = jest
.spyOn(redisClient, 'set') .spyOn(redisClient, 'set')