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.MemberRemoved;
15 import akka.cluster.ClusterEvent.MemberUp;
16 import akka.cluster.ClusterEvent.ReachableMember;
17 import akka.cluster.ClusterEvent.UnreachableMember;
18 import akka.cluster.Member;
19 import akka.cluster.MemberStatus;
20 import akka.cluster.UniqueAddress;
21 import org.opendaylight.controller.cluster.access.concepts.MemberName;
22 import org.opendaylight.controller.cluster.datastore.ClusterWrapper;
23 import scala.collection.immutable.Set;
25 public class MockClusterWrapper implements ClusterWrapper {
27 private Address selfAddress = new Address("akka", "test", "127.0.0.1", 2550);
28 private final MemberName currentMemberName;
30 public MockClusterWrapper() {
34 public MockClusterWrapper(final String currentMemberName) {
35 this.currentMemberName = MemberName.forName(currentMemberName);
39 public void subscribeToMemberEvents(final ActorRef actorRef) {
43 public MemberName getCurrentMemberName() {
44 return currentMemberName;
48 public Address getSelfAddress() {
52 public void setSelfAddress(final Address selfAddress) {
53 this.selfAddress = selfAddress;
56 public static void sendMemberUp(final ActorRef to, final String memberName, final String address) {
57 to.tell(createMemberUp(memberName, address), null);
60 public static void sendMemberRemoved(final ActorRef to, final String memberName, final String address) {
61 to.tell(createMemberRemoved(memberName, address), null);
64 public static MemberRemoved createMemberRemoved(final String memberName, final String address) {
65 UniqueAddress uniqueAddress = new UniqueAddress(AddressFromURIString.parse(address), 55L);
66 Member member = new Member(uniqueAddress, 1, MemberStatus.removed(), setOf(memberName));
68 return new MemberRemoved(member, MemberStatus.up());
71 public static MemberUp createMemberUp(final String memberName, final String address) {
72 UniqueAddress uniqueAddress = new UniqueAddress(AddressFromURIString.parse(address), 55L);
73 Member member = new Member(uniqueAddress, 1, MemberStatus.up(), setOf(memberName));
75 return new MemberUp(member);
78 public static UnreachableMember createUnreachableMember(final String memberName, final String address) {
79 UniqueAddress uniqueAddress = new UniqueAddress(AddressFromURIString.parse(address), 55L);
80 Member member = new Member(uniqueAddress, 1, MemberStatus.up(), setOf(memberName));
82 return new UnreachableMember(member);
85 public static ReachableMember createReachableMember(final String memberName, final String address) {
86 UniqueAddress uniqueAddress = new UniqueAddress(AddressFromURIString.parse(address), 55L);
87 Member member = new Member(uniqueAddress, 1, MemberStatus.up(), setOf(memberName));
89 return new ReachableMember(member);
92 @SuppressWarnings("unchecked")
93 private static Set<String> setOf(final String str) {
94 return scala.collection.immutable.Set$.MODULE$.<String>newBuilder().$plus$eq(str).result();