Extend service model
[aai/babel.git] / src / test / java / org / onap / aai / babel / xml / generator / api / AaiModelGeneratorTest.java
1 /**
2  * ============LICENSE_START=======================================================
3  * org.onap.aai
4  * ================================================================================
5  * Copyright (c) 2019 Nokia. 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.onap.aai.babel.xml.generator.api;
22
23
24 import org.apache.groovy.util.Maps;
25 import org.junit.Test;
26 import org.onap.aai.babel.util.ArtifactTestUtils;
27 import org.onap.aai.babel.xml.generator.XmlArtifactGenerationException;
28 import org.onap.aai.babel.xml.generator.model.Model;
29 import org.onap.aai.babel.xml.generator.model.Service;
30
31 import java.io.IOException;
32
33 import static org.assertj.core.api.Assertions.assertThat;
34
35 public class AaiModelGeneratorTest {
36
37     private AaiModelGenerator generator = new AaiModelGenerator();
38
39     @Test
40     public void shouldGenerateModelWithInstantiationType() throws XmlArtifactGenerationException, IOException {
41         new ArtifactTestUtils().loadWidgetMappings();
42         Model model = new Service();
43         model.populateModelIdentificationInformation(Maps.of("instantiationType", "macro"));
44
45
46         String generatedXml = generator.generateModelFor(model);
47
48         assertThat(generatedXml).containsSubsequence("   <model-vers>\n" +
49                 "        <model-ver>\n" +
50                 "            <orchestration-type>macro</orchestration-type>");
51     }
52 }