1 files changed, 138 insertions, 0 deletions
diff --git a/sca-java-1.x/tags/java-M1-final/java/sca/core/src/test/java/org/apache/tuscany/core/async/work/GeronimoWorkManagerTestCase.java b/sca-java-1.x/tags/java-M1-final/java/sca/core/src/test/java/org/apache/tuscany/core/async/work/GeronimoWorkManagerTestCase.java
new file mode 100644
index 0000000000..18570d54c3
--- /dev/null
+++ b/sca-java-1.x/tags/java-M1-final/java/sca/core/src/test/java/org/apache/tuscany/core/async/work/GeronimoWorkManagerTestCase.java
@@ -0,0 +1,138 @@
+/**
+ *
+ *  Copyright 2005 The Apache Software Foundation or its licensors, as applicable.
+ *
+ *  Licensed under the Apache License, Version 2.0 (the "License");
+ *  you may not use this file except in compliance with the License.
+ *  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *  Unless required by applicable law or agreed to in writing, software
+ *  distributed under the License is distributed on an "AS IS" BASIS,
+ *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *  See the License for the specific language governing permissions and
+ *  limitations under the License.
+ */
+package org.apache.tuscany.core.async.work;
+
+import javax.resource.spi.work.Work;
+import javax.resource.spi.work.WorkEvent;
+import javax.resource.spi.work.WorkListener;
+
+import junit.framework.TestCase;
+
+import org.apache.geronimo.connector.work.GeronimoWorkManager;
+import org.apache.geronimo.transaction.context.TransactionContextManager;
+
+/**
+ * Tests the Geronimo work manager
+ * 
+ * @version $Rev$ $Date$
+ */
+public class GeronimoWorkManagerTestCase extends TestCase {
+
+    private GeronimoWorkManager workManager;
+
+    protected void setUp() throws Exception {
+        TransactionContextManager transactionContextManager = new TransactionContextManager();
+     
+        workManager = new GeronimoWorkManager(2, transactionContextManager);
+        workManager.doStart();
+    }
+
+    public void testScheduleWork() throws Exception {
+        TestThread threads[] = startTestThreads(5, 10000, 100);
+        int accepted = 0;
+        int started = 0;
+        for (int i = 0; i < threads.length; i++) {
+            if (null != threads[i].listener.acceptedEvent) {
+                accepted++;
+            } else if (null != threads[i].listener.startedEvent) {
+                started++;
+            } else {
+                fail("incorrect state, expecting accepted or started");
+            }
+        }
+        assertTrue(accepted > 0);
+    }
+
+    private TestThread[] startTestThreads(int count, int timeout, int delay) throws Exception {
+        TestThread threads[] = new TestThread[count];
+        for (int i = 0; i < count; i++) {
+            TestWorkListener listener=new TestWorkListener();
+            threads[i] = new TestThread(listener, timeout, delay);
+        }
+        for (int i = 0; i < count; i++) {
+            threads[i].start();
+        }
+        for (int i = 0; i < count; i++) {
+            threads[i].join();
+        }
+        return threads;
+    }
+
+    private class TestThread extends Thread {
+        public final TestWorkListener listener;
+        private final int timeout;
+        private final int delay;
+
+        public TestThread(TestWorkListener listener, int timeout, int delay) {
+            this.listener = listener;
+            this.timeout = timeout;
+            this.delay = delay;
+        }
+
+        public void run() {
+            try {
+                workManager.scheduleWork(new TestWorker(delay), timeout, null, listener);
+            } catch (Exception e) {
+                e.printStackTrace();
+            }
+        }
+
+    }
+
+    public class TestWorker implements Work {
+        private final int delay;
+
+        public TestWorker(int delay) {
+            this.delay = delay;
+        }
+
+        public void release() {
+        }
+
+        public void run() {
+            try {
+                Thread.sleep(delay);
+            } catch (InterruptedException e) {
+                e.printStackTrace();
+            }
+        }
+
+    }
+
+    public class TestWorkListener implements WorkListener {
+        public WorkEvent acceptedEvent;
+        public WorkEvent rejectedEvent;
+        public WorkEvent startedEvent;
+        public WorkEvent completedEvent;
+
+        public void workAccepted(WorkEvent e) {
+            acceptedEvent = e;
+        }
+
+        public void workRejected(WorkEvent e) {
+            rejectedEvent = e;
+        }
+
+        public void workStarted(WorkEvent e) {
+            startedEvent = e;
+        }
+
+        public void workCompleted(WorkEvent e) {
+            completedEvent = e;
+        }
+    }
+}