mirror of
https://github.com/kevin-DL/services.git
synced 2026-01-12 03:05:14 +00:00
102 lines
2.7 KiB
Go
102 lines
2.7 KiB
Go
package handler_test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/micro/services/streams/handler"
|
|
pb "github.com/micro/services/streams/proto"
|
|
"github.com/stretchr/testify/assert"
|
|
"google.golang.org/protobuf/types/known/wrapperspb"
|
|
)
|
|
|
|
func TestRecentMessages(t *testing.T) {
|
|
h := testHandler(t)
|
|
h.Time = time.Now
|
|
|
|
// seed some data
|
|
ids := make([]string, 3)
|
|
convos := make(map[string][]*pb.Message, 3)
|
|
for i := 0; i < 3; i++ {
|
|
var convRsp pb.CreateConversationResponse
|
|
err := h.CreateConversation(context.TODO(), &pb.CreateConversationRequest{
|
|
Topic: "TestRecentMessages", GroupId: uuid.New().String(),
|
|
}, &convRsp)
|
|
assert.NoError(t, err)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
convos[convRsp.Conversation.Id] = make([]*pb.Message, 50)
|
|
ids[i] = convRsp.Conversation.Id
|
|
|
|
for j := 0; j < 50; j++ {
|
|
var rsp pb.CreateMessageResponse
|
|
err := h.CreateMessage(context.TODO(), &pb.CreateMessageRequest{
|
|
ConversationId: convRsp.Conversation.Id,
|
|
AuthorId: uuid.New().String(),
|
|
Text: fmt.Sprintf("Conversation %v, Message %v", i, j),
|
|
}, &rsp)
|
|
assert.NoError(t, err)
|
|
convos[convRsp.Conversation.Id][j] = rsp.Message
|
|
}
|
|
}
|
|
|
|
t.Run("MissingConversationIDs", func(t *testing.T) {
|
|
var rsp pb.RecentMessagesResponse
|
|
err := h.RecentMessages(context.TODO(), &pb.RecentMessagesRequest{}, &rsp)
|
|
assert.Equal(t, handler.ErrMissingConversationIDs, err)
|
|
assert.Nil(t, rsp.Messages)
|
|
})
|
|
|
|
t.Run("LimitSet", func(t *testing.T) {
|
|
var rsp pb.RecentMessagesResponse
|
|
err := h.RecentMessages(context.TODO(), &pb.RecentMessagesRequest{
|
|
ConversationIds: ids,
|
|
LimitPerConversation: &wrapperspb.Int32Value{Value: 10},
|
|
}, &rsp)
|
|
assert.NoError(t, err)
|
|
|
|
if len(rsp.Messages) != 30 {
|
|
t.Fatalf("Expected %v messages but got %v", 30, len(rsp.Messages))
|
|
return
|
|
}
|
|
var expected []*pb.Message
|
|
for _, msgs := range convos {
|
|
expected = append(expected, msgs[40:]...)
|
|
}
|
|
sortMessages(expected)
|
|
sortMessages(rsp.Messages)
|
|
for i, msg := range rsp.Messages {
|
|
assertMessagesMatch(t, expected[i], msg)
|
|
}
|
|
})
|
|
|
|
t.Run("NoLimitSet", func(t *testing.T) {
|
|
reducedIDs := ids[:2]
|
|
|
|
var rsp pb.RecentMessagesResponse
|
|
err := h.RecentMessages(context.TODO(), &pb.RecentMessagesRequest{
|
|
ConversationIds: reducedIDs,
|
|
}, &rsp)
|
|
assert.NoError(t, err)
|
|
|
|
if len(rsp.Messages) != 50 {
|
|
t.Fatalf("Expected %v messages but got %v", 50, len(rsp.Messages))
|
|
return
|
|
}
|
|
var expected []*pb.Message
|
|
for _, id := range reducedIDs {
|
|
expected = append(expected, convos[id][25:]...)
|
|
}
|
|
sortMessages(expected)
|
|
sortMessages(rsp.Messages)
|
|
for i, msg := range rsp.Messages {
|
|
assertMessagesMatch(t, expected[i], msg)
|
|
}
|
|
})
|
|
}
|