From a9df5921a1b5585db728108a85ed5c166b9b152d Mon Sep 17 00:00:00 2001
From: Julien Eluard <julien.eluard@gmail.com>
Date: Mon, 5 Feb 2024 14:49:47 +0100
Subject: [PATCH] chore: added tests

---
 .../test/e2e/network/gossipsub.test.ts        | 64 +++++++++++++++++++
 1 file changed, 64 insertions(+)

diff --git a/packages/beacon-node/test/e2e/network/gossipsub.test.ts b/packages/beacon-node/test/e2e/network/gossipsub.test.ts
index c7b3dbefbe77..fd1794bf549d 100644
--- a/packages/beacon-node/test/e2e/network/gossipsub.test.ts
+++ b/packages/beacon-node/test/e2e/network/gossipsub.test.ts
@@ -138,6 +138,70 @@ function runTests({useWorker}: {useWorker: boolean}): void {
     );
   });
 
+  it("Publish and receive an attesterSlashing", async function () {
+    let onAttesterSlashingChange: (payload: Uint8Array) => void;
+    const onAttesterSlashingChangePromise = new Promise<Uint8Array>((resolve) => (onAttesterSlashingChange = resolve));
+
+    const {netA, netB} = await mockModules({
+      [GossipType.attester_slashing]: async ({gossipData}: GossipHandlerParamGeneric<GossipType.attester_slashing>) => {
+        onAttesterSlashingChange(gossipData.serializedData);
+      },
+    });
+
+    await Promise.all([onPeerConnect(netA), onPeerConnect(netB), connect(netA, netB)]);
+    expect(netA.getConnectedPeerCount()).toBe(1);
+    expect(netB.getConnectedPeerCount()).toBe(1);
+
+    await netA.subscribeGossipCoreTopics();
+    await netB.subscribeGossipCoreTopics();
+
+    // Wait to have a peer connected to a topic
+    while (!netA.closed) {
+      await sleep(500);
+      if (await hasSomeMeshPeer(netA)) {
+        break;
+      }
+    }
+
+    const attesterSlashing = ssz.phase0.AttesterSlashing.defaultValue();
+    await netA.publishAttesterSlashing(attesterSlashing);
+
+    const received = await onAttesterSlashingChangePromise;
+    expect(Buffer.from(received)).toEqual(Buffer.from(ssz.phase0.AttesterSlashing.serialize(attesterSlashing)));
+  });
+
+  it("Publish and receive a proposerSlashing", async function () {
+    let onProposerSlashingChange: (payload: Uint8Array) => void;
+    const onProposerSlashingChangePromise = new Promise<Uint8Array>((resolve) => (onProposerSlashingChange = resolve));
+
+    const {netA, netB} = await mockModules({
+      [GossipType.proposer_slashing]: async ({gossipData}: GossipHandlerParamGeneric<GossipType.proposer_slashing>) => {
+        onProposerSlashingChange(gossipData.serializedData);
+      },
+    });
+
+    await Promise.all([onPeerConnect(netA), onPeerConnect(netB), connect(netA, netB)]);
+    expect(netA.getConnectedPeerCount()).toBe(1);
+    expect(netB.getConnectedPeerCount()).toBe(1);
+
+    await netA.subscribeGossipCoreTopics();
+    await netB.subscribeGossipCoreTopics();
+
+    // Wait to have a peer connected to a topic
+    while (!netA.closed) {
+      await sleep(500);
+      if (await hasSomeMeshPeer(netA)) {
+        break;
+      }
+    }
+
+    const proposerSlashing = ssz.phase0.ProposerSlashing.defaultValue();
+    await netA.publishProposerSlashing(proposerSlashing);
+
+    const received = await onProposerSlashingChangePromise;
+    expect(Buffer.from(received)).toEqual(Buffer.from(ssz.phase0.ProposerSlashing.serialize(proposerSlashing)));
+  });
+
   it("Publish and receive a LightClientOptimisticUpdate", async function () {
     let onLightClientOptimisticUpdate: (ou: Uint8Array) => void;
     const onLightClientOptimisticUpdatePromise = new Promise<Uint8Array>(