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
66 lines
1.8 KiB
Go
66 lines
1.8 KiB
Go
package message
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
|
|
)
|
|
|
|
func TestRequestBuilder_Delete(t *testing.T) {
|
|
ctx := context.Background()
|
|
sender, mock := testSender(t)
|
|
|
|
mock.ExpectCall(&tg.MessagesDeleteMessagesRequest{
|
|
ID: []int{1, 2, 3},
|
|
}).ThenResult(&tg.MessagesAffectedMessages{})
|
|
_, err := sender.Delete().Messages(ctx, 1, 2, 3)
|
|
require.NoError(t, err)
|
|
|
|
mock.ExpectCall(&tg.MessagesDeleteMessagesRequest{
|
|
ID: []int{1, 2, 3},
|
|
}).ThenRPCErr(testRPCError())
|
|
_, err = sender.Delete().Messages(ctx, 1, 2, 3)
|
|
require.Error(t, err)
|
|
}
|
|
|
|
func TestRequestBuilder_Revoke(t *testing.T) {
|
|
ctx := context.Background()
|
|
sender, mock := testSender(t)
|
|
|
|
mock.ExpectCall(&tg.MessagesDeleteMessagesRequest{
|
|
Revoke: true,
|
|
ID: []int{1, 2, 3},
|
|
}).ThenResult(&tg.MessagesAffectedMessages{})
|
|
_, err := sender.To(&tg.InputPeerChat{ChatID: 10}).Revoke().Messages(ctx, 1, 2, 3)
|
|
require.NoError(t, err)
|
|
|
|
mock.ExpectCall(&tg.MessagesDeleteMessagesRequest{
|
|
Revoke: true,
|
|
ID: []int{1, 2, 3},
|
|
}).ThenRPCErr(testRPCError())
|
|
_, err = sender.To(&tg.InputPeerChat{ChatID: 10}).Revoke().Messages(ctx, 1, 2, 3)
|
|
require.Error(t, err)
|
|
|
|
ch := &tg.InputPeerChannel{ChannelID: 10, AccessHash: 10}
|
|
inputCh := &tg.InputChannel{
|
|
ChannelID: ch.ChannelID,
|
|
AccessHash: ch.AccessHash,
|
|
}
|
|
mock.ExpectCall(&tg.ChannelsDeleteMessagesRequest{
|
|
Channel: inputCh,
|
|
ID: []int{1, 2, 3},
|
|
}).ThenResult(&tg.MessagesAffectedMessages{})
|
|
_, err = sender.To(ch).Revoke().Messages(ctx, 1, 2, 3)
|
|
require.NoError(t, err)
|
|
|
|
mock.ExpectCall(&tg.ChannelsDeleteMessagesRequest{
|
|
Channel: inputCh,
|
|
ID: []int{1, 2, 3},
|
|
}).ThenRPCErr(testRPCError())
|
|
_, err = sender.To(ch).Revoke().Messages(ctx, 1, 2, 3)
|
|
require.Error(t, err)
|
|
}
|