2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
6 * Copyright © 2017-2018 European Software Marketing Ltd.
7 * ================================================================================
8 * Licensed under the Apache License, Version 2.0 (the "License");
9 * you may not use this file except in compliance with the License.
10 * You may obtain a copy of the License at
12 * http://www.apache.org/licenses/LICENSE-2.0
14 * Unless required by applicable law or agreed to in writing, software
15 * distributed under the License is distributed on an "AS IS" BASIS,
16 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17 * See the License for the specific language governing permissions and
18 * limitations under the License.
19 * ============LICENSE_END=========================================================
22 package org.onap.aai.babel.testdata;
24 import com.google.gson.Gson;
25 import java.io.IOException;
26 import java.util.List;
27 import org.onap.aai.babel.csar.extractor.InvalidArchiveException;
28 import org.onap.aai.babel.csar.extractor.YamlExtractor;
29 import org.onap.aai.babel.csar.vnfcatalog.ToscaToCatalogException;
30 import org.onap.aai.babel.csar.vnfcatalog.VnfVendorImageExtractor;
31 import org.onap.aai.babel.service.data.BabelArtifact;
32 import org.onap.aai.babel.service.data.BabelRequest;
33 import org.onap.aai.babel.util.ArtifactTestUtils;
34 import org.onap.aai.babel.xml.generator.data.Artifact;
35 import org.onap.aai.babel.xml.generator.data.GeneratorUtil;
37 public enum CsarTest {
39 VNF_VENDOR_CSAR("catalog_csar.csar"),
40 NO_VNF_CONFIG_CSAR("noVnfConfiguration.csar"),
41 SD_WAN_CSAR_FILE("service-SdWanServiceTest-csar.csar"),
42 MISSING_METADATA_CSAR("service-MissingMetadataTest.csar"),
43 NO_YAML_FILES("noYmlFilesArchive.zip"),
44 PORT_MIRROR_CSAR("service_PortMirror.csar"),
45 VNFOD_SERVICE("service-Dev2devnfodservice17July-csar.csar"),
46 MULTIPLE_VNF_CSAR("catalog_csar_too_many_vnfConfigurations.csar");
49 private String filename;
50 private ArtifactTestUtils artifactTestUtils = new ArtifactTestUtils();
52 CsarTest(String filename) {
53 this.filename = filename;
56 public String getName() {
60 public byte[] getContent() throws IOException {
61 return artifactTestUtils.getCompressedArtifact(filename);
65 * Extract YAML Artifacts.
67 * @return the extracted artifacts
68 * @throws InvalidArchiveException if the CSAR is invalid
69 * @throws IOException for I/O errors
71 public List<Artifact> extractArtifacts() throws InvalidArchiveException, IOException {
72 return new YamlExtractor().extract(getContent(), getName(), "v1");
76 * Extract VNF Vendor Image Artifacts.
78 * @return the extracted artifacts
80 * @throws ToscaToCatalogException
83 public BabelArtifact extractVnfVendorImages() throws ToscaToCatalogException, IOException {
84 return new VnfVendorImageExtractor().extract(getContent());
87 public String getJsonRequest() throws IOException {
88 BabelRequest request = new BabelRequest();
89 request.setArtifactName(getName());
90 request.setArtifactVersion("1.0");
91 request.setCsar(new String(GeneratorUtil.encode(getContent())));
92 return new Gson().toJson(request);