mirror of https://github.com/usememos/memos.git
92 lines
2.0 KiB
Go
92 lines
2.0 KiB
Go
package test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net"
|
|
"os"
|
|
"testing"
|
|
|
|
// sqlite driver.
|
|
_ "modernc.org/sqlite"
|
|
|
|
"github.com/joho/godotenv"
|
|
|
|
"github.com/usememos/memos/internal/profile"
|
|
"github.com/usememos/memos/internal/version"
|
|
"github.com/usememos/memos/store"
|
|
"github.com/usememos/memos/store/db"
|
|
)
|
|
|
|
// NewTestingStore creates a new testing store with a fresh database.
|
|
// Each test gets its own isolated database:
|
|
// - SQLite: new temp file per test
|
|
// - MySQL/PostgreSQL: new database per test in shared container
|
|
func NewTestingStore(ctx context.Context, t *testing.T) *store.Store {
|
|
driver := getDriverFromEnv()
|
|
profile := getTestingProfileForDriver(t, driver)
|
|
dbDriver, err := db.NewDBDriver(profile)
|
|
if err != nil {
|
|
t.Fatalf("failed to create db driver: %v", err)
|
|
}
|
|
|
|
store := store.New(dbDriver, profile)
|
|
if err := store.Migrate(ctx); err != nil {
|
|
t.Fatalf("failed to migrate db: %v", err)
|
|
}
|
|
return store
|
|
}
|
|
|
|
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
|
|
}
|
|
|
|
// getTestingProfileForDriver creates a testing profile for a specific driver.
|
|
func getTestingProfileForDriver(t *testing.T, driver string) *profile.Profile {
|
|
// Attempt to load .env file if present (optional, for local development)
|
|
_ = godotenv.Load(".env")
|
|
|
|
// Get a temporary directory for the test data.
|
|
dir := t.TempDir()
|
|
mode := "prod"
|
|
port := getUnusedPort()
|
|
|
|
var dsn string
|
|
switch driver {
|
|
case "sqlite":
|
|
dsn = fmt.Sprintf("%s/memos_%s.db", dir, mode)
|
|
case "mysql":
|
|
dsn = GetMySQLDSN(t)
|
|
case "postgres":
|
|
dsn = GetPostgresDSN(t)
|
|
default:
|
|
t.Fatalf("unsupported driver: %s", driver)
|
|
}
|
|
|
|
return &profile.Profile{
|
|
Mode: mode,
|
|
Port: port,
|
|
Data: dir,
|
|
DSN: dsn,
|
|
Driver: driver,
|
|
Version: version.GetCurrentVersion(mode),
|
|
}
|
|
}
|
|
|
|
func getDriverFromEnv() string {
|
|
driver := os.Getenv("DRIVER")
|
|
if driver == "" {
|
|
driver = "sqlite"
|
|
}
|
|
return driver
|
|
}
|