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