Refactor MockConfiguration to extend ConfigurationImpl

[controller.git] / opendaylight / md-sal / sal-distributed-datastore / src / test / java / org / opendaylight / controller / cluster / datastore / utils / MockConfiguration.java

diff --git a/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/utils/MockConfiguration.java b/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/utils/MockConfiguration.java
index a8877c7a9fa6e1c11d7fc344bb426452f5a703c0..77097352489fa97a153761bb16817fcdd9fd6521 100644 (file)
--- a/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/utils/MockConfiguration.java
+++ b/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/utils/MockConfiguration.java
@@ -8,15 +8,36 @@
 
 package org.opendaylight.controller.cluster.datastore.utils;
 
-import org.opendaylight.controller.cluster.datastore.Configuration;
-
-import java.util.ArrayList;
+import com.google.common.collect.Sets;
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
+import org.opendaylight.controller.cluster.datastore.config.Configuration;
+import org.opendaylight.controller.cluster.datastore.config.ConfigurationImpl;
+import org.opendaylight.controller.cluster.datastore.config.ModuleConfig;
+import org.opendaylight.controller.cluster.datastore.config.ModuleShardConfigProvider;
+
+public class MockConfiguration extends ConfigurationImpl {
+    public MockConfiguration() {
+        this(Collections.singletonMap("default", Arrays.asList("member-1", "member-2")));
+    }
+
+    public MockConfiguration(final Map<String, List<String>> shardMembers) {
+        super(new ModuleShardConfigProvider() {
+            @Override
+            public Map<String, ModuleConfig> retrieveModuleConfigs(Configuration configuration) {
+                Map<String, ModuleConfig> retMap = new HashMap<String, ModuleConfig>();
+                for(Map.Entry<String, List<String>> e : shardMembers.entrySet()) {
+                    String shardName = e.getKey();
+                    ModuleConfig mc = new ModuleConfig(shardName);
+                    mc.addShardConfig(shardName, Sets.newHashSet(e.getValue()));
+                    retMap.put(mc.getName(), mc);
+                }
 
-public class MockConfiguration implements Configuration{
-    @Override public List<String> getMemberShardNames(String memberName) {
-        List<String> shardNames = new ArrayList<>();
-        shardNames.add("default");
-        return shardNames;
+                return retMap;
+            }
+        });
     }
 }