DroolsTest.java

package org.drools.integrationtests;

import java.io.StringReader;

import junit.framework.TestCase;

import org.drools.RuleBase;
import org.drools.RuleBaseFactory;
import org.drools.WorkingMemory;
import org.drools.compiler.PackageBuilder;

public class DroolsTest extends TestCase {
    private final static int NUM_FACTS = 20;

    private static int       counter;

    public static class Foo {
        private final int id;

        Foo(int id) {
            this.id = id;
        }

        public int getId() {
            return id;
        }
    }

    public static class Bar {
        private final int id;

        Bar(int id) {
            this.id = id;
        }

        public int getId() {
            return id;
        }
    }

    public static void test1() throws Exception {
        String str = "package org.drools.integrationtests;\n";
        str += "import org.drools.integrationtests.DroolsTest;\n";
        str += "import org.drools.integrationtests.DroolsTest.Foo;\n";
        str += "import org.drools.integrationtests.DroolsTest.Bar;\n";
        str += "rule test\n";
        str += "when\n";
        str += "      Foo($p : id, id < " + Integer.toString( NUM_FACTS ) + ")\n";
        str += "      Bar(id == $p)\n";
        str += "then\n";
        str += "   DroolsTest.incCounter();\n";
        str += "end\n";

        counter = 0;

        RuleBase rb = RuleBaseFactory.newRuleBase();
        WorkingMemory wm = rb.newStatefulSession();

        for ( int i = 0; i < NUM_FACTS; i++ ) {
            wm.insert( new Foo( i ) );
            wm.insert( new Bar( i ) );
        }
        PackageBuilder bld = new PackageBuilder();
        bld.addPackageFromDrl( new StringReader( str ) );
        if ( bld.hasErrors() ) {
            fail( bld.getErrors().toString() );
        }

        rb.addPackage( bld.getPackage() );
        wm.fireAllRules();
        System.out.println( counter + ":" + (counter == NUM_FACTS ? "passed" : "failed" ));
    }

    public static void incCounter() {
        ++counter;
    }
}