feature: Agent 配置重载功能优化 (closed TencentBlueKing#1769)

wyyalt · Oct 25, 2023 · 5de5615 · 5de5615
1 parent cc843c6
commit 5de5615
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/apps/backend/tests/components/collections/agent_new/test_render_and_push_gse_config.py b/apps/backend/tests/components/collections/agent_new/test_render_and_push_gse_config.py
@@ -11,6 +11,7 @@
 import base64
 
 from apps.backend.components.collections.agent_new import components
+from apps.backend.constants import ProxyConfigFile
 from apps.backend.subscription.steps.agent_adapter import legacy
 from apps.node_man import constants
 from apps.node_man.models import Host
@@ -75,8 +76,8 @@ def get_push_config_file_query_params(self):
     def tearDown(self) -> None:
 
         push_config_file_query_params = self.get_push_config_file_query_params()
-        # 1.0Proxy仅下发7个配置文件
-        self.assertEqual(len(push_config_file_query_params["file_list"]), 7)
+        # 对比1.0Proxy 配置文件
+        self.assertEqual([i["file_name"] for i in push_config_file_query_params["file_list"]], ProxyConfigFile.V1.value)
 
 
 class RenderAndPushGseV2ProxyGseConfigTestCase(RenderAndPushGseV1ProxyGseConfigTestCase):
@@ -92,5 +93,5 @@ def structure_common_inputs(self):
     def tearDown(self) -> None:
 
         push_config_file_query_params = self.get_push_config_file_query_params()
-        # 2.0Proxy仅下发3个配置文件
-        self.assertEqual(len(push_config_file_query_params["file_list"]), 3)
+        # 对比2.0Proxy配置文件
+        self.assertEqual([i["file_name"] for i in push_config_file_query_params["file_list"]], ProxyConfigFile.V2.value)