diff --git a/__tests__/fixtures/context.ts b/__tests__/fixtures/context.ts index cd1501d5..239114c3 100644 --- a/__tests__/fixtures/context.ts +++ b/__tests__/fixtures/context.ts @@ -3,11 +3,19 @@ import { PluginContext } from "rollup"; import { IContext } from "../../src/context"; +// if given a function, make sure to call it (for code coverage etc) +function returnText (message: string | (() => string)) { + if (typeof message === "string") + return message; + + return message(); +} + export function makeContext(): PluginContext & IContext { return { - error: jest.fn(), - warn: jest.fn(), - info: jest.fn(), - debug: jest.fn(), + error: jest.fn(returnText), + warn: jest.fn(returnText), + info: jest.fn(returnText), + debug: jest.fn(returnText), } as unknown as PluginContext & IContext; };