Add multi-oxm using schemaIngest library
[aai/data-router.git] / src / test / java / org / onap / aai / datarouter / 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.datarouter.schema;
22
23 import static org.junit.Assert.assertNotNull;
24 import static org.junit.Assert.assertTrue;
25 import java.util.ArrayList;
26 import org.eclipse.persistence.dynamic.DynamicType;
27 import org.eclipse.persistence.internal.helper.DatabaseField;
28 import org.eclipse.persistence.jaxb.dynamic.DynamicJAXBContext;
29 import org.eclipse.persistence.mappings.DatabaseMapping;
30 import org.junit.Test;
31 import org.onap.aai.datarouter.schema.OxmModelLoader;
32 import org.onap.aai.datarouter.util.EntityOxmReferenceHelper;
33 import org.onap.aai.datarouter.util.ExternalOxmModelProcessor;
34 import org.onap.aai.setup.Version;
35
36 public class OxmModelLoaderTest {
37
38     @Test
39     public void testLoadingMultipleOxmFiles() {
40         ArrayList<ExternalOxmModelProcessor> externalOxmModelProcessors = new ArrayList<ExternalOxmModelProcessor>();
41         externalOxmModelProcessors.add(EntityOxmReferenceHelper.getInstance());
42         OxmModelLoader.registerExternalOxmModelProcessors(externalOxmModelProcessors);
43         OxmModelLoader.loadModels();
44
45         DynamicJAXBContext jaxbContext = OxmModelLoader.getContextForVersion(Version.getLatest().toString());
46
47         DynamicType pserver = jaxbContext.getDynamicType("Pserver");
48         DynamicType genericVnf = jaxbContext.getDynamicType("GenericVnf");
49
50         assertNotNull(pserver);
51         assertNotNull(genericVnf);
52
53         DatabaseMapping mapping = pserver.getDescriptor().getMappings().firstElement();
54         if (mapping.isAbstractDirectMapping()) {
55             DatabaseField f = mapping.getField();
56             String keyName = f.getName().substring(0, f.getName().indexOf("/"));
57             assertTrue(keyName.equals("hostname"));
58         }
59
60         mapping = genericVnf.getDescriptor().getMappings().firstElement();
61         if (mapping.isAbstractDirectMapping()) {
62             DatabaseField f = mapping.getField();
63             String keyName = f.getName().substring(0, f.getName().indexOf("/"));
64             assertTrue(keyName.equals("vnf-id"));
65         }
66
67     }
68 }