memos/test/store/store.go
Athurg Gooth 5121e9f954
chore: move migration and seed code into driver (#2294)
Move migration and seed code into driver
2023-09-27 11:56:20 +08:00

29 lines
607 B
Go

package teststore
import (
"context"
"fmt"
"testing"
"github.com/usememos/memos/store"
"github.com/usememos/memos/store/sqlite"
"github.com/usememos/memos/test"
// sqlite driver.
_ "modernc.org/sqlite"
)
func NewTestingStore(ctx context.Context, t *testing.T) *store.Store {
profile := test.GetTestingProfile(t)
driver, err := sqlite.NewDriver(profile)
if err != nil {
fmt.Printf("failed to create db driver, error: %+v\n", err)
}
if err := driver.Migrate(ctx); err != nil {
fmt.Printf("failed to migrate db, error: %+v\n", err)
}
store := store.New(driver, profile)
return store
}