/*
* 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.ide.common.client.modeldriven;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.jar.JarInputStream;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import static org.junit.Assert.*;
import org.drools.ide.common.client.modeldriven.DropDownData;
import org.drools.ide.common.client.modeldriven.FactTypeFilter;
import org.drools.ide.common.client.modeldriven.ModelField;
import org.drools.ide.common.client.modeldriven.SuggestionCompletionEngine;
import org.drools.ide.common.client.modeldriven.ModelField.FIELD_CLASS_TYPE;
import org.drools.ide.common.client.modeldriven.brl.ActionFieldValue;
import org.drools.ide.common.client.modeldriven.brl.FactPattern;
import org.drools.ide.common.client.modeldriven.brl.SingleFieldConstraint;
import org.drools.ide.common.server.rules.SuggestionCompletionLoader;
import org.drools.lang.dsl.DSLTokenizedMappingFile;
public class SuggestionCompletionEngineTest {
@Test
public void testNestedImports() {
String pkg = "package org.test\n import org.drools.ide.common.client.modeldriven.SuggestionCompletionEngineTest.NestedClass";
SuggestionCompletionLoader loader = new SuggestionCompletionLoader();
SuggestionCompletionEngine engine = loader.getSuggestionEngine( pkg,
new ArrayList<JarInputStream>(),
new ArrayList<DSLTokenizedMappingFile>() );
assertEquals( "String",
engine.getFieldType( "SuggestionCompletionEngineTest$NestedClass",
"name" ) );
}
@Test
public void testStringNonNumeric() {
String pkg = "package org.test\n import org.drools.ide.common.client.modeldriven.Alert";
SuggestionCompletionLoader loader = new SuggestionCompletionLoader();
SuggestionCompletionEngine engine = loader.getSuggestionEngine( pkg,
new ArrayList<JarInputStream>(),
new ArrayList<DSLTokenizedMappingFile>() );
assertEquals( SuggestionCompletionEngine.TYPE_STRING,
engine.getFieldType( "Alert",
"message" ) );
}
@Test
public void testDataEnums() {
String pkg = "package org.test\n import org.drools.ide.common.client.modeldriven.SuggestionCompletionEngineTest.NestedClass";
SuggestionCompletionLoader loader = new SuggestionCompletionLoader();
List<String> enums = new ArrayList<String>();
enums.add( "'Person.age' : [42, 43] \n 'Person.sex' : ['M', 'F']" );
enums.add( "'Driver.sex' : ['M', 'F']" );
SuggestionCompletionEngine engine = loader.getSuggestionEngine( pkg,
new ArrayList<JarInputStream>(),
new ArrayList<DSLTokenizedMappingFile>(),
enums );
assertEquals( "String",
engine.getFieldType( "SuggestionCompletionEngineTest$NestedClass",
"name" ) );
assertEquals( 3,
engine.getDataEnumListsSize());
String[] items = engine.getDataEnumList( "Person.age" );
assertEquals( 2,
items.length );
assertEquals( "42",
items[0] );
assertEquals( "43",
items[1] );
items = engine.getEnums( new FactPattern( "Person" ),
"age" ).fixedList;
assertEquals( 2,
items.length );
assertEquals( "42",
items[0] );
assertEquals( "43",
items[1] );
assertNull( engine.getEnums( new FactPattern( "Nothing" ),
"age" ) );
assertEquals( null,
engine.getEnums( new FactPattern( "Something" ),
"else" ) );
}
@Test
public void testDataEnums3() {
String pkg = "package org.test\n import org.drools.ide.common.client.modeldriven.SuggestionCompletionEngineTest.NestedClass";
SuggestionCompletionLoader loader = new SuggestionCompletionLoader();
List<String> enums = new ArrayList<String>();
enums.add( "'Fact.f1' : ['a1', 'a2'] \n 'Fact.f2' : ['def1', 'def2', 'def3'] \n 'Fact.f2[f1=a2]' : ['c1', 'c2']" );
SuggestionCompletionEngine engine = loader.getSuggestionEngine( pkg,
new ArrayList<JarInputStream>(),
new ArrayList<DSLTokenizedMappingFile>(),
enums );
assertEquals( "String",
engine.getFieldType( "SuggestionCompletionEngineTest$NestedClass",
"name" ) );
FactPattern pat = new FactPattern( "Fact" );
SingleFieldConstraint f1 = new SingleFieldConstraint( "f1" );
f1.setValue("a1");
pat.addConstraint( f1 );
pat.addConstraint( new SingleFieldConstraint( "f2" ) );
DropDownData data = engine.getEnums( pat,
"f2" );
assertNotNull( data );
assertEquals( 3,
data.fixedList.length );
}
@Test
public void testDataEnums2() {
String pkg = "package org.test\n import org.drools.ide.common.client.modeldriven.SuggestionCompletionEngineTest.Fact";
SuggestionCompletionLoader loader = new SuggestionCompletionLoader();
List<String> enums = new ArrayList<String>();
enums.add( "'Fact.field1' : ['val1', 'val2'], 'Fact.field2' : ['val3', 'val4'], 'Fact.field2[field1=val1]' : ['f1val1a', 'f1val1b'], 'Fact.field2[field1=val2]' : ['f1val2a', 'f1val2b']" );
SuggestionCompletionEngine engine = loader.getSuggestionEngine( pkg,
new ArrayList<JarInputStream>(),
new ArrayList<DSLTokenizedMappingFile>(),
enums );
assertEquals( "String",
engine.getFieldType( "SuggestionCompletionEngineTest$Fact",
"field1" ) );
assertEquals( "String",
engine.getFieldType( "SuggestionCompletionEngineTest$Fact",
"field2" ) );
assertEquals( 4,
engine.getDataEnumListsSize() );
String[] items = engine.getDataEnumList( "Fact.field2" );
assertEquals( 2,
items.length );
assertEquals( "val3",
items[0] );
assertEquals( "val4",
items[1] );
FactPattern pat = new FactPattern( "Fact" );
SingleFieldConstraint sfc = new SingleFieldConstraint( "field2" );
pat.addConstraint( sfc );
items = engine.getEnums( pat,
"field2" ).fixedList;
assertEquals( 2,
items.length );
assertEquals( "val3",
items[0] );
assertEquals( "val4",
items[1] );
items = engine.getDataEnumList( "Fact.field1" );
assertEquals( 2,
items.length );
assertEquals( "val1",
items[0] );
assertEquals( "val2",
items[1] );
items = engine.getEnums( new FactPattern( "Fact" ),
"field1" ).fixedList;
assertEquals( 2,
items.length );
assertEquals( "val1",
items[0] );
assertEquals( "val2",
items[1] );
}
@Test
public void testCompletions() {
final SuggestionCompletionEngine com = new SuggestionCompletionEngine();
com.setFactTypes(new String[]{"Person", "Vehicle"});
com.setFieldsForTypes(new HashMap<String,ModelField[]>() {
{
put( "Person",
new ModelField[]{
new ModelField("age", Integer.class.getName(), FIELD_CLASS_TYPE.REGULAR_CLASS, SuggestionCompletionEngine.TYPE_NUMERIC),
new ModelField("rank", Integer.class.getName(),FIELD_CLASS_TYPE.REGULAR_CLASS, SuggestionCompletionEngine.TYPE_COMPARABLE),
new ModelField("name", String.class.getName(), FIELD_CLASS_TYPE.REGULAR_CLASS, SuggestionCompletionEngine.TYPE_STRING)
} );
put( "Vehicle",
new ModelField[]{
new ModelField("make", String.class.getName(), FIELD_CLASS_TYPE.REGULAR_CLASS, SuggestionCompletionEngine.TYPE_STRING),
new ModelField("type", String.class.getName(), FIELD_CLASS_TYPE.REGULAR_CLASS, SuggestionCompletionEngine.TYPE_STRING)
} );
}
});
com.setGlobalVariables(new HashMap<String, String>() {
{
put( "bar",
"Person" );
put( "baz",
"Vehicle" );
}
});
String[] c = com.getConditionalElements();
assertEquals( "not",
c[0] );
assertEquals( "exists",
c[1] );
assertEquals( "or",
c[2] );
c = com.getFactTypes();
assertEquals( 2,
c.length );
assertContains( "Person",
c );
assertContains( "Vehicle",
c );
c = com.getFieldCompletions( "Person" );
assertEquals( "age",
c[0] );
assertEquals( "rank",
c[1] );
assertEquals( "name",
c[2] );
c = com.getFieldCompletions( "Vehicle" );
assertEquals( "type",
c[1] );
assertEquals( "make",
c[0] );
c = com.getOperatorCompletions( "Person",
"name" );
assertEquals( 6,
c.length );
assertEquals( "==",
c[0] );
assertEquals( "!=",
c[1] );
assertEquals( "matches",
c[2] );
c = com.getOperatorCompletions( "Person",
"age" );
assertEquals( 8,
c.length );
assertEquals( c[0],
"==" );
assertEquals( c[1],
"!=" );
assertEquals( c[2],
"<" );
assertEquals( c[3],
">" );
c = com.getOperatorCompletions( "Person",
"rank" );
assertEquals( 8,
c.length );
assertEquals( c[0],
"==" );
assertEquals( c[1],
"!=" );
assertEquals( c[2],
"<" );
assertEquals( c[3],
">" );
c = com.getConnectiveOperatorCompletions( "Vehicle",
"make" );
assertEquals( 5,
c.length );
assertEquals( "|| ==",
c[0] );
c = com.getGlobalVariables();
assertEquals( 2,
c.length );
assertEquals( "baz",
c[0] );
assertEquals( "bar",
c[1] );
c = com.getFieldCompletionsForGlobalVariable( "bar" );
assertEquals( 3,
c.length );
assertEquals( "age",
c[0] );
assertEquals( "rank",
c[1] );
assertEquals( "name",
c[2] );
c = com.getFieldCompletionsForGlobalVariable( "baz" );
assertEquals( 2,
c.length );
assertEquals( "make",
c[0] );
assertEquals( "type",
c[1] );
//check that it has default operators for general objects
c = com.getOperatorCompletions( "Person",
"wankle" );
assertEquals( 4,
c.length );
assertEquals( "Numeric",
com.getFieldType( "Person",
"age" ) );
}
@Test
public void testAdd() {
final SuggestionCompletionEngine com = new SuggestionCompletionEngine();
com.setFactTypes(new String[]{"Foo"});
com.setFieldsForTypes(new HashMap<String,ModelField[]>() {
{
put( "Foo",
new ModelField[]{
new ModelField("a", String.class.getName(), FIELD_CLASS_TYPE.REGULAR_CLASS, "String")
});
}
});
assertEquals( 1,
com.getFactTypes().length );
assertEquals( "Foo",
com.getFactTypes()[0] );
assertEquals( 1,
com.getFieldCompletions( "Foo" ).length );
assertEquals( "a",
com.getFieldCompletions( "Foo" )[0] );
}
@Test
public void testSmartEnums() {
final SuggestionCompletionEngine sce = new SuggestionCompletionEngine();
sce.setDataEnumLists(new HashMap<String, String[]>());
sce.putDataEnumList( "Fact.type",
new String[]{"sex", "colour"} );
sce.putDataEnumList( "Fact.value[type=sex]",
new String[]{"M", "F"} );
sce.putDataEnumList( "Fact.value[type=colour]",
new String[]{"RED", "WHITE", "BLUE"} );
FactPattern pat = new FactPattern( "Fact" );
SingleFieldConstraint sfc = new SingleFieldConstraint( "type" );
sfc.setValue("sex");
pat.addConstraint( sfc );
String[] result = sce.getEnums( pat,
"value" ).fixedList;
assertEquals( 2,
result.length );
assertEquals( "M",
result[0] );
assertEquals( "F",
result[1] );
pat = new FactPattern( "Fact" );
sfc = new SingleFieldConstraint( "type" );
sfc.setValue("colour");
pat.addConstraint( sfc );
result = sce.getEnums( pat,
"value" ).fixedList;
assertEquals( 3,
result.length );
assertEquals( "RED",
result[0] );
assertEquals( "WHITE",
result[1] );
assertEquals( "BLUE",
result[2] );
result = sce.getEnums( pat,
"type" ).fixedList;
assertEquals( 2,
result.length );
assertEquals( "sex",
result[0] );
assertEquals( "colour",
result[1] );
ActionFieldValue[] vals = new ActionFieldValue[2];
vals[0] = new ActionFieldValue( "type",
"sex",
"blah" );
vals[1] = new ActionFieldValue( "value",
null,
"blah" );
result = sce.getEnums( "Fact",
vals,
"value" ).fixedList;
assertNotNull( result );
assertEquals( 2,
result.length );
assertEquals( "M",
result[0] );
assertEquals( "F",
result[1] );
assertNull( sce.getEnums( "Nothing",
vals,
"value" ) );
}
@Test
public void testSmartEnumsDependingOfSeveralFieldsTwo() {
final SuggestionCompletionEngine sce = new SuggestionCompletionEngine();
sce.setDataEnumLists(new HashMap<String, String[]>());
sce.putDataEnumList( "Fact.field1",
new String[]{"a1", "a2"} );
sce.putDataEnumList( "Fact.field2",
new String[]{"b1", "b2"} );
sce.putDataEnumList( "Fact.field3[field1=a1,field2=b1]",
new String[]{"c1", "c2", "c3"} );
sce.putDataEnumList( "Fact.field4[field1=a1]",
new String[]{"d1", "d2"} );
FactPattern pat = new FactPattern( "Fact" );
SingleFieldConstraint sfc = new SingleFieldConstraint( "field1" );
sfc.setValue("a1");
pat.addConstraint( sfc );
SingleFieldConstraint sfc2 = new SingleFieldConstraint( "field2" );
sfc2.setValue("b1");
pat.addConstraint( sfc2 );
String[] result = sce.getEnums( pat,
"field3" ).fixedList;
assertEquals( 3,
result.length );
assertEquals( "c1",
result[0] );
assertEquals( "c2",
result[1] );
assertEquals( "c3",
result[2] );
}
@Test
public void testSmartEnumsDependingOfSeveralFieldsFive() {
final SuggestionCompletionEngine sce = new SuggestionCompletionEngine();
sce.setDataEnumLists(new HashMap<String, String[]>());
sce.putDataEnumList( "Fact.field1",
new String[]{"a1", "a2"} );
sce.putDataEnumList( "Fact.field2",
new String[]{"b1", "b2"} );
sce.putDataEnumList( "Fact.field3",
new String[]{"c1", "c2", "c3"} );
sce.putDataEnumList( "Fact.longerField4",
new String[]{"d1", "d2"} );
sce.putDataEnumList( "Fact.field5",
new String[]{"e1", "e2"} );
sce.putDataEnumList( "Fact.field6[field1=a1, field2=b2, field3=c3,longerField4=d1,field5=e2]",
new String[]{"f1", "f2"} );
FactPattern pat = new FactPattern( "Fact" );
SingleFieldConstraint sfc = new SingleFieldConstraint( "field1" );
sfc.setValue("a1");
pat.addConstraint( sfc );
SingleFieldConstraint sfc2 = new SingleFieldConstraint( "field2" );
sfc2.setValue("b2");
pat.addConstraint( sfc2 );
SingleFieldConstraint sfc3 = new SingleFieldConstraint( "field3" );
sfc3.setValue("c3");
pat.addConstraint( sfc3 );
SingleFieldConstraint sfc4 = new SingleFieldConstraint( "longerField4" );
sfc4.setValue("d1");
pat.addConstraint( sfc4 );
assertNull( sce.getEnums( pat,
"field6" ) );
SingleFieldConstraint sfc5 = new SingleFieldConstraint( "field5" );
sfc5.setValue("e2");
pat.addConstraint( sfc5 );
String[] result2 = sce.getEnums( pat,
"field6" ).fixedList;
assertEquals( 2,
result2.length );
assertEquals( "f1",
result2[0] );
assertEquals( "f2",
result2[1] );
}
@Test
public void testSmarterLookupEnums() {
final SuggestionCompletionEngine sce = new SuggestionCompletionEngine();
sce.setDataEnumLists(new HashMap<String, String[]>());
sce.putDataEnumList( "Fact.type",
new String[]{"sex", "colour"} );
sce.putDataEnumList( "Fact.value[f1, f2]",
new String[]{"select something from database where x=@{f1} and y=@{f2}"} );
FactPattern fp = new FactPattern( "Fact" );
String[] drops = sce.getEnums( fp,
"type" ).fixedList;
assertEquals( 2,
drops.length );
assertEquals( "sex",
drops[0] );
assertEquals( "colour",
drops[1] );
Map<String, Object> lookupFields = sce.loadDataEnumLookupFields();
assertEquals( 1,
lookupFields.size() );
String[] flds = (String[]) lookupFields.get( "Fact.value" );
assertEquals( 2,
flds.length );
assertEquals( "f1",
flds[0] );
assertEquals( "f2",
flds[1] );
FactPattern pat = new FactPattern( "Fact" );
SingleFieldConstraint sfc = new SingleFieldConstraint( "f1" );
sfc.setValue("f1val");
pat.addConstraint( sfc );
sfc = new SingleFieldConstraint( "f2" );
sfc.setValue("f2val");
pat.addConstraint( sfc );
DropDownData dd = sce.getEnums( pat,
"value" );
assertNull( dd.fixedList );
assertNotNull( dd.queryExpression );
assertNotNull( dd.valuePairs );
assertEquals( 2,
dd.valuePairs.length );
assertEquals( "select something from database where x=@{f1} and y=@{f2}",
dd.queryExpression );
assertEquals( "f1=f1val",
dd.valuePairs[0] );
assertEquals( "f2=f2val",
dd.valuePairs[1] );
//and now for the RHS
ActionFieldValue[] vals = new ActionFieldValue[2];
vals[0] = new ActionFieldValue( "f1",
"f1val",
"blah" );
vals[1] = new ActionFieldValue( "f2",
"f2val",
"blah" );
dd = sce.getEnums( "Fact",
vals,
"value" );
assertNull( dd.fixedList );
assertNotNull( dd.queryExpression );
assertNotNull( dd.valuePairs );
assertEquals( 2,
dd.valuePairs.length );
assertEquals( "select something from database where x=@{f1} and y=@{f2}",
dd.queryExpression );
assertEquals( "f1=f1val",
dd.valuePairs[0] );
assertEquals( "f2=f2val",
dd.valuePairs[1] );
}
@Test
public void testSmarterLookupEnumsDifferentOrder() {
final SuggestionCompletionEngine sce = new SuggestionCompletionEngine();
sce.setDataEnumLists(new HashMap<String, String[]>());
sce.putDataEnumList( "Fact.type",
new String[]{"sex", "colour"} );
sce.putDataEnumList( "Fact.value[e1, e2]",
new String[]{"select something from database where x=@{e1} and y=@{e2}"} );
sce.putDataEnumList( "Fact.value[f1, f2]",
new String[]{"select something from database where x=@{f1} and y=@{f2}"} );
FactPattern fp = new FactPattern( "Fact" );
String[] drops = sce.getEnums( fp,
"type" ).fixedList;
assertEquals( 2,
drops.length );
assertEquals( "sex",
drops[0] );
assertEquals( "colour",
drops[1] );
Map<String, Object> lookupFields = sce.loadDataEnumLookupFields();
assertEquals( 1,
lookupFields.size() );
String[] flds = (String[]) lookupFields.get( "Fact.value" );
assertEquals( 2,
flds.length );
assertEquals( "f1",
flds[0] );
assertEquals( "f2",
flds[1] );
FactPattern pat = new FactPattern( "Fact" );
SingleFieldConstraint sfc = new SingleFieldConstraint( "f1" );
sfc.setValue("f1val");
pat.addConstraint( sfc );
sfc = new SingleFieldConstraint( "f2" );
sfc.setValue("f2val");
pat.addConstraint( sfc );
DropDownData dd = sce.getEnums( pat,
"value" );
assertNull( dd.fixedList );
assertNotNull( dd.queryExpression );
assertNotNull( dd.valuePairs );
assertEquals( 2,
dd.valuePairs.length );
assertEquals( "select something from database where x=@{f1} and y=@{f2}",
dd.queryExpression );
assertEquals( "f1=f1val",
dd.valuePairs[0] );
assertEquals( "f2=f2val",
dd.valuePairs[1] );
//and now for the RHS
ActionFieldValue[] vals = new ActionFieldValue[2];
vals[0] = new ActionFieldValue( "f1",
"f1val",
"blah" );
vals[1] = new ActionFieldValue( "f2",
"f2val",
"blah" );
dd = sce.getEnums( "Fact",
vals,
"value" );
assertNull( dd.fixedList );
assertNotNull( dd.queryExpression );
assertNotNull( dd.valuePairs );
assertEquals( 2,
dd.valuePairs.length );
assertEquals( "select something from database where x=@{f1} and y=@{f2}",
dd.queryExpression );
assertEquals( "f1=f1val",
dd.valuePairs[0] );
assertEquals( "f2=f2val",
dd.valuePairs[1] );
}
@Test
public void testSimpleEnums() {
final SuggestionCompletionEngine sce = new SuggestionCompletionEngine();
sce.setDataEnumLists(new HashMap<String, String[]>());
sce.putDataEnumList( "Fact.type",
new String[]{"sex", "colour"} );
assertEquals( 2,
sce.getEnumValues( "Fact",
"type" ).length );
assertEquals( "sex",
sce.getEnumValues( "Fact",
"type" )[0] );
assertEquals( "colour",
sce.getEnumValues( "Fact",
"type" )[1] );
}
private void assertContains(final String string,
final String[] c) {
for ( int i = 0; i < c.length; i++ ) {
if ( string.equals( c[i] ) ) {
return;
}
}
fail( "String array did not contain: " + string );
}
@Test
public void testGlobalAndFacts() {
final SuggestionCompletionEngine com = new SuggestionCompletionEngine();
com.setGlobalVariables(new HashMap<String, String>() {
{
put( "y", "Foo" );
}
});
com.setFieldsForTypes(new HashMap<String,ModelField[]>() {
{
put( "Foo",
new ModelField[]{
new ModelField("a", String.class.getName(), FIELD_CLASS_TYPE.REGULAR_CLASS, "String")
});
}
});
assertFalse( com.isGlobalVariable( "x" ) );
assertTrue( com.isGlobalVariable( "y" ) );
}
@Test
public void testDataDropDown() {
assertNull( DropDownData.create( null ) );
assertNull( DropDownData.create( null, null ) );
assertNotNull( DropDownData.create( new String[]{"hey"} ));
assertNotNull( DropDownData.create( "abc", new String[]{"hey"} ) );
}
@Test
public void testFilter() {
final SuggestionCompletionEngine sce = new SuggestionCompletionEngine();
sce.setFactTypes(new String[]{"Person", "Vehicle"});
sce.setFieldsForTypes(new HashMap<String,ModelField[]>() {
{
put( "Person",
new ModelField[]{
new ModelField("age", Integer.class.getName(), FIELD_CLASS_TYPE.REGULAR_CLASS, SuggestionCompletionEngine.TYPE_NUMERIC),
} );
put( "Vehicle",
new ModelField[]{
new ModelField("make", String.class.getName(), FIELD_CLASS_TYPE.REGULAR_CLASS, SuggestionCompletionEngine.TYPE_STRING),
} );
}
});
assertEquals(2, sce.getFactTypes().length);
sce.setFactTypeFilter(new FactTypeFilter() {
public boolean filter(String originalFact) {
return "Person".equals(originalFact);
}
});
assertEquals(1, sce.getFactTypes().length);
sce.setFilteringFacts(false);
assertEquals(2, sce.getFactTypes().length);
sce.setFilteringFacts(true);
assertEquals(1, sce.getFactTypes().length);
sce.setFactTypeFilter(null);
assertEquals(2, sce.getFactTypes().length);
}
public static class NestedClass {
private String name;
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
}
public static class Fact {
private String field1;
private String field2;
public String getField1() {
return field1;
}
public void setField1(String field1) {
this.field1 = field1;
}
public String getField2() {
return field2;
}
public void setField2(String field2) {
this.field2 = field2;
}
}
}