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
78 lines
2.0 KiB
Go
78 lines
2.0 KiB
Go
package message
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/bin"
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
|
|
)
|
|
|
|
func TestBuilder_Album(t *testing.T) {
|
|
ctx := context.Background()
|
|
sender, mock := testSender(t)
|
|
loc := &tg.InputPhoto{
|
|
ID: 10,
|
|
}
|
|
|
|
mock.ExpectFunc(func(b bin.Encoder) {
|
|
req, ok := b.(*tg.MessagesSendMultiMediaRequest)
|
|
require.True(t, ok)
|
|
require.Equal(t, &tg.InputPeerSelf{}, req.Peer)
|
|
require.Len(t, req.MultiMedia, 2)
|
|
for i := range req.MultiMedia {
|
|
require.Equal(t, req.MultiMedia[i].Media, &tg.InputMediaPhoto{ID: loc})
|
|
require.NotZero(t, req.MultiMedia[i].RandomID)
|
|
}
|
|
}).ThenResult(&tg.Updates{})
|
|
_, err := sender.Self().Album(ctx, Photo(loc), Photo(loc))
|
|
require.NoError(t, err)
|
|
|
|
doc := &tg.InputDocument{
|
|
ID: 10,
|
|
}
|
|
mock.ExpectFunc(func(b bin.Encoder) {
|
|
req, ok := b.(*tg.MessagesSendMultiMediaRequest)
|
|
require.True(t, ok)
|
|
require.Equal(t, &tg.InputPeerSelf{}, req.Peer)
|
|
require.Len(t, req.MultiMedia, 2)
|
|
for i := range req.MultiMedia {
|
|
require.Equal(t, req.MultiMedia[i].Media, &tg.InputMediaDocument{ID: doc})
|
|
require.NotZero(t, req.MultiMedia[i].RandomID)
|
|
}
|
|
}).ThenResult(&tg.Updates{})
|
|
_, err = sender.Self().Album(ctx, Document(doc), Document(doc))
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestBuilder_UploadMedia(t *testing.T) {
|
|
ctx := context.Background()
|
|
sender, mock := testSender(t)
|
|
file := &tg.InputFile{
|
|
ID: 10,
|
|
}
|
|
expected := &tg.MessageMediaEmpty{}
|
|
|
|
mock.ExpectCall(&tg.MessagesUploadMediaRequest{
|
|
Peer: &tg.InputPeerSelf{},
|
|
Media: &tg.InputMediaUploadedPhoto{
|
|
File: file,
|
|
},
|
|
}).ThenResult(expected)
|
|
|
|
r, err := sender.Self().UploadMedia(ctx, UploadedPhoto(file))
|
|
require.NoError(t, err)
|
|
require.Equal(t, expected, r)
|
|
|
|
mock.ExpectCall(&tg.MessagesUploadMediaRequest{
|
|
Peer: &tg.InputPeerSelf{},
|
|
Media: &tg.InputMediaUploadedPhoto{
|
|
File: file,
|
|
},
|
|
}).ThenRPCErr(testRPCError())
|
|
_, err = sender.Self().UploadMedia(ctx, UploadedPhoto(file))
|
|
require.Error(t, err)
|
|
}
|