Merge pull request #380 from whoshuu/expand-buffer-tests

Add tests for expanding an unpacker buffer
msgpack · Nov 1, 2015 · f58eb11 · f58eb11
2 parents 134beee + 434dae8
commit f58eb11
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/test/limit.cpp b/test/limit.cpp
@@ -553,3 +553,21 @@ TEST(limit, unpacker_array_over)
         EXPECT_TRUE(false);
     }
 }
+
+TEST(limit, unpacker_reserve)
+{
+    msgpack::unpacker u(nullptr, nullptr, MSGPACK_UNPACKER_INIT_BUFFER_SIZE,
+                        msgpack::unpack_limit());
+    std::size_t original_capacity = u.buffer_capacity();
+    u.reserve_buffer(original_capacity + 1u);
+    EXPECT_EQ((original_capacity + COUNTER_SIZE) * 2 - COUNTER_SIZE, u.buffer_capacity());
+}
+
+TEST(limit, unpacker_reserve_more_than_twice)
+{
+    msgpack::unpacker u(nullptr, nullptr, MSGPACK_UNPACKER_INIT_BUFFER_SIZE,
+                        msgpack::unpack_limit());
+    std::size_t original_capacity = u.buffer_capacity();
+    u.reserve_buffer(original_capacity * 3);
+    EXPECT_EQ((original_capacity + COUNTER_SIZE) * 4 - COUNTER_SIZE, u.buffer_capacity());
+}