/*
* Copyright 2006 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.
*/
package org.drools.rule.builder.dialect.java;
import java.io.InputStreamReader;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.HashSet;
import java.util.List;
import junit.framework.Assert;
import junit.framework.TestCase;
import org.drools.base.ClassTypeResolver;
import org.drools.base.TypeResolver;
import org.drools.compiler.Dialect;
import org.drools.compiler.DialectRegistry;
import org.drools.compiler.DrlParser;
import org.drools.compiler.PackageBuilder;
import org.drools.compiler.PackageBuilderConfiguration;
import org.drools.lang.descr.AttributeDescr;
import org.drools.lang.descr.PackageDescr;
import org.drools.lang.descr.RuleDescr;
import org.drools.rule.GroupElement;
import org.drools.rule.Package;
import org.drools.rule.Rule;
import org.drools.rule.builder.RuleBuildContext;
import org.drools.rule.builder.RuleBuilder;
/**
* @author etirelli
*
*/
public class RuleBuilderTest extends TestCase {
/* (non-Javadoc)
* @see junit.framework.TestCase#setUp()
*/
protected void setUp() throws Exception {
super.setUp();
}
/* (non-Javadoc)
* @see junit.framework.TestCase#tearDown()
*/
protected void tearDown() throws Exception {
super.tearDown();
}
/**
* Test method for {@link org.drools.rule.builder.RuleBuilder#build(org.drools.rule.Package, org.drools.lang.descr.RuleDescr)}.
*/
public void testBuild() throws Exception {
final DrlParser parser = new DrlParser();
final PackageDescr pkgDescr = parser.parse( new InputStreamReader( getClass().getResourceAsStream( "nestedConditionalElements.drl" ) ) );
// just checking there is no parsing errors
Assert.assertFalse( parser.getErrors().toString(),
parser.hasErrors() );
final Package pkg = new Package( "org.drools" );
final RuleDescr ruleDescr = (RuleDescr) pkgDescr.getRules().get( 0 );
final String ruleClassName = "RuleClassName.java";
ruleDescr.setClassName( ruleClassName );
ruleDescr.addAttribute( new AttributeDescr("dialect", "java") );
final TypeResolver typeResolver = new ClassTypeResolver( new HashSet(),
this.getClass().getClassLoader() );
// make an automatic import for the current package
typeResolver.addImport( pkgDescr.getName() + ".*" );
typeResolver.addImport( "java.lang.*" );
final RuleBuilder builder = new RuleBuilder( );
final PackageBuilder pkgBuilder = new PackageBuilder(pkg);
final PackageBuilderConfiguration conf = pkgBuilder.getPackageBuilderConfiguration();
Dialect dialect = pkgBuilder.getDialectRegistry().getDialect( "java" );
RuleBuildContext context = new RuleBuildContext(conf, pkg, ruleDescr, pkgBuilder.getDialectRegistry(), dialect);
builder.build( context );
Assert.assertTrue( context.getErrors().toString(),
context.getErrors().isEmpty() );
final Rule rule = context.getRule();
assertEquals( "There should be 2 rule level declarations",
2,
rule.getDeclarations().length );
// second GE should be a not
final GroupElement not = (GroupElement) rule.getLhs().getChildren().get( 1 );
assertTrue( not.isNot() );
// not has no outer declarations
assertTrue( not.getOuterDeclarations().isEmpty() );
assertEquals( 1,
not.getInnerDeclarations().size() );
assertTrue( not.getInnerDeclarations().keySet().contains( "$state" ) );
// second not
final GroupElement not2 = (GroupElement) ((GroupElement) not.getChildren().get( 0 )).getChildren().get( 1 );
assertTrue( not2.isNot() );
// not has no outer declarations
assertTrue( not2.getOuterDeclarations().isEmpty() );
assertEquals( 1,
not2.getInnerDeclarations().size() );
assertTrue( not2.getInnerDeclarations().keySet().contains( "$likes" ) );
}
public void testBuildAttributes() throws Exception {
Rule rule = new Rule( "my rule" );
List attributes = new ArrayList();
attributes.add( new AttributeDescr("dialect", "java") );
attributes.add( new AttributeDescr( "no-loop",
"true" ) );
attributes.add( new AttributeDescr( "enabled",
"false" ) );
attributes.add( new AttributeDescr( "ruleflow-group",
"mygroup" ) );
RuleBuildContext.setAttributes( rule, null, attributes );
assertTrue( rule.isNoLoop() );
assertFalse( rule.isEffective() );
assertEquals( "mygroup",
rule.getRuleFlowGroup() );
attributes = new ArrayList();
attributes.add( new AttributeDescr( "date-effective",
"10-Jul-1974" ) );
attributes.add( new AttributeDescr( "date-expires",
"10-Jul-2040" ) );
rule = new Rule( "myrule" );
RuleBuildContext.setAttributes( rule, null, attributes );
final Field eff = rule.getClass().getDeclaredField( "dateEffective" );
eff.setAccessible( true );
final Calendar effectiveDate = (Calendar) eff.get( rule );
assertNotNull( effectiveDate );
assertEquals( 1974,
effectiveDate.get( Calendar.YEAR ) );
final Field exp = rule.getClass().getDeclaredField( "dateExpires" );
exp.setAccessible( true );
final Calendar expiryDate = (Calendar) exp.get( rule );
assertEquals( 2040,
expiryDate.get( Calendar.YEAR ) );
assertNotNull( expiryDate );
}
}