From b443aa9ca19734af19e1b0ac9b64d6a74d36a4f0 Mon Sep 17 00:00:00 2001 From: xinran chen Date: Tue, 11 Jul 2023 23:48:48 +0800 Subject: [PATCH] fix --- miner-api/test/prover_task_test.go | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/miner-api/test/prover_task_test.go b/miner-api/test/prover_task_test.go index f1c4a81182..dc94758ee5 100644 --- a/miner-api/test/prover_task_test.go +++ b/miner-api/test/prover_task_test.go @@ -14,6 +14,7 @@ import ( "scroll-tech/common/database" "scroll-tech/common/docker" "scroll-tech/common/types" + "scroll-tech/database/migrate" "scroll-tech/miner-api/cmd/app" "scroll-tech/miner-api/internal/config" "scroll-tech/miner-api/internal/orm" @@ -105,11 +106,14 @@ var ( ) func insertSomeProverTasks(t *testing.T, db *gorm.DB) { - err := db.AutoMigrate(new(orm.ProverTask)) + sqlDB, err := db.DB() assert.NoError(t, err) + assert.NoError(t, migrate.ResetDB(sqlDB)) + ptdb := orm.NewProverTask(db) err = ptdb.SetProverTask(context.Background(), &task1) assert.NoError(t, err) + err = ptdb.SetProverTask(context.Background(), &task2) assert.NoError(t, err) }