/*
* Copyright 2015 Red Hat, Inc. and/or its affiliates.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
*
* 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.compiler.phreak;
import org.drools.core.common.InternalFactHandle;
import org.drools.core.common.InternalWorkingMemory;
import org.drools.core.common.PhreakPropagationContext;
import org.drools.core.reteoo.BetaMemory;
import org.drools.core.reteoo.BetaNode;
import org.drools.core.reteoo.LeftTuple;
import org.drools.core.reteoo.SegmentMemory;
import org.drools.core.reteoo.TupleMemory;
import java.util.ArrayList;
import java.util.List;
public class LeftMemory {
private Scenario scenario;
public LeftMemory(Scenario scenario, Object... objects) {
this.scenario = scenario;
scenario.getLeftMemory().addAll( getLeftTuples( objects ) );
}
public List<LeftTuple> getLeftTuples(Object... objects) {
BetaNode node = scenario.getBetaNode();
BetaMemory bm = scenario.getBm();
TupleMemory ltm = bm.getLeftTupleMemory();
InternalWorkingMemory wm = scenario.getWorkingMemory();
if ( objects == null ) {
objects = new Object[0];
}
List<LeftTuple> list = new ArrayList<LeftTuple>();
for ( Object object : objects ) {
InternalFactHandle fh = (InternalFactHandle) wm.insert( object );
LeftTuple expectedLeftTuple = node.createLeftTuple( fh, node, true );
expectedLeftTuple.setPropagationContext( new PhreakPropagationContext() );
list.add( expectedLeftTuple );
}
scenario.setTestLeftMemory( true );
return list;
}
public RightMemory right(Object... objects) {
return new RightMemory( scenario, objects );
}
public StagedBuilder preStaged(SegmentMemory sm) {
StagedBuilder stagedBuilder = new StagedBuilder( scenario, sm );
scenario.addPreStagedBuilder( stagedBuilder );
return stagedBuilder;
}
public StagedBuilder postStaged(SegmentMemory sm) {
StagedBuilder stagedBuilder = new StagedBuilder( scenario, sm );
scenario.addPostStagedBuilder( stagedBuilder );
return stagedBuilder;
}
public Scenario run() {
return this.scenario.run();
}
}