UTILS-17

[utils] / wicket / joe / src / test / java / org / wamblee / wicket / inject / InjectionBehaviorTest.java

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 d5c62024137d8940d3b524d792c80a095ccc2829..e9ba180ca6d767e54d0bd12c6d408c153b11d8de 100644 (file)
--- a/wicket/joe/src/test/java/org/wamblee/wicket/inject/InjectionBehaviorTest.java
+++ b/wicket/joe/src/test/java/org/wamblee/wicket/inject/InjectionBehaviorTest.java
@@ -33,9 +33,9 @@ import static org.mockito.Mockito.*;
 
 public class InjectionBehaviorTest {
 
-    Component component;
-    Injector injector;
-    InjectorFactory injectorFactory;
+    private Component component;
+    private Injector injector;
+    private InjectorFactory injectorFactory;
 
     @Before
     public void setUp() {
@@ -53,21 +53,24 @@ public class InjectionBehaviorTest {
 
     @Test
     public void testNoInjectionInitially() {
-        InjectionBehavior behavior = new InjectionBehavior();
-        behavior.beforeRender(component); 
+           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();
-        
-        behavior = ObjectSerializationUtils.deserialize(ObjectSerializationUtils.serialize(behavior), InjectionBehavior.class);
-       
-        behavior.beforeRender(component); 
-        verify(injector).inject(same(component));
+    public void testInjectOnlyOnceAfterDeserialisation() throws Exception {
+           Component component = mock(Component.class);
+        InjectionBehavior behavior = new InjectionBehavior(component);
+
+        behavior = ObjectSerializationUtils.deserialize(
+            ObjectSerializationUtils.serialize(behavior),
+            InjectionBehavior.class);
+
+        verify(injector).inject(any(Component.class));
         reset(injector);
-        behavior.beforeRender(component); 
+        behavior.beforeRender(component);
         verifyNoMoreInteractions(injector);
 
     }