Forces the test to fail if any worker encounter an exception

systay · Jun 13, 2012 · bb1bc7b · bb1bc7b
1 parent c665994
commit bb1bc7b
Showing 1 changed file with 18 additions and 5 deletions.
diff --git a/embedded-examples/src/test/java/org/neo4j/examples/GetOrCreateTest.java b/embedded-examples/src/test/java/org/neo4j/examples/GetOrCreateTest.java
@@ -25,6 +25,7 @@
 import java.util.List;
 import java.util.Map;
 import java.util.NoSuchElementException;
+import java.util.concurrent.atomic.AtomicReference;
 
 import org.junit.Test;
 import org.neo4j.graphdb.GraphDatabaseService;
@@ -68,7 +69,6 @@ class ThreadRunner implements Runnable
             this.impl = impl;
         }
 
-
         private Node createNode()
         {
             Transaction tx = graphdb().beginTx();
@@ -90,19 +90,28 @@ public void run()
             final Node lockNode = createNode();
             final List<List<Node>> results = new ArrayList<List<Node>>();
             final List<Thread> threads = new ArrayList<Thread>();
+            final AtomicReference<RuntimeException> failure = new AtomicReference<RuntimeException>();
             for ( int i = 0; i < 10; i++ )
             {
                 threads.add( new Thread()
                 {
                     @Override
                     public void run()
                     {
-                        List<Node> subresult = new ArrayList<Node>();
-                        for ( int j = 0; j < NUM_USERS; j++ )
+                        try
+                        {
+                            List<Node> subresult = new ArrayList<Node>();
+                            for ( int j = 0; j < NUM_USERS; j++ )
+                            {
+                                subresult.add( impl.getOrCreateUser( getUsername( j ), graphdb(), lockNode ) );
+                            }
+                            results.add( subresult );
+                        }
+                        catch ( RuntimeException e )
                         {
-                            subresult.add( impl.getOrCreateUser( getUsername( j ), graphdb(), lockNode ) );
+                            failure.compareAndSet( null, e );
+                            throw e;
                         }
-                        results.add( subresult );
                     }
                 } );
             }
@@ -121,6 +130,10 @@ public void run()
                     e.printStackTrace();
                 }
             }
+
+            if ( failure.get() != null )
+                throw failure.get();
+
             List<Node> first = results.remove( 0 );
             for ( List<Node> subresult : results )
             {