opendaylight/md-sal/sal-binding-it/src/test/java/org/opendaylight/controller/test/sal/binding/it/DataServiceTest.java

   1 package org.opendaylight.controller.test.sal.binding.it;
   2
   3 import static org.junit.Assert.*;
   4
   5 import java.util.concurrent.Future;
   6
   7 import org.junit.Before;
   8 import org.junit.Test;
   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.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
  15 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeRef;
  16 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeBuilder;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
  20 import org.opendaylight.yangtools.yang.binding.DataObject;
  21 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  22 import org.opendaylight.yangtools.yang.common.RpcResult;
  23
  24 public class DataServiceTest extends AbstractTest {
  25
  26     protected DataBrokerService consumerDataService;
  27
  28     @Before
  29     public void setUp() throws Exception {
  30     }
  31
  32     @Test
  33     public void test() throws Exception {
  34         BindingAwareConsumer consumer1 = new BindingAwareConsumer() {
  35
  36             @Override
  37             public void onSessionInitialized(ConsumerContext session) {
  38                 consumerDataService = session.getSALService(DataBrokerService.class);
  39             }
  40         };
  41         broker.registerConsumer(consumer1, getBundleContext());
  42
  43         assertNotNull(consumerDataService);
  44
  45
  46         DataModificationTransaction transaction = consumerDataService.beginTransaction();
  47         assertNotNull(transaction);
  48
  49         NodeRef node1 = createNodeRef("0");
  50         DataObject  node = consumerDataService.readConfigurationData(node1.getValue());
  51         assertNull(node);
  52         Node nodeData1 = createNode("0");
  53
  54         transaction.putConfigurationData(node1.getValue(), nodeData1);
  55         Future<RpcResult<TransactionStatus>> commitResult = transaction.commit();
  56         assertNotNull(commitResult);
  57
  58         RpcResult<TransactionStatus> result = commitResult.get();
  59
  60         assertNotNull(result);
  61         assertNotNull(result.getResult());
  62         assertEquals(TransactionStatus.COMMITED, result.getResult());
  63
  64         DataObject readedData = consumerDataService.readConfigurationData(node1.getValue());
  65         assertNotNull(readedData);
  66         assertEquals(nodeData1, readedData);
  67
  68
  69         DataModificationTransaction transaction2 = consumerDataService.beginTransaction();
  70         assertNotNull(transaction);
  71
  72         transaction2.removeConfigurationData(node1.getValue());
  73
  74         Future<RpcResult<TransactionStatus>> commitResult2 = transaction2.commit();
  75         assertNotNull(commitResult2);
  76
  77         RpcResult<TransactionStatus> result2 = commitResult2.get();
  78
  79         assertNotNull(result2);
  80         assertNotNull(result2.getResult());
  81         assertEquals(TransactionStatus.COMMITED, result2.getResult());
  82
  83         DataObject readedData2 = consumerDataService.readConfigurationData(node1.getValue());
  84         assertNull(readedData2);
  85
  86
  87     }
  88
  89
  90     private static NodeRef createNodeRef(String string) {
  91         NodeKey key = new NodeKey(new NodeId(string));
  92         InstanceIdentifier<Node> path = InstanceIdentifier.builder().node(Nodes.class).node(Node.class, key)
  93                 .toInstance();
  94
  95         return new NodeRef(path);
  96     }
  97
  98     private static Node createNode(String string) {
  99         NodeBuilder ret = new NodeBuilder();
 100         ret.setId(new NodeId(string));
 101         return ret.build();
 102     }
 103 }