Examples with Entity org.drools.traits.core.factmodel.Entity used on opensource projects

Example 6 with Entity

use of org.drools.traits.core.factmodel.Entity in project drools by kiegroup.

the class TraitTest method testPartitionWithSiblingsOnDelete.

@Test
public void testPartitionWithSiblingsOnDelete() {
    String drl = "import " + Entity.class.getName() + ";" + "global java.util.List list; " + "declare trait A @propertyReactive end " + "declare trait B extends A @propertyReactive end " + "declare trait C extends A @propertyReactive end " + "rule Trait when " + "    $core: Entity( ) " + "then " + "    don( $core, A.class ); " + "    don( $core, B.class ); " + "    don( $core, C.class ); " + "end " + "rule Shed when " + "   $s: String() " + "   $core : Entity() " + "then " + "   shed( $core, C.class ); " + "end " + "rule RA when A() then list.add( 'A' ); end " + "rule RB when B() then list.add( 'B' ); end " + "rule RC when C() then list.add( 'C' ); end " + " ";
    KieBase kbase = new KieHelper().addContent(drl, ResourceType.DRL).build();
    TraitFactoryImpl.setMode(mode, kbase);
    KieSession ksession = kbase.newKieSession();
    List list = new ArrayList();
    ksession.setGlobal("list", list);
    Entity e = new Entity();
    ksession.insert(e);
    ksession.fireAllRules();
    assertEquals(Arrays.asList('A', 'B', 'C'), list);
    ksession.insert("go");
    ksession.fireAllRules();
    Set<BitSet> s = checkOTNPartitioning(e, ksession);
    assertEquals(2, s.size());
    assertEquals(Arrays.asList('A', 'B', 'C'), list);
}

Example 7 with Entity

use of org.drools.traits.core.factmodel.Entity in project drools by kiegroup.

the class TraitTest method testWrapperContainsKey.

@Test
public void testWrapperContainsKey() {
    String source = "org/drools/compiler/factmodel/traits/testTraitDon.drl";
    KnowledgeBuilder kbuilder = KnowledgeBuilderFactory.newKnowledgeBuilder();
    Resource res = ResourceFactory.newClassPathResource(source);
    assertNotNull(res);
    kbuilder.add(res, ResourceType.DRL);
    if (kbuilder.hasErrors()) {
        fail(kbuilder.getErrors().toString());
    }
    InternalKnowledgeBase kb = KnowledgeBaseFactory.newKnowledgeBase();
    kb.addPackages(kbuilder.getKnowledgePackages());
    TraitFactoryImpl.setMode(mode, kb);
    TraitFactoryImpl tFactory = (TraitFactoryImpl) RuntimeComponentFactory.get().getTraitFactory(kb);
    try {
        FactType impClass = kb.getFactType("org.drools.compiler.trait.test", "Imp");
        TraitableBean imp = (TraitableBean) impClass.newInstance();
        impClass.set(imp, "name", "john");
        FactType traitClass = kb.getFactType("org.drools.compiler.trait.test", "Student");
        Class trait = traitClass.getFactClass();
        TraitProxyImpl proxy = (TraitProxyImpl) tFactory.getProxy(imp, trait);
        Map<String, Object> virtualFields = imp._getDynamicProperties();
        Map<String, Object> wrapper = proxy.getFields();
        assertTrue(wrapper.containsKey("name"));
        assertTrue(wrapper.containsKey("school"));
        assertTrue(wrapper.containsKey("age"));
        assertFalse(wrapper.containsKey("surname"));
        proxy.getFields().put("school", "skol");
        proxy.getFields().put("surname", "xxx");
        assertTrue(wrapper.containsKey("surname"));
        // FactType indClass = kb.getFactType("org.drools.compiler.trait.test","Entity");
        TraitableBean ind = new Entity();
        TraitProxyImpl proxy2 = (TraitProxyImpl) tFactory.getProxy(ind, trait);
        Map virtualFields2 = ind._getDynamicProperties();
        Map wrapper2 = proxy2.getFields();
        assertTrue(wrapper2.containsKey("name"));
        assertTrue(wrapper2.containsKey("school"));
        assertTrue(wrapper2.containsKey("age"));
        assertFalse(wrapper2.containsKey("surname"));
        traitClass.set(proxy2, "name", "john");
        proxy2.getFields().put("school", "skol");
        proxy2.getFields().put("surname", "xxx");
        assertTrue(wrapper2.containsKey("surname"));
        FactType traitClass2 = kb.getFactType("org.drools.compiler.trait.test", "Role");
        Class trait2 = traitClass2.getFactClass();
        TraitableBean ind2 = new Entity();
        TraitProxyImpl proxy99 = (TraitProxyImpl) tFactory.getProxy(ind2, trait2);
        Map<String, Object> wrapper99 = proxy99.getFields();
        assertFalse(wrapper99.containsKey("name"));
        assertFalse(wrapper99.containsKey("school"));
        assertFalse(wrapper99.containsKey("age"));
        assertFalse(wrapper99.containsKey("surname"));
        proxy99.getFields().put("surname", "xxx");
        proxy99.getFields().put("name", "xyz");
        proxy99.getFields().put("school", "skol");
        assertTrue(wrapper99.containsKey("name"));
        assertTrue(wrapper99.containsKey("school"));
        assertFalse(wrapper99.containsKey("age"));
        assertTrue(wrapper99.containsKey("surname"));
        assertEquals(3, proxy99.getFields().size());
        TraitableBean ind0 = new Entity();
        TraitProxyImpl proxy100 = (TraitProxyImpl) tFactory.getProxy(ind0, trait2);
        Map<String, Object> wrapper100 = proxy100.getFields();
        assertFalse(wrapper100.containsKey("name"));
        assertFalse(wrapper100.containsKey("school"));
        assertFalse(wrapper100.containsKey("age"));
        assertFalse(wrapper100.containsKey("surname"));
        TraitProxyImpl proxy101 = (TraitProxyImpl) tFactory.getProxy(ind0, trait);
        // object gains properties by virtue of another trait
        // so new props are accessible even using the old proxy
        assertTrue(wrapper100.containsKey("name"));
        assertTrue(wrapper100.containsKey("school"));
        assertTrue(wrapper100.containsKey("age"));
        assertFalse(wrapper100.containsKey("surname"));
    } catch (Exception e) {
        e.printStackTrace();
        fail(e.getMessage());
    }
}

