From 0fc8a1b5c0c4d3ab936c8c619e66f938ed127d95 Mon Sep 17 00:00:00 2001
From: Ivan Vilata-i-Balaguer <ivan@selidor.net>
Date: Tue, 10 May 2016 18:45:13 +0200
Subject: [PATCH] Add main function to run an individual network stress test

---
 .../io/bitsquare/p2p/network/NetworkStressTest.java | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/network/src/test/java/io/bitsquare/p2p/network/NetworkStressTest.java b/network/src/test/java/io/bitsquare/p2p/network/NetworkStressTest.java
index a3e56421c8..7fbd61f745 100644
--- a/network/src/test/java/io/bitsquare/p2p/network/NetworkStressTest.java
+++ b/network/src/test/java/io/bitsquare/p2p/network/NetworkStressTest.java
@@ -23,6 +23,9 @@ import org.jetbrains.annotations.NotNull;
 import org.junit.After;
 import org.junit.Before;
 import org.junit.Test;
+import org.junit.runner.JUnitCore;
+import org.junit.runner.Request;
+import org.junit.runner.Result;
 
 import java.io.File;
 import java.io.IOException;
@@ -83,6 +86,16 @@ public class NetworkStressTest {
     /** Number of direct messages to be sent by each peer. */
     private int directCount = DIRECT_COUNT_DEFAULT;
 
+    // Inspired by <https://stackoverflow.com/a/9288513> by Marc Peters.
+    public static void main(String[] args) {
+        Request request = (args.length == 0)
+                ? Request.aClass(NetworkStressTest.class)
+                : Request.method(NetworkStressTest.class, args[0]);
+
+        Result result = new JUnitCore().run(request);
+        System.exit(result.wasSuccessful() ? 0 : 1);
+    }
+
     @Before
     public void setUp() throws Exception {
         // Parse test parameter environment variables.