[2/2] Add varint encoding to Serializing Cache patch by Vijay; reviewed by jbellis,xedin for CASSANDRA-4138

[vi...@apache.org]

http://git-wip-us.apache.org/repos/asf/cassandra/blob/cb25a8fc/test/unit/org/apache/cassandra/utils/EncodedStreamsTest.java
----------------------------------------------------------------------
diff --git a/test/unit/org/apache/cassandra/utils/EncodedStreamsTest.java b/test/unit/org/apache/cassandra/utils/EncodedStreamsTest.java
new file mode 100644
index 0000000..1907c83
--- /dev/null
+++ b/test/unit/org/apache/cassandra/utils/EncodedStreamsTest.java
@@ -0,0 +1,160 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.apache.cassandra.utils;
+
+import static org.apache.cassandra.Util.*;
+
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+import java.io.DataInputStream;
+import java.io.DataOutputStream;
+import java.io.IOException;
+
+import org.apache.cassandra.SchemaLoader;
+import org.apache.cassandra.db.ColumnFamily;
+import org.apache.cassandra.db.DBTypeSizes;
+import org.apache.cassandra.utils.vint.EncodedDataInputStream;
+import org.apache.cassandra.utils.vint.EncodedDataOutputStream;
+
+import org.junit.Assert;
+import org.junit.Test;
+
+public class EncodedStreamsTest extends SchemaLoader
+{
+    private String tableName = "Keyspace1";
+    private String standardCFName = "Standard1";
+    private String counterCFName = "Counter1";
+    private String superCFName = "Super1";
+
+    @Test
+    public void testStreams() throws IOException
+    {
+        ByteArrayOutputStream byteArrayOStream1 = new ByteArrayOutputStream();
+        EncodedDataOutputStream odos = new EncodedDataOutputStream(byteArrayOStream1);
+
+        ByteArrayOutputStream byteArrayOStream2 = new ByteArrayOutputStream();
+        DataOutputStream dos = new DataOutputStream(byteArrayOStream2);
+        
+        for (short i = 0; i < 10000; i++)
+        {
+            dos.writeShort(i);
+            odos.writeShort(i);
+        }
+        dos.flush();
+        odos.flush();
+
+        for (int i = Short.MAX_VALUE; i < ((int)Short.MAX_VALUE + 10000); i++)
+        {
+            dos.writeInt(i);
+            odos.writeInt(i);
+        }
+        dos.flush();
+        odos.flush();
+
+        for (long i = Integer.MAX_VALUE; i < ((long)Integer.MAX_VALUE + 10000);i++)
+        {
+            dos.writeLong(i);
+            odos.writeLong(i);
+        }
+        dos.flush();
+        odos.flush();
+        Assert.assertTrue(byteArrayOStream1.size() < byteArrayOStream2.size());
+
+        ByteArrayInputStream byteArrayIStream1 = new ByteArrayInputStream(byteArrayOStream1.toByteArray());
+        EncodedDataInputStream idis = new EncodedDataInputStream(new DataInputStream(byteArrayIStream1));
+
+        // assert reading Short
+        for (int i = 0; i < 10000; i++)
+            Assert.assertEquals(i, idis.readShort());
+
+        // assert reading Integer
+        for (int i = Short.MAX_VALUE; i < ((int)Short.MAX_VALUE + 10000); i++)
+            Assert.assertEquals(i, idis.readInt());
+
+        // assert reading Long
+        for (long i = Integer.MAX_VALUE; i < ((long)Integer.MAX_VALUE) + 1000; i++)
+            Assert.assertEquals(i, idis.readLong());
+    }
+
+    private ColumnFamily createCF()
+    {
+        ColumnFamily cf = ColumnFamily.create(tableName, standardCFName);
+        cf.addColumn(column("vijay", "try", 1));
+        cf.addColumn(column("to", "be_nice", 1));
+        return cf;
+    }
+
+    private ColumnFamily createCounterCF()
+    {
+        ColumnFamily cf = ColumnFamily.create(tableName, counterCFName);
+        cf.addColumn(counterColumn("vijay", 1L, 1));
+        cf.addColumn(counterColumn("wants", 1000000, 1));
+        return cf;
+    }
+
+    private ColumnFamily createSuperCF()
+    {
+        ColumnFamily cf = ColumnFamily.create(tableName, superCFName);
+        cf.addColumn(superColumn(cf, "Avatar", column("$2,782,275,172", "2009", 1)));
+        cf.addColumn(superColumn(cf, "Titanic", column("$1,925,905,151", "1997", 1)));
+        return cf;
+    }
+
+    @Test
+    public void testCFSerialization() throws IOException
+    {
+        ByteArrayOutputStream byteArrayOStream1 = new ByteArrayOutputStream();
+        EncodedDataOutputStream odos = new EncodedDataOutputStream(byteArrayOStream1);
+        ColumnFamily.serializer.serialize(createCF(), odos);
+
+        ByteArrayInputStream byteArrayIStream1 = new ByteArrayInputStream(byteArrayOStream1.toByteArray());
+        EncodedDataInputStream odis = new EncodedDataInputStream(new DataInputStream(byteArrayIStream1));
+        ColumnFamily cf = ColumnFamily.serializer.deserialize(odis);
+        Assert.assertEquals(cf, createCF());
+        Assert.assertEquals(byteArrayOStream1.size(), (int)ColumnFamily.serializer().serializedSize(cf, DBTypeSizes.VINT));
+    }
+
+    @Test
+    public void testCounterCFSerialization() throws IOException
+    {
+        ByteArrayOutputStream byteArrayOStream1 = new ByteArrayOutputStream();
+        EncodedDataOutputStream odos = new EncodedDataOutputStream(byteArrayOStream1);
+        ColumnFamily.serializer.serialize(createCounterCF(), odos);
+
+        ByteArrayInputStream byteArrayIStream1 = new ByteArrayInputStream(byteArrayOStream1.toByteArray());
+        EncodedDataInputStream odis = new EncodedDataInputStream(new DataInputStream(byteArrayIStream1));
+        ColumnFamily cf = ColumnFamily.serializer.deserialize(odis);
+        Assert.assertEquals(cf, createCounterCF());
+        Assert.assertEquals(byteArrayOStream1.size(), (int)ColumnFamily.serializer().serializedSize(cf, DBTypeSizes.VINT));
+    }
+
+    @Test
+    public void testSuperCFSerialization() throws IOException
+    {
+        ByteArrayOutputStream byteArrayOStream1 = new ByteArrayOutputStream();
+        EncodedDataOutputStream odos = new EncodedDataOutputStream(byteArrayOStream1);
+        ColumnFamily.serializer.serialize(createSuperCF(), odos);
+
+        ByteArrayInputStream byteArrayIStream1 = new ByteArrayInputStream(byteArrayOStream1.toByteArray());
+        EncodedDataInputStream odis = new EncodedDataInputStream(new DataInputStream(byteArrayIStream1));
+        ColumnFamily cf = ColumnFamily.serializer.deserialize(odis);
+        Assert.assertEquals(cf, createSuperCF());
+        Assert.assertEquals(byteArrayOStream1.size(), (int)ColumnFamily.serializer().serializedSize(cf, DBTypeSizes.VINT));
+    }
+}
+