wsy182/Cloudreve
1
0
Fork 0
mirror of https://github.com/halejohn/Cloudreve.git synced 2026-01-26 09:34:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

test(hook): NewWebdavAfterUploadHook

Browse Source
This commit is contained in:
Aaron Liu
2023-04-15 09:21:29 +08:00
parent c5ffdbfcfb
commit 4d131db504
2 changed files with 24 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
pkg/filesystem/hooks_test.go
View File

@@ -683,3 +683,26 @@ func TestHookDeleteUploadSession(t *testing.T) {
_, ok := cache.Get(UploadSessionCachePrefix + "TestHookDeleteUploadSession")
a.False(ok)
}
func TestNewWebdavAfterUploadHook(t *testing.T) {
a := assert.New(t)
fs := &FileSystem{}
file := &fsctx.FileStream{
Model: &model.File{
Model: gorm.Model{ID: 1},
},
}
req, _ := http.NewRequest("get", "http://localhost", nil)
req.Header.Add("X-Oc-Mtime", "1681521402")
req.Header.Add("OC-Checksum", "checksum")
mock.ExpectBegin()
mock.ExpectExec("UPDATE(.+)files(.+)").WillReturnResult(sqlmock.NewResult(1, 1))
mock.ExpectCommit()
mock.ExpectBegin()
mock.ExpectExec("UPDATE(.+)files(.+)").WillReturnResult(sqlmock.NewResult(1, 1))
mock.ExpectCommit()
err := NewWebdavAfterUploadHook(req)(context.Background(), fs, file)
a.NoError(err)
a.NoError(mock.ExpectationsWereMet())
}