refactored SizeLimitException class

muga · muga · commit 08ad4fedb7c3 · 2011-11-06T02:12:18.000+09:00
diff --git a/src/main/java/org/msgpack/unpacker/SizeLimitException.java b/src/main/java/org/msgpack/unpacker/SizeLimitException.java
@@ -17,11 +17,11 @@
 //
 package org.msgpack.unpacker;
 
-import org.msgpack.MessageTypeException;
+import java.io.IOException;
 
 
 @SuppressWarnings("serial")
-public class SizeLimitException extends MessageTypeException {
+public class SizeLimitException extends IOException {
     public SizeLimitException() {
         super();
     }
diff --git a/src/test/java/org/msgpack/unpacker/TestSizeLimit.java b/src/test/java/org/msgpack/unpacker/TestSizeLimit.java
@@ -1,4 +1,4 @@
-package org.msgpack;
+package org.msgpack.unpacker;
 
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertTrue;
@@ -12,6 +12,7 @@
 import java.util.Map;
 
 import org.junit.Test;
+import org.msgpack.MessagePack;
 import org.msgpack.packer.Packer;
 import org.msgpack.template.ListTemplate;
 import org.msgpack.template.MapTemplate;
diff --git a/src/test/java/org/msgpack/unpacker/TestUnpackerInterator.java b/src/test/java/org/msgpack/unpacker/TestUnpackerInterator.java
@@ -0,0 +1,40 @@
+package org.msgpack.unpacker;
+
+import static org.junit.Assert.assertEquals;
+
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+
+import org.junit.Test;
+import org.msgpack.MessagePack;
+import org.msgpack.packer.Packer;
+
+public class TestUnpackerInterator {
+
+    @Test
+    public void testSample() throws Exception {
+        MessagePack msgpack = new MessagePack();
+
+        ByteArrayOutputStream out = new ByteArrayOutputStream();
+        Packer packer = msgpack.createPacker(out);
+        packer.write(1);
+        packer.write(2);
+        packer.write(3);
+        byte[] bytes = out.toByteArray();
+
+        Unpacker unpacker = msgpack.createUnpacker(new ByteArrayInputStream(bytes));
+        UnpackerIterator iter = unpacker.iterator();
+        unpacker.resetReadByteCount();
+        iter.hasNext();
+        iter.next();
+        assertEquals(1, unpacker.getReadByteCount());
+        unpacker.resetReadByteCount();
+        iter.hasNext();
+        iter.next();
+        assertEquals(1, unpacker.getReadByteCount());
+        unpacker.resetReadByteCount();
+        iter.hasNext();
+        iter.next();
+        assertEquals(1, unpacker.getReadByteCount());
+    }
+}
