move gotd fork into repo. (#111)

- 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

pkg/gotd/telegram/query/messages/iter_test.go

@@ -0,0 +1,107 @@
+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)
+}