gerrit.onap Code Review - policy/common.git/blob

   1 /*
   2  * ============LICENSE_START=======================================================
   3  * ONAP
   4  * ================================================================================
   5  * Copyright (C) 2019 AT&T Intellectual Property. 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.policy.common.endpoints.http.server.test;
  22
  23 import static org.assertj.core.api.Assertions.assertThatThrownBy;
  24 import static org.junit.Assert.assertEquals;
  25 import static org.junit.Assert.assertFalse;
  26 import static org.junit.Assert.assertTrue;
  27
  28 import java.io.ByteArrayInputStream;
  29 import java.io.ByteArrayOutputStream;
  30 import java.io.IOException;
  31 import java.io.InputStream;
  32 import java.io.OutputStream;
  33 import java.util.HashMap;
  34 import java.util.Map;
  35 import javax.ws.rs.core.MediaType;
  36 import org.junit.Before;
  37 import org.junit.Test;
  38 import org.onap.policy.common.endpoints.http.server.YamlMessageBodyHandler;
  39 import org.yaml.snakeyaml.error.YAMLException;
  40
  41 public class YamlMessageBodyHandlerTest {
  42     private static final String EXPECTED_EXCEPTION = "expected exception";
  43
  44     private static final String GEN_TYPE = "some-type";
  45     private static final String[] subtypes = {"yaml"};
  46
  47     @SuppressWarnings("rawtypes")
  48     private static final Class GEN_CLASS = MyObject.class;
  49
  50     @SuppressWarnings("unchecked")
  51     private static final Class<Object> CLASS_OBJ = GEN_CLASS;
  52
  53     private YamlMessageBodyHandler hdlr;
  54
  55     @Before
  56     public void setUp() {
  57         hdlr = new YamlMessageBodyHandler();
  58     }
  59
  60     @Test
  61     public void testIsWriteable() {
  62         for (String subtype : subtypes) {
  63             assertTrue("writeable " + subtype, hdlr.isWriteable(null, null, null, new MediaType(GEN_TYPE, subtype)));
  64
  65         }
  66
  67         // the remaining should be FALSE
  68
  69         // null media type
  70         assertFalse(hdlr.isWriteable(null, null, null, null));
  71
  72         // null subtype
  73         assertFalse(hdlr.isWriteable(null, null, null, new MediaType(GEN_TYPE, null)));
  74
  75         // text subtype
  76         assertFalse(hdlr.isWriteable(null, null, null, MediaType.TEXT_HTML_TYPE));
  77     }
  78
  79     @Test
  80     public void testGetSize() {
  81         assertEquals(-1, hdlr.getSize(null, null, null, null, null));
  82     }
  83
  84     @Test
  85     public void testWriteTo_testReadFrom() throws Exception {
  86         ByteArrayOutputStream outstr = new ByteArrayOutputStream();
  87         MyObject obj1 = new MyObject(10);
  88         hdlr.writeTo(obj1, obj1.getClass(), CLASS_OBJ, null, null, null, outstr);
  89
  90         Object obj2 = hdlr.readFrom(CLASS_OBJ, CLASS_OBJ, null, null, null,
  91                         new ByteArrayInputStream(outstr.toByteArray()));
  92         assertEquals(obj1.toString(), obj2.toString());
  93     }
  94
  95     @Test
  96     public void testWriteTo_DifferentTypes() throws Exception {
  97         ByteArrayOutputStream outstr = new ByteArrayOutputStream();
  98
  99         // use a derived type, but specify the base type when writing
 100         MyObject obj1 = new DerivedObject(10);
 101         hdlr.writeTo(obj1, obj1.getClass(), CLASS_OBJ, null, null, null, outstr);
 102
 103         Object obj2 = hdlr.readFrom(CLASS_OBJ, CLASS_OBJ, null, null, null,
 104                         new ByteArrayInputStream(outstr.toByteArray()));
 105         assertEquals(obj1.toString(), obj2.toString());
 106     }
 107
 108     @Test
 109     public void testWriteTo_Ex() throws Exception {
 110         OutputStream outstr = new OutputStream() {
 111             @Override
 112             public void write(int value) throws IOException {
 113                 throw new IOException(EXPECTED_EXCEPTION);
 114             }
 115         };
 116
 117         MyObject obj1 = new MyObject(10);
 118
 119         assertThatThrownBy(() -> hdlr.writeTo(obj1, obj1.getClass(), CLASS_OBJ, null, null, null, outstr))
 120                         .isInstanceOf(IOException.class);
 121
 122         outstr.close();
 123     }
 124
 125     @Test
 126     public void testIsReadable() {
 127         for (String subtype : subtypes) {
 128             assertTrue("readable " + subtype, hdlr.isReadable(null, null, null, new MediaType(GEN_TYPE, subtype)));
 129
 130         }
 131
 132         // the remaining should be FALSE
 133
 134         // null media type
 135         assertFalse(hdlr.isReadable(null, null, null, null));
 136
 137         // null subtype
 138         assertFalse(hdlr.isReadable(null, null, null, new MediaType(GEN_TYPE, null)));
 139
 140         // text subtype
 141         assertFalse(hdlr.isReadable(null, null, null, MediaType.TEXT_HTML_TYPE));
 142     }
 143
 144     @Test
 145     public void testReadFrom_DifferentTypes() throws Exception {
 146         ByteArrayOutputStream outstr = new ByteArrayOutputStream();
 147         MyObject obj1 = new MyObject(10);
 148         hdlr.writeTo(obj1, obj1.getClass(), CLASS_OBJ, null, null, null, outstr);
 149
 150         // use a derived type, but specify the base type when reading
 151         @SuppressWarnings("rawtypes")
 152         Class clazz = DerivedObject.class;
 153
 154         @SuppressWarnings("unchecked")
 155         Class<Object> objclazz = clazz;
 156
 157         Object obj2 = hdlr.readFrom(objclazz, CLASS_OBJ, null, null, null,
 158                         new ByteArrayInputStream(outstr.toByteArray()));
 159         assertEquals(obj1.toString(), obj2.toString());
 160     }
 161
 162     @Test
 163     public void testReadFrom_Ex() throws Exception {
 164         InputStream inpstr = new InputStream() {
 165             @Override
 166             public int read() throws IOException {
 167                 throw new IOException(EXPECTED_EXCEPTION);
 168             }
 169         };
 170
 171         assertThatThrownBy(() -> hdlr.readFrom(CLASS_OBJ, CLASS_OBJ, null, null, null, inpstr))
 172                         .isInstanceOf(YAMLException.class);
 173
 174         inpstr.close();
 175     }
 176
 177     @Test
 178     public void testReadFrom_Invalid() throws Exception {
 179         InputStream inpstr = new ByteArrayInputStream("plain text".getBytes());
 180
 181         assertThatThrownBy(() -> hdlr.readFrom(CLASS_OBJ, CLASS_OBJ, null, null, null, inpstr))
 182                         .isInstanceOf(YAMLException.class);
 183
 184         inpstr.close();
 185     }
 186
 187     @Test
 188     public void testMapDouble() throws Exception {
 189         MyMap map = new MyMap();
 190         map.props = new HashMap<>();
 191         map.props.put("plainString", "def");
 192         map.props.put("negInt", -10);
 193         map.props.put("doubleVal", 12.5);
 194         map.props.put("posLong", 100000000000L);
 195
 196         ByteArrayOutputStream outstr = new ByteArrayOutputStream();
 197         hdlr.writeTo(map, map.getClass(), map.getClass(), null, null, null, outstr);
 198
 199         Object obj2 = hdlr.readFrom(Object.class, map.getClass(), null, null, null,
 200                         new ByteArrayInputStream(outstr.toByteArray()));
 201         assertEquals(map.toString(), obj2.toString());
 202
 203         map = (MyMap) obj2;
 204
 205         assertEquals(-10, map.props.get("negInt"));
 206         assertEquals(100000000000L, map.props.get("posLong"));
 207         assertEquals(12.5, map.props.get("doubleVal"));
 208     }
 209
 210     public static class DerivedObject extends MyObject {
 211         public DerivedObject(int id) {
 212             super(id);
 213         }
 214     }
 215
 216     public static class MyObject {
 217         private int id;
 218
 219         public MyObject() {
 220             super();
 221         }
 222
 223         public MyObject(int id) {
 224             this.id = id;
 225         }
 226
 227         @Override
 228         public String toString() {
 229             return "MyObject [id=" + id + "]";
 230         }
 231     }
 232
 233     private static class MyMap {
 234         private Map<String, Object> props;
 235
 236         @Override
 237         public String toString() {
 238             return props.toString();
 239         }
 240     }
 241 }