opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/utils/MockClusterWrapper.java

   1 /*
   2  * Copyright (c) 2014 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   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
   7  */
   8 package org.opendaylight.controller.cluster.datastore.utils;
   9
  10 import akka.actor.ActorRef;
  11 import akka.actor.Address;
  12 import akka.actor.AddressFromURIString;
  13 import akka.cluster.ClusterEvent.MemberRemoved;
  14 import akka.cluster.ClusterEvent.MemberUp;
  15 import akka.cluster.ClusterEvent.ReachableMember;
  16 import akka.cluster.ClusterEvent.UnreachableMember;
  17 import akka.cluster.Member;
  18 import akka.cluster.MemberStatus;
  19 import akka.cluster.UniqueAddress;
  20 import akka.util.Version;
  21 import org.opendaylight.controller.cluster.access.concepts.MemberName;
  22 import org.opendaylight.controller.cluster.datastore.ClusterWrapper;
  23 import scala.collection.immutable.Set.Set1;
  24
  25 public class MockClusterWrapper implements ClusterWrapper {
  26
  27     private Address selfAddress = new Address("akka", "test", "127.0.0.1", 2550);
  28     private final MemberName currentMemberName;
  29
  30     public MockClusterWrapper() {
  31         this("member-1");
  32     }
  33
  34     public MockClusterWrapper(final String currentMemberName) {
  35         this.currentMemberName = MemberName.forName(currentMemberName);
  36     }
  37
  38     @Override
  39     public void subscribeToMemberEvents(final ActorRef actorRef) {
  40     }
  41
  42     @Override
  43     public MemberName getCurrentMemberName() {
  44         return currentMemberName;
  45     }
  46
  47     @Override
  48     public Address getSelfAddress() {
  49         return selfAddress;
  50     }
  51
  52     public void setSelfAddress(final Address selfAddress) {
  53         this.selfAddress = selfAddress;
  54     }
  55
  56     public static void sendMemberUp(final ActorRef to, final String memberName, final String address) {
  57         to.tell(createMemberUp(memberName, address), null);
  58     }
  59
  60     public static void sendMemberRemoved(final ActorRef to, final String memberName, final String address) {
  61         to.tell(createMemberRemoved(memberName, address), null);
  62     }
  63
  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(), new Set1<>(memberName), Version.Zero());
  67
  68         return new MemberRemoved(member, MemberStatus.up());
  69     }
  70
  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(), new Set1<>(memberName), Version.Zero());
  74
  75         return new MemberUp(member);
  76     }
  77
  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(), new Set1<>(memberName), Version.Zero());
  81
  82         return new UnreachableMember(member);
  83     }
  84
  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(), new Set1<>(memberName), Version.Zero());
  88
  89         return new ReachableMember(member);
  90     }
  91 }