Git-Mirrors/haveno
1
0
Fork 0
mirror of https://github.com/haveno-dex/haveno.git synced 2025-12-14 23:44:51 -05:00
Code Issues Projects Releases Packages Wiki Activity

Customize direct message count per peer from environment variable

Browse source
This commit is contained in:
Ivan Vilata-i-Balaguer 2016-05-03 14:43:55 +02:00
parent e7e45f497e
commit 97e6b942d3
1 changed files with 31 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

45
network/src/test/java/io/bitsquare/p2p/network/NetworkStressTest.java
View file

@ -40,6 +40,13 @@ public class NetworkStressTest {
// Constants // Constants
/** Environment variable to specify the number of peers in the test. */
private static final String NPEERS_ENVVAR = "STRESS_TEST_NPEERS";
/** Environment variable to specify a persistent test data directory. */
private static final String TEST_DIR_ENVVAR = "STRESS_TEST_DIR";
/** Environment variable to specify the number of direct messages sent per peer. */
private static final String DIRECT_COUNT_ENVVAR = "STRESS_TEST_NDIRECT";
/** Numeric identifier of the regtest Bitcoin network. */ /** Numeric identifier of the regtest Bitcoin network. */
private static final int REGTEST_NETWORK_ID = 2; private static final int REGTEST_NETWORK_ID = 2;
@ -47,13 +54,9 @@ public class NetworkStressTest {
private static final int NPEERS_DEFAULT = 4; private static final int NPEERS_DEFAULT = 4;
/** Minimum number of peers for the test to work. */ /** Minimum number of peers for the test to work. */
private static final int NPEERS_MIN = 2; private static final int NPEERS_MIN = 2;
/** Environment variable to specify the number of peers in the test. */ /** Default number of direct messages to be sent by each peer. */
private static final String NPEERS_ENVVAR = "STRESS_TEST_NPEERS"; private static final int DIRECT_COUNT_DEFAULT = 100;
/** Environment variable to specify a persistent test data directory. */
private static final String TEST_DIR_ENVVAR = "STRESS_TEST_DIR";
/** Number of direct messages to be sent by each peer. */
private static int DIRECT_COUNT = 100;
/** Minimum delay between direct messages in milliseconds, 25% larger than throttle limit. */ /** Minimum delay between direct messages in milliseconds, 25% larger than throttle limit. */
private static long MIN_DIRECT_DELAY_MILLIS = Math.round(1.25 * (1.0 / Connection.MSG_THROTTLE_PER_SEC) * 1000); private static long MIN_DIRECT_DELAY_MILLIS = Math.round(1.25 * (1.0 / Connection.MSG_THROTTLE_PER_SEC) * 1000);
/** Maximum delay between direct messages in milliseconds, 10 times larger than minimum. */ /** Maximum delay between direct messages in milliseconds, 10 times larger than minimum. */
@ -75,17 +78,31 @@ public class NetworkStressTest {
/** A barrier to wait for concurrent bootstrap of peers. */ /** A barrier to wait for concurrent bootstrap of peers. */
private CountDownLatch bootstrapLatch; private CountDownLatch bootstrapLatch;
/** Number of direct messages to be sent by each peer. */
private int directCount = DIRECT_COUNT_DEFAULT;
@Before @Before
public void setUp() throws Exception { public void setUp() throws Exception {
// Parse test parameter environment variables.
/** Number of peer nodes to create. */ /** Number of peer nodes to create. */
int nPeers = NPEERS_DEFAULT; int nPeers = NPEERS_DEFAULT;
final String nPeersEnv = System.getenv(NPEERS_ENVVAR); final String nPeersEnv = System.getenv(NPEERS_ENVVAR);
if (nPeersEnv != null && !nPeersEnv.equals("")) if (nPeersEnv != null && !nPeersEnv.equals(""))
nPeers = Integer.parseInt(nPeersEnv); nPeers = Integer.parseInt(nPeersEnv);
if (nPeers < NPEERS_MIN) if (nPeers < NPEERS_MIN)
throw new IllegalArgumentException( throw new IllegalArgumentException(
String.format("Test needs at least %d peer nodes to work: %d", NPEERS_MIN, nPeers)); String.format("Test needs at least %d peer nodes to work: %d", NPEERS_MIN, nPeers)
);
final String nDirectEnv = System.getenv(DIRECT_COUNT_ENVVAR);
if (nDirectEnv != null && !nDirectEnv.equals(""))
directCount = Integer.parseInt(nDirectEnv);
if (directCount < 0)
throw new IllegalArgumentException(
String.format("Direct messages sent per peer must not be negative: %d", directCount)
);
prelimDataLatch = new CountDownLatch(nPeers); prelimDataLatch = new CountDownLatch(nPeers);
bootstrapLatch = new CountDownLatch(nPeers); bootstrapLatch = new CountDownLatch(nPeers);
@ -189,8 +206,8 @@ public class NetworkStressTest {
// Test each peer sending a direct message to another random peer. // Test each peer sending a direct message to another random peer.
final int nPeers = peerNodes.size(); final int nPeers = peerNodes.size();
BooleanProperty sentDirectFailed = new SimpleBooleanProperty(false); BooleanProperty sentDirectFailed = new SimpleBooleanProperty(false);
final CountDownLatch sentDirectLatch = new CountDownLatch(DIRECT_COUNT * nPeers); final CountDownLatch sentDirectLatch = new CountDownLatch(directCount * nPeers);
final CountDownLatch receivedDirectLatch = new CountDownLatch(DIRECT_COUNT * nPeers); final CountDownLatch receivedDirectLatch = new CountDownLatch(directCount * nPeers);
final long sendStartMillis = System.currentTimeMillis(); final long sendStartMillis = System.currentTimeMillis();
for (final P2PService srcPeer : peerNodes) { for (final P2PService srcPeer : peerNodes) {
// Make the peer ready for receiving direct messages. // Make the peer ready for receiving direct messages.
@ -203,7 +220,7 @@ public class NetworkStressTest {
}); });
long nextSendMillis = System.currentTimeMillis(); long nextSendMillis = System.currentTimeMillis();
for (int i = 0; i < DIRECT_COUNT; i++) { for (int i = 0; i < directCount; i++) {
// Select a random peer and send a direct message to it... // Select a random peer and send a direct message to it...
final int dstPeerIdx = (int) (Math.random() * nPeers); final int dstPeerIdx = (int) (Math.random() * nPeers);
final P2PService dstPeer = peerNodes.get(dstPeerIdx); final P2PService dstPeer = peerNodes.get(dstPeerIdx);
@ -233,17 +250,17 @@ public class NetworkStressTest {
// Since receiving is completed before sending is reported to be complete, // Since receiving is completed before sending is reported to be complete,
// all receiving checks should end before all sending checks to avoid deadlocking. // all receiving checks should end before all sending checks to avoid deadlocking.
/** Time to transmit all messages in the worst random case, and with no computation delays. */ /** Time to transmit all messages in the worst random case, and with no computation delays. */
final long idealMaxDirectDelay = MAX_DIRECT_DELAY_MILLIS * DIRECT_COUNT; final long idealMaxDirectDelay = MAX_DIRECT_DELAY_MILLIS * directCount;
// Wait for peers to complete receiving. We are generous here. // Wait for peers to complete receiving. We are generous here.
org.junit.Assert.assertTrue("timed out while receiving direct messages", org.junit.Assert.assertTrue("timed out while receiving direct messages",
receivedDirectLatch.await(2 * idealMaxDirectDelay, TimeUnit.MILLISECONDS)); receivedDirectLatch.await(2 * idealMaxDirectDelay, TimeUnit.MILLISECONDS));
print("receiving %d direct messages per peer took %ss", DIRECT_COUNT, print("receiving %d direct messages per peer took %ss", directCount,
(System.currentTimeMillis() - sendStartMillis)/1000.0); (System.currentTimeMillis() - sendStartMillis)/1000.0);
// Wait for peers to complete sending. // Wait for peers to complete sending.
// This should be nearly instantaneous after waiting for reception is completed. // This should be nearly instantaneous after waiting for reception is completed.
org.junit.Assert.assertTrue("timed out while sending direct messages", org.junit.Assert.assertTrue("timed out while sending direct messages",
sentDirectLatch.await(idealMaxDirectDelay / 10, TimeUnit.MILLISECONDS)); sentDirectLatch.await(idealMaxDirectDelay / 10, TimeUnit.MILLISECONDS));
print("sending %d direct messages per peer took %ss", DIRECT_COUNT, print("sending %d direct messages per peer took %ss", directCount,
(System.currentTimeMillis() - sendStartMillis)/1000.0); (System.currentTimeMillis() - sendStartMillis)/1000.0);
org.junit.Assert.assertFalse("some peer(s) failed to send a direct message", sentDirectFailed.get()); org.junit.Assert.assertFalse("some peer(s) failed to send a direct message", sentDirectFailed.get());
} }