nl.vpro.test.util.serialize.SerializeTestUtil Maven / Gradle / Ivy

Go to download
/*
 * Copyright (C) 2012 All rights reserved
 * VPRO The Netherlands
 */
package nl.vpro.test.util.serialize;

import java.io.*;

import org.assertj.core.api.AbstractObjectAssert;
import org.assertj.core.api.Assertions;

import static org.assertj.core.api.Assertions.assertThat;


/**
 *
 * @author Michiel Meeuwissen
 * @since 0.50
 */
public class SerializeTestUtil {

    public static  byte[] serialize(T object) throws IOException {
        ByteArrayOutputStream bytes  = new ByteArrayOutputStream();
        ObjectOutputStream out = new ObjectOutputStream(bytes);
        out.writeObject(object);
        out.close();
        return bytes.toByteArray();

    }

    @SuppressWarnings("unchecked")
    public static  T deserialize(byte[] bytes, Class clazz) throws IOException {
        try(ObjectInputStream in = new ObjectInputStream(new ByteArrayInputStream(bytes))) {
            return (T) in.readObject();
        } catch (ClassNotFoundException e) {
            throw new RuntimeException(e);
        }
    }

    @SuppressWarnings("unchecked")
    public static  T roundTrip(T input) throws IOException {
        return (T) deserialize(serialize(input), input.getClass());
    }

    @SuppressWarnings("unchecked")
    public static  T roundTripAndEquals(T input) throws IOException {
        T newT = (T) deserialize(serialize(input), input.getClass());
        assertThat(newT).isEqualTo(input);
        return newT;

    }

    public static  SerializeTestUtil.ObjectAssert assertThatSerialized(T o) {
        return new SerializeTestUtil.ObjectAssert<>(o);
    }


    public static class ObjectAssert, A> extends AbstractObjectAssert {

        A rounded;

        protected ObjectAssert(A actual) {
            super(actual, SerializeTestUtil.ObjectAssert.class);
        }



        public AbstractObjectAssert andRounded() throws IOException {
            if (rounded == null) {
                rounded = roundTrip(actual);
            }
            return Assertions.assertThat(rounded);
        }

        public A get() throws IOException {
            if (rounded == null) {
                rounded = roundTrip(actual);
            }
            return rounded;
        }

    }

}