mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-27 17:25:16 +00:00
45 lines
829 B
Go
45 lines
829 B
Go
|
package service
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"testing"
|
||
|
|
||
|
"github.com/c9s/rockhopper"
|
||
|
"github.com/stretchr/testify/assert"
|
||
|
)
|
||
|
|
||
|
func prepareDB(t *testing.T) (*rockhopper.DB, error) {
|
||
|
dialect, err := rockhopper.LoadDialect("sqlite3")
|
||
|
if !assert.NoError(t, err) {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
assert.NotNil(t, dialect)
|
||
|
|
||
|
db, err := rockhopper.Open("sqlite3", dialect, ":memory:")
|
||
|
if !assert.NoError(t, err) {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
assert.NotNil(t, db)
|
||
|
|
||
|
_, err = db.CurrentVersion()
|
||
|
if !assert.NoError(t, err) {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
var loader rockhopper.SqlMigrationLoader
|
||
|
migrations, err := loader.Load("../../migrations/sqlite3")
|
||
|
if !assert.NoError(t, err) {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
assert.NotEmpty(t, migrations)
|
||
|
|
||
|
ctx := context.Background()
|
||
|
err = rockhopper.Up(ctx, db, migrations, 0, 0)
|
||
|
assert.NoError(t, err)
|
||
|
|
||
|
return db, err
|
||
|
}
|