对接

2026-09-23 17:42:55 +08:00
parent 53e8599bdd
commit 0a8ecbe09b
75 changed files with 2299 additions and 276 deletions
@@ -74,3 +74,39 @@ func TestCatalogRepoScansCategoryAndRoomFields(t *testing.T) {
 		t.Fatalf("room get scan result=%+v", got)
 	}
 }
+
+func TestCatalogRepoFiltersRoomsByCategoryEnabled(t *testing.T) {
+	db := openCatalogTestDB(t)
+	if err := db.Exec(`INSERT INTO sys_users(id, username, nick_name, header_img) VALUES (2, 'anchor', 'Anchor', 'avatar.png')`).Error; err != nil {
+		t.Fatal(err)
+	}
+	enabledCategory := categoryPO{Name: "Games", Enabled: true}
+	disabledCategory := categoryPO{Name: "Hidden", Enabled: false}
+	if err := db.Create(&enabledCategory).Error; err != nil {
+		t.Fatal(err)
+	}
+	if err := db.Create(&disabledCategory).Error; err != nil {
+		t.Fatal(err)
+	}
+	if err := db.Model(&disabledCategory).Update("enabled", false).Error; err != nil {
+		t.Fatal(err)
+	}
+	for _, room := range []*roomPO{
+		{Name: "Visible", CategoryID: enabledCategory.ID, CoverURL: "visible.png", AnchorUserID: 2, Enabled: true},
+		{Name: "Hidden", CategoryID: disabledCategory.ID, CoverURL: "hidden.png", AnchorUserID: 2, Enabled: true},
+	} {
+		if err := db.Create(room).Error; err != nil {
+			t.Fatal(err)
+		}
+	}
+
+	repo := NewCatalogRepo(catalogTestProvider{db: db})
+	enabled := true
+	rooms, total, err := repo.ListRooms(context.Background(), bizlive.RoomFilter{CategoryEnabled: &enabled}, bizlive.ListOptions{Limit: 10})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if total != 1 || len(rooms) != 1 || rooms[0].Name != "Visible" {
+		t.Fatalf("enabled-category rooms total=%d rooms=%+v", total, rooms)
+	}
+}