- @Override public List<String> getMemberShardNames(String memberName) {
- List<String> shardNames = new ArrayList<>();
- shardNames.add("default");
+ private Map<String, List<String>> shardMembers = ImmutableMap.<String, List<String>>builder().
+ put("default", Arrays.asList("member-1", "member-2")).
+ /*put("astronauts", Arrays.asList("member-2", "member-3")).*/build();
+
+ public MockConfiguration() {
+ }
+
+ public MockConfiguration(Map<String, List<String>> shardMembers) {
+ this.shardMembers = shardMembers;
+ }
+
+ @Override
+ public Collection<String> getMemberShardNames(final String memberName) {
+ ArrayList<String> shardNames = new ArrayList<String>();
+ for(Map.Entry<String, List<String>> shard : shardMembers.entrySet()) {
+ if (shard.getValue().contains(memberName)) {
+ shardNames.add(shard.getKey());
+ }
+ }