Git-Mirrors/haveno
1
0
Fork 0
mirror of https://github.com/haveno-dex/haveno.git synced 2025-04-21 16:16:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

Skeleton of mailbox send stress test

Browse Source
This commit is contained in:
Ivan Vilata-i-Balaguer 2016-05-12 12:08:30 +02:00
parent c27aabda2e
commit 3ab270f5c5
1 changed files with 69 additions and 49 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

118
network/src/test/java/io/bitsquare/p2p/network/NetworkStressTest.java
View File

@ -360,58 +360,41 @@ public class NetworkStressTest {
// (so it can get its mailbox messages),
// and the new first online node sends messages.
// This is repeated until all nodes have been online and offline.
final CountDownLatch shutDownLatch = new CountDownLatch(1);
P2PService peer = peerNodes.get(0);
peer.shutDown(shutDownLatch::countDown);
shutDownLatch.await();
peer = createPeerNode(0, peerPorts.get(0));
peerNodes.set(0, peer);
final CountDownLatch bootLatch = new CountDownLatch(1);
peer.start(new P2PServiceListener() {
@Override
public void onRequestingDataCompleted() {
}
@Override
public void onNoSeedNodeAvailable() {
}
@Override
public void onNoPeersAvailable() {
}
@Override
public void onBootstrapComplete() {
bootLatch.countDown();
}
@Override
public void onTorNodeReady() {
}
@Override
public void onHiddenServicePublished() {
}
@Override
public void onSetupFailed(Throwable throwable) {
}
});
bootLatch.await();
// Put the second half of peers offline.
final int nPeers = peerNodes.size();
for (int firstOnline = 0, firstOffline = (int)Math.ceil(nPeers/2.0);
firstOnline < nPeers;
firstOnline++, firstOffline = ++firstOffline%nPeers) {
System.out.println("firstOnline "+firstOnline+" firstOffline "+firstOffline);
final CountDownLatch halfShutDown = new CountDownLatch(nPeers / 2);
int firstPeerDown = (int)Math.ceil(nPeers / 2.0);
for (P2PService peer : peerNodes.subList(firstPeerDown, nPeers)) {
peer.shutDown(halfShutDown::countDown);
}
assertLatch("timed out while stopping a half of the peers",
halfShutDown, 10, TimeUnit.SECONDS);
print("stopped a half of the peers for mailbox test");
// Cycle through peers sending to others, stopping the peer
// and starting one of the stopped peers.
for (int firstOnline = 0, firstOffline = firstPeerDown;
firstOnline < nPeers;
firstOnline++, firstOffline = ++firstOffline % nPeers) {
// TODO: Make first online node send messages to other nodes.
// When done, put first online peer offline.
final CountDownLatch stopLatch = new CountDownLatch(1);
peerNodes.get(firstOnline).shutDown(stopLatch::countDown);
stopLatch.await(10, TimeUnit.SECONDS);
print("put peer %d offline", firstOnline);
// When done, put first offline peer online.
final CountDownLatch startLatch = new CountDownLatch(1);
final P2PService startedPeer = createPeerNode(firstOffline, peerPorts.get(firstOffline));
// TODO: Setup message listeners.
peerNodes.set(firstOffline, startedPeer);
startedPeer.start(new MailboxStartListener(startLatch));
startLatch.await(10, TimeUnit.SECONDS);
print("put peer %d online", firstOffline);
}
// TODO: Wait for nodes to receive messages.
}
private void print(String message, Object... args) {
@ -575,6 +558,43 @@ public class NetworkStressTest {
countDownAndPrint(bootstrapLatch, 'b');
}
}
private class MailboxStartListener implements P2PServiceListener {
private final CountDownLatch startLatch;
MailboxStartListener(CountDownLatch startLatch) {
this.startLatch = startLatch;
}
@Override
public void onRequestingDataCompleted() {
}
@Override
public void onNoSeedNodeAvailable() {
}
@Override
public void onNoPeersAvailable() {
}
@Override
public void onBootstrapComplete() {
startLatch.countDown();
}
@Override
public void onTorNodeReady() {
}
@Override
public void onHiddenServicePublished() {
}
@Override
public void onSetupFailed(Throwable throwable) {
}
}
}
// Message classes