2 * Copyright (c) 2014 Cisco Systems, Inc. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
9 package org.opendaylight.controller.cluster.datastore.utils;
11 import akka.actor.ActorRef;
12 import akka.actor.Address;
13 import akka.actor.AddressFromURIString;
14 import akka.cluster.ClusterEvent;
15 import akka.cluster.MemberStatus;
16 import akka.cluster.UniqueAddress;
17 import java.util.HashSet;
19 import org.opendaylight.controller.cluster.datastore.ClusterWrapper;
20 import scala.collection.JavaConversions;
22 public class MockClusterWrapper implements ClusterWrapper{
24 private Address selfAddress = new Address("akka.tcp", "test", "127.0.0.1", 2550);
25 private String currentMemberName = "member-1";
27 public MockClusterWrapper() {
30 public MockClusterWrapper(String currentMemberName) {
31 this.currentMemberName = currentMemberName;
35 public void subscribeToMemberEvents(ActorRef actorRef) {
39 public String getCurrentMemberName() {
40 return currentMemberName;
44 public Address getSelfAddress() {
48 public void setSelfAddress(Address selfAddress) {
49 this.selfAddress = selfAddress;
52 public static void sendMemberUp(ActorRef to, String memberName, String address){
53 to.tell(createMemberUp(memberName, address), null);
56 public static void sendMemberRemoved(ActorRef to, String memberName, String address){
57 to.tell(createMemberRemoved(memberName, address), null);
60 private static ClusterEvent.MemberRemoved createMemberRemoved(String memberName, String address) {
61 akka.cluster.UniqueAddress uniqueAddress = new UniqueAddress(
62 AddressFromURIString.parse(address), 55);
64 Set<String> roles = new HashSet<>();
66 roles.add(memberName);
68 akka.cluster.Member member = new akka.cluster.Member(uniqueAddress, 1, MemberStatus
70 JavaConversions.asScalaSet(roles).<String>toSet());
72 return new ClusterEvent.MemberRemoved(member, MemberStatus.up());
77 public static ClusterEvent.MemberUp createMemberUp(String memberName, String address) {
78 akka.cluster.UniqueAddress uniqueAddress = new UniqueAddress(
79 AddressFromURIString.parse(address), 55);
81 Set<String> roles = new HashSet<>();
83 roles.add(memberName);
85 akka.cluster.Member member = new akka.cluster.Member(uniqueAddress, 1, MemberStatus.up(),
86 JavaConversions.asScalaSet(roles).<String>toSet());
88 return new ClusterEvent.MemberUp(member);
91 public static ClusterEvent.UnreachableMember createUnreachableMember(String memberName, String address) {
92 akka.cluster.UniqueAddress uniqueAddress = new UniqueAddress(
93 AddressFromURIString.parse(address), 55);
95 Set<String> roles = new HashSet<>();
97 roles.add(memberName);
99 akka.cluster.Member member = new akka.cluster.Member(uniqueAddress, 1, MemberStatus.up(),
100 JavaConversions.asScalaSet(roles).<String>toSet());
102 return new ClusterEvent.UnreachableMember(member);
105 public static ClusterEvent.ReachableMember createReachableMember(String memberName, String address) {
106 akka.cluster.UniqueAddress uniqueAddress = new UniqueAddress(
107 AddressFromURIString.parse(address), 55);
109 Set<String> roles = new HashSet<>();
111 roles.add(memberName);
113 akka.cluster.Member member = new akka.cluster.Member(uniqueAddress, 1, MemberStatus.up(),
114 JavaConversions.asScalaSet(roles).<String>toSet());
116 return new ClusterEvent.ReachableMember(member);