更新 update 门户站点界面和后台功能

server/unified-management/internal/db/store_test.go

@@ -6,6 +6,7 @@ import (
 	"encoding/json"
 	"os"
 	"path/filepath"
+	"strings"
 	"testing"
 
 	"ymhut-box/server/unified-management/internal/config"
@@ -273,3 +274,75 @@ func TestChangeAdminPasswordRejectsWeakPasswords(t *testing.T) {
 		}
 	}
 }
+
+func TestMySQLSchemaAvoidsTextKeys(t *testing.T) {
+	statements := strings.Join(schemaStatements(dialectFor("mysql")), "\n")
+	for _, forbidden := range []string{
+		"TEXT NOT NULL UNIQUE",
+		"TEXT PRIMARY KEY",
+		"TEXT NOT NULL PRIMARY KEY",
+		"key VARCHAR(191) NOT NULL PRIMARY KEY",
+	} {
+		if strings.Contains(statements, forbidden) {
+			t.Fatalf("mysql schema contains forbidden fragment %q:\n%s", forbidden, statements)
+		}
+	}
+	if !strings.Contains(statements, "`key` VARCHAR(191) NOT NULL PRIMARY KEY") {
+		t.Fatalf("system_settings.key must be quoted for MySQL:\n%s", statements)
+	}
+}
+
+func TestDashboardOverviewKeepsChecksForDeletedSources(t *testing.T) {
+	root := t.TempDir()
+	path := filepath.Join(root, "unified.sqlite")
+	store, err := Open(&config.Config{
+		StorageDir: root,
+		Database: config.DatabaseConfig{
+			Provider:               "sqlite",
+			SQLitePath:             path,
+			FailoverEnabled:        true,
+			HealthIntervalSec:      3600,
+			MaxOpenConns:           1,
+			MaxIdleConns:           1,
+			ConnMaxLifetimeSeconds: 60,
+		},
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer store.Close()
+
+	source, err := store.UpsertSource(Source{
+		CategoryID:    "video",
+		CategoryName:  "视频",
+		SourceID:      "video-demo",
+		Name:          "演示接口",
+		APIURL:        "https://example.com/video.json",
+		Enabled:       true,
+		ClientVisible: true,
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if err := store.RecordSourceCheck(source.ID, "ok", 123, ""); err != nil {
+		t.Fatal(err)
+	}
+	if err := store.DeleteSource(source.SourceID); err != nil {
+		t.Fatal(err)
+	}
+
+	overview, err := store.DashboardOverview(10)
+	if err != nil {
+		t.Fatal(err)
+	}
+	checks, ok := overview["heartbeats"].([]map[string]any)
+	if !ok {
+		t.Fatalf("heartbeats has unexpected type %T", overview["heartbeats"])
+	}
+	if len(checks) != 1 {
+		t.Fatalf("expected deleted source check to remain visible, got %d", len(checks))
+	}
+	if checks[0]["sourceId"] == "" || checks[0]["name"] == "" {
+		t.Fatalf("deleted source check should have fallback sourceId/name: %#v", checks[0])
+	}
+}