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
This commit is contained in:
@@ -0,0 +1,167 @@
|
||||
package peers
|
||||
|
||||
import (
|
||||
"context"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/require"
|
||||
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
|
||||
)
|
||||
|
||||
func TestManager_findPeerClass(t *testing.T) {
|
||||
user := getTestSelf()
|
||||
chat := &tg.Chat{
|
||||
ID: 10,
|
||||
}
|
||||
channel := &tg.Channel{
|
||||
ID: 10,
|
||||
}
|
||||
|
||||
m := &Manager{}
|
||||
type args struct {
|
||||
p tg.PeerClass
|
||||
users []tg.UserClass
|
||||
chats []tg.ChatClass
|
||||
}
|
||||
tests := []struct {
|
||||
name string
|
||||
args args
|
||||
want Peer
|
||||
wantOk bool
|
||||
}{
|
||||
{
|
||||
name: "User",
|
||||
args: args{
|
||||
p: &tg.PeerUser{UserID: user.GetID()},
|
||||
users: []tg.UserClass{user},
|
||||
},
|
||||
want: User{
|
||||
raw: user,
|
||||
m: m,
|
||||
},
|
||||
wantOk: true,
|
||||
},
|
||||
{
|
||||
name: "Chat",
|
||||
args: args{
|
||||
p: &tg.PeerChat{ChatID: 10},
|
||||
chats: []tg.ChatClass{chat},
|
||||
},
|
||||
want: Chat{
|
||||
raw: chat,
|
||||
m: m,
|
||||
},
|
||||
wantOk: true,
|
||||
},
|
||||
{
|
||||
name: "Channel",
|
||||
args: args{
|
||||
p: &tg.PeerChannel{ChannelID: 10},
|
||||
chats: []tg.ChatClass{channel},
|
||||
},
|
||||
want: Channel{
|
||||
raw: channel,
|
||||
m: m,
|
||||
},
|
||||
wantOk: true,
|
||||
},
|
||||
{name: "NilPeer"},
|
||||
}
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
a := require.New(t)
|
||||
r, ok := m.findPeerClass(tt.args.p, tt.args.users, tt.args.chats)
|
||||
if tt.wantOk {
|
||||
a.Equal(tt.want, r)
|
||||
a.True(ok)
|
||||
} else {
|
||||
a.False(ok)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestManager_Resolve(t *testing.T) {
|
||||
testUser := getTestSelf()
|
||||
inputs := []struct {
|
||||
Name string
|
||||
Input string
|
||||
}{
|
||||
{"Domain", "@gotduser"},
|
||||
{"Deeplink", "https://t.me/gotduser"},
|
||||
}
|
||||
for _, tt := range inputs {
|
||||
t.Run(tt.Name, func(t *testing.T) {
|
||||
a := require.New(t)
|
||||
mock, m := testManager(t)
|
||||
|
||||
username := "gotduser"
|
||||
mock.ExpectCall(&tg.ContactsResolveUsernameRequest{
|
||||
Username: username,
|
||||
}).ThenResult(&tg.ContactsResolvedPeer{
|
||||
Peer: &tg.PeerUser{UserID: testUser.GetID()},
|
||||
Users: []tg.UserClass{
|
||||
&tg.User{ID: testUser.GetID(), AccessHash: 10, Username: username},
|
||||
},
|
||||
}).ExpectCall(&tg.ContactsResolveUsernameRequest{
|
||||
Username: username,
|
||||
}).ThenRPCErr(getTestError())
|
||||
|
||||
ctx := context.Background()
|
||||
|
||||
r, err := m.Resolve(ctx, tt.Input)
|
||||
a.NoError(err)
|
||||
a.IsType(&tg.InputPeerUser{}, r.InputPeer())
|
||||
a.Equal(testUser.GetID(), r.ID())
|
||||
|
||||
_, err = m.Resolve(ctx, tt.Input)
|
||||
a.Error(err)
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestManager_ResolvePhone(t *testing.T) {
|
||||
a := require.New(t)
|
||||
mock, m := testManager(t)
|
||||
// To count contacts hash.
|
||||
m.me.Store(&tg.User{
|
||||
ID: 1,
|
||||
})
|
||||
|
||||
phone := "+79001234567"
|
||||
phone2 := "+79011234567"
|
||||
mock.ExpectCall(&tg.ContactsGetContactsRequest{
|
||||
Hash: 0,
|
||||
}).ThenRPCErr(getTestError())
|
||||
ctx := context.Background()
|
||||
|
||||
_, err := m.Resolve(ctx, phone)
|
||||
a.Error(err)
|
||||
|
||||
resp := &tg.ContactsContacts{
|
||||
Contacts: []tg.Contact{{
|
||||
UserID: 10,
|
||||
Mutual: false,
|
||||
}},
|
||||
SavedCount: 1,
|
||||
Users: []tg.UserClass{
|
||||
&tg.User{ID: 10, AccessHash: 10, Username: "rustmustdie", Phone: cleanupPhone(phone)},
|
||||
},
|
||||
}
|
||||
mock.ExpectCall(&tg.ContactsGetContactsRequest{
|
||||
Hash: 0,
|
||||
}).ThenResult(resp)
|
||||
|
||||
r, err := m.Resolve(ctx, phone)
|
||||
a.NoError(err)
|
||||
a.IsType(&tg.InputPeerUser{}, r.InputPeer())
|
||||
a.Equal(int64(10), r.ID())
|
||||
|
||||
mock.ExpectCall(&tg.ContactsGetContactsRequest{
|
||||
Hash: contactsHash(1, resp),
|
||||
}).ThenResult(&tg.ContactsContactsNotModified{})
|
||||
|
||||
_, err = m.Resolve(ctx, phone2)
|
||||
a.Error(err)
|
||||
}
|
||||
Reference in New Issue
Block a user