Function createJobTester

packages/testing/src/index.ts:110–213

(vi: typeof vitestVi)

export const createJobTester =
  (vi: typeof vitestVi) =>
  async <
    TTrigger extends Trigger<EventSpecification<any>>,
    TIntegrations extends Record<string, TriggerIntegration> = {},
    TTasks extends Record<string, any> = {},
  >(
    job: Job<TTrigger, TIntegrations>,
    opts: {
      payload?: TriggerEventType<TTrigger>;
      tasks?: TTasks;
    } = {}
  ): Promise<
    {
      output: any;
      status: string;
      tasks: Record<keyof TTasks, Mock> & Record<string, Mock>;
    } & Record<string, any>
  > => {
    const mockSendEvent = (client: TriggerClient) =>
      vi
        .spyOn(client, "sendEvent")
        .mockImplementation(
          async (unparsedEvent: SendEvent, unparsedOptions: SendEventOptions = {}) => {
            const body = SendEventBodySchema.parse({
              event: unparsedEvent,
              options: unparsedOptions,
            });
            const { event, options } = body;

            const timestamp = new Date();
            const deliverAt =
              options?.deliverAt ||
              (options?.deliverAfter
                ? new Date(Date.now() + options.deliverAfter * 1000)
                : undefined);

            const eventLog = {
              id: event.id,
              name: event.name,
              payload: event.payload ?? {},
              context: event.context,
              timestamp,
              deliverAt,
              deliveredAt: !deliverAt ? timestamp : undefined,
              cancelledAt: undefined,
            };

            return eventLog;
          }
        );

    const { client, trigger } = job;

    mockSendEvent(client!);
    const eventLog = await client!.sendEvent({
      name: typeof trigger.event.name === "string" ? trigger.event.name : trigger.event.name[0],
      payload: opts.payload,
    });