From 4d131db504059b975d7d24a96a14d0f64d52f059 Mon Sep 17 00:00:00 2001 From: Aaron Liu <912394456@qq.com> Date: Sat, 15 Apr 2023 09:21:29 +0800 Subject: [PATCH] test(hook): NewWebdavAfterUploadHook --- assets | 2 +- pkg/filesystem/hooks_test.go | 23 +++++++++++++++++++++++ 2 files changed, 24 insertions(+), 1 deletion(-) diff --git a/assets b/assets index ca831db..88e4b7f 160000 --- a/assets +++ b/assets @@ -1 +1 @@ -Subproject commit ca831dbe194a6508f67bcf646a7b47c087043546 +Subproject commit 88e4b7fbf3d5e5806ad0cedce99d845ef21704c7 diff --git a/pkg/filesystem/hooks_test.go b/pkg/filesystem/hooks_test.go index 6f03f4d..cc660ce 100644 --- a/pkg/filesystem/hooks_test.go +++ b/pkg/filesystem/hooks_test.go @@ -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()) + +}