Use akka-multi-dc in eos native

[controller.git] / opendaylight / md-sal / eos-dom-akka / src / test / java / org / opendaylight / controller / eos / akka / AbstractNativeEosTest.java

diff --git a/opendaylight/md-sal/eos-dom-akka/src/test/java/org/opendaylight/controller/eos/akka/AbstractNativeEosTest.java b/opendaylight/md-sal/eos-dom-akka/src/test/java/org/opendaylight/controller/eos/akka/AbstractNativeEosTest.java
index 8d10119c796acaa9db370af38f951e116c3f681b..201d6c659be329a1fa73e135cd1cce01327ffc73 100644 (file)
--- a/opendaylight/md-sal/eos-dom-akka/src/test/java/org/opendaylight/controller/eos/akka/AbstractNativeEosTest.java
+++ b/opendaylight/md-sal/eos-dom-akka/src/test/java/org/opendaylight/controller/eos/akka/AbstractNativeEosTest.java
@@ -37,6 +37,8 @@ import org.opendaylight.controller.eos.akka.bootstrap.EOSMain;
 import org.opendaylight.controller.eos.akka.bootstrap.command.BootstrapCommand;
 import org.opendaylight.controller.eos.akka.bootstrap.command.GetRunningContext;
 import org.opendaylight.controller.eos.akka.bootstrap.command.RunningContext;
+import org.opendaylight.controller.eos.akka.owner.supervisor.command.ActivateDataCenter;
+import org.opendaylight.controller.eos.akka.owner.supervisor.command.DeactivateDataCenter;
 import org.opendaylight.controller.eos.akka.owner.supervisor.command.MemberReachableEvent;
 import org.opendaylight.controller.eos.akka.owner.supervisor.command.MemberUnreachableEvent;
 import org.opendaylight.controller.eos.akka.owner.supervisor.command.OwnerSupervisorCommand;
@@ -56,6 +58,8 @@ public abstract class AbstractNativeEosTest {
     public static final DOMEntity ENTITY_1 = new DOMEntity("test-type", "entity-1");
     public static final DOMEntity ENTITY_2 = new DOMEntity("test-type-2", "entity-2");
 
+    protected static final String DEFAULT_DATACENTER = "dc-default";
+
     protected static final List<String> TWO_NODE_SEED_NODES =
             List.of("akka://ClusterSystem@127.0.0.1:2550",
                     "akka://ClusterSystem@127.0.0.1:2551");
@@ -65,10 +69,17 @@ public abstract class AbstractNativeEosTest {
                     "akka://ClusterSystem@127.0.0.1:2551",
                     "akka://ClusterSystem@127.0.0.1:2552");
 
+    protected static final List<String> DATACENTER_SEED_NODES =
+            List.of("akka://ClusterSystem@127.0.0.1:2550",
+                    "akka://ClusterSystem@127.0.0.1:2551",
+                    "akka://ClusterSystem@127.0.0.1:2552",
+                    "akka://ClusterSystem@127.0.0.1:2553");
+
     private static final String REMOTE_PROTOCOL = "akka";
     private static final String PORT_PARAM = "akka.remote.artery.canonical.port";
     private static final String ROLE_PARAM = "akka.cluster.roles";
     private static final String SEED_NODES_PARAM = "akka.cluster.seed-nodes";
+    private static final String DATA_CENTER_PARAM = "akka.cluster.multi-data-center.self-data-center";
 
 
     protected static ClusterNode startupRemote(final int port, final List<String> roles)
@@ -120,6 +131,35 @@ public abstract class AbstractNativeEosTest {
                 runningContext.getCandidateRegistry(), runningContext.getOwnerSupervisor());
     }
 
