use of org.drools.core.reteoo.LeftInputAdapterNode in project drools by kiegroup.
the class RemoveRuleTest method testPopulatedSingleRuleNoSharing.
@Test
public void testPopulatedSingleRuleNoSharing() throws Exception {
InternalKnowledgeBase kbase = KnowledgeBaseFactory.newKnowledgeBase();
InternalWorkingMemory wm = ((InternalWorkingMemory) kbase.newKieSession());
wm.insert(new A(1));
wm.insert(new B(1));
wm.insert(new C(1));
wm.insert(new C(2));
wm.insert(new D(1));
wm.insert(new E(1));
wm.fireAllRules();
kbase.addPackages(buildKnowledgePackage("r1", " A() B() C(object == 2) D() E()\n"));
List list = new ArrayList();
wm.setGlobal("list", list);
ObjectTypeNode aotn = getObjectTypeNode(kbase, A.class);
LeftInputAdapterNode liaNode = (LeftInputAdapterNode) aotn.getObjectSinkPropagator().getSinks()[0];
LiaNodeMemory lm = (LiaNodeMemory) wm.getNodeMemory(liaNode);
SegmentMemory sm = lm.getSegmentMemory();
assertNotNull(sm.getStagedLeftTuples().getInsertFirst());
wm.fireAllRules();
BetaMemory bMem = (BetaMemory) sm.getNodeMemories().get(1);
assertEquals(1, bMem.getLeftTupleMemory().size());
assertEquals(1, bMem.getRightTupleMemory().size());
BetaMemory eMem = (BetaMemory) sm.getNodeMemories().get(4);
assertEquals(1, eMem.getLeftTupleMemory().size());
assertEquals(1, eMem.getRightTupleMemory().size());
NodeMemories nms = wm.getNodeMemories();
assertEquals(12, countNodeMemories(nms));
assertNull(sm.getStagedLeftTuples().getInsertFirst());
assertEquals(1, list.size());
assertEquals("r1", ((Match) list.get(0)).getRule().getName());
kbase.removeRule("org.kie", "r1");
// still has OTN
assertEquals(6, countNodeMemories(nms));
}
use of org.drools.core.reteoo.LeftInputAdapterNode in project drools by kiegroup.
the class RemoveRuleTest method testPopulatedSharedLiaNode.
@Test
public void testPopulatedSharedLiaNode() throws Exception {
InternalKnowledgeBase kbase1 = buildKnowledgeBase("r1", " A() B(1;) C() D() E()\n");
InternalWorkingMemory wm = ((InternalWorkingMemory) kbase1.newKieSession());
List list = new ArrayList();
wm.setGlobal("list", list);
wm.insert(new A(1));
wm.insert(new A(2));
wm.insert(new A(3));
wm.insert(new B(1));
wm.insert(new B(2));
wm.insert(new C(1));
wm.insert(new D(1));
wm.insert(new E(1));
wm.fireAllRules();
assertEquals(3, list.size());
assertEquals(7, countNodeMemories(wm.getNodeMemories()));
kbase1.addPackages(buildKnowledgePackage("r2", " a : A() B(2;) C() D() E()\n"));
wm.fireAllRules();
assertEquals(17, countNodeMemories(wm.getNodeMemories()));
ObjectTypeNode aotn = getObjectTypeNode(kbase1, A.class);
LeftInputAdapterNode liaNode = (LeftInputAdapterNode) aotn.getObjectSinkPropagator().getSinks()[0];
JoinNode b1Node = (JoinNode) liaNode.getSinkPropagator().getFirstLeftTupleSink();
JoinNode b2Node = (JoinNode) liaNode.getSinkPropagator().getLastLeftTupleSink();
JoinNode c1Node = (JoinNode) b1Node.getSinkPropagator().getLastLeftTupleSink();
LiaNodeMemory lm = (LiaNodeMemory) wm.getNodeMemory(liaNode);
SegmentMemory sm = lm.getSegmentMemory();
BetaMemory b1Mem = (BetaMemory) wm.getNodeMemory(b1Node);
assertSame(sm.getFirst(), b1Mem.getSegmentMemory());
assertEquals(3, b1Mem.getLeftTupleMemory().size());
assertEquals(1, b1Mem.getRightTupleMemory().size());
BetaMemory b2Mem = (BetaMemory) wm.getNodeMemory(b2Node);
SegmentMemory b2Smem = sm.getFirst().getNext();
assertSame(b2Smem, b2Mem.getSegmentMemory());
assertEquals(3, b2Mem.getLeftTupleMemory().size());
assertEquals(1, b2Mem.getRightTupleMemory().size());
assertEquals(6, list.size());
BetaMemory c1Mem = (BetaMemory) wm.getNodeMemory(c1Node);
assertSame(b1Mem.getSegmentMemory(), c1Mem.getSegmentMemory());
assertNotSame(b1Mem.getSegmentMemory(), b2Mem.getSegmentMemory());
wm.fireAllRules();
assertEquals(6, list.size());
assertEquals(17, countNodeMemories(wm.getNodeMemories()));
kbase1.removeRule("org.kie", "r2");
assertEquals(12, countNodeMemories(wm.getNodeMemories()));
assertSame(sm, b1Mem.getSegmentMemory());
assertSame(sm, c1Mem.getSegmentMemory());
assertNull(sm.getFirst());
assertEquals(3, b1Mem.getLeftTupleMemory().size());
assertEquals(1, b1Mem.getRightTupleMemory().size());
// SegmentMemory b2Smem = sm.getFirst().remove();
assertSame(b2Smem, b2Mem.getSegmentMemory());
wm.insert(new A(1));
wm.fireAllRules();
assertEquals("r1", ((Match) list.get(6)).getRule().getName());
// only one more added, as second rule as removed
assertEquals(7, list.size());
}
Aggregations