45 lines
1.8 KiB
Go
45 lines
1.8 KiB
Go
package mtprotoedge
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/gotd/td/mt"
|
|
"github.com/gotd/td/proto"
|
|
"github.com/gotd/td/tg"
|
|
)
|
|
|
|
// TestNewSessionCreatedUniqueIDPerSession 验证两次 session 建立收到的
|
|
// new_session_created.unique_id 互不相同。客户端按 unique_id 去重，复用同一值
|
|
// 会让断线重连后的 new_session_created 被吞掉，依赖它触发的差分补拉随之丢失。
|
|
func TestNewSessionCreatedUniqueIDPerSession(t *testing.T) {
|
|
const dc = 2
|
|
addr, pub, _ := startTestServer(t, Options{DC: dc})
|
|
conn, auth, cipher := dialHandshake(t, addr, dc, pub)
|
|
|
|
clientMsgID := proto.NewMessageIDGen(time.Now)
|
|
firstMsgID := clientMsgID.New(proto.MessageFromClient)
|
|
sendEncryptedWithSeq(t, conn, cipher, auth, firstMsgID, 1, &tg.HelpGetConfigRequest{})
|
|
first := collectReplies(t, conn, cipher, auth.AuthKey, mt.NewSessionCreatedTypeID)
|
|
var created1 mt.NewSessionCreated
|
|
if err := created1.Decode(mustHave(t, first, mt.NewSessionCreatedTypeID, "first new_session_created")); err != nil {
|
|
t.Fatalf("decode first new_session_created: %v", err)
|
|
}
|
|
|
|
nextSessionID := auth.SessionID + 1
|
|
if nextSessionID == 0 {
|
|
nextSessionID++
|
|
}
|
|
secondMsgID := clientMsgID.New(proto.MessageFromClient)
|
|
body := encodeClientMessageBodyForTest(t, &tg.HelpGetConfigRequest{})
|
|
sendEncryptedWithSessionSaltAndSeq(t, conn, cipher, auth, nextSessionID, auth.ServerSalt, secondMsgID, 1, body)
|
|
second := collectReplies(t, conn, cipher, auth.AuthKey, mt.NewSessionCreatedTypeID)
|
|
var created2 mt.NewSessionCreated
|
|
if err := created2.Decode(mustHave(t, second, mt.NewSessionCreatedTypeID, "second new_session_created")); err != nil {
|
|
t.Fatalf("decode second new_session_created: %v", err)
|
|
}
|
|
|
|
if created1.UniqueID == created2.UniqueID {
|
|
t.Fatalf("new_session_created.unique_id reused across sessions: %d", created1.UniqueID)
|
|
}
|
|
}
|