DroolsConfigurationTest.java

package org.opennms.netmgt.correlation.drools;


import java.util.Arrays;
import java.util.Collection;


import org.junit.runners.Parameterized.Parameters;
import org.opennms.core.test.xml.XmlTest;
import org.opennms.netmgt.correlation.drools.config.EngineConfiguration;
import org.opennms.netmgt.correlation.drools.config.Global;
import org.opennms.netmgt.correlation.drools.config.RuleSet;

import bsh.ParseException;

public class DroolsConfigurationTest extends XmlTest<EngineConfiguration> {

	public DroolsConfigurationTest(String schemaFile, EngineConfiguration sampleObject, String sampleXml) {
		super(sampleObject, sampleXml, schemaFile);
	}
	
	static class ConfigBuilder {
		EngineConfiguration m_engineConfiguration = new EngineConfiguration();
		RuleSet m_currentRuleSet;
		
		public EngineConfiguration get() {
			return m_engineConfiguration;
		}

		public ConfigBuilder addRuleSet(String name) {
			m_currentRuleSet = new RuleSet();
			m_currentRuleSet.setName(name);
			
			m_engineConfiguration.addRuleSet(m_currentRuleSet);
			return this;
		}
		
		public ConfigBuilder addRuleFile(String path) {
			m_currentRuleSet.addRuleFile(path);
			return this;
		}
		
		public ConfigBuilder addEvent(String uei) {
			m_currentRuleSet.addEvent(uei);
			return this;
		}
		
		public ConfigBuilder setAppContext(String path) {
			m_currentRuleSet.setAppContext(path);
			return this;
		}
		
		public <T> ConfigBuilder addGlobalValue(String name, Class<T> type, T val) {
			Global global = new Global();
			global.setName(name);
			global.setType(type.getName());
			global.setValue(val.toString());
			
			m_currentRuleSet.addGlobal(global);
			
			return this;
		}
		
		public ConfigBuilder addGlobalRef(String name, Class<?> type, String ref) {
			Global global = new Global();
			global.setName(name);
			global.setType(type.getName());
			global.setRef(ref);
			
			m_currentRuleSet.addGlobal(global);
			
			return this;
		}
		
	}
	
	@Parameters
	public static Collection<Object[]> data() throws ParseException {
		   		return Arrays.asList(new Object[][] {
		   				{   "target/classes/xsds/drools-engine.xsd",
		   					new ConfigBuilder().get(), 
		   					"<engine-configuration/>"
		   				},
		   				{	"target/classes/xsds/drools-engine.xsd", 
		   					new ConfigBuilder()
		   					.addRuleSet("locationMonitorRules")
		   						.addRuleFile("file:src/test/opennms-home/etc/LocationMonitorRules.drl")
		   					 	.addEvent("uei.opennms.org/remote/nodes/nodeLostService")
		   					 	.addEvent("uei.opennms.org/remote/nodes/nodeRegainedService")
		   					 	.addGlobalValue("WIDE_SPREAD_THRESHOLD", Integer.class, 3)
		   					 	.addGlobalValue("FLAP_INTERVAL", Long.class, 1000L)
		   					 	.addGlobalValue("FLAP_COUNT", Integer.class, 3)
		   					.addRuleSet("nodeParentRules")
		   					    .addRuleFile("file:src/test/opennms-home/etc/NodeParentRules.drl")
		   					    .addEvent("uei.opennms.org/nodes/nodeDown")
		   					    .addEvent("uei.opennms.org/nodes/nodeUp")
		   					    .setAppContext("file:src/test/opennms-home/etc/nodeParentRules-context.xml")
		   					    .addGlobalValue("POLL_INTERVAL", Long.class, 3000L)
		   					    .addGlobalRef("nodeService", NodeService.class, "nodeService")
		   					.get(), 
		   					"<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n" + 
		   					"<engine-configuration>\n" + 
		   					"  <rule-set name=\"locationMonitorRules\">\n" + 
		   					"    <rule-file>file:src/test/opennms-home/etc/LocationMonitorRules.drl</rule-file>\n" + 
		   					"    <event>uei.opennms.org/remote/nodes/nodeLostService</event>\n" + 
		   					"    <event>uei.opennms.org/remote/nodes/nodeRegainedService</event>\n" + 
		   					"    <global name=\"WIDE_SPREAD_THRESHOLD\" type=\"java.lang.Integer\" value=\"3\"/>\n" + 
		   					"    <global name=\"FLAP_INTERVAL\" type=\"java.lang.Long\" value=\"1000\" />\n" + 
		   					"    <global name=\"FLAP_COUNT\" type=\"java.lang.Integer\" value=\"3\" />\n" + 
		   					"  </rule-set>\n" + 
		   					"  <rule-set name=\"nodeParentRules\">\n" + 
		   					"    <rule-file>file:src/test/opennms-home/etc/NodeParentRules.drl</rule-file>\n" + 
		   					"    <event>uei.opennms.org/nodes/nodeDown</event>\n" + 
		   					"    <event>uei.opennms.org/nodes/nodeUp</event>\n" + 
		   					"    <app-context>file:src/test/opennms-home/etc/nodeParentRules-context.xml</app-context>\n" + 
		   					"    <global name=\"POLL_INTERVAL\" type=\"java.lang.Long\" value=\"3000\" />\n" + 
		   					"    <global name=\"nodeService\" type=\"org.opennms.netmgt.correlation.drools.NodeService\" ref=\"nodeService\" />\n" + 
		   					"  </rule-set>\n" + 
		   					"</engine-configuration>\n" + 
		   					"" 
		   				} 
		   		});

	}


}