[jboss-svn-commits] JBL Code SVN: r19822 - labs/jbossrules/trunk/drools-compiler/src/test/java/org/drools/integrationtests.

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

Author: tirelli
Date: 2008-05-01 18:06:30 -0400 (Thu, 01 May 2008)
New Revision: 19822

Modified:
   labs/jbossrules/trunk/drools-compiler/src/test/java/org/drools/integrationtests/FirstOrderLogicTest.java
Log:
JBRULES-1594: fixing test

Modified: labs/jbossrules/trunk/drools-compiler/src/test/java/org/drools/integrationtests/FirstOrderLogicTest.java
===================================================================
--- labs/jbossrules/trunk/drools-compiler/src/test/java/org/drools/integrationtests/FirstOrderLogicTest.java	2008-05-01 22:05:56 UTC (rev 19821)
+++ labs/jbossrules/trunk/drools-compiler/src/test/java/org/drools/integrationtests/FirstOrderLogicTest.java	2008-05-01 22:06:30 UTC (rev 19822)
@@ -22,7 +22,9 @@
 import org.drools.lang.descr.PackageDescr;
 import org.drools.rule.Package;
 import org.drools.rule.Rule;
+import org.drools.util.DroolsStreamUtils;
 
+import java.io.ByteArrayInputStream;
 import java.io.IOException;
 import java.io.InputStreamReader;
 import java.io.Reader;
@@ -43,6 +45,15 @@
                                             config );
     }
 
+    private StatefulSession serializeWorkingMemory(RuleBase ruleBase,
+                                                   StatefulSession wm) throws IOException,
+                                                                      ClassNotFoundException {
+        byte[] serializedSession = DroolsStreamUtils.streamOut( wm );
+        wm.dispose();
+        wm = ruleBase.newStatefulSession( new ByteArrayInputStream( serializedSession ) );
+        return wm;
+    }
+
     public void testCollect() throws Exception {
 
         // read in the source
@@ -977,7 +988,7 @@
         wm.insert( new Double( 10 ) );
         wm.insert( new Integer( 2 ) );
 
-        ruleBase = serializeRuleBase( ruleBase );
+        ruleBase = SerializationHelper.serializeObject( ruleBase );
         wm = serializeWorkingMemory( ruleBase,
                                      wm );
         results = (List) wm.getGlobal( "results" );