chore: adjust store test for mysql

This commit is contained in:
Steven
2023-09-29 09:15:54 +08:00
parent 255254eb69
commit 87ddeb2c79
13 changed files with 359 additions and 163 deletions

View File

@@ -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
}