Muse
6a5ee8f77b
Display L-UI; panel binary/process l-ui, agent lui-agent. Dirs /usr/local/l-ui, /etc/l-ui, /var/log/l-ui; DB /etc/l-ui/l-ui.db (name file drives it via GetName), probe.db beside it, cert /etc/l-ui/cert, bin /usr/local/l-ui/bin. Services l-ui.service / lui-agent.service, OpenRC lui-agent, l-ui.sh, cmd/lui-agent. Env prefix XUI_ -> LUI_ (code reads new prefix only). Release assets l-ui-linux-* / lui-agent-linux-*; Gitea repo rogue/l-ui is a placeholder pending user confirmation (see selfupdate.go note). Go module path github.com/mhsanaei/3x-ui/v3 intentionally kept. Versions unchanged: panel V2.6.0 / agent A2.5.0.
63 lines
1.7 KiB
Go
63 lines
1.7 KiB
Go
package database
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/mhsanaei/3x-ui/v3/internal/database/model"
|
|
|
|
"gorm.io/driver/postgres"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/logger"
|
|
)
|
|
|
|
func assertClientHwidSchema(t *testing.T, db *gorm.DB) {
|
|
t.Helper()
|
|
if !db.Migrator().HasColumn(&model.ClientRecord{}, "limit_hwid") {
|
|
t.Fatalf("clients.limit_hwid missing")
|
|
}
|
|
if !db.Migrator().HasTable(&model.ClientHwid{}) {
|
|
t.Fatalf("client_hwids table missing")
|
|
}
|
|
for _, col := range []string{"sub_id", "hwid_hash", "first_seen", "last_seen", "user_agent", "device_os", "os_version", "device_model"} {
|
|
if !db.Migrator().HasColumn(&model.ClientHwid{}, col) {
|
|
t.Fatalf("client_hwids.%s missing", col)
|
|
}
|
|
}
|
|
if !db.Migrator().HasIndex(&model.ClientHwid{}, "idx_client_hwids_sub_hash") {
|
|
t.Fatalf("client_hwids unique hash index missing")
|
|
}
|
|
}
|
|
|
|
func TestClientHwidSchemaSQLite(t *testing.T) {
|
|
dbDir := t.TempDir()
|
|
t.Setenv("LUI_DB_FOLDER", dbDir)
|
|
if err := InitDB(filepath.Join(dbDir, "l-ui.db")); err != nil {
|
|
t.Fatalf("InitDB: %v", err)
|
|
}
|
|
t.Cleanup(func() { _ = CloseDB() })
|
|
assertClientHwidSchema(t, GetDB())
|
|
}
|
|
|
|
func TestClientHwidSchemaPostgres(t *testing.T) {
|
|
dsn := strings.TrimSpace(os.Getenv("LUI_TEST_PG_DSN"))
|
|
if dsn == "" {
|
|
t.Skip("set LUI_TEST_PG_DSN to a reachable Postgres to run this test")
|
|
}
|
|
db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{Logger: logger.Discard})
|
|
if err != nil {
|
|
t.Fatalf("open postgres: %v", err)
|
|
}
|
|
sqlDB, err := db.DB()
|
|
if err != nil {
|
|
t.Fatalf("postgres db handle: %v", err)
|
|
}
|
|
t.Cleanup(func() { _ = sqlDB.Close() })
|
|
if err := db.AutoMigrate(&model.ClientRecord{}, &model.ClientHwid{}); err != nil {
|
|
t.Fatalf("automigrate postgres: %v", err)
|
|
}
|
|
assertClientHwidSchema(t, db)
|
|
}
|