[jboss-svn-commits] JBL Code SVN: r7087 - labs/jbossrules/trunk/drools-core/src/test/java/org/drools/reteoo

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

Author: mark.proctor at jboss.com
Date: 2006-10-24 11:18:19 -0400 (Tue, 24 Oct 2006)
New Revision: 7087

Modified:
   labs/jbossrules/trunk/drools-core/src/test/java/org/drools/reteoo/MockTupleSink.java
   labs/jbossrules/trunk/drools-core/src/test/java/org/drools/reteoo/TupleSourceTest.java
Log:
-Fixed TupleSourceTest

Modified: labs/jbossrules/trunk/drools-core/src/test/java/org/drools/reteoo/MockTupleSink.java
===================================================================
--- labs/jbossrules/trunk/drools-core/src/test/java/org/drools/reteoo/MockTupleSink.java	2006-10-24 15:13:00 UTC (rev 7086)
+++ labs/jbossrules/trunk/drools-core/src/test/java/org/drools/reteoo/MockTupleSink.java	2006-10-24 15:18:19 UTC (rev 7087)
@@ -30,7 +30,7 @@
 
 public class MockTupleSink extends TupleSource
     implements
-    TupleSink,
+    TupleSinkNode,
     NodeMemory {
     /**
      * 
@@ -39,6 +39,9 @@
     private final List        asserted         = new ArrayList();
     private final List        retracted        = new ArrayList();
 
+    private TupleSinkNode           previousTupleSinkNode;
+    private TupleSinkNode           nextTupleSinkNode;
+    
     public MockTupleSink() {
         super( 0 );
     }
@@ -113,5 +116,41 @@
         // TODO Auto-generated method stub
         return Collections.EMPTY_LIST;
     }
+    
+    /**
+     * Returns the next node
+     * @return
+     *      The next TupleSinkNode
+     */
+    public TupleSinkNode getNextTupleSinkNode() {
+        return this.nextTupleSinkNode;
+    }
 
+    /**
+     * Sets the next node 
+     * @param next
+     *      The next TupleSinkNode
+     */
+    public void setNextTupleSinkNode(final TupleSinkNode next) {
+        this.nextTupleSinkNode = next;
+    }
+
+    /**
+     * Returns the previous node
+     * @return
+     *      The previous TupleSinkNode
+     */
+    public TupleSinkNode getPreviousTupleSinkNode() {
+        return this.previousTupleSinkNode;
+    }
+
+    /**
+     * Sets the previous node 
+     * @param previous
+     *      The previous TupleSinkNode
+     */
+    public void setPreviousTupleSinkNode(final TupleSinkNode previous) {
+        this.previousTupleSinkNode = previous;
+    }    
+
 }
\ No newline at end of file

Modified: labs/jbossrules/trunk/drools-core/src/test/java/org/drools/reteoo/TupleSourceTest.java
===================================================================
--- labs/jbossrules/trunk/drools-core/src/test/java/org/drools/reteoo/TupleSourceTest.java	2006-10-24 15:13:00 UTC (rev 7086)
+++ labs/jbossrules/trunk/drools-core/src/test/java/org/drools/reteoo/TupleSourceTest.java	2006-10-24 15:18:19 UTC (rev 7087)
@@ -42,7 +42,7 @@
         field.setAccessible( true );
         TupleSinkPropagator sink = (TupleSinkPropagator) field.get( source );
 
-        assertNull( sink );
+        assertSame( EmptyTupleSinkAdapter.getInstance(), sink );
 
         final MockTupleSink sink1 = new MockTupleSink();
         source.addTupleSink( sink1 );
@@ -79,6 +79,13 @@
                     sink.getClass() );
         assertEquals( 1,
                       sink.getSinks().length );
+        
+        source.removeTupleSink( sink3 );
+        sink = (TupleSinkPropagator) field.get( source );
+        assertSame( EmptyTupleSinkAdapter.getInstance(),
+                    sink );
+        assertEquals( 0,
+                      sink.getSinks().length );        
     }
 
 }
\ No newline at end of file