Pass test gossipHandlers

ChainSafe · Mar 1, 2023 · 0e803d0 · 0e803d0
1 parent 86450f8
commit 0e803d0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/packages/beacon-node/src/network/network.ts b/packages/beacon-node/src/network/network.ts
@@ -151,6 +151,7 @@ export class Network implements INetwork {
     chain,
     reqRespHandlers,
     signal,
+    gossipHandlers,
   }: NetworkInitModules): Promise<Network> {
     const clock = chain.clock;
     const peersData = new PeersData();
@@ -229,7 +230,7 @@ export class Network implements INetwork {
     );
 
     const networkProcessor = new NetworkProcessor(
-      {attnetsService, chain, config, logger, metrics, peerRpcScores, events: networkEventBus},
+      {attnetsService, chain, config, logger, metrics, peerRpcScores, events: networkEventBus, gossipHandlers},
       opts
     );
 

diff --git a/packages/beacon-node/src/network/processor/worker.ts b/packages/beacon-node/src/network/processor/worker.ts
@@ -1,7 +1,7 @@
 import {IBeaconChain} from "../../chain/interface.js";
 import {Metrics} from "../../metrics/metrics.js";
 import {NetworkEvent, NetworkEventBus} from "../events.js";
-import {GossipValidatorFn} from "../gossip/interface.js";
+import {GossipHandlers, GossipValidatorFn} from "../gossip/interface.js";
 import {getGossipHandlers, GossipHandlerOpts, ValidatorFnsModules} from "./gossipHandlers.js";
 import {getGossipValidatorFn, ValidatorFnModules} from "./gossipValidatorFn.js";
 import {PendingGossipsubMessage} from "./types.js";
@@ -11,6 +11,8 @@ export type NetworkWorkerModules = ValidatorFnsModules &
     chain: IBeaconChain;
     events: NetworkEventBus;
     metrics: Metrics | null;
+    // Optionally pass custom GossipHandlers, for testing
+    gossipHandlers?: GossipHandlers;
   };
 
 export class NetworkWorker {
@@ -21,7 +23,7 @@ export class NetworkWorker {
   constructor(modules: NetworkWorkerModules, opts: GossipHandlerOpts) {
     this.events = modules.events;
     this.metrics = modules.metrics;
-    this.gossipValidatorFn = getGossipValidatorFn(getGossipHandlers(modules, opts), modules);
+    this.gossipValidatorFn = getGossipValidatorFn(modules.gossipHandlers ?? getGossipHandlers(modules, opts), modules);
   }
 
   async processPendingGossipsubMessage(message: PendingGossipsubMessage): Promise<void> {