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,128 @@
|
||||
package peer
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/require"
|
||||
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
|
||||
)
|
||||
|
||||
type badPeer struct {
|
||||
tg.PeerClass
|
||||
}
|
||||
|
||||
type mockResult struct {
|
||||
Entities
|
||||
}
|
||||
|
||||
func (m mockResult) MapUsers() (r tg.UserClassArray) {
|
||||
for _, e := range m.Entities.Users() {
|
||||
r = append(r, e)
|
||||
}
|
||||
|
||||
return r
|
||||
}
|
||||
|
||||
func (m mockResult) MapChats() (r tg.ChatClassArray) {
|
||||
for _, e := range m.Entities.Chats() {
|
||||
r = append(r, e)
|
||||
}
|
||||
|
||||
for _, e := range m.Entities.Channels() {
|
||||
r = append(r, e)
|
||||
}
|
||||
|
||||
return r
|
||||
}
|
||||
|
||||
func TestEntities(t *testing.T) {
|
||||
users := map[int64]*tg.User{
|
||||
10: {ID: 10, AccessHash: 10},
|
||||
}
|
||||
chats := map[int64]*tg.Chat{
|
||||
10: {ID: 10},
|
||||
}
|
||||
channels := map[int64]*tg.Channel{
|
||||
10: {ID: 10, AccessHash: 10},
|
||||
}
|
||||
ent := NewEntities(users, chats, channels)
|
||||
ctx := tg.Entities{
|
||||
Users: users,
|
||||
Chats: chats,
|
||||
Channels: channels,
|
||||
}
|
||||
result := mockResult{Entities: ent}
|
||||
|
||||
tests := []struct {
|
||||
name string
|
||||
filler func() Entities
|
||||
}{
|
||||
{"NewEntities", func() Entities {
|
||||
return ent
|
||||
}},
|
||||
{"EntitiesFromResult", func() Entities {
|
||||
return EntitiesFromResult(result)
|
||||
}},
|
||||
{"FillFromResult", func() Entities {
|
||||
e := NewEntities(
|
||||
map[int64]*tg.User{},
|
||||
map[int64]*tg.Chat{},
|
||||
map[int64]*tg.Channel{},
|
||||
)
|
||||
e.FillFromResult(result)
|
||||
return e
|
||||
}},
|
||||
{"EntitiesFromUpdate", func() Entities {
|
||||
return EntitiesFromUpdate(ctx)
|
||||
}},
|
||||
{"FillFromUpdate", func() Entities {
|
||||
e := NewEntities(
|
||||
map[int64]*tg.User{},
|
||||
map[int64]*tg.Chat{},
|
||||
map[int64]*tg.Channel{},
|
||||
)
|
||||
e.FillFromUpdate(ctx)
|
||||
return e
|
||||
}},
|
||||
}
|
||||
|
||||
for _, test := range tests {
|
||||
t.Run(test.name, func(t *testing.T) {
|
||||
a := require.New(t)
|
||||
e := test.filler()
|
||||
|
||||
_, err := e.ExtractPeer(badPeer{})
|
||||
a.Error(err)
|
||||
|
||||
user, err := e.ExtractPeer(&tg.PeerUser{UserID: 10})
|
||||
peerUser := &tg.InputPeerUser{
|
||||
UserID: 10,
|
||||
AccessHash: 10,
|
||||
}
|
||||
a.Equal(peerUser, user)
|
||||
a.NoError(err)
|
||||
_, err = e.ExtractPeer(&tg.PeerUser{UserID: 11})
|
||||
a.Error(err)
|
||||
|
||||
chat, err := e.ExtractPeer(&tg.PeerChat{ChatID: 10})
|
||||
peerChat := &tg.InputPeerChat{
|
||||
ChatID: 10,
|
||||
}
|
||||
a.Equal(peerChat, chat)
|
||||
a.NoError(err)
|
||||
_, err = e.ExtractPeer(&tg.PeerChat{ChatID: 11})
|
||||
a.Error(err)
|
||||
|
||||
channel, err := e.ExtractPeer(&tg.PeerChannel{ChannelID: 10})
|
||||
peerChannel := &tg.InputPeerChannel{
|
||||
ChannelID: 10,
|
||||
AccessHash: 10,
|
||||
}
|
||||
a.Equal(peerChannel, channel)
|
||||
a.NoError(err)
|
||||
_, err = e.ExtractPeer(&tg.PeerChannel{ChannelID: 11})
|
||||
a.Error(err)
|
||||
})
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user