X-Git-Url: http://wamblee.org/gitweb/?a=blobdiff_plain;f=system%2Fgeneral%2Fsrc%2Ftest%2Fjava%2Forg%2Fwamblee%2Fsystem%2Fgraph%2FGraphTest.java;h=675a06bd6c1c7247c713a9b1af79fc85b40afb27;hb=8557fbe8c7ea4e1cbcbf10d3c4e8c60c9c1e312b;hp=7a5a87971204255aa1c945af36b5b972fc826a04;hpb=34b0dc0bca5a2d99394b9692b4ea14ce9e2141b7;p=utils

diff --git a/system/general/src/test/java/org/wamblee/system/graph/GraphTest.java b/system/general/src/test/java/org/wamblee/system/graph/GraphTest.java
index 7a5a8797..675a06bd 100644
--- a/system/general/src/test/java/org/wamblee/system/graph/GraphTest.java
+++ b/system/general/src/test/java/org/wamblee/system/graph/GraphTest.java
@@ -18,9 +18,10 @@ package org.wamblee.system.graph;
 import java.util.Arrays;
 import java.util.List;
 
-import org.easymock.classextension.EasyMock;
+import static org.mockito.Mockito.*;
 import org.wamblee.test.AssertionUtils;
 
+
 import junit.framework.TestCase;
 
 public class GraphTest extends TestCase {
@@ -156,12 +157,13 @@ public class GraphTest extends TestCase {
         graph.addNode(x);
         graph.addNode(y);
         graph.addEdge(e);
-        Visitor visitor = EasyMock.createMock(Visitor.class);
-        visitor.visitNode(x);
-        visitor.visitNode(y);
-        visitor.visitEdge(e);
-        EasyMock.replay(visitor);
+        Visitor visitor = mock(Visitor.class); 
+        
         graph.accept(visitor);
-        EasyMock.verify(visitor);
+        verify(visitor).visitNode(x);
+        verify(visitor).visitNode(y);
+        verify(visitor).visitEdge(e);
+        
+        verifyNoMoreInteractions(visitor);
     }
 }