如何在返回带有上下文参数的函数的函数中模拟 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。createContext如果你想尝试自己构建它，他们会在底层使用 Koa 。

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);
    });
});