fixed a race condition when the method is invoked too quickly and

lprimak · lprimak · commit f61c3217e7a9 · 2018-01-28T11:39:28.000-06:00
is never registered as being run, thus making the test wait forever
diff --git a/jms/send-receive/src/test/java/org/javaee7/jms/send/receive/mdb/ReceptionSynchronizer.java b/jms/send-receive/src/test/java/org/javaee7/jms/send/receive/mdb/ReceptionSynchronizer.java
@@ -56,6 +56,8 @@ void registerInvocation(Method m) {
         synchronized (barrier) {
             if (barrier.containsKey(m)) {
                 latch = barrier.get(m);
+            } else {
+                barrier.put(m, new CountDownLatch(0));
             }
         }
         if (latch != null) {
@@ -85,11 +87,6 @@ private static void waitFor(Method method, int timeoutInMillis) throws Interrupt
         synchronized (barrier) {
             if (barrier.containsKey(method)) {
                 latch = barrier.get(method);
-                if (latch.getCount() == 0) {
-                    throw new IllegalStateException("The invocation already happened");
-                } else {
-                    throw new IllegalStateException("Sorry, I only serve the first one");
-                }
             } else {
                 latch = new CountDownLatch(1);
                 barrier.put(method, latch);
@@ -99,5 +96,4 @@ private static void waitFor(Method method, int timeoutInMillis) throws Interrupt
             throw new AssertionError("Expected method has not been invoked in " + timeoutInMillis + "ms");
         }
     }
-
 }

Original file line number	Diff line number	Diff line change
`@@ -56,6 +56,8 @@ void registerInvocation(Method m) {`
`56`	`56`	`synchronized (barrier) {`
`57`	`57`	`if (barrier.containsKey(m)) {`
`58`	`58`	`latch = barrier.get(m);`
	`59`	`+ } else {`
	`60`	`+ barrier.put(m, new CountDownLatch(0));`
`59`	`61`	`}`
`60`	`62`	`}`
`61`	`63`	`if (latch != null) {`
`@@ -85,11 +87,6 @@ private static void waitFor(Method method, int timeoutInMillis) throws Interrupt`
`85`	`87`	`synchronized (barrier) {`
`86`	`88`	`if (barrier.containsKey(method)) {`
`87`	`89`	`latch = barrier.get(method);`
`88`		`- if (latch.getCount() == 0) {`
`89`		`- throw new IllegalStateException("The invocation already happened");`
`90`		`- } else {`
`91`		`- throw new IllegalStateException("Sorry, I only serve the first one");`
`92`		`- }`
`93`	`90`	`} else {`
`94`	`91`	`latch = new CountDownLatch(1);`
`95`	`92`	`barrier.put(method, latch);`
`@@ -99,5 +96,4 @@ private static void waitFor(Method method, int timeoutInMillis) throws Interrupt`
`99`	`96`	`throw new AssertionError("Expected method has not been invoked in " + timeoutInMillis + "ms");`
`100`	`97`	`}`
`101`	`98`	`}`
`102`		`-`
`103`	`99`	`}`