Adam Van Ymeren
7a04f298d2
- update to latest telegram layer - remove some references to fields in tg.Entities that don't exist in the schema - originally added here: https://github.com/beeper/td/commit/820929062a2ba0104397bc01235ab58a9cff780e - referenced here - https://github.com/mautrix/telegramgo/commit/124f0967ed195b5a380c9bd02e170ada9710dde3 - https://github.com/mautrix/telegramgo/commit/4205047aab2e0639217148b5d125bfaab668bd8e
108 lines
2.9 KiB
Go
108 lines
2.9 KiB
Go
package messages
|
|
|
|
import (
|
|
"context"
|
|
"strconv"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/tgmock"
|
|
)
|
|
|
|
func generateMessages(count int) []tg.MessageClass {
|
|
r := make([]tg.MessageClass, 0, count)
|
|
|
|
for i := 0; i < count; i++ {
|
|
r = append(r, &tg.Message{
|
|
ID: i,
|
|
PeerID: &tg.PeerUser{UserID: 10},
|
|
Message: strconv.Itoa(i),
|
|
})
|
|
}
|
|
|
|
return r
|
|
}
|
|
|
|
func messagesClass(r []tg.MessageClass, count int) tg.MessagesMessagesClass {
|
|
return &tg.MessagesChannelMessages{
|
|
Messages: r,
|
|
Count: count,
|
|
}
|
|
}
|
|
|
|
func TestIterator(t *testing.T) {
|
|
ctx := context.Background()
|
|
mock := tgmock.NewRequire(t)
|
|
limit := 10
|
|
totalMessages := 3 * limit
|
|
expected := generateMessages(totalMessages)
|
|
raw := tg.NewClient(mock)
|
|
|
|
mock.ExpectCall(&tg.MessagesSearchRequest{
|
|
Q: "query",
|
|
Peer: &tg.InputPeerSelf{},
|
|
OffsetID: 0,
|
|
FromID: &tg.InputPeerEmpty{},
|
|
Filter: &tg.InputMessagesFilterEmpty{},
|
|
SavedPeerID: &tg.InputPeerEmpty{},
|
|
Limit: limit,
|
|
}).ThenResult(messagesClass(expected[2*limit:3*limit], totalMessages))
|
|
mock.ExpectCall(&tg.MessagesSearchRequest{
|
|
Q: "query",
|
|
Peer: &tg.InputPeerSelf{},
|
|
OffsetID: 20,
|
|
FromID: &tg.InputPeerEmpty{},
|
|
Filter: &tg.InputMessagesFilterEmpty{},
|
|
SavedPeerID: &tg.InputPeerEmpty{},
|
|
Limit: limit,
|
|
}).ThenResult(messagesClass(expected[limit:2*limit], totalMessages))
|
|
mock.ExpectCall(&tg.MessagesSearchRequest{
|
|
Q: "query",
|
|
Peer: &tg.InputPeerSelf{},
|
|
OffsetID: 10,
|
|
FromID: &tg.InputPeerEmpty{},
|
|
Filter: &tg.InputMessagesFilterEmpty{},
|
|
SavedPeerID: &tg.InputPeerEmpty{},
|
|
Limit: limit,
|
|
}).ThenResult(messagesClass(expected[:limit], totalMessages))
|
|
mock.ExpectCall(&tg.MessagesSearchRequest{
|
|
Q: "query",
|
|
Peer: &tg.InputPeerSelf{},
|
|
OffsetID: 0,
|
|
FromID: &tg.InputPeerEmpty{},
|
|
Filter: &tg.InputMessagesFilterEmpty{},
|
|
SavedPeerID: &tg.InputPeerEmpty{},
|
|
Limit: limit,
|
|
}).ThenResult(messagesClass(expected[:0], totalMessages))
|
|
|
|
iter := NewQueryBuilder(raw).Search(&tg.InputPeerSelf{}).
|
|
Filter(&tg.InputMessagesFilterEmpty{}).
|
|
Q("query").BatchSize(10).Iter()
|
|
i := 0
|
|
for iter.Next(ctx) {
|
|
require.Equal(t, expected[len(expected)-i-1], iter.Value().Msg)
|
|
i++
|
|
}
|
|
require.NoError(t, iter.Err())
|
|
require.Equal(t, totalMessages, i)
|
|
|
|
total, err := iter.Total(ctx)
|
|
require.NoError(t, err)
|
|
require.Equal(t, totalMessages, total)
|
|
|
|
mock.ExpectCall(&tg.MessagesSearchRequest{
|
|
Q: "query",
|
|
Peer: &tg.InputPeerSelf{},
|
|
OffsetID: 0,
|
|
FromID: &tg.InputPeerEmpty{},
|
|
Filter: &tg.InputMessagesFilterEmpty{},
|
|
SavedPeerID: &tg.InputPeerEmpty{},
|
|
Limit: 1,
|
|
}).ThenResult(messagesClass(expected[:0], totalMessages))
|
|
total, err = iter.FetchTotal(ctx)
|
|
require.NoError(t, err)
|
|
require.Equal(t, totalMessages, total)
|
|
}
|