testing - 如何在返回带有上下文参数的函数的函数中模拟 koa 上下文

Question

我有一个文件 myFunction.ts，它定义并导出了一个函数:

export function MyFunction() {
    return async (ctx: Koa.Context) => {
        //some work happens in here
    };
}

该函数在 Koa 中间件文件中被调用，并继承了之前中间件的 ctx。在这种情况下，如何创建模拟 ctx 来测试 MyFunction()？

Answer 1

我使用 Shopify 的库 @shopify/jest-koa-mocks .如果您想自己探索构建它，他们会在后台使用 Koa 的 createContext。

import { createMockContext } from '@shopify/jest-koa-mocks';
import { myFunction } from './myFunction.js';

describe('middleware being tested', () => {

    let ctx;
    let myMiddleware;

    beforeEach(() => {
        ctx = createMockContext();
        ctx.addAnythingYouWant = jest.fn();
        myMiddleware = myFunction();
    });

    test('calls function addAnythingYouWant', async () => {
        await myMiddleware(ctx);
        expect(ctx.addAnythingYouWant).toHaveBeenCalledTimes(1);
    });
});