Add Identify to more tests that need it

Peergos · Jul 13, 2023 · 3b46a22 · 3b46a22
1 parent 45d1683
commit 3b46a22
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/test/java/org/peergos/FindPeerTest.java b/src/test/java/org/peergos/FindPeerTest.java
@@ -5,6 +5,7 @@
 import io.libp2p.core.multiformats.*;
 import org.junit.*;
 import org.peergos.blockstore.*;
+import org.peergos.protocol.*;
 import org.peergos.protocol.dht.*;
 
 import java.util.*;
@@ -19,6 +20,7 @@ public void findLongRunningNode() {
                 new RamProviderStore(), new RamRecordStore(), blockstore1, (c, b, p, a) -> CompletableFuture.completedFuture(true));
         Host node1 = builder1.build();
         node1.start().join();
+        IdentifyBuilder.addIdentifyProtocol(node1);
 
         try {
             // bootstrap node 1

diff --git a/src/test/java/org/peergos/PingTest.java b/src/test/java/org/peergos/PingTest.java
@@ -6,6 +6,7 @@
 import io.libp2p.protocol.*;
 import org.junit.*;
 import org.peergos.blockstore.*;
+import org.peergos.protocol.*;
 import org.peergos.protocol.bitswap.*;
 
 import java.util.*;
@@ -39,7 +40,9 @@ public void replyIdentifyOnNewDial() {
         Host node1 = HostBuilder.build(TestPorts.getPort(), List.of(new Ping(), new Bitswap(new BitswapEngine(new RamBlockstore(), (c, b, p, a) -> CompletableFuture.completedFuture(true)))));
         Host node2 = HostBuilder.build(TestPorts.getPort(), List.of(new Ping(), new Bitswap(new BitswapEngine(new RamBlockstore(), (c, b, p, a) -> CompletableFuture.completedFuture(true)))));
         node1.start().join();
+        IdentifyBuilder.addIdentifyProtocol(node1);
         node2.start().join();
+        IdentifyBuilder.addIdentifyProtocol(node2);
         try {
             // ping from 1 to 2
             Multiaddr address2 = node2.listenAddresses().get(0);