fix test

netbirdio · Jul 12, 2024 · a5533c7 · a5533c7
1 parent 5ebbef5
commit a5533c7
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/management/server/account_test.go b/management/server/account_test.go
@@ -1256,6 +1256,16 @@ func TestAccountManager_NetworkUpdates_DeleteGroup(t *testing.T) {
 		},
 	}
 
+	if err := manager.DeletePolicy(context.Background(), account.Id, account.Policies[0].ID, userID); err != nil {
+		t.Errorf("delete default rule: %v", err)
+		return
+	}
+
+	if err := manager.SavePolicy(context.Background(), account.Id, userID, &policy); err != nil {
+		t.Errorf("save policy: %v", err)
+		return
+	}
+
 	wg := sync.WaitGroup{}
 	wg.Add(1)
 	go func() {
@@ -1269,7 +1279,10 @@ func TestAccountManager_NetworkUpdates_DeleteGroup(t *testing.T) {
 	}()
 
 	// clean policy is pre requirement for delete group
-	_ = manager.DeletePolicy(context.Background(), account.Id, policy.ID, userID)
+	if err := manager.DeletePolicy(context.Background(), account.Id, policy.ID, userID); err != nil {
+		t.Errorf("delete default rule: %v", err)
+		return
+	}
 
 	if err := manager.DeleteGroup(context.Background(), account.Id, "", group.ID); err != nil {
 		t.Errorf("delete group: %v", err)