package filesystem import ( "context" "errors" "os" "testing" "github.com/cloudreve/Cloudreve/v3/pkg/filesystem/response" testMock "github.com/stretchr/testify/mock" "github.com/DATA-DOG/go-sqlmock" model "github.com/cloudreve/Cloudreve/v3/models" "github.com/cloudreve/Cloudreve/v3/pkg/cache" "github.com/cloudreve/Cloudreve/v3/pkg/conf" "github.com/cloudreve/Cloudreve/v3/pkg/filesystem/fsctx" "github.com/cloudreve/Cloudreve/v3/pkg/serializer" "github.com/cloudreve/Cloudreve/v3/pkg/util" "github.com/jinzhu/gorm" "github.com/stretchr/testify/assert" ) func TestFileSystem_ListPhysical(t *testing.T) { asserts := assert.New(t) fs := &FileSystem{ User: &model.User{ Model: gorm.Model{ ID: 1, }, }, Policy: &model.Policy{Type: "mock"}, } ctx := context.Background() // 未知存储策略 { fs.Policy.Type = "unknown" res, err := fs.ListPhysical(ctx, "/") asserts.Equal(ErrUnknownPolicyType, err) asserts.Empty(res) fs.Policy.Type = "mock" } // 无法列取目录 { testHandler := new(FileHeaderMock) testHandler.On("List", testMock.Anything, "/", testMock.Anything).Return([]response.Object{}, errors.New("error")) fs.Handler = testHandler res, err := fs.ListPhysical(ctx, "/") asserts.EqualError(err, "error") asserts.Empty(res) } // 成功 { testHandler := new(FileHeaderMock) testHandler.On("List", testMock.Anything, "/", testMock.Anything).Return( []response.Object{{IsDir: true, Name: "1"}, {IsDir: false, Name: "2"}}, nil, ) fs.Handler = testHandler res, err := fs.ListPhysical(ctx, "/") asserts.NoError(err) asserts.Len(res, 1) asserts.Equal("1", res[0].Name) } } func TestFileSystem_List(t *testing.T) { asserts := assert.New(t) fs := &FileSystem{User: &model.User{ Model: gorm.Model{ ID: 1, }, }} ctx := context.Background() // 成功,子目录包含文件和路径,不使用路径处理钩子 // 根目录 mock.ExpectQuery("SELECT(.+)"). WithArgs(1). WillReturnRows(sqlmock.NewRows([]string{"id", "name", "owner_id"}).AddRow(1, "/", 1)) // folder mock.ExpectQuery("SELECT(.+)"). WithArgs(1, 1, "folder"). WillReturnRows(sqlmock.NewRows([]string{"id", "name", "owner_id"}).AddRow(5, "folder", 1)) mock.ExpectQuery("SELECT(.+)folder(.+)").WillReturnRows(sqlmock.NewRows([]string{"id", "name"}).AddRow(6, "sub_folder1").AddRow(7, "sub_folder2")) mock.ExpectQuery("SELECT(.+)file(.+)").WillReturnRows(sqlmock.NewRows([]string{"id", "name"}).AddRow(6, "sub_file1.txt").AddRow(7, "sub_file2.txt")) objects, err := fs.List(ctx, "/folder", nil) asserts.Len(objects, 4) asserts.NoError(err) asserts.NoError(mock.ExpectationsWereMet()) // 成功,子目录包含文件和路径,不使用路径处理钩子,包含分享key // 根目录 mock.ExpectQuery("SELECT(.+)"). WithArgs(1). WillReturnRows(sqlmock.NewRows([]string{"id", "name", "owner_id"}).AddRow(1, "/", 1)) // folder mock.ExpectQuery("SELECT(.+)"). WithArgs(1, 1, "folder"). WillReturnRows(sqlmock.NewRows([]string{"id", "name", "owner_id"}).AddRow(5, "folder", 1)) mock.ExpectQuery("SELECT(.+)folder(.+)").WillReturnRows(sqlmock.NewRows([]string{"id", "name"}).AddRow(6, "sub_folder1").AddRow(7, "sub_folder2")) mock.ExpectQuery("SELECT(.+)file(.+)").WillReturnRows(sqlmock.NewRows([]string{"id", "name"}).AddRow(6, "sub_file1.txt").AddRow(7, "sub_file2.txt")) ctxWithKey := context.WithValue(ctx, fsctx.ShareKeyCtx, "share") objects, err = fs.List(ctxWithKey, "/folder", nil) asserts.Len(objects, 4) asserts.Equal("share", objects[3].Key) asserts.NoError(err) asserts.NoError(mock.ExpectationsWereMet()) // 成功,子目录包含文件和路径,使用路径处理钩子 mock.ExpectQuery("SELECT(.+)"). WithArgs(1). WillReturnRows(sqlmock.NewRows([]string{"id", "name", "owner_id"}).AddRow(1, "/", 1)) // folder mock.ExpectQuery("SELECT(.+)"). WithArgs(1, 1, "folder"). WillReturnRows(sqlmock.NewRows([]string{"id", "name", "owner_id"}).AddRow(2, "folder", 1)) mock.ExpectQuery("SELECT(.+)folder(.+)").WillReturnRows(sqlmock.NewRows([]string{"id", "name", "position"}).AddRow(6, "sub_folder1", "/folder").AddRow(7, "sub_folder2", "/folder")) mock.ExpectQuery("SELECT(.+)file(.+)").WillReturnRows(sqlmock.NewRows([]string{"id", "name", "dir"}).AddRow(6, "sub_file1.txt", "/folder").AddRow(7, "sub_file2.txt", "/folder")) objects, err = fs.List(ctx, "/folder", func(s string) string { return "prefix" + s }) asserts.Len(objects, 4) asserts.NoError(err) asserts.NoError(mock.ExpectationsWereMet()) for _, value := range objects { asserts.Contains(value.Path, "prefix/") } // 成功,子目录包含路径,使用路径处理钩子 mock.ExpectQuery("SELECT(.+)"). WithArgs(1). WillReturnRows(sqlmock.NewRows([]string{"id", "name", "owner_id"}).AddRow(1, "/", 1)) // folder mock.ExpectQuery("SELECT(.+)"). WithArgs(1, 1, "folder"). WillReturnRows(sqlmock.NewRows([]string{"id", "name", "owner_id"}).AddRow(2, "folder", 1)) mock.ExpectQuery("SELECT(.+)folder(.+)").WillReturnRows(sqlmock.NewRows([]string{"id", "name", "position"})) mock.ExpectQuery("SELECT(.+)file(.+)").WillReturnRows(sqlmock.NewRows([]string{"id", "name", "dir"}).AddRow(6, "sub_file1.txt", "/folder").AddRow(7, "sub_file2.txt", "/folder")) objects, err = fs.List(ctx, "/folder", func(s string) string { return "prefix" + s }) asserts.Len(objects, 2) asserts.NoError(err) asserts.NoError(mock.ExpectationsWereMet()) for _, value := range objects { asserts.Contains(value.Path, "prefix/") } // 成功,子目录下为空,使用路径处理钩子 mock.ExpectQuery("SELECT(.+)"). WithArgs(1). WillReturnRows(sqlmock.NewRows([]string{"id", "name", "owner_id"}).AddRow(1, "/", 1)) // folder mock.ExpectQuery("SELECT(.+)"). WithArgs(1, 1, "folder"). WillReturnRows(sqlmock.NewRows([]string{"id", "name", "owner_id"}).AddRow(2, "folder", 1)) mock.ExpectQuery("SELECT(.+)folder(.+)").WillReturnRows(sqlmock.NewRows([]string{"id", "name", "position"})) mock.ExpectQuery("SELECT(.+)file(.+)").WillReturnRows(sqlmock.NewRows([]string{"id", "name", "dir"})) objects, err = fs.List(ctx, "/folder", func(s string) string { return "prefix" + s }) asserts.Len(objects, 0) asserts.NoError(err) asserts.NoError(mock.ExpectationsWereMet()) // 成功,子目录路径不存在 mock.ExpectQuery("SELECT(.+)"). WithArgs(1). WillReturnRows(sqlmock.NewRows([]string{"id", "name", "owner_id"}).AddRow(1, "/", 1)) // folder mock.ExpectQuery("SELECT(.+)"). WithArgs(1, 1, "folder"). WillReturnRows(sqlmock.NewRows([]string{"id", "name", "owner_id"})) objects, err = fs.List(ctx, "/folder", func(s string) string { return "prefix" + s }) asserts.Len(objects, 0) asserts.NoError(mock.ExpectationsWereMet()) } func TestFileSystem_CreateDirectory(t *testing.T) { asserts := assert.New(t) fs := &FileSystem{User: &model.User{ Model: gorm.Model{ ID: 1, }, }} ctx := context.Background() // 目录名非法 _, err := fs.CreateDirectory(ctx, "/ad/a+?") asserts.Equal(ErrIllegalObjectName, err) // 存在同名文件 // 根目录 mock.ExpectQuery("SELECT(.+)"). WithArgs(1). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(1, 1)) // ad mock.ExpectQuery("SELECT(.+)"). WithArgs(1, 1, "ad"). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(2, 1)) mock.ExpectQuery("SELECT(.+)files").WillReturnRows(sqlmock.NewRows([]string{"id", "name"}).AddRow(1, "ab")) _, err = fs.CreateDirectory(ctx, "/ad/ab") asserts.Equal(ErrFileExisted, err) asserts.NoError(mock.ExpectationsWereMet()) // 存在同名目录,直接返回 mock.ExpectQuery("SELECT(.+)"). WithArgs(1). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(1, 1)) // ad mock.ExpectQuery("SELECT(.+)"). WithArgs(1, 1, "ad"). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(2, 1)) mock.ExpectQuery("SELECT(.+)files").WillReturnRows(sqlmock.NewRows([]string{"id", "name"})) // ab mock.ExpectQuery("SELECT(.+)"). WithArgs("ab", 2, 1). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(3, 1)) res, err := fs.CreateDirectory(ctx, "/ad/ab") asserts.NoError(err) asserts.EqualValues(3, res.ID) asserts.NoError(mock.ExpectationsWereMet()) // 成功创建 mock.ExpectQuery("SELECT(.+)"). WithArgs(1). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(1, 1)) // ad mock.ExpectQuery("SELECT(.+)"). WithArgs(1, 1, "ad"). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(2, 1)) mock.ExpectQuery("SELECT(.+)files").WillReturnRows(sqlmock.NewRows([]string{"id", "name"})) mock.ExpectQuery("SELECT(.+)"). WithArgs("ab", 2, 1). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"})) mock.ExpectBegin() mock.ExpectExec("INSERT(.+)").WillReturnResult(sqlmock.NewResult(1, 1)) mock.ExpectCommit() _, err = fs.CreateDirectory(ctx, "/ad/ab") asserts.NoError(err) asserts.NoError(mock.ExpectationsWereMet()) // 成功创建, 递归创建父目录 // 根目录 mock.ExpectQuery("SELECT(.+)"). WithArgs(1). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(1, 1)) // ad mock.ExpectQuery("SELECT(.+)"). WithArgs(1, 1, "ad"). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"})) // 根目录 mock.ExpectQuery("SELECT(.+)"). WithArgs(1). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(1, 1)) mock.ExpectQuery("SELECT(.+)files").WillReturnRows(sqlmock.NewRows([]string{"id", "name"})) // 创建ad mock.ExpectQuery("SELECT(.+)"). WithArgs("ad", 1, 1). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"})) mock.ExpectBegin() mock.ExpectExec("INSERT(.+)").WillReturnResult(sqlmock.NewResult(2, 1)) mock.ExpectCommit() mock.ExpectQuery("SELECT(.+)files").WillReturnRows(sqlmock.NewRows([]string{"id", "name"})) // 创建ab mock.ExpectQuery("SELECT(.+)"). WithArgs("ab", 2, 1). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"})) mock.ExpectBegin() mock.ExpectExec("INSERT(.+)").WillReturnResult(sqlmock.NewResult(1, 1)) mock.ExpectCommit() _, err = fs.CreateDirectory(ctx, "/ad/ab") asserts.NoError(err) asserts.NoError(mock.ExpectationsWereMet()) // 底层创建失败 // 成功创建 mock.ExpectQuery("SELECT(.+)"). WithArgs(1). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(1, 1)) // ad mock.ExpectQuery("SELECT(.+)"). WithArgs(1, 1, "ad"). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"})) // 根目录 mock.ExpectQuery("SELECT(.+)"). WithArgs(1). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(1, 1)) mock.ExpectQuery("SELECT(.+)files").WillReturnRows(sqlmock.NewRows([]string{"id", "name"})) // 创建ad mock.ExpectQuery("SELECT(.+)"). WithArgs("ad", 1, 1). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"})) mock.ExpectBegin() mock.ExpectExec("INSERT(.+)").WillReturnResult(sqlmock.NewResult(2, 1)).WillReturnError(errors.New("error")) mock.ExpectRollback() mock.ExpectQuery("SELECT(.+)"). WillReturnError(errors.New("error")) _, err = fs.CreateDirectory(ctx, "/ad/ab") asserts.Error(err) asserts.NoError(mock.ExpectationsWereMet()) // 直接创建根目录 mock.ExpectQuery("SELECT(.+)"). WithArgs(1). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(1, 1)) _, err = fs.CreateDirectory(ctx, "/") asserts.NoError(err) asserts.NoError(mock.ExpectationsWereMet()) // 直接创建根目录, 重设根目录 fs.Root = &model.Folder{} _, err = fs.CreateDirectory(ctx, "/") asserts.NoError(err) asserts.NoError(mock.ExpectationsWereMet()) } func TestFileSystem_ListDeleteFiles(t *testing.T) { conf.DatabaseConfig.Type = "mysql" asserts := assert.New(t) fs := &FileSystem{User: &model.User{ Model: gorm.Model{ ID: 1, }, }} // 成功 { mock.ExpectQuery("SELECT(.+)").WillReturnRows(sqlmock.NewRows([]string{"id", "name"}).AddRow(1, "1.txt").AddRow(2, "2.txt")) err := fs.ListDeleteFiles(context.Background(), []uint{1}) asserts.NoError(err) asserts.NoError(mock.ExpectationsWereMet()) } // 失败 { mock.ExpectQuery("SELECT(.+)").WillReturnError(errors.New("error")) err := fs.ListDeleteFiles(context.Background(), []uint{1}) asserts.Error(err) asserts.Equal(serializer.CodeDBError, err.(serializer.AppError).Code) asserts.NoError(mock.ExpectationsWereMet()) } } func TestFileSystem_ListDeleteDirs(t *testing.T) { conf.DatabaseConfig.Type = "mysql" asserts := assert.New(t) fs := &FileSystem{User: &model.User{ Model: gorm.Model{ ID: 1, }, }} // 成功 { mock.ExpectQuery("SELECT(.+)"). WillReturnRows( sqlmock.NewRows([]string{"id", "parent_id"}). AddRow(1, 0). AddRow(2, 0). AddRow(3, 0), ) mock.ExpectQuery("SELECT(.+)files(.+)"). WithArgs(1, 2, 3). WillReturnRows( sqlmock.NewRows([]string{"id", "name"}). AddRow(4, "1.txt"). AddRow(5, "2.txt"). AddRow(6, "3.txt"), ) err := fs.ListDeleteDirs(context.Background(), []uint{1}) asserts.NoError(err) asserts.Len(fs.FileTarget, 3) asserts.Len(fs.DirTarget, 3) asserts.NoError(mock.ExpectationsWereMet()) } // 成功,忽略根目录 { mock.ExpectQuery("SELECT(.+)"). WillReturnRows( sqlmock.NewRows([]string{"id", "parent_id"}). AddRow(1, 0). AddRow(2, nil). AddRow(3, 0), ) mock.ExpectQuery("SELECT(.+)files(.+)"). WithArgs(1, 3). WillReturnRows( sqlmock.NewRows([]string{"id", "name"}). AddRow(4, "1.txt"). AddRow(5, "2.txt"). AddRow(6, "3.txt"), ) fs.CleanTargets() err := fs.ListDeleteDirs(context.Background(), []uint{1}) asserts.NoError(err) asserts.Len(fs.FileTarget, 3) asserts.Len(fs.DirTarget, 2) asserts.NoError(mock.ExpectationsWereMet()) } // 检索文件发生错误 { mock.ExpectQuery("SELECT(.+)"). WillReturnRows( sqlmock.NewRows([]string{"id", "parent_id"}). AddRow(1, 0). AddRow(2, 0). AddRow(3, 0), ) mock.ExpectQuery("SELECT(.+)"). WithArgs(1, 2, 3). WillReturnError(errors.New("error")) fs.CleanTargets() err := fs.ListDeleteDirs(context.Background(), []uint{1}) asserts.Error(err) asserts.Len(fs.DirTarget, 3) asserts.NoError(mock.ExpectationsWereMet()) } // 检索目录发生错误 { mock.ExpectQuery("SELECT(.+)"). WillReturnError(errors.New("error")) err := fs.ListDeleteDirs(context.Background(), []uint{1}) asserts.Error(err) asserts.NoError(mock.ExpectationsWereMet()) } } func TestFileSystem_Delete(t *testing.T) { conf.DatabaseConfig.Type = "mysql" asserts := assert.New(t) cache.Set("pack_size_1", uint64(0), 0) fs := &FileSystem{User: &model.User{ Model: gorm.Model{ ID: 0, }, Storage: 3, Group: model.Group{MaxStorage: 3}, }} ctx := context.Background() //全部未成功,强制 { fs.CleanTargets() mock.ExpectQuery("SELECT(.+)"). WillReturnRows( sqlmock.NewRows([]string{"id", "parent_id"}). AddRow(1, 0). AddRow(2, 0). AddRow(3, 0), ) mock.ExpectQuery("SELECT(.+)"). WithArgs(1, 2, 3). WillReturnRows( sqlmock.NewRows([]string{"id", "name", "source_name", "policy_id", "size"}). AddRow(4, "1.txt", "1.txt", 365, 1), ) mock.ExpectQuery("SELECT(.+)").WillReturnRows(sqlmock.NewRows([]string{"id", "name", "source_name", "policy_id", "size"}).AddRow(1, "2.txt", "2.txt", 365, 2)) // 两次查询软连接 mock.ExpectQuery("SELECT(.+)files(.+)"). WillReturnRows(sqlmock.NewRows([]string{"id", "policy_id", "source_name"})) mock.ExpectQuery("SELECT(.+)files(.+)"). WillReturnRows(sqlmock.NewRows([]string{"id", "policy_id", "source_name"})) // 查询上传策略 mock.ExpectQuery("SELECT(.+)").WillReturnRows(sqlmock.NewRows([]string{"id", "type"}).AddRow(365, "local")) // 删除文件记录 mock.ExpectBegin() mock.ExpectExec("DELETE(.+)"). WillReturnResult(sqlmock.NewResult(0, 1)) mock.ExpectExec("DELETE(.+)"). WillReturnResult(sqlmock.NewResult(0, 1)) mock.ExpectExec("UPDATE(.+)users(.+)storage(.+)").WillReturnResult(sqlmock.NewResult(1, 1)) mock.ExpectCommit() // 删除对应分享 mock.ExpectBegin() mock.ExpectExec("UPDATE(.+)shares"). WillReturnResult(sqlmock.NewResult(0, 3)) mock.ExpectCommit() // 删除目录 mock.ExpectBegin() mock.ExpectExec("DELETE(.+)"). WillReturnResult(sqlmock.NewResult(0, 3)) mock.ExpectCommit() // 删除对应分享 mock.ExpectBegin() mock.ExpectExec("UPDATE(.+)shares"). WillReturnResult(sqlmock.NewResult(0, 3)) mock.ExpectCommit() fs.FileTarget = []model.File{} fs.DirTarget = []model.Folder{} err := fs.Delete(ctx, []uint{1}, []uint{1}, true) asserts.NoError(err) } //全部成功 { fs.CleanTargets() file, err := os.Create(util.RelativePath("1.txt")) file2, err := os.Create(util.RelativePath("2.txt")) file.Close() file2.Close() asserts.NoError(err) mock.ExpectQuery("SELECT(.+)"). WillReturnRows( sqlmock.NewRows([]string{"id", "parent_id"}). AddRow(1, 0). AddRow(2, 0). AddRow(3, 0), ) mock.ExpectQuery("SELECT(.+)"). WithArgs(1, 2, 3). WillReturnRows( sqlmock.NewRows([]string{"id", "name", "source_name", "policy_id", "size"}). AddRow(4, "1.txt", "1.txt", 602, 1), ) mock.ExpectQuery("SELECT(.+)").WillReturnRows(sqlmock.NewRows([]string{"id", "name", "source_name", "policy_id", "size"}).AddRow(1, "2.txt", "2.txt", 602, 2)) // 两次查询软连接 mock.ExpectQuery("SELECT(.+)files(.+)"). WillReturnRows(sqlmock.NewRows([]string{"id", "policy_id", "source_name"})) mock.ExpectQuery("SELECT(.+)files(.+)"). WillReturnRows(sqlmock.NewRows([]string{"id", "policy_id", "source_name"})) // 查询上传策略 mock.ExpectQuery("SELECT(.+)").WillReturnRows(sqlmock.NewRows([]string{"id", "type"}).AddRow(602, "local")) // 删除文件记录 mock.ExpectBegin() mock.ExpectExec("DELETE(.+)"). WillReturnResult(sqlmock.NewResult(0, 1)) mock.ExpectExec("DELETE(.+)"). WillReturnResult(sqlmock.NewResult(0, 1)) mock.ExpectExec("UPDATE(.+)users(.+)storage(.+)").WillReturnResult(sqlmock.NewResult(1, 1)) mock.ExpectCommit() // 删除对应分享 mock.ExpectBegin() mock.ExpectExec("UPDATE(.+)shares"). WillReturnResult(sqlmock.NewResult(0, 3)) mock.ExpectCommit() // 删除目录 mock.ExpectBegin() mock.ExpectExec("DELETE(.+)"). WillReturnResult(sqlmock.NewResult(0, 3)) mock.ExpectCommit() // 删除对应分享 mock.ExpectBegin() mock.ExpectExec("UPDATE(.+)shares"). WillReturnResult(sqlmock.NewResult(0, 3)) mock.ExpectCommit() fs.FileTarget = []model.File{} fs.DirTarget = []model.Folder{} err = fs.Delete(ctx, []uint{1}, []uint{1}, false) asserts.NoError(err) } } func TestFileSystem_Copy(t *testing.T) { asserts := assert.New(t) cache.Set("pack_size_1", uint64(0), 0) fs := &FileSystem{User: &model.User{ Model: gorm.Model{ ID: 1, }, Storage: 3, Group: model.Group{MaxStorage: 3}, }} ctx := context.Background() // 目录不存在 { mock.ExpectQuery("SELECT(.+)").WillReturnRows( sqlmock.NewRows([]string{"name"}), ) mock.ExpectQuery("SELECT(.+)").WillReturnRows( sqlmock.NewRows([]string{"name"}), ) err := fs.Copy(ctx, []uint{}, []uint{}, "/src", "/dst") asserts.Equal(ErrPathNotExist, err) asserts.NoError(mock.ExpectationsWereMet()) } // 复制目录出错 { // 根目录 mock.ExpectQuery("SELECT(.+)"). WithArgs(1). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(1, 1)) // 1 mock.ExpectQuery("SELECT(.+)"). WithArgs(1, 1, "dst"). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(2, 1)) // 根目录 mock.ExpectQuery("SELECT(.+)"). WithArgs(1). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(1, 1)) // 1 mock.ExpectQuery("SELECT(.+)"). WithArgs(1, 1, "src"). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(2, 1)) err := fs.Copy(ctx, []uint{1}, []uint{}, "/src", "/dst") asserts.Error(err) asserts.NoError(mock.ExpectationsWereMet()) } } func TestFileSystem_Move(t *testing.T) { asserts := assert.New(t) cache.Set("pack_size_1", uint64(0), 0) fs := &FileSystem{User: &model.User{ Model: gorm.Model{ ID: 1, }, Storage: 3, Group: model.Group{MaxStorage: 3}, }} ctx := context.Background() // 目录不存在 { mock.ExpectQuery("SELECT(.+)").WillReturnRows( sqlmock.NewRows([]string{"name"}), ) err := fs.Move(ctx, []uint{}, []uint{}, "/src", "/dst") asserts.Equal(ErrPathNotExist, err) asserts.NoError(mock.ExpectationsWereMet()) } // 移动目录出错 { // 根目录 mock.ExpectQuery("SELECT(.+)"). WithArgs(1). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(1, 1)) // 1 mock.ExpectQuery("SELECT(.+)"). WithArgs(1, 1, "dst"). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(2, 1)) // 根目录 mock.ExpectQuery("SELECT(.+)"). WithArgs(1). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(1, 1)) // 1 mock.ExpectQuery("SELECT(.+)"). WithArgs(1, 1, "src"). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(2, 1)) err := fs.Move(ctx, []uint{1}, []uint{}, "/src", "/dst") asserts.Error(err) asserts.NoError(mock.ExpectationsWereMet()) } } func TestFileSystem_Rename(t *testing.T) { asserts := assert.New(t) fs := &FileSystem{User: &model.User{ Model: gorm.Model{ ID: 1, }, }, Policy: &model.Policy{}, } ctx := context.Background() // 重命名文件 成功 { mock.ExpectQuery("SELECT(.+)files(.+)"). WithArgs(10, 1). WillReturnRows(sqlmock.NewRows([]string{"id", "name"}).AddRow(10, "old.text")) mock.ExpectBegin() mock.ExpectExec("UPDATE(.+)files(.+)SET(.+)"). WithArgs("new.txt", 10). WillReturnResult(sqlmock.NewResult(1, 1)) mock.ExpectCommit() err := fs.Rename(ctx, []uint{}, []uint{10}, "new.txt") asserts.NoError(mock.ExpectationsWereMet()) asserts.NoError(err) } // 重命名文件 不存在 { mock.ExpectQuery("SELECT(.+)files(.+)"). WithArgs(10, 1). WillReturnRows(sqlmock.NewRows([]string{"id", "name"})) err := fs.Rename(ctx, []uint{}, []uint{10}, "new.txt") asserts.NoError(mock.ExpectationsWereMet()) asserts.Error(err) asserts.Equal(ErrPathNotExist, err) } // 重命名文件 失败 { mock.ExpectQuery("SELECT(.+)files(.+)"). WithArgs(10, 1). WillReturnRows(sqlmock.NewRows([]string{"id", "name"}).AddRow(10, "old.text")) mock.ExpectBegin() mock.ExpectExec("UPDATE(.+)files(.+)SET(.+)"). WithArgs("new.txt", 10). WillReturnError(errors.New("error")) mock.ExpectRollback() err := fs.Rename(ctx, []uint{}, []uint{10}, "new.txt") asserts.NoError(mock.ExpectationsWereMet()) asserts.Error(err) asserts.Equal(ErrFileExisted, err) } // 重命名目录 成功 { mock.ExpectQuery("SELECT(.+)folders(.+)"). WithArgs(10, 1). WillReturnRows(sqlmock.NewRows([]string{"id", "name"}).AddRow(10, "old")) mock.ExpectBegin() mock.ExpectExec("UPDATE(.+)folders(.+)SET(.+)"). WithArgs("new", 10). WillReturnResult(sqlmock.NewResult(1, 1)) mock.ExpectCommit() err := fs.Rename(ctx, []uint{10}, []uint{}, "new") asserts.NoError(mock.ExpectationsWereMet()) asserts.NoError(err) } // 重命名目录 不存在 { mock.ExpectQuery("SELECT(.+)folders(.+)"). WithArgs(10, 1). WillReturnRows(sqlmock.NewRows([]string{"id", "name"})) err := fs.Rename(ctx, []uint{10}, []uint{}, "new") asserts.NoError(mock.ExpectationsWereMet()) asserts.Error(err) asserts.Equal(ErrPathNotExist, err) } // 重命名目录 失败 { mock.ExpectQuery("SELECT(.+)folders(.+)"). WithArgs(10, 1). WillReturnRows(sqlmock.NewRows([]string{"id", "name"}).AddRow(10, "old")) mock.ExpectBegin() mock.ExpectExec("UPDATE(.+)folders(.+)SET(.+)"). WithArgs("new", 10). WillReturnError(errors.New("error")) mock.ExpectRollback() err := fs.Rename(ctx, []uint{10}, []uint{}, "new") asserts.NoError(mock.ExpectationsWereMet()) asserts.Error(err) asserts.Equal(ErrFileExisted, err) } // 未选中任何对象 { err := fs.Rename(ctx, []uint{}, []uint{}, "new") asserts.Error(err) asserts.Equal(ErrPathNotExist, err) } // 新名字是目录,不合法 { err := fs.Rename(ctx, []uint{10}, []uint{}, "ne/w") asserts.Error(err) asserts.Equal(ErrIllegalObjectName, err) } // 新名字是文件,不合法 { err := fs.Rename(ctx, []uint{}, []uint{10}, "ne/w") asserts.Error(err) asserts.Equal(ErrIllegalObjectName, err) } // 新名字是文件,扩展名不合法 { fs.Policy.OptionsSerialized.FileType = []string{"txt"} err := fs.Rename(ctx, []uint{}, []uint{10}, "1.jpg") asserts.Error(err) asserts.Equal(ErrIllegalObjectName, err) } // 新名字是目录,不应该检测扩展名 { fs.Policy.OptionsSerialized.FileType = []string{"txt"} mock.ExpectQuery("SELECT(.+)folders(.+)"). WithArgs(10, 1). WillReturnRows(sqlmock.NewRows([]string{"id", "name"})) err := fs.Rename(ctx, []uint{10}, []uint{}, "new") asserts.NoError(mock.ExpectationsWereMet()) asserts.Error(err) asserts.Equal(ErrPathNotExist, err) } } func TestFileSystem_SaveTo(t *testing.T) { asserts := assert.New(t) fs := &FileSystem{User: &model.User{ Model: gorm.Model{ ID: 1, }, }} ctx := context.Background() // 单文件 失败 { // 根目录 mock.ExpectQuery("SELECT(.+)"). WithArgs(1). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(1, 1)) mock.ExpectQuery("SELECT(.+)").WillReturnError(errors.New("error")) fs.SetTargetFile(&[]model.File{{Name: "test.txt"}}) err := fs.SaveTo(ctx, "/") asserts.NoError(mock.ExpectationsWereMet()) asserts.Error(err) } // 目录 成功 { // 根目录 mock.ExpectQuery("SELECT(.+)"). WithArgs(1). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(1, 1)) mock.ExpectQuery("SELECT(.+)").WillReturnError(errors.New("error")) fs.SetTargetDir(&[]model.Folder{{Name: "folder"}}) err := fs.SaveTo(ctx, "/") asserts.NoError(mock.ExpectationsWereMet()) asserts.Error(err) } // 父目录不存在 { // 根目录 mock.ExpectQuery("SELECT(.+)"). WithArgs(1). WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"})) fs.SetTargetDir(&[]model.Folder{{Name: "folder"}}) err := fs.SaveTo(ctx, "/") asserts.NoError(mock.ExpectationsWereMet()) asserts.Error(err) } }