2 * Copyright 2016-2017, Nokia Corporation
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
17 package org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.restapi;
19 import com.google.common.collect.Lists;
20 import junit.framework.TestCase;
21 import org.apache.http.entity.ContentType;
22 import org.junit.Before;
23 import org.junit.Test;
24 import org.mockito.InjectMocks;
25 import org.mockito.Mock;
26 import org.mockito.Mockito;
27 import org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.packagetransformer.CbamVnfPackageBuilder;
28 import org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.packagetransformer.CbamVnfdBuilder;
29 import org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.packagetransformer.OnapVnfdBuilder;
30 import org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.util.TestUtil;
31 import org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.vnfm.TestBase;
32 import org.springframework.http.HttpHeaders;
33 import org.springframework.http.HttpStatus;
34 import org.springframework.mock.web.DelegatingServletOutputStream;
36 import javax.servlet.http.HttpServletRequest;
37 import javax.servlet.http.Part;
38 import java.io.ByteArrayInputStream;
39 import java.io.ByteArrayOutputStream;
40 import java.io.PrintStream;
41 import java.util.Arrays;
43 import static org.mockito.Mockito.verify;
44 import static org.mockito.Mockito.when;
45 import static org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.vnfm.CatalogManager.getFileInZip;
46 import static org.springframework.test.util.ReflectionTestUtils.setField;
49 public class TestConverterApi extends TestBase {
53 private ConverterApi converterApi;
55 private HttpServletRequest httpRequest;
58 public void initMocks() throws Exception {
59 setField(ConverterApi.class, "logger", logger);
65 public void test() throws Exception {
66 ByteArrayOutputStream bos = new ByteArrayOutputStream();
67 PrintStream actualOut = new PrintStream(bos, true);
68 when(systemFunctions.out()).thenReturn(actualOut);
69 when(systemFunctions.loadFile("cbam.pre.collectConnectionPoints.js")).thenCallRealMethod();
70 when(systemFunctions.loadFile("cbam.collectConnectionPoints.js")).thenCallRealMethod();
71 when(systemFunctions.loadFile("cbam.post.collectConnectionPoints.js")).thenCallRealMethod();
72 when(systemFunctions.loadFile("TOSCA.meta")).thenCallRealMethod();
73 when(systemFunctions.loadFile("MainServiceTemplate.meta")).thenCallRealMethod();
74 when(httpResponse.getOutputStream()).thenReturn(new DelegatingServletOutputStream(actualOut));
75 Part part = Mockito.mock(Part.class);
76 when(part.getInputStream()).thenReturn(new ByteArrayInputStream(TestUtil.loadFile("unittests/packageconverter/cbam.package.zip")));
77 when(httpRequest.getParts()).thenReturn(Lists.newArrayList(part));
79 converterApi.convert(httpResponse, httpRequest);
81 verifyVnfPackageWritterToOutputStream(bos);
82 verify(httpResponse).addHeader(HttpHeaders.CONTENT_TYPE, ContentType.APPLICATION_OCTET_STREAM.getMimeType());
83 verify(httpResponse).setStatus(HttpStatus.OK.value());
84 verify(httpResponse).addHeader(HttpHeaders.CONTENT_LENGTH, Integer.toString(bos.toByteArray().length));
85 verify(httpResponse).addHeader(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=\"" + "core.csar" + "\"");
88 private void verifyVnfPackageWritterToOutputStream(ByteArrayOutputStream bos) throws Exception {
89 String cbamVnfd = new String(TestUtil.loadFile("unittests/packageconverter/cbam.package.zip.vnfd"));
90 String expectedOnapVnfd = new OnapVnfdBuilder().toOnapVnfd(cbamVnfd);
91 assertFileInZip(bos.toByteArray(), "TOSCA-Metadata/TOSCA.meta", TestUtil.loadFile("TOSCA.meta"));
92 assertFileInZip(bos.toByteArray(), "Definitions/MainServiceTemplate.yaml", expectedOnapVnfd.getBytes());
93 assertFileInZip(bos.toByteArray(), "MainServiceTemplate.yaml", expectedOnapVnfd.getBytes());
94 assertFileInZip(bos.toByteArray(), "MainServiceTemplate.meta", TestUtil.loadFile("MainServiceTemplate.meta"));
95 ByteArrayOutputStream actualModifiedCbamVnfPackage = getFileInZip(new ByteArrayInputStream(bos.toByteArray()), "Artifacts/Deployment/OTHER/cbam.package.zip");
96 byte[] expectedModifiedCbamPackage = new CbamVnfPackageBuilder().toModifiedCbamVnfPackage(TestUtil.loadFile("unittests/packageconverter/cbam.package.zip"), "vnfdloc/a.yaml", new CbamVnfdBuilder().build(cbamVnfd));
97 assertItenticalZips(expectedModifiedCbamPackage, actualModifiedCbamVnfPackage.toByteArray());
101 public void testDownloaderPage() throws Exception {
102 ByteArrayOutputStream bos = new ByteArrayOutputStream();
103 PrintStream actualOut = new PrintStream(bos, true);
104 when(httpResponse.getOutputStream()).thenReturn(new DelegatingServletOutputStream(actualOut));
105 when(systemFunctions.loadFile("upload.html")).thenCallRealMethod();
107 converterApi.getUploadPageForConvertingVnfd(httpResponse);
109 TestCase.assertTrue(Arrays.equals(TestUtil.loadFile("upload.html"), bos.toByteArray()));
110 verify(httpResponse).addHeader(HttpHeaders.CONTENT_LENGTH, Integer.toString(bos.toByteArray().length));