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.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;
  24
  25 import com.google.inject.Inject;
  26
  27 public class DataServiceTest extends AbstractTest {
  28
  29     protected DataBrokerService consumerDataService;
  30
  31
  32     @Inject
  33     Broker broker2;
  34
  35     @Before
  36     public void setUp() throws Exception {
  37     }
  38
  39     @Test
  40     public void test() throws Exception {
  41         BindingAwareConsumer consumer1 = new BindingAwareConsumer() {
  42
  43             @Override
  44             public void onSessionInitialized(ConsumerContext session) {
  45                 consumerDataService = session.getSALService(DataBrokerService.class);
  46             }
  47         };
  48         broker.registerConsumer(consumer1, getBundleContext());
  49
  50         assertNotNull(consumerDataService);
  51
  52
  53         DataModificationTransaction transaction = consumerDataService.beginTransaction();
  54         assertNotNull(transaction);
  55
  56         NodeRef node1 = createNodeRef("0");
  57         DataObject  node = consumerDataService.readConfigurationData(node1.getValue());
  58         assertNull(node);
  59         Node nodeData1 = createNode("0");
  60
  61         transaction.putConfigurationData(node1.getValue(), nodeData1);
  62         Future<RpcResult<TransactionStatus>> commitResult = transaction.commit();
  63         assertNotNull(commitResult);
  64
  65         RpcResult<TransactionStatus> result = commitResult.get();
  66
  67         assertNotNull(result);
  68         assertNotNull(result.getResult());
  69         assertEquals(TransactionStatus.COMMITED, result.getResult());
  70
  71         Node readedData = (Node) consumerDataService.readConfigurationData(node1.getValue());
  72         assertNotNull(readedData);
  73         assertEquals(nodeData1.getKey(), readedData.getKey());
  74
  75
  76         DataModificationTransaction transaction2 = consumerDataService.beginTransaction();
  77         assertNotNull(transaction);
  78
  79         transaction2.removeConfigurationData(node1.getValue());
  80
  81         Future<RpcResult<TransactionStatus>> commitResult2 = transaction2.commit();
  82         assertNotNull(commitResult2);
  83
  84         RpcResult<TransactionStatus> result2 = commitResult2.get();
  85
  86         assertNotNull(result2);
  87         assertNotNull(result2.getResult());
  88         assertEquals(TransactionStatus.COMMITED, result2.getResult());
  89
  90         DataObject readedData2 = consumerDataService.readConfigurationData(node1.getValue());
  91         assertNull(readedData2);
  92
  93
  94     }
  95
  96
  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)
 100                 .toInstance();
 101
 102         return new NodeRef(path);
 103     }
 104
 105     private static Node createNode(String string) {
 106         NodeBuilder ret = new NodeBuilder();
 107         NodeId id = new NodeId(string);
 108         ret.setKey(new NodeKey(id));
 109         ret.setId(id);
 110         return ret.build();
 111     }
 112 }