-import java.util.Set;
-import org.opendaylight.controller.cluster.datastore.Configuration;
-import org.opendaylight.controller.cluster.datastore.shardstrategy.ShardStrategy;
-
-public class MockConfiguration implements Configuration{
- @Override public List<String> getMemberShardNames(final String memberName) {
- return Arrays.asList("default");
- }
-
- @Override public Optional<String> getModuleNameFromNameSpace(
- final String nameSpace) {
- return Optional.absent();
- }
-
- @Override
- public Map<String, ShardStrategy> getModuleNameToShardStrategyMap() {
- return Collections.emptyMap();
- }
-
- @Override public List<String> getShardNamesFromModuleName(
- final String moduleName) {
- return Collections.emptyList();
- }
-
- @Override public List<String> getMembersFromShardName(final String shardName) {
- if("default".equals(shardName)) {
- return Arrays.asList("member-1", "member-2");
- } else if("astronauts".equals(shardName)){
- return Arrays.asList("member-2", "member-3");
- }
-
- return Collections.emptyList();
+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")));