patch 4.0

deps/oblib/unittest/rpc/test_obrpc_stat.cpp

@@ -17,29 +17,45 @@ using namespace oceanbase::common;
 using namespace oceanbase::rpc;
 using namespace oceanbase::obrpc;
 
-class TestObrpcStat : public ::testing::Test {
+class TestObrpcStat
+    : public ::testing::Test
+{
 public:
   virtual void SetUp()
-  {}
+  {
+  }
 
   virtual void TearDown()
-  {}
+  {
+  }
 };
 
+namespace oceanbase
+{
+namespace rpc
+{
+RpcStatService *get_stat_srv_by_tenant_id(uint64_t)
+{
+  static RpcStatService stat;
+  return &stat;
+}
+}
+}
+
 TEST_F(TestObrpcStat, Basic)
 {
   RpcStatPiece piece;
   piece.size_ = 100;
   piece.time_ = ObTimeUtility::current_time();
-  RPC_STAT(OB_BOOTSTRAP, piece);
+  RPC_STAT(OB_BOOTSTRAP, 1, piece);
 
   const int64_t idx = ObRpcPacketSet::instance().idx_of_pcode(OB_BOOTSTRAP);
   RpcStatItem item;
-  ASSERT_EQ(OB_SUCCESS, RPC_STAT_GET(idx, item));
+  ASSERT_EQ(OB_SUCCESS, RPC_STAT_GET(idx, 1, item));
   EXPECT_EQ(piece.size_, item.size_);
 }
 
-int main(int argc, char* argv[])
+int main(int argc, char *argv[])
 {
   ::testing::InitGoogleTest(&argc, argv);
   return RUN_ALL_TESTS();