Add test utilities for dependents #30

Merged
roysc merged 5 commits from utils-for-server-tests into main 2024-07-13 11:09:40 +00:00
Showing only changes of commit 6fe622e4bf - Show all commits

View File

@ -24,7 +24,9 @@ import (
"testing"
"github.com/cerc-io/plugeth-statediff/indexer/database/sql"
"github.com/cerc-io/plugeth-statediff/indexer/database/sql/postgres"
"github.com/cerc-io/plugeth-statediff/indexer/shared/schema"
"github.com/jmoiron/sqlx"
)
// DedupFile removes duplicates from the given file
@ -65,6 +67,12 @@ func TearDownDB(t *testing.T, db sql.Database) {
}
}
func ClearSqlxDB(sqlxdb *sqlx.DB) error {
driver := postgres.NewSQLXDriver(context.Background(), sqlxdb)
db := postgres.NewPostgresDB(driver, false)
return ClearDB(db)
}
func ClearDB(db sql.Database) error {
ctx := context.Background()
tx, err := db.Begin(ctx)