init push

unittest/storage/test_ob_col_map.cpp

@@ -0,0 +1,74 @@
+/**
+ * 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 "storage/ob_col_map.h"
+#include "lib/allocator/page_arena.h"
+#include <gtest/gtest.h>
+#include <gmock/gmock.h>
+
+namespace oceanbase {
+using namespace common;
+using namespace storage;
+
+namespace unittest {
+void test_col_map(const int64_t col_count)
+{
+  int ret = OB_SUCCESS;
+  ObColMap col_map;
+
+  ret = col_map.init(col_count);
+  ASSERT_EQ(OB_SUCCESS, ret);
+
+  for (int64_t i = 0; i < col_count; i++) {
+    ret = col_map.set_refactored(static_cast<uint64_t>(i), col_count - i);
+    ASSERT_EQ(OB_SUCCESS, ret);
+  }
+
+  int64_t col_idx = 0;
+  for (int64_t i = col_count - 1; OB_SUCC(ret) && i >= 0; i--) {
+    ret = col_map.get_refactored(static_cast<uint64_t>(i), col_idx);
+    ASSERT_EQ(OB_SUCCESS, ret);
+    ASSERT_EQ(col_idx, col_count - i);
+  }
+
+  if (col_count < 1024 /* FIRST_LEVEL_MAX_COL_NUM */) {
+    ASSERT_EQ(NULL, col_map.get(1023));
+  } else if (col_count < 65535 /* FINAL_LEVEL_MAX_COL_NUM */) {
+    ASSERT_EQ(NULL, col_map.get(65534));
+  }
+}
+
+TEST(TestObColMap, test_col_map_first_level)
+{
+  // less than FIRST_LEVEL_MAX_COL_NUM = 1024
+  int64_t col_count = 752;
+  test_col_map(col_count);
+}
+
+TEST(TestObColMap, test_col_map_final_level)
+{
+  // greater than FIRST_LEVEL_MAX_COL_NUM = 1024
+  int64_t col_count = 12281;
+  test_col_map(col_count);
+}
+
+}  // namespace unittest
+}  // namespace oceanbase
+
+int main(int argc, char** argv)
+{
+  OB_LOGGER.set_file_name("test_ob_col_map.log");
+  OB_LOGGER.set_log_level("WARN");
+  CLOG_LOG(INFO, "begin unittest: test_ob_col_map");
+  ::testing::InitGoogleTest(&argc, argv);
+  return RUN_ALL_TESTS();
+}