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