restconf/restconf-nb-bierman02/src/test/java/org/opendaylight/controller/sal/restconf/impl/test/RestPutConfigTest.java

   1 /*
   2  * Copyright (c) 2015 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
   9 package org.opendaylight.controller.sal.restconf.impl.test;
  10
  11 import com.google.common.util.concurrent.CheckedFuture;
  12 import com.google.common.util.concurrent.Futures;
  13 import java.util.HashSet;
  14 import javax.ws.rs.core.MultivaluedMap;
  15 import javax.ws.rs.core.Response.Status;
  16 import javax.ws.rs.core.UriInfo;
  17 import org.junit.Before;
  18 import org.junit.Test;
  19 import org.junit.runner.RunWith;
  20 import org.mockito.Mock;
  21 import org.mockito.Mockito;
  22 import org.mockito.runners.MockitoJUnitRunner;
  23 import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
  24 import org.opendaylight.controller.md.sal.rest.common.TestRestconfUtils;
  25 import org.opendaylight.netconf.sal.restconf.impl.BrokerFacade;
  26 import org.opendaylight.netconf.sal.restconf.impl.ControllerContext;
  27 import org.opendaylight.netconf.sal.restconf.impl.PutResult;
  28 import org.opendaylight.netconf.sal.restconf.impl.RestconfDocumentedException;
  29 import org.opendaylight.netconf.sal.restconf.impl.RestconfImpl;
  30 import org.opendaylight.restconf.common.context.InstanceIdentifierContext;
  31 import org.opendaylight.restconf.common.context.NormalizedNodeContext;
  32 import org.opendaylight.yangtools.yang.common.QName;
  33 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  34 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
  35 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
  36 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  37 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
  38
  39 @RunWith(MockitoJUnitRunner.class)
  40 public class RestPutConfigTest {
  41
  42     private RestconfImpl restconfService;
  43     private ControllerContext controllerCx;
  44     private SchemaContext schemaCx;
  45
  46     @Mock
  47     private BrokerFacade brokerFacade;
  48
  49     @Before
  50     public void init() {
  51         this.restconfService = RestconfImpl.getInstance();
  52         this.controllerCx = ControllerContext.getInstance();
  53         this.schemaCx = TestRestconfUtils.loadSchemaContext("/test-config-data/yang1/", null);
  54         this.controllerCx.setSchemas(this.schemaCx);
  55         this.restconfService.setControllerContext(this.controllerCx);
  56     }
  57
  58     @Test
  59     public void testPutConfigData() {
  60         final String identifier = "test-interface:interfaces/interface/key";
  61         final InstanceIdentifierContext<?> iiCx = this.controllerCx.toInstanceIdentifier(identifier);
  62         final MapEntryNode data = Mockito.mock(MapEntryNode.class);
  63         final QName qName = QName.create("urn:ietf:params:xml:ns:yang:test-interface", "2014-07-01", "interface");
  64         final QName qNameKey = QName.create("urn:ietf:params:xml:ns:yang:test-interface", "2014-07-01", "name");
  65         final NodeIdentifierWithPredicates identWithPredicates =
  66                 new NodeIdentifierWithPredicates(qName, qNameKey, "key");
  67         Mockito.when(data.getNodeType()).thenReturn(qName);
  68         Mockito.when(data.getIdentifier()).thenReturn(identWithPredicates);
  69         final NormalizedNodeContext payload = new NormalizedNodeContext(iiCx, data);
  70
  71         mockingBrokerPut(iiCx.getInstanceIdentifier(), data);
  72
  73         final UriInfo uriInfo = Mockito.mock(UriInfo.class);
  74         final MultivaluedMap<String, String> value = Mockito.mock(MultivaluedMap.class);
  75         Mockito.when(value.entrySet()).thenReturn(new HashSet<>());
  76         Mockito.when(uriInfo.getQueryParameters()).thenReturn(value);
  77         this.restconfService.updateConfigurationData(identifier, payload, uriInfo);
  78     }
  79
  80     @Test
  81     public void testPutConfigDataCheckOnlyLastElement() {
  82         final String identifier = "test-interface:interfaces/interface/key/sub-interface/subkey";
  83         final InstanceIdentifierContext<?> iiCx = this.controllerCx.toInstanceIdentifier(identifier);
  84         final MapEntryNode data = Mockito.mock(MapEntryNode.class);
  85         final QName qName = QName.create("urn:ietf:params:xml:ns:yang:test-interface", "2014-07-01", "sub-interface");
  86         final QName qNameSubKey = QName.create("urn:ietf:params:xml:ns:yang:test-interface", "2014-07-01", "sub-name");
  87         final NodeIdentifierWithPredicates identWithPredicates =
  88                 new NodeIdentifierWithPredicates(qName, qNameSubKey, "subkey");
  89         Mockito.when(data.getNodeType()).thenReturn(qName);
  90         Mockito.when(data.getIdentifier()).thenReturn(identWithPredicates);
  91         final NormalizedNodeContext payload = new NormalizedNodeContext(iiCx, data);
  92
  93         mockingBrokerPut(iiCx.getInstanceIdentifier(), data);
  94
  95         final UriInfo uriInfo = Mockito.mock(UriInfo.class);
  96         final MultivaluedMap<String, String> value = Mockito.mock(MultivaluedMap.class);
  97         Mockito.when(value.entrySet()).thenReturn(new HashSet<>());
  98         Mockito.when(uriInfo.getQueryParameters()).thenReturn(value);
  99         this.restconfService.updateConfigurationData(identifier, payload, uriInfo);
 100     }
 101
 102     @Test(expected = RestconfDocumentedException.class)
 103     public void testPutConfigDataMissingUriKey() {
 104         final String identifier = "test-interface:interfaces/interface";
 105         this.controllerCx.toInstanceIdentifier(identifier);
 106     }
 107
 108     @Test(expected = RestconfDocumentedException.class)
 109     public void testPutConfigDataDiferentKey() {
 110         final String identifier = "test-interface:interfaces/interface/key";
 111         final InstanceIdentifierContext<?> iiCx = this.controllerCx.toInstanceIdentifier(identifier);
 112         final MapEntryNode data = Mockito.mock(MapEntryNode.class);
 113         final QName qName = QName.create("urn:ietf:params:xml:ns:yang:test-interface", "2014-07-01", "interface");
 114         final QName qNameKey = QName.create("urn:ietf:params:xml:ns:yang:test-interface", "2014-07-01", "name");
 115         final NodeIdentifierWithPredicates identWithPredicates =
 116                 new NodeIdentifierWithPredicates(qName, qNameKey, "notSameKey");
 117         Mockito.when(data.getNodeType()).thenReturn(qName);
 118         Mockito.when(data.getIdentifier()).thenReturn(identWithPredicates);
 119         final NormalizedNodeContext payload = new NormalizedNodeContext(iiCx, data);
 120
 121         mockingBrokerPut(iiCx.getInstanceIdentifier(), data);
 122
 123         final UriInfo uriInfo = Mockito.mock(UriInfo.class);
 124         final MultivaluedMap<String, String> value = Mockito.mock(MultivaluedMap.class);
 125         Mockito.when(value.entrySet()).thenReturn(new HashSet<>());
 126         Mockito.when(uriInfo.getQueryParameters()).thenReturn(value);
 127         this.restconfService.updateConfigurationData(identifier, payload, uriInfo);
 128     }
 129
 130     private void mockingBrokerPut(final YangInstanceIdentifier yii, final NormalizedNode<?, ?> data) {
 131         final PutResult result = Mockito.mock(PutResult.class);
 132         final CheckedFuture<Void, TransactionCommitFailedException> checkedFuture =
 133                 Futures.immediateCheckedFuture(null);
 134         Mockito.when(this.brokerFacade.commitConfigurationDataPut(this.schemaCx, yii, data, null, null))
 135                 .thenReturn(result);
 136         Mockito.when(result.getFutureOfPutData()).thenReturn(checkedFuture);
 137         Mockito.when(result.getStatus()).thenReturn(Status.OK);
 138         this.restconfService.setBroker(this.brokerFacade);
 139     }
 140 }