You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
71 lines
1.9 KiB
71 lines
1.9 KiB
package filesystem
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
model "github.com/cloudreve/Cloudreve/v3/models"
|
|
"github.com/cloudreve/Cloudreve/v3/pkg/cache"
|
|
"github.com/cloudreve/Cloudreve/v3/pkg/filesystem/response"
|
|
"github.com/cloudreve/Cloudreve/v3/pkg/request"
|
|
testMock "github.com/stretchr/testify/mock"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestFileSystem_GetThumb(t *testing.T) {
|
|
asserts := assert.New(t)
|
|
fs := &FileSystem{User: &model.User{}}
|
|
|
|
// 非图像文件
|
|
{
|
|
fs.SetTargetFile(&[]model.File{{}})
|
|
_, err := fs.GetThumb(context.Background(), 1)
|
|
asserts.Equal(err, ErrObjectNotExist)
|
|
}
|
|
|
|
// 成功
|
|
{
|
|
cache.Set("setting_thumb_width", "10", 0)
|
|
cache.Set("setting_thumb_height", "10", 0)
|
|
cache.Set("setting_preview_timeout", "50", 0)
|
|
testHandller2 := new(FileHeaderMock)
|
|
testHandller2.On("Thumb", testMock.Anything, "").Return(&response.ContentResponse{}, nil)
|
|
fs.CleanTargets()
|
|
fs.SetTargetFile(&[]model.File{{PicInfo: "1,1", Policy: model.Policy{Type: "mock"}}})
|
|
fs.FileTarget[0].Policy.ID = 1
|
|
fs.Handler = testHandller2
|
|
res, err := fs.GetThumb(context.Background(), 1)
|
|
asserts.NoError(err)
|
|
asserts.EqualValues(50, res.MaxAge)
|
|
}
|
|
}
|
|
|
|
func TestFileSystem_ThumbWorker(t *testing.T) {
|
|
asserts := assert.New(t)
|
|
|
|
asserts.NotPanics(func() {
|
|
getThumbWorker().addWorker()
|
|
getThumbWorker().releaseWorker()
|
|
})
|
|
}
|
|
|
|
func TestFileSystem_GenerateThumbnail(t *testing.T) {
|
|
fs := &FileSystem{User: &model.User{}}
|
|
|
|
// 无法生成缩略图
|
|
{
|
|
fs.SetTargetFile(&[]model.File{{}})
|
|
fs.GenerateThumbnail(context.Background(), &model.File{})
|
|
}
|
|
|
|
// 无法获取文件数据
|
|
{
|
|
testHandller := new(FileHeaderMock)
|
|
testHandller.On("Get", testMock.Anything, "").Return(request.NopRSCloser{}, errors.New("error"))
|
|
fs.Handler = testHandller
|
|
fs.GenerateThumbnail(context.Background(), &model.File{Name: "test.png"})
|
|
testHandller.AssertExpectations(t)
|
|
}
|
|
}
|