* Copyright 2010 JBoss Inc

*

* 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.

*/

    @Before

    public void setUp() throws Exception {

        XMLUnit.setIgnoreComments( true );

        XMLUnit.setIgnoreWhitespace( true );

        XMLUnit.setIgnoreAttributeOrder( true );

        XMLUnit.setNormalizeWhitespace( true );

        XMLUnit.setNormalize( true );

    }

    private void assertXMLEqual(String expectedXml,

                                String resultXml) {

        try {

            Diff diff = new Diff( expectedXml,

                                  resultXml );

            diff.overrideElementQualifier( new RecursiveElementNameAndTextQualifier() );

            XMLAssert.assertXMLEqual( diff,

                                      true );

        } catch ( Exception e ) {

            throw new RuntimeException( "XML Assertion failure",

                                        e );

        }

    }

    @Test

    public void testListenForChanges() throws Exception {

        String str = "";

        str += "package org.drools \n";

        str += "import org.drools.Cheese \n";

        str += "import org.drools.ChangeCollector \n";

        str += "rule rule1 \n";

        str += "  when \n";

        str += "    $c : Cheese(price==25) \n";

        str += " \n";

        str += "  then \n";

        str += "end\n";

        str += "rule rule2 \n";

        str += "  when \n";

        str += "    p : Person(name=='mic') \n";

        str += "    c : Cheese(price != 42) \n";

        str += " \n";

        str += "  then \n";

        str += "    c.setPrice( 42 ); \n";

        str += "    update(c); \n";

        str += "end\n";

        str += "rule rule3 \n";

        str += "  when \n";

        str += "    p : Person(name=='mark') \n";

        str += "    c : Cheese(price == 42) \n";

        str += " \n";

        str += "  then \n";

        str += "    retract(c); \n";

        str += "end\n";

        str += "rule ruleBootStrap \n";

        str += "salience 10000\n";

        str += "  when \n";

        str += "    $c : ChangeCollector() \n";

        str += " \n";

        str += "  then \n";

        str += "    kcontext.getKnowledgeRuntime().addEventListener($c); \n";

        str += "end\n";

        str += "rule ruleCleanup \n";

        str += "salience -10000\n";

        str += "  when \n";

        str += "    $c : ChangeCollector() \n";

        str += " \n";

        str += "  then \n";

        str += "    kcontext.getKnowledgeRuntime().removeEventListener($c); \n";

        str += "    retract($c); \n";

        str += "end\n";

        String inXml = "";

        inXml += "{\"batch-execution\":{\"commands\":[";

        inXml += "{\"insert\":{\"object\":{\"org.drools.Cheese\":{\"type\":\"stilton\",\"price\":25,\"oldPrice\":0}}, \"out-identifier\":\"outStilton\" } }";

        inXml += ", {\"fire-all-rules\":\"\"}";

        inXml += "]}}";

        inXml = roundTripFromXml( inXml );

               

        StatefulKnowledgeSession ksession = getSessionStateful( ResourceFactory.newByteArrayResource( str.getBytes() ) );

        ResultHandlerImpl resultHandler = new ResultHandlerImpl();

        getPipelineStateful( ksession ).insert( inXml,

                                                resultHandler );

        inXml = "";

        inXml += "{\"batch-execution\":{\"commands\":[";

        inXml += "   {\"insert\":{\"object\":{\"org.drools.Person\":{\"name\":\"mic\"} }, \"out-identifier\":\"person\" } }";

        inXml += ",  {\"insert\":{\"object\":{\"org.drools.ChangeCollector\":{} }, \"out-identifier\":\"changes\" } }";

        inXml += ",  {\"fire-all-rules\":\"\"}";

        inXml += "]}}";

        inXml = roundTripFromXml( inXml );

                       

       

        resultHandler = new ResultHandlerImpl();

        getPipelineStateful( ksession ).insert( inXml,

                                                resultHandler );

        String outXml = (String) resultHandler.getObject();

        outXml = roundTripFromXml( outXml );

       

        ExecutionResults result = (ExecutionResults) BatchExecutionHelper.newJSonMarshaller().fromXML( outXml );

        result = ( ExecutionResults ) roundTripFromObject( result );

        ChangeCollector collector = ( ChangeCollector ) result.getValue( "changes" );

        Cheese c = ( Cheese ) collector.getChanges().get( 0 );

        assertEquals( 42, c.getPrice() );

        inXml = "";

        inXml += "{\"batch-execution\":{\"commands\":[";

        inXml += "   {\"insert\":{\"object\":{\"org.drools.Person\":{\"name\":\"mark\"} }, \"out-identifier\":\"person\" } }";

        inXml += ",  {\"insert\":{\"object\":{\"org.drools.ChangeCollector\":{} }, \"out-identifier\":\"changes\" } }";

        inXml += ",  {\"fire-all-rules\":\"\"}";

        inXml += "]}}";

        inXml = roundTripFromXml( inXml );

       

        resultHandler = new ResultHandlerImpl();

        getPipelineStateful( ksession ).insert( inXml,

                                                resultHandler );

       

        outXml = (String) resultHandler.getObject();

        outXml = roundTripFromXml( outXml );

       

        result = (ExecutionResults) BatchExecutionHelper.newJSonMarshaller().fromXML( outXml );

        result = ( ExecutionResults ) roundTripFromObject( result );

        collector = ( ChangeCollector ) result.getValue( "changes" );

        assertEquals( "stilton", collector.getRetracted().get( 0 ) );

    }

    @Test

    public void testInsertWithDefaults() throws Exception {

        String str = "";

        str += "package org.drools \n";

        str += "import org.drools.Cheese \n";

        str += "rule rule1 \n";

        str += "  when \n";

        str += "    $c : Cheese() \n";

        str += " \n";

        str += "  then \n";

        str += "    $c.setPrice( $c.getPrice() + 5 ); \n";

        str += "end\n";

        String inXml = "";

        inXml += "{\"batch-execution\":{\"commands\":[";

        inXml += "{\"insert\":{\"object\":{\"org.drools.Cheese\":{\"type\":\"stilton\",\"price\":25,\"oldPrice\":0}}, \"out-identifier\":\"outStilton\" } }";

        inXml += ", {\"fire-all-rules\":\"\"}";

        inXml += "]}}";

        inXml = roundTripFromXml( inXml );

       

        StatefulKnowledgeSession ksession = getSessionStateful( ResourceFactory.newByteArrayResource( str.getBytes() ) );

        ResultHandlerImpl resultHandler = new ResultHandlerImpl();

        getPipelineStateful( ksession ).insert( inXml,

                                                resultHandler );

        String outXml = (String) resultHandler.getObject();

        outXml = roundTripFromXml( outXml );

        ExecutionResults result = (ExecutionResults) BatchExecutionHelper.newJSonMarshaller().fromXML( outXml );

        result = ( ExecutionResults ) roundTripFromObject( result );

       

        Cheese stilton = (Cheese) result.getValue( "outStilton" );

        assertEquals( 30,

                      stilton.getPrice() );

        FactHandle factHandle = (FactHandle) result.getFactHandle( "outStilton" );

        stilton = (Cheese) ksession.getObject( factHandle );

        assertEquals( 30,

                      stilton.getPrice() );

//        expectedXml = "{\"execution-results\":{\"results\":{\"outStilton\":{\"org.drools.Cheese\":{\"type\":\"stilton\",\"oldPrice\":0,\"price\":30}}},\"fact-handles\":{\"outStilton\":\"" + ((InternalFactHandle) result.getFactHandle( "outStilton" )).toExternalForm() + "\"}}}";

//        String expectedXml = "";

//        expectedXml += "<execution-results>\n";

//        expectedXml += "  <result identifier=\"outStilton\">\n";

//        expectedXml += "    <org.drools.Cheese>\n";

//        expectedXml += "      <type>stilton</type>\n";

//        expectedXml += "      <oldPrice>0</oldPrice>\n";

//        expectedXml += "      <price>30</price>\n";

//        expectedXml += "    </org.drools.Cheese>\n";

//        expectedXml += "  </result>\n";

//        expectedXml += "  <fact-handle identifier=\"outStilton\" externalForm=\"" + ((InternalFactHandle) result.getFactHandle( "outStilton" )).toExternalForm() + "\" /> \n";

//        expectedXml += "</execution-results>\n";

//

//        assertXMLEqual( expectedXml,

//                        outXml );

//        JSONAssert.assertEquals( expectedXml, outXml );

    }

    @Test

    public void testInsertWithReturnObjectFalse() throws Exception {

        String str = "";

        str += "package org.drools \n";

        str += "import org.drools.Cheese \n";

        str += "rule rule1 \n";

        str += "  when \n";

        str += "    $c : Cheese() \n";

        str += " \n";

        str += "  then \n";

        str += "    $c.setPrice( $c.getPrice() + 5 ); \n";

        str += "end\n";

        String inXml = "";

        inXml += "{\"batch-execution\":{\"commands\":[";

        inXml += "{\"insert\":{\"object\":{\"org.drools.Cheese\":{\"type\":\"stilton\",\"price\":25,\"oldPrice\":0}}, \"return-object\":false, \"out-identifier\":\"outStilton\" } }";

        inXml += ", {\"fire-all-rules\":\"\"}";

        inXml += "]}}";

        inXml = roundTripFromXml( inXml );

        StatefulKnowledgeSession ksession = getSessionStateful( ResourceFactory.newByteArrayResource( str.getBytes() ) );

        ResultHandlerImpl resultHandler = new ResultHandlerImpl();

        getPipelineStateful( ksession ).insert( inXml,

                                                resultHandler );

        String outXml = (String) resultHandler.getObject();

        outXml = roundTripFromXml( outXml );

        ExecutionResults result = (ExecutionResults) BatchExecutionHelper.newJSonMarshaller().fromXML( outXml );

        result = ( ExecutionResults ) roundTripFromObject ( result );

        assertNull( result.getValue( "outStilton" ) );

        FactHandle factHandle = (FactHandle) result.getFactHandle( "outStilton" );

        Cheese stilton = (Cheese) ksession.getObject( factHandle );

        assertEquals( 30,

                      stilton.getPrice() );

//        expectedXml += "<execution-results>\n";

//        expectedXml += "  <fact-handle identifier=\"outStilton\" externalForm=\"" + ((InternalFactHandle) result.getFactHandle( "outStilton" )).toExternalForm() + "\" /> \n";

//        expectedXml += "</execution-results>\n";

//

//        assertXMLEqual( expectedXml,

//                        outXml );

    }

    @Test

    public void testGetObject() throws Exception {

        String str = "";

        str += "package org.drools \n";

        str += "import org.drools.Cheese \n";

        str += "rule rule1 \n";

        str += "  when \n";

        str += "    $c : Cheese() \n";

        str += " \n";

        str += "  then \n";

        str += "    $c.setPrice( $c.getPrice() + 5 ); \n";

        str += "end\n";

        String inXml = "";

        inXml += "{\"batch-execution\":{\"commands\":[";

        inXml += "{\"insert\":{\"object\":{\"org.drools.Cheese\":{\"type\":\"stilton\",\"price\":25,\"oldPrice\":0}}, \"out-identifier\":\"outStilton\" } }";

        inXml += ", {\"fire-all-rules\":{\"max\":10}}";

        inXml += "]}}";

        inXml = roundTripFromXml( inXml );

       

        StatefulKnowledgeSession ksession = getSessionStateful( ResourceFactory.newByteArrayResource( str.getBytes() ) );

        ResultHandlerImpl resultHandler = new ResultHandlerImpl();

        getPipelineStateful( ksession ).insert( inXml,

                                                resultHandler );

        String outXml = (String) resultHandler.getObject();

        outXml = roundTripFromXml( outXml );

        ExecutionResults result = (ExecutionResults) BatchExecutionHelper.newJSonMarshaller().fromXML( outXml );

        result = ( ExecutionResults) roundTripFromObject( result );

       

        Cheese stilton = (Cheese) result.getValue( "outStilton" );

        assertEquals( 30,

                      stilton.getPrice() );

        FactHandle factHandle = (FactHandle) result.getFactHandle( "outStilton" );

        inXml = "";

        inXml += "{\"batch-execution\":{\"commands\":[";

        inXml += "  {\"get-object\":{ ";

        inXml += "      \"out-identifier\":'outStilton',";

        inXml += "      \"fact-handle\":'" + factHandle.toExternalForm() + "'}}";

        inXml += "]}}";

        inXml = roundTripFromXml( inXml );

     

        getPipelineStateful( ksession ).insert( inXml,

                                                resultHandler );

        outXml = (String) resultHandler.getObject();

        outXml = roundTripFromXml( outXml );

       

        result = (ExecutionResults) BatchExecutionHelper.newJSonMarshaller().fromXML( outXml );

        result = ( ExecutionResults) roundTripFromObject( result );

        stilton = (Cheese) result.getValue( "outStilton" );

        assertEquals( 30,

                      stilton.getPrice() );

    }

    @Test

    public void testRetractObject() throws Exception {

        String str = "";

        str += "package org.drools \n";

        str += "import org.drools.Cheese \n";

        str += "rule rule1 \n";

        str += "  when \n";

        str += "    $c : Cheese() \n";

        str += " \n";

        str += "  then \n";

        str += "    $c.setPrice( $c.getPrice() + 5 ); \n";

        str += "end\n";

        String inXml = "";

        inXml += "{\"batch-execution\":{\"commands\":[";

        inXml += "{\"insert\":{\"object\":{\"org.drools.Cheese\":{\"type\":\"stilton\",\"price\":25,\"oldPrice\":0}}, \"out-identifier\":\"outStilton\" } }";

        inXml += ", {\"fire-all-rules\":{\"max\":10}}";

        inXml += "]}}";

        inXml = roundTripFromXml( inXml );

        StatefulKnowledgeSession ksession = getSessionStateful( ResourceFactory.newByteArrayResource( str.getBytes() ) );

        ResultHandlerImpl resultHandler = new ResultHandlerImpl();

        getPipelineStateful( ksession ).insert( inXml,

                                                resultHandler );

        String outXml = (String) resultHandler.getObject();

        outXml = roundTripFromXml( outXml );

        ExecutionResults result = (ExecutionResults) BatchExecutionHelper.newJSonMarshaller().fromXML( outXml );

        result = ( ExecutionResults) roundTripFromObject( result );

       

        Cheese stilton = (Cheese) result.getValue( "outStilton" );

        assertEquals( 30,

                      stilton.getPrice() );

        FactHandle factHandle = (FactHandle) result.getFactHandle( "outStilton" );

        inXml = "";

        inXml += "{\"batch-execution\":{\"commands\":[";

        inXml += "  { \"retract\":{\"fact-handle\":'" + factHandle.toExternalForm() + "'}}";

        inXml += "]}}";

        inXml = roundTripFromXml( inXml );

        getPipelineStateful( ksession ).insert( inXml,

                                                resultHandler );

        inXml = "";

        inXml += "{\"batch-execution\":{\"commands\":[";

        inXml += "  {\"get-object\":{ ";

        inXml += "      \"out-identifier\":'outStilton',";

        inXml += "      \"fact-handle\":'" + factHandle.toExternalForm() + "'}}";

        inXml += "]}}";

        inXml = roundTripFromXml( inXml );

        getPipelineStateful( ksession ).insert( inXml,

                                                resultHandler );

        outXml = (String) resultHandler.getObject();

        result = (ExecutionResults) BatchExecutionHelper.newJSonMarshaller().fromXML( outXml );

        result = ( ExecutionResults) roundTripFromObject( result );

        assertNull( result.getValue( "outStilton" ) );

    }

    @Test

    public void testModifyObject() throws Exception {

        String str = "";

        str += "package org.drools \n";

        str += "import org.drools.Cheese \n";

        str += "rule rule1 \n";

        str += "  when \n";

        str += "    $c : Cheese() \n";

        str += " \n";

        str += "  then \n";

        str += "    $c.setPrice( $c.getPrice() + 5 ); \n";

        str += "end\n";

        String inXml = "";

        inXml += "{\"batch-execution\":{\"commands\":[";

        inXml += "{\"insert\":{\"object\":{\"org.drools.Cheese\":{\"type\":\"stilton\",\"price\":25,\"oldPrice\":0}}, \"out-identifier\":\"outStilton\" } }";

        inXml += ", {\"fire-all-rules\":\"\"}";

        inXml += "]}}";

        inXml = roundTripFromXml( inXml );

        StatefulKnowledgeSession ksession = getSessionStateful( ResourceFactory.newByteArrayResource( str.getBytes() ) );

        ResultHandlerImpl resultHandler = new ResultHandlerImpl();

        getPipelineStateful( ksession ).insert( inXml,

                                                resultHandler );

        String outXml = (String) resultHandler.getObject();

        outXml = roundTripFromXml( outXml );

        ExecutionResults result = (ExecutionResults) BatchExecutionHelper.newJSonMarshaller().fromXML( outXml );

        result = ( ExecutionResults ) roundTripFromObject( result );

        Cheese stilton = (Cheese) result.getValue( "outStilton" );

        assertEquals( 30,

                      stilton.getPrice() );

        FactHandle factHandle = ((FactHandle) result.getFactHandle( "outStilton" ));

//        String expectedXml = "";

//        expectedXml += "<execution-results>\n";

//        expectedXml += "  <result identifier=\"outStilton\">\n";

//        expectedXml += "    <org.drools.Cheese>\n";

//        expectedXml += "      <type>stilton</type>\n";

//        expectedXml += "      <oldPrice>0</oldPrice>\n";

//        expectedXml += "      <price>30</price>\n";

//        expectedXml += "    </org.drools.Cheese>\n";

//        expectedXml += "  </result>\n";

//        expectedXml += "  <fact-handle identifier=\"outStilton\" externalForm=\"" + factHandle.toExternalForm() + "\" /> \n";

//        expectedXml += "</execution-results>\n";

//

//        assertXMLEqual( expectedXml,

//                        outXml );

       

        inXml = "";

        inXml += "{\"batch-execution\":{\"commands\":[";

        inXml += "  {\"modify\":{\"fact-handle\":'" + factHandle.toExternalForm() + "'";

        inXml += ",                \"setters\":[{\"accessor\":\"oldPrice\",\"set\":42}, {\"accessor\":\"price\",\"set\":50}]";

        inXml += "} }";

        inXml += ", {\"fire-all-rules\":\"\"}";

        inXml += "]}}";

        inXml = roundTripFromXml( inXml );

       

        getPipelineStateful( ksession ).insert( inXml,

                                                resultHandler );

        inXml = "";

        inXml += "{\"batch-execution\":{\"commands\":[";

        inXml += "  {\"get-object\":{ ";

        inXml += "      \"out-identifier\":'outCheddar',";

        inXml += "      \"fact-handle\":'" + factHandle.toExternalForm() + "'}}";

        inXml += "]}}";

        inXml = roundTripFromXml( inXml );

       

        getPipelineStateful( ksession ).insert( inXml,

                                                resultHandler );

        outXml = (String) resultHandler.getObject();

        outXml = roundTripFromXml( outXml );

        result = (ExecutionResults) BatchExecutionHelper.newJSonMarshaller().fromXML( outXml );

        result = ( ExecutionResults ) roundTripFromObject ( result );

        Cheese cheddar = (Cheese) result.getValue( "outCheddar" );

        assertEquals( 42,

                      cheddar.getOldPrice() );

        assertEquals( 55,

                      cheddar.getPrice() );

        //now test for code injection:

        ModifyCommand.ALLOW_MODIFY_EXPRESSIONS = false;

        inXml = "";

        inXml += "{\"batch-execution\":{\"commands\":[";

        inXml += "  {\"modify\":{\"fact-handle\":'" + factHandle.toExternalForm() + "'";

        inXml += ",                \"setters\":[{\"accessor\":\"type\",\"set\":\"44; System.exit(1);\"}, {\"accessor\":\"price\",\"set\":50}]";

        inXml += "} }";

        inXml += ", {\"fire-all-rules\":\"\"}";

        inXml += "]}}";

        inXml = roundTripFromXml( inXml );

        getPipelineStateful( ksession ).insert( inXml,

                                                resultHandler );

        outXml = (String) resultHandler.getObject();

        result = (ExecutionResults) BatchExecutionHelper.newJSonMarshaller().fromXML( outXml );

        ModifyCommand.ALLOW_MODIFY_EXPRESSIONS = true;

    }

    @Test

    public void testInsertElements() throws Exception {

        String str = "";

        str += "package org.drools \n";

        str += "import org.drools.Cheese \n";

        str += "global java.util.List list1 \n";

        str += "rule rule1 \n";

        str += "  when \n";

        str += "    $c : Cheese() \n";

        str += " \n";

        str += "  then \n";

        str += "    $c.setPrice( $c.getPrice() + 5 ); \n";

        str += "     list1.add( $c );";

        str += "end\n";

       

        String inXml = "";

        inXml += "{\"batch-execution\":{\"commands\":[";

        inXml += "   {\"set-global\":{\"identifier\":\"list1\",\"out\"=true";

        inXml += "                   ,\"object\":{\"list\":{\"object\":[]}}";

        inXml += "   } } "; //       

        inXml += ",  {\"insert-elements\":{\"objects\":[";

        inXml += "   {   \"org.drools.Cheese\":{\"type\":\"stilton\",\"price\":25,\"oldPrice\":0}}, ";

        inXml += "   {   \"org.drools.Cheese\":{\"type\":\"stilton\",\"price\":30,\"oldPrice\":0}} ";

        inXml += "   ]}}";

        inXml += "]}}";

        inXml = roundTripFromXml( inXml );

        StatelessKnowledgeSession ksession = getSession2( ResourceFactory.newByteArrayResource( str.getBytes() ) );

        ResultHandlerImpl resultHandler = new ResultHandlerImpl();

        getPipeline( ksession ).insert( inXml,

                                        resultHandler );

        String outXml = (String) resultHandler.getObject();

        outXml = roundTripFromXml( outXml );

       

//        expectedXml += "<execution-results>\n";

//        expectedXml += "  <result identifier='list'>\n";

//        expectedXml += "    <list>\n";

//        expectedXml += "      <org.drools.Cheese>\n";

//        expectedXml += "        <type>stilton</type>\n";

//        expectedXml += "        <price>35</price>\n";

//        expectedXml += "        <oldPrice>0</oldPrice>\n";

//        expectedXml += "      </org.drools.Cheese>\n";

//        expectedXml += "      <org.drools.Cheese>\n";

//        expectedXml += "        <type>stilton</type>\n";

//        expectedXml += "        <price>30</price>\n";

//        expectedXml += "        <oldPrice>0</oldPrice>\n";

//        expectedXml += "      </org.drools.Cheese>\n";

//        expectedXml += "    </list>\n";

//        expectedXml += "  </result>\n";

//        expectedXml += "</execution-results>\n";

//

//        assertXMLEqual( expectedXml,

//                        outXml );

//

        ExecutionResults result = (ExecutionResults) BatchExecutionHelper.newJSonMarshaller().fromXML( outXml );

        result = ( ExecutionResults ) roundTripFromObject( result );

        List list = (List) result.getValue( "list1" );

        Cheese stilton25 = new Cheese( "stilton",

                                       30 );

        Cheese stilton30 = new Cheese( "stilton",

                                       35 );

        Set expectedList = new HashSet();

        expectedList.add( stilton25 );

        expectedList.add( stilton30 );

        assertEquals( expectedList,

                      new HashSet( list ) );

    }

    @Test

    public void testFactHandleReturn() throws Exception {

        String str = "";

        str += "package org.drools \n";

        str += "import org.drools.Cheese \n";

        str += "global java.util.List list1 \n";

        str += "rule rule1 \n";

        str += "  when \n";

        str += "    $c : Cheese() \n";

        str += " \n";

        str += "  then \n";

        str += "    System.err.println(42); \n";

        str += "end\n";

        String inXml = "";

        inXml += "{\"batch-execution\":{\"commands\":[";

        inXml += "  {\"get-global\":{\"identifier\":\"list1\",\"out-identifier\":\"out-list\"}}";

        inXml += "]}}";

        inXml = roundTripFromXml( inXml );

        StatefulKnowledgeSession ksession = getSessionStateful( ResourceFactory.newByteArrayResource( str.getBytes() ) );

        FactHandle fh = ksession.insert( new Person( "mic",

                                                     42 ) );

        List<FactHandle> list = new ArrayList<FactHandle>();

        list.add( fh );

        ksession.setGlobal( "list1",

                            list );

        ResultHandlerImpl resultHandler = new ResultHandlerImpl();

        getPipelineStateful( ksession ).insert( inXml,

                                                resultHandler );

        getPipelineStateful( ksession ).insert( inXml,

                                                resultHandler );

        String outXml = (String) resultHandler.getObject();

        outXml = roundTripFromXml( outXml );

       

        ExecutionResults result = (ExecutionResults) BatchExecutionHelper.newJSonMarshaller().fromXML( outXml );

        result = ( ExecutionResults ) roundTripFromObject( result );

        List outList = ( List ) result.getValue( "out-list" );

        assertEquals( 1, outList.size() );

        assertEquals( fh.toExternalForm(), ((FactHandle)outList.get(0)).toExternalForm() );

        assertNotSame( fh, outList.get( 0 ));

       

//        String expectedXml = "";

//        expectedXml += "<execution-results>\n" + "  <result identifier=\"out-list\">\n" + "    <list>\n" + "      <fact-handle externalForm=\"" + fh.toExternalForm() + "\"/>\n" + "    </list>\n" + "  </result>\n" + "</execution-results>";

//

//        assertXMLEqual( expectedXml,

//                        outXml );

    }

    @Test

    public void testInsertElementsWithReturnObjects() throws Exception {

        String str = "";

        str += "package org.drools \n";

        str += "import org.drools.Cheese \n";

        str += "global java.util.List list1 \n";

        str += "rule rule1 \n";

        str += "  when \n";

        str += "    $c : Cheese() \n";

        str += " \n";

        str += "  then \n";

        str += "    $c.setPrice( $c.getPrice() + 5 ); \n";

        str += "    list1.add( $c );";

        str += "end\n";

       

        String inXml = "";

        inXml += "{\"batch-execution\":{\"commands\":[";

        inXml += "   {\"set-global\":{\"identifier\":\"list1\",\"out\"=true";

        inXml += "                   ,\"object\":{\"list\":{\"object\":[]}}";

        inXml += "   } } "; //       

        inXml += ",  {\"insert-elements\":{\"out-identifier\":\"myfacts\",\"return-objects\":true,\"objects\":[";

        inXml += "   {   \"org.drools.Cheese\":{\"type\":\"stilton\",\"price\":25,\"oldPrice\":0}}, ";

        inXml += "   {   \"org.drools.Cheese\":{\"type\":\"stilton\",\"price\":30,\"oldPrice\":0}} ";

        inXml += "   ]}}";

        inXml += ", {\"fire-all-rules\":\"\"}";

        inXml += "]}}";

        inXml = roundTripFromXml( inXml );

        StatefulKnowledgeSession ksession = getSessionStateful( ResourceFactory.newByteArrayResource( str.getBytes() ) );

        ResultHandlerImpl resultHandler = new ResultHandlerImpl();

        getPipelineStateful( ksession ).insert( inXml,

                                                resultHandler );

        String outXml = (String) resultHandler.getObject();

        outXml = roundTripFromXml( outXml );

       

        ExecutionResults result = (ExecutionResults) BatchExecutionHelper.newJSonMarshaller().fromXML( outXml );

        result = ( ExecutionResults) roundTripFromObject( result );

       

        List list1 = ( List ) result.getValue( "list1" );

        assertEquals( 2, list1.size() );

        assertTrue( list1.contains( new Cheese("stilton", 35) ) );

        assertTrue( list1.contains( new Cheese("stilton", 30) ) );

       

        List myFacts = ( List ) result.getValue( "myfacts" );

        assertEquals( 2, list1.size() );

        assertTrue( myFacts.contains( new Cheese("stilton", 35) ) );

        assertTrue( myFacts.contains( new Cheese("stilton", 30) ) );

       

        List factHandles = ( List ) result.getFactHandle( "myfacts" );

        List list = new ArrayList();

        list.add( ksession.getObject( ((InternalFactHandle)factHandles.get( 0 )) ) );

        list.add( ksession.getObject( ((InternalFactHandle)factHandles.get( 1 )) ) );

        assertTrue( list.contains( new Cheese("stilton", 35) ) );

        assertTrue( list.contains( new Cheese("stilton", 30) ) );

//        expectedXml += "<execution-results>\n";

//        expectedXml += "  <result identifier='list'>\n";

//        expectedXml += "    <list>\n";

//        expectedXml += "      <org.drools.Cheese>\n";

//        expectedXml += "        <type>stilton</type>\n";

//        expectedXml += "        <price>35</price>\n";

//        expectedXml += "        <oldPrice>0</oldPrice>\n";

//        expectedXml += "      </org.drools.Cheese>\n";

//        expectedXml += "      <org.drools.Cheese>\n";

//        expectedXml += "        <type>stilton</type>\n";

//        expectedXml += "        <price>30</price>\n";

//        expectedXml += "        <oldPrice>0</oldPrice>\n";

//        expectedXml += "      </org.drools.Cheese>\n";

//        expectedXml += "    </list>\n";

//        expectedXml += "  </result>\n";

//

//        expectedXml += "  <result identifier=\"myfacts\">\n";

//        expectedXml += "  <list>\n";

//        expectedXml += "    <org.drools.Cheese reference=\"../../../result/list/org.drools.Cheese[2]\"/>\n";

//        expectedXml += "    <org.drools.Cheese reference=\"../../../result/list/org.drools.Cheese\"/>\n";

//        expectedXml += "  </list>\n";

//        expectedXml += "  </result>\n";

//        expectedXml += "  <fact-handles identifier=\"myfacts\">\n";

//        for ( FactHandle factHandle : factHandles ) {

//            if ( ((Cheese) ksession.getObject( factHandle )).getPrice() == 30 ) {

//                expectedXml += "  <fact-handle externalForm=\"" + factHandle.toExternalForm() + "\"/>\n";

//            }

//        }

//

//        for ( FactHandle factHandle : factHandles ) {

//            if ( ((Cheese) ksession.getObject( factHandle )).getPrice() == 35 ) {

//                expectedXml += "  <fact-handle externalForm=\"" + factHandle.toExternalForm() + "\"/>\n";

//            }

//        }

//        expectedXml += "  </fact-handles>\n";

//

//        expectedXml += "</execution-results>\n";

//

//        assertXMLEqual( expectedXml,

//                        outXml );

    }

    @Test

    public void testSetGlobal() throws Exception {

        String str = "";

        str += "package org.drools \n";

        str += "import org.drools.Cheese \n";

        str += "global java.util.List list1 \n";

        str += "global java.util.List list2 \n";

        str += "global java.util.List list3 \n";

        str += "rule rule1 \n";

        str += "  when \n";

        str += "    $c : Cheese() \n";

        str += " \n";

        str += "  then \n";

        str += "    $c.setPrice( 30 ); \n";

        str += "    list1.add( $c ); \n";

        str += "    list2.add( $c ); \n";

        str += "    list3.add( $c ); \n";

        str += "end\n";

       

        String inXml = "";

        inXml += "{\"batch-execution\":{\"commands\":[";

        inXml += " {\"set-global\":{\"identifier\":\"list1\"";

        inXml += "                  ,\"object\":{\"list\":{\"object\":[]}}";

        inXml += "} } ";

        inXml += ", {\"set-global\":{\"identifier\":\"list2\",\"out\"=true";

        inXml += "                  ,\"object\":{\"list\":{\"object\":[]}}";

        inXml += "} } ";

        inXml += ", {\"set-global\":{\"identifier\":\"list3\",\"out-identifier\"=\"outList3\"";

        inXml += "                  ,\"object\":{\"list\":{\"object\":[]}}";

        inXml += "} } ";

        inXml += ", {\"insert\":{\"object\":{\"org.drools.Cheese\":{\"type\":\"stilton\",\"price\":5}}, \"out-identifier\":\"outStilton\" } }";

        inXml += "]}}";

        inXml = roundTripFromXml( inXml );

        StatelessKnowledgeSession ksession = getSession2( ResourceFactory.newByteArrayResource( str.getBytes() ) );

        ResultHandlerImpl resultHandler = new ResultHandlerImpl();

        getPipeline( ksession ).insert( inXml,

                                        resultHandler );

        String outXml = (String) resultHandler.getObject();

        outXml = roundTripFromXml( outXml );

//        expectedXml += "<execution-results>\n";

//        expectedXml += "  <result identifier='list2'>\n";

//        expectedXml += "    <list>\n";

//        expectedXml += "      <org.drools.Cheese>\n";

//        expectedXml += "        <type>stilton</type>\n";

//        expectedXml += "        <price>30</price>\n";

//        expectedXml += "        <oldPrice>0</oldPrice>\n";

//        expectedXml += "      </org.drools.Cheese>\n";

//        expectedXml += "    </list>\n";

//        expectedXml += "  </result>\n";

//        expectedXml += "  <result identifier='outList3'>\n";

//        expectedXml += "    <list>\n";

//        expectedXml += "      <org.drools.Cheese reference='../../../result/list/org.drools.Cheese'/>\n";

//        expectedXml += "    </list>\n";

//        expectedXml += "  </result>\n";

//        expectedXml += "</execution-results>\n";

//

//        assertXMLEqual( expectedXml,

//                        outXml );

//

        ExecutionResults result = (ExecutionResults) BatchExecutionHelper.newJSonMarshaller().fromXML( outXml );

        Cheese stilton = new Cheese( "stilton",

                                     30 );

        assertNull( result.getValue( "list1" ) );

        List list2 = (List) result.getValue( "list2" );

        assertEquals( 1,

                      list2.size() );

        assertEquals( stilton,

                      list2.get( 0 ) );

        List list3 = (List) result.getValue( "outList3" );

        assertEquals( 1,

                      list3.size() );

        assertEquals( stilton,

                      list3.get( 0 ) );

    }

    @Test

    public void testGetGlobal() throws Exception {

        String str = "";

        str += "package org.drools \n";

        str += "import org.drools.Cheese \n";

        str += "global java.util.List list1 \n";

        str += "rule rule1 \n";

        str += "  when \n";

        str += "    $c : Cheese() \n";

        str += " \n";

        str += "  then \n";

        str += "    list1.add( $c ); \n";

        str += "end\n";

       

        String inXml = "";

        inXml += "{\"batch-execution\":{\"commands\":[";

        inXml += "   {\"set-global\":{\"identifier\":\"list1\",\"out\"=true";

        inXml += "                   ,\"object\":{\"list\":{\"object\":[]}}";

        inXml += "   } } "; //       

        inXml += ",  {\"insert\":{\"object\":{\"org.drools.Cheese\":{\"type\":\"stilton\",\"price\":25,\"oldPrice\":0}}, \"out-identifier\":\"outStilton\" } }";

        inXml += ",  {\"get-global\":{\"identifier\":\"list1\",\"out-identifier\"=\"out-list\"}}";

        inXml += "]}}";

        inXml = roundTripFromXml( inXml );

        StatelessKnowledgeSession ksession = getSession2( ResourceFactory.newByteArrayResource( str.getBytes() ) );

        ResultHandlerImpl resultHandler = new ResultHandlerImpl();

        getPipeline( ksession ).insert( inXml,

                                        resultHandler );

        String outXml = (String) resultHandler.getObject();

        outXml = roundTripFromXml( outXml );

       

        ExecutionResults result = (ExecutionResults) BatchExecutionHelper.newJSonMarshaller().fromXML( outXml );

        result = ( ExecutionResults ) roundTripFromObject( result );

        List resultsList = ( List ) result.getValue("out-list");

        assertEquals(1, resultsList.size() );

        assertEquals(new Cheese("stilton",25), resultsList.get(0));

//        expectedXml += "<execution-results>\n";

//        expectedXml += "  <result identifier=\"out-list\">\n";

//        expectedXml += "    <list>\n";

//        expectedXml += "      <org.drools.Cheese>\n";

//        expectedXml += "        <type>stilton</type>\n";

//        expectedXml += "        <price>25</price>\n";

//        expectedXml += "        <oldPrice>0</oldPrice>\n";

//        expectedXml += "      </org.drools.Cheese>\n";

//        expectedXml += "    </list>\n";

//        expectedXml += "  </result>\n";

//        expectedXml += "</execution-results>\n";

//

//        assertXMLEqual( expectedXml,

//                        outXml );

    }

    @Test

    public void testGetObjects() throws Exception {

        String str = "";

        str += "package org.drools \n";

        str += "import org.drools.Cheese \n";

        str += "rule rule1 \n";

        str += "  when \n";

        str += "    $c : Cheese() \n";

        str += " \n";

        str += "  then \n";

        str += "    $c.setPrice( $c.getPrice() + 5 ); \n";

        str += "end\n";

        String inXml ="";

        inXml =  "{\"batch-execution\":{\"commands\":[";

        inXml += "  {\"insert-elements\":{\"objects\":[";

        inXml += "   {   \"org.drools.Cheese\":{\"type\":\"stilton\",\"price\":25,\"oldPrice\":0}}, ";

        inXml += "   {   \"org.drools.Cheese\":{\"type\":\"stilton\",\"price\":30,\"oldPrice\":0}} ";

        inXml += "   ]}}";

        inXml += ",  {\"get-objects\":{\"out-identifier\":\"list1\"}}";

        inXml += "]}}";

        inXml = roundTripFromXml( inXml );

       

        StatelessKnowledgeSession ksession = getSession2( ResourceFactory.newByteArrayResource( str.getBytes() ) );

        ResultHandlerImpl resultHandler = new ResultHandlerImpl();

        getPipeline( ksession ).insert( inXml,

                                        resultHandler );

        String outXml = (String) resultHandler.getObject();

        outXml = roundTripFromXml( outXml );

//        expectedXml += "<execution-results>";

//        expectedXml += "  <result identifier='list'>";

//        expectedXml += "    <list>";

//        expectedXml += "      <org.drools.Cheese>";

//        expectedXml += "        <type>stilton</type>";

//        expectedXml += "        <price>30</price>";

//        expectedXml += "        <oldPrice>0</oldPrice>";

//        expectedXml += "      </org.drools.Cheese>";

//        expectedXml += "      <org.drools.Cheese>";

//        expectedXml += "        <type>stilton</type>";

//        expectedXml += "        <price>35</price>";

//        expectedXml += "        <oldPrice>0</oldPrice>";

//        expectedXml += "      </org.drools.Cheese>";

//        expectedXml += "    </list>";

//        expectedXml += "  </result>";

//        expectedXml += "</execution-results>";

//

//        assertXMLEqual( expectedXml,

//                        outXml );

//

        ExecutionResults result = (ExecutionResults) BatchExecutionHelper.newJSonMarshaller().fromXML( outXml );

        result = ( ExecutionResults ) roundTripFromObject( result );

        List list = (List) result.getValue( "list1" );

        Cheese stilton25 = new Cheese( "stilton",

                                       30 );

        Cheese stilton30 = new Cheese( "stilton",

                                       35 );

        Set expectedList = new HashSet();

        expectedList.add( stilton25 );

        expectedList.add( stilton30 );

        assertEquals( expectedList,

                      new HashSet( list ) );

    }

    @Test

    public void testQuery() throws Exception {

        String str = "";

        str += "package org.drools.test  \n";

        str += "import org.drools.Cheese \n";

        str += "query cheeses \n";

        str += "    stilton : Cheese(type == 'stilton') \n";

        str += "    cheddar : Cheese(type == 'cheddar', price == stilton.price) \n";

        str += "end\n";

        str += "query cheesesWithParams(String a, String b) \n";

        str += "    stilton : Cheese(type == a) \n";

        str += "    cheddar : Cheese(type == b, price == stilton.price) \n";

        str += "end\n";

        String inXml ="";

        inXml =  "{\"batch-execution\":{\"commands\":[";

        inXml += "  {\"insert-elements\":{\"objects\":[";

        inXml += "   {   \"org.drools.Cheese\":{\"type\":\"stilton\",\"price\":2}}, ";

        inXml += "   {   \"org.drools.Cheese\":{\"type\":\"cheddar\",\"price\":1}}, ";

        inXml += "   {   \"org.drools.Cheese\":{\"type\":\"cheddar\",\"price\":2}}, ";

        inXml += "   {   \"org.drools.Cheese\":{\"type\":\"stilton\",\"price\":1}}  ";

        inXml += "   ]}}";

        inXml += ",  {\"query\":{\"out-identifier\":\"cheeses\",\"name\":\"cheeses\"}}";

        inXml += "]}}";

        inXml = roundTripFromXml( inXml );

       

        StatefulKnowledgeSession ksession = getSessionStateful( ResourceFactory.newByteArrayResource( str.getBytes() ) );

        ResultHandlerImpl resultHandler = new ResultHandlerImpl();

        getPipelineStateful( ksession ).insert( inXml,

                                                resultHandler );

        String outXml = (String) resultHandler.getObject();

        outXml = roundTripFromXml( outXml );

//        Iterator<QueryResultsRow> it2 = ksession.getQueryResults( "cheesesWithParams",

//                                                                  new String[]{"stilton", "cheddar"} ).iterator();

//        QueryResultsRow row = null;

//

//        String expectedXml = "";

//        expectedXml += "<execution-results>\n";

//        expectedXml += "  <result identifier='cheeses'>\n";

//        expectedXml += "    <query-results>\n";

//        expectedXml += "      <identifiers>\n";

//        expectedXml += "        <identifier>stilton</identifier>\n";

//        expectedXml += "        <identifier>cheddar</identifier>\n";

//        expectedXml += "      </identifiers>\n";

//        expectedXml += "      <row>\n";

//        row = it1.next();

//        expectedXml += "        <org.drools.Cheese>\n";

//        expectedXml += "          <type>stilton</type>\n";

//        expectedXml += "          <price>1</price>\n";

//        expectedXml += "          <oldPrice>0</oldPrice>\n";

//        expectedXml += "        </org.drools.Cheese>\n";

//        expectedXml += "        <fact-handle externalForm='" + row.getFactHandle( "stilton" ).toExternalForm() + "' />";

//        expectedXml += "        <org.drools.Cheese>\n";

//        expectedXml += "          <type>cheddar</type>\n";

//        expectedXml += "          <price>1</price>\n";

//        expectedXml += "          <oldPrice>0</oldPrice>\n";

//        expectedXml += "        </org.drools.Cheese>\n";

//        expectedXml += "        <fact-handle externalForm='" + row.getFactHandle( "cheddar" ).toExternalForm() + "' />";

//        expectedXml += "      </row>\n";

//        expectedXml += "      <row>\n";

//        row = it1.next();

//        expectedXml += "        <org.drools.Cheese>\n";

//        expectedXml += "          <type>stilton</type>\n";

//        expectedXml += "          <price>2</price>\n";

//        expectedXml += "          <oldPrice>0</oldPrice>\n";

//        expectedXml += "        </org.drools.Cheese>\n";

//        expectedXml += "        <fact-handle externalForm='" + row.getFactHandle( "stilton" ).toExternalForm() + "' />";

//        expectedXml += "        <org.drools.Cheese>\n";

//        expectedXml += "          <type>cheddar</type>\n";

//        expectedXml += "          <price>2</price>\n";

//        expectedXml += "          <oldPrice>0</oldPrice>\n";

//        expectedXml += "        </org.drools.Cheese>\n";

//        expectedXml += "        <fact-handle externalForm='" + row.getFactHandle( "cheddar" ).toExternalForm() + "' />";

//        expectedXml += "      </row>\n";

//        expectedXml += "    </query-results>\n";

//        expectedXml += "  </result>\n";

//        expectedXml += "  <result identifier='cheeses2'>\n";

//        expectedXml += "    <query-results>\n";

//        expectedXml += "      <identifiers>\n";

//        expectedXml += "        <identifier>stilton</identifier>\n";

//        expectedXml += "        <identifier>cheddar</identifier>\n";

//        expectedXml += "      </identifiers>\n";

//        expectedXml += "      <row>\n";

//        row = it2.next();

//        expectedXml += "        <org.drools.Cheese reference=\"../../../../result/query-results/row/org.drools.Cheese\"/>\n";

//        expectedXml += "        <fact-handle externalForm='" + row.getFactHandle( "stilton" ).toExternalForm() + "' />";

//        expectedXml += "        <org.drools.Cheese reference=\"../../../../result/query-results/row/org.drools.Cheese[2]\"/>\n";

//        expectedXml += "        <fact-handle externalForm='" + row.getFactHandle( "cheddar" ).toExternalForm() + "' />";

//        expectedXml += "      </row>\n";

//        expectedXml += "      <row>\n";

//        row = it2.next();

//        expectedXml += "        <org.drools.Cheese reference=\"../../../../result/query-results/row[2]/org.drools.Cheese\"/>\n";

//        expectedXml += "        <fact-handle externalForm='" + row.getFactHandle( "stilton" ).toExternalForm() + "' />";

//        expectedXml += "        <org.drools.Cheese reference=\"../../../../result/query-results/row[2]/org.drools.Cheese[2]\"/>\n";

//        expectedXml += "        <fact-handle externalForm='" + row.getFactHandle( "cheddar" ).toExternalForm() + "' />";

//        expectedXml += "      </row>\n";

//        expectedXml += "    </query-results>\n";

//        expectedXml += "  </result>\n";

//        expectedXml += "</execution-results>\n";

//

//        assertXMLEqual( expectedXml,

//                        outXml );

//

        ExecutionResults batchResult = (ExecutionResults) BatchExecutionHelper.newJSonMarshaller().fromXML( outXml );

        batchResult = ( ExecutionResults ) roundTripFromObject( batchResult );

        Cheese stilton1 = new Cheese( "stilton",

                                      1 );

        Cheese cheddar1 = new Cheese( "cheddar",

                                      1 );

        Cheese stilton2 = new Cheese( "stilton",

                                      2 );

        Cheese cheddar2 = new Cheese( "cheddar",

                                      2 );

        Set set = new HashSet();

        List list = new ArrayList();

        list.add( stilton1 );

        list.add( cheddar1 );

        set.add( list );

        list = new ArrayList();

        list.add( stilton2 );

        list.add( cheddar2 );

        set.add( list );

        org.drools.runtime.rule.QueryResults results = (org.drools.runtime.rule.QueryResults) batchResult.getValue( "cheeses" );

        assertEquals( 2,

                      results.size() );

        assertEquals( 2,

                      results.getIdentifiers().length );

        Set newSet = new HashSet();

        for ( org.drools.runtime.rule.QueryResultsRow result : results ) {

            list = new ArrayList();

            list.add( result.get( "stilton" ) );

            list.add( result.get( "cheddar" ) );

            newSet.add( list );

        }

        assertEquals( set,

                      newSet );

    }

    @Test

    public void testManualFireAllRules() throws Exception {

        String str = "";

        str += "package org.drools \n";

        str += "import org.drools.Cheese \n";

        str += "global java.util.List list1 \n";

        str += "rule rule1 \n";

        str += "  when \n";

        str += "    $c : Cheese() \n";

        str += " \n";

        str += "  then \n";

        str += "    $c.setPrice( $c.getPrice() + 5 ); \n";

        str += "    list1.add( $c );";

        str += "end\n";

        String inXml = "";

        inXml += "{\"batch-execution\":{\"commands\":[";

        inXml += "   {\"set-global\":{\"identifier\":\"list1\",\"out\"=true";

        inXml += "                   ,\"object\":{\"list\":{\"object\":[]}}";

        inXml += "   } } "; //       

        inXml += ",  {\"insert-elements\":{\"objects\":[";

        inXml += "   {   \"org.drools.Cheese\":{\"type\":\"stilton\",\"price\":25,\"oldPrice\":0}}, ";

        inXml += "   {   \"org.drools.Cheese\":{\"type\":\"stilton\",\"price\":30,\"oldPrice\":0}} ";

        inXml += "   ]}}";

        inXml += ", {\"fire-all-rules\":\"\"}";

        inXml += ", {\"insert\":{\"object\":{\"org.drools.Cheese\":{\"type\":\"brie\",\"price\":10,\"oldPrice\":0}}, \"out-identifier\":\"outBrie\" } }";

        inXml += "]}}";

        inXml = roundTripFromXml( inXml );

       

        StatelessKnowledgeSession ksession = getSession2( ResourceFactory.newByteArrayResource( str.getBytes() ) );

        ResultHandlerImpl resultHandler = new ResultHandlerImpl();

        getPipeline( ksession ).insert( inXml,

                                        resultHandler );

        String outXml = (String) resultHandler.getObject();

        FactHandle factHandle = (FactHandle) ((ExecutionResults) BatchExecutionHelper.newJSonMarshaller().fromXML( outXml )).getFactHandle( "outBrie" );

//        expectedXml += "<execution-results>\n";