durabull/tests/integration/query.integration.test.ts

A durable workflow engine built on top of BullMQ and Redis

import {
  Durabull,
  WorkflowStub,
  Workflow,
  QueryMethod,
  startWorkflowWorker,
} from '../../src';
import { Worker } from 'bullmq';

jest.setTimeout(60000);

let workflowWorker: Worker;
let durabull: Durabull;
const testQueuePrefix = `test-query-${Date.now()}`;

beforeAll(async () => {
  durabull = new Durabull({
    redisUrl: process.env.REDIS_URL || 'redis://redis:6379',
    queues: {
      workflow: `${testQueuePrefix}-workflow`,
      activity: `${testQueuePrefix}-activity`,
    },
  });

  durabull.setActive();
  workflowWorker = startWorkflowWorker(durabull);
  await new Promise((resolve) => setTimeout(resolve, 1000));
});

afterAll(async () => {
  if (workflowWorker) {
    await workflowWorker.close();
  }
  await new Promise((resolve) => setTimeout(resolve, 1000));
  
  // eslint-disable-next-line @typescript-eslint/no-var-requires
  const { closeQueues } = require('../../src/queues');
  // eslint-disable-next-line @typescript-eslint/no-var-requires
  const { closeStorage } = require('../../src/runtime/storage');
  await closeQueues();
  await closeStorage();
});

describe('Integration: Workflow Queries', () => {
  class QueryTestWorkflow extends Workflow<[], string> {
    private count = 0;

    @QueryMethod()
    getCount(): number {
      return this.count;
    }

    // eslint-disable-next-line require-yield
    async *execute(): AsyncGenerator<unknown, string, unknown> {
      this.count = 10;
      await WorkflowStub.timer(1);
      this.count = 20;
      return 'done';
    }
  }

  beforeAll(() => {
    durabull.registerWorkflow('QueryTestWorkflow', QueryTestWorkflow);
  });

  it('should query workflow state during execution', async () => {
    const wf = await WorkflowStub.make('QueryTestWorkflow');
    await wf.start();

    await new Promise((resolve) => setTimeout(resolve, 500));

    // eslint-disable-next-line @typescript-eslint/no-explicit-any
    const count1 = await wf.query<any>(QueryTestWorkflow).getCount();
    expect(count1).toBe(10);

    await new Promise((resolve) => setTimeout(resolve, 1500));

    // eslint-disable-next-line @typescript-eslint/no-explicit-any
    const count2 = await wf.query<any>(QueryTestWorkflow).getCount();
    expect(count2).toBe(20);
    
    expect(await wf.output()).toBe('done');
  });
});