Bug 2194: Modify FindPrimary to check for leader
[controller.git] / opendaylight / md-sal / sal-distributed-datastore / src / test / java / org / opendaylight / controller / cluster / datastore / utils / MessageCollectorActor.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.UntypedActor;
13 import akka.pattern.Patterns;
14 import akka.util.Timeout;
15 import com.google.common.collect.Lists;
16 import com.google.common.util.concurrent.Uninterruptibles;
17 import java.util.ArrayList;
18 import java.util.List;
19 import java.util.concurrent.TimeUnit;
20 import org.junit.Assert;
21 import scala.concurrent.Await;
22 import scala.concurrent.Future;
23 import scala.concurrent.duration.Duration;
24 import scala.concurrent.duration.FiniteDuration;
25
26 /**
27  * MessageCollectorActor collects messages as it receives them. It can send
28  * those collected messages to any sender which sends it the "messages" message
29  * <p>
30  *     This class would be useful as a mock to test whether messages were sent
31  *     to a remote actor or not.
32  * </p>
33  */
34 public class MessageCollectorActor extends UntypedActor {
35     private final List<Object> messages = new ArrayList<>();
36
37     @Override public void onReceive(Object message) throws Exception {
38         if(message instanceof String){
39             if("messages".equals(message)){
40                 getSender().tell(new ArrayList(messages), getSelf());
41             }
42         } else {
43             messages.add(message);
44         }
45     }
46
47     public void clear() {
48         messages.clear();
49     }
50
51     public static List<Object> getAllMessages(ActorRef actor) throws Exception {
52         FiniteDuration operationDuration = Duration.create(5, TimeUnit.SECONDS);
53         Timeout operationTimeout = new Timeout(operationDuration);
54         Future<Object> future = Patterns.ask(actor, "messages", operationTimeout);
55
56         try {
57             return (List<Object>) Await.result(future, operationDuration);
58         } catch (Exception e) {
59             throw e;
60         }
61     }
62
63     /**
64      * Get the first message that matches the specified class
65      * @param actor
66      * @param clazz
67      * @return
68      */
69     public static Object getFirstMatching(ActorRef actor, Class<?> clazz) throws Exception {
70         List<Object> allMessages = getAllMessages(actor);
71
72         for(Object message : allMessages){
73             if(message.getClass().equals(clazz)){
74                 return message;
75             }
76         }
77
78         return null;
79     }
80
81     public static List<Object> getAllMatching(ActorRef actor, Class<?> clazz) throws Exception {
82         List<Object> allMessages = getAllMessages(actor);
83
84         List<Object> output = Lists.newArrayList();
85
86         for(Object message : allMessages){
87             if(message.getClass().equals(clazz)){
88                 output.add(message);
89             }
90         }
91
92         return output;
93     }
94
95     public static <T> T expectFirstMatching(ActorRef actor, Class<T> clazz) {
96         int count = 5000 / 50;
97         for(int i = 0; i < count; i++) {
98             try {
99                 T message = (T) getFirstMatching(actor, clazz);
100                 if(message != null) {
101                     return message;
102                 }
103             } catch (Exception e) {}
104
105             Uninterruptibles.sleepUninterruptibly(50, TimeUnit.MILLISECONDS);
106         }
107
108         Assert.fail("Did not receive message of type " + clazz);
109         return null;
110     }
111 }

©2013 OpenDaylight, A Linux Foundation Collaborative Project. All Rights Reserved.
OpenDaylight is a registered trademark of The OpenDaylight Project, Inc.
Linux Foundation and OpenDaylight are registered trademarks of the Linux Foundation.
Linux is a registered trademark of Linus Torvalds.