opendaylight/md-sal/sal-binding-broker/src/test/java/org/opendaylight/controller/sal/binding/test/bugfix/MultipleAugmentationPuts.java

   1 /*
   2  * Copyright (c) 2014 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.controller.sal.binding.test.bugfix;
   9
  10 import static org.junit.Assert.assertEquals;
  11 import static org.junit.Assert.assertNotNull;
  12 import static org.junit.Assert.assertNull;
  13
  14 import java.math.BigInteger;
  15 import java.util.ArrayList;
  16 import java.util.Collections;
  17 import java.util.List;
  18 import java.util.Map;
  19
  20 import org.junit.Test;
  21 import org.opendaylight.controller.md.sal.common.api.TransactionStatus;
  22 import org.opendaylight.controller.md.sal.common.api.data.DataChangeEvent;
  23 import org.opendaylight.controller.sal.binding.api.data.DataChangeListener;
  24 import org.opendaylight.controller.sal.binding.api.data.DataModificationTransaction;
  25 import org.opendaylight.controller.sal.binding.test.AbstractDataServiceTest;
  26 import org.opendaylight.controller.sal.binding.test.AugmentationVerifier;
  27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev100924.Counter32;
  28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev100924.Counter64;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNode;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNodeBuilder;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeBuilder;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.NodeMeterStatistics;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.NodeMeterStatisticsBuilder;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.nodes.node.meter.MeterStatisticsBuilder;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.MeterId;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.meter.statistics.DurationBuilder;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.meter.statistics.reply.MeterStats;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.meter.statistics.reply.MeterStatsBuilder;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.meter.statistics.reply.MeterStatsKey;
  44 import org.opendaylight.yangtools.yang.binding.Augmentation;
  45 import org.opendaylight.yangtools.yang.binding.DataObject;
  46 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  47 import org.opendaylight.yangtools.yang.common.QName;
  48 import org.opendaylight.yangtools.yang.common.RpcResult;
  49 import org.opendaylight.yangtools.yang.data.api.CompositeNode;
  50
  51 public class MultipleAugmentationPuts extends AbstractDataServiceTest implements DataChangeListener {
  52
  53     private static final QName NODE_ID_QNAME = QName.create(Node.QNAME, "id");
  54     private static final String NODE_ID = "openflow:1";
  55
  56     private static final NodeKey NODE_KEY = new NodeKey(new NodeId(NODE_ID));
  57
  58     private static final Map<QName, Object> NODE_KEY_BI = Collections.<QName, Object> singletonMap(NODE_ID_QNAME,
  59             NODE_ID);
  60
  61     private static final InstanceIdentifier<Nodes> NODES_INSTANCE_ID_BA = InstanceIdentifier.builder(Nodes.class) //
  62             .toInstance();
  63
  64     private static final InstanceIdentifier<Node> NODE_INSTANCE_ID_BA = InstanceIdentifier
  65             .builder(NODES_INSTANCE_ID_BA) //
  66             .child(Node.class, NODE_KEY).toInstance();
  67
  68     private static final org.opendaylight.yangtools.yang.data.api.InstanceIdentifier NODE_INSTANCE_ID_BI = //
  69     org.opendaylight.yangtools.yang.data.api.InstanceIdentifier.builder() //
  70             .node(Nodes.QNAME) //
  71             .nodeWithKey(Node.QNAME, NODE_KEY_BI) //
  72             .toInstance();
  73     private DataChangeEvent<InstanceIdentifier<?>, DataObject> receivedChangeEvent;
  74
  75     /**
  76      * Test for Bug 148
  77      *
  78      * @throws Exception
  79      */
  80     @Test
  81     public void testAugmentSerialization() throws Exception {
  82
  83         baDataService.registerDataChangeListener(NODES_INSTANCE_ID_BA, this);
  84
  85         Node flowCapableNode = createTestNode(FlowCapableNode.class, flowCapableNodeAugmentation());
  86         commitNodeAndVerifyTransaction(flowCapableNode);
  87
  88         assertNotNull(receivedChangeEvent);
  89         verifyNode((Nodes) receivedChangeEvent.getUpdatedOperationalSubtree(), flowCapableNode);
  90
  91         Nodes nodes = checkForNodes();
  92         verifyNode(nodes, flowCapableNode).assertHasAugmentation(FlowCapableNode.class);
  93         ;
  94         assertBindingIndependentVersion(NODE_INSTANCE_ID_BI);
  95 //        Node meterStatsNode = createTestNode(NodeMeterStatistics.class, nodeMeterStatistics());
  96 //        commitNodeAndVerifyTransaction(meterStatsNode);
  97 //
  98 //        assertNotNull(receivedChangeEvent);
  99 //        verifyNode((Nodes) receivedChangeEvent.getUpdatedOperationalSubtree(), meterStatsNode);
 100 //
 101 //        assertBindingIndependentVersion(NODE_INSTANCE_ID_BI);
 102 //
 103 //        Node mergedNode = (Node) baDataService.readOperationalData(NODE_INSTANCE_ID_BA);
 104 //
 105 //        AugmentationVerifier.from(mergedNode) //
 106 //                .assertHasAugmentation(FlowCapableNode.class) //
 107 //                .assertHasAugmentation(NodeMeterStatistics.class);
 108 //
 109 //        assertBindingIndependentVersion(NODE_INSTANCE_ID_BI);
 110 //
 111 //        Node meterStatsNodeWithDuration = createTestNode(NodeMeterStatistics.class, nodeMeterStatistics(5, true));
 112 //        commitNodeAndVerifyTransaction(meterStatsNodeWithDuration);
 113 //
 114 //
 115 //        Node nodeWithUpdatedList = (Node) baDataService.readOperationalData(NODE_INSTANCE_ID_BA);
 116 //        AugmentationVerifier.from(nodeWithUpdatedList) //
 117 //                .assertHasAugmentation(FlowCapableNode.class) //
 118 //                .assertHasAugmentation(NodeMeterStatistics.class);
 119 //
 120 //        List<MeterStats> meterStats = nodeWithUpdatedList.getAugmentation(NodeMeterStatistics.class).getMeterStatistics().getMeterStats();
 121 //        assertNotNull(meterStats);
 122 //        assertFalse(meterStats.isEmpty());
 123 //        assertBindingIndependentVersion(NODE_INSTANCE_ID_BI);
 124         testNodeRemove();
 125     }
 126
 127     private <T extends Augmentation<Node>> Node createTestNode(Class<T> augmentationClass, T augmentation) {
 128         NodeBuilder nodeBuilder = new NodeBuilder();
 129         nodeBuilder.setId(new NodeId(NODE_ID));
 130         nodeBuilder.setKey(NODE_KEY);
 131         nodeBuilder.addAugmentation(augmentationClass, augmentation);
 132         return nodeBuilder.build();
 133     }
 134
 135     private DataModificationTransaction commitNodeAndVerifyTransaction(Node original) throws Exception {
 136         DataModificationTransaction transaction = baDataService.beginTransaction();
 137         transaction.putOperationalData(NODE_INSTANCE_ID_BA, original);
 138         RpcResult<TransactionStatus> result = transaction.commit().get();
 139         assertEquals(TransactionStatus.COMMITED, result.getResult());
 140         return transaction;
 141     }
 142
 143     private void testNodeRemove() throws Exception {
 144         DataModificationTransaction transaction = baDataService.beginTransaction();
 145         transaction.removeOperationalData(NODE_INSTANCE_ID_BA);
 146         RpcResult<TransactionStatus> result = transaction.commit().get();
 147         assertEquals(TransactionStatus.COMMITED, result.getResult());
 148
 149         Node node = (Node) baDataService.readOperationalData(NODE_INSTANCE_ID_BA);
 150         assertNull(node);
 151     }
 152
 153     private AugmentationVerifier<Node> verifyNode(Nodes nodes, Node original) {
 154         assertNotNull(nodes);
 155         assertNotNull(nodes.getNode());
 156         assertEquals(1, nodes.getNode().size());
 157         Node readedNode = nodes.getNode().get(0);
 158         assertEquals(original.getId(), readedNode.getId());
 159         assertEquals(original.getKey(), readedNode.getKey());
 160         return new AugmentationVerifier<Node>(readedNode);
 161     }
 162
 163     private void assertBindingIndependentVersion(org.opendaylight.yangtools.yang.data.api.InstanceIdentifier nodeId) {
 164         CompositeNode node = biDataService.readOperationalData(nodeId);
 165         assertNotNull(node);
 166     }
 167
 168     private Nodes checkForNodes() {
 169         return (Nodes) baDataService.readOperationalData(NODES_INSTANCE_ID_BA);
 170     }
 171
 172     private NodeMeterStatistics nodeMeterStatistics() {
 173         return nodeMeterStatistics(10, false);
 174     }
 175
 176     private NodeMeterStatistics nodeMeterStatistics(int count, boolean setDuration) {
 177         NodeMeterStatisticsBuilder nmsb = new NodeMeterStatisticsBuilder();
 178         MeterStatisticsBuilder meterStats = new MeterStatisticsBuilder();
 179
 180         List<MeterStats> stats = new ArrayList<>(count);
 181         for (int i = 0; i <= count; i++) {
 182             MeterStatsBuilder statistic = new MeterStatsBuilder();
 183             statistic.setKey(new MeterStatsKey(new MeterId((long) i)));
 184             statistic.setByteInCount(new Counter64(BigInteger.valueOf(34590 + i)));
 185             statistic.setFlowCount(new Counter32(4569L + i));
 186
 187             if (setDuration) {
 188                 DurationBuilder duration = new DurationBuilder();
 189                 duration.setNanosecond(new Counter32(70L));
 190                 statistic.setDuration(duration.build());
 191             }
 192
 193             stats.add(statistic.build());
 194         }
 195        // meterStats.setMeterStats(stats);
 196         nmsb.setMeterStatistics(meterStats.build());
 197         return nmsb.build();
 198     }
 199
 200     private FlowCapableNode flowCapableNodeAugmentation() {
 201         FlowCapableNodeBuilder fnub = new FlowCapableNodeBuilder();
 202         fnub.setHardware("Hardware Foo");
 203         fnub.setManufacturer("Manufacturer Foo");
 204         fnub.setSerialNumber("Serial Foo");
 205         fnub.setDescription("Description Foo");
 206         fnub.setSoftware("JUnit emulated");
 207         FlowCapableNode fnu = fnub.build();
 208         return fnu;
 209     }
 210
 211     @Override
 212     public void onDataChanged(DataChangeEvent<InstanceIdentifier<?>, DataObject> change) {
 213         receivedChangeEvent = change;
 214     }
 215
 216 }