Git-Mirrors/haveno
1
0
Fork 0
mirror of https://github.com/haveno-dex/haveno.git synced 2025-08-07 06:02:33 -04:00
Code Issues Projects Releases Packages Wiki Activity

Make the number of peers in network stress test selectable with environment variable

Browse source
This commit is contained in:
Ivan Vilata-i-Balaguer 2016-04-27 14:56:43 +02:00
parent bfbda2c38e
commit a9973808b2
1 changed files with 18 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -33,8 +33,6 @@ import java.util.concurrent.TimeUnit;
public class NetworkStressTest { public class NetworkStressTest {
// Test parameters // Test parameters
/** Number of peer nodes to create. */
private static final int NPEERS = 4;
/** Whether to log messages less important than warnings. */ /** Whether to log messages less important than warnings. */
private static final boolean USE_DETAILED_LOGGING = false; private static final boolean USE_DETAILED_LOGGING = false;
@ -42,6 +40,10 @@ public class NetworkStressTest {
/** 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;
/** Default number of peers in the test. */
private static final int NPEERS_DEFAULT = 4;
/** 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. */ /** Environment variable to specify a persistent test data directory. */
private static final String TEST_DIR_ENVVAR = "STRESS_TEST_DIR"; private static final String TEST_DIR_ENVVAR = "STRESS_TEST_DIR";
@ -55,16 +57,25 @@ public class NetworkStressTest {
private List<P2PService> peerNodes = new ArrayList<>(); private List<P2PService> peerNodes = new ArrayList<>();
/** A barrier to wait for concurrent reception of preliminary data in peers. */ /** A barrier to wait for concurrent reception of preliminary data in peers. */
private CountDownLatch prelimDataLatch = new CountDownLatch(NPEERS); private CountDownLatch prelimDataLatch;
/** A barrier to wait for concurrent bootstrap of peers. */ /** A barrier to wait for concurrent bootstrap of peers. */
private CountDownLatch bootstrapLatch = new CountDownLatch(NPEERS); private CountDownLatch bootstrapLatch;
@Before @Before
public void setUp() throws Exception { public void setUp() throws Exception {
/** Number of peer nodes to create. */
int nPeers = NPEERS_DEFAULT;
final String nPeersEnv = System.getenv(NPEERS_ENVVAR);
if (nPeersEnv != null && !nPeersEnv.equals(""))
nPeers = Integer.parseInt(nPeersEnv);
prelimDataLatch = new CountDownLatch(nPeers);
bootstrapLatch = new CountDownLatch(nPeers);
/** A property where threads can indicate setup failure of local services (Tor node, hidden service). */ /** A property where threads can indicate setup failure of local services (Tor node, hidden service). */
BooleanProperty localServicesFailed = new SimpleBooleanProperty(false); final BooleanProperty localServicesFailed = new SimpleBooleanProperty(false);
/** A barrier to wait for concurrent setup of local services (Tor node, hidden service). */ /** A barrier to wait for concurrent setup of local services (Tor node, hidden service). */
final CountDownLatch localServicesLatch = new CountDownLatch(1 /*seed node*/ + NPEERS); final CountDownLatch localServicesLatch = new CountDownLatch(1 /*seed node*/ + nPeers);
// Set a security provider to allow key generation. // Set a security provider to allow key generation.
Security.addProvider(new BouncyCastleProvider()); Security.addProvider(new BouncyCastleProvider());
@ -89,7 +100,7 @@ public class NetworkStressTest {
} else { } else {
seedNodesRepository.setTorSeedNodeAddresses(seedNodes); seedNodesRepository.setTorSeedNodeAddresses(seedNodes);
} }
for (int p = 0; p < NPEERS; p++) { for (int p = 0; p < nPeers; p++) {
final int peerPort = Utils.findFreeSystemPort(); final int peerPort = Utils.findFreeSystemPort();
final File peerDir = new File(testDataDir.toFile(), String.format("peer-%06d", p)); final File peerDir = new File(testDataDir.toFile(), String.format("peer-%06d", p));
final File peerTorDir = new File(peerDir, "tor"); final File peerTorDir = new File(peerDir, "tor");