Example 8 with Entity

use of org.drools.traits.core.factmodel.Entity in project drools by kiegroup.

the class TraitTest method testTupleIntegrityOnModification.

@Test
public void testTupleIntegrityOnModification() {
    String drl = "package test " + "import " + Entity.class.getName() + ";" + "global java.util.List list; " + "declare trait A @propertyReactive value : int end " + "rule Trait when " + "    $core: Entity( ) " + "then " + "    A o = don( $core, A.class ); " + "    System.out.println( 'Found ! ' + o ); " + "end " + "rule Test when " + "   $x: A( value == 0 ) " + "then " + "   list.add( 0 ); " + "end " + "rule Check when " + "   $x: A( value == 42 ) " + "then " + "   list.add( 42 ); " + "end " + "rule Mood when " + "  $x : A( value != 42 ) " + "then " + "  modify ( $x ) { setValue( 42 ); } " + "end ";
    KieBase kbase = getKieBaseFromString(drl);
    TraitFactoryImpl.setMode(mode, kbase);
    KieSession ksession = kbase.newKieSession();
    List list = new ArrayList();
    ksession.setGlobal("list", list);
    ksession.insert(new Entity());
    ksession.fireAllRules();
    for (final Object o : ksession.getObjects(new ObjectFilter() {

        @Override
        public boolean accept(Object object) {
            return object.getClass().getName().contains("test.A");
        }
    })) {
        InternalFactHandle handle = (InternalFactHandle) ksession.getFactHandle(o);
        LeftTuple first = handle.getFirstLeftTuple();
        assertTrue(first instanceof RuleTerminalNodeLeftTuple);
        assertEquals("Check", ((RuleTerminalNodeLeftTuple) first).getRule().getName());
    }
    assertEquals(Arrays.asList(0, 42), list);
}

Example 9 with Entity

use of org.drools.traits.core.factmodel.Entity in project drools by kiegroup.

the class TraitTest method testAlphaNodeSharing.

@Test
public void testAlphaNodeSharing() {
    String drl = "package test; " + "import " + Entity.class.getName() + " " + "declare trait Person\n" + "    name : String\n" + "end\n" + "rule Init " + "when " + "then " + "    don( new Entity(), Person.class ); " + "end\n" + "rule One when" + "    $core: Entity( this isA Person ) " + "then " + "end " + "rule Two when" + "    $core: Entity( this isA Person ) " + "then " + "end " + "\n";
    final KieBase kbase = getKieBaseFromString(drl);
    TraitFactoryImpl.setMode(mode, kbase);
    KieSession kSession = kbase.newKieSession();
    assertEquals(3, kSession.fireAllRules());
    NamedEntryPoint nep = ((NamedEntryPoint) kSession.getEntryPoint(EntryPointId.DEFAULT.getEntryPointId()));
    ObjectTypeNode otn = nep.getEntryPointNode().getObjectTypeNodes().get(new ClassObjectType(Entity.class));
    assertNotNull(otn);
    assertEquals(1, otn.getObjectSinkPropagator().getSinks().length);
}

Example 10 with Entity

use of org.drools.traits.core.factmodel.Entity in project drools by kiegroup.

the class MetadataTest method testDonWithAttributes.

@Test
public void testDonWithAttributes() {
    Entity entity = new Entity("123");
    entity._setDynamicProperties(new HashMap());
    SubKlass klass = SubKlass_.donSubKlass(entity).setTraitFactory(createStandaloneTraitFactory()).prop("hello").subProp(32).call();
    assertEquals("hello", klass.getProp());
    assertEquals(32, (int) klass.getSubProp());
}