From f9b37a3359b4acf5da568c176a21eaa9215296c8 Mon Sep 17 00:00:00 2001
From: HFO4 <912394456@qq.com>
Date: Tue, 19 Nov 2019 18:10:18 +0800
Subject: [PATCH] Fix: test failed due to missing context

---
 pkg/filesystem/context.go    |  4 ++--
 pkg/filesystem/filesystem.go | 19 +++++++++++++++----
 pkg/filesystem/hooks.go      | 17 ++++++++++++-----
 pkg/filesystem/hooks_test.go | 16 +++++++++++-----
 4 files changed, 40 insertions(+), 16 deletions(-)

diff --git a/pkg/filesystem/context.go b/pkg/filesystem/context.go
index 59ddc2b..3c7406f 100644
--- a/pkg/filesystem/context.go
+++ b/pkg/filesystem/context.go
@@ -7,6 +7,6 @@ const (
 	GinCtx key = iota
 	// SavePathCtx 文件物理路径
 	SavePathCtx
-	// FileCtx 上传的文件
-	FileCtx
+	// FileHeaderCtx 上传的文件
+	FileHeaderCtx
 )
diff --git a/pkg/filesystem/filesystem.go b/pkg/filesystem/filesystem.go
index 89c014d..e3cb67f 100644
--- a/pkg/filesystem/filesystem.go
+++ b/pkg/filesystem/filesystem.go
@@ -73,6 +73,16 @@ func NewFileSystem(user *model.User) (*FileSystem, error) {
 	}, nil
 }
 
+/* ============
+	 文件相关
+   ============
+*/
+
+// AddFile 新增文件记录
+func (fs *FileSystem) AddFile(parent *model.Folder) error {
+	return nil
+}
+
 /* ================
 	 上传处理相关
    ================
@@ -80,7 +90,7 @@ func NewFileSystem(user *model.User) (*FileSystem, error) {
 
 // Upload 上传文件
 func (fs *FileSystem) Upload(ctx context.Context, file FileHeader) (err error) {
-	ctx = context.WithValue(ctx, FileCtx, file)
+	ctx = context.WithValue(ctx, FileHeaderCtx, file)
 
 	// 上传前的钩子
 	if fs.BeforeUpload != nil {
@@ -162,9 +172,10 @@ func (fs *FileSystem) CancelUpload(ctx context.Context, path string, file FileHe
 */
 
 // IsPathExist 返回给定目录是否存在
-func (fs *FileSystem) IsPathExist(path string) bool {
-	_, err := model.GetFolderByPath(path, fs.User.ID)
-	return err == nil
+// 如果存在就返回目录
+func (fs *FileSystem) IsPathExist(path string) (bool, model.Folder) {
+	folder, err := model.GetFolderByPath(path, fs.User.ID)
+	return err == nil, folder
 }
 
 // IsFileExist 返回给定路径的文件是否存在
diff --git a/pkg/filesystem/hooks.go b/pkg/filesystem/hooks.go
index f939b44..3d4d3a6 100644
--- a/pkg/filesystem/hooks.go
+++ b/pkg/filesystem/hooks.go
@@ -9,7 +9,7 @@ import (
 
 // GenericBeforeUpload 通用上传前处理钩子，包含数据库操作
 func GenericBeforeUpload(ctx context.Context, fs *FileSystem) error {
-	file := ctx.Value(FileCtx).(FileHeader)
+	file := ctx.Value(FileHeaderCtx).(FileHeader)
 
 	// 验证单文件尺寸
 	if !fs.ValidateFileSize(ctx, file.GetSize()) {
@@ -35,7 +35,7 @@ func GenericBeforeUpload(ctx context.Context, fs *FileSystem) error {
 
 // GenericAfterUploadCanceled 通用上传取消处理钩子，包含数据库操作
 func GenericAfterUploadCanceled(ctx context.Context, fs *FileSystem) error {
-	file := ctx.Value(FileCtx).(FileHeader)
+	file := ctx.Value(FileHeaderCtx).(FileHeader)
 
 	filePath := ctx.Value(SavePathCtx).(string)
 	// 删除临时文件
@@ -56,20 +56,27 @@ func GenericAfterUploadCanceled(ctx context.Context, fs *FileSystem) error {
 // GenericAfterUpload 文件上传完成后，包含数据库操作
 func GenericAfterUpload(ctx context.Context, fs *FileSystem) error {
 	// 文件存放的虚拟路径
-	virtualPath := ctx.Value(FileCtx).(FileHeader).GetVirtualPath()
+	virtualPath := ctx.Value(FileHeaderCtx).(FileHeader).GetVirtualPath()
 
 	// 检查路径是否存在
-	if !fs.IsPathExist(virtualPath) {
+	isExist, folder := fs.IsPathExist(virtualPath)
+	if !isExist {
 		return errors.New("路径\"" + virtualPath + "\"不存在")
 	}
 
 	// 检查文件是否存在
 	if fs.IsFileExist(path.Join(
 		virtualPath,
-		ctx.Value(FileCtx).(FileHeader).GetFileName(),
+		ctx.Value(FileHeaderCtx).(FileHeader).GetFileName(),
 	)) {
 		return errors.New("同名文件已存在")
 	}
 
+	// 向数据库中插入记录
+	err := fs.AddFile(&folder)
+	if err != nil {
+		return errors.New("无法插入文件记录")
+	}
+
 	return nil
 }
diff --git a/pkg/filesystem/hooks_test.go b/pkg/filesystem/hooks_test.go
index 528820c..c6d808f 100644
--- a/pkg/filesystem/hooks_test.go
+++ b/pkg/filesystem/hooks_test.go
@@ -10,11 +10,11 @@ import (
 
 func TestGenericBeforeUpload(t *testing.T) {
 	asserts := assert.New(t)
-	ctx := context.Background()
 	file := local.FileData{
 		Size: 5,
 		Name: "1.txt",
 	}
+	ctx := context.WithValue(context.Background(), FileHeaderCtx, file)
 	fs := FileSystem{
 		User: &model.User{
 			Storage: 0,
@@ -30,12 +30,18 @@ func TestGenericBeforeUpload(t *testing.T) {
 		},
 	}
 
-	asserts.Error(GenericBeforeUpload(ctx, &fs, file))
+	asserts.Error(GenericBeforeUpload(ctx, &fs))
+
 	file.Size = 1
 	file.Name = "1"
-	asserts.Error(GenericBeforeUpload(ctx, &fs, file))
+	ctx = context.WithValue(context.Background(), FileHeaderCtx, file)
+	asserts.Error(GenericBeforeUpload(ctx, &fs))
+
 	file.Name = "1.txt"
-	asserts.NoError(GenericBeforeUpload(ctx, &fs, file))
+	ctx = context.WithValue(context.Background(), FileHeaderCtx, file)
+	asserts.NoError(GenericBeforeUpload(ctx, &fs))
+
 	file.Name = "1.t/xt"
-	asserts.Error(GenericBeforeUpload(ctx, &fs, file))
+	ctx = context.WithValue(context.Background(), FileHeaderCtx, file)
+	asserts.Error(GenericBeforeUpload(ctx, &fs))
 }