feature: 提供 Agent 包管理后台接口 (closed TencentBlueKing#1683)

ping15 · Nov 20, 2023 · 59bf39d · 59bf39d
1 parent 8994249
commit 59bf39d
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/apps/node_man/tests/test_views/test_package_manage_views.py b/apps/node_man/tests/test_views/test_package_manage_views.py
@@ -31,7 +31,7 @@ def clear_agent_data(cls):
         GsePackageDesc.objects.all().delete()
         Tag.objects.all().delete()
 
-    @patch("apps.node_man.handlers.cmdb.get_request_username", return_value="admin")
+    @patch("apps.node_man.permissions.package_manage.PackageManagePermission.has_permission", return_value=True)
     def test_list(self, *args, **kwargs):
         # 和之前的builder.make加起来100
         create_gse_package(99, start_id=1000)
@@ -46,7 +46,7 @@ def test_list(self, *args, **kwargs):
         self.assertEqual(result["data"]["total"], 100)
         self.assertEqual(len(result["data"]["list"]), 2)
 
-    @patch("apps.node_man.handlers.cmdb.get_request_username", return_value="admin")
+    @patch("apps.node_man.permissions.package_manage.PackageManagePermission.has_permission", return_value=True)
     def test_list_with_filter_condition(self, *args, **kwargs):
         # 不筛选
         result = self.client.get(path="/api/agent/package/")
@@ -122,21 +122,21 @@ def collect_all_tag_names(cls, tags):
 
         return tag_name_set
 
-    @patch("apps.node_man.handlers.cmdb.get_request_username", return_value="admin")
+    @patch("apps.node_man.permissions.package_manage.PackageManagePermission.has_permission", return_value=True)
     def test_update(self, *args, **kwargs):
         first_gse_package = GsePackages.objects.first()
         self.assertEqual(first_gse_package.is_ready, True)
         self.client.put(path=f"/api/agent/package/{first_gse_package.id}/", data={"is_ready": False})
         self.assertEqual(GsePackages.objects.first().is_ready, False)
 
-    @patch("apps.node_man.handlers.cmdb.get_request_username", return_value="admin")
+    @patch("apps.node_man.permissions.package_manage.PackageManagePermission.has_permission", return_value=True)
     def test_destroy(self, *args, **kwargs):
         gse_packages = GsePackages.objects.all()
         self.assertEqual(len(gse_packages), 1)
         self.client.delete(path=f"/api/agent/package/{gse_packages.first().id}/")
         self.assertEqual(len(GsePackages.objects.all()), 0)
 
-    @patch("apps.node_man.handlers.cmdb.get_request_username", return_value="admin")
+    @patch("apps.node_man.permissions.package_manage.PackageManagePermission.has_permission", return_value=True)
     def test_quick_search_condition(self, *args, **kwargs):
         result = self.client.get(path="/api/agent/package/quick_search_condition/")
         print(f"result2={result}")
@@ -158,7 +158,7 @@ def test_quick_search_condition(self, *args, **kwargs):
                     ],
                 )
 
-    @patch("apps.node_man.handlers.cmdb.get_request_username", return_value="admin")
+    @patch("apps.node_man.permissions.package_manage.PackageManagePermission.has_permission", return_value=True)
     def test_filter_condition_with_agent_pkg_manage(self, *args, **kwargs):
         result = MetaHandler().filter_condition("agent_pkg_manage")
         print(f"result3={result}")