opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/DataChangeListenerRegistrationProxyTest.java

   1 package org.opendaylight.controller.cluster.datastore;
   2
   3 import akka.actor.ActorRef;
   4 import akka.actor.Props;
   5 import junit.framework.Assert;
   6 import org.junit.Test;
   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;
  17
  18 import java.util.List;
  19
  20 import static junit.framework.TestCase.assertEquals;
  21 import static junit.framework.TestCase.assertNotNull;
  22 import static junit.framework.TestCase.assertTrue;
  23
  24 public class DataChangeListenerRegistrationProxyTest extends AbstractActorTest{
  25
  26     private ActorRef dataChangeListenerActor = getSystem().actorOf(Props.create(DoNothingActor.class));
  27
  28     private static class MockDataChangeListener implements
  29         AsyncDataChangeListener<YangInstanceIdentifier, NormalizedNode<?, ?>> {
  30
  31         @Override public void onDataChanged(
  32             AsyncDataChangeEvent<YangInstanceIdentifier, NormalizedNode<?, ?>> change) {
  33             throw new UnsupportedOperationException("onDataChanged");
  34         }
  35     }
  36
  37     @Test
  38     public void testGetInstance() throws Exception {
  39         final Props props = Props.create(MessageCollectorActor.class);
  40         final ActorRef actorRef = getSystem().actorOf(props);
  41
  42         MockDataChangeListener listener =
  43             new MockDataChangeListener();
  44         DataChangeListenerRegistrationProxy proxy =
  45             new DataChangeListenerRegistrationProxy(
  46                 getSystem().actorSelection(actorRef.path()),
  47                 listener, dataChangeListenerActor);
  48
  49         Assert.assertEquals(listener, proxy.getInstance());
  50
  51     }
  52
  53     @Test
  54     public void testClose() throws Exception {
  55         final Props props = Props.create(MessageCollectorActor.class);
  56         final ActorRef actorRef = getSystem().actorOf(props);
  57
  58         DataChangeListenerRegistrationProxy proxy =
  59             new DataChangeListenerRegistrationProxy(
  60                 getSystem().actorSelection(actorRef.path()),
  61                 new MockDataChangeListener(), dataChangeListenerActor);
  62
  63         proxy.close();
  64
  65         //Check if it was received by the remote actor
  66         ActorContext
  67             testContext = new ActorContext(getSystem(), getSystem().actorOf(Props.create(DoNothingActor.class)),new MockClusterWrapper(), new MockConfiguration());
  68         Object messages = testContext
  69             .executeOperation(actorRef, "messages");
  70
  71         assertNotNull(messages);
  72
  73         assertTrue(messages instanceof List);
  74
  75         List<Object> listMessages = (List<Object>) messages;
  76
  77         assertEquals(1, listMessages.size());
  78
  79         assertTrue(listMessages.get(0).getClass()
  80             .equals(CloseDataChangeListenerRegistration.SERIALIZABLE_CLASS));
  81     }
  82
  83     @Test
  84     public void testCloseWhenRegistrationIsNull() throws Exception {
  85         final Props props = Props.create(MessageCollectorActor.class);
  86         final ActorRef actorRef = getSystem().actorOf(props);
  87
  88         DataChangeListenerRegistrationProxy proxy =
  89             new DataChangeListenerRegistrationProxy(
  90                 new MockDataChangeListener(), dataChangeListenerActor);
  91
  92         proxy.close();
  93
  94         //Check if it was received by the remote actor
  95         ActorContext
  96             testContext = new ActorContext(getSystem(), getSystem().actorOf(Props.create(DoNothingActor.class)),new MockClusterWrapper(), new MockConfiguration());
  97         Object messages = testContext
  98             .executeOperation(actorRef, "messages");
  99
 100         assertNotNull(messages);
 101
 102         assertTrue(messages instanceof List);
 103
 104         List<Object> listMessages = (List<Object>) messages;
 105
 106         assertEquals(0, listMessages.size());
 107     }
 108 }