MiaoStars/cloudreve
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Test: tasks pkg

Browse Source
This commit is contained in:
HFO4
2021-11-20 17:14:45 +08:00
parent 9056ef9171
commit 96b84bb5e5
7 changed files with 44 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pkg/task/transfer_test.go
View File

@@ -138,7 +138,7 @@ func TestNewTransferTask(t *testing.T) {
mock.ExpectBegin()
mock.ExpectExec("INSERT(.+)").WillReturnResult(sqlmock.NewResult(1, 1))
mock.ExpectCommit()
job, err := NewTransferTask(1, []string{}, "/", "/", false)
job, err := NewTransferTask(1, []string{}, "/", "/", false, 0, nil)
asserts.NoError(mock.ExpectationsWereMet())
asserts.NotNil(job)
asserts.NoError(err)
@@ -150,7 +150,7 @@ func TestNewTransferTask(t *testing.T) {
mock.ExpectBegin()
mock.ExpectExec("INSERT(.+)").WillReturnError(errors.New("error"))
mock.ExpectRollback()
job, err := NewTransferTask(1, []string{}, "/", "/", false)
job, err := NewTransferTask(1, []string{}, "/", "/", false, 0, nil)
asserts.NoError(mock.ExpectationsWereMet())
asserts.Nil(job)
asserts.Error(err)