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 PNF_VENDOR_CSAR("service-Testpnfsvc-csar.csar"),
41 NO_VNF_CONFIG_CSAR("noVnfConfiguration.csar"),
42 SD_WAN_CSAR_FILE("service-SdWanServiceTest-csar.csar"),
43 COS_AVPN_CSAR_FILE("service_CosAvpn_csar.csar"),
44 MISSING_METADATA_CSAR("service-MissingMetadataTest.csar"),
45 NO_YAML_FILES("noYmlFilesArchive.zip"),
46 PORT_MIRROR_CSAR("service_PortMirror.csar"),
47 MULTIPLE_VNF_CSAR("catalog_csar_too_many_vnfConfigurations.csar"),
48 NETWORK_COLLECTION_CSAR_FILE("service_NetworkCollection.csar"),
49 RG_COLLECTOR_615_CSAR_FILE("service-RgCollector615-csar.csar"),
50 VNFOD_SERVICE("service-Dev2devnfodservice17July-csar.csar"),
51 CHILD_RESOURCE_CSAR_FILE("service-NetworkCloudVnfServiceMock-csar.csar"),
52 SERVICE_PROXY_CSAR_FILE("service-S1-csar.csar");
56 private String filename;
57 private ArtifactTestUtils artifactTestUtils = new ArtifactTestUtils();
59 CsarTest(String filename) {
60 this.filename = filename;
63 public String getName() {
67 public byte[] getContent() throws IOException {
68 return artifactTestUtils.getCompressedArtifact(filename);
72 * Extract YAML Artifacts.
74 * @return the extracted artifacts
75 * @throws InvalidArchiveException
76 * if the CSAR is invalid
80 public List<Artifact> extractArtifacts() throws InvalidArchiveException, IOException {
81 return new YamlExtractor().extract(getContent(), getName(), "v1");
85 * Extract VNF Vendor Image Artifacts.
87 * @return the extracted artifacts
88 * @throws ToscaToCatalogException
89 * if the CSAR content is not valid
91 * if an I/O exception occursSince:
93 public BabelArtifact extractVnfVendorImages() throws ToscaToCatalogException, IOException {
94 return new VnfVendorImageExtractor().extract(getContent());
98 * Create a BabelRequest containing the encoded CSAR content.
100 * @return a new Babel request for this CSAR
101 * @throws IOException
102 * if an I/O exception occurs
104 public String getJsonRequest() throws IOException {
105 BabelRequest request = new BabelRequest();
106 request.setArtifactName(getName());
107 request.setArtifactVersion("1.0");
108 request.setCsar(new String(GeneratorUtil.encode(getContent())));
109 return new Gson().toJson(request);
113 * Create a BabelRequest containing the encoded CSAR content by passing in the artifact version.
115 * @return a new Babel request for this CSAR
116 * @throws IOException
117 * if an I/O exception occurs
119 public String getJsonRequestWithArtifactVersion(String artifactVersion) throws IOException {
120 BabelRequest request = new BabelRequest();
121 request.setArtifactName(getName());
122 request.setArtifactVersion(artifactVersion);
123 request.setCsar(new String(GeneratorUtil.encode(getContent())));
124 return new Gson().toJson(request);