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
109 lines
2.7 KiB
Go
109 lines
2.7 KiB
Go
package message
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"io/fs"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/telegram/uploader"
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/telegram/uploader/source"
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
|
|
)
|
|
|
|
type mockUploader struct {
|
|
file tg.InputFileClass
|
|
}
|
|
|
|
func (m mockUploader) FromFile(ctx context.Context, f uploader.File, name string) (tg.InputFileClass, error) {
|
|
return m.file, nil
|
|
}
|
|
|
|
func (m mockUploader) FromPath(ctx context.Context, path, name string) (tg.InputFileClass, error) {
|
|
return m.file, nil
|
|
}
|
|
|
|
func (m mockUploader) FromFS(ctx context.Context, filesystem fs.FS, path, name string) (tg.InputFileClass, error) {
|
|
return m.file, nil
|
|
}
|
|
|
|
func (m mockUploader) FromReader(ctx context.Context, name string, f io.Reader) (tg.InputFileClass, error) {
|
|
return m.file, nil
|
|
}
|
|
|
|
func (m mockUploader) FromBytes(ctx context.Context, name string, b []byte) (tg.InputFileClass, error) {
|
|
return m.file, nil
|
|
}
|
|
|
|
func (m mockUploader) FromURL(ctx context.Context, rawURL string) (tg.InputFileClass, error) {
|
|
return m.file, nil
|
|
}
|
|
|
|
func (m mockUploader) FromSource(ctx context.Context, src source.Source, rawURL string) (tg.InputFileClass, error) {
|
|
return m.file, nil
|
|
}
|
|
|
|
func TestUpload(t *testing.T) {
|
|
ctx := context.Background()
|
|
sender, mock := testSender(t)
|
|
|
|
f := &tg.InputFile{
|
|
ID: 1,
|
|
Parts: 1,
|
|
Name: "10.jpg",
|
|
MD5Checksum: "abc",
|
|
}
|
|
upd := mockUploader{file: f}
|
|
dialog := sender.WithUploader(upd).Self()
|
|
|
|
expectSendMedia(t, &tg.InputMediaUploadedPhoto{
|
|
File: f,
|
|
}, mock)
|
|
_, err := dialog.Upload(FromPath("abc.jpg", "")).Photo(ctx)
|
|
require.NoError(t, err)
|
|
|
|
expectSendMedia(t, &tg.InputMediaUploadedDocument{
|
|
File: f,
|
|
ForceFile: true,
|
|
}, mock)
|
|
_, err = dialog.Upload(FromReader("abc.jpg", nil)).File(ctx)
|
|
require.NoError(t, err)
|
|
|
|
expectSendMedia(t, &tg.InputMediaUploadedDocument{
|
|
File: f,
|
|
ForceFile: true,
|
|
}, mock)
|
|
_, err = dialog.Upload(FromFS(nil, "abc.jpg", "")).File(ctx)
|
|
require.NoError(t, err)
|
|
|
|
expectSendMedia(t, &tg.InputMediaUploadedDocument{
|
|
File: f,
|
|
ForceFile: true,
|
|
}, mock)
|
|
_, err = dialog.Upload(FromBytes("abc.jpg", nil)).File(ctx)
|
|
require.NoError(t, err)
|
|
|
|
expectSendMedia(t, &tg.InputMediaUploadedDocument{
|
|
File: f,
|
|
ForceFile: true,
|
|
}, mock)
|
|
_, err = dialog.Upload(FromFile(nil)).File(ctx)
|
|
require.NoError(t, err)
|
|
|
|
expectSendMedia(t, &tg.InputMediaUploadedDocument{
|
|
File: f,
|
|
ForceFile: true,
|
|
}, mock)
|
|
_, err = dialog.Upload(FromURL("http://example.com")).File(ctx)
|
|
require.NoError(t, err)
|
|
|
|
expectSendMedia(t, &tg.InputMediaUploadedDocument{
|
|
File: f,
|
|
ForceFile: true,
|
|
}, mock)
|
|
_, err = dialog.Upload(FromSource(source.NewHTTPSource(), "http://example.com")).File(ctx)
|
|
require.NoError(t, err)
|
|
}
|