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 public class DataChangeListenerRegistrationProxyTest extends AbstractActorTest{
  21
  22     private ActorRef dataChangeListenerActor = getSystem().actorOf(Props.create(DoNothingActor.class));
  23
  24     private static class MockDataChangeListener implements
  25         AsyncDataChangeListener<YangInstanceIdentifier, NormalizedNode<?, ?>> {
  26
  27         @Override public void onDataChanged(
  28             AsyncDataChangeEvent<YangInstanceIdentifier, NormalizedNode<?, ?>> change) {
  29             throw new UnsupportedOperationException("onDataChanged");
  30         }
  31     }
  32
  33     @Test
  34     public void testGetInstance() throws Exception {
  35         final Props props = Props.create(MessageCollectorActor.class);
  36         final ActorRef actorRef = getSystem().actorOf(props);
  37
  38         MockDataChangeListener listener =
  39             new MockDataChangeListener();
  40         DataChangeListenerRegistrationProxy proxy =
  41             new DataChangeListenerRegistrationProxy(
  42                 getSystem().actorSelection(actorRef.path()),
  43                 listener, dataChangeListenerActor);
  44
  45         Assert.assertEquals(listener, proxy.getInstance());
  46
  47     }
  48
  49     @Test
  50     public void testClose() throws Exception {
  51         final Props props = Props.create(MessageCollectorActor.class);
  52         final ActorRef actorRef = getSystem().actorOf(props);
  53
  54         DataChangeListenerRegistrationProxy proxy =
  55             new DataChangeListenerRegistrationProxy(
  56                 getSystem().actorSelection(actorRef.path()),
  57                 new MockDataChangeListener(), dataChangeListenerActor);
  58
  59         proxy.close();
  60
  61         //Check if it was received by the remote actor
  62         ActorContext
  63             testContext = new ActorContext(getSystem(), getSystem().actorOf(Props.create(DoNothingActor.class)),new MockClusterWrapper(), new MockConfiguration());
  64         Object messages = testContext
  65             .executeLocalOperation(actorRef, "messages",
  66                 ActorContext.ASK_DURATION);
  67
  68         Assert.assertNotNull(messages);
  69
  70         Assert.assertTrue(messages instanceof List);
  71
  72         List<Object> listMessages = (List<Object>) messages;
  73
  74         Assert.assertEquals(1, listMessages.size());
  75
  76         Assert.assertTrue(listMessages.get(0).getClass().equals(CloseDataChangeListenerRegistration.SERIALIZABLE_CLASS));
  77     }
  78 }