opendaylight/config/config-manager/src/test/java/org/opendaylight/controller/config/manager/impl/osgi/BundleContextBackedModuleFactoriesResolverTest.java

   1 package org.opendaylight.controller.config.manager.impl.osgi;
   2
   3 import static org.hamcrest.CoreMatchers.containsString;
   4 import static org.junit.Assert.assertEquals;
   5 import static org.junit.Assert.assertThat;
   6 import static org.junit.Assert.assertTrue;
   7 import static org.junit.Assert.fail;
   8 import static org.mockito.Mockito.doReturn;
   9 import static org.mockito.Mockito.mock;
  10 import com.google.common.collect.Lists;
  11 import java.util.Map;
  12 import org.junit.Before;
  13 import org.junit.Test;
  14 import org.mockito.Mock;
  15 import org.mockito.MockitoAnnotations;
  16 import org.opendaylight.controller.config.spi.ModuleFactory;
  17 import org.osgi.framework.Bundle;
  18 import org.osgi.framework.BundleContext;
  19 import org.osgi.framework.ServiceReference;
  20
  21 public class BundleContextBackedModuleFactoriesResolverTest {
  22
  23     @Mock
  24     private BundleContext bundleContext;
  25     private BundleContextBackedModuleFactoriesResolver resolver;
  26     private ServiceReference<?> s1;
  27     private ServiceReference<?> s2;
  28     private ModuleFactory f1;
  29     private ModuleFactory f2;
  30
  31     @Before
  32     public void setUp() throws Exception {
  33         MockitoAnnotations.initMocks(this);
  34         s1 = getServiceRef();
  35         s2 = getServiceRef();
  36         doReturn(Lists.newArrayList(s1, s2)).when(bundleContext).getServiceReferences(ModuleFactory.class, null);
  37         f1 = getMockFactory("f1");
  38         doReturn(f1).when(bundleContext).getService(s1);
  39         f2 = getMockFactory("f2");
  40         doReturn(f2).when(bundleContext).getService(s2);
  41         resolver = new BundleContextBackedModuleFactoriesResolver(bundleContext);
  42     }
  43
  44     private ModuleFactory getMockFactory(final String name) {
  45         ModuleFactory mock = mock(ModuleFactory.class);
  46         doReturn(name).when(mock).toString();
  47         doReturn(name).when(mock).getImplementationName();
  48         return mock;
  49     }
  50
  51     private ServiceReference<?> getServiceRef() {
  52         ServiceReference<?> mock = mock(ServiceReference.class);
  53         doReturn("serviceRef").when(mock).toString();
  54         final Bundle bundle = mock(Bundle.class);
  55         doReturn(bundleContext).when(bundle).getBundleContext();
  56         doReturn(bundle).when(mock).getBundle();
  57         return mock;
  58     }
  59
  60     @Test
  61     public void testGetAllFactories() throws Exception {
  62         Map<String, Map.Entry<ModuleFactory, BundleContext>> allFactories = resolver.getAllFactories();
  63         assertEquals(2, allFactories.size());
  64         assertTrue(allFactories.containsKey(f1.getImplementationName()));
  65         assertEquals(f1, allFactories.get(f1.getImplementationName()).getKey());
  66         assertEquals(bundleContext, allFactories.get(f1.getImplementationName()).getValue());
  67         assertTrue(allFactories.containsKey(f2.getImplementationName()));
  68         assertEquals(f2, allFactories.get(f2.getImplementationName()).getKey());
  69         assertEquals(bundleContext, allFactories.get(f2.getImplementationName()).getValue());
  70     }
  71
  72     @Test
  73     public void testDuplicateFactories() throws Exception {
  74         doReturn(f1).when(bundleContext).getService(s2);
  75         try {
  76             resolver.getAllFactories();
  77         } catch (Exception e) {
  78             assertThat(e.getMessage(), containsString(f1.getImplementationName()));
  79             assertThat(e.getMessage(), containsString("unique"));
  80             return;
  81         }
  82
  83         fail("Should fail with duplicate factory name");
  84     }
  85
  86     @Test(expected = NullPointerException.class)
  87     public void testNullFactory() throws Exception {
  88         doReturn(null).when(bundleContext).getService(s2);
  89         resolver.getAllFactories();
  90     }
  91
  92     @Test(expected = IllegalStateException.class)
  93     public void testNullFactoryName() throws Exception {
  94         doReturn(null).when(f1).getImplementationName();
  95         resolver.getAllFactories();
  96     }
  97
  98     @Test(expected = NullPointerException.class)
  99     public void testNullBundleName() throws Exception {
 100         doReturn(null).when(s1).getBundle();
 101         resolver.getAllFactories();
 102     }
 103 }