Skip to content

Commit

Permalink
fix(downloader): fix after case update
Browse files Browse the repository at this point in the history
  • Loading branch information
ernado committed Mar 10, 2021
1 parent 6df86eb commit b642a14
Show file tree
Hide file tree
Showing 3 changed files with 8 additions and 8 deletions.
4 changes: 2 additions & 2 deletions telegram/downloader/cdn.go
Expand Up @@ -64,7 +64,7 @@ func (c cdn) decrypt(src []byte, offset int) ([]byte, error) {
}

func (c cdn) Chunk(ctx context.Context, offset, limit int) (chunk, error) {
r, err := c.cdn.UploadGetCdnFile(ctx, &tg.UploadGetCDNFileRequest{
r, err := c.cdn.UploadGetCDNFile(ctx, &tg.UploadGetCDNFileRequest{
Offset: offset,
Limit: limit,
FileToken: c.redirect.FileToken,
Expand All @@ -91,7 +91,7 @@ func (c cdn) Chunk(ctx context.Context, offset, limit int) (chunk, error) {
}

func (c cdn) Hashes(ctx context.Context, offset int) ([]tg.FileHash, error) {
return c.client.UploadGetCdnFileHashes(ctx, &tg.UploadGetCDNFileHashesRequest{
return c.client.UploadGetCDNFileHashes(ctx, &tg.UploadGetCDNFileHashesRequest{
FileToken: c.redirect.FileToken,
Offset: offset,
})
Expand Down
6 changes: 3 additions & 3 deletions telegram/downloader/client.go
Expand Up @@ -8,16 +8,16 @@ import (

// CDN represents Telegram RPC client to CDN server.
type CDN interface {
UploadGetCdnFile(ctx context.Context, request *tg.UploadGetCDNFileRequest) (tg.UploadCDNFileClass, error)
UploadGetCDNFile(ctx context.Context, request *tg.UploadGetCDNFileRequest) (tg.UploadCDNFileClass, error)
}

// Client represents Telegram RPC client.
type Client interface {
UploadGetFile(ctx context.Context, request *tg.UploadGetFileRequest) (tg.UploadFileClass, error)
UploadGetFileHashes(ctx context.Context, request *tg.UploadGetFileHashesRequest) ([]tg.FileHash, error)

UploadReuploadCdnFile(ctx context.Context, request *tg.UploadReuploadCDNFileRequest) ([]tg.FileHash, error)
UploadGetCdnFileHashes(ctx context.Context, request *tg.UploadGetCDNFileHashesRequest) ([]tg.FileHash, error)
UploadReuploadCDNFile(ctx context.Context, request *tg.UploadReuploadCDNFileRequest) ([]tg.FileHash, error)
UploadGetCDNFileHashes(ctx context.Context, request *tg.UploadGetCDNFileHashesRequest) ([]tg.FileHash, error)

UploadGetWebFile(ctx context.Context, request *tg.UploadGetWebFileRequest) (*tg.UploadWebFile, error)
}
Expand Down
6 changes: 3 additions & 3 deletions telegram/downloader/downloader_test.go
Expand Up @@ -68,11 +68,11 @@ func (m mock) UploadGetFileHashes(ctx context.Context, request *tg.UploadGetFile
return m.hashes.Hashes(ctx, request.Offset)
}

func (m mock) UploadReuploadCdnFile(ctx context.Context, request *tg.UploadReuploadCDNFileRequest) ([]tg.FileHash, error) {
func (m mock) UploadReuploadCDNFile(ctx context.Context, request *tg.UploadReuploadCDNFileRequest) ([]tg.FileHash, error) {
panic("implement me")
}

func (m mock) UploadGetCdnFile(ctx context.Context, request *tg.UploadGetCDNFileRequest) (tg.UploadCDNFileClass, error) {
func (m mock) UploadGetCDNFile(ctx context.Context, request *tg.UploadGetCDNFileRequest) (tg.UploadCDNFileClass, error) {
if m.err {
return nil, testErr
}
Expand Down Expand Up @@ -100,7 +100,7 @@ func (m mock) UploadGetCdnFile(ctx context.Context, request *tg.UploadGetCDNFile
}, nil
}

func (m mock) UploadGetCdnFileHashes(ctx context.Context, request *tg.UploadGetCDNFileHashesRequest) ([]tg.FileHash, error) {
func (m mock) UploadGetCDNFileHashes(ctx context.Context, request *tg.UploadGetCDNFileHashesRequest) ([]tg.FileHash, error) {
if m.hashesErr {
return nil, testErr
}
Expand Down

0 comments on commit b642a14

Please sign in to comment.