mirror of
https://github.com/usememos/memos.git
synced 2024-11-17 04:07:30 +08:00
31 lines
692 B
Go
31 lines
692 B
Go
package teststore
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/usememos/memos/store"
|
|
"github.com/usememos/memos/store/db"
|
|
"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)
|
|
db := db.NewDB(profile)
|
|
if err := db.Open(); err != nil {
|
|
fmt.Printf("failed to open db, error: %+v\n", err)
|
|
}
|
|
if err := db.Migrate(ctx); err != nil {
|
|
fmt.Printf("failed to migrate db, error: %+v\n", err)
|
|
}
|
|
|
|
driver := sqlite.NewDriver(db.DBInstance)
|
|
|
|
store := store.New(db.DBInstance, driver, profile)
|
|
return store
|
|
}
|