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

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

Author: mark.proctor at jboss.com
Date: 2009-01-12 10:54:47 -0500 (Mon, 12 Jan 2009)
New Revision: 24651

Modified:
   labs/jbossrules/trunk/drools-core/src/test/java/org/drools/runtime/pipeline/impl/StatefulKnowledgeSessionPipelineTest.java
Log:
JBRULES-1903 Pipeline support for DataLoaders and Feeders 
-Fixed unit test failure

Modified: labs/jbossrules/trunk/drools-core/src/test/java/org/drools/runtime/pipeline/impl/StatefulKnowledgeSessionPipelineTest.java
===================================================================
--- labs/jbossrules/trunk/drools-core/src/test/java/org/drools/runtime/pipeline/impl/StatefulKnowledgeSessionPipelineTest.java	2009-01-12 14:22:28 UTC (rev 24650)
+++ labs/jbossrules/trunk/drools-core/src/test/java/org/drools/runtime/pipeline/impl/StatefulKnowledgeSessionPipelineTest.java	2009-01-12 15:54:47 UTC (rev 24651)
@@ -5,6 +5,9 @@
 import org.drools.KnowledgeBase;
 import org.drools.KnowledgeBaseFactory;
 import org.drools.runtime.StatefulKnowledgeSession;
+import org.drools.runtime.pipeline.Action;
+import org.drools.runtime.pipeline.Pipeline;
+import org.drools.runtime.pipeline.PipelineFactory;
 import org.drools.runtime.pipeline.ResultHandler;
 
 import junit.framework.TestCase;
@@ -14,13 +17,15 @@
         KnowledgeBase kbase = KnowledgeBaseFactory.newKnowledgeBase();
         StatefulKnowledgeSession ksession = kbase.newStatefulKnowledgeSession();
         
-        StatefulKnowledgeSessionInsertStage stage1 = new StatefulKnowledgeSessionInsertStage();
-        MvelAction mvelAction = new MvelAction( "context.resultHandler.handleResult( context.handles )");
-        stage1.setReceiver( mvelAction );
+        Action executeResultHandler = PipelineFactory.newExecuteResultHandler();
         
-        StatefulKnowledgeSessionPipelineImpl pipeline = new StatefulKnowledgeSessionPipelineImpl(ksession);
-        pipeline.setReceiver( stage1 );
+        StatefulKnowledgeSessionInsertStage insertStage = new StatefulKnowledgeSessionInsertStage();
+        insertStage.setReceiver( executeResultHandler );
         
+        
+        Pipeline pipeline = PipelineFactory.newStatefulKnowledgeSessionPipeline(ksession);
+        pipeline.setReceiver( insertStage );
+        
         assertEquals( 0, ksession.getObjects().size() );
         
         ResultHandlerImpl resultHanadle = new ResultHandlerImpl();