opendaylight/md-sal/sal-binding-dom-it/src/test/java/org/opendaylight/controller/sal/binding/test/bugfix/DeleteNestedAugmentationListenParentTest.java

   1 package org.opendaylight.controller.sal.binding.test.bugfix;
   2
   3 import static org.junit.Assert.assertFalse;
   4
   5 import java.util.concurrent.ExecutionException;
   6
   7 import org.junit.Test;
   8 import org.opendaylight.controller.md.sal.common.api.data.DataChangeEvent;
   9 import org.opendaylight.controller.sal.binding.api.data.DataChangeListener;
  10 import org.opendaylight.controller.sal.binding.api.data.DataModificationTransaction;
  11 import org.opendaylight.controller.sal.binding.test.AbstractDataServiceTest;
  12 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNode;
  13 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowId;
  14 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.Table;
  15 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.TableKey;
  16 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.table.Flow;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.table.FlowBuilder;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.table.FlowKey;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.FlowStatisticsData;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.FlowStatisticsDataBuilder;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.statistics.rev130819.flow.statistics.FlowStatisticsBuilder;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.flow.MatchBuilder;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
  27 import org.opendaylight.yangtools.concepts.ListenerRegistration;
  28 import org.opendaylight.yangtools.yang.binding.DataObject;
  29 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  30
  31 import com.google.common.util.concurrent.SettableFuture;
  32
  33 public class DeleteNestedAugmentationListenParentTest extends AbstractDataServiceTest {
  34
  35     private static final NodeKey NODE_KEY = new NodeKey(new NodeId("foo"));
  36
  37     private static final TableKey TABLE_KEY = new TableKey((short) 0);
  38
  39     private static final FlowKey FLOW_KEY = new FlowKey(new FlowId("100"));
  40
  41     private static final InstanceIdentifier<FlowCapableNode> LISTENER_PATH = InstanceIdentifier.builder(Nodes.class) //
  42             .child(Node.class)
  43             .augmentation(FlowCapableNode.class).build();
  44
  45
  46     private static final InstanceIdentifier<FlowCapableNode> NODE_AUGMENT_PATH = InstanceIdentifier.builder(Nodes.class)
  47             .child(Node.class,NODE_KEY)
  48             .augmentation(FlowCapableNode.class)
  49             .build();
  50
  51     private static final InstanceIdentifier<Flow> FLOW_PATH = NODE_AUGMENT_PATH.builder()
  52             .child(Table.class,TABLE_KEY)
  53             .child(Flow.class,FLOW_KEY)
  54             .build();
  55
  56
  57     @Test
  58     public void deleteChildListenParent() throws InterruptedException, ExecutionException {
  59         DataModificationTransaction initTx = baDataService.beginTransaction();
  60
  61         initTx.putOperationalData(FLOW_PATH, flow());
  62         initTx.commit().get();
  63
  64         final SettableFuture<DataChangeEvent<InstanceIdentifier<?>, DataObject>> event = SettableFuture.create();
  65
  66         ListenerRegistration<DataChangeListener> listenerReg = baDataService.registerDataChangeListener(FLOW_PATH, new DataChangeListener() {
  67
  68             @Override
  69             public void onDataChanged(final DataChangeEvent<InstanceIdentifier<?>, DataObject> change) {
  70                 event.set(change);
  71             }
  72         });
  73
  74         DataModificationTransaction deleteTx = baDataService.beginTransaction();
  75         deleteTx.removeOperationalData(FLOW_PATH.augmentation(FlowStatisticsData.class));
  76         deleteTx.commit().get();
  77
  78         DataChangeEvent<InstanceIdentifier<?>, DataObject> receivedEvent = event.get();
  79         assertFalse(receivedEvent.getRemovedOperationalData().contains(NODE_AUGMENT_PATH));
  80     }
  81
  82     private Flow flow() {
  83         FlowBuilder builder = new FlowBuilder()
  84             .setKey(FLOW_KEY)
  85             .addAugmentation(FlowStatisticsData.class,new FlowStatisticsDataBuilder()
  86                     .setFlowStatistics(new FlowStatisticsBuilder()
  87                         .setBarrier(true)
  88                         .setMatch(new MatchBuilder()
  89                             .build())
  90                         .build())
  91                     .build())
  92         ;//.build();
  93         return builder.build();
  94     }
  95
  96 }