X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=opendaylight%2Fconfig%2Fconfig-manager%2Fsrc%2Ftest%2Fjava%2Forg%2Fopendaylight%2Fcontroller%2Fconfig%2Fmanager%2Fimpl%2Ffactoriesresolver%2FHardcodedModuleFactoriesResolver.java;h=dd6588f3f63cb7c0f06589f1890cb68c3f94b619;hb=f3714846c0eedf09fe83844c0dc30889f790fbc2;hp=04f651f6d301e39d890aea81270e05c458d33853;hpb=2c6fd09ad16f5c2968390a674865efbbe5f1fcec;p=controller.git

diff --git a/opendaylight/config/config-manager/src/test/java/org/opendaylight/controller/config/manager/impl/factoriesresolver/HardcodedModuleFactoriesResolver.java b/opendaylight/config/config-manager/src/test/java/org/opendaylight/controller/config/manager/impl/factoriesresolver/HardcodedModuleFactoriesResolver.java
index 04f651f6d3..dd6588f3f6 100644
--- a/opendaylight/config/config-manager/src/test/java/org/opendaylight/controller/config/manager/impl/factoriesresolver/HardcodedModuleFactoriesResolver.java
+++ b/opendaylight/config/config-manager/src/test/java/org/opendaylight/controller/config/manager/impl/factoriesresolver/HardcodedModuleFactoriesResolver.java
@@ -7,21 +7,72 @@
  */
 package org.opendaylight.controller.config.manager.impl.factoriesresolver;
 
+import org.mockito.Matchers;
+import org.mockito.Mockito;
+import org.opendaylight.controller.config.spi.ModuleFactory;
+import org.osgi.framework.BundleContext;
+import org.osgi.framework.ServiceRegistration;
+
+import java.io.Closeable;
 import java.util.Arrays;
+import java.util.Dictionary;
 import java.util.List;
+import java.util.Map;
+import java.util.HashMap;
+import java.util.AbstractMap;
 
-import org.opendaylight.controller.config.spi.ModuleFactory;
+import static org.mockito.Matchers.any;
+import static org.mockito.Mockito.doNothing;
+import static org.mockito.Mockito.doReturn;
+import static org.mockito.Mockito.mock;
 
 public class HardcodedModuleFactoriesResolver implements ModuleFactoriesResolver {
-    private final List<ModuleFactory> list;
+    private Map<String, Map.Entry<ModuleFactory, BundleContext>> factories;
+
+    public HardcodedModuleFactoriesResolver(BundleContext bundleContext, ModuleFactory... list) {
+        List<ModuleFactory> factoryList = Arrays.asList(list);
+        this.factories = new HashMap<>(factoryList.size());
+        for (ModuleFactory moduleFactory : list) {
+            StringBuffer errors = new StringBuffer();
+            String moduleName = moduleFactory.getImplementationName();
+            if (moduleName == null || moduleName.isEmpty()) {
+                throw new IllegalStateException(
+                        "Invalid implementation name for " + moduleFactory);
+            }
+            String error = null;
+            Map.Entry<ModuleFactory, BundleContext> conflicting = factories.get(moduleName);
+            if (conflicting != null) {
+                error = String
+                        .format("Module name is not unique. Found two conflicting factories with same name '%s': " +
+                                "\n\t%s\n\t%s\n", moduleName, conflicting.getKey(), moduleFactory);
+
+            }
+
+            if (error == null) {
+                factories.put(moduleName, new AbstractMap.SimpleEntry<>(moduleFactory,
+                        bundleContext));
+            } else {
+                errors.append(error);
+            }
+            if (errors.length() > 0) {
+                throw new IllegalArgumentException(errors.toString());
+            }
+        }
+    }
 
-    public HardcodedModuleFactoriesResolver(ModuleFactory... list) {
-        this.list = Arrays.asList(list);
+    private static BundleContext mockBundleContext() {
+        BundleContext bundleContext = Mockito.mock(BundleContext.class);
+        ServiceRegistration<ModuleFactory> serviceRegistration = mock(ServiceRegistration.class);
+        doNothing().when(serviceRegistration).unregister();
+        doReturn(serviceRegistration).when(bundleContext).registerService(
+                Matchers.any(String[].class), any(Closeable.class),
+                any(Dictionary.class));
+        return bundleContext;
     }
 
     @Override
-    public List<ModuleFactory> getAllFactories() {
-        return list;
+    public Map<String, Map.Entry<ModuleFactory, BundleContext>> getAllFactories() {
+        return factories;
     }
 
 }