chore(ci): fix ci

PromptPal · Mar 11, 2024 · 7039400 · 7039400
1 parent ff8c667
commit 7039400
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/schema/call_test.go b/schema/call_test.go
@@ -11,7 +11,9 @@ import (
 	"time"
 
 	"github.com/PromptPal/PromptPal/config"
+	"github.com/PromptPal/PromptPal/ent/project"
 	"github.com/PromptPal/PromptPal/ent/prompt"
+	"github.com/PromptPal/PromptPal/ent/promptcall"
 	"github.com/PromptPal/PromptPal/ent/schema"
 	dbSchema "github.com/PromptPal/PromptPal/ent/schema"
 	"github.com/PromptPal/PromptPal/routes"
@@ -295,6 +297,8 @@ func (s *callTestSuite) TestPerformCall() {
 }
 
 func (s *callTestSuite) TearDownSuite() {
+	service.EntClient.PromptCall.Delete().Where(promptcall.HasProjectWith(project.ID(s.pjID))).ExecX(context.Background())
+	service.EntClient.Prompt.DeleteOneID(s.promptID).ExecX(context.Background())
 	service.EntClient.Project.DeleteOneID(s.pjID).ExecX(context.Background())
 	service.Close()
 }