IndexBlockTask.test.ts

import "reflect-metadata";
import {
  BlockWithResult,
  InMemoryDatabase,
  LocalTaskQueue,
  LocalTaskWorkerModule,
  TaskPayload,
} from "@proto-kit/sequencer";

import { Indexer } from "../src/Indexer";
import { IndexBlockTask } from "../src/tasks/IndexBlockTask";

describe("IndexBlockTask", () => {
  const indexer = Indexer.from({
    Database: InMemoryDatabase,
    TaskQueue: LocalTaskQueue,
    LocalTaskWorkerModule: LocalTaskWorkerModule.from({
      IndexBlockTask: IndexBlockTask,
    }),
  });

  indexer.configurePartial({
    Database: {},
    TaskQueue: {},
    LocalTaskWorkerModule: {
      IndexBlockTask: {},
    },
  });

  it("should listen to block indexing tasks", async () => {
    await indexer.start();
    const taskQueue = indexer.resolve("TaskQueue");
    const localTaskWorker = indexer.resolve("LocalTaskWorkerModule");
    const indexBlockTask = localTaskWorker.resolve("IndexBlockTask");
    const queue = await taskQueue.getQueue(indexBlockTask.name);
    const block = BlockWithResult.createEmpty();

    const payload = await indexBlockTask.inputSerializer().toJSON(block);

    const task: TaskPayload = {
      name: indexBlockTask.name,
      payload,
      flowId: "",
      sequencerId: "test-sequencer",
    };

    await queue.addTask(task);

    // LocalTaskQueue voids all the pending promises, so we need this hack
    await new Promise<void>((resolve) => {
      setTimeout(resolve, 1000);
    });

    const storage = indexer.resolve("BlockStorage");
    const latestBlock = await storage.getLatestBlock();

    expect(latestBlock?.block.hash).toBe(block.block.hash);
  });
});