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