[jboss-svn-commits] JBL Code SVN: r32362 - labs/jbossrules/trunk/drools-compiler/src/test/java/org/drools/guvnor/client/modeldriven.

[jboss-svn-commits at lists.jboss.org]

Author: baunax
Date: 2010-04-01 16:09:19 -0400 (Thu, 01 Apr 2010)
New Revision: 32362

Modified:
   labs/jbossrules/trunk/drools-compiler/src/test/java/org/drools/guvnor/client/modeldriven/SuggestionCompletionEngineTest.java
Log:
added test for fact filtering

Modified: labs/jbossrules/trunk/drools-compiler/src/test/java/org/drools/guvnor/client/modeldriven/SuggestionCompletionEngineTest.java
===================================================================
--- labs/jbossrules/trunk/drools-compiler/src/test/java/org/drools/guvnor/client/modeldriven/SuggestionCompletionEngineTest.java	2010-04-01 18:29:35 UTC (rev 32361)
+++ labs/jbossrules/trunk/drools-compiler/src/test/java/org/drools/guvnor/client/modeldriven/SuggestionCompletionEngineTest.java	2010-04-01 20:09:19 UTC (rev 32362)
@@ -719,6 +719,45 @@
 
     }
 
+    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;