added src/test/java/org/msgpack/simple

frsyuki · frsyuki · commit 0169c776822a · 2011-11-29T12:49:09.000-08:00
diff --git a/src/test/java/org/msgpack/simple/TestSimpleDynamicTyping.java b/src/test/java/org/msgpack/simple/TestSimpleDynamicTyping.java
@@ -0,0 +1,69 @@
+package org.msgpack.simple;
+
+import org.msgpack.MessagePack;
+import org.msgpack.type.Value;
+import org.msgpack.type.IntegerValue;
+import org.msgpack.type.RawValue;
+import org.msgpack.type.ArrayValue;
+
+import java.io.IOException;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertArrayEquals;
+import org.junit.Test;
+
+
+public class TestSimpleDynamicTyping {
+    @Test
+    @SuppressWarnings("unused")
+    public void testTypes() throws IOException {
+        MessagePack msgpack = new MessagePack();
+
+        byte[] raw = msgpack.write(new int[] {1,2,3});
+        Value v = msgpack.read(raw);
+
+        if(v.isArrayValue()) {
+            // ArrayValue extends List<Value>
+            ArrayValue array = v.asArrayValue();
+            int n0 = array.get(0).asIntegerValue().intValue();
+            assertEquals(1, n0);
+            int n1 = array.get(1).asIntegerValue().intValue();
+            assertEquals(2, n1);
+            int n2 = array.get(2).asIntegerValue().intValue();
+            assertEquals(3, n2);
+
+        } else if(v.isIntegerValue()) {
+            // IntegerValue extends Number
+            int num = v.asIntegerValue().intValue();
+
+        } else if(v.isRawValue()) {
+            // getString() or getByteArray()
+            String str = v.asRawValue().getString();
+        }
+        // other types:
+        //   NilValue asNilValue() / isNilValue()
+        //   BooleanValue asBooleanValue() / isBooleanValue()
+        //   IntegerValue asIntegerValue() / isIntegerValue()
+        //   FloatValue asFloatValue() / isFloatValue()
+        //   ArrayValue asArrayValue() / isArrayValue()
+        //   MapValue asMapValue() / isMapValue()
+        //   RawValue asRawValue() / isRawValue
+    }
+
+    @Test
+    public void testSimpleConvert() throws IOException {
+        MessagePack msgpack = new MessagePack();
+
+        byte[] raw = msgpack.write(new int[] {1,2,3});
+        Value v = msgpack.read(raw);
+
+        // convert from dynamic type (Value) to static type (int[])
+        int[] array = msgpack.convert(v, new int[3]);
+        assertArrayEquals(new int[] {1,2,3}, array);
+
+        // unconvert from static type (int[]) to dynamic type (Value)
+        Value v2 = msgpack.unconvert(array);
+        assertEquals(v, v2);
+    }
+}
+
diff --git a/src/test/java/org/msgpack/simple/TestSimplePackUnpack.java b/src/test/java/org/msgpack/simple/TestSimplePackUnpack.java
@@ -3,23 +3,33 @@
 import java.nio.ByteBuffer;
 import java.io.IOException;
 
-import org.junit.Test;
 import org.msgpack.MessagePack;
 import org.msgpack.type.Value;
 
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertArrayEquals;
+import org.junit.Test;
+
 
 public class TestSimplePackUnpack {
     @SuppressWarnings("unused")
     @Test
     public void testSimplePackUnpack() throws IOException {
         MessagePack msgpack = new MessagePack();
+
+        // serialize
         byte[] raw = msgpack.write(new int[] {1,2,3});
 
-        Value v = msgpack.read(raw);
+        // deserialize to static type
         int[] a = msgpack.read(raw, new int[3]);
+        assertArrayEquals(new int[] {1,2,3}, a);
+
+        // deserialize to dynamic type (see TestSimpleDynamicTyping.java)
+        Value v = msgpack.read(raw);
 
-        Value vb = msgpack.read(ByteBuffer.wrap(raw));
+        // ByteBuffer is also supported
         int[] ab = msgpack.read(ByteBuffer.wrap(raw), new int[3]);
+        assertArrayEquals(new int[] {1,2,3}, ab);
     }
 }
 
diff --git a/src/test/java/org/msgpack/simple/TestSimpleStreaming.java b/src/test/java/org/msgpack/simple/TestSimpleStreaming.java
@@ -0,0 +1,70 @@
+package org.msgpack;
+
+import org.msgpack.MessagePack;
+import org.msgpack.packer.Packer;
+import org.msgpack.unpacker.Unpacker;
+import org.msgpack.packer.BufferPacker;
+import org.msgpack.unpacker.BufferUnpacker;
+
+import java.io.ByteArrayOutputStream;
+import java.io.ByteArrayInputStream;
+import java.io.IOException;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertArrayEquals;
+import org.junit.Test;
+
+
+public class TestSimpleStreaming {
+    @SuppressWarnings("unused")
+    @Test
+    public void testSimpleStreamingSerialize() throws IOException {
+        MessagePack msgpack = new MessagePack();
+
+        // streaming serialize
+        ByteArrayOutputStream out = new ByteArrayOutputStream();
+        Packer packer = msgpack.createPacker(out);
+
+        packer.write(new String[] {"a", "b", "c"});
+        packer.write(new int[] {1, 2, 3});
+        packer.write(9.1);
+
+        // streaming deserialize
+        ByteArrayInputStream in = new ByteArrayInputStream(out.toByteArray());
+        Unpacker unpacker = msgpack.createUnpacker(in);
+
+        String[] msg1 = unpacker.read(new String[3]);
+        int[] msg2 = unpacker.read(new int[3]);
+        double msg3 = unpacker.read(double.class);
+
+        assertArrayEquals(new String[] {"a", "b", "c"}, msg1);
+        assertArrayEquals(new int[] {1, 2, 3}, msg2);
+        assertEquals(9.1, msg3, 0.001);
+    }
+
+    @SuppressWarnings("unused")
+    @Test
+    public void testBufferPackUnpacker() throws IOException {
+        MessagePack msgpack = new MessagePack();
+
+        // streaming serialize into efficient buffer
+        BufferPacker packer = msgpack.createBufferPacker();
+
+        packer.write(new String[] {"a", "b", "c"});
+        packer.write(new int[] {1, 2, 3});
+        packer.write(9.1);
+
+        // streaming deserialize from the buffer
+        // BufferUnpacker reduces copies
+        BufferUnpacker unpacker = msgpack.createBufferUnpacker(packer.toByteArray());
+
+        String[] msg1 = unpacker.read(new String[3]);
+        int[] msg2 = unpacker.read(new int[3]);
+        double msg3 = unpacker.read(double.class);
+
+        assertArrayEquals(new String[] {"a", "b", "c"}, msg1);
+        assertArrayEquals(new int[] {1, 2, 3}, msg2);
+        assertEquals(9.1, msg3, 0.01);
+    }
+}
+
