model/basic-model/src/test/java/org/onap/policy/apex/model/basicmodel/handling/ApexModelReaderTest.java

   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 import java.lang.reflect.Field;
  36
  37 import javax.xml.bind.JAXBException;
  38 import javax.xml.bind.PropertyException;
  39 import javax.xml.bind.Unmarshaller;
  40 import javax.xml.transform.stream.StreamSource;
  41
  42 import org.junit.Test;
  43 import org.mockito.Mock;
  44 import org.mockito.Mockito;
  45 import org.mockito.MockitoAnnotations;
  46 import org.onap.policy.apex.model.basicmodel.concepts.ApexException;
  47 import org.onap.policy.apex.model.basicmodel.concepts.AxModel;
  48 import org.onap.policy.apex.model.basicmodel.handling.ApexModelReader;
  49 import org.onap.policy.apex.model.basicmodel.handling.ApexModelWriter;
  50
  51 public class ApexModelReaderTest {
  52     @Mock
  53     private Unmarshaller unmarshallerMock;
  54
  55     @Test
  56     public void testModelReader() throws IOException, ApexException {
  57         AxModel model = new DummyApexBasicModelCreator().getModel();
  58         AxModel invalidModel = new DummyApexBasicModelCreator().getInvalidModel();
  59
  60         ApexModelWriter<AxModel> modelWriter = new ApexModelWriter<AxModel>(AxModel.class);
  61         modelWriter.setValidateFlag(true);
  62         modelWriter.setJsonOutput(true);
  63
  64         ByteArrayOutputStream baos = new ByteArrayOutputStream();
  65         modelWriter.write(model, baos);
  66
  67         ByteArrayOutputStream baosInvalid = new ByteArrayOutputStream();
  68         modelWriter.setValidateFlag(false);
  69         modelWriter.write(invalidModel, baosInvalid);
  70
  71         ApexModelReader<AxModel> modelReader = new ApexModelReader<AxModel>(AxModel.class, true);
  72
  73         modelReader.setValidateFlag(true);
  74         assertTrue(modelReader.getValidateFlag());
  75
  76         ByteArrayInputStream bais = new ByteArrayInputStream(baos.toByteArray());
  77         AxModel readModel = modelReader.read(bais);
  78         assertEquals(model, readModel);
  79
  80         ByteArrayInputStream baisInvalid = new ByteArrayInputStream(baosInvalid.toByteArray());
  81         try {
  82             modelReader.read(baisInvalid);
  83             fail("test should throw an exceptino here");
  84         }
  85         catch (Exception e) {
  86             assertTrue(e.getMessage().startsWith("Apex concept validation failed"));
  87         }
  88
  89         modelReader.setValidateFlag(false);
  90         assertFalse(modelReader.getValidateFlag());
  91
  92         ByteArrayInputStream bais2 = new ByteArrayInputStream(baos.toByteArray());
  93         AxModel readModel2 = modelReader.read(bais2);
  94         assertEquals(model, readModel2);
  95
  96         modelWriter.setJsonOutput(false);
  97
  98         ByteArrayOutputStream baosXml = new ByteArrayOutputStream();
  99         modelWriter.write(model, baosXml);
 100
 101         ByteArrayInputStream baisXml = new ByteArrayInputStream(baosXml.toByteArray());
 102         AxModel readModelXml = modelReader.read(baisXml);
 103         assertEquals(model, readModelXml);
 104
 105         String dummyString = "SomeDummyText";
 106         ByteArrayInputStream baisDummy = new ByteArrayInputStream(dummyString.getBytes());
 107         try {
 108             modelReader.read(baisDummy);
 109             fail("test should throw an exception here");
 110         }
 111         catch (Exception e) {
 112             assertEquals("format of input for Apex concept is neither JSON nor XML", e.getMessage());
 113         }
 114
 115         try {
 116             ByteArrayInputStream nullBais = null;
 117             modelReader.read(nullBais);
 118             fail("test should throw an exception here");
 119         }
 120         catch (Exception e) {
 121             assertEquals("concept stream may not be null", e.getMessage());
 122         }
 123
 124         try {
 125             FileInputStream fis = new FileInputStream(new File("somewhere/over/the/rainbow"));
 126             modelReader.read(fis);
 127             fail("test should throw an exception here");
 128         }
 129         catch (Exception e) {
 130             assertTrue(e.getMessage().contains("rainbow"));
 131         }
 132
 133         File tempFile = File.createTempFile("Apex", "Dummy");
 134         try {
 135             BufferedReader br = new BufferedReader(new FileReader(tempFile));
 136             br.close();
 137             modelReader.read(br);
 138             fail("test should throw an exception here");
 139         }
 140         catch (Exception e) {
 141             assertEquals("Unable to read Apex concept ", e.getMessage());
 142         }
 143         finally {
 144             tempFile.delete();
 145         }
 146
 147         modelReader.setSchema(null);
 148
 149         tempFile = File.createTempFile("Apex", "Dummy");
 150         try {
 151             modelReader.setSchema(tempFile.getCanonicalPath());
 152             fail("test should throw an exception here");
 153         }
 154         catch (Exception e) {
 155             assertEquals("Unable to load schema", e.getMessage());
 156         }
 157         finally {
 158             tempFile.delete();
 159         }
 160
 161         modelReader.setSchema("xml/example.xsd");
 162     }
 163
 164     @Test
 165     public void testSetInputTypeError() throws ApexModelException {
 166         MockitoAnnotations.initMocks(this);
 167
 168         ApexModelReader<AxModel> modelReader = new ApexModelReader<AxModel>(AxModel.class, true);
 169
 170         try {
 171             Field marshallerField = modelReader.getClass().getDeclaredField("unmarshaller");
 172             marshallerField.setAccessible(true);
 173             marshallerField.set(modelReader, unmarshallerMock);
 174             marshallerField.setAccessible(false);
 175         } catch (Exception validationException) {
 176             fail("test should not throw an exception");
 177         }
 178
 179         try {
 180             Mockito.doThrow(new JAXBException("Exception marshalling to JSON")).when(unmarshallerMock)
 181                 .unmarshal((StreamSource)Mockito.anyObject(), Mockito.anyObject());
 182
 183             modelReader.read("{Hello}");
 184             fail("Test should throw an exception here");
 185         } catch (Exception jaxbe) {
 186             assertEquals("Unable to unmarshal Apex concept ", jaxbe.getMessage());
 187         }
 188
 189         try {
 190             Mockito.doThrow(new PropertyException("Exception setting JAXB property")).when(unmarshallerMock)
 191                 .setProperty(Mockito.anyString(), Mockito.anyString());
 192             modelReader.read("{Hello}");
 193             fail("Test should throw an exception here");
 194         } catch (Exception jaxbe) {
 195             assertEquals("JAXB error setting unmarshaller for JSON input", jaxbe.getMessage());
 196         }
 197
 198         try {
 199             Mockito.doThrow(new PropertyException("Exception setting JAXB property")).when(unmarshallerMock)
 200                 .setProperty(Mockito.anyString(), Mockito.anyString());
 201             modelReader.read("<?xml version=\"1.0\" encoding=\"UTF-8\"?>");
 202             fail("Test should throw an exception here");
 203         } catch (Exception jaxbe) {
 204             assertEquals("JAXB error setting unmarshaller for XML input", jaxbe.getMessage());
 205         }
 206     }
 207 }