Add MockNode that allows for simulating connections

p2p/src/test/java/bisq/network/p2p/MockNode.java

@@ -0,0 +1,86 @@
+/*
+ * This file is part of Bisq.
+ *
+ * Bisq is free software: you can redistribute it and/or modify it
+ * under the terms of the GNU Affero General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or (at
+ * your option) any later version.
+ *
+ * Bisq is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE. See the GNU Affero General Public
+ * License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with Bisq. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package bisq.network.p2p;
+
+import bisq.core.network.p2p.seed.DefaultSeedNodeRepository;
+
+import bisq.network.p2p.network.Connection;
+import bisq.network.p2p.network.InboundConnection;
+import bisq.network.p2p.network.NetworkNode;
+import bisq.network.p2p.network.OutboundConnection;
+import bisq.network.p2p.network.Statistic;
+import bisq.network.p2p.peers.PeerManager;
+import bisq.network.p2p.peers.peerexchange.PeerList;
+
+import bisq.common.ClockWatcher;
+import bisq.common.proto.persistable.PersistenceProtoResolver;
+import bisq.common.storage.CorruptedDatabaseFilesHandler;
+import bisq.common.storage.Storage;
+
+import java.io.File;
+
+import java.util.HashSet;
+import java.util.Set;
+
+import lombok.Getter;
+
+import static org.mockito.Mockito.doNothing;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
+public class MockNode {
+    @Getter
+    public NetworkNode networkNode;
+    @Getter
+    public PeerManager peerManager;
+    @Getter
+    public Set<Connection> connections;
+    @Getter
+    public int maxConnections;
+
+    public MockNode(int maxConnections) {
+        this.maxConnections = maxConnections;
+        networkNode = mock(NetworkNode.class);
+        Storage<PeerList> storage = new Storage<>(mock(File.class), mock(PersistenceProtoResolver.class), mock(CorruptedDatabaseFilesHandler.class));
+        peerManager = new PeerManager(networkNode, mock(DefaultSeedNodeRepository.class), new ClockWatcher(), maxConnections, storage);
+        connections = new HashSet<>();
+        when(networkNode.getAllConnections()).thenReturn(connections);
+    }
+
+    public void addInboundConnection(Connection.PeerType peerType) {
+        InboundConnection inboundConnection = mock(InboundConnection.class);
+        when(inboundConnection.getPeerType()).thenReturn(peerType);
+        Statistic statistic = mock(Statistic.class);
+        long lastActivityTimestamp = System.currentTimeMillis();
+        when(statistic.getLastActivityTimestamp()).thenReturn(lastActivityTimestamp);
+        when(inboundConnection.getStatistic()).thenReturn(statistic);
+        doNothing().when(inboundConnection).run();
+        connections.add(inboundConnection);
+    }
+
+    public void addOutboundConnection(Connection.PeerType peerType) {
+        OutboundConnection outboundConnection = mock(OutboundConnection.class);
+        when(outboundConnection.getPeerType()).thenReturn(peerType);
+        Statistic statistic = mock(Statistic.class);
+        long lastActivityTimestamp = System.currentTimeMillis();
+        when(statistic.getLastActivityTimestamp()).thenReturn(lastActivityTimestamp);
+        when(outboundConnection.getStatistic()).thenReturn(statistic);
+        doNothing().when(outboundConnection).run();
+        connections.add(outboundConnection);
+    }
+}