predicate-search/src/test/java/com/yahoo/search/predicate/serialization/SerializationTestHelper.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.search.predicate.serialization;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;

import static org.junit.jupiter.api.Assertions.assertArrayEquals;

/**
 * @author bjorncs
 */
public class SerializationTestHelper {

    private SerializationTestHelper() {}

    public static <T> void assertSerializationDeserializationMatches
            (T object, Serializer<T> serializer, Deserializer<T> deserializer) throws IOException {

        ByteArrayOutputStream byteArrayOut = new ByteArrayOutputStream(4096);
        DataOutputStream out = new DataOutputStream(byteArrayOut);
        serializer.serialize(object, out);
        out.flush();

        byte[] bytes = byteArrayOut.toByteArray();
        DataInputStream in = new DataInputStream(new ByteArrayInputStream(bytes));
        T newObject = deserializer.deserialize(in);

        byteArrayOut = new ByteArrayOutputStream(4096);
        out = new DataOutputStream(byteArrayOut);
        serializer.serialize(newObject, out);
        byte[] newBytes = byteArrayOut.toByteArray();
        assertArrayEquals(bytes, newBytes);
    }

    @FunctionalInterface
    public interface Serializer<T> {
        void serialize(T object, DataOutputStream out) throws IOException;
    }

    @FunctionalInterface
    public interface Deserializer<T> {
        T deserialize(DataInputStream in) throws IOException;
    }

}