X-Git-Url: http://wamblee.org/gitweb/?a=blobdiff_plain;f=support%2Fcdi%2Fsrc%2Ftest%2Fjava%2Forg%2Fwamblee%2Fcdi%2FBeanManagerLookupTest.java;h=5b7f51d495981a426e9729bb76fbcaf0a65cc2c9;hb=a5c9a040af910244a90db82e00e9de967a8ea1e6;hp=8115ac49825c438bdefc021c649f1cd452f6a6bb;hpb=8fee9047edd0fe3acc5643aeea20cdc89a684b48;p=utils

diff --git a/support/cdi/src/test/java/org/wamblee/cdi/BeanManagerLookupTest.java b/support/cdi/src/test/java/org/wamblee/cdi/BeanManagerLookupTest.java
index 8115ac49..5b7f51d4 100644
--- a/support/cdi/src/test/java/org/wamblee/cdi/BeanManagerLookupTest.java
+++ b/support/cdi/src/test/java/org/wamblee/cdi/BeanManagerLookupTest.java
@@ -16,15 +16,16 @@
 package org.wamblee.cdi;
 
 import static junit.framework.Assert.*;
+import static org.mockito.Mockito.*;
 
 import javax.enterprise.inject.spi.BeanManager;
 import javax.naming.InitialContext;
 
 import org.junit.Test;
-import org.wamblee.support.jndi.StubInitialContextFactory;
-
-public class BeanManagerLookupTest {
+import org.wamblee.test.jndi.StubInitialContextFactory;
 
+public class BeanManagerLookupTest extends BaseTestFixture {
+  
     @Test
     public void testLookupInJndi() throws Exception {
         BeanManagerSetup setup = new BeanManagerSetup();
@@ -33,12 +34,34 @@ public class BeanManagerLookupTest {
         InitialContext ctx = new InitialContext(); 
         ctx.bind(BeanManagerLookup.BEAN_MANAGER_JNDI, setup.getBeanManager());
         try {
-            ctx = new InitialContext(); 
-            BeanManager manager = (BeanManager) ctx.lookup(BeanManagerLookup.BEAN_MANAGER_JNDI);
-            assertSame(setup.getBeanManager(), manager);
+            assertSame(setup.getBeanManager(), BeanManagerLookup.lookup());
         } finally {
             setup.shutdown();
             StubInitialContextFactory.unregister();
         }
     }
+    
+    @Test
+    public void testWithJndiButWithOverride() throws Exception { 
+        BeanManagerSetup setup = new BeanManagerSetup();
+        setup.initialize();
+        StubInitialContextFactory.register();
+        InitialContext ctx = new InitialContext(); 
+        ctx.bind(BeanManagerLookup.BEAN_MANAGER_JNDI, setup.getBeanManager());
+        try {
+            BeanManager override = mock(BeanManager.class);
+            BeanManagerLookup.setBeanManager(override);
+            assertSame(override, BeanManagerLookup.lookup());
+        } finally {
+            setup.shutdown();
+            StubInitialContextFactory.unregister();
+        }
+    }
+    
+    @Test
+    public void testNoJndiButWithOverride() throws Exception { 
+        BeanManager override = mock(BeanManager.class);
+        BeanManagerLookup.setBeanManager(override);
+        assertSame(override, BeanManagerLookup.lookup());
+    }
 }