[jboss-svn-commits] JBL Code SVN: r25800 - labs/jbossrules/trunk/drools-eclipse/org.drools.eclipse.test/src/test/java/org/drools/eclipse/editors/rete.

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

Author: KrisVerlaenen
Date: 2009-03-24 12:54:42 -0400 (Tue, 24 Mar 2009)
New Revision: 25800

Modified:
   labs/jbossrules/trunk/drools-eclipse/org.drools.eclipse.test/src/test/java/org/drools/eclipse/editors/rete/ReteooLayoutFactoryTest.java
Log:
JBRULES-2003: Rete viewer should support new rete nodes
 - updated tests accordingly

Modified: labs/jbossrules/trunk/drools-eclipse/org.drools.eclipse.test/src/test/java/org/drools/eclipse/editors/rete/ReteooLayoutFactoryTest.java
===================================================================
--- labs/jbossrules/trunk/drools-eclipse/org.drools.eclipse.test/src/test/java/org/drools/eclipse/editors/rete/ReteooLayoutFactoryTest.java	2009-03-24 16:13:22 UTC (rev 25799)
+++ labs/jbossrules/trunk/drools-eclipse/org.drools.eclipse.test/src/test/java/org/drools/eclipse/editors/rete/ReteooLayoutFactoryTest.java	2009-03-24 16:54:42 UTC (rev 25800)
@@ -20,6 +20,7 @@
 import org.drools.lang.descr.PackageDescr;
 import org.drools.reteoo.AlphaNodeVertex;
 import org.drools.reteoo.BaseVertex;
+import org.drools.reteoo.EntryPointNodeVertex;
 import org.drools.reteoo.LeftInputAdapterNodeVertex;
 import org.drools.reteoo.ObjectTypeNodeVertex;
 import org.drools.reteoo.ReteVertex;
@@ -70,11 +71,11 @@
 
         int rownum = rows.getDepth();
 
-        assertEquals( 5,
+        assertEquals( 6,
                       rownum );
 
-        int[] expectedDepths = new int[]{-1, 0, 1, 2, 3};
-        int[] expectedSizes = new int[]{1, 1, 2, 2, 2};
+        int[] expectedDepths = new int[]{-1, 0, 1, 2, 3, 4};
+        int[] expectedSizes = new int[]{1, 1, 1, 2, 2, 2};
 
         for ( int j = 0; j < rownum; j++ ) {
             final Row row = rows.get( j );
@@ -122,8 +123,10 @@
 
                      } );
 
-        Class[] expectedTypes = new Class[]{ReteVertex.class, ObjectTypeNodeVertex.class, AlphaNodeVertex.class, AlphaNodeVertex.class, LeftInputAdapterNodeVertex.class, LeftInputAdapterNodeVertex.class, RuleTerminalNodeVertex.class,
-                RuleTerminalNodeVertex.class};
+        Class[] expectedTypes = new Class[]{ReteVertex.class, EntryPointNodeVertex.class,
+    		ObjectTypeNodeVertex.class, AlphaNodeVertex.class, AlphaNodeVertex.class,
+    		LeftInputAdapterNodeVertex.class, LeftInputAdapterNodeVertex.class,
+    		RuleTerminalNodeVertex.class, RuleTerminalNodeVertex.class};
 
         for ( int i = 0; i < yOrder.length; i++ ) {
             assertEquals( expectedTypes[i],