Infrastructure/Cloudreve
0
Fork 0
Code Issues Pull requests Activity

test: delete file while user not found

Browse source
This commit is contained in:
Aaron Liu 2023-02-07 20:18:13 +08:00
parent d67d0512f8
commit 489a2bab4f
1 changed files with 19 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
models/file_test.go
View file

@ -365,7 +365,7 @@ func TestDeleteFiles(t *testing.T) {
// uid 不一致 // uid 不一致
{ {
err := DeleteFiles([]*File{{}}, 1) err := DeleteFiles([]*File{{UserID: 2}}, 1)
a.Contains("user id not consistent", err.Error()) a.Contains("user id not consistent", err.Error())
} }
@ -375,7 +375,7 @@ func TestDeleteFiles(t *testing.T) {
mock.ExpectExec("DELETE(.+)"). mock.ExpectExec("DELETE(.+)").
WillReturnError(errors.New("error")) WillReturnError(errors.New("error"))
mock.ExpectRollback() mock.ExpectRollback()
err := DeleteFiles([]*File{{}}, 0) err := DeleteFiles([]*File{{UserID: 1}}, 1)
a.NoError(mock.ExpectationsWereMet()) a.NoError(mock.ExpectationsWereMet())
a.Error(err) a.Error(err)
} }
@ -387,7 +387,7 @@ func TestDeleteFiles(t *testing.T) {
WillReturnResult(sqlmock.NewResult(1, 1)) WillReturnResult(sqlmock.NewResult(1, 1))
mock.ExpectExec("UPDATE(.+)storage(.+)").WillReturnError(errors.New("error")) mock.ExpectExec("UPDATE(.+)storage(.+)").WillReturnError(errors.New("error"))
mock.ExpectRollback() mock.ExpectRollback()
err := DeleteFiles([]*File{{}}, 0) err := DeleteFiles([]*File{{UserID: 1}}, 1)
a.NoError(mock.ExpectationsWereMet()) a.NoError(mock.ExpectationsWereMet())
a.Error(err) a.Error(err)
} }
@ -398,7 +398,7 @@ func TestDeleteFiles(t *testing.T) {
mock.ExpectExec("DELETE(.+)"). mock.ExpectExec("DELETE(.+)").
WillReturnResult(sqlmock.NewResult(1, 0)) WillReturnResult(sqlmock.NewResult(1, 0))
mock.ExpectRollback() mock.ExpectRollback()
err := DeleteFiles([]*File{{Size: 1}, {Size: 2}}, 0) err := DeleteFiles([]*File{{Size: 1, UserID: 1}, {Size: 2, UserID: 1}}, 1)
a.NoError(mock.ExpectationsWereMet()) a.NoError(mock.ExpectationsWereMet())
a.Error(err) a.Error(err)
a.Contains("file size is dirty", err.Error()) a.Contains("file size is dirty", err.Error())
@ -411,9 +411,22 @@ func TestDeleteFiles(t *testing.T) {
WillReturnResult(sqlmock.NewResult(2, 1)) WillReturnResult(sqlmock.NewResult(2, 1))
mock.ExpectExec("DELETE(.+)"). mock.ExpectExec("DELETE(.+)").
WillReturnResult(sqlmock.NewResult(2, 1)) WillReturnResult(sqlmock.NewResult(2, 1))
mock.ExpectExec("UPDATE(.+)storage(.+)").WithArgs(uint64(3), sqlmock.AnyArg()).WillReturnResult(sqlmock.NewResult(1, 1)) mock.ExpectExec("UPDATE(.+)storage(.+)").WithArgs(uint64(3), sqlmock.AnyArg(), uint(1)).WillReturnResult(sqlmock.NewResult(1, 1))
mock.ExpectCommit() mock.ExpectCommit()
err := DeleteFiles([]*File{{Size: 1}, {Size: 2}}, 0) err := DeleteFiles([]*File{{Size: 1, UserID: 1}, {Size: 2, UserID: 1}}, 1)
a.NoError(mock.ExpectationsWereMet())
a.NoError(err)
}
// 成功, 关联用户不存在
{
mock.ExpectBegin()
mock.ExpectExec("DELETE(.+)").
WillReturnResult(sqlmock.NewResult(2, 1))
mock.ExpectExec("DELETE(.+)").
WillReturnResult(sqlmock.NewResult(2, 1))
mock.ExpectCommit()
err := DeleteFiles([]*File{{Size: 1, UserID: 1}, {Size: 2, UserID: 1}}, 0)
a.NoError(mock.ExpectationsWereMet()) a.NoError(mock.ExpectationsWereMet())
a.NoError(err) a.NoError(err)
} }