1 files changed, 46 insertions, 0 deletions
diff --git a/test/i2ctcp/main.cpp b/test/i2ctcp/main.cpp
new file mode 100644
index 0000000..1f0c3ff
--- /dev/null
+++ b/test/i2ctcp/main.cpp
@@ -0,0 +1,46 @@
+#include <algorithm>
+#include <gtest/gtest.h>
+
+#include "i2ctcpv1.h"
+
+using std::min;
+
+const uint8_t data[] = { 0xff, 0x00, 0xde, 0xad, 0xbe, 0xef, };
+const size_t data_len = sizeof(data);
+const size_t chunk_size = 6;
+
+char * send_data = nullptr;
+size_t send_size = 0;
+
+TEST(i2ctcp, send) {
+	i2ctcp_msg_t send_msg = {
+		.addr = 0x1122,
+		.data = (char *) data,
+		.length = data_len,
+	};
+
+	ASSERT_TRUE(i2ctcp_write(&send_msg, &send_data, &send_size));
+	ASSERT_NE(send_data, nullptr);
+	ASSERT_GE(send_size, 0);
+}
+
+TEST(i2ctcp, recv) {
+	i2ctcp_msg_t recv_msg;
+	i2ctcp_read_reset(&recv_msg);
+	for (size_t i = 0; i < send_size; i += chunk_size) {
+		size_t expected_size = min(send_size, i + chunk_size) - i;
+
+		int parsed = i2ctcp_read(&recv_msg, send_data + i, expected_size);
+		EXPECT_GE(parsed, 0);
+
+		if (i + expected_size == send_size)
+			EXPECT_EQ(parsed, 0);
+		else
+			EXPECT_GT(parsed, 0);
+	}
+
+	ASSERT_NE(recv_msg.data, nullptr);
+	ASSERT_EQ(recv_msg.length, data_len);
+	ASSERT_EQ(0, memcmp(recv_msg.data, data, data_len));
+}
+