[jboss-svn-commits] JBL Code SVN: r20223 - labs/jbossrules/trunk/drools-core/src/test/java/org/drools/time/impl.

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

Author: mark.proctor at jboss.com
Date: 2008-05-29 15:58:03 -0400 (Thu, 29 May 2008)
New Revision: 20223

Added:
   labs/jbossrules/trunk/drools-core/src/test/java/org/drools/time/impl/JDKTimerServiceTest.java
Removed:
   labs/jbossrules/trunk/drools-core/src/test/java/org/drools/time/impl/jdk/
Log:
JBRULES-1625 Create Scheduler API with JDK implementation
-Basic refactoring for for TimerService, instead of Scheduler

Added: labs/jbossrules/trunk/drools-core/src/test/java/org/drools/time/impl/JDKTimerServiceTest.java
===================================================================
--- labs/jbossrules/trunk/drools-core/src/test/java/org/drools/time/impl/JDKTimerServiceTest.java	                        (rev 0)
+++ labs/jbossrules/trunk/drools-core/src/test/java/org/drools/time/impl/JDKTimerServiceTest.java	2008-05-29 19:58:03 UTC (rev 20223)
@@ -0,0 +1,131 @@
+package org.drools.time.impl;
+
+import java.util.ArrayList;
+import java.util.Date;
+import java.util.List;
+import java.util.Stack;
+
+import junit.framework.TestCase;
+
+import org.drools.time.Job;
+import org.drools.time.JobContext;
+import org.drools.time.JobHandle;
+import org.drools.time.TimerService;
+import org.drools.time.TimerServiceFactory;
+import org.drools.time.Trigger;
+
+public class JDKTimerServiceTest extends TestCase {
+    
+    public void testSingleExecutionJob() throws Exception {
+        TimerService timeService = TimerServiceFactory.getTimerService(); 
+        Trigger trigger = new DelayedTrigger( 100 );
+        HelloWorldJobContext ctx = new HelloWorldJobContext( "hello world", timeService);
+        timeService.scheduleJob( new HelloWorldJob(), ctx,  trigger);        
+        Thread.sleep( 500 );
+        assertEquals( 1, ctx.getList().size() ); 
+    }    
+    
+    public void testRepeatedExecutionJob() throws Exception {
+        TimerService timeService = TimerServiceFactory.getTimerService(); 
+        Trigger trigger = new DelayedTrigger(  new long[] { 100, 100, 100} );
+        HelloWorldJobContext ctx = new HelloWorldJobContext( "hello world", timeService);
+        timeService.scheduleJob( new HelloWorldJob(), ctx,  trigger);        
+        Thread.sleep( 500 );
+        
+        assertEquals( 3, ctx.getList().size() );
+    }    
+        
+    
+	public void testRepeatedExecutionJobWithRemove() throws Exception {
+	    TimerService timeService = TimerServiceFactory.getTimerService();
+		Trigger trigger = new DelayedTrigger( new long[] { 100, 100, 100, 100, 100 } );
+		HelloWorldJobContext ctx = new HelloWorldJobContext( "hello world", timeService);
+		ctx.setLimit( 3 );
+		timeService.scheduleJob( new HelloWorldJob(), ctx,  trigger);		
+		Thread.sleep( 1000 );
+		
+		assertEquals( 4, ctx.getList().size() );
+	}
+	
+	public static class HelloWorldJob implements Job {
+        public void execute(JobContext c) {
+            HelloWorldJobContext ctx = (HelloWorldJobContext) c;
+            int counter = ctx.increaseCounter();
+            if ( counter > 3 ) {
+                ctx.timeService.removeJob( ctx.getJobHandle() );
+            }
+            ctx.getList().add( ((HelloWorldJobContext)ctx).getMessage() + " : " + counter);
+        }	    
+	}
+	
+	public static class HelloWorldJobContext implements JobContext {
+	    private String message;
+	    private  TimerService timeService;
+	    private JobHandle jobHandle;
+	    
+	    private List list;
+	    
+	    private int counter;	    
+	    private int limit;
+	    
+	    public HelloWorldJobContext(String message, TimerService timeService) {
+	        this.message = message;
+	        this.timeService = timeService;
+	        this.list = new ArrayList();
+	    }
+	    
+	    public String getMessage() {
+	        return this.message;
+	    }
+	    
+	    public int increaseCounter() {
+	        return this.counter++;
+	    }
+
+        public JobHandle getJobHandle() {
+            return this.jobHandle;
+        }
+
+        public void setJobHandle(JobHandle jobHandle) {
+            this.jobHandle = jobHandle;
+        }
+
+        public int getLimit() {
+            return limit;
+        }
+
+        public void setLimit(int limit) {
+            this.limit = limit;
+        }
+
+        public List getList() {
+            return list;
+        }
+	    
+	    
+	}
+	
+	public static class DelayedTrigger implements Trigger {
+	    public Stack<Long> stack;
+	    
+	    public DelayedTrigger(long delay) {
+	        this( new long[] { delay } );
+	    }
+	    
+        public DelayedTrigger(long[] delay) {
+            this.stack = new Stack<Long>();
+            for( int i = delay.length-1; i >= 0; i-- ) {
+                this.stack.push( delay[i] );
+            }
+        }	    
+
+        public Date getNextFireTime() {
+            if ( !this.stack.isEmpty() ) {
+                return new Date( this.stack.pop() );    
+            } else {
+                return null;
+            }
+        }
+	    
+	}
+}