X-Git-Url: http://wamblee.org/gitweb/?a=blobdiff_plain;ds=sidebyside;f=wicket%2Fjoe%2Fsrc%2Ftest%2Fjava%2Forg%2Fwamblee%2Fwicket%2Fmodel%2FDetachableEntityTest.java;h=3346be63e0ad23ef30ae1c49b13aaab5843d957d;hb=1df323c6b726222262194aecd6d23ed0da194458;hp=e4cf731711bc897dea589c919ae96a2ca9550ce5;hpb=51cc217ee697147805ea23dd777511b82efe94ce;p=utils

diff --git a/wicket/joe/src/test/java/org/wamblee/wicket/model/DetachableEntityTest.java b/wicket/joe/src/test/java/org/wamblee/wicket/model/DetachableEntityTest.java
index e4cf7317..3346be63 100644
--- a/wicket/joe/src/test/java/org/wamblee/wicket/model/DetachableEntityTest.java
+++ b/wicket/joe/src/test/java/org/wamblee/wicket/model/DetachableEntityTest.java
@@ -28,22 +28,27 @@ public class DetachableEntityTest {
         
     }
     
+    private X initialValue; 
     private Detachable<X> detachable;
     private DetachableEntity<X> entity; 
     
     @Before
     public void setUp() { 
+        initialValue = mock(X.class);
         detachable = mock(Detachable.class);
+        when(detachable.get()).thenReturn(initialValue);
         entity = new DetachableEntity<X>(detachable);
     }
 
     @Test
     public void testGetObject() {
         X x = new X(); 
+        reset(detachable);
         when(detachable.get()).thenReturn(x);
         X value = entity.getObject();
-        assertSame(x, value);
-        verify(detachable).get();
+        // we did not detach it yet so the old value should be returned.
+        assertSame(initialValue, value);
+        verifyNoMoreInteractions(detachable);
     }
 
     @Test
@@ -51,9 +56,12 @@ public class DetachableEntityTest {
         X x = new X(); 
         when(detachable.get()).thenReturn(x);
         
-        entity.onDetach();
+        entity.detach();
         verify(detachable).detach();
         
+        reset(detachable);
+        when(detachable.get()).thenReturn(x);
+        
         X y = entity.getObject();
         assertSame(x, y); 
         verify(detachable).get();