mirror of
https://github.com/usememos/memos.git
synced 2025-06-05 22:09:59 +02:00
chore: adjust store test for mysql
This commit is contained in:
18
test/test.go
18
test/test.go
@@ -3,6 +3,7 @@ package test
|
||||
import (
|
||||
"fmt"
|
||||
"net"
|
||||
"os"
|
||||
"testing"
|
||||
|
||||
"github.com/usememos/memos/server/profile"
|
||||
@@ -27,11 +28,26 @@ func GetTestingProfile(t *testing.T) *profile.Profile {
|
||||
dir := t.TempDir()
|
||||
mode := "dev"
|
||||
port := getUnusedPort()
|
||||
driver := getDriverFromEnv()
|
||||
dsn := os.Getenv("DSN")
|
||||
if driver == "sqlite" {
|
||||
dsn = fmt.Sprintf("%s/memos_%s.db", dir, mode)
|
||||
}
|
||||
println("dsn", dsn, driver)
|
||||
return &profile.Profile{
|
||||
Mode: mode,
|
||||
Port: port,
|
||||
Data: dir,
|
||||
DSN: fmt.Sprintf("%s/memos_%s.db", dir, mode),
|
||||
DSN: dsn,
|
||||
Driver: driver,
|
||||
Version: version.GetCurrentVersion(mode),
|
||||
}
|
||||
}
|
||||
|
||||
func getDriverFromEnv() string {
|
||||
driver := os.Getenv("DRIVER")
|
||||
if driver == "" {
|
||||
driver = "sqlite"
|
||||
}
|
||||
return driver
|
||||
}
|
||||
|
Reference in New Issue
Block a user