X-Git-Url: http://wamblee.org/gitweb/?a=blobdiff_plain;f=support%2Fgeneral%2Fsrc%2Ftest%2Fjava%2Forg%2Fwamblee%2Fgeneral%2FThreadSpecificProxyFactoryTest.java;h=0cffd5a0d38e6a2e4668b45c188dc4b7210b0625;hb=26805fc0810098c4bd8009a35c8719478e74153e;hp=9b52817e5251bb241437cd6b685f428b86cee52a;hpb=f07884740605cfc19ccb411cb65224792c9185bb;p=utils

diff --git a/support/general/src/test/java/org/wamblee/general/ThreadSpecificProxyFactoryTest.java b/support/general/src/test/java/org/wamblee/general/ThreadSpecificProxyFactoryTest.java
index 9b52817e..0cffd5a0 100644
--- a/support/general/src/test/java/org/wamblee/general/ThreadSpecificProxyFactoryTest.java
+++ b/support/general/src/test/java/org/wamblee/general/ThreadSpecificProxyFactoryTest.java
@@ -23,6 +23,7 @@ import org.junit.After;
 import org.junit.Before;
 import org.junit.Test;
 import org.wamblee.general.ThreadSpecificProxyFactory;
+import org.wamblee.general.ThreadSpecificProxyFactory.CreationCallback;
 
 public class ThreadSpecificProxyFactoryTest {
 
@@ -58,6 +59,21 @@ public class ThreadSpecificProxyFactoryTest {
         verify(svc).execute(10);
     }
 
+    @Test
+    public void testCreationCallback() throws Exception {
+        CreationCallback callback = mock(CreationCallback.class);
+        Service initialService = mock(Service.class);
+        when(callback.create()).thenReturn(initialService);
+
+        factory = new ThreadSpecificProxyFactory<Service>(Service.class,
+            callback);
+        proxy = factory.getProxy();
+        Service svc = factory.get();
+
+        assertSame(initialService, svc);
+        verify(callback).create();
+    }
+
     @Test
     public void testInvokeThroughProxyWithException() throws Exception {
         Service svc = mock(Service.class);
@@ -102,11 +118,12 @@ public class ThreadSpecificProxyFactoryTest {
 
     @Test(expected = IllegalArgumentException.class)
     public void testNotAnInterface() {
-        ThreadSpecificProxyFactory f = new ThreadSpecificProxyFactory(String.class);
+        ThreadSpecificProxyFactory f = new ThreadSpecificProxyFactory(
+            String.class);
     }
-    
+
     @Test
-    public void testProxyToStringWhileUninitialized() { 
+    public void testProxyToStringWhileUninitialized() {
         String val = proxy.toString();
     }
 }