20815ae72781938dde6958f771f06da55b2bb96e
[policy/apex-pdp.git] /
1 /*
2  * ============LICENSE_START=======================================================
3  *  Copyright (C) 2016-2018 Ericsson. All rights reserved.
4  * ================================================================================
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at
8  * 
9  *      http://www.apache.org/licenses/LICENSE-2.0
10  * 
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  * 
17  * SPDX-License-Identifier: Apache-2.0
18  * ============LICENSE_END=========================================================
19  */
20
21 package org.onap.policy.apex.model.basicmodel.handling;
22
23 import static org.junit.Assert.assertEquals;
24 import static org.junit.Assert.assertFalse;
25 import static org.junit.Assert.assertTrue;
26 import static org.junit.Assert.fail;
27
28 import java.io.BufferedReader;
29 import java.io.ByteArrayInputStream;
30 import java.io.ByteArrayOutputStream;
31 import java.io.File;
32 import java.io.FileInputStream;
33 import java.io.FileReader;
34 import java.io.IOException;
35
36 import org.junit.Test;
37 import org.onap.policy.apex.model.basicmodel.concepts.ApexException;
38 import org.onap.policy.apex.model.basicmodel.concepts.AxModel;
39 import org.onap.policy.apex.model.basicmodel.handling.ApexModelReader;
40 import org.onap.policy.apex.model.basicmodel.handling.ApexModelWriter;
41
42 public class ApexModelReaderTest {
43
44     @Test
45     public void testModelReader() throws IOException, ApexException {
46         AxModel model = new DummyApexBasicModelCreator().getModel();
47         AxModel invalidModel = new DummyApexBasicModelCreator().getInvalidModel();
48         
49         ApexModelWriter<AxModel> modelWriter = new ApexModelWriter<AxModel>(AxModel.class);
50         modelWriter.setValidateFlag(true);
51         modelWriter.setJsonOutput(true);
52         
53         ByteArrayOutputStream baos = new ByteArrayOutputStream();
54         modelWriter.write(model, baos);
55         
56         ByteArrayOutputStream baosInvalid = new ByteArrayOutputStream();
57         modelWriter.setValidateFlag(false);
58         modelWriter.write(invalidModel, baosInvalid);
59         
60         ApexModelReader<AxModel> modelReader = new ApexModelReader<AxModel>(AxModel.class, true);
61         
62         modelReader.setValidateFlag(true);
63         assertTrue(modelReader.getValidateFlag());
64         
65         ByteArrayInputStream bais = new ByteArrayInputStream(baos.toByteArray());
66         AxModel readModel = modelReader.read(bais);
67         assertEquals(model, readModel);
68         
69         ByteArrayInputStream baisInvalid = new ByteArrayInputStream(baosInvalid.toByteArray());
70         try {
71             modelReader.read(baisInvalid);
72             fail("test should throw an exceptino here");
73         }
74         catch (Exception e) {
75             assertTrue(e.getMessage().startsWith("Apex concept validation failed"));
76         }
77         
78         modelReader.setValidateFlag(false);
79         assertFalse(modelReader.getValidateFlag());
80         
81         ByteArrayInputStream bais2 = new ByteArrayInputStream(baos.toByteArray());
82         AxModel readModel2 = modelReader.read(bais2);
83         assertEquals(model, readModel2);
84         
85         modelWriter.setJsonOutput(false);
86         
87         ByteArrayOutputStream baosXml = new ByteArrayOutputStream();
88         modelWriter.write(model, baosXml);
89         
90         ByteArrayInputStream baisXml = new ByteArrayInputStream(baosXml.toByteArray());
91         AxModel readModelXml = modelReader.read(baisXml);
92         assertEquals(model, readModelXml);
93         
94         String dummyString = "SomeDummyText";
95         ByteArrayInputStream baisDummy = new ByteArrayInputStream(dummyString.getBytes());
96         try {
97             modelReader.read(baisDummy);
98             fail("test should throw an exception here");
99         }
100         catch (Exception e) {
101             assertEquals("format of input for Apex concept is neither JSON nor XML", e.getMessage());
102         }
103         
104         try {
105             ByteArrayInputStream nullBais = null;
106             modelReader.read(nullBais);
107             fail("test should throw an exception here");
108         }
109         catch (Exception e) {
110             assertEquals("concept stream may not be null", e.getMessage());
111         }
112         
113         try {
114             FileInputStream fis = new FileInputStream(new File("somewhere/over/the/rainbow"));
115             modelReader.read(fis);
116             fail("test should throw an exception here");
117         }
118         catch (Exception e) {
119             assertTrue(e.getMessage().contains("rainbow"));
120         }
121         
122         File tempFile = File.createTempFile("Apex", "Dummy");
123         try {
124             BufferedReader br = new BufferedReader(new FileReader(tempFile));
125             br.close();
126             modelReader.read(br);
127             fail("test should throw an exception here");
128         }
129         catch (Exception e) {
130             assertEquals("Unable to read Apex concept ", e.getMessage());
131         }
132         finally {
133             tempFile.delete();
134         }
135         
136         modelReader.setSchema(null);
137         
138         tempFile = File.createTempFile("Apex", "Dummy");
139         try {
140             modelReader.setSchema(tempFile.getCanonicalPath());
141             fail("test should throw an exception here");
142         }
143         catch (Exception e) {
144             assertEquals("Unable to load schema", e.getMessage());
145         }
146         finally {
147             tempFile.delete();
148         }
149         
150         modelReader.setSchema("xml/example.xsd");
151     }
152 }