package org.drools.compiler.compiler.xml.changeset;
import java.io.File;
import java.io.IOException;
import java.io.StringReader;
import java.net.URL;
import java.net.URLClassLoader;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import org.drools.compiler.CommonTestMethodBase;
import org.drools.compiler.compiler.PackageBuilderConfiguration;
import org.drools.core.io.impl.UrlResource;
import org.drools.core.xml.XmlChangeSetReader;
import org.junit.Test;
import org.kie.internal.ChangeSet;
import org.kie.internal.KnowledgeBase;
import org.kie.internal.KnowledgeBaseFactory;
import org.kie.internal.builder.DecisionTableConfiguration;
import org.kie.internal.builder.DecisionTableInputType;
import org.kie.internal.builder.KnowledgeBuilder;
import org.kie.internal.builder.KnowledgeBuilderConfiguration;
import org.kie.internal.builder.KnowledgeBuilderFactory;
import org.kie.internal.io.ResourceFactory;
import org.kie.api.io.Resource;
import org.kie.api.io.ResourceType;
import org.kie.internal.runtime.StatefulKnowledgeSession;
import org.xml.sax.SAXException;
public class ChangeSetTest extends CommonTestMethodBase {
@Test
public void testXmlParser() throws SAXException,
IOException {
PackageBuilderConfiguration conf = new PackageBuilderConfiguration();
XmlChangeSetReader xmlReader = new XmlChangeSetReader( conf.getSemanticModules() );
xmlReader.setClassLoader( ChangeSetTest.class.getClassLoader(), ChangeSetTest.class );
String str = "";
str += "<change-set ";
str += "xmlns='http://drools.org/drools-5.0/change-set' ";
str += "xmlns:xs='http://www.w3.org/2001/XMLSchema-instance' ";
str += "xs:schemaLocation='http://drools.org/drools-5.0/change-set http://anonsvn.jboss.org/repos/labs/labs/jbossrules/trunk/drools-api/src/main/resources/change-set-1.0.0.xsd' >";
str += " <add> ";
str += " <resource source='http://www.domain.com/test.drl' type='DRL' />";
str += " <resource source='http://www.domain.com/test.xls' type='DTABLE' >";
str += " <decisiontable-conf worksheet-name='sheet10' input-type='XLS' />";
str += " </resource>";
str += " </add> ";
str += "</change-set>";
StringReader reader = new StringReader( str );
ChangeSet changeSet = xmlReader.read( reader );
assertEquals( 2,
changeSet.getResourcesAdded().size() );
UrlResource resource = ( UrlResource ) ((List)changeSet.getResourcesAdded()).get( 0 );
assertEquals( "http://www.domain.com/test.drl",
resource.getURL().toString() );
assertEquals( ResourceType.DRL,
resource.getResourceType() );
resource = ( UrlResource ) ((List)changeSet.getResourcesAdded()).get( 1 );
assertEquals( "http://www.domain.com/test.xls",
resource.getURL().toString() );
assertEquals( ResourceType.DTABLE,
resource.getResourceType() );
DecisionTableConfiguration dtConf = (DecisionTableConfiguration) resource.getConfiguration();
assertEquals( DecisionTableInputType.XLS,
dtConf.getInputType() );
}
@Test
public void testIntegregation() {
KnowledgeBuilder kbuilder = KnowledgeBuilderFactory.newKnowledgeBuilder();
kbuilder.add( ResourceFactory.newClassPathResource("changeset1Test.xml",
getClass()),
ResourceType.CHANGE_SET );
assertFalse( kbuilder.hasErrors() );
KnowledgeBase kbase = KnowledgeBaseFactory.newKnowledgeBase();
kbase.addKnowledgePackages( kbuilder.getKnowledgePackages() );
StatefulKnowledgeSession ksession = createKnowledgeSession(kbase);
List list = new ArrayList();
ksession.setGlobal( "list",
list );
ksession.fireAllRules();
ksession.dispose();
assertEquals( 2,
list.size() );
assertTrue( list.containsAll( Arrays.asList( new String[]{"rule1", "rule2"} ) ) );
}
@Test
public void testBasicAuthentication() throws SAXException,
IOException {
PackageBuilderConfiguration conf = new PackageBuilderConfiguration();
XmlChangeSetReader xmlReader = new XmlChangeSetReader( conf.getSemanticModules() );
xmlReader.setClassLoader( ChangeSetTest.class.getClassLoader(), ChangeSetTest.class );
String str = "";
str += "<change-set ";
str += "xmlns='http://drools.org/drools-5.0/change-set' ";
str += "xmlns:xs='http://www.w3.org/2001/XMLSchema-instance' ";
str += "xs:schemaLocation='http://drools.org/drools-5.0/change-set http://anonsvn.jboss.org/repos/labs/labs/jbossrules/trunk/drools-api/src/main/resources/change-set-1.0.0.xsd' >";
str += " <add> ";
str += " <resource source='http://localhost:8081/jboss-brms/org.kie.guvnor.Guvnor/package/defaultPackage/LATEST' type='PKG' basicAuthentication='enabled' username='admin' password='pwd'/>";
str += " </add> ";
str += "</change-set>";
StringReader reader = new StringReader( str );
ChangeSet changeSet = xmlReader.read( reader );
assertEquals( 1,
changeSet.getResourcesAdded().size() );
UrlResource resource = ( UrlResource ) ((List)changeSet.getResourcesAdded()).get( 0 );
assertEquals( "http://localhost:8081/jboss-brms/org.kie.guvnor.Guvnor/package/defaultPackage/LATEST",
resource.getURL().toString() );
assertEquals( "enabled", resource.getBasicAuthentication() );
assertEquals( "admin", resource.getUsername() );
assertEquals( "pwd", resource.getPassword() );
assertEquals( ResourceType.PKG,
resource.getResourceType() );
}
@Test(timeout = 10000)
public void testCustomClassLoader() throws Exception {
// JBRULES-3630
String absolutePath = new File("file").getAbsolutePath();
URL url = ChangeSetTest.class.getResource(ChangeSetTest.class.getSimpleName() + ".class");
File file = new File( url.toURI() );
File jar = null;
while ( true ) {
file = file.getParentFile();
jar = new File( file, "/src/test/resources/org/drools/compiler/compiler/xml/changeset/changeset.jar" );
if ( jar.exists() ) {
break;
}
}
ClassLoader classLoader = URLClassLoader.newInstance(new URL[]{jar.toURI().toURL()}, getClass().getClassLoader());
Resource changeSet = ResourceFactory.newClassPathResource("changeset1.xml", classLoader);
KnowledgeBuilderConfiguration conf = KnowledgeBuilderFactory.newKnowledgeBuilderConfiguration(null, classLoader);
KnowledgeBuilder builder = KnowledgeBuilderFactory.newKnowledgeBuilder(conf);
builder.add(changeSet, ResourceType.CHANGE_SET);
}
}