mirror of
https://github.com/usememos/memos.git
synced 2025-06-05 22:09:59 +02:00
chore: update package name
This commit is contained in:
@ -18,7 +18,7 @@ import (
|
||||
"github.com/usememos/memos/server"
|
||||
"github.com/usememos/memos/server/profile"
|
||||
"github.com/usememos/memos/store"
|
||||
"github.com/usememos/memos/store/sqlite"
|
||||
"github.com/usememos/memos/store/db"
|
||||
"github.com/usememos/memos/test"
|
||||
)
|
||||
|
||||
@ -31,15 +31,15 @@ type TestingServer struct {
|
||||
|
||||
func NewTestingServer(ctx context.Context, t *testing.T) (*TestingServer, error) {
|
||||
profile := test.GetTestingProfile(t)
|
||||
driver, err := sqlite.NewDriver(profile)
|
||||
dbDriver, err := db.NewDBDriver(profile)
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "failed to create db driver")
|
||||
}
|
||||
if err := driver.Migrate(ctx); err != nil {
|
||||
if err := dbDriver.Migrate(ctx); err != nil {
|
||||
return nil, errors.Wrap(err, "failed to migrate db")
|
||||
}
|
||||
|
||||
store := store.New(driver, profile)
|
||||
store := store.New(dbDriver, profile)
|
||||
server, err := server.NewServer(ctx, profile, store)
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "failed to create server")
|
||||
|
@ -11,30 +11,20 @@ import (
|
||||
_ "modernc.org/sqlite"
|
||||
|
||||
"github.com/usememos/memos/store"
|
||||
"github.com/usememos/memos/store/mysql"
|
||||
"github.com/usememos/memos/store/sqlite"
|
||||
"github.com/usememos/memos/store/db"
|
||||
"github.com/usememos/memos/test"
|
||||
)
|
||||
|
||||
func NewTestingStore(ctx context.Context, t *testing.T) *store.Store {
|
||||
profile := test.GetTestingProfile(t)
|
||||
var driver store.Driver
|
||||
var err error
|
||||
switch profile.Driver {
|
||||
case "sqlite":
|
||||
driver, err = sqlite.NewDriver(profile)
|
||||
case "mysql":
|
||||
driver, err = mysql.NewDriver(profile)
|
||||
default:
|
||||
panic(fmt.Sprintf("unknown db driver: %s", profile.Driver))
|
||||
}
|
||||
dbDriver, err := db.NewDBDriver(profile)
|
||||
if err != nil {
|
||||
fmt.Printf("failed to create db driver, error: %+v\n", err)
|
||||
}
|
||||
if err := driver.Migrate(ctx); err != nil {
|
||||
if err := dbDriver.Migrate(ctx); err != nil {
|
||||
fmt.Printf("failed to migrate db, error: %+v\n", err)
|
||||
}
|
||||
|
||||
store := store.New(driver, profile)
|
||||
store := store.New(dbDriver, profile)
|
||||
return store
|
||||
}
|
||||
|
Reference in New Issue
Block a user