add diagnose mgr unittest & placeholder for RPC request

unittest/storage/compaction/test_diagnose_info_mgr.cpp

@@ -0,0 +1,193 @@
+/**
+ * Copyright (c) 2023 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.
+ */
+#define USING_LOG_PREFIX STORAGE
+#include <gtest/gtest.h>
+#include <gmock/gmock.h>
+#define private public
+#define protected public
+#include <string.h>
+#include "storage/compaction/ob_compaction_diagnose.h"
+#include "mittest/mtlenv/mock_tenant_module_env.h"
+#include "share/rc/ob_tenant_base.h"
+namespace oceanbase
+{
+using namespace common;
+using namespace compaction;
+using namespace storage;
+
+namespace unittest
+{
+
+class TestDiagnoseInfoMgr : public ::testing::Test
+{
+public:
+  TestDiagnoseInfoMgr()
+  : ls_id_(TEST_LS_ID),
+    tenant_base_(TEST_TENANT_ID),
+    suspect_info_mgr_(NULL),
+    diagnose_tablet_mgr_(NULL),
+    is_inited_(false)
+  {}
+  ~TestDiagnoseInfoMgr() = default;
+  void SetUp();
+  void TearDown();
+  int gene_suspect_info(
+    const ObDiagnoseInfoPrio &prio,
+    const ObMergeType &merge_type,
+    const ObTabletID &tablet_id,
+    ObScheduleSuspectInfo &info);
+  static constexpr int64_t TEST_TENANT_ID = 1001;
+  static constexpr int64_t TEST_LS_ID = 9001;
+  share::ObLSID ls_id_;
+  ObTenantBase tenant_base_;
+  ObScheduleSuspectInfoMgr *suspect_info_mgr_;
+  ObDiagnoseTabletMgr *diagnose_tablet_mgr_;
+  ObDiagnoseInfoParam<2, 0> param_;
+  bool is_inited_;
+};
+
+void TestDiagnoseInfoMgr::SetUp()
+{
+  if (!is_inited_) {
+    ObMallocAllocator::get_instance()->create_and_add_tenant_allocator(TEST_TENANT_ID);
+    is_inited_ = true;
+  }
+  if (OB_ISNULL(suspect_info_mgr_)) {
+    suspect_info_mgr_ = OB_NEW(ObScheduleSuspectInfoMgr, ObModIds::TEST);
+  }
+  tenant_base_.set(suspect_info_mgr_);
+
+  if (OB_ISNULL(diagnose_tablet_mgr_)) {
+    diagnose_tablet_mgr_ = OB_NEW(ObDiagnoseTabletMgr, ObModIds::TEST);
+  }
+  tenant_base_.set(diagnose_tablet_mgr_);
+
+  ObTenantEnv::set_tenant(&tenant_base_);
+  ASSERT_EQ(OB_SUCCESS, tenant_base_.init());
+
+  ObMallocAllocator *ma = ObMallocAllocator::get_instance();
+  ASSERT_EQ(OB_SUCCESS, ma->set_tenant_limit(TEST_TENANT_ID, 1LL << 30));
+}
+
+void TestDiagnoseInfoMgr::TearDown()
+{
+  if (OB_NOT_NULL(suspect_info_mgr_)) {
+    suspect_info_mgr_->destroy();
+    suspect_info_mgr_ = nullptr;
+  }
+  if (OB_NOT_NULL(diagnose_tablet_mgr_)) {
+    diagnose_tablet_mgr_->destroy();
+    diagnose_tablet_mgr_ = nullptr;
+  }
+  tenant_base_.destroy();
+  ObTenantEnv::set_tenant(nullptr);
+}
+
+int TestDiagnoseInfoMgr::gene_suspect_info(
+  const ObDiagnoseInfoPrio &prio,
+  const ObMergeType &merge_type,
+  const ObTabletID &tablet_id,
+  ObScheduleSuspectInfo &info)
+{
+  int ret = OB_SUCCESS;
+  info.priority_ = static_cast<uint32_t>(prio);
+  info.merge_type_ = merge_type;
+  info.ls_id_ = ls_id_;
+  info.tablet_id_ = tablet_id;
+  info.info_param_ = &param_;
+  return ret;
+}
+
+bool judge_equal(const ObScheduleSuspectInfo &a, const ObScheduleSuspectInfo &b)
+{
+  return a.priority_ == b.priority_
+    && a.ls_id_ == b.ls_id_
+    && a.tablet_id_ == b.tablet_id_
+    && a.merge_type_ == b.merge_type_;
+}
+
+TEST_F(TestDiagnoseInfoMgr, test_add_del_suspect_info)
+{
+  int ret = OB_SUCCESS;
+  ObArenaAllocator allocator;
+  const ObTabletID tablet_id(1);
+  ASSERT_EQ(OB_SUCCESS, ObScheduleSuspectInfoMgr::mtl_init(suspect_info_mgr_));
+
+  ObScheduleSuspectInfo info;
+  ObScheduleSuspectInfo ret_info;
+  ret = gene_suspect_info(ObDiagnoseInfoPrio::DIAGNOSE_PRIORITY_LOW, MINOR_MERGE, tablet_id, info);
+  ASSERT_EQ(OB_SUCCESS, ret);
+  ret = suspect_info_mgr_->add_suspect_info(info.hash(), info);
+  ASSERT_EQ(OB_SUCCESS, ret);
+
+  // high priority could cover low priority
+  info.priority_ = static_cast<uint32_t>(ObDiagnoseInfoPrio::DIAGNOSE_PRIORITY_HIGH);
+  ret = suspect_info_mgr_->add_suspect_info(info.hash(), info);
+  ASSERT_EQ(OB_SUCCESS, ret);
+
+  ret = suspect_info_mgr_->get_with_param(info.hash(), ret_info, allocator);
+  ASSERT_EQ(OB_SUCCESS, ret);
+  ASSERT_EQ(true, judge_equal(info, ret_info));
+
+  // mid priority couldn't cover high priority
+  info.priority_ = static_cast<uint32_t>(ObDiagnoseInfoPrio::DIAGNOSE_PRIORITY_MID);
+  ret = suspect_info_mgr_->add_suspect_info(info.hash(), info);
+  ASSERT_EQ(OB_SUCCESS, ret);
+
+  ret = suspect_info_mgr_->get_with_param(info.hash(), ret_info, allocator);
+  ASSERT_EQ(OB_SUCCESS, ret);
+  // still be high priority
+  ASSERT_EQ(static_cast<uint32_t>(ObDiagnoseInfoPrio::DIAGNOSE_PRIORITY_HIGH), ret_info.priority_);
+}
+
+TEST_F(TestDiagnoseInfoMgr, test_diagnose_tablet_mgr)
+{
+  int ret = OB_SUCCESS;
+  const ObTabletID tablet_id(1);
+  ASSERT_EQ(OB_SUCCESS, ObDiagnoseTabletMgr::mtl_init(diagnose_tablet_mgr_));
+
+  ret = diagnose_tablet_mgr_->add_diagnose_tablet(ls_id_, tablet_id, TYPE_DIAGNOSE_TABLET_MAX);
+  ASSERT_EQ(OB_INVALID_ARGUMENT, ret);
+
+  ret = diagnose_tablet_mgr_->add_diagnose_tablet(ls_id_, tablet_id, TYPE_MINOR_MERGE);
+  ASSERT_EQ(OB_SUCCESS, ret);
+
+  ret = diagnose_tablet_mgr_->add_diagnose_tablet(ls_id_, tablet_id, TYPE_MINOR_MERGE);
+  ASSERT_EQ(OB_SUCCESS, ret);
+  // same diagnose type is registed, return success anyway
+  ret = diagnose_tablet_mgr_->add_diagnose_tablet(ls_id_, tablet_id, TYPE_MINOR_MERGE);
+  ASSERT_EQ(OB_SUCCESS, ret);
+
+  // same diagnose type is registed, return success anyway
+  ret = diagnose_tablet_mgr_->add_diagnose_tablet(ls_id_, tablet_id, TYPE_RS_MAJOR_MERGE);
+  ASSERT_EQ(OB_SUCCESS, ret);
+
+  ret = diagnose_tablet_mgr_->delete_diagnose_tablet(ls_id_, tablet_id, TYPE_MINOR_MERGE);
+  ASSERT_EQ(OB_SUCCESS, ret);
+  ASSERT_EQ(1, diagnose_tablet_mgr_->diagnose_tablet_map_.size());
+  // after remove all flag, registed tablet is deleted
+  ret = diagnose_tablet_mgr_->delete_diagnose_tablet(ls_id_, tablet_id, TYPE_RS_MAJOR_MERGE);
+  ASSERT_EQ(OB_SUCCESS, ret);
+  ASSERT_EQ(0, diagnose_tablet_mgr_->diagnose_tablet_map_.size());
+}
+
+}//end namespace unittest
+}//end namespace oceanbase
+
+int main(int argc, char **argv)
+{
+  system("rm -f test_diagnose_info_mgr.log*");
+  OB_LOGGER.set_file_name("test_diagnose_info_mgr.log");
+  OB_LOGGER.set_log_level("DEBUG");
+  testing::InitGoogleTest(&argc, argv);
+  return RUN_ALL_TESTS();
+}