20 lines
395 B
Go
20 lines
395 B
Go
|
package dbtest
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"fmt"
|
||
|
|
||
|
"entgo.io/ent/dialect"
|
||
|
)
|
||
|
|
||
|
func TestDatabaseForDialect(ctx context.Context, dbDialect string) (string, DbCloser, error) {
|
||
|
switch dbDialect {
|
||
|
case dialect.Postgres:
|
||
|
return TestPostgresDatabase(ctx)
|
||
|
case dialect.SQLite:
|
||
|
return TestSQLiteDB(), NoOpCloser(nil), nil
|
||
|
default:
|
||
|
return "", nil, fmt.Errorf("unsupported dialect: %s", dbDialect)
|
||
|
}
|
||
|
}
|