1 /*
   2  * Copyright (c) 2012, 2013, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.  Oracle designates this
   8  * particular file as subject to the "Classpath" exception as provided
   9  * by Oracle in the LICENSE file that accompanied this code.
  10  *
  11  * This code is distributed in the hope that it will be useful, but WITHOUT
  12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  14  * version 2 for more details (a copy is included in the LICENSE file that
  15  * accompanied this code).
  16  *
  17  * You should have received a copy of the GNU General Public License version
  18  * 2 along with this work; if not, write to the Free Software Foundation,
  19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  20  *
  21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  22  * or visit www.oracle.com if you need additional information or have any
  23  * questions.
  24  */
  25 
  26 /*
  27  * Copyright (c) 2011-2012, Stephen Colebourne & Michael Nascimento Santos
  28  *
  29  * All rights reserved.
  30  *
  31  * Redistribution and use in source and binary forms, with or without
  32  * modification, are permitted provided that the following conditions are met:
  33  *
  34  *  * Redistributions of source code must retain the above copyright notice,
  35  *    this list of conditions and the following disclaimer.
  36  *
  37  *  * Redistributions in binary form must reproduce the above copyright notice,
  38  *    this list of conditions and the following disclaimer in the documentation
  39  *    and/or other materials provided with the distribution.
  40  *
  41  *  * Neither the name of JSR-310 nor the names of its contributors
  42  *    may be used to endorse or promote products derived from this software
  43  *    without specific prior written permission.
  44  *
  45  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  46  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  47  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  48  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
  49  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  50  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  51  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  52  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  53  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
  54  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  55  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  56  */
  57 package tck.java.time;
  58 
  59 import static org.testng.Assert.assertEquals;
  60 import static org.testng.Assert.assertSame;
  61 
  62 import java.io.ByteArrayInputStream;
  63 import java.io.ByteArrayOutputStream;
  64 import java.io.DataInputStream;
  65 import java.io.IOException;
  66 import java.io.ObjectInputStream;
  67 import java.io.ObjectOutputStream;
  68 import java.io.ObjectStreamConstants;
  69 import java.io.Serializable;
  70 import java.lang.reflect.Field;
  71 
  72 /**
  73  * Base test class.
  74  */
  75 public abstract class AbstractTCKTest {
  76 
  77     protected static boolean isIsoLeap(long year) {
  78         if (year % 4 != 0) {
  79             return false;
  80         }
  81         if (year % 100 == 0 && year % 400 != 0) {
  82             return false;
  83         }
  84         return true;
  85     }
  86 
  87     protected static void assertSerializable(Object object) throws IOException, ClassNotFoundException {
  88         assertEquals(object instanceof Serializable, true);
  89         Object deserializedObject = writeThenRead(object);
  90         assertEquals(deserializedObject, object);
  91     }
  92 
  93     protected static void assertSerializableSame(Object object) throws IOException, ClassNotFoundException {
  94         assertEquals(object instanceof Serializable, true);
  95         Object deserializedObject = writeThenRead(object);
  96         assertSame(deserializedObject, object);
  97     }
  98 
  99     private static Object writeThenRead(Object object) throws IOException, ClassNotFoundException {
 100         ByteArrayOutputStream baos = new ByteArrayOutputStream();
 101         try (ObjectOutputStream oos = new ObjectOutputStream(baos) ) {
 102             oos.writeObject(object);
 103         }
 104         try (ObjectInputStream ois = new ObjectInputStream(new ByteArrayInputStream(baos.toByteArray()))) {
 105             return ois.readObject();
 106         }
 107     }
 108 
 109     protected static void assertSerializedBySer(Object object, byte[] expectedBytes, byte[]... matches) throws Exception {
 110         String serClass = object.getClass().getPackage().getName() + ".Ser";
 111         Class<?> serCls = Class.forName(serClass);
 112         Field field = serCls.getDeclaredField("serialVersionUID");
 113         field.setAccessible(true);
 114         long serVer = (Long) field.get(null);
 115         ByteArrayOutputStream baos = new ByteArrayOutputStream();
 116         try (ObjectOutputStream oos = new ObjectOutputStream(baos) ) {
 117             oos.writeObject(object);
 118         }
 119         byte[] bytes = baos.toByteArray();
 120         ByteArrayInputStream bais = new ByteArrayInputStream(bytes);
 121         try (DataInputStream dis = new DataInputStream(bais)) {
 122             assertEquals(dis.readShort(), ObjectStreamConstants.STREAM_MAGIC);
 123             assertEquals(dis.readShort(), ObjectStreamConstants.STREAM_VERSION);
 124             assertEquals(dis.readByte(), ObjectStreamConstants.TC_OBJECT);
 125             assertEquals(dis.readByte(), ObjectStreamConstants.TC_CLASSDESC);
 126             assertEquals(dis.readUTF(), serClass);
 127             assertEquals(dis.readLong(), serVer);
 128             assertEquals(dis.readByte(), ObjectStreamConstants.SC_EXTERNALIZABLE | ObjectStreamConstants.SC_BLOCK_DATA);
 129             assertEquals(dis.readShort(), 0);  // number of fields
 130             assertEquals(dis.readByte(), ObjectStreamConstants.TC_ENDBLOCKDATA);  // end of classdesc
 131             assertEquals(dis.readByte(), ObjectStreamConstants.TC_NULL);  // no superclasses
 132             if (expectedBytes.length < 256) {
 133                 assertEquals(dis.readByte(), ObjectStreamConstants.TC_BLOCKDATA);
 134                 assertEquals(dis.readUnsignedByte(), expectedBytes.length);
 135             } else {
 136                 assertEquals(dis.readByte(), ObjectStreamConstants.TC_BLOCKDATALONG);
 137                 assertEquals(dis.readInt(), expectedBytes.length);
 138             }
 139             byte[] input = new byte[expectedBytes.length];
 140             dis.readFully(input);
 141             assertEquals(input, expectedBytes);
 142             if (matches.length > 0) {
 143                 for (byte[] match : matches) {
 144                     boolean matched = false;
 145                     while (matched == false) {
 146                         try {
 147                             dis.mark(1000);
 148                             byte[] possible = new byte[match.length];
 149                             dis.readFully(possible);
 150                             assertEquals(possible, match);
 151                             matched = true;
 152                         } catch (AssertionError ex) {
 153                             dis.reset();
 154                             dis.readByte();  // ignore
 155                         }
 156                     }
 157                 }
 158             } else {
 159                 assertEquals(dis.readByte(), ObjectStreamConstants.TC_ENDBLOCKDATA);  // end of blockdata
 160                 assertEquals(dis.read(), -1);
 161             }
 162         }
 163     }
 164 
 165 }