test CallWithDB added

Forget-C · Dec 4, 2021 · 7f271ef · 7f271ef
1 parent 5fc7e5a
commit 7f271ef
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/test/base_test.go b/test/base_test.go
@@ -49,9 +49,7 @@ func TestBaseSqlDB(t *testing.T) {
 	dbr = db.Model(&BarrierModel{}).Where("gid=?", "gid2").Find(&[]BarrierModel{})
 	asserts.Equal(dbr.RowsAffected, int64(0))
 	barrier.BarrierID = 0
-	tx2, err := db.ToSQLDB().Begin()
-	asserts.Nil(err)
-	err = barrier.Call(tx2, func(tx *sql.Tx) error {
+	err = barrier.CallWithDB(db.ToSQLDB(), func(tx *sql.Tx) error {
 		dtmimp.Logf("submit gid2")
 		return nil
 	})