1 package org.opendaylight.controller.cluster.datastore;
3 import akka.actor.ActorRef;
4 import akka.actor.Props;
5 import junit.framework.Assert;
7 import org.opendaylight.controller.cluster.datastore.messages.CloseDataChangeListenerRegistration;
8 import org.opendaylight.controller.cluster.datastore.utils.ActorContext;
9 import org.opendaylight.controller.cluster.datastore.utils.DoNothingActor;
10 import org.opendaylight.controller.cluster.datastore.utils.MessageCollectorActor;
11 import org.opendaylight.controller.cluster.datastore.utils.MockClusterWrapper;
12 import org.opendaylight.controller.cluster.datastore.utils.MockConfiguration;
13 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
14 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeListener;
15 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
16 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
18 import java.util.List;
20 import static junit.framework.TestCase.assertEquals;
21 import static junit.framework.TestCase.assertNotNull;
22 import static junit.framework.TestCase.assertTrue;
24 public class DataChangeListenerRegistrationProxyTest extends AbstractActorTest{
26 private ActorRef dataChangeListenerActor = getSystem().actorOf(Props.create(DoNothingActor.class));
28 private static class MockDataChangeListener implements
29 AsyncDataChangeListener<YangInstanceIdentifier, NormalizedNode<?, ?>> {
31 @Override public void onDataChanged(
32 AsyncDataChangeEvent<YangInstanceIdentifier, NormalizedNode<?, ?>> change) {
33 throw new UnsupportedOperationException("onDataChanged");
38 public void testGetInstance() throws Exception {
39 final Props props = Props.create(MessageCollectorActor.class);
40 final ActorRef actorRef = getSystem().actorOf(props);
42 MockDataChangeListener listener =
43 new MockDataChangeListener();
44 DataChangeListenerRegistrationProxy proxy =
45 new DataChangeListenerRegistrationProxy(
46 getSystem().actorSelection(actorRef.path()),
47 listener, dataChangeListenerActor);
49 Assert.assertEquals(listener, proxy.getInstance());
54 public void testClose() throws Exception {
55 final Props props = Props.create(MessageCollectorActor.class);
56 final ActorRef actorRef = getSystem().actorOf(props);
58 DataChangeListenerRegistrationProxy proxy =
59 new DataChangeListenerRegistrationProxy(
60 getSystem().actorSelection(actorRef.path()),
61 new MockDataChangeListener(), dataChangeListenerActor);
65 //Check if it was received by the remote actor
67 testContext = new ActorContext(getSystem(), getSystem().actorOf(Props.create(DoNothingActor.class)),new MockClusterWrapper(), new MockConfiguration());
68 Object messages = testContext
69 .executeLocalOperation(actorRef, "messages");
71 assertNotNull(messages);
73 assertTrue(messages instanceof List);
75 List<Object> listMessages = (List<Object>) messages;
77 assertEquals(1, listMessages.size());
79 assertTrue(listMessages.get(0).getClass()
80 .equals(CloseDataChangeListenerRegistration.SERIALIZABLE_CLASS));
84 public void testCloseWhenRegistrationIsNull() throws Exception {
85 final Props props = Props.create(MessageCollectorActor.class);
86 final ActorRef actorRef = getSystem().actorOf(props);
88 DataChangeListenerRegistrationProxy proxy =
89 new DataChangeListenerRegistrationProxy(
90 new MockDataChangeListener(), dataChangeListenerActor);
94 //Check if it was received by the remote actor
96 testContext = new ActorContext(getSystem(), getSystem().actorOf(Props.create(DoNothingActor.class)),new MockClusterWrapper(), new MockConfiguration());
97 Object messages = testContext
98 .executeLocalOperation(actorRef, "messages");
100 assertNotNull(messages);
102 assertTrue(messages instanceof List);
104 List<Object> listMessages = (List<Object>) messages;
106 assertEquals(0, listMessages.size());