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 index 97723b383..1b8aa0070 100644 --- 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}")