fix: suit test (#7752)

TencentBlueKing · Jan 19, 2024 · f4672bc · f4672bc
1 parent be70314
commit f4672bc
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/test/host_server/host_test.go b/src/test/host_server/host_test.go
@@ -102,6 +102,7 @@ var _ = Describe("host test", func() {
 
 	Describe("add host test", func() {
 		It("add host using api", func() {
+			test.DeleteAllHosts()
 			input := map[string]interface{}{
 				"bk_biz_id": bizId,
 				"host_info": map[string]interface{}{

diff --git a/src/test/test.go b/src/test/test.go
@@ -233,7 +233,12 @@ func DeleteAllHosts() {
 func DeleteAllObjects() {
 	ctx := context.Background()
 
-	delCond := mapstr.MapStr{"creator": mapstr.MapStr{common.BKDBNE: "cc_system"}}
+	innerObjs := []string{common.BKInnerObjIDBizSet, common.BKInnerObjIDApp, common.BKInnerObjIDSet,
+		common.BKInnerObjIDModule, common.BKInnerObjIDHost, common.BKInnerObjIDProc, common.BKInnerObjIDPlat,
+		common.BKInnerObjIDProject, common.BKInnerObjIDSwitch, common.BKInnerObjIDRouter, common.BKInnerObjIDBlance,
+		common.BKInnerObjIDFirewall, common.BKInnerObjIDWeblogic, common.BKInnerObjIDTomcat, common.BKInnerObjIDApache}
+
+	delCond := mapstr.MapStr{common.BKObjIDField: mapstr.MapStr{common.BKDBNIN: innerObjs}}
 	objects := make([]metadata.Object, 0)
 	err := GetDB().Table(common.BKTableNameObjDes).Find(delCond).Fields(common.BKObjIDField, common.BkSupplierAccount).
 		All(ctx, &objects)