chore: update protocol health for lightpush

waku-org · Oct 9, 2024 · e84eb62 · e84eb62
1 parent 829d183
commit e84eb62
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/packages/sdk/src/protocols/light_push/light_push.ts b/packages/sdk/src/protocols/light_push/light_push.ts
@@ -1,5 +1,10 @@
 import type { Peer, PeerId } from "@libp2p/interface";
-import { ConnectionManager, LightPushCodec, LightPushCore } from "@waku/core";
+import {
+  ConnectionManager,
+  getHealthManager,
+  LightPushCodec,
+  LightPushCore
+} from "@waku/core";
 import {
   Failure,
   type IEncoder,
@@ -113,6 +118,11 @@ class LightPush extends BaseProtocolSDK implements ILightPush {
       }
     }
 
+    getHealthManager().updateProtocolHealth(
+      this.protocol.multicodec,
+      successes.length
+    );
+
     return {
       successes,
       failures

diff --git a/packages/tests/tests/health-manager/protocols.spec.ts b/packages/tests/tests/health-manager/protocols.spec.ts
@@ -17,7 +17,7 @@ import {
 
 const NUM_NODES = [0, 1, 2, 3];
 
-describe("Health Manager", function () {
+describe.only("Health Manager", function () {
   this.timeout(10_000);
 
   let waku: LightNode;