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
79 lines
1.8 KiB
Go
79 lines
1.8 KiB
Go
package peer
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/tgerr"
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/tgmock"
|
|
)
|
|
|
|
func Test_plainResolver_Resolve(t *testing.T) {
|
|
mock := tgmock.NewRequire(t)
|
|
raw := tg.NewClient(mock)
|
|
|
|
domain := "adcd"
|
|
mock.ExpectCall(&tg.ContactsResolveUsernameRequest{
|
|
Username: domain,
|
|
}).ThenResult(&tg.ContactsResolvedPeer{
|
|
Peer: &tg.PeerUser{UserID: 10},
|
|
Users: []tg.UserClass{
|
|
&tg.User{ID: 10, AccessHash: 10, Username: domain},
|
|
},
|
|
}).ExpectCall(&tg.ContactsResolveUsernameRequest{
|
|
Username: domain,
|
|
}).ThenRPCErr(&tgerr.Error{
|
|
Code: 1337,
|
|
Message: "TEST_ERROR",
|
|
Type: "TEST_ERROR",
|
|
})
|
|
|
|
ctx := context.Background()
|
|
resolver := plainResolver{raw: raw}
|
|
|
|
r, err := resolver.ResolveDomain(ctx, domain)
|
|
require.IsType(t, &tg.InputPeerUser{}, r)
|
|
require.Equal(t, int64(10), r.(*tg.InputPeerUser).UserID)
|
|
require.NoError(t, err)
|
|
|
|
_, err = resolver.ResolveDomain(ctx, domain)
|
|
require.Error(t, err)
|
|
}
|
|
|
|
func Test_plainResolver_ResolvePhone(t *testing.T) {
|
|
mock := tgmock.New(t)
|
|
raw := tg.NewClient(mock)
|
|
|
|
phone := "adcd"
|
|
mock.ExpectCall(&tg.ContactsGetContactsRequest{
|
|
Hash: 0,
|
|
}).ThenResult(&tg.ContactsContacts{
|
|
Contacts: []tg.Contact{{
|
|
UserID: 10,
|
|
Mutual: false,
|
|
}},
|
|
SavedCount: 1,
|
|
Users: []tg.UserClass{
|
|
&tg.User{ID: 10, AccessHash: 10, Username: "rustmustdie", Phone: phone},
|
|
},
|
|
}).Expect().ThenRPCErr(&tgerr.Error{
|
|
Code: 1337,
|
|
Message: "TEST_ERROR",
|
|
Type: "TEST_ERROR",
|
|
})
|
|
|
|
ctx := context.Background()
|
|
resolver := plainResolver{raw: raw}
|
|
|
|
r, err := resolver.ResolvePhone(ctx, phone)
|
|
require.NoError(t, err)
|
|
require.IsType(t, &tg.InputPeerUser{}, r)
|
|
require.Equal(t, int64(10), r.(*tg.InputPeerUser).UserID)
|
|
|
|
_, err = resolver.ResolvePhone(ctx, phone)
|
|
require.Error(t, err)
|
|
}
|