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 MULTIPLE_VNF_CSAR("catalog_csar_too_many_vnfConfigurations.csar"),
46 NETWORK_COLLECTION_CSAR_FILE("service_NetworkCollection.csar"),
47 SERVICE_PROXY_CSAR_FILE("service-S1-csar.csar");
50 private String filename;
51 private ArtifactTestUtils artifactTestUtils = new ArtifactTestUtils();
53 CsarTest(String filename) {
54 this.filename = filename;
57 public String getName() {
61 public byte[] getContent() throws IOException {
62 return artifactTestUtils.getCompressedArtifact(filename);
66 * Extract YAML Artifacts.
68 * @return the extracted artifacts
69 * @throws InvalidArchiveException if the CSAR is invalid
70 * @throws IOException for I/O errors
72 public List<Artifact> extractArtifacts() throws InvalidArchiveException, IOException {
73 return new YamlExtractor().extract(getContent(), getName(), "v1");
77 * Extract VNF Vendor Image Artifacts.
79 * @return the extracted artifacts
81 * @throws ToscaToCatalogException
84 public BabelArtifact extractVnfVendorImages() throws ToscaToCatalogException, IOException {
85 return new VnfVendorImageExtractor().extract(getContent());
88 public String getJsonRequest() throws IOException {
89 BabelRequest request = new BabelRequest();
90 request.setArtifactName(getName());
91 request.setArtifactVersion("1.0");
92 request.setCsar(new String(GeneratorUtil.encode(getContent())));
93 return new Gson().toJson(request);