package mtprotoedge

import (
	"testing"

	"go.uber.org/zap/zaptest"
)

// TestReceivesUpdatesForAuthKeyRequiresMembershipSync 验证「完全就绪」查询同时要求
// receivesUpdates 与 channel membership 路由建立成功。membership 同步失败时该查询
// 必须返回 false，让按 RPC 置位的短路放行重试——否则该 session 会以「已置位但
// byMemberChannel 缺失」的状态静默漏收超级群推送，且 channel 维度没有 pts 缺口
// 信号可供客户端自愈。
func TestReceivesUpdatesForAuthKeyRequiresMembershipSync(t *testing.T) {
	sm := NewSessionManager(zaptest.NewLogger(t))
	raw := [8]byte{1, 2, 3}
	c := &Conn{sessionID: 42, authKeyID: raw}
	sm.Register(c)
	sm.BindUserForAuthKey(raw, 42, 100)

	sm.SetReceivesUpdatesForAuthKey(raw, 42, true)
	if sm.ReceivesUpdatesForAuthKey(raw, 42) {
		t.Fatal("ready before membership sync — a failed sync would never be retried")
	}

	sm.SetSessionChannelMemberships(raw, 42, 100, []int64{7})
	if !sm.ReceivesUpdatesForAuthKey(raw, 42) {
		t.Fatal("not ready after successful membership sync")
	}

	// 没有任何频道的账号：空列表的成功同步同样算就绪。
	sm.SetSessionChannelMemberships(raw, 42, 100, nil)
	if !sm.ReceivesUpdatesForAuthKey(raw, 42) {
		t.Fatal("not ready after successful empty membership sync")
	}

	// userID 与连接当前绑定不一致（换号竞态）时不算就绪，等正确身份重试。
	sm.SetSessionChannelMemberships(raw, 42, 999, []int64{7})
	if sm.ReceivesUpdatesForAuthKey(raw, 42) {
		t.Fatal("ready after membership sync for mismatched user")
	}
	sm.SetSessionChannelMemberships(raw, 42, 100, []int64{7})

	// 登出清除就绪标志。
	sm.BindUserForAuthKey(raw, 42, 0)
	if sm.ReceivesUpdatesForAuthKey(raw, 42) {
		t.Fatal("still ready after user unbind")
	}
}