Update router core new schema ingest lib
[aai/router-core.git] / src / test / java / org / onap / aai / schema / OxmModelLoaderTest.java
1 /**
2  * ============LICENSE_START=======================================================
3  * org.onap.aai
4  * ================================================================================
5  * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
6  * Copyright © 2017-2018 Amdocs
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
11  *
12  *       http://www.apache.org/licenses/LICENSE-2.0
13  *
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=========================================================
20  */
21 package org.onap.aai.schema;
22
23
24 import static org.junit.Assert.assertNotNull;
25 import static org.junit.Assert.assertTrue;
26
27 import java.util.ArrayList;
28
29 import org.eclipse.persistence.dynamic.DynamicType;
30 import org.eclipse.persistence.internal.helper.DatabaseField;
31 import org.eclipse.persistence.jaxb.dynamic.DynamicJAXBContext;
32 import org.eclipse.persistence.mappings.DatabaseMapping;
33 import org.junit.Test;
34 import org.junit.runner.RunWith;
35 import org.onap.aai.nodes.NodeIngestor;
36 import org.onap.aai.setup.SchemaLocationsBean;
37 import org.onap.aai.setup.SchemaVersions;
38 import org.onap.aai.util.EntityOxmReferenceHelper;
39 import org.onap.aai.util.ExternalOxmModelProcessor;
40 import org.springframework.beans.factory.annotation.Autowired;
41 import org.springframework.test.context.ContextConfiguration;
42 import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
43
44 @RunWith(SpringJUnit4ClassRunner.class)
45 @ContextConfiguration("file:src/test/resources/spring-beans/data-router-oxm.xml")
46 public class OxmModelLoaderTest {
47
48     @Autowired
49     private SchemaVersions schemaVersions;
50     @Autowired
51     private SchemaLocationsBean schemaLocationsBean;
52     @Autowired
53     private NodeIngestor nodeIngestor;
54
55     @Test
56     public void testLoadingMultipleOxmFiles() {
57
58         ArrayList<ExternalOxmModelProcessor> externalOxmModelProcessors = new ArrayList<ExternalOxmModelProcessor>();
59         externalOxmModelProcessors.add(EntityOxmReferenceHelper.getInstance());
60         OxmModelLoader.registerExternalOxmModelProcessors(externalOxmModelProcessors);
61         OxmModelLoader.setNodeIngestor(nodeIngestor);
62         OxmModelLoader.loadModels(schemaVersions, schemaLocationsBean);
63
64         DynamicJAXBContext jaxbContext = OxmModelLoader.getContextForVersion("v13", schemaVersions, schemaLocationsBean);
65
66         DynamicType pserver = jaxbContext.getDynamicType("Pserver");
67         DynamicType genericVnf = jaxbContext.getDynamicType("GenericVnf");
68
69         assertNotNull(pserver);
70         assertNotNull(genericVnf);
71
72         DatabaseMapping mapping = pserver.getDescriptor().getMappings().firstElement();
73         if (mapping.isAbstractDirectMapping()) {
74             DatabaseField f = mapping.getField();
75             String keyName = f.getName().substring(0, f.getName().indexOf("/"));
76             assertTrue(keyName.equals("hostname"));
77         }
78
79         mapping = genericVnf.getDescriptor().getMappings().firstElement();
80         if (mapping.isAbstractDirectMapping()) {
81             DatabaseField f = mapping.getField();
82             String keyName = f.getName().substring(0, f.getName().indexOf("/"));
83             assertTrue(keyName.equals("vnf-id"));
84         }
85       
86     }
87     
88 }