/*
* Copyright 2011 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.factmodel.traits;
import java.io.*;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.Comparator;
import java.util.Date;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import org.drools.KnowledgeBase;
import org.drools.KnowledgeBaseFactory;
import org.drools.Person;
import org.drools.builder.KnowledgeBuilder;
import org.drools.builder.KnowledgeBuilderFactory;
import org.drools.builder.ResourceType;
import org.drools.command.Command;
import org.drools.command.CommandFactory;
import org.drools.common.DroolsObjectInputStream;
import org.drools.definition.type.FactType;
import org.drools.event.rule.AfterActivationFiredEvent;
import org.drools.event.rule.AgendaEventListener;
import org.drools.io.Resource;
import org.drools.io.ResourceFactory;
import org.drools.io.impl.ByteArrayResource;
import org.drools.runtime.StatefulKnowledgeSession;
import org.drools.runtime.StatelessKnowledgeSession;
import org.junit.AfterClass;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Ignore;
import org.junit.Test;
import org.mockito.ArgumentCaptor;
import java.util.*;
import static org.mockito.Mockito.*;
import static org.hamcrest.CoreMatchers.is;
import static org.junit.Assert.*;
public class TraitTest {
private static long t0;
@BeforeClass
public static void init() {
t0 = new Date().getTime();
}
@AfterClass
public static void finish() {
System.out.println( "TIME : " + ( new Date().getTime() - t0 ) );
}
@Before
public void reset() {
TraitRegistry.reset();
TraitFactory.reset();
}
private StatefulKnowledgeSession getSession( String... ruleFiles ) {
KnowledgeBuilder knowledgeBuilder = KnowledgeBuilderFactory.newKnowledgeBuilder();
for (String file : ruleFiles) {
knowledgeBuilder.add (ResourceFactory.newClassPathResource(file), ResourceType.DRL);
}
if (knowledgeBuilder.hasErrors()) {
throw new RuntimeException(knowledgeBuilder.getErrors().toString());
}
KnowledgeBase kbase = KnowledgeBaseFactory.newKnowledgeBase();
kbase.addKnowledgePackages(knowledgeBuilder.getKnowledgePackages());
StatefulKnowledgeSession session = kbase.newStatefulKnowledgeSession();
return session;
}
private StatefulKnowledgeSession getSessionFromString( String drl ) {
KnowledgeBuilder knowledgeBuilder = KnowledgeBuilderFactory.newKnowledgeBuilder();
knowledgeBuilder.add( ResourceFactory.newByteArrayResource( drl.getBytes() ),
ResourceType.DRL );
if (knowledgeBuilder.hasErrors()) {
throw new RuntimeException( knowledgeBuilder.getErrors().toString() );
}
KnowledgeBase kbase = KnowledgeBaseFactory.newKnowledgeBase();
kbase.addKnowledgePackages( knowledgeBuilder.getKnowledgePackages() );
StatefulKnowledgeSession session = kbase.newStatefulKnowledgeSession();
return session;
}
@Test
public void testTraitWrapper_GetAndSet() {
String source = "org/drools/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() );
}
KnowledgeBase kb = KnowledgeBaseFactory.newKnowledgeBase();
kb.addKnowledgePackages( kbuilder.getKnowledgePackages() );
try {
FactType impClass = kb.getFactType( "org.test",
"Imp" );
TraitableBean imp = (TraitableBean) impClass.newInstance();
Class trait = kb.getFactType( "org.test",
"Student" ).getFactClass();
TraitProxy proxy = (TraitProxy) new TraitFactory( kb ).getProxy( imp,
trait );
Map<String, Object> virtualFields = imp.getDynamicProperties();
Map<String, Object> wrapper = proxy.getFields();
wrapper.put( "name",
"john" );
wrapper.put( "virtualField",
"xyz" );
wrapper.entrySet();
assertEquals( 4,
wrapper.size() );
assertEquals( 2,
virtualFields.size() );
assertEquals( "john",
wrapper.get( "name" ) );
assertEquals( "xyz",
wrapper.get( "virtualField" ) );
assertEquals( "john",
impClass.get( imp,
"name" ) );
} catch (Exception e) {
e.printStackTrace();
fail( e.getMessage() );
}
}
@Test
public void testTraitShed() {
String source = "org/drools/factmodel/traits/testTraitShed.drl";
StatefulKnowledgeSession ks = getSession( source );
List info = new ArrayList();
ks.setGlobal( "list",
info );
assertTrue( info.isEmpty() );
ks.fireAllRules();
assertTrue( info.contains( "Student" ) );
assertEquals( 1,
info.size() );
ks.insert( "hire" );
ks.fireAllRules();
Collection c = ks.getObjects();
assertTrue( info.contains( "Worker" ) );
assertEquals( 2,
info.size() );
ks.insert( "check" );
ks.fireAllRules();
assertEquals( 4,
info.size() );
assertTrue( info.contains( "Conflict" ) );
assertTrue( info.contains( "Nothing" ) );
}
@Test
public void testTraitDon() {
String source = "org/drools/factmodel/traits/testTraitDon.drl";
StatefulKnowledgeSession ks = getSession( source );
List info = new ArrayList();
ks.setGlobal( "list",
info );
ks.fireAllRules();
Collection<Object> wm = ks.getObjects();
ks.insert( "die" );
ks.fireAllRules();
assertTrue( info.contains( "DON" ) );
assertTrue( info.contains( "SHED" ) );
}
@Test
public void testMixin() {
String source = "org/drools/factmodel/traits/testTraitMixin.drl";
StatefulKnowledgeSession ks = getSession( source );
List info = new ArrayList();
ks.setGlobal( "list",
info );
ks.fireAllRules();
assertTrue( info.contains( "27" ) );
}
@Test
public void testTraitMethodsWithObjects() {
String source = "org/drools/factmodel/traits/testTraitWrapping.drl";
StatefulKnowledgeSession ks = getSession( source );
List errors = new ArrayList();
ks.setGlobal( "list",
errors );
ks.fireAllRules();
if (!errors.isEmpty()) {
System.err.println( errors.toString() );
}
assertTrue( errors.isEmpty() );
}
@Test
public void testTraitMethodsWithPrimitives() {
String source = "org/drools/factmodel/traits/testTraitWrappingPrimitives.drl";
StatefulKnowledgeSession ks = getSession( source );
List errors = new ArrayList();
ks.setGlobal( "list",
errors );
ks.fireAllRules();
if (!errors.isEmpty()) {
System.err.println( errors );
}
assertTrue( errors.isEmpty() );
}
@Test
public void testTraitProxy() {
String source = "org/drools/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() );
}
KnowledgeBase kb = KnowledgeBaseFactory.newKnowledgeBase();
kb.addKnowledgePackages( kbuilder.getKnowledgePackages() );
TraitFactory traitBuilder = new TraitFactory( kb );
try {
FactType impClass = kb.getFactType( "org.test",
"Imp" );
TraitableBean imp = (TraitableBean) impClass.newInstance();
impClass.set( imp,
"name",
"aaa" );
Class trait = kb.getFactType( "org.test",
"Student" ).getFactClass();
Class trait2 = kb.getFactType( "org.test",
"Role" ).getFactClass();
assertNotNull( trait );
TraitProxy proxy = (TraitProxy) traitBuilder.getProxy( imp,
trait );
proxy.getFields().put( "field",
"xyz" );
// proxy.getFields().put("name", "aaa");
assertNotNull( proxy );
TraitProxy proxy2 = (TraitProxy) traitBuilder.getProxy( imp,
trait );
assertSame( proxy2,
proxy );
TraitProxy proxy3 = (TraitProxy) traitBuilder.getProxy( imp,
trait2 );
assertNotNull( proxy3 );
assertEquals( "xyz",
proxy3.getFields().get( "field" ) );
assertEquals( "aaa",
proxy3.getFields().get( "name" ) );
TraitableBean imp2 = (TraitableBean) impClass.newInstance();
impClass.set( imp2,
"name",
"aaa" );
TraitProxy proxy4 = (TraitProxy) traitBuilder.getProxy( imp2,
trait );
// proxy4.getFields().put("name", "aaa");
proxy4.getFields().put( "field",
"xyz" );
assertEquals( proxy2,
proxy4 );
} catch (InstantiationException e) {
e.printStackTrace();
fail( e.getMessage() );
} catch (IllegalAccessException e) {
e.printStackTrace();
fail( e.getMessage() );
}
}
@Test
public void testTraitWrapper_Size() {
String source = "org/drools/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() );
}
KnowledgeBase kb = KnowledgeBaseFactory.newKnowledgeBase();
kb.addKnowledgePackages( kbuilder.getKnowledgePackages() );
TraitFactory traitBuilder = new TraitFactory( kb );
try {
FactType impClass = kb.getFactType( "org.test",
"Imp" );
TraitableBean imp = (TraitableBean) impClass.newInstance();
FactType traitClass = kb.getFactType( "org.test",
"Student" );
Class trait = traitClass.getFactClass();
TraitProxy proxy = (TraitProxy) traitBuilder.getProxy( imp,
trait );
Map<String, Object> virtualFields = imp.getDynamicProperties();
Map<String, Object> wrapper = proxy.getFields();
assertEquals( 3,
wrapper.size() );
assertEquals( 1,
virtualFields.size() );
impClass.set( imp,
"name",
"john" );
assertEquals( 3,
wrapper.size() );
assertEquals( 1,
virtualFields.size() );
proxy.getFields().put( "school",
"skol" );
assertEquals( 3,
wrapper.size() );
assertEquals( 1,
virtualFields.size() );
proxy.getFields().put( "surname",
"xxx" );
assertEquals( 4,
wrapper.size() );
assertEquals( 2,
virtualFields.size() );
// FactType indClass = kb.getFactType("org.test","Entity");
// TraitableBean ind = (TraitableBean) indClass.newInstance();
TraitableBean ind = new Entity();
TraitProxy proxy2 = (TraitProxy) traitBuilder.getProxy( ind,
trait );
Map virtualFields2 = ind.getDynamicProperties();
Map wrapper2 = proxy2.getFields();
assertEquals( 3,
wrapper2.size() );
assertEquals( 3,
virtualFields2.size() );
traitClass.set( proxy2,
"name",
"john" );
assertEquals( 3,
wrapper2.size() );
assertEquals( 3,
virtualFields2.size() );
proxy2.getFields().put( "school",
"skol" );
assertEquals( 3,
wrapper2.size() );
assertEquals( 3,
virtualFields2.size() );
proxy2.getFields().put( "surname",
"xxx" );
assertEquals( 4,
wrapper2.size() );
assertEquals( 4,
virtualFields2.size() );
FactType traitClass2 = kb.getFactType( "org.test",
"Role" );
Class trait2 = traitClass2.getFactClass();
// TraitableBean ind2 = (TraitableBean) indClass.newInstance();
TraitableBean ind2 = new Entity();
TraitProxy proxy99 = (TraitProxy) traitBuilder.getProxy( ind2,
trait2 );
proxy99.getFields().put( "surname",
"xxx" );
proxy99.getFields().put( "name",
"xyz" );
proxy99.getFields().put( "school",
"skol" );
assertEquals( 3,
proxy99.getFields().size() );
TraitProxy proxy100 = (TraitProxy) traitBuilder.getProxy( ind2,
trait );
assertEquals( 4,
proxy100.getFields().size() );
} catch (Exception e) {
e.printStackTrace();
fail( e.getMessage() );
}
}
@Test
public void testTraitWrapper_IsEmpty() {
String source = "org/drools/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() );
}
KnowledgeBase kb = KnowledgeBaseFactory.newKnowledgeBase();
kb.addKnowledgePackages( kbuilder.getKnowledgePackages() );
TraitFactory traitBuilder = new TraitFactory( kb );
try {
FactType impClass = kb.getFactType( "org.test",
"Imp" );
TraitableBean imp = (TraitableBean) impClass.newInstance();
FactType studentClass = kb.getFactType( "org.test",
"Student" );
Class trait = studentClass.getFactClass();
TraitProxy proxy = (TraitProxy) traitBuilder.getProxy( imp,
trait );
Map<String, Object> virtualFields = imp.getDynamicProperties();
Map<String, Object> wrapper = proxy.getFields();
assertFalse( wrapper.isEmpty() );
studentClass.set( proxy,
"name",
"john" );
assertFalse( wrapper.isEmpty() );
studentClass.set( proxy,
"name",
null );
assertFalse( wrapper.isEmpty() );
studentClass.set( proxy,
"age",
32 );
assertFalse( wrapper.isEmpty() );
studentClass.set( proxy,
"age",
null );
assertFalse( wrapper.isEmpty() );
// FactType indClass = kb.getFactType("org.test","Entity");
TraitableBean ind = new Entity();
FactType RoleClass = kb.getFactType( "org.test",
"Role" );
Class trait2 = RoleClass.getFactClass();
TraitProxy proxy2 = (TraitProxy) traitBuilder.getProxy( ind,
trait2 );
Map<String, Object> wrapper2 = proxy2.getFields();
assertTrue( wrapper2.isEmpty() );
proxy2.getFields().put( "name",
"john" );
assertFalse( wrapper2.isEmpty() );
proxy2.getFields().put( "name",
null );
assertFalse( wrapper2.isEmpty() );
} catch (Exception e) {
e.printStackTrace();
fail( e.getMessage() );
}
}
@Test
public void testTraitWrapper_ContainsKey() {
String source = "org/drools/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() );
}
KnowledgeBase kb = KnowledgeBaseFactory.newKnowledgeBase();
kb.addKnowledgePackages( kbuilder.getKnowledgePackages() );
TraitFactory traitBuilder = new TraitFactory( kb );
try {
FactType impClass = kb.getFactType( "org.test",
"Imp" );
TraitableBean imp = (TraitableBean) impClass.newInstance();
impClass.set( imp,
"name",
"john" );
FactType traitClass = kb.getFactType( "org.test",
"Student" );
Class trait = traitClass.getFactClass();
TraitProxy proxy = (TraitProxy) traitBuilder.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.test","Entity");
TraitableBean ind = new Entity();
TraitProxy proxy2 = (TraitProxy) traitBuilder.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.test",
"Role" );
Class trait2 = traitClass2.getFactClass();
TraitableBean ind2 = new Entity();
TraitProxy proxy99 = (TraitProxy) traitBuilder.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();
TraitProxy proxy100 = (TraitProxy) traitBuilder.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" ) );
TraitProxy proxy101 = (TraitProxy) traitBuilder.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() );
}
}
@Test
public void testTraitWrapper_keySetAndValues() {
String source = "org/drools/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() );
}
KnowledgeBase kb = KnowledgeBaseFactory.newKnowledgeBase();
kb.addKnowledgePackages( kbuilder.getKnowledgePackages() );
TraitFactory traitBuilder = new TraitFactory( kb );
try {
FactType impClass = kb.getFactType( "org.test",
"Imp" );
TraitableBean imp = (TraitableBean) impClass.newInstance();
FactType traitClass = kb.getFactType( "org.test",
"Student" );
Class trait = traitClass.getFactClass();
TraitProxy proxy = (TraitProxy) traitBuilder.getProxy( imp,
trait );
impClass.set( imp,
"name",
"john" );
proxy.getFields().put( "surname",
"xxx" );
proxy.getFields().put( "name2",
"john" );
proxy.getFields().put( "nfield",
null );
Set set = new HashSet();
set.add( "name" );
set.add( "surname" );
set.add( "age" );
set.add( "school" );
set.add( "name2" );
set.add( "nfield" );
assertEquals( 6,
proxy.getFields().keySet().size() );
assertEquals( set,
proxy.getFields().keySet() );
Collection col1 = proxy.getFields().values();
Collection col2 = Arrays.asList( "john",
null,
0,
"xxx",
"john",
null );
Comparator comp = new Comparator() {
public int compare( Object o1, Object o2 ) {
if (o1 == null && o2 != null) {
return 1;
}
if (o1 != null && o2 == null) {
return -1;
}
if (o1 == null && o2 == null) {
return 0;
}
return o1.toString().compareTo( o2.toString() );
}
};
Collections.sort( (List) col1,
comp );
Collections.sort( (List) col2,
comp );
assertEquals( col1,
col2 );
assertTrue( proxy.getFields().containsValue( null ) );
assertTrue( proxy.getFields().containsValue( "john" ) );
assertTrue( proxy.getFields().containsValue( 0 ) );
assertTrue( proxy.getFields().containsValue( "xxx" ) );
assertFalse( proxy.getFields().containsValue( "randomString" ) );
assertFalse( proxy.getFields().containsValue( -96 ) );
} catch (Exception e) {
e.printStackTrace();
fail( e.getMessage() );
}
}
@Test
public void testTraitWrapper_ClearAndRemove() {
String source = "org/drools/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() );
}
KnowledgeBase kb = KnowledgeBaseFactory.newKnowledgeBase();
kb.addKnowledgePackages( kbuilder.getKnowledgePackages() );
TraitFactory traitBuilder = new TraitFactory( kb );
try {
FactType impClass = kb.getFactType( "org.test",
"Imp" );
TraitableBean imp = (TraitableBean) impClass.newInstance();
impClass.set( imp,
"name",
"john" );
FactType traitClass = kb.getFactType( "org.test",
"Student" );
Class trait = traitClass.getFactClass();
TraitProxy proxy = (TraitProxy) traitBuilder.getProxy( imp,
trait );
proxy.getFields().put( "surname",
"xxx" );
proxy.getFields().put( "name2",
"john" );
proxy.getFields().put( "nfield",
null );
Set set = new HashSet();
set.add( "name" );
set.add( "surname" );
set.add( "age" );
set.add( "school" );
set.add( "name2" );
set.add( "nfield" );
assertEquals( 6,
proxy.getFields().keySet().size() );
assertEquals( set,
proxy.getFields().keySet() );
proxy.getFields().clear();
Map<String, Object> fields = proxy.getFields();
assertEquals( 3,
fields.size() );
assertTrue( fields.containsKey( "age" ) );
assertTrue( fields.containsKey( "school" ) );
assertTrue( fields.containsKey( "name" ) );
assertEquals( 0,
fields.get( "age" ) );
assertNull( fields.get( "school" ) );
assertNotNull( fields.get( "name" ) );
proxy.getFields().put( "surname",
"xxx" );
proxy.getFields().put( "name2",
"john" );
proxy.getFields().put( "nfield",
null );
proxy.getFields().put( "age",
24 );
assertEquals( "john",
proxy.getFields().get( "name" ) );
assertEquals( "xxx",
proxy.getFields().get( "surname" ) );
assertEquals( "john",
proxy.getFields().get( "name2" ) );
assertEquals( null,
proxy.getFields().get( "nfield" ) );
assertEquals( 24,
proxy.getFields().get( "age" ) );
assertEquals( null,
proxy.getFields().get( "school" ) );
proxy.getFields().remove( "surname" );
proxy.getFields().remove( "name2" );
proxy.getFields().remove( "age" );
proxy.getFields().remove( "school" );
proxy.getFields().remove( "nfield" );
assertEquals( 3,
proxy.getFields().size() );
assertEquals( 0,
proxy.getFields().get( "age" ) );
assertEquals( null,
proxy.getFields().get( "school" ) );
assertEquals( "john",
proxy.getFields().get( "name" ) );
assertEquals( null,
proxy.getFields().get( "nfield" ) );
assertFalse( proxy.getFields().containsKey( "nfield" ) );
assertEquals( null,
proxy.getFields().get( "name2" ) );
assertFalse( proxy.getFields().containsKey( "name2" ) );
assertEquals( null,
proxy.getFields().get( "surname" ) );
assertFalse( proxy.getFields().containsKey( "surname" ) );
} catch (Exception e) {
e.printStackTrace();
fail( e.getMessage() );
}
}
@Test
public void testIsA() {
String source = "org/drools/factmodel/traits/testTraitIsA.drl";
StatefulKnowledgeSession ks = getSession( source );
List info = new ArrayList();
ks.setGlobal( "list",
info );
ks.fireAllRules();
int num = 10;
System.out.println( info );
assertEquals( num,
info.size() );
for (int j = 0; j < num; j++) {
assertTrue( info.contains( "" + j ) );
}
}
@Test
public void testTraitOverrideType() {
String source = "org/drools/factmodel/traits/testTraitOverride.drl";
StatefulKnowledgeSession ks = getSession( source );
List info = new ArrayList();
ks.setGlobal( "list",
info );
ks.fireAllRules();
Collection wm = ks.getObjects();
assertTrue( info.contains( "OK" ) );
}
@Test
public void testTraitLegacy() {
String source = "org/drools/factmodel/traits/testTraitLegacyTrait.drl";
StatefulKnowledgeSession ks = getSession( source );
List info = new ArrayList();
ks.setGlobal( "list",
info );
ks.fireAllRules();
System.err.println( " -------------- " + ks.getObjects().size() + " ---------------- " );
for (Object o : ks.getObjects()) {
System.err.println( "\t\t" + o );
}
System.err.println( " -------------- ---------------- " );
System.err.println( info );
System.err.println( " -------------- ---------------- " );
assertEquals( 5,
info.size() );
assertTrue( info.contains( "OK" ) );
assertTrue( info.contains( "OK2" ) );
assertTrue( info.contains( "OK3" ) );
assertTrue( info.contains( "OK4" ) );
assertTrue( info.contains( "OK5" ) );
}
@Test
public void testTraitCollections() {
String source = "org/drools/factmodel/traits/testTraitCollections.drl";
StatefulKnowledgeSession ks = getSession( source );
List info = new ArrayList();
ks.setGlobal( "list",
info );
ks.fireAllRules();
System.err.println( " -------------- " + ks.getObjects().size() + " ---------------- " );
for (Object o : ks.getObjects()) {
System.err.println( "\t\t" + o );
}
System.err.println( " -------------- ---------------- " );
System.err.println( info );
System.err.println( " -------------- ---------------- " );
assertEquals( 1,
info.size() );
assertTrue( info.contains( "OK" ) );
}
@Test
public void testTraitCore() {
String source = "org/drools/factmodel/traits/testTraitLegacyCore.drl";
StatefulKnowledgeSession ks = getSession( source );
List info = new ArrayList();
ks.setGlobal( "list",
info );
ks.fireAllRules();
System.err.println( " -------------- " + ks.getObjects().size() + " ---------------- " );
for (Object o : ks.getObjects()) {
System.err.println( "\t\t" + o );
}
System.err.println( " -------------- ---------------- " );
System.err.println( info );
System.err.println( " -------------- ---------------- " );
assertTrue( info.contains( "OK" ) );
assertTrue( info.contains( "OK2" ) );
assertEquals( 2,
info.size() );
}
@Test
public void testIsAOperator() {
String source = "org/drools/factmodel/traits/testTraitIsA2.drl";
StatefulKnowledgeSession ksession = getSession( source );
AgendaEventListener ael = mock( AgendaEventListener.class );
ksession.addEventListener( ael );
Person student = new Person( "student", 18 );
ksession.insert( student );
ksession.fireAllRules();
ArgumentCaptor<AfterActivationFiredEvent> cap = ArgumentCaptor.forClass( AfterActivationFiredEvent.class );
verify( ael,
times( 3 ) ).afterActivationFired( cap.capture() );
List<AfterActivationFiredEvent> values = cap.getAllValues();
assertThat( values.get( 0 ).getActivation().getRule().getName(),
is( "create student" ) );
assertThat( values.get( 1 ).getActivation().getRule().getName(),
is( "print student" ) );
assertThat( values.get( 2 ).getActivation().getRule().getName(),
is( "print school" ) );
}
@Test
public void testTraitDeclared() {
List<Integer> trueTraits = new ArrayList<Integer>();
List<Integer> untrueTraits = new ArrayList<Integer>();
StatefulKnowledgeSession session = getSession("org/drools/factmodel/traits/testDeclaredFactTrait.drl");
session.setGlobal("trueTraits", trueTraits);
session.setGlobal("untrueTraits", untrueTraits);
session.fireAllRules();
session.dispose();
assertTrue(trueTraits.contains(1));
assertFalse(trueTraits.contains(2));
assertTrue(untrueTraits.contains(2));
assertFalse(untrueTraits.contains(1));
}
@Test
public void testTraitPojo() {
List<Integer> trueTraits = new ArrayList<Integer>();
List<Integer> untrueTraits = new ArrayList<Integer>();
StatefulKnowledgeSession session = getSession("org/drools/factmodel/traits/testPojoFactTrait.drl");
session.setGlobal("trueTraits", trueTraits);
session.setGlobal("untrueTraits", untrueTraits);
session.fireAllRules();
session.dispose();
assertTrue(trueTraits.contains(1));
assertFalse(trueTraits.contains(2));
assertTrue(untrueTraits.contains(2));
assertFalse(untrueTraits.contains(1));
}
@Test
public void testManyTraits() {
String source = "" +
"import org.drools.Message;" +
"" +
"global java.util.List list; \n" +
"" +
"declare Message\n" +
" @Traitable\n" +
" end\n" +
"\n" +
" declare trait NiceMessage\n" +
" message : String\n" +
" end\n" +
"" +
"rule \"Nice\"\n" +
"when\n" +
" $n : NiceMessage( $m : message )\n" +
"then\n" +
" System.out.println( $m );\n" +
"end" +
"\n" +
" rule load\n" +
" when\n" +
"\n" +
" then\n" +
" Message message = new Message();\n" +
" message.setMessage(\"Hello World\");\n" +
" insert(message);\n" +
" don( message, NiceMessage.class );\n" +
"\n" +
" Message unreadMessage = new Message();\n" +
" unreadMessage.setMessage(\"unread\");\n" +
" insert(unreadMessage);\n" +
" don( unreadMessage, NiceMessage.class );\n" +
"\n" +
" Message oldMessage = new Message();\n" +
" oldMessage.setMessage(\"old\");\n" +
" insert(oldMessage);\n" +
" don( oldMessage, NiceMessage.class );" +
" list.add(\"OK\");\n" +
" end";
StatefulKnowledgeSession ksession = getSessionFromString( source );
TraitFactory.setMode( TraitFactory.VirtualPropertyMode.MAP );
List list = new ArrayList();
ksession.setGlobal( "list", list );
Person student = new Person( "student", 18 );
ksession.insert( student );
ksession.fireAllRules();
assertEquals( 1, list.size() );
assertTrue( list.contains( "OK" ) );
}
@Test
public void testTraitManyTimes() {
StatefulKnowledgeSession ksession = getSession( "org/drools/factmodel/traits/testTraitDonMultiple.drl" );
List list = new ArrayList();
ksession.setGlobal( "list", list );
ksession.fireAllRules();
assertEquals( 1, list.size() );
assertEquals( 0, list.get( 0 ) );
assertFalse( list.contains( 1 ) );
}
@Test
@Ignore
// BZ #748752
public void testTraitsInBatchExecution() {
String str = "package org.jboss.qa.brms.traits\n" +
"import org.drools.Person;\n" +
"import org.drools.factmodel.traits.Traitable;\n" +
"" +
"global java.util.List list;" +
"" +
"declare Person \n" +
" @Traitable \n" +
"end \n" +
"" +
"declare trait Student\n" +
" school : String\n" +
"end\n" +
"\n" +
"rule \"create student\" \n" +
" when\n" +
" $student : Person( age < 26 )\n" +
" then\n" +
" Student s = don( $student, Student.class );\n" +
" s.setSchool(\"Masaryk University\");\n" +
"end\n" +
"\n" +
"rule \"print student\"\n" +
" when\n" +
" student : Person( this isA Student )\n" +
" then" +
" list.add( 1 );\n" +
" System.out.println(\"Person is a student: \" + student);\n" +
"end\n" +
"\n" +
"rule \"print school\"\n" +
" when\n" +
" Student( $school : school )\n" +
" then\n" +
" list.add( 2 );\n" +
" System.out.println(\"Student is attending \" + $school);\n" +
"end";
KnowledgeBuilder kbuilder = KnowledgeBuilderFactory.newKnowledgeBuilder();
kbuilder.add( new ByteArrayResource( str.getBytes() ), ResourceType.DRL );
if (kbuilder.hasErrors()) {
throw new RuntimeException(kbuilder.getErrors().toString());
}
List list = new ArrayList();
KnowledgeBase kbase = kbuilder.newKnowledgeBase();
StatelessKnowledgeSession ksession = kbase.newStatelessKnowledgeSession();
ksession.setGlobal( "list", list );
List<Command<?>> commands = new ArrayList<Command<?>>();
Person student = new Person("student", 18);
commands.add(CommandFactory.newInsert(student));
System.out.println("Starting execution...");
commands.add(CommandFactory.newFireAllRules());
ksession.execute(CommandFactory.newBatchExecution(commands));
System.out.println("Finished...");
assertEquals( 2, list.size() );
assertTrue( list.contains( 1 ) );
assertTrue( list.contains( 2 ) );
}
}