X-Git-Url: http://wamblee.org/gitweb/?a=blobdiff_plain;f=wicket%2Fjoe%2Fsrc%2Ftest%2Fjava%2Forg%2Fwamblee%2Fwicket%2Finject%2FInjectionBehaviorTest.java;fp=wicket%2Fjoe%2Fsrc%2Ftest%2Fjava%2Forg%2Fwamblee%2Fwicket%2Finject%2FInjectionBehaviorTest.java;h=e9ba180ca6d767e54d0bd12c6d408c153b11d8de;hb=ec59f519fe8286b7fadad2b7712f6b32f2b31939;hp=d54de1ce4265af9f344611e97fc58df40d7e08b0;hpb=514d7b3dd07385a4150514b8fe09ab7c4ff791c3;p=utils

diff --git a/wicket/joe/src/test/java/org/wamblee/wicket/inject/InjectionBehaviorTest.java b/wicket/joe/src/test/java/org/wamblee/wicket/inject/InjectionBehaviorTest.java
index d54de1ce..e9ba180c 100644
--- a/wicket/joe/src/test/java/org/wamblee/wicket/inject/InjectionBehaviorTest.java
+++ b/wicket/joe/src/test/java/org/wamblee/wicket/inject/InjectionBehaviorTest.java
@@ -53,21 +53,22 @@ public class InjectionBehaviorTest {
 
     @Test
     public void testNoInjectionInitially() {
-        InjectionBehavior behavior = new InjectionBehavior();
+	    Component component = mock(Component.class);
+        InjectionBehavior behavior = new InjectionBehavior(component);
         behavior.beforeRender(component);
         verifyNoMoreInteractions(injector);
     }
 
     @Test
     public void testInjectOnlyOnceAfterDeserialisation() throws Exception {
-        InjectionBehavior behavior = new InjectionBehavior();
+	    Component component = mock(Component.class);
+        InjectionBehavior behavior = new InjectionBehavior(component);
 
         behavior = ObjectSerializationUtils.deserialize(
             ObjectSerializationUtils.serialize(behavior),
             InjectionBehavior.class);
 
-        behavior.beforeRender(component);
-        verify(injector).inject(same(component));
+        verify(injector).inject(any(Component.class));
         reset(injector);
         behavior.beforeRender(component);
         verifyNoMoreInteractions(injector);