1 package org.opendaylight.controller.config.manager.impl.osgi;
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;
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;
21 public class BundleContextBackedModuleFactoriesResolverTest {
24 private BundleContext bundleContext;
25 private BundleContextBackedModuleFactoriesResolver resolver;
26 private ServiceReference<?> s1;
27 private ServiceReference<?> s2;
28 private ModuleFactory f1;
29 private ModuleFactory f2;
32 public void setUp() throws Exception {
33 MockitoAnnotations.initMocks(this);
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);
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();
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();
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());
73 public void testDuplicateFactories() throws Exception {
74 doReturn(f1).when(bundleContext).getService(s2);
76 resolver.getAllFactories();
77 } catch (Exception e) {
78 assertThat(e.getMessage(), containsString(f1.getImplementationName()));
79 assertThat(e.getMessage(), containsString("unique"));
83 fail("Should fail with duplicate factory name");
86 @Test(expected = NullPointerException.class)
87 public void testNullFactory() throws Exception {
88 doReturn(null).when(bundleContext).getService(s2);
89 resolver.getAllFactories();
92 @Test(expected = IllegalStateException.class)
93 public void testNullFactoryName() throws Exception {
94 doReturn(null).when(f1).getImplementationName();
95 resolver.getAllFactories();
98 @Test(expected = NullPointerException.class)
99 public void testNullBundleName() throws Exception {
100 doReturn(null).when(s1).getBundle();
101 resolver.getAllFactories();