chore: fix testing profile

This commit is contained in:
Steven 2023-12-17 13:59:15 +08:00
parent 9173c8f19a
commit a9071d629a

View File

@ -3,6 +3,7 @@ package test
import ( import (
"fmt" "fmt"
"net" "net"
"os"
"testing" "testing"
"github.com/yourselfhosted/slash/server/profile" "github.com/yourselfhosted/slash/server/profile"
@ -27,11 +28,25 @@ func GetTestingProfile(t *testing.T) *profile.Profile {
dir := t.TempDir() dir := t.TempDir()
mode := "dev" mode := "dev"
port := getUnusedPort() port := getUnusedPort()
driver := getDriverFromEnv()
dsn := os.Getenv("DSN")
if driver == "sqlite" {
dsn = fmt.Sprintf("%s/slash_%s.db", dir, mode)
}
return &profile.Profile{ return &profile.Profile{
Mode: mode, Mode: mode,
Port: port, Port: port,
Data: dir, Data: dir,
DSN: fmt.Sprintf("%s/slash_%s.db", dir, mode), DSN: dsn,
Driver: driver,
Version: version.GetCurrentVersion(mode), Version: version.GetCurrentVersion(mode),
} }
} }
func getDriverFromEnv() string {
driver := os.Getenv("DRIVER")
if driver == "" {
driver = "sqlite"
}
return driver
}