1 files changed, 56 insertions, 0 deletions
diff --git a/test/pbdrv/msg.cpp b/test/pbdrv/msg.cpp
new file mode 100644
index 0000000..ca23bc7
--- /dev/null
+++ b/test/pbdrv/msg.cpp
@@ -0,0 +1,56 @@
+#include <gtest/gtest.h>
+
+#include "pb-msg.h"
+#include "pb-send.h"
+#include "pb-mod.h"
+
+TEST(pb_msg_rw, cmd_req_read) {
+	pb_cmd_prop_t cmd = {
+		.propid = 2,
+	};
+	pb_msg_t msg_write = {
+		.type = PB_CMD_PROP,
+		.action = PB_ACTION_REQ,
+		.sender = 0xff,
+		.cmd = &cmd,
+	};
+	pb_buf_t buf = pb_msg_write(&msg_write);
+
+	ASSERT_NE(buf.data, nullptr);
+	ASSERT_GE(buf.size, 0);
+
+	pb_msg_t * msg_read = pb_msg_read(&buf);
+	pb_buf_free(&buf);
+
+	ASSERT_EQ(buf.data, nullptr);
+
+	EXPECT_EQ(msg_write.type, msg_read->type);
+	EXPECT_EQ(msg_write.sender, msg_read->sender);
+	EXPECT_NE(nullptr, msg_read->cmd);
+	pb_cmd_prop_t * prop = (pb_cmd_prop_t *) msg_read->cmd;
+	EXPECT_EQ(prop->propid, cmd.propid);
+
+	pb_msg_free(msg_read);
+}
+
+TEST(pb_msg_rw, cmd_req_magic) {
+	pb_buf_t buf = pb_send_magic_req();
+
+	ASSERT_NE(buf.data, nullptr);
+	ASSERT_GE(buf.size, 0);
+
+	pb_msg_t * msg_read = pb_msg_read(&buf);
+	pb_buf_free(&buf);
+
+	ASSERT_EQ(buf.data, nullptr);
+
+	EXPECT_EQ(msg_read->type, PB_CMD_MAGIC);
+	EXPECT_EQ(msg_read->sender, PB_MOD_ADDR);
+	EXPECT_NE(msg_read->cmd, nullptr);
+	pb_cmd_magic_t * magic = (pb_cmd_magic_t *) msg_read->cmd;
+	EXPECT_EQ(magic->_magic_size, sizeof(pb_cmd_magic_req));
+	EXPECT_EQ(0, memcmp(pb_cmd_magic_req, magic->magic, magic->_magic_size));
+
+	pb_msg_free(msg_read);
+}
+