polish

alibaba · Dec 28, 2024 · 842453b · 842453b
1 parent bc0cd0d
commit 842453b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/core/app_config/AppConfig.h b/core/app_config/AppConfig.h
@@ -534,6 +534,7 @@ class AppConfig {
     friend class InputContainerStdioUnittest;
     friend class BatcherUnittest;
     friend class EnterpriseSLSClientManagerUnittest;
+    friend class FlusherRunnerUnittest;
 #endif
 };
 

diff --git a/core/unittest/sender/FlusherRunnerUnittest.cpp b/core/unittest/sender/FlusherRunnerUnittest.cpp
@@ -31,6 +31,10 @@ class FlusherRunnerUnittest : public ::testing::Test {
     void TestPushToHttpSink();
 
 protected:
+    static void SetUpTestCase() {
+        AppConfig::GetInstance()->mSendRequestGlobalConcurrency = 10;
+    }
+
     void TearDown() override {
         SenderQueueManager::GetInstance()->Clear();
         HttpSink::GetInstance()->mQueue.Clear();
@@ -47,8 +51,6 @@ void FlusherRunnerUnittest::TestDispatch() {
         flusher->SetMetricsRecordRef("name", "1");
         flusher->Init(Json::Value(), tmp);
 
-        AppConfig::GetInstance()->mSendRequestGlobalConcurrency = 10;
-
         auto item = make_unique<SenderQueueItem>("content", 10, flusher.get(), flusher->GetQueueKey());
         auto realItem = item.get();
         flusher->PushToQueue(std::move(item));