1 package org.opendaylight.controller.test.sal.binding.it;
3 import static org.junit.Assert.*;
5 import java.util.concurrent.Future;
7 import org.junit.Before;
9 import org.opendaylight.controller.md.sal.common.api.TransactionStatus;
10 import org.opendaylight.controller.sal.binding.api.BindingAwareConsumer;
11 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker.ConsumerContext;
12 import org.opendaylight.controller.sal.binding.api.data.DataBrokerService;
13 import org.opendaylight.controller.sal.binding.api.data.DataModificationTransaction;
14 import org.opendaylight.controller.sal.core.api.Broker;
15 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
16 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeRef;
17 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeBuilder;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
21 import org.opendaylight.yangtools.yang.binding.DataObject;
22 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
23 import org.opendaylight.yangtools.yang.common.RpcResult;
25 import com.google.inject.Inject;
27 public class DataServiceTest extends AbstractTest {
29 protected DataBrokerService consumerDataService;
36 public void setUp() throws Exception {
40 public void test() throws Exception {
41 BindingAwareConsumer consumer1 = new BindingAwareConsumer() {
44 public void onSessionInitialized(ConsumerContext session) {
45 consumerDataService = session.getSALService(DataBrokerService.class);
48 broker.registerConsumer(consumer1, getBundleContext());
50 assertNotNull(consumerDataService);
53 DataModificationTransaction transaction = consumerDataService.beginTransaction();
54 assertNotNull(transaction);
56 NodeRef node1 = createNodeRef("0");
57 DataObject node = consumerDataService.readConfigurationData(node1.getValue());
59 Node nodeData1 = createNode("0");
61 transaction.putConfigurationData(node1.getValue(), nodeData1);
62 Future<RpcResult<TransactionStatus>> commitResult = transaction.commit();
63 assertNotNull(commitResult);
65 RpcResult<TransactionStatus> result = commitResult.get();
67 assertNotNull(result);
68 assertNotNull(result.getResult());
69 assertEquals(TransactionStatus.COMMITED, result.getResult());
71 Node readedData = (Node) consumerDataService.readConfigurationData(node1.getValue());
72 assertNotNull(readedData);
73 assertEquals(nodeData1.getKey(), readedData.getKey());
76 DataModificationTransaction transaction2 = consumerDataService.beginTransaction();
77 assertNotNull(transaction);
79 transaction2.removeConfigurationData(node1.getValue());
81 Future<RpcResult<TransactionStatus>> commitResult2 = transaction2.commit();
82 assertNotNull(commitResult2);
84 RpcResult<TransactionStatus> result2 = commitResult2.get();
86 assertNotNull(result2);
87 assertNotNull(result2.getResult());
88 assertEquals(TransactionStatus.COMMITED, result2.getResult());
90 DataObject readedData2 = consumerDataService.readConfigurationData(node1.getValue());
91 assertNull(readedData2);
97 private static NodeRef createNodeRef(String string) {
98 NodeKey key = new NodeKey(new NodeId(string));
99 InstanceIdentifier<Node> path = InstanceIdentifier.builder().node(Nodes.class).node(Node.class, key)
102 return new NodeRef(path);
105 private static Node createNode(String string) {
106 NodeBuilder ret = new NodeBuilder();
107 NodeId id = new NodeId(string);
108 ret.setKey(new NodeKey(id));