Files
mautrix-telegram/pkg/gotd/telegram/transfer_test.go
T
2025-06-27 20:03:37 -07:00

41 lines
878 B
Go

package telegram
import (
"bytes"
"context"
"testing"
"github.com/stretchr/testify/require"
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
"go.mau.fi/mautrix-telegram/pkg/gotd/tgmock"
)
func TestTransfer(t *testing.T) {
ctx := context.Background()
dc := 1
mockClient(func(a *tgmock.Mock, client *Client) {
user := &tg.User{ID: 10, Username: "abc10"}
auth := &tg.AuthAuthorization{
User: user,
}
exported := bytes.Repeat([]byte{10}, 10)
a.ExpectCall(&tg.AuthExportAuthorizationRequest{
DCID: dc,
}).ThenResult(&tg.AuthExportedAuthorization{
ID: user.ID,
Bytes: exported,
}).ExpectCall(&tg.AuthImportAuthorizationRequest{
ID: user.ID,
Bytes: exported,
}).ThenResult(&tg.AuthAuthorization{
User: user,
})
r, err := client.transfer(ctx, tg.NewClient(client), dc)
require.NoError(t, err)
require.Equal(t, auth, r)
})(t)
}