Infrastructure/Cloudreve
0
Fork 0
Code Issues Pull requests Activity

fix: incorrect progress count (#1379)

Browse source 
* fix: incorrect progress count

文件中转 已完成文件计数 应在文件成功上传后+1 #1367

* fix failed ut

Co-authored-by: HFO4 <912394456@qq.com>
This commit is contained in:
小白-白 2022-07-12 19:30:41 +08:00 • committed by GitHub
parent c99b36f788
commit 9869671633
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
pkg/task/tranfer.go
View file

@ -96,9 +96,8 @@ func (job *TransferTask) Do() {
return
}
for index, file := range job.TaskProps.Src {
job.TaskModel.SetProgress(index)
successCount := 0
for _, file := range job.TaskProps.Src {
dst := path.Join(job.TaskProps.Dst, filepath.Base(file))
if job.TaskProps.TrimPath {
// 保留原始目录
@ -132,6 +131,9 @@ func (job *TransferTask) Do() {
if err != nil {
job.SetErrorMsg("文件转存失败", err)
} else {
successCount++
job.TaskModel.SetProgress(successCount)
}
}

10
pkg/task/transfer_test.go
View file

@ -88,11 +88,6 @@ func TestTransferTask_Do(t *testing.T) {
}
task.TaskProps.Src = []string{"test/not_exist"}
task.TaskProps.Parent = "test/not_exist"
// 更新进度
mock.ExpectBegin()
mock.ExpectExec("UPDATE(.+)").WillReturnResult(sqlmock.NewResult(1,
1))
mock.ExpectCommit()
// 更新错误
mock.ExpectBegin()
mock.ExpectExec("UPDATE(.+)").WillReturnResult(sqlmock.NewResult(1,
@ -113,11 +108,6 @@ func TestTransferTask_Do(t *testing.T) {
task.TaskProps.Src = []string{"test/not_exist"}
task.TaskProps.Parent = "test/not_exist"
task.TaskProps.TrimPath = true
// 更新进度
mock.ExpectBegin()
mock.ExpectExec("UPDATE(.+)").WillReturnResult(sqlmock.NewResult(1,
1))
mock.ExpectCommit()
// 更新错误
mock.ExpectBegin()
mock.ExpectExec("UPDATE(.+)").WillReturnResult(sqlmock.NewResult(1,