diff --git a/pkg/filesystem/local/handller_test.go b/pkg/filesystem/local/handller_test.go index b5a0ab1..71da4ca 100644 --- a/pkg/filesystem/local/handller_test.go +++ b/pkg/filesystem/local/handller_test.go @@ -2,6 +2,7 @@ package local import ( "context" + "github.com/HFO4/cloudreve/pkg/conf" "github.com/HFO4/cloudreve/pkg/util" "github.com/stretchr/testify/assert" "io" @@ -89,3 +90,25 @@ func TestHandler_Get(t *testing.T) { asserts.Error(err) asserts.Nil(rs) } + +func TestHandler_Thumb(t *testing.T) { + asserts := assert.New(t) + handler := Handler{} + ctx := context.Background() + file, err := os.Create("TestHandler_Thumb" + conf.ThumbConfig.FileSuffix) + asserts.NoError(err) + file.Close() + + // 正常 + { + thumb, err := handler.Thumb(ctx, "TestHandler_Thumb") + asserts.NoError(err) + asserts.NotNil(thumb.Content) + } + + // 不存在 + { + _, err := handler.Thumb(ctx, "not_exist") + asserts.Error(err) + } +}