HFO4
5 years ago
parent
b1490a665c
commit
e722c33cd5
@ -1,190 +0,0 @@
|
|||||||
package filesystem
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
"github.com/DATA-DOG/go-sqlmock"
|
|
||||||
model "github.com/HFO4/cloudreve/models"
|
|
||||||
"github.com/HFO4/cloudreve/pkg/cache"
|
|
||||||
"github.com/HFO4/cloudreve/pkg/conf"
|
|
||||||
"github.com/HFO4/cloudreve/pkg/filesystem/response"
|
|
||||||
"github.com/HFO4/cloudreve/pkg/util"
|
|
||||||
"github.com/jinzhu/gorm"
|
|
||||||
"github.com/stretchr/testify/assert"
|
|
||||||
testMock "github.com/stretchr/testify/mock"
|
|
||||||
"image"
|
|
||||||
"image/jpeg"
|
|
||||||
"os"
|
|
||||||
"testing"
|
|
||||||
)
|
|
||||||
|
|
||||||
func CreateTestImage() *os.File {
|
|
||||||
file, err := os.Create("TestFileSystem_GenerateThumbnail.jpeg")
|
|
||||||
alpha := image.NewAlpha(image.Rect(0, 0, 500, 200))
|
|
||||||
jpeg.Encode(file, alpha, nil)
|
|
||||||
if err != nil {
|
|
||||||
fmt.Println(err)
|
|
||||||
}
|
|
||||||
_, _ = file.Seek(0, 0)
|
|
||||||
return file
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestFileSystem_GetThumb(t *testing.T) {
|
|
||||||
asserts := assert.New(t)
|
|
||||||
ctx := context.Background()
|
|
||||||
cache.Set("setting_preview_timeout", "60", 0)
|
|
||||||
|
|
||||||
// 正常
|
|
||||||
{
|
|
||||||
fs := FileSystem{
|
|
||||||
User: &model.User{
|
|
||||||
Model: gorm.Model{ID: 1},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
testHandler := new(FileHeaderMock)
|
|
||||||
testHandler.On("Thumb", testMock.Anything, "123.jpg").Return(&response.ContentResponse{URL: "123"}, nil)
|
|
||||||
fs.Handler = testHandler
|
|
||||||
mock.ExpectQuery("SELECT(.+)").
|
|
||||||
WithArgs(10, 1).
|
|
||||||
WillReturnRows(
|
|
||||||
sqlmock.NewRows(
|
|
||||||
[]string{"id", "pic_info", "source_name", "policy_id"}).
|
|
||||||
AddRow(10, "10,10", "123.jpg", 154),
|
|
||||||
)
|
|
||||||
mock.ExpectQuery("SELECT(.+)").
|
|
||||||
WillReturnRows(
|
|
||||||
sqlmock.NewRows(
|
|
||||||
[]string{"id", "type"}).
|
|
||||||
AddRow(154, "mock"),
|
|
||||||
)
|
|
||||||
|
|
||||||
res, err := fs.GetThumb(ctx, 10)
|
|
||||||
asserts.NoError(mock.ExpectationsWereMet())
|
|
||||||
testHandler.AssertExpectations(t)
|
|
||||||
asserts.NoError(err)
|
|
||||||
asserts.Equal("123", res.URL)
|
|
||||||
}
|
|
||||||
|
|
||||||
// 文件不存在
|
|
||||||
{
|
|
||||||
fs := FileSystem{
|
|
||||||
User: &model.User{
|
|
||||||
Model: gorm.Model{ID: 1},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
mock.ExpectQuery("SELECT(.+)").
|
|
||||||
WithArgs(10, 1).
|
|
||||||
WillReturnRows(
|
|
||||||
sqlmock.NewRows(
|
|
||||||
[]string{"id", "pic_info", "source_name"}),
|
|
||||||
)
|
|
||||||
|
|
||||||
_, err := fs.GetThumb(ctx, 10)
|
|
||||||
asserts.NoError(mock.ExpectationsWereMet())
|
|
||||||
asserts.Error(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestFileSystem_GenerateThumbnail(t *testing.T) {
|
|
||||||
asserts := assert.New(t)
|
|
||||||
fs := FileSystem{
|
|
||||||
User: &model.User{
|
|
||||||
Model: gorm.Model{ID: 1},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
ctx := context.Background()
|
|
||||||
|
|
||||||
// 成功
|
|
||||||
{
|
|
||||||
src := CreateTestImage()
|
|
||||||
testHandler := new(FileHeaderMock)
|
|
||||||
testHandler.On("Get", testMock.Anything, "TestFileSystem_GenerateThumbnail.jpeg").Return(src, nil)
|
|
||||||
fs.Handler = testHandler
|
|
||||||
|
|
||||||
mock.ExpectBegin()
|
|
||||||
mock.ExpectExec("UPDATE(.+)").WillReturnResult(sqlmock.NewResult(1, 1))
|
|
||||||
mock.ExpectCommit()
|
|
||||||
|
|
||||||
file := &model.File{
|
|
||||||
Model: gorm.Model{ID: 1},
|
|
||||||
Name: "123.jpg",
|
|
||||||
SourceName: "TestFileSystem_GenerateThumbnail.jpeg",
|
|
||||||
}
|
|
||||||
|
|
||||||
fs.GenerateThumbnail(ctx, file)
|
|
||||||
asserts.NoError(mock.ExpectationsWereMet())
|
|
||||||
testHandler.AssertExpectations(t)
|
|
||||||
asserts.True(util.Exists("TestFileSystem_GenerateThumbnail.jpeg" + conf.ThumbConfig.FileSuffix))
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
// 成功,不进行数据库更新
|
|
||||||
{
|
|
||||||
src := CreateTestImage()
|
|
||||||
testHandler := new(FileHeaderMock)
|
|
||||||
testHandler.On("Get", testMock.Anything, "TestFileSystem_GenerateThumbnail.jpeg").Return(src, nil)
|
|
||||||
fs.Handler = testHandler
|
|
||||||
|
|
||||||
file := &model.File{
|
|
||||||
Name: "123.jpg",
|
|
||||||
SourceName: "TestFileSystem_GenerateThumbnail.jpeg",
|
|
||||||
}
|
|
||||||
|
|
||||||
fs.GenerateThumbnail(ctx, file)
|
|
||||||
asserts.NoError(mock.ExpectationsWereMet())
|
|
||||||
testHandler.AssertExpectations(t)
|
|
||||||
asserts.True(util.Exists("TestFileSystem_GenerateThumbnail.jpeg" + conf.ThumbConfig.FileSuffix))
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
// 更新信息失败后删除文件
|
|
||||||
{
|
|
||||||
src := CreateTestImage()
|
|
||||||
testHandler := new(FileHeaderMock)
|
|
||||||
testHandler.On("Get", testMock.Anything, "TestFileSystem_GenerateThumbnail.jpeg").Return(src, nil)
|
|
||||||
testHandler.On("Delete", testMock.Anything, testMock.Anything).Return([]string{}, nil)
|
|
||||||
fs.Handler = testHandler
|
|
||||||
|
|
||||||
mock.ExpectBegin()
|
|
||||||
mock.ExpectExec("UPDATE(.+)").WillReturnError(errors.New("error"))
|
|
||||||
mock.ExpectRollback()
|
|
||||||
|
|
||||||
file := &model.File{
|
|
||||||
Model: gorm.Model{ID: 1},
|
|
||||||
Name: "123.jpg",
|
|
||||||
SourceName: "TestFileSystem_GenerateThumbnail.jpeg",
|
|
||||||
}
|
|
||||||
|
|
||||||
fs.GenerateThumbnail(ctx, file)
|
|
||||||
asserts.NoError(mock.ExpectationsWereMet())
|
|
||||||
testHandler.AssertExpectations(t)
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
// 不能生成缩略图
|
|
||||||
{
|
|
||||||
file := &model.File{
|
|
||||||
Model: gorm.Model{ID: 1},
|
|
||||||
Name: "123.123",
|
|
||||||
SourceName: "TestFileSystem_GenerateThumbnail.jpeg",
|
|
||||||
}
|
|
||||||
|
|
||||||
fs.GenerateThumbnail(ctx, file)
|
|
||||||
asserts.NoError(mock.ExpectationsWereMet())
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestFileSystem_GenerateThumbnailSize(t *testing.T) {
|
|
||||||
asserts := assert.New(t)
|
|
||||||
fs := FileSystem{
|
|
||||||
User: &model.User{
|
|
||||||
Model: gorm.Model{ID: 1},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
asserts.NotPanics(func() {
|
|
||||||
_, _ = fs.GenerateThumbnailSize(0, 0)
|
|
||||||
})
|
|
||||||
|
|
||||||
}
|
|
||||||
@ -0,0 +1,8 @@
|
|||||||
|
package task
|
||||||
|
|
||||||
|
import "errors"
|
||||||
|
|
||||||
|
var (
|
||||||
|
// ErrUnknownTaskType 未知任务类型
|
||||||
|
ErrUnknownTaskType = errors.New("未知任务类型")
|
||||||
|
)
|
||||||
Loading…
Reference in new issue