Feat: after uploading finished hook

models/folder.go

@@ -19,6 +19,6 @@ type Folder struct {
 // GetFolderByPath 根据绝对路径和UID查找目录
 func GetFolderByPath(path string, uid uint) (Folder, error) {
 	var folder Folder
-	result := DB.Where("owner = ? AND position_absolute = ?", uid, path).Find(&folder)
+	result := DB.Where("owner_id = ? AND position_absolute = ?", uid, path).Find(&folder)
 	return folder, result.Error
 }

models/folder_test.go

@@ -1,7 +1,7 @@
 package model
 
 import (
-	"fmt"
+	"github.com/DATA-DOG/go-sqlmock"
 	"github.com/stretchr/testify/assert"
 	"testing"
 )
@@ -9,12 +9,16 @@ import (
 func TestGetFolderByPath(t *testing.T) {
 	asserts := assert.New(t)
 
-	//policyRows := sqlmock.NewRows([]string{"id", "name"}).
-	//	AddRow(1, "默认上传策略")
-	//mock.ExpectQuery("^SELECT (.+)").WillReturnRows(policyRows)
-
-	folder,_ := GetFolderByPath("/测试/test",1)
-	fmt.Println(folder)
+	folderRows := sqlmock.NewRows([]string{"id", "name"}).
+		AddRow(1, "test")
+	mock.ExpectQuery("^SELECT (.+)").WillReturnRows(folderRows)
+	folder, _ := GetFolderByPath("/测试/test", 1)
+	asserts.Equal("test", folder.Name)
+	asserts.NoError(mock.ExpectationsWereMet())
+
+	folderRows = sqlmock.NewRows([]string{"id", "name"})
+	mock.ExpectQuery("^SELECT (.+)").WillReturnRows(folderRows)
+	folder, err := GetFolderByPath("/测试/test", 1)
+	asserts.Error(err)
 	asserts.NoError(mock.ExpectationsWereMet())
-	asserts.NoError(mock.)
 }