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,123 @@
|
||||
package peers
|
||||
|
||||
import (
|
||||
"context"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/require"
|
||||
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
|
||||
)
|
||||
|
||||
func TestManager_getUserFull(t *testing.T) {
|
||||
a := require.New(t)
|
||||
ctx := context.Background()
|
||||
mock, m := testManager(t)
|
||||
testUserFull := getTestUserFull()
|
||||
input := &tg.InputUser{
|
||||
UserID: testUserFull.ID,
|
||||
AccessHash: 10,
|
||||
}
|
||||
|
||||
mock.ExpectCall(&tg.UsersGetFullUserRequest{ID: input}).ThenResult(&tg.UsersUserFull{
|
||||
FullUser: testUserFull,
|
||||
})
|
||||
|
||||
v, err := m.getUserFull(ctx, input)
|
||||
a.NoError(err)
|
||||
a.Equal(&testUserFull, v)
|
||||
|
||||
// Test caching.
|
||||
v, err = m.getUserFull(ctx, input)
|
||||
a.NoError(err)
|
||||
a.Equal(&testUserFull, v)
|
||||
}
|
||||
|
||||
func TestManager_updateUserFull(t *testing.T) {
|
||||
a := require.New(t)
|
||||
ctx := context.Background()
|
||||
mock, m := testManager(t)
|
||||
testUserFull := getTestUserFull()
|
||||
input := &tg.InputUser{
|
||||
UserID: testUserFull.ID,
|
||||
AccessHash: 10,
|
||||
}
|
||||
|
||||
mock.ExpectCall(&tg.UsersGetFullUserRequest{ID: input}).ThenRPCErr(getTestError())
|
||||
_, err := m.updateUserFull(ctx, input)
|
||||
a.Error(err)
|
||||
}
|
||||
|
||||
func TestManager_getChatFull(t *testing.T) {
|
||||
a := require.New(t)
|
||||
ctx := context.Background()
|
||||
mock, m := testManager(t)
|
||||
testChatFull := getTestChatFull()
|
||||
input := testChatFull.ID
|
||||
|
||||
mock.ExpectCall(&tg.MessagesGetFullChatRequest{ChatID: input}).ThenResult(&tg.MessagesChatFull{
|
||||
FullChat: testChatFull,
|
||||
})
|
||||
|
||||
v, err := m.getChatFull(ctx, input)
|
||||
a.NoError(err)
|
||||
a.Equal(testChatFull, v)
|
||||
|
||||
// Test caching.
|
||||
v, err = m.getChatFull(ctx, input)
|
||||
a.NoError(err)
|
||||
a.Equal(testChatFull, v)
|
||||
}
|
||||
|
||||
func TestManager_updateChatFull(t *testing.T) {
|
||||
a := require.New(t)
|
||||
ctx := context.Background()
|
||||
mock, m := testManager(t)
|
||||
|
||||
testChatFull := getTestChatFull()
|
||||
input := testChatFull.ID
|
||||
|
||||
mock.ExpectCall(&tg.MessagesGetFullChatRequest{ChatID: input}).ThenRPCErr(getTestError())
|
||||
_, err := m.updateChatFull(ctx, input)
|
||||
a.Error(err)
|
||||
}
|
||||
|
||||
func TestManager_getChannelFull(t *testing.T) {
|
||||
a := require.New(t)
|
||||
ctx := context.Background()
|
||||
mock, m := testManager(t)
|
||||
testChannelFull := getTestChannelFull()
|
||||
input := &tg.InputChannel{
|
||||
ChannelID: testChannelFull.ID,
|
||||
AccessHash: 10,
|
||||
}
|
||||
|
||||
mock.ExpectCall(&tg.ChannelsGetFullChannelRequest{Channel: input}).ThenResult(&tg.MessagesChatFull{
|
||||
FullChat: testChannelFull,
|
||||
})
|
||||
|
||||
v, err := m.getChannelFull(ctx, input)
|
||||
a.NoError(err)
|
||||
a.Equal(testChannelFull, v)
|
||||
|
||||
// Test caching.
|
||||
v, err = m.getChannelFull(ctx, input)
|
||||
a.NoError(err)
|
||||
a.Equal(testChannelFull, v)
|
||||
}
|
||||
|
||||
func TestManager_updateChannelFull(t *testing.T) {
|
||||
a := require.New(t)
|
||||
ctx := context.Background()
|
||||
mock, m := testManager(t)
|
||||
|
||||
testChannelFull := getTestChannelFull()
|
||||
input := &tg.InputChannel{
|
||||
ChannelID: testChannelFull.ID,
|
||||
AccessHash: 10,
|
||||
}
|
||||
|
||||
mock.ExpectCall(&tg.ChannelsGetFullChannelRequest{Channel: input}).ThenRPCErr(getTestError())
|
||||
_, err := m.updateChannelFull(ctx, input)
|
||||
a.Error(err)
|
||||
}
|
||||
Reference in New Issue
Block a user