+    protected static ClusterNode startupWithDatacenter(final int port, final List<String> roles,
+                                                       final List<String> seedNodes, final String dataCenter)
+            throws ExecutionException, InterruptedException {
+        final Map<String, Object> overrides = new HashMap<>();
+        overrides.put(PORT_PARAM, port);
+        overrides.put(ROLE_PARAM, roles);
+        if (!seedNodes.isEmpty()) {
+            overrides.put(SEED_NODES_PARAM, seedNodes);
+        }
+        overrides.put(DATA_CENTER_PARAM, dataCenter);
+
+        final Config config = ConfigFactory.parseMap(overrides)
+                .withFallback(ConfigFactory.load());
+
+        // Create a classic Akka system since thats what we will have in osgi
+        final akka.actor.ActorSystem system = akka.actor.ActorSystem.create("ClusterSystem", config);
+        final ActorRef<BootstrapCommand> eosBootstrap =
+                Adapter.spawn(system, EOSMain.create(), "EOSBootstrap");
+
+        final CompletionStage<RunningContext> ask = AskPattern.ask(eosBootstrap,
+                GetRunningContext::new,
+                Duration.ofSeconds(5),
+                Adapter.toTyped(system.scheduler()));
+        final RunningContext runningContext = ask.toCompletableFuture().get();
+
+        return new ClusterNode(port, roles, system, eosBootstrap, runningContext.getListenerRegistry(),
+                runningContext.getCandidateRegistry(), runningContext.getOwnerSupervisor());
+    }
+
     private static Behavior<BootstrapCommand> rootBehavior() {
         return Behaviors.setup(context -> EOSMain.create());
     }
@@ -152,20 +192,22 @@ public abstract class AbstractNativeEosTest {
         return listener;
     }
 
-    protected static void reachableMember(final ClusterNode node, final String role) {
+    protected static void reachableMember(final ClusterNode node, final String... role) {
         reachableMember(node.getOwnerSupervisor(), role);
     }
 
-    protected static void reachableMember(final ActorRef<OwnerSupervisorCommand> ownerSupervisor, final String role) {
+    protected static void reachableMember(final ActorRef<OwnerSupervisorCommand> ownerSupervisor,
+                                          final String... role) {
         ownerSupervisor.tell(new MemberReachableEvent(
                 new Address(REMOTE_PROTOCOL, "ClusterSystem@127.0.0.1:2550"), Set.of(role)));
     }
 
-    protected static void unreachableMember(final ClusterNode node, final String role) {
+    protected static void unreachableMember(final ClusterNode node, final String... role) {
         unreachableMember(node.getOwnerSupervisor(), role);
     }
 
-    protected static void unreachableMember(final ActorRef<OwnerSupervisorCommand> ownerSupervisor, final String role) {
+    protected static void unreachableMember(final ActorRef<OwnerSupervisorCommand> ownerSupervisor,
+                                            final String... role) {
         ownerSupervisor.tell(new MemberUnreachableEvent(
                 new Address(REMOTE_PROTOCOL, "ClusterSystem@127.0.0.1:2550"), Set.of(role)));
     }
@@ -192,6 +234,14 @@ public abstract class AbstractNativeEosTest {
         });
     }
 
+    protected static void activateDatacenter(final ClusterNode clusterNode) {
+        clusterNode.getOwnerSupervisor().tell(ActivateDataCenter.INSTANCE);
+    }
+
+    protected static void deactivateDatacenter(final ClusterNode clusterNode) {
+        clusterNode.getOwnerSupervisor().tell(DeactivateDataCenter.INSTANCE);
+    }
+
     protected static void verifyListenerState(final MockEntityOwnershipListener listener, final DOMEntity entity,
                                               final boolean hasOwner, final boolean isOwner, final boolean wasOwner) {
         await().until(() -> !listener.getChanges().isEmpty());
@@ -267,13 +317,13 @@ public abstract class AbstractNativeEosTest {
 
     protected static final class MockEntityOwnershipListener implements DOMEntityOwnershipListener {
 
-        private final Logger log = LoggerFactory.getLogger(MockEntityOwnershipListener.class);
+        private final Logger log;
 
         private final List<DOMEntityOwnershipChange> changes = new ArrayList<>();
         private final String member;
 
         public MockEntityOwnershipListener(final String member) {
-
+            log = LoggerFactory.getLogger("EOS-listener-" + member);
             this.member = member;
         }