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

   1
   2 package org.opendaylight.controller.cluster.datastore;
   3
   4 import akka.actor.ActorRef;
   5 import akka.actor.Props;
   6 import java.util.HashMap;
   7 import java.util.HashSet;
   8 import java.util.List;
   9 import java.util.Map;
  10 import java.util.Set;
  11 import org.junit.Assert;
  12 import org.junit.Test;
  13 import org.opendaylight.controller.cluster.datastore.messages.DataChanged;
  14 import org.opendaylight.controller.cluster.datastore.utils.ActorContext;
  15 import org.opendaylight.controller.cluster.datastore.utils.DoNothingActor;
  16 import org.opendaylight.controller.cluster.datastore.utils.MessageCollectorActor;
  17 import org.opendaylight.controller.cluster.datastore.utils.MockClusterWrapper;
  18 import org.opendaylight.controller.cluster.datastore.utils.MockConfiguration;
  19 import org.opendaylight.controller.md.cluster.datastore.model.CompositeModel;
  20 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
  21 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  22 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  23 public class DataChangeListenerProxyTest extends AbstractActorTest {
  24
  25   private static class MockDataChangedEvent implements AsyncDataChangeEvent<YangInstanceIdentifier, NormalizedNode<?, ?>> {
  26     Map<YangInstanceIdentifier,NormalizedNode<?,?>> createdData = new HashMap<>();
  27     Map<YangInstanceIdentifier,NormalizedNode<?,?>> updatedData = new HashMap<>();
  28     Map<YangInstanceIdentifier,NormalizedNode<?,?>> originalData = new HashMap<>();
  29
  30
  31
  32     @Override
  33     public Map<YangInstanceIdentifier, NormalizedNode<?, ?>> getCreatedData() {
  34       createdData.put(YangInstanceIdentifier.builder().build(), CompositeModel.createDocumentOne(CompositeModel.createTestContext()));
  35       return createdData;
  36     }
  37
  38     @Override
  39     public Map<YangInstanceIdentifier, NormalizedNode<?, ?>> getUpdatedData() {
  40       updatedData.put(YangInstanceIdentifier.builder().build(), CompositeModel.createTestContainer());
  41       return updatedData;
  42
  43     }
  44
  45     @Override
  46     public Set<YangInstanceIdentifier> getRemovedPaths() {
  47       Set<YangInstanceIdentifier>ids = new HashSet<>();
  48       ids.add( CompositeModel.TEST_PATH);
  49       return ids;
  50     }
  51
  52     @Override
  53     public Map<YangInstanceIdentifier, NormalizedNode<?, ?>> getOriginalData() {
  54       originalData.put(YangInstanceIdentifier.builder().build(), CompositeModel.createFamily());
  55       return originalData;
  56     }
  57
  58     @Override public NormalizedNode<?, ?> getOriginalSubtree() {
  59       return CompositeModel.createFamily() ;
  60     }
  61
  62     @Override public NormalizedNode<?, ?> getUpdatedSubtree() {
  63       return CompositeModel.createTestContainer();
  64     }
  65   }
  66
  67
  68   @Test
  69     public void testOnDataChanged() throws Exception {
  70         final Props props = Props.create(MessageCollectorActor.class);
  71         final ActorRef actorRef = getSystem().actorOf(props);
  72
  73         DataChangeListenerProxy dataChangeListenerProxy = new DataChangeListenerProxy(
  74                 getSystem().actorSelection(actorRef.path()));
  75
  76         dataChangeListenerProxy.onDataChanged(new MockDataChangedEvent());
  77
  78         //Check if it was received by the remote actor
  79         ActorContext
  80             testContext = new ActorContext(getSystem(), getSystem().actorOf(Props.create(DoNothingActor.class)), new MockClusterWrapper(), new MockConfiguration());
  81         Object messages = testContext
  82             .executeOperation(actorRef, "messages");
  83
  84         Assert.assertNotNull(messages);
  85
  86         Assert.assertTrue(messages instanceof List);
  87
  88         List<?> listMessages = (List<?>) messages;
  89
  90         Assert.assertEquals(1, listMessages.size());
  91
  92         Assert.assertTrue(listMessages.get(0).getClass().equals(DataChanged.class));
  93
  94     }
  95 }