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,87 @@
|
||||
package fileid_test
|
||||
|
||||
import (
|
||||
"context"
|
||||
"os"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/go-faster/errors"
|
||||
"go.uber.org/zap"
|
||||
"go.uber.org/zap/zaptest"
|
||||
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/fileid"
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/telegram"
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/telegram/downloader"
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/testutil"
|
||||
)
|
||||
|
||||
func runBot(ctx context.Context, token, fileID string, logger *zap.Logger) error {
|
||||
bot := telegram.NewClient(telegram.TestAppID, telegram.TestAppHash, telegram.Options{
|
||||
Logger: logger,
|
||||
})
|
||||
d := downloader.NewDownloader()
|
||||
|
||||
return bot.Run(ctx, func(ctx context.Context) error {
|
||||
auth, err := bot.Auth().Bot(ctx, token)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "auth bot")
|
||||
}
|
||||
user, ok := auth.User.AsNotEmpty()
|
||||
if !ok {
|
||||
return errors.Errorf("unexpected type %T", auth.User)
|
||||
}
|
||||
_ = user
|
||||
|
||||
decoded, err := fileid.DecodeFileID(fileID)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "decode FileID")
|
||||
}
|
||||
|
||||
loc, ok := decoded.AsInputFileLocation()
|
||||
if !ok {
|
||||
return errors.Errorf("can't map %q", fileID)
|
||||
}
|
||||
|
||||
filename := "file.dat"
|
||||
switch decoded.Type {
|
||||
case fileid.Thumbnail, fileid.ProfilePhoto, fileid.Photo:
|
||||
filename = "file.jpg"
|
||||
case fileid.Video,
|
||||
fileid.Animation,
|
||||
fileid.VideoNote:
|
||||
filename = "file.mp4"
|
||||
case fileid.Audio:
|
||||
filename = "file.mp3"
|
||||
case fileid.Voice:
|
||||
filename = "file.ogg"
|
||||
case fileid.Sticker:
|
||||
filename = "file.png"
|
||||
}
|
||||
|
||||
if _, err := d.Download(bot.API(), loc).ToPath(ctx, filename); err != nil {
|
||||
return errors.Wrap(err, "download")
|
||||
}
|
||||
return nil
|
||||
})
|
||||
}
|
||||
|
||||
func TestExternalE2ECheckFileID(t *testing.T) {
|
||||
testutil.SkipExternal(t)
|
||||
token := os.Getenv("GOTD_E2E_BOT_TOKEN")
|
||||
if token == "" {
|
||||
t.Skip("Set GOTD_E2E_BOT_TOKEN env to run test.")
|
||||
}
|
||||
fileID := os.Getenv("GOTD_E2E_FILE_ID")
|
||||
if fileID == "" {
|
||||
t.Skip("Set GOTD_E2E_FILE_ID env to run test.")
|
||||
}
|
||||
logger := zaptest.NewLogger(t)
|
||||
|
||||
ctx, cancel := context.WithTimeout(context.Background(), time.Minute)
|
||||
defer cancel()
|
||||
|
||||
if err := runBot(ctx, token, fileID, logger.Named("bot")); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user