[jboss-cvs] JBossAS SVN: r86535 - projects/jboss-jca/trunk/core/src/test/java/org/jboss/jca/test/core/spec/chapter10/section3.

[jboss-cvs-commits at lists.jboss.org]

Author: jesper.pedersen
Date: 2009-03-31 09:43:13 -0400 (Tue, 31 Mar 2009)
New Revision: 86535

Modified:
   projects/jboss-jca/trunk/core/src/test/java/org/jboss/jca/test/core/spec/chapter10/section3/WorkManagementModelTestCase.java
Log:
Fix testOneThreadPickWorkInstance

Modified: projects/jboss-jca/trunk/core/src/test/java/org/jboss/jca/test/core/spec/chapter10/section3/WorkManagementModelTestCase.java
===================================================================
--- projects/jboss-jca/trunk/core/src/test/java/org/jboss/jca/test/core/spec/chapter10/section3/WorkManagementModelTestCase.java	2009-03-31 13:33:57 UTC (rev 86534)
+++ projects/jboss-jca/trunk/core/src/test/java/org/jboss/jca/test/core/spec/chapter10/section3/WorkManagementModelTestCase.java	2009-03-31 13:43:13 UTC (rev 86535)
@@ -86,29 +86,22 @@
    {
       WorkManager workManager = bootstrap.lookup("WorkManager", WorkManager.class);
       
-      final CountDownLatch startA = new CountDownLatch(1);
-      final CountDownLatch doneA = new CountDownLatch(1);
-      final CountDownLatch startB = new CountDownLatch(1);
-      final CountDownLatch doneB = new CountDownLatch(1);
-      long threadIdA;
-      long threadIdB;
+      CountDownLatch start = new CountDownLatch(1);
+      CountDownLatch done = new CountDownLatch(2);
 
-      LongRunningWork mwA = new LongRunningWork(startA, doneA);
-      LongRunningWork mwB = new LongRunningWork(startB, doneB);
+      LongRunningWork mwA = new LongRunningWork(start, done);
+      LongRunningWork mwB = new LongRunningWork(start, done);
 
-      startA.countDown();
       workManager.startWork(mwA);
-      threadIdA = mwA.getThreadId();
-      
-      startB.countDown();
       workManager.startWork(mwB);
-      threadIdB = mwB.getThreadId();
-      
-      doneA.await();
-      doneB.await();
-      
-      assertNotSame(threadIdA, threadIdB);
 
+      start.countDown();
+
+      done.await();
+
+      assertFalse(mwA.getThreadId() == mwB.getThreadId());
+      assertTrue(mwA.hasPostRun());
+      assertTrue(mwB.hasPostRun());
    }
    
    /**