1 files changed, 76 insertions, 0 deletions
diff --git a/test/pbdrv/send.cpp b/test/pbdrv/send.cpp
new file mode 100644
index 0000000..7562ca4
--- /dev/null
+++ b/test/pbdrv/send.cpp
@@ -0,0 +1,76 @@
+#include <gtest/gtest.h>
+
+#include "pb-mod.h"
+#include "pb-msg.h"
+#include "pb-send.h"
+
+/**
+ * Only one of each command type is tested due to the (de)serializers being the
+ * same (i.e. you're not testing more by changing the command action).
+ */
+
+TEST(pb_send_rw, cmd_state_req) {
+	pb_hook_mod_state_write(PB_GS_NOINIT);
+
+	pb_buf_t buf = pb_send_state_req();
+	ASSERT_NE(buf.data, nullptr);
+	ASSERT_GE(buf.size, 0);
+
+	pb_msg_t * msg = pb_msg_read(&buf);
+	EXPECT_EQ(msg->type, PB_CMD_STATE);
+	EXPECT_EQ(msg->action, PB_ACTION_REQ);
+	EXPECT_EQ(msg->sender, PB_MOD_ADDR);
+	EXPECT_NE(nullptr, msg->cmd);
+
+	pb_cmd_state_t * cmd = (pb_cmd_state_t *) msg->cmd;
+	EXPECT_EQ(cmd->state, pb_hook_mod_state_read());
+
+	pb_buf_free(&buf);
+	ASSERT_EQ(buf.data, nullptr);
+
+	pb_msg_free(msg);
+}
+
+TEST(pb_send_rw, cmd_read_req) {
+	pb_buf_t buf = pb_send_read_req(2);
+	ASSERT_NE(buf.data, nullptr);
+	ASSERT_GE(buf.size, 0);
+
+	pb_msg_t * msg = pb_msg_read(&buf);
+
+	EXPECT_EQ(msg->type, PB_CMD_PROP);
+	EXPECT_EQ(msg->action, PB_ACTION_REQ);
+	EXPECT_EQ(msg->sender, PB_MOD_ADDR);
+	EXPECT_NE(nullptr, msg->cmd);
+
+	pb_cmd_prop_t * cmd = (pb_cmd_prop_t *) msg->cmd;
+	EXPECT_EQ(cmd->propid, 2);
+
+	pb_buf_free(&buf);
+	ASSERT_EQ(buf.data, nullptr);
+
+	pb_msg_free(msg);
+}
+
+TEST(pb_send_rw, cmd_magic_req) {
+	pb_buf_t buf = pb_send_magic_req();
+	ASSERT_NE(buf.data, nullptr);
+	ASSERT_GE(buf.size, 0);
+
+	pb_msg_t * msg = pb_msg_read(&buf);
+
+	EXPECT_EQ(msg->type, PB_CMD_MAGIC);
+	EXPECT_EQ(msg->action, PB_ACTION_REQ);
+	EXPECT_EQ(msg->sender, PB_MOD_ADDR);
+	EXPECT_NE(msg->cmd, nullptr);
+
+	pb_cmd_magic_t * cmd = (pb_cmd_magic_t *) msg->cmd;
+	EXPECT_EQ(cmd->_magic_size, sizeof(pb_cmd_magic_req));
+	EXPECT_EQ(0, memcmp(pb_cmd_magic_req, cmd->magic, cmd->_magic_size));
+
+	pb_buf_free(&buf);
+	ASSERT_EQ(buf.data, nullptr);
+
+	pb_msg_free(msg);
+}
+