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
   9 package org.opendaylight.controller.cluster.datastore.utils;
  10
  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 org.opendaylight.controller.cluster.datastore.ClusterWrapper;
  18 import scala.collection.JavaConversions;
  19 import java.util.HashSet;
  20 import java.util.Set;
  21
  22 public class MockClusterWrapper implements ClusterWrapper{
  23
  24     private Address selfAddress = new Address("akka.tcp", "test", "127.0.0.1", 2550);
  25
  26     @Override
  27     public void subscribeToMemberEvents(ActorRef actorRef) {
  28     }
  29
  30     @Override
  31     public String getCurrentMemberName() {
  32         return "member-1";
  33     }
  34
  35     @Override
  36     public Address getSelfAddress() {
  37         return selfAddress;
  38     }
  39
  40     public void setSelfAddress(Address selfAddress) {
  41         this.selfAddress = selfAddress;
  42     }
  43
  44     public static void sendMemberUp(ActorRef to, String memberName, String address){
  45         to.tell(createMemberUp(memberName, address), null);
  46     }
  47
  48     public static void sendMemberRemoved(ActorRef to, String memberName, String address){
  49         to.tell(createMemberRemoved(memberName, address), null);
  50     }
  51
  52     private static ClusterEvent.MemberRemoved createMemberRemoved(String memberName, String address) {
  53         akka.cluster.UniqueAddress uniqueAddress = new UniqueAddress(
  54             AddressFromURIString.parse(address), 55);
  55
  56         Set<String> roles = new HashSet<>();
  57
  58         roles.add(memberName);
  59
  60         akka.cluster.Member member = new akka.cluster.Member(uniqueAddress, 1, MemberStatus
  61             .removed(),
  62             JavaConversions.asScalaSet(roles).<String>toSet());
  63
  64         return new ClusterEvent.MemberRemoved(member, MemberStatus.up());
  65
  66     }
  67
  68
  69     private static ClusterEvent.MemberUp createMemberUp(String memberName, String address) {
  70         akka.cluster.UniqueAddress uniqueAddress = new UniqueAddress(
  71             AddressFromURIString.parse(address), 55);
  72
  73         Set<String> roles = new HashSet<>();
  74
  75         roles.add(memberName);
  76
  77         akka.cluster.Member member = new akka.cluster.Member(uniqueAddress, 1, MemberStatus.up(),
  78             JavaConversions.asScalaSet(roles).<String>toSet());
  79
  80         return new ClusterEvent.MemberUp(member);
  81     }
  82 }