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
113 lines
2.4 KiB
Go
113 lines
2.4 KiB
Go
package message
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
|
|
)
|
|
|
|
func TestSender_JoinLink(t *testing.T) {
|
|
formats := []struct {
|
|
fmt string
|
|
wantErr bool
|
|
}{
|
|
{`t.me/+%s`, false},
|
|
{`t.me/+%s/`, false},
|
|
{`https://t.me/+%s`, false},
|
|
{`https://t.me/+%s/`, false},
|
|
{`t.me/joinchat/%s`, false},
|
|
{`t.me/joinchat/%s/`, false},
|
|
{`https://t.me/joinchat/%s`, false},
|
|
{`https://t.me/joinchat/%s/`, false},
|
|
{`tg:join?invite=%s`, false},
|
|
{`tg://join?invite=%s`, false},
|
|
}
|
|
inputs := []struct {
|
|
value string
|
|
wantErr bool
|
|
}{
|
|
{"AAAAAAAAAAAAAAAAAA", false},
|
|
{"", true},
|
|
}
|
|
|
|
for _, format := range formats {
|
|
for _, input := range inputs {
|
|
link := fmt.Sprintf(format.fmt, input.value)
|
|
t.Run(link, func(t *testing.T) {
|
|
ctx := context.Background()
|
|
sender, mock := testSender(t)
|
|
|
|
wantErr := format.wantErr || input.wantErr
|
|
if !wantErr {
|
|
mock.ExpectCall(&tg.MessagesImportChatInviteRequest{
|
|
Hash: input.value,
|
|
}).ThenResult(&tg.Updates{})
|
|
}
|
|
|
|
_, err := sender.JoinLink(ctx, link)
|
|
if wantErr {
|
|
require.Error(t, err)
|
|
} else {
|
|
require.NoError(t, err)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestRequestBuilder_Join(t *testing.T) {
|
|
ctx := context.Background()
|
|
sender, mock := testSender(t)
|
|
peer := &tg.InputPeerChannel{
|
|
ChannelID: 10,
|
|
AccessHash: 10,
|
|
}
|
|
|
|
mock.ExpectCall(&tg.ChannelsJoinChannelRequest{
|
|
Channel: &tg.InputChannel{
|
|
ChannelID: 10,
|
|
AccessHash: 10,
|
|
},
|
|
}).ThenResult(&tg.Updates{})
|
|
_, err := sender.To(peer).Join(ctx)
|
|
require.NoError(t, err)
|
|
|
|
mock.ExpectCall(&tg.ChannelsJoinChannelRequest{
|
|
Channel: &tg.InputChannel{
|
|
ChannelID: 10,
|
|
AccessHash: 10,
|
|
},
|
|
}).ThenRPCErr(testRPCError())
|
|
_, err = sender.To(peer).Join(ctx)
|
|
require.Error(t, err)
|
|
}
|
|
|
|
func TestRequestBuilder_Leave(t *testing.T) {
|
|
ctx := context.Background()
|
|
sender, mock := testSender(t)
|
|
peer := &tg.InputPeerChannel{
|
|
ChannelID: 10,
|
|
AccessHash: 10,
|
|
}
|
|
ch := &tg.InputChannel{
|
|
ChannelID: peer.ChannelID,
|
|
AccessHash: peer.AccessHash,
|
|
}
|
|
|
|
mock.ExpectCall(&tg.ChannelsLeaveChannelRequest{
|
|
Channel: ch,
|
|
}).ThenResult(&tg.Updates{})
|
|
_, err := sender.To(peer).Leave(ctx)
|
|
require.NoError(t, err)
|
|
|
|
mock.ExpectCall(&tg.ChannelsLeaveChannelRequest{
|
|
Channel: ch,
|
|
}).ThenRPCErr(testRPCError())
|
|
_, err = sender.To(peer).Leave(ctx)
|
|
require.Error(t, err)
|
|
}
|