Initial OpenECOMP A&AI Model Loader commit
[aai/model-loader.git] / src / test / java / org / openecomp / modelloader / entity / model / ModelArtifactParserTest.java
1 /*-
2  * ============LICENSE_START=======================================================
3  * MODEL LOADER SERVICE
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
10  * 
11  *      http://www.apache.org/licenses/LICENSE-2.0
12  * 
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=========================================================
19  */
20
21 package org.openecomp.modelloader.entity.model;
22
23 import static org.junit.Assert.assertTrue;
24
25 import java.nio.file.Files;
26 import java.nio.file.Paths;
27 import java.util.List;
28
29 import org.junit.Test;
30 import org.openecomp.modelloader.entity.Artifact;
31
32 public class ModelArtifactParserTest {
33
34   @Test
35   public void testParseModelFileNoDeps() throws Exception {
36     final String MODEL_FILE = "src/test/resources/models/vnf-model.xml";
37
38     try {
39       byte[] xmlBytes = Files.readAllBytes(Paths.get(MODEL_FILE));
40
41       ModelArtifactParser parser = new ModelArtifactParser();
42       List<Artifact> modelList = parser.parse(xmlBytes, "test-artifact");
43
44       assertTrue(modelList.size() == 1);
45
46       ModelArtifact model = (ModelArtifact) modelList.get(0);
47       System.out.println(model.toString());
48
49       assertTrue(model.getNameVersionId().equalsIgnoreCase("model-vid"));
50       assertTrue(model.getType().toString().equalsIgnoreCase("MODEL"));
51       assertTrue(model.getDependentModelIds().size() == 0);
52     } catch (Exception e) {
53       e.printStackTrace();
54       assertTrue(false);
55     }
56   }
57
58   @Test
59   public void testParseModelFileDeps() throws Exception {
60     final String MODEL_FILE = "src/test/resources/models/wan-connector-model.xml";
61
62     try {
63       byte[] xmlBytes = Files.readAllBytes(Paths.get(MODEL_FILE));
64
65       ModelArtifactParser parser = new ModelArtifactParser();
66       List<Artifact> modelList = parser.parse(xmlBytes, "test-artifact");
67
68       assertTrue(modelList.size() == 1);
69
70       ModelArtifact model = (ModelArtifact) modelList.get(0);
71       System.out.println(model.toString());
72
73       assertTrue(model.getNameVersionId().equalsIgnoreCase("93d9d45d-7eec-4371-9083-675e4c353de3"));
74       assertTrue(model.getType().toString().equalsIgnoreCase("MODEL"));
75       assertTrue(model.getDependentModelIds().size() == 7);
76       assertTrue(model.getDependentModelIds().contains("d09dd9da-0148-46cd-a947-591afc844d24"));
77       assertTrue(model.getDependentModelIds().contains("ae16244f-4d29-4801-a559-e25f2db2a4c3"));
78       assertTrue(model.getDependentModelIds().contains("a6d9de88-4046-4b78-a59e-5691243d292a"));
79     } catch (Exception e) {
80       e.printStackTrace();
81       assertTrue(false);
82     }
83   }
84
85   @Test
86   public void testParseCompleteModel() throws Exception {
87     final String MODEL_FILE = "src/test/resources/models/complete-model.xml";
88
89     try {
90       byte[] xmlBytes = Files.readAllBytes(Paths.get(MODEL_FILE));
91
92       ModelArtifactParser parser = new ModelArtifactParser();
93       List<Artifact> modelList = parser.parse(xmlBytes, "test-artifact");
94
95       for (Artifact art : modelList) {
96         ModelArtifact model = (ModelArtifact) art;
97         System.out.println(model.toString());
98       }
99
100       assertTrue(modelList.size() == 14);
101
102       ModelArtifact modelVdc = (ModelArtifact) modelList.get(0);
103       assertTrue(
104           modelVdc.getNameVersionId().equalsIgnoreCase("997fc7-fca1-451f-b953-9a1e6197b4d6"));
105       assertTrue(modelVdc.getType().toString().equalsIgnoreCase("MODEL"));
106       assertTrue(modelVdc.getDependentModelIds().size() == 1);
107       assertTrue(modelVdc.getDependentModelIds().contains("93d9d45d-7eec-4371-9083-675e4c353de3"));
108
109       ModelArtifact modelPserver = (ModelArtifact) modelList.get(9);
110       assertTrue(
111           modelPserver.getNameVersionId().equalsIgnoreCase("f2b24d95-c582-48d5-b2d6-c5b3a94ce812"));
112       assertTrue(modelPserver.getType().toString().equalsIgnoreCase("MODEL"));
113       assertTrue(modelPserver.getDependentModelIds().size() == 2);
114       assertTrue(
115           modelPserver.getDependentModelIds().contains("35be1acf-1298-48c6-a128-66850083b8bd"));
116       assertTrue(
117           modelPserver.getDependentModelIds().contains("759dbd4a-2473-46f3-a932-48d987c9b4a1"));
118     } catch (Exception e) {
119       e.printStackTrace();
120       assertTrue(false);
121     }
122   }
123
124   @Test
125   public void testParseNamedQuery() throws Exception {
126     final String MODEL_FILE = "src/test/resources/models/named-query-wan-connector.xml";
127
128     try {
129       byte[] xmlBytes = Files.readAllBytes(Paths.get(MODEL_FILE));
130
131       ModelArtifactParser parser = new ModelArtifactParser();
132       List<Artifact> modelList = parser.parse(xmlBytes, "test-artifact");
133
134       assertTrue(modelList.size() == 1);
135
136       ModelArtifact model = (ModelArtifact) modelList.get(0);
137       System.out.println(model.toString());
138
139       assertTrue(model.getNameVersionId().equalsIgnoreCase("94cac189-8d88-4d63-a194-f44214e080ff"));
140       assertTrue(model.getType().toString().equalsIgnoreCase("NAMED_QUERY"));
141       assertTrue(model.getDependentModelIds().size() == 4);
142       assertTrue(model.getDependentModelIds().contains("d09dd9da-0148-46cd-a947-591afc844d24"));
143       assertTrue(model.getDependentModelIds().contains("997fc7-fca1-451f-b953-9a1e6197b4d6"));
144       assertTrue(model.getDependentModelIds().contains("897df7ea-8938-42b0-bc57-46e913a4d93b"));
145       assertTrue(model.getDependentModelIds().contains("f2b24d95-c582-48d5-b2d6-c5b3a94ce812"));
146     } catch (Exception e) {
147       e.printStackTrace();
148       assertTrue(false);
149     }
150   }
151 }