2 * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
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
8 package org.opendaylight.restconf.nb.rfc8040.rests.utils;
10 import static org.hamcrest.CoreMatchers.containsString;
11 import static org.hamcrest.MatcherAssert.assertThat;
12 import static org.junit.Assert.assertEquals;
13 import static org.junit.Assert.assertThrows;
14 import static org.mockito.ArgumentMatchers.any;
15 import static org.mockito.Mockito.doNothing;
16 import static org.mockito.Mockito.doReturn;
17 import static org.mockito.Mockito.verify;
18 import static org.opendaylight.yangtools.util.concurrent.FluentFutures.immediateFailedFluentFuture;
19 import static org.opendaylight.yangtools.util.concurrent.FluentFutures.immediateFalseFluentFuture;
21 import com.google.common.util.concurrent.Futures;
22 import java.net.URLDecoder;
23 import java.nio.charset.StandardCharsets;
24 import java.util.Collection;
25 import java.util.Optional;
26 import javax.ws.rs.core.Response;
27 import javax.ws.rs.core.UriBuilder;
28 import javax.ws.rs.core.UriInfo;
29 import org.junit.Before;
30 import org.junit.Test;
31 import org.junit.runner.RunWith;
32 import org.mockito.Mock;
33 import org.mockito.junit.MockitoJUnitRunner;
34 import org.opendaylight.mdsal.common.api.CommitInfo;
35 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
36 import org.opendaylight.mdsal.dom.api.DOMDataBroker;
37 import org.opendaylight.mdsal.dom.api.DOMDataTreeReadWriteTransaction;
38 import org.opendaylight.mdsal.dom.spi.DefaultDOMRpcResult;
39 import org.opendaylight.netconf.dom.api.NetconfDataTreeService;
40 import org.opendaylight.restconf.common.context.InstanceIdentifierContext;
41 import org.opendaylight.restconf.common.errors.RestconfDocumentedException;
42 import org.opendaylight.restconf.nb.rfc8040.TestRestconfUtils;
43 import org.opendaylight.restconf.nb.rfc8040.WriteDataParams;
44 import org.opendaylight.restconf.nb.rfc8040.legacy.NormalizedNodePayload;
45 import org.opendaylight.restconf.nb.rfc8040.rests.transactions.MdsalRestconfStrategy;
46 import org.opendaylight.restconf.nb.rfc8040.rests.transactions.NetconfRestconfStrategy;
47 import org.opendaylight.yangtools.yang.common.QName;
48 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
49 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
50 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
51 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
52 import org.opendaylight.yangtools.yang.data.api.schema.LeafNode;
53 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
54 import org.opendaylight.yangtools.yang.data.api.schema.MapNode;
55 import org.opendaylight.yangtools.yang.data.impl.schema.Builders;
56 import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
57 import org.opendaylight.yangtools.yang.test.util.YangParserTestUtils;
58 import org.w3c.dom.DOMException;
60 @RunWith(MockitoJUnitRunner.StrictStubs.class)
61 public class PostDataTransactionUtilTest {
62 private static final String PATH_FOR_NEW_SCHEMA_CONTEXT = "/jukebox";
65 private DOMDataTreeReadWriteTransaction readWrite;
67 private UriInfo uriInfo;
69 private DOMDataBroker mockDataBroker;
71 private NetconfDataTreeService netconfService;
73 private ContainerNode buildBaseCont;
74 private EffectiveModelContext schema;
75 private YangInstanceIdentifier iid2;
76 private YangInstanceIdentifier iidList;
77 private MapNode buildList;
80 public void setUp() throws Exception {
82 YangParserTestUtils.parseYangFiles(TestRestconfUtils.loadFiles(PATH_FOR_NEW_SCHEMA_CONTEXT));
84 final QName baseQName = QName.create("http://example.com/ns/example-jukebox", "2015-04-04", "jukebox");
85 final QName containerQname = QName.create(baseQName, "player");
86 final QName leafQname = QName.create(baseQName, "gap");
87 final QName listQname = QName.create(baseQName, "playlist");
88 final QName listKeyQname = QName.create(baseQName, "name");
89 final NodeIdentifierWithPredicates nodeWithKey = NodeIdentifierWithPredicates.of(listQname, listKeyQname,
91 iid2 = YangInstanceIdentifier.builder()
94 iidList = YangInstanceIdentifier.builder()
99 final LeafNode<?> buildLeaf = Builders.leafBuilder()
100 .withNodeIdentifier(new NodeIdentifier(leafQname))
103 final ContainerNode buildPlayerCont = Builders.containerBuilder()
104 .withNodeIdentifier(new NodeIdentifier(containerQname))
105 .withChild(buildLeaf)
107 buildBaseCont = Builders.containerBuilder()
108 .withNodeIdentifier(new NodeIdentifier(baseQName))
109 .withChild(buildPlayerCont)
112 final LeafNode<Object> content = Builders.leafBuilder()
113 .withNodeIdentifier(new NodeIdentifier(QName.create(baseQName, "name")))
114 .withValue("name of band")
116 final LeafNode<Object> content2 = Builders.leafBuilder()
117 .withNodeIdentifier(new NodeIdentifier(QName.create(baseQName, "description")))
118 .withValue("band description")
120 final MapEntryNode mapEntryNode = Builders.mapEntryBuilder()
121 .withNodeIdentifier(nodeWithKey)
125 buildList = Builders.mapBuilder()
126 .withNodeIdentifier(new NodeIdentifier(listQname))
127 .withChild(mapEntryNode)
130 doReturn(UriBuilder.fromUri("http://localhost:8181/restconf/16/")).when(uriInfo).getBaseUriBuilder();
131 doReturn(readWrite).when(mockDataBroker).newReadWriteTransaction();
133 doReturn(Futures.immediateFuture(new DefaultDOMRpcResult())).when(netconfService).lock();
134 doReturn(Futures.immediateFuture(new DefaultDOMRpcResult())).when(netconfService).unlock();
138 public void testPostContainerData() {
139 final InstanceIdentifierContext iidContext =
140 new InstanceIdentifierContext(iid2, null, null, schema);
141 final NormalizedNodePayload payload = NormalizedNodePayload.of(iidContext, buildBaseCont);
143 doReturn(immediateFalseFluentFuture()).when(readWrite).exists(LogicalDatastoreType.CONFIGURATION, iid2);
144 final NodeIdentifier identifier =
145 ((ContainerNode) ((Collection<?>) payload.getData().body()).iterator().next()).getIdentifier();
146 final YangInstanceIdentifier node = iid2.node(identifier);
147 doNothing().when(readWrite).put(LogicalDatastoreType.CONFIGURATION, node.getParent(), payload.getData());
148 doReturn(CommitInfo.emptyFluentFuture()).when(readWrite).commit();
149 doReturn(Futures.immediateFuture(new DefaultDOMRpcResult())).when(netconfService).commit();
150 doReturn(Futures.immediateFuture(new DefaultDOMRpcResult())).when(netconfService)
151 .create(LogicalDatastoreType.CONFIGURATION, iid2, payload.getData(), Optional.empty());
153 Response response = PostDataTransactionUtil.postData(uriInfo, payload,
154 new MdsalRestconfStrategy(mockDataBroker), schema, WriteDataParams.empty());
155 assertEquals(201, response.getStatus());
156 verify(readWrite).exists(LogicalDatastoreType.CONFIGURATION, iid2);
157 verify(readWrite).put(LogicalDatastoreType.CONFIGURATION, iid2, payload.getData());
159 response = PostDataTransactionUtil.postData(uriInfo, payload,
160 new NetconfRestconfStrategy(netconfService), schema, WriteDataParams.empty());
161 assertEquals(201, response.getStatus());
162 verify(netconfService).create(LogicalDatastoreType.CONFIGURATION, iid2, payload.getData(), Optional.empty());
166 public void testPostListData() {
167 final InstanceIdentifierContext iidContext = new InstanceIdentifierContext(iidList, null, null, schema);
168 final NormalizedNodePayload payload = NormalizedNodePayload.of(iidContext, buildList);
170 final MapNode data = (MapNode) payload.getData();
171 final MapEntryNode entryNode = data.body().iterator().next();
172 final NodeIdentifierWithPredicates identifier = entryNode.getIdentifier();
173 final YangInstanceIdentifier node = iidList.node(identifier);
174 doReturn(immediateFalseFluentFuture()).when(readWrite).exists(LogicalDatastoreType.CONFIGURATION, node);
175 doNothing().when(readWrite).put(LogicalDatastoreType.CONFIGURATION, node, entryNode);
176 doReturn(CommitInfo.emptyFluentFuture()).when(readWrite).commit();
177 doReturn(Futures.immediateFuture(new DefaultDOMRpcResult())).when(netconfService)
178 .merge(any(), any(), any(), any());
179 doReturn(Futures.immediateFuture(new DefaultDOMRpcResult())).when(netconfService).commit();
180 doReturn(Futures.immediateFuture(new DefaultDOMRpcResult())).when(netconfService).create(
181 LogicalDatastoreType.CONFIGURATION, node, entryNode, Optional.empty());
183 Response response = PostDataTransactionUtil.postData(uriInfo, payload,
184 new MdsalRestconfStrategy(mockDataBroker), schema, WriteDataParams.empty());
185 assertEquals(201, response.getStatus());
186 assertThat(URLDecoder.decode(response.getLocation().toString(), StandardCharsets.UTF_8),
187 containsString(identifier.getValue(identifier.keySet().iterator().next()).toString()));
188 verify(readWrite).exists(LogicalDatastoreType.CONFIGURATION, node);
189 verify(readWrite).put(LogicalDatastoreType.CONFIGURATION, node, entryNode);
191 response = PostDataTransactionUtil.postData(uriInfo, payload,
192 new NetconfRestconfStrategy(netconfService), schema, WriteDataParams.empty());
193 assertEquals(201, response.getStatus());
194 assertThat(URLDecoder.decode(response.getLocation().toString(), StandardCharsets.UTF_8),
195 containsString(identifier.getValue(identifier.keySet().iterator().next()).toString()));
196 verify(netconfService).create(LogicalDatastoreType.CONFIGURATION, node, entryNode,
201 public void testPostDataFail() {
202 final InstanceIdentifierContext iidContext = new InstanceIdentifierContext(iid2, null, null, schema);
203 final NormalizedNodePayload payload = NormalizedNodePayload.of(iidContext, buildBaseCont);
205 doReturn(immediateFalseFluentFuture()).when(readWrite).exists(LogicalDatastoreType.CONFIGURATION,
207 final NodeIdentifier identifier =
208 ((ContainerNode) ((Collection<?>) payload.getData().body()).iterator().next()).getIdentifier();
209 final YangInstanceIdentifier node = iid2.node(identifier);
210 doNothing().when(readWrite).put(LogicalDatastoreType.CONFIGURATION, node.getParent(), payload.getData());
211 final DOMException domException = new DOMException((short) 414, "Post request failed");
212 doReturn(immediateFailedFluentFuture(domException)).when(readWrite).commit();
213 doReturn(immediateFailedFluentFuture(domException)).when(netconfService)
214 .create(any(), any(), any(), any());
215 doReturn(Futures.immediateFuture(new DefaultDOMRpcResult())).when(netconfService).discardChanges();
216 doReturn(Futures.immediateFuture(new DefaultDOMRpcResult())).when(netconfService).unlock();
218 RestconfDocumentedException ex = assertThrows(RestconfDocumentedException.class,
219 () -> PostDataTransactionUtil.postData(uriInfo, payload, new MdsalRestconfStrategy(mockDataBroker), schema,
220 WriteDataParams.empty()));
221 assertEquals(1, ex.getErrors().size());
222 assertThat(ex.getErrors().get(0).getErrorInfo(), containsString(domException.getMessage()));
224 verify(readWrite).exists(LogicalDatastoreType.CONFIGURATION, iid2);
225 verify(readWrite).put(LogicalDatastoreType.CONFIGURATION, iid2, payload.getData());
227 ex = assertThrows(RestconfDocumentedException.class, () -> PostDataTransactionUtil.postData(uriInfo, payload,
228 new NetconfRestconfStrategy(netconfService), schema, WriteDataParams.empty()));
229 assertEquals(1, ex.getErrors().size());
230 assertThat(ex.getErrors().get(0).getErrorInfo(), containsString(domException.getMessage()));
232 verify(netconfService).create(LogicalDatastoreType.CONFIGURATION, iid2, payload.getData(), Optional.empty());