Bull Mock

bull.ts

import { ObjectId } from 'mongodb';

import type Bull from 'bull';

const fakeJob = (data, opts): Bull.Job => {
  const dataSerialized = JSON.parse(JSON.stringify(data));
  const job = {
    data: dataSerialized,
    opts,
    id: opts?.jobId || new ObjectId().toHexString(),
    progress: jest.fn(),
    attemptsMade: 0,
  } as unknown as Bull.Job;
  return job;
};

export default class CustomQueue {
  name: string

  constructor(name) {
    this.name = name;
  }

  processFn: Bull.Queue['process']

  process = (fn) => {
    console.log(`Registered function ${this.name}`);
    this.processFn = fn;
  }

  add: Bull.Queue['add'] = async (data, opts) => {
    const job = fakeJob(data, opts);
    console.log(`Running ${this.name}`, data);
    // eslint-disable-next-line
    // @ts-ignore
    const results = await this.processFn(job);
    console.log(`Done running ${this.name}`, results);
    return job;
  }

  removeJobs = jest.fn();

  // eslint-disable-next-line
  // @ts-ignore
  addBulk: Bull.Queue['addBulk'] = async (array) => {
    console.log(`AddBulk ${this.name}`);
    // eslint-disable-next-line
    // @ts-ignore
    const result = await Promise.all(array.map(async ({ data, opts }) => {
      const job = fakeJob(data, opts);
      // eslint-disable-next-line
    // @ts-ignore
      await this.processFn(job);
      return job;
    }));
    console.log(`addBulk Done: ${this.name}`);

    return result;
  }
}