/*
* Copyright 2010 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.reteoo.test.dsl;
import java.beans.IntrospectionException;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import org.drools.common.SingleBetaConstraints;
import org.drools.reteoo.ExistsNode;
import org.drools.reteoo.LeftTupleSource;
import org.drools.reteoo.MockObjectSource;
import org.drools.reteoo.MockTupleSource;
import org.drools.reteoo.ObjectSource;
import org.drools.reteoo.builder.BuildContext;
import org.drools.rule.BehaviorManager;
import org.drools.rule.Declaration;
import org.drools.rule.Pattern;
import org.drools.spi.BetaNodeFieldConstraint;
import org.mockito.Mockito;
public class ExistsNodeStep
implements
Step {
private ReteTesterHelper reteTesterHelper;
public ExistsNodeStep(ReteTesterHelper reteTesterHelper) {
this.reteTesterHelper = reteTesterHelper;
}
public void execute(Map<String, Object> context,
List<String[]> args) {
BuildContext buildContext = (BuildContext) context.get( "BuildContext" );
if ( args.size() != 0 ) {
String[] a = args.get( 0 );
String name = a[0];
String leftInput = a[1];
String rightInput = a[2];
LeftTupleSource leftTupleSource;
if ( "mock".equals( leftInput ) ) {
leftTupleSource = Mockito.mock( LeftTupleSource.class );
} else {
leftTupleSource = (LeftTupleSource) context.get( leftInput );
}
ObjectSource rightObjectSource;
if ( "mock".equals( rightInput ) ) {
rightObjectSource = Mockito.mock( ObjectSource.class );
} else {
rightObjectSource = (ObjectSource) context.get( rightInput );
}
a = args.get( 1 );
String type = a[0].trim();
String fieldName = a[1].trim();
String operator = a[2].trim();
String var = a[3].trim();
Pattern rightSidePattern = null;
try {
rightSidePattern = reteTesterHelper.getPattern( 1,
type );
} catch ( Exception e ) {
throw new IllegalArgumentException( "Not possible to process arguments: "+Arrays.toString( a ));
}
Declaration declr = (Declaration) context.get( var );
BetaNodeFieldConstraint betaConstraint;
try {
betaConstraint = this.reteTesterHelper.getBoundVariableConstraint( rightSidePattern,
fieldName,
declr,
operator );
} catch ( IntrospectionException e ) {
throw new IllegalArgumentException();
}
SingleBetaConstraints constraints = new SingleBetaConstraints( betaConstraint,
buildContext.getRuleBase().getConfiguration() );
ExistsNode existsNode = new ExistsNode( buildContext.getNextId(),
leftTupleSource,
rightObjectSource,
constraints,
BehaviorManager.NO_BEHAVIORS,
buildContext );
existsNode.attach();
context.put( name,
existsNode );
} else {
throw new IllegalArgumentException( "Cannot arguments " + args );
}
}
}