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