tmate/msgpack/test/pack_unpack_c.cc

#include <msgpack.h>
#include <gtest/gtest.h>
#include <stdio.h>

TEST(pack, num)
{
	msgpack_sbuffer* sbuf = msgpack_sbuffer_new();
	msgpack_packer* pk = msgpack_packer_new(sbuf, msgpack_sbuffer_write);

	EXPECT_EQ(0, msgpack_pack_int(pk, 1));

	msgpack_sbuffer_free(sbuf);
	msgpack_packer_free(pk);
}


TEST(pack, array)
{
	msgpack_sbuffer* sbuf = msgpack_sbuffer_new();
	msgpack_packer* pk = msgpack_packer_new(sbuf, msgpack_sbuffer_write);

	EXPECT_EQ(0, msgpack_pack_array(pk, 3));
	EXPECT_EQ(0, msgpack_pack_int(pk, 1));
	EXPECT_EQ(0, msgpack_pack_int(pk, 2));
	EXPECT_EQ(0, msgpack_pack_int(pk, 3));

	msgpack_sbuffer_free(sbuf);
	msgpack_packer_free(pk);
}


TEST(unpack, sequence)
{
	msgpack_sbuffer* sbuf = msgpack_sbuffer_new();
	msgpack_packer* pk = msgpack_packer_new(sbuf, msgpack_sbuffer_write);

	EXPECT_EQ(0, msgpack_pack_int(pk, 1));
	EXPECT_EQ(0, msgpack_pack_int(pk, 2));
	EXPECT_EQ(0, msgpack_pack_int(pk, 3));

	msgpack_packer_free(pk);

	bool success;
	size_t offset = 0;

	msgpack_unpacked msg;
	msgpack_unpacked_init(&msg);

	success = msgpack_unpack_next(&msg, sbuf->data, sbuf->size, &offset);
	EXPECT_TRUE(success);
	EXPECT_EQ(MSGPACK_OBJECT_POSITIVE_INTEGER, msg.data.type);
	EXPECT_EQ(1, msg.data.via.u64);

	success = msgpack_unpack_next(&msg, sbuf->data, sbuf->size, &offset);
	EXPECT_TRUE(success);
	EXPECT_EQ(MSGPACK_OBJECT_POSITIVE_INTEGER, msg.data.type);
	EXPECT_EQ(2, msg.data.via.u64);

	success = msgpack_unpack_next(&msg, sbuf->data, sbuf->size, &offset);
	EXPECT_TRUE(success);
	EXPECT_EQ(MSGPACK_OBJECT_POSITIVE_INTEGER, msg.data.type);
	EXPECT_EQ(3, msg.data.via.u64);

	success = msgpack_unpack_next(&msg, sbuf->data, sbuf->size, &offset);
	EXPECT_FALSE(success);

	msgpack_sbuffer_free(sbuf);
	msgpack_unpacked_destroy(&msg);
}