owpengram-server/internal/store/postgres/channel_multi_pin_integration_test.go

package postgres

import (
	"context"
	"errors"
	"testing"

	"telesrv/internal/domain"
)

// TestChannelMultiPin 验证超级群多置顶模型：多条共存、filterPinned 搜索、
// 上限、unpinAll 批量清除、删除被置顶消息后的缓存重算。
func TestChannelMultiPin(t *testing.T) {
	pool := testPool(t)
	ctx := context.Background()
	suffix := randomSuffix(t)

	users := NewUserStore(pool)
	owner, err := users.Create(ctx, domain.User{AccessHash: 91, Phone: "+1676" + suffix + "01", FirstName: "MultiPinOwner"})
	if err != nil {
		t.Fatalf("create owner: %v", err)
	}
	member, err := users.Create(ctx, domain.User{AccessHash: 92, Phone: "+1676" + suffix + "02", FirstName: "MultiPinMember"})
	if err != nil {
		t.Fatalf("create member: %v", err)
	}
	t.Cleanup(func() {
		_, _ = pool.Exec(ctx, "DELETE FROM users WHERE id = ANY($1::bigint[])", []int64{owner.ID, member.ID})
	})

	channels := NewChannelStore(pool)
	created, err := channels.CreateChannel(ctx, domain.CreateChannelRequest{
		CreatorUserID: owner.ID,
		Title:         "MultiPin " + suffix,
		Megagroup:     true,
		MemberUserIDs: []int64{member.ID},
		Date:          1700000920,
	})
	if err != nil {
		t.Fatalf("create channel: %v", err)
	}
	channelID := created.Channel.ID

	const pinCount = 11
	ids := make([]int, 0, pinCount)
	for i := 0; i < pinCount; i++ {
		sent, err := channels.SendChannelMessage(ctx, domain.SendChannelMessageRequest{
			UserID:    owner.ID,
			ChannelID: channelID,
			RandomID:  int64(852000 + i),
			Message:   "pin-target",
			Date:      1700000921 + i,
		})
		if err != nil {
			t.Fatalf("send %d: %v", i, err)
		}
		ids = append(ids, sent.Message.ID)
	}

	pin := func(id int, pinned bool) (domain.UpdateChannelPinnedMessageResult, error) {
		return channels.UpdatePinnedMessage(ctx, domain.UpdateChannelPinnedMessageRequest{
			UserID:    owner.ID,
			ChannelID: channelID,
			MessageID: id,
			Pinned:    pinned,
			Date:      1700000950,
		})
	}

	// 多条共存：pin 前三条，互不替代。
	for _, id := range ids[:3] {
		res, err := pin(id, true)
		if err != nil {
			t.Fatalf("pin %d: %v", id, err)
		}
		if !res.Event.Pinned || len(res.Event.MessageIDs) != 1 || res.Event.MessageIDs[0] != id {
			t.Fatalf("pin event = %+v, want pinned [%d]", res.Event, id)
		}
	}
	view, err := channels.GetChannel(ctx, owner.ID, channelID)
	if err != nil {
		t.Fatalf("get channel: %v", err)
	}
	if view.Channel.PinnedMessageID != ids[2] {
		t.Fatalf("pinned_message_id = %d, want latest pinned %d", view.Channel.PinnedMessageID, ids[2])
	}
	history, err := channels.ListChannelHistory(ctx, owner.ID, domain.ChannelHistoryFilter{
		ChannelID: channelID, PinnedOnly: true, Limit: 50,
	})
	if err != nil {
		t.Fatalf("filterPinned search: %v", err)
	}
	if len(history.Messages) != 3 {
		t.Fatalf("filterPinned messages = %d, want 3 coexisting pins: %+v", len(history.Messages), history.Messages)
	}
	for _, msg := range history.Messages {
		if !msg.Pinned {
			t.Fatalf("filterPinned message %d lacks pinned flag", msg.ID)
		}
	}
	// 普通历史页的消息行直接携带 pinned 标志（多置顶都标，不只最新）。
	page, err := channels.ListChannelHistory(ctx, member.ID, domain.ChannelHistoryFilter{ChannelID: channelID, Limit: 50})
	if err != nil {
		t.Fatalf("list history: %v", err)
	}
	flagged := 0
	for _, msg := range page.Messages {
		if msg.Pinned {
			flagged++
		}
	}
	if flagged != 3 {
		t.Fatalf("history pinned flags = %d, want 3", flagged)
	}

	// unpin 最新一条：其它置顶保留，缓存回落到次新。
	if _, err := pin(ids[2], false); err != nil {
		t.Fatalf("unpin latest: %v", err)
	}
	view, err = channels.GetChannel(ctx, owner.ID, channelID)
	if err != nil {
		t.Fatalf("get channel after unpin: %v", err)
	}
	if view.Channel.PinnedMessageID != ids[1] {
		t.Fatalf("pinned_message_id after unpin = %d, want %d", view.Channel.PinnedMessageID, ids[1])
	}

	// 无数量上限（对齐官方）：剩余全部 pin 上，共 pinCount 条同时置顶。
	for _, id := range ids[2:] {
		if _, err := pin(id, true); err != nil {
			t.Fatalf("pin %d: %v", id, err)
		}
	}
	history, err = channels.ListChannelHistory(ctx, owner.ID, domain.ChannelHistoryFilter{
		ChannelID: channelID, PinnedOnly: true, Limit: 50,
	})
	if err != nil {
		t.Fatalf("filterPinned all pinned: %v", err)
	}
	if len(history.Messages) != pinCount {
		t.Fatalf("filterPinned messages = %d, want %d coexisting pins", len(history.Messages), pinCount)
	}

	// 删除被置顶的最新消息：置顶集合自动排除，缓存重算。
	latest := ids[pinCount-1]
	if _, err := channels.DeleteChannelMessages(ctx, domain.DeleteChannelMessagesRequest{
		UserID: owner.ID, ChannelID: channelID, IDs: []int{latest}, Date: 1700000961,
	}); err != nil {
		t.Fatalf("delete pinned message: %v", err)
	}
	view, err = channels.GetChannel(ctx, owner.ID, channelID)
	if err != nil {
		t.Fatalf("get channel after delete: %v", err)
	}
	if view.Channel.PinnedMessageID != ids[pinCount-2] {
		t.Fatalf("pinned_message_id after delete = %d, want %d", view.Channel.PinnedMessageID, ids[pinCount-2])
	}

	// unpinAll：一条事件批量携带剩余全部置顶 id，缓存清零，再次调用 no-op。
	res, err := channels.UnpinAllChannelMessages(ctx, domain.UnpinAllChannelMessagesRequest{
		UserID: owner.ID, ChannelID: channelID, Date: 1700000962,
	})
	if err != nil {
		t.Fatalf("unpin all: %v", err)
	}
	if res.Event.Pinned || len(res.Event.MessageIDs) != pinCount-1 {
		t.Fatalf("unpin all event = %+v, want pinned=false with %d ids", res.Event, pinCount-1)
	}
	if res.Channel.PinnedMessageID != 0 {
		t.Fatalf("pinned_message_id after unpin all = %d, want 0", res.Channel.PinnedMessageID)
	}
	if _, err := channels.UnpinAllChannelMessages(ctx, domain.UnpinAllChannelMessagesRequest{
		UserID: owner.ID, ChannelID: channelID, Date: 1700000963,
	}); !errors.Is(err, domain.ErrChannelNotModified) {
		t.Fatalf("unpin all again err = %v, want ErrChannelNotModified", err)
	}
	history, err = channels.ListChannelHistory(ctx, owner.ID, domain.ChannelHistoryFilter{
		ChannelID: channelID, PinnedOnly: true, Limit: 50,
	})
	if err != nil {
		t.Fatalf("filterPinned after unpin all: %v", err)
	}
	if len(history.Messages) != 0 {
		t.Fatalf("filterPinned after unpin all = %+v, want empty", history.Messages)
	}
}