2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright (C) 2019 Ericsson
6 * ================================================================================
7 * Licensed under the Apache License, Version 2.0 (the "License");
8 * you may not use this file except in compliance with the License.
9 * You may obtain a copy of the License at
11 * http://www.apache.org/licenses/LICENSE-2.0
13 * Unless required by applicable law or agreed to in writing, software
14 * distributed under the License is distributed on an "AS IS" BASIS,
15 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16 * See the License for the specific language governing permissions and
17 * limitations under the License.
19 * ============LICENSE_END=========================================================
22 package org.onap.appc.artifact.handler;
24 import com.google.common.util.concurrent.CheckedFuture;
25 import static org.junit.Assert.assertEquals;
26 import static org.junit.Assert.assertThat;
27 import static org.junit.Assert.assertTrue;
28 import java.util.concurrent.ExecutionException;
29 import java.util.concurrent.Future;
30 import org.junit.Before;
31 import org.junit.Test;
32 import org.mockito.Mockito;
33 import org.onap.appc.artifact.handler.utils.ArtifactHandlerProviderUtil;
34 import org.onap.appc.artifact.handler.utils.SdcArtifactHandlerConstants;
35 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
36 import org.opendaylight.controller.md.sal.binding.api.NotificationPublishService;
37 import org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction;
38 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker;
39 import org.opendaylight.controller.sal.binding.api.RpcProviderRegistry;
40 import org.opendaylight.yangtools.yang.common.RpcResult;
41 import org.opendaylight.yang.gen.v1.org.onap.appc.artifacthandler.rev170321.ArtifactHandlerService;
42 import org.opendaylight.yang.gen.v1.org.onap.appc.artifacthandler.rev170321.UploadartifactInput;
43 import org.opendaylight.yang.gen.v1.org.onap.appc.artifacthandler.rev170321.UploadartifactInputBuilder;
44 import org.opendaylight.yang.gen.v1.org.onap.appc.artifacthandler.rev170321.UploadartifactOutput;
45 import org.opendaylight.yang.gen.v1.org.onap.appc.artifacthandler.rev170321.document.parameters.DocumentParameters;
46 import org.opendaylight.yang.gen.v1.org.onap.appc.artifacthandler.rev170321.request.information.RequestInformation;
47 import org.powermock.reflect.Whitebox;
49 public class ArtifactHandlerProviderTest {
51 private DataBroker dataBroker = Mockito.mock(DataBroker.class);
52 private RpcProviderRegistry rpcRegistry = Mockito.mock(RpcProviderRegistry.class);
53 private NotificationPublishService notificationService = Mockito.mock(NotificationPublishService.class);
54 private ReadWriteTransaction writeTransaction = Mockito.mock(ReadWriteTransaction.class);
55 private CheckedFuture checkedFuture = Mockito.mock(CheckedFuture.class);
56 private BindingAwareBroker.RpcRegistration<ArtifactHandlerService> rpcRegistration;
58 private ArtifactHandlerProvider artifactHandlerProvider;
63 (BindingAwareBroker.RpcRegistration<ArtifactHandlerService>) Mockito.mock(BindingAwareBroker.RpcRegistration.class);
64 Mockito.doReturn(writeTransaction).when(dataBroker).newReadWriteTransaction();
65 Mockito.doReturn(checkedFuture).when(writeTransaction).submit();
70 public void testUploadArtifactNullInput() throws InterruptedException, ExecutionException {
71 artifactHandlerProvider = new ArtifactHandlerProvider(dataBroker, notificationService, rpcRegistry);
72 UploadartifactInput uploadArtifactInput = Mockito.mock(UploadartifactInput.class);
73 Future<RpcResult<UploadartifactOutput>> output = (Future<RpcResult<UploadartifactOutput>>) artifactHandlerProvider.uploadartifact(uploadArtifactInput);
74 assertTrue(output.get().getResult() instanceof UploadartifactOutput);
78 public void testUploadArtifactException() throws Exception {
79 artifactHandlerProvider = Mockito.spy(new ArtifactHandlerProvider(dataBroker, notificationService, rpcRegistry));
80 UploadartifactInputBuilder builder = new UploadartifactInputBuilder();
81 DocumentParameters mockDocumentParameters = Mockito.mock(DocumentParameters.class);
82 Mockito.doReturn("ARTIFACT CONTENTS").when(mockDocumentParameters).getArtifactContents();
83 Mockito.doReturn("ARTIFACT NAME").when(mockDocumentParameters).getArtifactName();
84 builder.setDocumentParameters(mockDocumentParameters);
85 RequestInformation mockRequestInformation = Mockito.mock(RequestInformation.class);
86 Mockito.doReturn("REQUEST ID").when(mockRequestInformation).getRequestId();
87 Mockito.doReturn(SdcArtifactHandlerConstants.DESIGN_TOOL).when(mockRequestInformation).getSource();
88 builder.setRequestInformation(mockRequestInformation);
89 UploadartifactInput uploadArtifactInput = builder.build();
90 ArtifactHandlerProviderUtil mockProvider = Mockito.mock(ArtifactHandlerProviderUtil.class);
91 Mockito.doThrow(new Exception()).when(mockProvider).processTemplate(Mockito.anyString());
92 Mockito.doReturn(mockProvider).when(artifactHandlerProvider).getArtifactHandlerProviderUtil(Mockito.any(UploadartifactInput.class));
93 Future<RpcResult<UploadartifactOutput>> output = (Future<RpcResult<UploadartifactOutput>>) artifactHandlerProvider.uploadartifact(uploadArtifactInput);
94 assertTrue(output.get().getResult() instanceof UploadartifactOutput);
98 public void testUploadArtifact() throws InterruptedException, ExecutionException {
99 artifactHandlerProvider = Mockito.spy(new ArtifactHandlerProvider(dataBroker, notificationService, rpcRegistry));
100 UploadartifactInputBuilder builder = new UploadartifactInputBuilder();
101 DocumentParameters mockDocumentParameters = Mockito.mock(DocumentParameters.class);
102 Mockito.doReturn("ARTIFACT CONTENTS").when(mockDocumentParameters).getArtifactContents();
103 Mockito.doReturn("ARTIFACT NAME").when(mockDocumentParameters).getArtifactName();
104 builder.setDocumentParameters(mockDocumentParameters);
105 RequestInformation mockRequestInformation = Mockito.mock(RequestInformation.class);
106 Mockito.doReturn("REQUEST ID").when(mockRequestInformation).getRequestId();
107 Mockito.doReturn(SdcArtifactHandlerConstants.DESIGN_TOOL).when(mockRequestInformation).getSource();
108 builder.setRequestInformation(mockRequestInformation);
109 UploadartifactInput uploadArtifactInput = builder.build();
110 ArtifactHandlerProviderUtil mockProvider = Mockito.mock(ArtifactHandlerProviderUtil.class);
111 Mockito.doReturn(mockProvider).when(artifactHandlerProvider).getArtifactHandlerProviderUtil(Mockito.any(UploadartifactInput.class));
112 Future<RpcResult<UploadartifactOutput>> output = (Future<RpcResult<UploadartifactOutput>>) artifactHandlerProvider.uploadartifact(uploadArtifactInput);
113 assertTrue(output.get().getResult() instanceof UploadartifactOutput);
117 public void testUploadArtifact2() throws InterruptedException, ExecutionException {
118 artifactHandlerProvider = Mockito.spy(new ArtifactHandlerProvider(dataBroker, notificationService, rpcRegistry));
119 UploadartifactInputBuilder builder = new UploadartifactInputBuilder();
120 DocumentParameters mockDocumentParameters = Mockito.mock(DocumentParameters.class);
121 Mockito.doReturn("ARTIFACT CONTENTS").when(mockDocumentParameters).getArtifactContents();
122 Mockito.doReturn("ARTIFACT NAME").when(mockDocumentParameters).getArtifactName();
123 builder.setDocumentParameters(mockDocumentParameters);
124 RequestInformation mockRequestInformation = Mockito.mock(RequestInformation.class);
125 Mockito.doReturn("REQUEST ID").when(mockRequestInformation).getRequestId();
126 builder.setRequestInformation(mockRequestInformation);
127 UploadartifactInput uploadArtifactInput = builder.build();
128 ArtifactHandlerProviderUtil mockProvider = Mockito.mock(ArtifactHandlerProviderUtil.class);
129 Mockito.doReturn(mockProvider).when(artifactHandlerProvider).getArtifactHandlerProviderUtil(Mockito.any(UploadartifactInput.class));
130 Future<RpcResult<UploadartifactOutput>> output = (Future<RpcResult<UploadartifactOutput>>) artifactHandlerProvider.uploadartifact(uploadArtifactInput);
131 assertTrue(output.get().getResult() instanceof UploadartifactOutput);
135 public void testUploadArtifact3() throws InterruptedException, ExecutionException {
136 artifactHandlerProvider = Mockito.spy(new ArtifactHandlerProvider(dataBroker, notificationService, rpcRegistry));
137 UploadartifactInputBuilder builder = new UploadartifactInputBuilder();
138 DocumentParameters mockDocumentParameters = Mockito.mock(DocumentParameters.class);
139 Mockito.doReturn("ARTIFACT CONTENTS").when(mockDocumentParameters).getArtifactContents();
140 Mockito.doReturn("ARTIFACT NAME").when(mockDocumentParameters).getArtifactName();
141 builder.setDocumentParameters(mockDocumentParameters);
142 RequestInformation mockRequestInformation = Mockito.mock(RequestInformation.class);
143 Mockito.doReturn("REQUEST ID").when(mockRequestInformation).getRequestId();
144 Mockito.doReturn(SdcArtifactHandlerConstants.ACTION).when(mockRequestInformation).getSource();
145 builder.setRequestInformation(mockRequestInformation);
146 UploadartifactInput uploadArtifactInput = builder.build();
147 ArtifactHandlerProviderUtil mockProvider = Mockito.mock(ArtifactHandlerProviderUtil.class);
148 Mockito.doReturn(mockProvider).when(artifactHandlerProvider).getArtifactHandlerProviderUtil(Mockito.any(UploadartifactInput.class));
149 Future<RpcResult<UploadartifactOutput>> output = (Future<RpcResult<UploadartifactOutput>>) artifactHandlerProvider.uploadartifact(uploadArtifactInput);
150 assertTrue(output.get().getResult() instanceof UploadartifactOutput);