1 files changed, 48 insertions, 0 deletions
diff --git a/predicate-search/src/test/java/com/yahoo/search/predicate/serialization/SerializationTestHelper.java b/predicate-search/src/test/java/com/yahoo/search/predicate/serialization/SerializationTestHelper.java
new file mode 100644
index 00000000000..47746b15d49
--- /dev/null
+++ b/predicate-search/src/test/java/com/yahoo/search/predicate/serialization/SerializationTestHelper.java
@@ -0,0 +1,48 @@
+// Copyright 2016 Yahoo Inc. 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.Assert.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;
+    }
+
+}