chore: add rand sleep ut (#1844)

alibaba · Nov 6, 2024 · 0b6ddd2 · 0b6ddd2
1 parent 67e0d7b
commit 0b6ddd2
Showing 1 changed file with 44 additions and 2 deletions.
diff --git a/core/unittest/prometheus/TargetSubscriberSchedulerUnittest.cpp b/core/unittest/prometheus/TargetSubscriberSchedulerUnittest.cpp
@@ -34,6 +34,7 @@ class TargetSubscriberSchedulerUnittest : public ::testing::Test {
     void OnInitScrapeJobEvent();
     void TestProcess();
     void TestParseTargetGroups();
+    void TestBuildScrapeSchedulerSet();
 
 protected:
     void SetUp() override {
@@ -125,6 +126,25 @@ class TargetSubscriberSchedulerUnittest : public ::testing::Test {
                 "__meta_kubernetes_service_labelpresent_component": "true",
                 "__meta_kubernetes_endpoint_ready": "true"
             }
+        },
+        {
+            "targets": [
+                "192.168.22.33:6443"
+            ],
+            "labels": {
+                "__address__": "192.168.22.33:6443",
+                "__meta_kubernetes_endpoint_port_protocol": "TCP",
+                "__meta_kubernetes_service_label_provider": "kubernetes",
+                "__meta_kubernetes_endpoints_name": "kubernetes",
+                "__meta_kubernetes_service_name": "kubernetes",
+                "__meta_kubernetes_endpoints_labelpresent_endpointslice_kubernetes_io_skip_mirror": "true",
+                "__meta_kubernetes_service_labelpresent_provider": "true",
+                "__meta_kubernetes_endpoint_port_name": "https",
+                "__meta_kubernetes_namespace": "default",
+                "__meta_kubernetes_service_label_component": "apiserver",
+                "__meta_kubernetes_service_labelpresent_component": "true",
+                "__meta_kubernetes_endpoint_ready": "true"
+            }
         }
     ])JSON";
         }
@@ -160,7 +180,7 @@ void TargetSubscriberSchedulerUnittest::TestProcess() {
     // if status code is 200
     mHttpResponse.SetStatusCode(200);
     targetSubscriber->OnSubscription(mHttpResponse, 0);
-    APSARA_TEST_EQUAL(2UL, targetSubscriber->mScrapeSchedulerMap.size());
+    APSARA_TEST_EQUAL(3UL, targetSubscriber->mScrapeSchedulerMap.size());
 }
 
 void TargetSubscriberSchedulerUnittest::TestParseTargetGroups() {
@@ -169,12 +189,34 @@ void TargetSubscriberSchedulerUnittest::TestParseTargetGroups() {
 
     std::vector<Labels> newScrapeSchedulerSet;
     APSARA_TEST_TRUE(targetSubscriber->ParseScrapeSchedulerGroup(*mHttpResponse.GetBody<string>(), newScrapeSchedulerSet));
-    APSARA_TEST_EQUAL(2UL, newScrapeSchedulerSet.size());
+    APSARA_TEST_EQUAL(3UL, newScrapeSchedulerSet.size());
+}
+
+void TargetSubscriberSchedulerUnittest::TestBuildScrapeSchedulerSet() {
+    // prepare data
+    std::shared_ptr<TargetSubscriberScheduler> targetSubscriber = std::make_shared<TargetSubscriberScheduler>();
+    APSARA_TEST_TRUE(targetSubscriber->Init(mConfig["ScrapeConfig"]));
+    std::vector<Labels> newScrapeSchedulerSet;
+    APSARA_TEST_TRUE(targetSubscriber->ParseScrapeSchedulerGroup(*mHttpResponse.GetBody<string>(), newScrapeSchedulerSet));
+    APSARA_TEST_EQUAL(3UL, newScrapeSchedulerSet.size());
+
+    auto result = targetSubscriber->BuildScrapeSchedulerSet(newScrapeSchedulerSet);
+
+    vector<pair<string, chrono::steady_clock::time_point>> startTimeList;
+    startTimeList.reserve(result.size());
+    for (auto& it : result) {
+        startTimeList.emplace_back(it.second->GetId(), it.second->GetNextExecTime());
+    }
+    APSARA_TEST_EQUAL(3UL, startTimeList.size());
+    APSARA_TEST_NOT_EQUAL(startTimeList[0].second, startTimeList[1].second);
+    APSARA_TEST_NOT_EQUAL(startTimeList[1].second, startTimeList[2].second);
+    APSARA_TEST_NOT_EQUAL(startTimeList[0].second, startTimeList[2].second);
 }
 
 UNIT_TEST_CASE(TargetSubscriberSchedulerUnittest, OnInitScrapeJobEvent)
 UNIT_TEST_CASE(TargetSubscriberSchedulerUnittest, TestProcess)
 UNIT_TEST_CASE(TargetSubscriberSchedulerUnittest, TestParseTargetGroups)
+UNIT_TEST_CASE(TargetSubscriberSchedulerUnittest, TestBuildScrapeSchedulerSet)
 
 } // namespace logtail