DESKTOP-L8OUL9F\gfl0034
d293de34f5
补上会员价格按平台筛选，并修正广场机器人测试的 Feed 调用，使 go test 能通过。 Co-authored-by: Cursor <cursoragent@cursor.com>
96 lines
3.5 KiB
Go
96 lines
3.5 KiB
Go
package tcp
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"strconv"
|
|
"testing"
|
|
|
|
"spark-server/internal/infrastructure/id"
|
|
convlogic "spark-server/internal/module/conversation/logic"
|
|
convmodel "spark-server/internal/module/conversation/model"
|
|
delivermodel "spark-server/internal/module/deliver/model"
|
|
friendlogic "spark-server/internal/module/friend/logic"
|
|
friendmodel "spark-server/internal/module/friend/model"
|
|
inboxmodel "spark-server/internal/module/inbox/model"
|
|
msglogic "spark-server/internal/module/message/logic"
|
|
msgmodel "spark-server/internal/module/message/model"
|
|
userentity "spark-server/internal/module/user/entity"
|
|
"spark-server/internal/module/user/logictest"
|
|
"spark-server/internal/pkg/errs"
|
|
"spark-server/internal/pkg/reqctx"
|
|
"spark-server/internal/transport/tcp/protocol"
|
|
"spark-server/internal/transport/tcp/session"
|
|
)
|
|
|
|
func TestReadMessageRegistered(t *testing.T) {
|
|
svc, b, convID, msgID := readFixture(t)
|
|
srv := &Server{messages: svc, router: NewRouter()}
|
|
srv.registerDefaultHandlers()
|
|
conn := session.NewConn("c", "127.0.0.1", 4)
|
|
conn.Authenticate(b, 9)
|
|
conn.SetRequestID("req-read")
|
|
|
|
missing, err := srv.router.Dispatch(t.Context(), conn, protocol.Frame{
|
|
Version: protocol.CurrentVersion, Command: protocol.CmdReadMessage, Body: []byte(`{}`),
|
|
})
|
|
if missing != nil || !errors.Is(err, errs.ErrInvalidArgument) {
|
|
t.Fatalf("missing fields frame=%v err=%v", missing, err)
|
|
}
|
|
|
|
resp, err := srv.router.Dispatch(t.Context(), conn, protocol.Frame{
|
|
Version: protocol.CurrentVersion, Command: protocol.CmdReadMessage, SequenceID: 4,
|
|
Body: []byte(`{"conversation_id":` + strconv.FormatInt(convID, 10) + `,"msg_id":` + strconv.FormatInt(msgID, 10) + `}`),
|
|
})
|
|
if err != nil || resp == nil || resp.Command != protocol.CmdReadMessage {
|
|
t.Fatalf("read %+v %v", resp, err)
|
|
}
|
|
env, err := protocol.DecodeEnvelope(resp.Body)
|
|
if err != nil || env.Code != protocol.CodeOK {
|
|
t.Fatalf("envelope %+v %v", env, err)
|
|
}
|
|
var data msglogic.ReadResult
|
|
if err := json.Unmarshal(env.Data, &data); err != nil || data.MsgID != msgID || data.ConversationID != convID {
|
|
t.Fatalf("data %+v %v", data, err)
|
|
}
|
|
}
|
|
|
|
func readFixture(t *testing.T) (*msglogic.Service, int64, int64, int64) {
|
|
t.Helper()
|
|
ids, err := id.New(1)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
users := logictest.Users(ids, nil)
|
|
chats := convmodel.NewMemoryP2P()
|
|
p2p := convlogic.NewP2PService(chats, ids)
|
|
q := delivermodel.NewMemoryQueue()
|
|
msgs := msgmodel.NewMemoryRepo()
|
|
friends := friendlogic.NewService(friendmodel.NewMemoryRepo(), users, ids, friendlogic.NewGreeter(p2p, msgs, ids, q))
|
|
svc := msglogic.NewService(msgs, chats, p2p, friends, nil, ids, q)
|
|
svc.BindInbox(inboxmodel.NewMemoryStore())
|
|
ctx := reqctx.With(t.Context(), &reqctx.RequestContext{Language: "zh-CN"})
|
|
a, err := users.AuthenticateEmail(ctx, "test01@spark.com", "1", &userentity.LoginDevice{DeviceID: 1, DeviceType: userentity.DeviceIOS})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
b, err := users.AuthenticateEmail(ctx, "test02@spark.com", "1", &userentity.LoginDevice{DeviceID: 2, DeviceType: userentity.DeviceWeb})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
applied, err := friends.Apply(ctx, a.ID, &friendlogic.ApplyInput{UID: b.ID})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := friends.Accept(ctx, b.ID, friendlogic.AcceptInput{RequestID: applied.RequestID}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
sent, err := svc.SendMessage(ctx, a.ID, 1, &msglogic.SendInput{
|
|
PeerID: b.ID, ClientMsgID: "read-1", Content: json.RawMessage(`{"t":1,"text":{"body":"hi"}}`),
|
|
})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
return svc, b.ID, sent.ConversationID, sent.MsgID
|
|
}
|