Fixed a few issues

Author: rpanic

packages/indexer/test/IndexerNotifier.test.ts

@@ -20,7 +20,12 @@ import {
   StateServiceQueryModule,
   TestingAppChain,
 } from "@proto-kit/sdk";
-import { LocalTaskQueue, Sequencer, TaskPayload } from "@proto-kit/sequencer";
+import {
+  LocalTaskQueue,
+  Sequencer,
+  TaskPayload,
+  VanillaTaskWorkerModules,
+} from "@proto-kit/sequencer";
 
 import { IndexerNotifier, IndexBlockTaskParametersSerializer } from "../src";
 
@@ -89,16 +94,7 @@ function createAppChain() {
       BlockTrigger: {},
       Mempool: {},
       BlockProducerModule: {},
-      LocalTaskWorkerModule: {
-        StateTransitionReductionTask: {},
-        StateTransitionTask: {},
-        RuntimeProvingTask: {},
-        BlockBuildingTask: {},
-        BlockProvingTask: {},
-        BlockReductionTask: {},
-        CircuitCompilerTask: {},
-        WorkerRegistrationTask: {},
-      },
+      LocalTaskWorkerModule: VanillaTaskWorkerModules.defaultConfig(),
       BaseLayer: {},
       BatchProducerModule: {},
       TaskQueue: {
@@ -176,7 +172,7 @@ describe("IndexerNotifier", () => {
 
     await appChain.start();
     await sendTransactions(appChain, 2);
-  });
+  }, 20000);
 
   it("should create a task for every unproven block produced", async () => {
     const { block } = container

packages/persistance/test-integration/utils.ts

@@ -146,16 +146,7 @@ export function createPrismaAppchain(
       BlockTrigger: {},
       Mempool: {},
       BatchProducerModule: {},
-      LocalTaskWorkerModule: {
-        StateTransitionTask: {},
-        RuntimeProvingTask: {},
-        StateTransitionReductionTask: {},
-        BlockReductionTask: {},
-        BlockProvingTask: {},
-        BlockBuildingTask: {},
-        CircuitCompilerTask: {},
-        WorkerRegistrationTask: {},
-      },
+      LocalTaskWorkerModule: VanillaTaskWorkerModules.defaultConfig(),
       BaseLayer: {},
       BlockProducerModule: {},
       TaskQueue: {

packages/sequencer/src/sequencer/executor/Sequencer.ts

@@ -113,12 +113,12 @@ export class Sequencer<Modules extends SequencerModulesRecord>
     // and start the modules in the order they were resolved.
     for (const moduleName of orderedModules) {
       const sequencerModule = this.resolve(moduleName);
-      // eslint-disable-next-line no-await-in-loop
-      await sequencerModule.start();
 
       log.info(
         `Starting sequencer module ${moduleName} (${sequencerModule.constructor.name})`
       );
+      // eslint-disable-next-line no-await-in-loop
+      await sequencerModule.start();
     }
     if (!moduleClassNames.includes("SequencerStartupModule")) {
       log.warn("SequencerStartupModule is not defined.");

packages/sequencer/test/integration/BlockProduction.test.ts

@@ -24,7 +24,12 @@ import { Bool, Field, PrivateKey, PublicKey, Struct, UInt64 } from "o1js";
 import "reflect-metadata";
 import { container } from "tsyringe";
 
-import { BatchStorage, HistoricalBatchStorage, Sequencer } from "../../src";
+import {
+  BatchStorage,
+  HistoricalBatchStorage,
+  Sequencer,
+  VanillaTaskWorkerModules,
+} from "../../src";
 import {
   DefaultTestingSequencerModules,
   testingSequencerFromModules,
@@ -133,7 +138,7 @@ describe("block production", () => {
         Mempool: {},
         BatchProducerModule: {},
         BlockProducerModule: {},
-        LocalTaskWorkerModule: {},
+        LocalTaskWorkerModule: VanillaTaskWorkerModules.defaultConfig(),
         BaseLayer: {},
         TaskQueue: {},
         FeeStrategy: {},

packages/sequencer/test/integration/BlockProductionSize.test.ts

@@ -7,7 +7,12 @@ import { Bool, PrivateKey, Struct, UInt64 } from "o1js";
 import "reflect-metadata";
 import { container } from "tsyringe";
 
-import { ManualBlockTrigger, PrivateMempool, Sequencer } from "../../src";
+import {
+  ManualBlockTrigger,
+  PrivateMempool,
+  Sequencer,
+  VanillaTaskWorkerModules,
+} from "../../src";
 import {
   DefaultTestingSequencerModules,
   testingSequencerFromModules,
@@ -76,11 +81,12 @@ describe("block limit", () => {
         BlockProducerModule: {
           maximumBlockSize: maxBlockSize,
         },
-        LocalTaskWorkerModule: {},
+        LocalTaskWorkerModule: VanillaTaskWorkerModules.defaultConfig(),
         BaseLayer: {},
         TaskQueue: {},
         FeeStrategy: {},
         ProtocolStartupModule: {},
+        SequencerStartupModule: {},
       },
       Runtime: {
         Balance: {},

packages/sequencer/test/integration/Mempool.test.ts

@@ -13,6 +13,7 @@ import {
   Sequencer,
   SequencerModule,
   StorageDependencyFactory,
+  VanillaTaskWorkerModules,
 } from "../../src";
 import {
   DefaultTestingSequencerModules,
@@ -109,7 +110,7 @@ describe.each([["InMemory", InMemoryDatabase]])(
           FeeStrategy: {},
           BatchProducerModule: {},
           BlockProducerModule: {},
-          LocalTaskWorkerModule: {},
+          LocalTaskWorkerModule: VanillaTaskWorkerModules.defaultConfig(),
           BaseLayer: {},
           TaskQueue: {},
           SequencerStartupModule: {},

packages/sequencer/test/integration/Proven.test.ts

@@ -26,6 +26,7 @@ import {
   ProvenSettlementPermissions,
   SettlementModule,
   SettlementProvingTask,
+  VanillaTaskWorkerModules,
   WithdrawalQueue,
 } from "../../src";
 import { SettlementStartupModule } from "../../src/sequencer/SettlementStartupModule";
@@ -36,7 +37,7 @@ import { ProvenBalance } from "./mocks/ProvenBalance";
 
 const timeout = 300000;
 
-describe("Proven", () => {
+describe.skip("Proven", () => {
   let test: BlockTestService;
 
   let appChain: AppChain<any, any, any, any>;
@@ -95,7 +96,7 @@ describe("Proven", () => {
           Mempool: {},
           BatchProducerModule: {},
           BlockProducerModule: {},
-          LocalTaskWorkerModule: {},
+          LocalTaskWorkerModule: VanillaTaskWorkerModules.defaultConfig(),
           TaskQueue: {},
           FeeStrategy: {},
           SequencerStartupModule: {},

packages/sequencer/test/integration/StorageIntegration.test.ts

@@ -19,6 +19,7 @@ import {
   StateRecord,
   StorageDependencyFactory,
   BlockStorage,
+  VanillaTaskWorkerModules,
 } from "../../src";
 import {
   DefaultTestingSequencerModules,
@@ -104,7 +105,7 @@ describe.each([["InMemory", InMemoryDatabase]])(
           Mempool: {},
           BatchProducerModule: {},
           BlockProducerModule: {},
-          LocalTaskWorkerModule: {},
+          LocalTaskWorkerModule: VanillaTaskWorkerModules.defaultConfig(),
           BaseLayer: {},
           TaskQueue: {},
           FeeStrategy: {},
@@ -119,7 +120,7 @@ describe.each([["InMemory", InMemoryDatabase]])(
         },
       });
 
-      await appChain.start();
+      await appChain.start(false);
 
       runtime = appChain.runtime;
       sequencer = appChain.sequencer;

packages/sequencer/test/protocol/production/sequencing/atomic-block-production.test.ts

@@ -11,6 +11,7 @@ import {
   BlockQueue,
   ManualBlockTrigger,
   TransactionExecutionService,
+  VanillaTaskWorkerModules,
 } from "../../../../src";
 import { ProtocolStateTestHook } from "../../../integration/mocks/ProtocolStateTestHook";
 import {
@@ -57,7 +58,7 @@ describe("atomic block production", () => {
         Mempool: {},
         BatchProducerModule: {},
         BlockProducerModule: {},
-        LocalTaskWorkerModule: {},
+        LocalTaskWorkerModule: VanillaTaskWorkerModules.defaultConfig(),
         BaseLayer: {},
         TaskQueue: {},
         FeeStrategy: {},

packages/sequencer/test/settlement/Settlement.ts

@@ -55,6 +55,7 @@ import {
   MinaBaseLayerConfig,
   SignedSettlementPermissions,
   ProvenSettlementPermissions,
+  VanillaTaskWorkerModules,
 } from "../../src";
 import { BlockProofSerializer } from "../../src/protocol/production/tasks/serializers/BlockProofSerializer";
 import { testingSequencerFromModules } from "../TestingSequencer";
@@ -168,7 +169,7 @@ export const settlementTestFn = (
           BlockTrigger: {},
           Mempool: {},
           BatchProducerModule: {},
-          LocalTaskWorkerModule: {},
+          LocalTaskWorkerModule: VanillaTaskWorkerModules.defaultConfig(),
           OutgoingMessageQueue: {},
           BaseLayer: baseLayerConfig,
           BlockProducerModule: {},