init push

deps/oblib/unittest/rpc/test_mysql_packet.cpp

@@ -0,0 +1,161 @@
+/**
+ * Copyright (c) 2021 OceanBase
+ * OceanBase CE is licensed under Mulan PubL v2.
+ * You can use this software according to the terms and conditions of the Mulan PubL v2.
+ * You may obtain a copy of Mulan PubL v2 at:
+ *          http://license.coscl.org.cn/MulanPubL-2.0
+ * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+ * EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+ * MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+ * See the Mulan PubL v2 for more details.
+ */
+
+#include <gtest/gtest.h>
+#include "rpc/obmysql/ob_mysql_packet.h"
+
+using namespace oceanbase::common;
+using namespace oceanbase::rpc;
+using namespace oceanbase::obmysql;
+
+class ObFakeMySQLPacket : public ObMySQLPacket {
+  public:
+  ObFakeMySQLPacket() : content_len_(0)
+  {}
+  int64_t content_len_;
+
+  protected:
+  virtual int serialize(char* start, const int64_t len, int64_t& pos) const
+  {
+    int ret = OB_SUCCESS;
+    if ((NULL == start) || (content_len_ < 0)) {
+      ret = OB_INVALID_ARGUMENT;
+    } else if ((len - pos) < content_len_) {
+      ret = OB_BUF_NOT_ENOUGH;
+    } else {
+      for (int i = 0; i < content_len_; i++) {
+        start[pos + i] = 'a';
+      }
+      pos += content_len_;
+    }
+    return ret;
+  }
+};
+
+class TestMySQLPacket : public ::testing::Test {
+  public:
+  TestMySQLPacket()
+  {}
+
+  virtual void SetUp()
+  {}
+
+  virtual void TearDown()
+  {}
+};
+
+TEST_F(TestMySQLPacket, common)
+{
+  ObFakeMySQLPacket fake_pkt;
+  fake_pkt.set_seq(255);
+  fake_pkt.content_len_ = 1;
+  int64_t pkt_count = 0;
+  int64_t pos = 0;
+  int ret = OB_SUCCESS;
+  int64_t len = 100;
+  char* buf = new char[len];
+  ret = fake_pkt.encode(buf, len, pos, pkt_count);
+  EXPECT_EQ(OB_SUCCESS, ret);
+  EXPECT_EQ(1, pkt_count);
+  delete[] buf;
+  buf = NULL;
+
+  len = 16 * 1024 * 1024 - 1 + 4;
+  buf = new char[len];
+  fake_pkt.content_len_ = 16 * 1024 * 1024 - 1;
+  pkt_count = 0;
+  pos = 0;
+  ret = fake_pkt.encode(buf, len, pos, pkt_count);
+  EXPECT_EQ(OB_SIZE_OVERFLOW, ret);
+  EXPECT_EQ(0, pkt_count);
+  EXPECT_EQ(0, pos);
+  delete[] buf;
+  buf = NULL;
+
+  len = 16 * 1024 * 1024 - 1 + 4 + 4;
+  buf = new char[len];
+  fake_pkt.content_len_ = 16 * 1024 * 1024 - 1;
+  pkt_count = 0;
+  pos = 0;
+  ret = fake_pkt.encode(buf, len, pos, pkt_count);
+  EXPECT_EQ(OB_SUCCESS, ret);
+  EXPECT_EQ(2, pkt_count);
+  EXPECT_EQ(len, pos);
+  delete[] buf;
+  buf = NULL;
+
+  len = 16 * 1024 * 1024 - 1 + 4 + 4;
+  buf = new char[len];
+  fake_pkt.content_len_ = 16 * 1024 * 1024 - 1;
+  pkt_count = 0;
+  pos = 0;
+  ret = fake_pkt.encode(buf, len, pos, pkt_count);
+  EXPECT_EQ(OB_SUCCESS, ret);
+  EXPECT_EQ(2, pkt_count);
+  EXPECT_EQ(len, pos);
+  delete[] buf;
+  buf = NULL;
+
+  len = 16 * 1024 * 1024 - 2 + 4;
+  buf = new char[len];
+  fake_pkt.content_len_ = 16 * 1024 * 1024 - 2;
+  pkt_count = 0;
+  pos = 0;
+  ret = fake_pkt.encode(buf, len, pos, pkt_count);
+  EXPECT_EQ(OB_SUCCESS, ret);
+  EXPECT_EQ(1, pkt_count);
+  EXPECT_EQ(len, pos);
+  delete[] buf;
+  buf = NULL;
+
+  len = 16 * 1024 * 1024 - 1 + 1 + 4 + 4;
+  buf = new char[len];
+  fake_pkt.content_len_ = 16 * 1024 * 1024 - 1 + 1;
+  pkt_count = 0;
+  pos = 0;
+  ret = fake_pkt.encode(buf, len, pos, pkt_count);
+  EXPECT_EQ(OB_SUCCESS, ret);
+  EXPECT_EQ(2, pkt_count);
+  EXPECT_EQ(len, pos);
+  delete[] buf;
+  buf = NULL;
+
+  len = (16 * 1024 * 1024 - 1 + 4) * 5 + 4;
+  buf = new char[len];
+  fake_pkt.content_len_ = (16 * 1024 * 1024 - 1) * 5;
+  pkt_count = 0;
+  pos = 0;
+  ret = fake_pkt.encode(buf, len, pos, pkt_count);
+  EXPECT_EQ(OB_SUCCESS, ret);
+  EXPECT_EQ(6, pkt_count);
+  EXPECT_EQ(len, pos);
+  delete[] buf;
+  buf = NULL;
+
+  len = (16 * 1024 * 1024 - 1 + 4);
+  buf = new char[len];
+  fake_pkt.content_len_ = (16 * 1024 * 1024 - 1) * 5;
+  pkt_count = 0;
+  pos = 0;
+  ret = fake_pkt.encode(buf, len, pos, pkt_count);
+  EXPECT_EQ(OB_BUF_NOT_ENOUGH, ret);
+  EXPECT_EQ(0, pkt_count);
+  EXPECT_EQ(0, pos);
+  delete[] buf;
+  buf = NULL;
+}
+
+int main(int argc, char* argv[])
+{
+  ::testing::InitGoogleTest(&argc, argv);
+  return RUN_ALL_TESTS();
+}