package org.drools.compiler.util.debug;
import java.awt.Dimension;
import java.util.ArrayList;
import org.drools.compiler.Cheese;
import org.drools.compiler.Cheesery;
import org.drools.compiler.CommonTestMethodBase;
import org.drools.compiler.Person;
import org.drools.compiler.Worker;
import org.drools.core.util.debug.SessionInspector;
import org.drools.core.util.debug.SessionReporter;
import org.drools.core.util.debug.StatefulKnowledgeSessionInfo;
import org.junit.Ignore;
import org.junit.Test;
import org.kie.internal.KnowledgeBase;
import org.kie.internal.KnowledgeBaseFactory;
import org.kie.internal.builder.KnowledgeBuilder;
import org.kie.internal.builder.KnowledgeBuilderFactory;
import org.kie.internal.io.ResourceFactory;
import org.kie.internal.runtime.StatefulKnowledgeSession;
import org.kie.api.io.ResourceType;
public class SessionInspectorTest extends CommonTestMethodBase {
@Test
public void testGetSessionInfo() {
KnowledgeBuilder kbuilder = KnowledgeBuilderFactory.newKnowledgeBuilder();
kbuilder.add( ResourceFactory.newClassPathResource( "org/drools/compiler/integrationtests/test_SubNetworks.drl" ),
ResourceType.DRL );
kbuilder.add( ResourceFactory.newClassPathResource( "org/drools/compiler/integrationtests/test_AccumulateWithFromChaining.drl" ),
ResourceType.DRL );
kbuilder.add( ResourceFactory.newClassPathResource( "org/drools/compiler/integrationtests/test_CollectResultsBetaConstraint.drl" ),
ResourceType.DRL );
kbuilder.add( ResourceFactory.newClassPathResource( "org/drools/compiler/integrationtests/test_QueryMemoryLeak.drl" ),
ResourceType.DRL );
assertFalse( kbuilder.getErrors().toString(),
kbuilder.hasErrors() );
KnowledgeBase kbase = KnowledgeBaseFactory.newKnowledgeBase();
kbase.addKnowledgePackages( kbuilder.getKnowledgePackages() );
StatefulKnowledgeSession ksession = createKnowledgeSession(kbase);
ksession.setGlobal( "results", new ArrayList<Object>() );
ksession.insert( new Dimension( 100, 50 ) );
ksession.insert( new Dimension( 130, 80 ) );
ksession.insert( new Dimension( 50, 40 ) );
ksession.insert( new Dimension( 50, 40 ) );
Cheesery cheesery = new Cheesery();
cheesery.addCheese( new Cheese( "brie", 10 ));
cheesery.addCheese( new Cheese( "brie", 10 ));
cheesery.addCheese( new Cheese( "brie", 10 ));
cheesery.addCheese( new Cheese( "brie", 10 ));
cheesery.addCheese( new Cheese( "muzzarella", 10 ));
cheesery.addCheese( new Cheese( "muzzarella", 10 ));
cheesery.addCheese( new Cheese( "muzzarella", 10 ));
cheesery.addCheese( new Cheese( "muzzarella", 10 ));
cheesery.addCheese( new Cheese( "muzzarella", 10 ));
cheesery.addCheese( new Cheese( "muzzarella", 10 ));
cheesery.addCheese( new Cheese( "muzzarella", 10 ));
cheesery.addCheese( new Cheese( "stilton", 10 ));
cheesery.addCheese( new Cheese( "stilton", 10 ));
cheesery.addCheese( new Cheese( "stilton", 10 ));
cheesery.addCheese( new Cheese( "stilton", 10 ));
cheesery.addCheese( new Cheese( "stilton", 10 ));
cheesery.addCheese( new Cheese( "stilton", 10 ));
cheesery.addCheese( new Cheese( "stilton", 10 ));
ksession.insert( cheesery );
ksession.insert( new Person( "Bob", "muzzarella") );
ksession.insert( new Person( "Mark", "brie") );
ksession.insert( new Cheese( "brie", 10 ) );
ksession.insert( new Cheese( "brie", 10 ) );
ksession.insert( new Cheese( "brie", 10 ) );
ksession.insert( new Cheese( "brie", 10 ) );
ksession.insert( new Cheese( "muzzarella", 10 ) );
ksession.insert( new Cheese( "muzzarella", 10 ) );
ksession.insert( new Cheese( "muzzarella", 10 ) );
ksession.insert( new Cheese( "muzzarella", 10 ) );
ksession.insert( new Cheese( "Stilton", 10 ) );
ksession.insert( new Cheese( "Stilton", 10 ) );
ksession.insert( new Cheese( "Stilton", 10 ) );
ksession.insert( new Double( 10 ) );
ksession.insert( new Double( 11 ) );
ksession.insert( new Double( 12 ) );
ksession.insert( new Double( 13 ) );
ksession.insert( new Double( 14 ) );
ksession.insert( new Integer( 15 ) );
ksession.insert( new Integer( 16 ) );
ksession.insert( new Integer( 17 ) );
ksession.insert( new Integer( 18 ) );
org.kie.api.runtime.rule.FactHandle handle = ksession.insert( new Worker( ) );
ksession.retract( handle );
SessionInspector inspector = new SessionInspector( ksession );
StatefulKnowledgeSessionInfo info = inspector.getSessionInfo();
String report = SessionReporter.generateReport( "simple", info, null );
assertNotNull( report );
}
@Test
public void testGetSessionInfoWithCustomTemplate() {
if ( System.getProperty("java.vendor").toUpperCase().contains("IBM") ) {
return; //Does not work in the IBM JDK due to a bug in MVEL
}
KnowledgeBuilder kbuilder = KnowledgeBuilderFactory.newKnowledgeBuilder();
kbuilder.add( ResourceFactory.newClassPathResource( "org/drools/compiler/integrationtests/test_SubNetworks.drl" ),
ResourceType.DRL );
kbuilder.add( ResourceFactory.newClassPathResource( "org/drools/compiler/integrationtests/test_AccumulateWithFromChaining.drl" ),
ResourceType.DRL );
kbuilder.add( ResourceFactory.newClassPathResource( "org/drools/compiler/integrationtests/test_CollectResultsBetaConstraint.drl" ),
ResourceType.DRL );
kbuilder.add( ResourceFactory.newClassPathResource("org/drools/compiler/integrationtests/test_QueryMemoryLeak.drl"),
ResourceType.DRL );
assertFalse( kbuilder.getErrors().toString(),
kbuilder.hasErrors() );
KnowledgeBase kbase = KnowledgeBaseFactory.newKnowledgeBase();
kbase.addKnowledgePackages( kbuilder.getKnowledgePackages() );
StatefulKnowledgeSession ksession = createKnowledgeSession(kbase);
ksession.setGlobal( "results", new ArrayList<Object>() );
ksession.insert( new Dimension( 100, 50 ) );
ksession.insert( new Dimension( 130, 80 ) );
ksession.insert( new Dimension( 50, 40 ) );
ksession.insert( new Dimension( 50, 40 ) );
Cheesery cheesery = new Cheesery();
cheesery.addCheese( new Cheese( "brie", 10 ));
cheesery.addCheese( new Cheese( "brie", 10 ));
cheesery.addCheese( new Cheese( "brie", 10 ));
cheesery.addCheese( new Cheese( "brie", 10 ));
cheesery.addCheese( new Cheese( "muzzarella", 10 ));
cheesery.addCheese( new Cheese( "muzzarella", 10 ));
cheesery.addCheese( new Cheese( "muzzarella", 10 ));
cheesery.addCheese( new Cheese( "muzzarella", 10 ));
cheesery.addCheese( new Cheese( "muzzarella", 10 ));
cheesery.addCheese( new Cheese( "muzzarella", 10 ));
cheesery.addCheese( new Cheese( "muzzarella", 10 ));
cheesery.addCheese( new Cheese( "stilton", 10 ));
cheesery.addCheese( new Cheese( "stilton", 10 ));
cheesery.addCheese( new Cheese( "stilton", 10 ));
cheesery.addCheese( new Cheese( "stilton", 10 ));
cheesery.addCheese( new Cheese( "stilton", 10 ));
cheesery.addCheese( new Cheese( "stilton", 10 ));
cheesery.addCheese( new Cheese( "stilton", 10 ));
ksession.insert( cheesery );
ksession.insert( new Person( "Bob", "muzzarella") );
ksession.insert( new Person( "Mark", "brie") );
ksession.insert( new Cheese( "brie", 10 ) );
ksession.insert( new Cheese( "brie", 10 ) );
ksession.insert( new Cheese( "brie", 10 ) );
ksession.insert( new Cheese( "brie", 10 ) );
ksession.insert( new Cheese( "muzzarella", 10 ) );
ksession.insert( new Cheese( "muzzarella", 10 ) );
ksession.insert( new Cheese( "muzzarella", 10 ) );
ksession.insert( new Cheese( "muzzarella", 10 ) );
ksession.insert( new Cheese( "Stilton", 10 ) );
ksession.insert( new Cheese( "Stilton", 10 ) );
ksession.insert( new Cheese( "Stilton", 10 ) );
ksession.insert( new Double( 10 ) );
ksession.insert( new Double( 11 ) );
ksession.insert( new Double( 12 ) );
ksession.insert( new Double( 13 ) );
ksession.insert( new Double( 14 ) );
ksession.insert( new Integer( 15 ) );
ksession.insert( new Integer( 16 ) );
ksession.insert( new Integer( 17 ) );
ksession.insert( new Integer( 18 ) );
org.kie.api.runtime.rule.FactHandle handle = ksession.insert( new Worker( ) );
ksession.retract( handle );
SessionInspector inspector = new SessionInspector( ksession );
StatefulKnowledgeSessionInfo info = inspector.getSessionInfo();
SessionReporter.addNamedTemplate( "topten", getClass().getResourceAsStream( "customreports.mvel" ) );
String report = SessionReporter.generateReport( "topten", info, null );
assertNotNull( report );
}
}