cassandra-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From jbel...@apache.org
Subject svn commit: r1149623 - /cassandra/trunk/test/unit/org/apache/cassandra/utils/BytesReadTrackerTest.java
Date Fri, 22 Jul 2011 14:41:24 GMT
Author: jbellis
Date: Fri Jul 22 14:41:23 2011
New Revision: 1149623

URL: http://svn.apache.org/viewvc?rev=1149623&view=rev
Log:
belatedly add BRTT

Added:
    cassandra/trunk/test/unit/org/apache/cassandra/utils/BytesReadTrackerTest.java

Added: cassandra/trunk/test/unit/org/apache/cassandra/utils/BytesReadTrackerTest.java
URL: http://svn.apache.org/viewvc/cassandra/trunk/test/unit/org/apache/cassandra/utils/BytesReadTrackerTest.java?rev=1149623&view=auto
==============================================================================
--- cassandra/trunk/test/unit/org/apache/cassandra/utils/BytesReadTrackerTest.java (added)
+++ cassandra/trunk/test/unit/org/apache/cassandra/utils/BytesReadTrackerTest.java Fri Jul
22 14:41:23 2011
@@ -0,0 +1,209 @@
+/*
+ * 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.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
+
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+import java.io.DataInputStream;
+import java.io.DataOutputStream;
+
+import org.junit.Test;
+
+public class BytesReadTrackerTest
+{
+
+    @Test
+    public void testBytesRead() throws Exception
+    {
+        byte[] testData;
+
+        ByteArrayOutputStream baos = new ByteArrayOutputStream();
+        DataOutputStream dos = new DataOutputStream(baos);
+        try
+        {
+            // boolean
+            dos.writeBoolean(true);
+            // byte
+            dos.writeByte(0x1);
+            // char
+            dos.writeChar('a');
+            // short
+            dos.writeShort(1);
+            // int
+            dos.writeInt(1);
+            // long
+            dos.writeLong(1L);
+            // float
+            dos.writeFloat(1.0f);
+            // double
+            dos.writeDouble(1.0d);
+
+            // String
+            dos.writeUTF("abc");
+            testData = baos.toByteArray();
+        }
+        finally
+        {
+            dos.close();
+        }
+
+        DataInputStream dis = new DataInputStream(new ByteArrayInputStream(testData));
+        BytesReadTracker tracker = new BytesReadTracker(dis);
+
+        try
+        {
+            // boolean = 1byte
+            boolean bool = tracker.readBoolean();
+            assertTrue(bool);
+            assertEquals(1, tracker.getBytesRead());
+            // byte = 1byte
+            byte b = tracker.readByte();
+            assertEquals(b, 0x1);
+            assertEquals(2, tracker.getBytesRead());
+            // char = 1byte
+            char c = tracker.readChar();
+            assertEquals('a', c);
+            assertEquals(3, tracker.getBytesRead());
+            // short = 2bytes
+            short s = tracker.readShort();
+            assertEquals(1, s);
+            assertEquals((short) 5, tracker.getBytesRead());
+            // int = 4bytes
+            int i = tracker.readInt();
+            assertEquals(1, i);
+            assertEquals(9, tracker.getBytesRead());
+            // long = 8bytes
+            long l = tracker.readLong();
+            assertEquals(1L, l);
+            assertEquals(17, tracker.getBytesRead());
+            // float = 4bytes
+            float f = tracker.readFloat();
+            assertEquals(1.0f, f, 0);
+            assertEquals(21, tracker.getBytesRead());
+            // double = 8bytes
+            double d = tracker.readDouble();
+            assertEquals(1.0d, d, 0);
+            assertEquals(29, tracker.getBytesRead());
+            // String("abc") = 2(string size) + 3 = 5 bytes
+            String str = tracker.readUTF();
+            assertEquals("abc", str);
+            assertEquals(34, tracker.getBytesRead());
+        }
+        finally
+        {
+            dis.close();
+        }
+
+        tracker.reset();
+        assertEquals(0, tracker.getBytesRead());
+    }
+
+    @Test
+    public void testUnsignedRead() throws Exception
+    {
+        byte[] testData;
+
+        ByteArrayOutputStream baos = new ByteArrayOutputStream();
+        DataOutputStream dos = new DataOutputStream(baos);
+        try
+        {
+            // byte
+            dos.writeByte(0x1);
+            // short
+            dos.writeShort(1);
+            testData = baos.toByteArray();
+        }
+        finally
+        {
+            dos.close();
+        }
+
+        DataInputStream dis = new DataInputStream(new ByteArrayInputStream(testData));
+        BytesReadTracker tracker = new BytesReadTracker(dis);
+
+        try
+        {
+            // byte = 1byte
+            int b = tracker.readUnsignedByte();
+            assertEquals(b, 1);
+            assertEquals(1, tracker.getBytesRead());
+            // short = 2bytes
+            int s = tracker.readUnsignedShort();
+            assertEquals(1, s);
+            assertEquals(3, tracker.getBytesRead());
+        }
+        finally
+        {
+            dis.close();
+        }
+    }
+
+    @Test
+    public void testSkipBytesAndReadFully() throws Exception
+    {
+        String testStr = "1234567890";
+        byte[] testData = testStr.getBytes();
+
+        DataInputStream dis = new DataInputStream(new ByteArrayInputStream(testData));
+        BytesReadTracker tracker = new BytesReadTracker(dis);
+
+        try
+        {
+            // read first 5 bytes
+            byte[] out = new byte[5];
+            tracker.readFully(out, 0, 5);
+            assertEquals("12345", new String(out));
+            assertEquals(5, tracker.getBytesRead());
+
+            // then skip 2 bytes
+            tracker.skipBytes(2);
+            assertEquals(7, tracker.getBytesRead());
+
+            // and read the rest
+            out = new byte[3];
+            tracker.readFully(out);
+            assertEquals("890", new String(out));
+            assertEquals(10, tracker.getBytesRead());
+        }
+        finally
+        {
+            dis.close();
+        }
+    }
+
+    @Test(expected = UnsupportedOperationException.class)
+    public void testReadLine() throws Exception
+    {
+        DataInputStream dis = new DataInputStream(new ByteArrayInputStream("1".getBytes()));
+        BytesReadTracker tracker = new BytesReadTracker(dis);
+
+        try
+        {
+            // throws UnsupportedOperationException
+            tracker.readLine();
+        }
+        finally
+        {
+            dis.close();
+        }
+    }
+}



Mime
View raw message