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

   1 package org.opendaylight.controller.config.manager.impl.osgi;
   2
   3 import static org.mockito.Matchers.any;
   4 import static org.mockito.Mockito.doNothing;
   5 import static org.mockito.Mockito.doReturn;
   6 import static org.mockito.Mockito.mock;
   7 import static org.mockito.Mockito.verify;
   8
   9 import java.util.Dictionary;
  10 import org.junit.Test;
  11 import org.mockito.Mockito;
  12 import org.opendaylight.controller.config.manager.impl.osgi.mapping.BindingContextProvider;
  13 import org.opendaylight.controller.config.manager.impl.osgi.mapping.RefreshingSCPModuleInfoRegistry;
  14 import org.opendaylight.yangtools.concepts.ObjectRegistration;
  15 import org.opendaylight.yangtools.sal.binding.generator.api.ClassLoadingStrategy;
  16 import org.opendaylight.yangtools.sal.binding.generator.api.ModuleInfoRegistry;
  17 import org.opendaylight.yangtools.sal.binding.generator.util.BindingRuntimeContext;
  18 import org.opendaylight.yangtools.yang.binding.YangModuleInfo;
  19 import org.opendaylight.yangtools.yang.model.api.SchemaContextProvider;
  20 import org.osgi.framework.BundleContext;
  21 import org.osgi.framework.ServiceRegistration;
  22
  23 public class RefreshingSCPModuleInfoRegistryTest {
  24     @Test
  25     public void testConstructor() throws Exception {
  26         ModuleInfoRegistry reg = mock(ModuleInfoRegistry.class);
  27         SchemaContextProvider prov = mock(SchemaContextProvider.class);
  28         doReturn("string").when(prov).toString();
  29         BundleContext ctxt = mock(BundleContext.class);
  30         ServiceRegistration<?> servReg = mock(ServiceRegistration.class);
  31         doReturn(servReg).when(ctxt).registerService(any(Class.class), any(SchemaContextProvider.class), any(Dictionary.class));
  32         doReturn(servReg).when(ctxt).registerService(Mockito.anyString(), any(Object.class), any(Dictionary.class));
  33         doNothing().when(servReg).setProperties(any(Dictionary.class));
  34
  35         final ClassLoadingStrategy classLoadingStrat = mock(ClassLoadingStrategy.class);
  36         final BindingContextProvider codecRegistryProvider = mock(BindingContextProvider.class);
  37         doNothing().when(codecRegistryProvider).update(classLoadingStrat, prov);
  38         final BindingRuntimeContext bindingRuntimeContext = mock(BindingRuntimeContext.class);
  39         doReturn("B-runtime-context").when(bindingRuntimeContext).toString();
  40         doReturn(bindingRuntimeContext).when(codecRegistryProvider).getBindingContext();
  41
  42         RefreshingSCPModuleInfoRegistry scpreg = new RefreshingSCPModuleInfoRegistry(reg, prov, classLoadingStrat, codecRegistryProvider, ctxt);
  43
  44         doNothing().when(servReg).unregister();
  45
  46         YangModuleInfo modInfo = mock(YangModuleInfo.class);
  47         doReturn("").when(modInfo).toString();
  48         ObjectRegistration<YangModuleInfo> ymi = mock(ObjectRegistration.class);
  49         doReturn(ymi).when(reg).registerModuleInfo(modInfo);
  50
  51         scpreg.registerModuleInfo(modInfo);
  52
  53         verify(codecRegistryProvider).update(classLoadingStrat, prov);
  54
  55         scpreg.close();
  56
  57         Mockito.verify(servReg, Mockito.times(1)).setProperties(any(Dictionary.class));
  58         Mockito.verify(servReg, Mockito.times(1)).unregister();
  59     }
  60 }