1 package org.opendaylight.controller.sal.binding.test.bugfix;
3 import static org.junit.Assert.assertEquals;
4 import static org.junit.Assert.assertNotNull;
5 import static org.junit.Assert.assertNull;
7 import java.math.BigInteger;
8 import java.util.ArrayList;
9 import java.util.Collections;
10 import java.util.List;
13 import org.junit.Test;
14 import org.opendaylight.controller.md.sal.common.api.TransactionStatus;
15 import org.opendaylight.controller.md.sal.common.api.data.DataChangeEvent;
16 import org.opendaylight.controller.sal.binding.api.data.DataChangeListener;
17 import org.opendaylight.controller.sal.binding.api.data.DataModificationTransaction;
18 import org.opendaylight.controller.sal.binding.test.AbstractDataServiceTest;
19 import org.opendaylight.controller.sal.binding.test.AugmentationVerifier;
20 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev100924.Counter32;
21 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev100924.Counter64;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNode;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNodeBuilder;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeBuilder;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.NodeMeterStatistics;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.NodeMeterStatisticsBuilder;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.nodes.node.meter.MeterStatisticsBuilder;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.MeterId;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.meter.statistics.DurationBuilder;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.meter.statistics.reply.MeterStats;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.meter.statistics.reply.MeterStatsBuilder;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.meter.statistics.reply.MeterStatsKey;
37 import org.opendaylight.yangtools.yang.binding.Augmentation;
38 import org.opendaylight.yangtools.yang.binding.DataObject;
39 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
40 import org.opendaylight.yangtools.yang.common.QName;
41 import org.opendaylight.yangtools.yang.common.RpcResult;
42 import org.opendaylight.yangtools.yang.data.api.CompositeNode;
44 public class MultipleAugmentationPuts extends AbstractDataServiceTest implements DataChangeListener {
46 private static final QName NODE_ID_QNAME = QName.create(Node.QNAME, "id");
47 private static final String NODE_ID = "openflow:1";
49 private static final NodeKey NODE_KEY = new NodeKey(new NodeId(NODE_ID));
51 private static final Map<QName, Object> NODE_KEY_BI = Collections.<QName, Object> singletonMap(NODE_ID_QNAME,
54 private static final InstanceIdentifier<Nodes> NODES_INSTANCE_ID_BA = InstanceIdentifier.builder(Nodes.class) //
57 private static final InstanceIdentifier<Node> NODE_INSTANCE_ID_BA = InstanceIdentifier
58 .builder(NODES_INSTANCE_ID_BA) //
59 .child(Node.class, NODE_KEY).toInstance();
61 private static final org.opendaylight.yangtools.yang.data.api.InstanceIdentifier NODE_INSTANCE_ID_BI = //
62 org.opendaylight.yangtools.yang.data.api.InstanceIdentifier.builder() //
64 .nodeWithKey(Node.QNAME, NODE_KEY_BI) //
66 private DataChangeEvent<InstanceIdentifier<?>, DataObject> receivedChangeEvent;
74 public void testAugmentSerialization() throws Exception {
76 baDataService.registerDataChangeListener(NODES_INSTANCE_ID_BA, this);
78 Node flowCapableNode = createTestNode(FlowCapableNode.class, flowCapableNodeAugmentation());
79 commitNodeAndVerifyTransaction(flowCapableNode);
81 assertNotNull(receivedChangeEvent);
82 verifyNode((Nodes) receivedChangeEvent.getUpdatedOperationalSubtree(), flowCapableNode);
84 Nodes nodes = checkForNodes();
85 verifyNode(nodes, flowCapableNode).assertHasAugmentation(FlowCapableNode.class);
87 assertBindingIndependentVersion(NODE_INSTANCE_ID_BI);
88 // Node meterStatsNode = createTestNode(NodeMeterStatistics.class, nodeMeterStatistics());
89 // commitNodeAndVerifyTransaction(meterStatsNode);
91 // assertNotNull(receivedChangeEvent);
92 // verifyNode((Nodes) receivedChangeEvent.getUpdatedOperationalSubtree(), meterStatsNode);
94 // assertBindingIndependentVersion(NODE_INSTANCE_ID_BI);
96 // Node mergedNode = (Node) baDataService.readOperationalData(NODE_INSTANCE_ID_BA);
98 // AugmentationVerifier.from(mergedNode) //
99 // .assertHasAugmentation(FlowCapableNode.class) //
100 // .assertHasAugmentation(NodeMeterStatistics.class);
102 // assertBindingIndependentVersion(NODE_INSTANCE_ID_BI);
104 // Node meterStatsNodeWithDuration = createTestNode(NodeMeterStatistics.class, nodeMeterStatistics(5, true));
105 // commitNodeAndVerifyTransaction(meterStatsNodeWithDuration);
108 // Node nodeWithUpdatedList = (Node) baDataService.readOperationalData(NODE_INSTANCE_ID_BA);
109 // AugmentationVerifier.from(nodeWithUpdatedList) //
110 // .assertHasAugmentation(FlowCapableNode.class) //
111 // .assertHasAugmentation(NodeMeterStatistics.class);
113 // List<MeterStats> meterStats = nodeWithUpdatedList.getAugmentation(NodeMeterStatistics.class).getMeterStatistics().getMeterStats();
114 // assertNotNull(meterStats);
115 // assertFalse(meterStats.isEmpty());
116 // assertBindingIndependentVersion(NODE_INSTANCE_ID_BI);
120 private <T extends Augmentation<Node>> Node createTestNode(Class<T> augmentationClass, T augmentation) {
121 NodeBuilder nodeBuilder = new NodeBuilder();
122 nodeBuilder.setId(new NodeId(NODE_ID));
123 nodeBuilder.setKey(NODE_KEY);
124 nodeBuilder.addAugmentation(augmentationClass, augmentation);
125 return nodeBuilder.build();
128 private DataModificationTransaction commitNodeAndVerifyTransaction(Node original) throws Exception {
129 DataModificationTransaction transaction = baDataService.beginTransaction();
130 transaction.putOperationalData(NODE_INSTANCE_ID_BA, original);
131 RpcResult<TransactionStatus> result = transaction.commit().get();
132 assertEquals(TransactionStatus.COMMITED, result.getResult());
136 private void testNodeRemove() throws Exception {
137 DataModificationTransaction transaction = baDataService.beginTransaction();
138 transaction.removeOperationalData(NODE_INSTANCE_ID_BA);
139 RpcResult<TransactionStatus> result = transaction.commit().get();
140 assertEquals(TransactionStatus.COMMITED, result.getResult());
142 Node node = (Node) baDataService.readOperationalData(NODE_INSTANCE_ID_BA);
146 private AugmentationVerifier<Node> verifyNode(Nodes nodes, Node original) {
147 assertNotNull(nodes);
148 assertNotNull(nodes.getNode());
149 assertEquals(1, nodes.getNode().size());
150 Node readedNode = nodes.getNode().get(0);
151 assertEquals(original.getId(), readedNode.getId());
152 assertEquals(original.getKey(), readedNode.getKey());
153 return new AugmentationVerifier<Node>(readedNode);
156 private void assertBindingIndependentVersion(org.opendaylight.yangtools.yang.data.api.InstanceIdentifier nodeId) {
157 CompositeNode node = biDataService.readOperationalData(nodeId);
161 private Nodes checkForNodes() {
162 return (Nodes) baDataService.readOperationalData(NODES_INSTANCE_ID_BA);
165 private NodeMeterStatistics nodeMeterStatistics() {
166 return nodeMeterStatistics(10, false);
169 private NodeMeterStatistics nodeMeterStatistics(int count, boolean setDuration) {
170 NodeMeterStatisticsBuilder nmsb = new NodeMeterStatisticsBuilder();
171 MeterStatisticsBuilder meterStats = new MeterStatisticsBuilder();
173 List<MeterStats> stats = new ArrayList<>(count);
174 for (int i = 0; i <= count; i++) {
175 MeterStatsBuilder statistic = new MeterStatsBuilder();
176 statistic.setKey(new MeterStatsKey(new MeterId((long) i)));
177 statistic.setByteInCount(new Counter64(BigInteger.valueOf(34590 + i)));
178 statistic.setFlowCount(new Counter32(4569L + i));
181 DurationBuilder duration = new DurationBuilder();
182 duration.setNanosecond(new Counter32(70L));
183 statistic.setDuration(duration.build());
186 stats.add(statistic.build());
188 // meterStats.setMeterStats(stats);
189 nmsb.setMeterStatistics(meterStats.build());
193 private FlowCapableNode flowCapableNodeAugmentation() {
194 FlowCapableNodeBuilder fnub = new FlowCapableNodeBuilder();
195 fnub.setHardware("Hardware Foo");
196 fnub.setManufacturer("Manufacturer Foo");
197 fnub.setSerialNumber("Serial Foo");
198 fnub.setDescription("Description Foo");
199 fnub.setSoftware("JUnit emulated");
200 FlowCapableNode fnu = fnub.build();
205 public void onDataChanged(DataChangeEvent<InstanceIdentifier<?>, DataObject> change) {
206 receivedChangeEvent = change;