mirror of
https://github.com/kevin-DL/services.git
synced 2026-01-12 11:15:12 +00:00
115 lines
2.8 KiB
Go
115 lines
2.8 KiB
Go
package handler_test
|
|
|
|
import (
|
|
"sort"
|
|
"strconv"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/micro/services/threads/handler"
|
|
pb "github.com/micro/services/threads/proto"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestListMessages(t *testing.T) {
|
|
h := testHandler(t)
|
|
h.Time = time.Now
|
|
|
|
// seed some data
|
|
var convRsp pb.CreateThreadResponse
|
|
err := h.CreateThread(microAccountCtx(), &pb.CreateThreadRequest{
|
|
Topic: "TestListMessages", GroupId: uuid.New().String(),
|
|
}, &convRsp)
|
|
assert.NoError(t, err)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
msgs := make([]*pb.Message, 50)
|
|
for i := 0; i < len(msgs); i++ {
|
|
var rsp pb.CreateMessageResponse
|
|
err := h.CreateMessage(microAccountCtx(), &pb.CreateMessageRequest{
|
|
ThreadId: convRsp.Thread.Id,
|
|
AuthorId: uuid.New().String(),
|
|
Text: strconv.Itoa(i),
|
|
}, &rsp)
|
|
assert.NoError(t, err)
|
|
msgs[i] = rsp.Message
|
|
}
|
|
|
|
t.Run("MissingThreadID", func(t *testing.T) {
|
|
var rsp pb.ListMessagesResponse
|
|
err := h.ListMessages(microAccountCtx(), &pb.ListMessagesRequest{}, &rsp)
|
|
assert.Equal(t, handler.ErrMissingThreadID, err)
|
|
assert.Nil(t, rsp.Messages)
|
|
})
|
|
|
|
t.Run("NoOffset", func(t *testing.T) {
|
|
var rsp pb.ListMessagesResponse
|
|
err := h.ListMessages(microAccountCtx(), &pb.ListMessagesRequest{
|
|
ThreadId: convRsp.Thread.Id,
|
|
}, &rsp)
|
|
assert.NoError(t, err)
|
|
|
|
if len(rsp.Messages) != handler.DefaultLimit {
|
|
t.Fatalf("Expected %v messages but got %v", handler.DefaultLimit, len(rsp.Messages))
|
|
return
|
|
}
|
|
expected := msgs[25:]
|
|
sortMessages(rsp.Messages)
|
|
for i, msg := range rsp.Messages {
|
|
assertMessagesMatch(t, expected[i], msg)
|
|
}
|
|
})
|
|
|
|
t.Run("LimitSet", func(t *testing.T) {
|
|
var rsp pb.ListMessagesResponse
|
|
err := h.ListMessages(microAccountCtx(), &pb.ListMessagesRequest{
|
|
ThreadId: convRsp.Thread.Id,
|
|
Limit: 10,
|
|
}, &rsp)
|
|
assert.NoError(t, err)
|
|
|
|
if len(rsp.Messages) != 10 {
|
|
t.Fatalf("Expected %v messages but got %v", 10, len(rsp.Messages))
|
|
return
|
|
}
|
|
expected := msgs[40:]
|
|
sortMessages(rsp.Messages)
|
|
for i, msg := range rsp.Messages {
|
|
assertMessagesMatch(t, expected[i], msg)
|
|
}
|
|
})
|
|
|
|
t.Run("OffsetAndLimit", func(t *testing.T) {
|
|
var rsp pb.ListMessagesResponse
|
|
err := h.ListMessages(microAccountCtx(), &pb.ListMessagesRequest{
|
|
ThreadId: convRsp.Thread.Id,
|
|
Limit: 5,
|
|
Offset: 30,
|
|
}, &rsp)
|
|
assert.NoError(t, err)
|
|
|
|
if len(rsp.Messages) != 5 {
|
|
t.Fatalf("Expected %v messages but got %v", 5, len(rsp.Messages))
|
|
return
|
|
}
|
|
expected := msgs[15:20]
|
|
sortMessages(rsp.Messages)
|
|
for i, msg := range rsp.Messages {
|
|
assertMessagesMatch(t, expected[i], msg)
|
|
}
|
|
})
|
|
}
|
|
|
|
// sortMessages by the time they were sent
|
|
func sortMessages(msgs []*pb.Message) {
|
|
sort.Slice(msgs, func(i, j int) bool {
|
|
if msgs[i].SentAt == "" || msgs[j].SentAt == "" {
|
|
return true
|
|
}
|
|
return handler.ParseTime(msgs[i].SentAt).Before(handler.ParseTime(msgs[j].SentAt))
|
|
})
|
|
}
|