memos/test/test.go

59 lines
1.1 KiB
Go
Raw Normal View History

2023-04-01 22:47:19 +08:00
package test
import (
"fmt"
"net"
2023-09-29 09:15:54 +08:00
"os"
2023-04-01 22:47:19 +08:00
"testing"
2024-01-06 16:55:13 +08:00
"github.com/joho/godotenv"
2023-04-01 22:47:19 +08:00
"github.com/usememos/memos/server/profile"
"github.com/usememos/memos/server/version"
)
func getUnusedPort() int {
// Get a random unused port
listener, err := net.Listen("tcp", "localhost:0")
if err != nil {
panic(err)
}
defer listener.Close()
// Get the port number
port := listener.Addr().(*net.TCPAddr).Port
return port
}
2023-04-01 22:47:19 +08:00
func GetTestingProfile(t *testing.T) *profile.Profile {
2024-01-06 16:55:13 +08:00
if err := godotenv.Load(".env"); err != nil {
t.Fatal("failed to load .env file", err)
}
2023-04-01 22:47:19 +08:00
// Get a temporary directory for the test data.
dir := t.TempDir()
mode := "dev"
port := getUnusedPort()
2023-09-29 09:15:54 +08:00
driver := getDriverFromEnv()
dsn := os.Getenv("DSN")
if driver == "sqlite" {
dsn = fmt.Sprintf("%s/memos_%s.db", dir, mode)
}
2023-04-01 22:47:19 +08:00
return &profile.Profile{
Mode: mode,
Port: port,
2023-04-01 22:47:19 +08:00
Data: dir,
2023-09-29 09:15:54 +08:00
DSN: dsn,
Driver: driver,
2023-04-01 22:47:19 +08:00
Version: version.GetCurrentVersion(mode),
}
}
2023-09-29 09:15:54 +08:00
func getDriverFromEnv() string {
driver := os.Getenv("DRIVER")
if driver == "" {
driver = "sqlite"
}
return driver
}