Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
hexbee
Cloudreve
提交
030fd4ac
C
Cloudreve
项目概览
hexbee
/
Cloudreve
通知
2
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
C
Cloudreve
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
030fd4ac
编写于
4月 27, 2020
作者:
H
HFO4
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Test: task.ImportTask
上级
9eeb4b6d
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
240 addition
and
0 deletion
+240
-0
pkg/task/import_test.go
pkg/task/import_test.go
+240
-0
未找到文件。
pkg/task/import_test.go
0 → 100644
浏览文件 @
030fd4ac
package
task
import
(
"errors"
"github.com/DATA-DOG/go-sqlmock"
model
"github.com/HFO4/cloudreve/models"
"github.com/HFO4/cloudreve/pkg/cache"
"github.com/HFO4/cloudreve/pkg/util"
"github.com/jinzhu/gorm"
"github.com/stretchr/testify/assert"
"testing"
)
func
TestImportTask_Props
(
t
*
testing
.
T
)
{
asserts
:=
assert
.
New
(
t
)
task
:=
&
ImportTask
{
User
:
&
model
.
User
{},
}
asserts
.
NotEmpty
(
task
.
Props
())
asserts
.
Equal
(
ImportTaskType
,
task
.
Type
())
asserts
.
EqualValues
(
0
,
task
.
Creator
())
asserts
.
Nil
(
task
.
Model
())
}
func
TestImportTask_SetStatus
(
t
*
testing
.
T
)
{
asserts
:=
assert
.
New
(
t
)
task
:=
&
ImportTask
{
User
:
&
model
.
User
{},
TaskModel
:
&
model
.
Task
{
Model
:
gorm
.
Model
{
ID
:
1
},
},
}
mock
.
ExpectBegin
()
mock
.
ExpectExec
(
"UPDATE(.+)"
)
.
WillReturnResult
(
sqlmock
.
NewResult
(
1
,
1
))
mock
.
ExpectCommit
()
task
.
SetStatus
(
3
)
asserts
.
NoError
(
mock
.
ExpectationsWereMet
())
}
func
TestImportTask_SetError
(
t
*
testing
.
T
)
{
asserts
:=
assert
.
New
(
t
)
task
:=
&
ImportTask
{
User
:
&
model
.
User
{},
TaskModel
:
&
model
.
Task
{
Model
:
gorm
.
Model
{
ID
:
1
},
},
}
mock
.
ExpectBegin
()
mock
.
ExpectExec
(
"UPDATE(.+)"
)
.
WillReturnResult
(
sqlmock
.
NewResult
(
1
,
1
))
mock
.
ExpectCommit
()
task
.
SetErrorMsg
(
"error"
,
nil
)
asserts
.
NoError
(
mock
.
ExpectationsWereMet
())
asserts
.
Equal
(
"error"
,
task
.
GetError
()
.
Msg
)
}
func
TestImportTask_Do
(
t
*
testing
.
T
)
{
asserts
:=
assert
.
New
(
t
)
task
:=
&
ImportTask
{
User
:
&
model
.
User
{},
TaskModel
:
&
model
.
Task
{
Model
:
gorm
.
Model
{
ID
:
1
},
},
TaskProps
:
ImportProps
{
PolicyID
:
63
,
Src
:
""
,
Recursive
:
false
,
Dst
:
""
,
},
}
// 存储策略不存在
{
cache
.
Deletes
([]
string
{
"63"
},
"policy_"
)
mock
.
ExpectQuery
(
"SELECT(.+)policies(.+)"
)
.
WillReturnRows
(
sqlmock
.
NewRows
([]
string
{
"id"
}))
// 设定失败状态
mock
.
ExpectBegin
()
mock
.
ExpectExec
(
"UPDATE(.+)"
)
.
WillReturnResult
(
sqlmock
.
NewResult
(
1
,
1
))
mock
.
ExpectCommit
()
task
.
Do
()
asserts
.
NoError
(
mock
.
ExpectationsWereMet
())
asserts
.
NotEmpty
(
task
.
Err
.
Error
)
task
.
Err
=
nil
}
// 无法分配 Filesystem
{
cache
.
Deletes
([]
string
{
"63"
},
"policy_"
)
mock
.
ExpectQuery
(
"SELECT(.+)policies(.+)"
)
.
WillReturnRows
(
sqlmock
.
NewRows
([]
string
{
"id"
,
"type"
})
.
AddRow
(
63
,
"unknown"
))
// 设定失败状态
mock
.
ExpectBegin
()
mock
.
ExpectExec
(
"UPDATE(.+)"
)
.
WillReturnResult
(
sqlmock
.
NewResult
(
1
,
1
))
mock
.
ExpectCommit
()
task
.
Do
()
asserts
.
NoError
(
mock
.
ExpectationsWereMet
())
asserts
.
NotEmpty
(
task
.
Err
.
Msg
)
task
.
Err
=
nil
}
// 成功列取,但是文件为空
{
cache
.
Deletes
([]
string
{
"63"
},
"policy_"
)
task
.
TaskProps
.
Src
=
"TestImportTask_Do/empty"
mock
.
ExpectQuery
(
"SELECT(.+)policies(.+)"
)
.
WillReturnRows
(
sqlmock
.
NewRows
([]
string
{
"id"
,
"type"
})
.
AddRow
(
63
,
"local"
))
// 设定listing状态
mock
.
ExpectBegin
()
mock
.
ExpectExec
(
"UPDATE(.+)"
)
.
WillReturnResult
(
sqlmock
.
NewResult
(
1
,
1
))
mock
.
ExpectCommit
()
// 设定inserting状态
mock
.
ExpectBegin
()
mock
.
ExpectExec
(
"UPDATE(.+)"
)
.
WillReturnResult
(
sqlmock
.
NewResult
(
1
,
1
))
mock
.
ExpectCommit
()
task
.
Do
()
asserts
.
NoError
(
mock
.
ExpectationsWereMet
())
asserts
.
Nil
(
task
.
Err
)
task
.
Err
=
nil
}
// 创建测试文件
f
,
_
:=
util
.
CreatNestedFile
(
util
.
RelativePath
(
"tests/TestImportTask_Do/test.txt"
))
f
.
Close
()
// 成功列取,包含一个文件一个目录,父目录创建失败
{
cache
.
Deletes
([]
string
{
"63"
},
"policy_"
)
task
.
TaskProps
.
Src
=
"tests"
task
.
TaskProps
.
Dst
=
"/"
task
.
TaskProps
.
Recursive
=
true
mock
.
ExpectQuery
(
"SELECT(.+)policies(.+)"
)
.
WillReturnRows
(
sqlmock
.
NewRows
([]
string
{
"id"
,
"type"
})
.
AddRow
(
63
,
"local"
))
// 设定listing状态
mock
.
ExpectBegin
()
mock
.
ExpectExec
(
"UPDATE(.+)"
)
.
WillReturnResult
(
sqlmock
.
NewResult
(
1
,
1
))
mock
.
ExpectCommit
()
// 设定inserting状态
mock
.
ExpectBegin
()
mock
.
ExpectExec
(
"UPDATE(.+)"
)
.
WillReturnResult
(
sqlmock
.
NewResult
(
1
,
1
))
mock
.
ExpectCommit
()
// 查找父目录,但是不存在
mock
.
ExpectQuery
(
"SELECT(.+)folders"
)
.
WillReturnRows
(
sqlmock
.
NewRows
([]
string
{
"id"
}))
// 创建文件时查找父目录,仍然不存在
mock
.
ExpectQuery
(
"SELECT(.+)folders"
)
.
WillReturnRows
(
sqlmock
.
NewRows
([]
string
{
"id"
}))
task
.
Do
()
asserts
.
NoError
(
mock
.
ExpectationsWereMet
())
asserts
.
Nil
(
task
.
Err
)
task
.
Err
=
nil
}
// 成功列取,包含一个文件一个目录, 全部操作成功
{
cache
.
Deletes
([]
string
{
"63"
},
"policy_"
)
task
.
TaskProps
.
Src
=
"tests"
task
.
TaskProps
.
Dst
=
"/"
task
.
TaskProps
.
Recursive
=
true
mock
.
ExpectQuery
(
"SELECT(.+)policies(.+)"
)
.
WillReturnRows
(
sqlmock
.
NewRows
([]
string
{
"id"
,
"type"
})
.
AddRow
(
63
,
"local"
))
// 设定listing状态
mock
.
ExpectBegin
()
mock
.
ExpectExec
(
"UPDATE(.+)"
)
.
WillReturnResult
(
sqlmock
.
NewResult
(
1
,
1
))
mock
.
ExpectCommit
()
// 设定inserting状态
mock
.
ExpectBegin
()
mock
.
ExpectExec
(
"UPDATE(.+)"
)
.
WillReturnResult
(
sqlmock
.
NewResult
(
1
,
1
))
mock
.
ExpectCommit
()
// 查找父目录,存在
mock
.
ExpectQuery
(
"SELECT(.+)folders"
)
.
WillReturnRows
(
sqlmock
.
NewRows
([]
string
{
"id"
})
.
AddRow
(
1
))
// 查找同名文件,不存在
mock
.
ExpectQuery
(
"SELECT(.+)files"
)
.
WillReturnRows
(
sqlmock
.
NewRows
([]
string
{
"id"
}))
// 创建目录
mock
.
ExpectBegin
()
mock
.
ExpectExec
(
"INSERT(.+)folders(.+)"
)
.
WillReturnResult
(
sqlmock
.
NewResult
(
2
,
1
))
mock
.
ExpectCommit
()
// 插入文件记录
mock
.
ExpectBegin
()
mock
.
ExpectExec
(
"INSERT(.+)files(.+)"
)
.
WillReturnResult
(
sqlmock
.
NewResult
(
2
,
1
))
mock
.
ExpectCommit
()
task
.
Do
()
asserts
.
NoError
(
mock
.
ExpectationsWereMet
())
asserts
.
Nil
(
task
.
Err
)
task
.
Err
=
nil
}
}
func
TestNewImportTask
(
t
*
testing
.
T
)
{
asserts
:=
assert
.
New
(
t
)
// 成功
{
mock
.
ExpectQuery
(
"SELECT(.+)"
)
.
WillReturnRows
(
sqlmock
.
NewRows
([]
string
{
"id"
})
.
AddRow
(
1
))
mock
.
ExpectBegin
()
mock
.
ExpectExec
(
"INSERT(.+)"
)
.
WillReturnResult
(
sqlmock
.
NewResult
(
1
,
1
))
mock
.
ExpectCommit
()
job
,
err
:=
NewImportTask
(
1
,
1
,
"/"
,
"/"
,
false
)
asserts
.
NoError
(
mock
.
ExpectationsWereMet
())
asserts
.
NotNil
(
job
)
asserts
.
NoError
(
err
)
}
// 失败
{
mock
.
ExpectQuery
(
"SELECT(.+)"
)
.
WillReturnRows
(
sqlmock
.
NewRows
([]
string
{
"id"
})
.
AddRow
(
1
))
mock
.
ExpectBegin
()
mock
.
ExpectExec
(
"INSERT(.+)"
)
.
WillReturnError
(
errors
.
New
(
"error"
))
mock
.
ExpectRollback
()
job
,
err
:=
NewImportTask
(
1
,
1
,
"/"
,
"/"
,
false
)
asserts
.
NoError
(
mock
.
ExpectationsWereMet
())
asserts
.
Nil
(
job
)
asserts
.
Error
(
err
)
}
}
func
TestNewImportTaskFromModel
(
t
*
testing
.
T
)
{
asserts
:=
assert
.
New
(
t
)
// 成功
{
mock
.
ExpectQuery
(
"SELECT(.+)"
)
.
WillReturnRows
(
sqlmock
.
NewRows
([]
string
{
"id"
})
.
AddRow
(
1
))
job
,
err
:=
NewImportTaskFromModel
(
&
model
.
Task
{
Props
:
"{}"
})
asserts
.
NoError
(
mock
.
ExpectationsWereMet
())
asserts
.
NoError
(
err
)
asserts
.
NotNil
(
job
)
}
// JSON解析失败
{
mock
.
ExpectQuery
(
"SELECT(.+)"
)
.
WillReturnRows
(
sqlmock
.
NewRows
([]
string
{
"id"
})
.
AddRow
(
1
))
job
,
err
:=
NewImportTaskFromModel
(
&
model
.
Task
{
Props
:
"?"
})
asserts
.
NoError
(
mock
.
ExpectationsWereMet
())
asserts
.
Error
(
err
)
asserts
.
Nil
(
job
)
}
}
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录