2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
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.
18 * ============LICENSE_END=========================================================
21 package org.openecomp.sdc.tosca.services.impl;
23 import org.junit.Assert;
24 import org.junit.Test;
25 import org.openecomp.core.utilities.file.FileContentHandler;
26 import org.openecomp.sdc.tosca.datatypes.ToscaServiceModel;
27 import org.openecomp.sdc.tosca.datatypes.model.ServiceTemplate;
28 import org.openecomp.sdc.tosca.services.ToscaConstants;
29 import org.openecomp.sdc.tosca.services.ToscaUtil;
32 import java.io.FileOutputStream;
33 import java.io.IOException;
34 import java.nio.file.Files;
35 import java.nio.file.Paths;
36 import java.util.Enumeration;
37 import java.util.HashMap;
39 import java.util.zip.ZipEntry;
40 import java.util.zip.ZipFile;
42 public class ToscaFileOutputServiceCsarImplTest {
44 private ToscaFileOutputServiceCsarImpl toscaFileOutputServiceCSARImpl =
45 new ToscaFileOutputServiceCsarImpl();
48 public void testCreationMetaFile() {
49 String createdMeta = toscaFileOutputServiceCSARImpl.createMetaFile("entryFile.yaml");
51 "TOSCA-Meta-File-Version: 1.0\n" +
52 "CSAR-Version: 1.1\n" +
53 "Created-By: ASDC Onboarding portal\n" +
54 "Entry-Definitions: Definitions" + File.separator + "entryFile.yaml";
55 Assert.assertEquals(createdMeta.replaceAll("\\s+", ""), expectedMeta.replaceAll("\\s+", ""));
59 public void testCSARFileCreationWithExternalArtifacts() throws IOException {
60 ServiceTemplate mainServiceTemplate = new ServiceTemplate();
61 Map<String, String> metadata1 = new HashMap<>();
62 metadata1.put("Template_author", "OPENECOMP");
63 metadata1.put(ToscaConstants.ST_METADATA_TEMPLATE_NAME,"ST1");
64 metadata1.put("Template_version", "1.0.0");
65 mainServiceTemplate.setMetadata(metadata1);
66 mainServiceTemplate.setTosca_definitions_version("tosca_simple_yaml_1_0_0");
67 mainServiceTemplate.setDescription("testing desc tosca service template");
69 ServiceTemplate additionalServiceTemplate = new ServiceTemplate();
70 Map<String, String> metadata2 = new HashMap<>();
71 metadata2.put("Template_author", "OPENECOMP");
72 metadata2.put(ToscaConstants.ST_METADATA_TEMPLATE_NAME, "ST2");
73 metadata2.put("Template_version", "1.0.0");
74 additionalServiceTemplate.setTosca_definitions_version("tosca_simple_yaml_1_0_0");
75 additionalServiceTemplate.setDescription("testing desc tosca service template");
76 additionalServiceTemplate.setMetadata(metadata2);
78 Map<String, ServiceTemplate> definitionsInput = new HashMap<>();
80 .put(ToscaUtil.getServiceTemplateFileName(mainServiceTemplate), mainServiceTemplate);
81 definitionsInput.put(ToscaUtil.getServiceTemplateFileName(additionalServiceTemplate),
82 additionalServiceTemplate);
85 Map<String, byte[]> dummyHeatArtifacts = new HashMap<>();
86 String file1Content = "this is file number 1";
87 String file2Content = "this is file number 2";
88 String file1 = "file1.xml";
89 dummyHeatArtifacts.put(file1, file1Content.getBytes());
90 String file2 = "file2.yml";
91 dummyHeatArtifacts.put(file2, file2Content.getBytes());
94 FileContentHandler heatFiles = new FileContentHandler();
95 heatFiles.putAll(dummyHeatArtifacts);
96 Map<String, byte[]> licenseArtifacts = new HashMap<>();
98 FileContentHandler licenseArtifactsFiles = new FileContentHandler();
100 licenseArtifacts.put(
101 ToscaFileOutputServiceCsarImpl.EXTERNAL_ARTIFACTS_FOLDER_NAME + File.separator +
102 "license-file-1.xml", file1Content.getBytes());
103 licenseArtifacts.put(
104 ToscaFileOutputServiceCsarImpl.EXTERNAL_ARTIFACTS_FOLDER_NAME + File.separator +
105 "license-file-2.xml", file1Content.getBytes());
107 licenseArtifactsFiles.putAll(licenseArtifacts);
109 byte[] csarFile = toscaFileOutputServiceCSARImpl.createOutputFile(
110 new ToscaServiceModel(heatFiles, definitionsInput,
111 ToscaUtil.getServiceTemplateFileName(mainServiceTemplate)), licenseArtifactsFiles);
113 String resultFileName = "resultFile.zip";
114 File file = new File(resultFileName);
115 FileOutputStream fos = new FileOutputStream(file);
119 ZipFile zipFile = new ZipFile(resultFileName);
121 Enumeration<? extends ZipEntry> entries = zipFile.entries();
124 while (entries.hasMoreElements()) {
126 entries.nextElement();
128 Assert.assertEquals(7, count);
130 Files.delete(Paths.get(file.getPath()));
134 public void testCSARFileCreation_noArtifacts() throws IOException {
135 ServiceTemplate serviceTemplate = new ServiceTemplate();
136 Map<String, String> metadata = new HashMap<>();
137 metadata.put("Template_author", "OPENECOMP");
138 metadata.put(ToscaConstants.ST_METADATA_TEMPLATE_NAME, "Test");
139 metadata.put("Template_version", "1.0.0");
140 serviceTemplate.setTosca_definitions_version("tosca_simple_yaml_1_0_0");
141 serviceTemplate.setDescription("testing desc tosca service template");
142 serviceTemplate.setMetadata(metadata);
143 Map<String, ServiceTemplate> definitionsInput = new HashMap<>();
144 String serviceTemplateFileName = ToscaUtil.getServiceTemplateFileName(serviceTemplate);
145 definitionsInput.put(serviceTemplateFileName, serviceTemplate);
146 byte[] csarFile = toscaFileOutputServiceCSARImpl
147 .createOutputFile(new ToscaServiceModel(null, definitionsInput, serviceTemplateFileName),
151 String resultFileName = "resultFile.zip";
152 File file = new File(resultFileName);
153 FileOutputStream fos = new FileOutputStream(file);
157 ZipFile zipFile = new ZipFile(resultFileName);
159 Enumeration<? extends ZipEntry> entries = zipFile.entries();
162 while (entries.hasMoreElements()) {
164 entries.nextElement();
166 Assert.assertEquals(2, count);
168 Files.delete(Paths.get(file.getPath()));