Remove: mutipart file upload

pkg/filesystem/upload_test.go

@@ -0,0 +1,80 @@
+package filesystem
+
+import (
+	"context"
+	"errors"
+	model "github.com/HFO4/cloudreve/models"
+	"github.com/HFO4/cloudreve/pkg/filesystem/local"
+	"github.com/gin-gonic/gin"
+	"github.com/jinzhu/gorm"
+	"github.com/stretchr/testify/assert"
+	testMock "github.com/stretchr/testify/mock"
+	"io"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+)
+
+type FileHeaderMock struct {
+	testMock.Mock
+}
+
+func (m FileHeaderMock) Put(ctx context.Context, file io.ReadCloser, dst string) error {
+	args := m.Called(ctx, file, dst)
+	return args.Error(0)
+}
+
+func (m FileHeaderMock) Delete(ctx context.Context, files []string) ([]string, error) {
+	args := m.Called(ctx, files)
+	return args.Get(0).([]string), args.Error(1)
+}
+
+func TestFileSystem_Upload(t *testing.T) {
+	asserts := assert.New(t)
+
+	// 正常
+	testHandller := new(FileHeaderMock)
+	testHandller.On("Put", testMock.Anything, testMock.Anything, testMock.Anything).Return(nil)
+	fs := &FileSystem{
+		Handler: testHandller,
+		User: &model.User{
+			Model: gorm.Model{
+				ID: 1,
+			},
+			Policy: model.Policy{
+				AutoRename:  false,
+				DirNameRule: "{path}",
+			},
+		},
+	}
+	ctx, cancel := context.WithCancel(context.Background())
+	c, _ := gin.CreateTestContext(httptest.NewRecorder())
+	c.Request, _ = http.NewRequest("POST", "/", nil)
+	ctx = context.WithValue(ctx, GinCtx, c)
+	cancel()
+	file := local.FileStream{
+		Size:        5,
+		VirtualPath: "/",
+		Name:        "1.txt",
+	}
+	err := fs.Upload(ctx, file)
+	asserts.NoError(err)
+
+	// BeforeUpload 返回错误
+	fs.BeforeUpload = func(ctx context.Context, fs *FileSystem) error {
+		return errors.New("error")
+	}
+	err = fs.Upload(ctx, file)
+	asserts.Error(err)
+	fs.BeforeUpload = nil
+	testHandller.AssertExpectations(t)
+
+	// 上传文件失败
+	testHandller2 := new(FileHeaderMock)
+	testHandller2.On("Put", testMock.Anything, testMock.Anything, testMock.Anything).Return(errors.New("error"))
+	fs.Handler = testHandller2
+	err = fs.Upload(ctx, file)
+	asserts.Error(err)
+	testHandller2.AssertExpectations(t)
+
+}