Reduce the use of AttrBuilders
[netconf.git] / restconf / restconf-nb-bierman02 / src / test / java / org / opendaylight / controller / sal / restconf / impl / test / URIParametersParsing.java
1 /*
2  * Copyright (c) 2014 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 package org.opendaylight.controller.sal.restconf.impl.test;
9
10 import static org.junit.Assert.assertNotNull;
11 import static org.junit.Assert.assertTrue;
12 import static org.mockito.ArgumentMatchers.eq;
13 import static org.mockito.Mockito.mock;
14 import static org.mockito.Mockito.when;
15
16 import com.google.common.base.Preconditions;
17 import java.io.FileNotFoundException;
18 import java.util.Set;
19 import javax.ws.rs.core.MultivaluedMap;
20 import javax.ws.rs.core.UriBuilder;
21 import javax.ws.rs.core.UriInfo;
22 import org.junit.Before;
23 import org.junit.Test;
24 import org.mockito.Mockito;
25 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
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.RestconfImpl;
30 import org.opendaylight.netconf.sal.streams.listeners.ListenerAdapter;
31 import org.opendaylight.netconf.sal.streams.listeners.Notificator;
32 import org.opendaylight.restconf.common.context.InstanceIdentifierContext;
33 import org.opendaylight.restconf.common.context.NormalizedNodeContext;
34 import org.opendaylight.restconf.common.util.DataChangeScope;
35 import org.opendaylight.yang.gen.v1.urn.sal.restconf.event.subscription.rev140708.NotificationOutputTypeGrouping.NotificationOutputType;
36 import org.opendaylight.yangtools.yang.common.QName;
37 import org.opendaylight.yangtools.yang.common.Revision;
38 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
39 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.AugmentationIdentifier;
40 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.InstanceIdentifierBuilder;
41 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
42 import org.opendaylight.yangtools.yang.data.api.schema.AugmentationNode;
43 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
44 import org.opendaylight.yangtools.yang.data.api.schema.LeafNode;
45 import org.opendaylight.yangtools.yang.data.impl.schema.Builders;
46 import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.DataContainerNodeBuilder;
47 import org.opendaylight.yangtools.yang.model.api.AugmentationSchemaNode;
48 import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
49 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
50 import org.opendaylight.yangtools.yang.model.api.LeafSchemaNode;
51 import org.opendaylight.yangtools.yang.model.api.Module;
52 import org.opendaylight.yangtools.yang.model.api.RpcDefinition;
53 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
54 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
55 import org.opendaylight.yangtools.yang.model.util.SchemaNodeUtils;
56 import org.opendaylight.yangtools.yang.parser.spi.meta.ReactorException;
57
58 public class URIParametersParsing {
59
60     private RestconfImpl restconf;
61     private BrokerFacade mockedBrokerFacade;
62     private ControllerContext controllerContext;
63
64     @Before
65     public void init() throws FileNotFoundException, ReactorException {
66         this.mockedBrokerFacade = mock(BrokerFacade.class);
67         this.controllerContext = TestRestconfUtils.newControllerContext(
68                 TestUtils.loadSchemaContext("/datastore-and-scope-specification"));
69         this.restconf = RestconfImpl.newInstance(mockedBrokerFacade, controllerContext);
70     }
71
72     @Test
73     public void resolveURIParametersConcreteValues() {
74         resolveURIParameters("OPERATIONAL", "SUBTREE", LogicalDatastoreType.OPERATIONAL, DataChangeScope.SUBTREE);
75     }
76
77     @Test
78     public void resolveURIParametersDefaultValues() {
79         resolveURIParameters(null, null, LogicalDatastoreType.CONFIGURATION, DataChangeScope.BASE);
80     }
81
82     private void resolveURIParameters(final String datastore, final String scope,
83             final LogicalDatastoreType datastoreExpected, final DataChangeScope scopeExpected) {
84
85         final InstanceIdentifierBuilder iiBuilder = YangInstanceIdentifier.builder();
86         iiBuilder.node(QName.create("", "dummyStreamName"));
87
88         final String datastoreValue = datastore == null ? "CONFIGURATION" : datastore;
89         final String scopeValue = scope == null ? "BASE" : scope + "";
90         Notificator.createListener(iiBuilder.build(), "dummyStreamName/datastore=" + datastoreValue + "/scope="
91                 + scopeValue, NotificationOutputType.XML, controllerContext);
92
93         final UriInfo mockedUriInfo = mock(UriInfo.class);
94         @SuppressWarnings("unchecked")
95         final MultivaluedMap<String, String> mockedMultivaluedMap = mock(MultivaluedMap.class);
96         when(mockedMultivaluedMap.getFirst(eq("datastore"))).thenReturn(datastoreValue);
97         when(mockedMultivaluedMap.getFirst(eq("scope"))).thenReturn(scopeValue);
98
99         when(mockedUriInfo.getQueryParameters(eq(false))).thenReturn(mockedMultivaluedMap);
100
101         final UriBuilder uriBuilder = UriBuilder.fromUri("www.whatever.com");
102         when(mockedUriInfo.getAbsolutePathBuilder()).thenReturn(uriBuilder);
103
104         this.restconf.invokeRpc("sal-remote:create-data-change-event-subscription",
105             prepareDomRpcNode(datastoreValue, scopeValue), mockedUriInfo);
106
107         final ListenerAdapter listener =
108                 Notificator.getListenerFor("data-change-event-subscription/opendaylight-inventory:nodes/datastore="
109                 + datastoreValue + "/scope=" + scopeValue);
110         assertNotNull(listener);
111     }
112
113     private NormalizedNodeContext prepareDomRpcNode(final String datastore, final String scope) {
114         final SchemaContext schema = this.controllerContext.getGlobalSchema();
115         final Module rpcSalRemoteModule = schema.findModule("sal-remote", Revision.of("2014-01-14")).get();
116         final Set<RpcDefinition> setRpcs = rpcSalRemoteModule.getRpcs();
117         final QName rpcQName =
118                 QName.create(rpcSalRemoteModule.getQNameModule(), "create-data-change-event-subscription");
119         final QName rpcInputQName =
120                 QName.create("urn:opendaylight:params:xml:ns:yang:controller:md:sal:remote", "2014-01-14", "input");
121         final RpcDefinition rpcDef = Mockito.mock(RpcDefinition.class);
122         ContainerSchemaNode rpcInputSchemaNode = null;
123         for (final RpcDefinition rpc : setRpcs) {
124             if (rpcQName.isEqualWithoutRevision(rpc.getQName())) {
125                 rpcInputSchemaNode = SchemaNodeUtils.getRpcDataSchema(rpc, rpcInputQName);
126                 break;
127             }
128         }
129         assertNotNull("RPC ContainerSchemaNode was not found!", rpcInputSchemaNode);
130
131         final DataContainerNodeBuilder<NodeIdentifier, ContainerNode> container =
132                 Builders.containerBuilder(rpcInputSchemaNode);
133
134         final QName pathQName =
135                 QName.create("urn:opendaylight:params:xml:ns:yang:controller:md:sal:remote", "2014-01-14", "path");
136         final DataSchemaNode pathSchemaNode = rpcInputSchemaNode.getDataChildByName(pathQName);
137         assertTrue(pathSchemaNode instanceof LeafSchemaNode);
138         final LeafNode<Object> pathNode = Builders.leafBuilder((LeafSchemaNode) pathSchemaNode)
139                 .withValue(YangInstanceIdentifier.builder()
140                         .node(QName.create("urn:opendaylight:inventory", "2013-08-19", "nodes")).build()).build();
141         container.withChild(pathNode);
142
143         final AugmentationSchemaNode augmentationSchema = rpcInputSchemaNode.getAvailableAugmentations().iterator()
144                 .next();
145         Preconditions.checkNotNull(augmentationSchema);
146         final DataContainerNodeBuilder<AugmentationIdentifier, AugmentationNode> augmentationBuilder =
147                 Builders.augmentationBuilder(augmentationSchema);
148
149         final QName dataStoreQName = QName.create("urn:sal:restconf:event:subscription", "2014-07-08", "datastore");
150         final DataSchemaNode dsSchemaNode = augmentationSchema.getDataChildByName(dataStoreQName);
151         assertTrue(dsSchemaNode instanceof LeafSchemaNode);
152         final LeafNode<Object> dsNode = Builders.leafBuilder((LeafSchemaNode) dsSchemaNode)
153                 .withValue(datastore).build();
154         augmentationBuilder.withChild(dsNode);
155
156         final QName scopeQName = QName.create("urn:sal:restconf:event:subscription", "2014-07-08", "scope");
157         final DataSchemaNode scopeSchemaNode = augmentationSchema.getDataChildByName(scopeQName);
158         assertTrue(scopeSchemaNode instanceof LeafSchemaNode);
159         final LeafNode<Object> scopeNode = Builders.leafBuilder((LeafSchemaNode) scopeSchemaNode)
160                 .withValue(scope).build();
161         augmentationBuilder.withChild(scopeNode);
162
163         final QName outputQName =
164                 QName.create("urn:sal:restconf:event:subscription", "2014-07-08", "notification-output-type");
165         final DataSchemaNode outputSchemaNode = augmentationSchema.getDataChildByName(outputQName);
166         assertTrue(outputSchemaNode instanceof LeafSchemaNode);
167         final LeafNode<Object> outputNode =
168                 Builders.leafBuilder((LeafSchemaNode) outputSchemaNode).withValue("XML").build();
169         augmentationBuilder.withChild(outputNode);
170
171         container.withChild(augmentationBuilder.build());
172
173         when(rpcDef.getInput()).thenReturn(rpcInputSchemaNode);
174         when(rpcDef.getPath()).thenReturn(SchemaPath.create(true, rpcQName));
175         when(rpcDef.getQName()).thenReturn(rpcQName);
176
177         return new NormalizedNodeContext(new InstanceIdentifierContext<>(null, rpcDef, null, schema),
178                 container.build());
179     }
180 }