test(wopi): add tests for wopi client

pkg/wopi/discovery.go

@@ -3,7 +3,6 @@ package wopi
 import (
 	"encoding/xml"
 	"fmt"
-	"github.com/cloudreve/Cloudreve/v3/pkg/cache"
 	"github.com/cloudreve/Cloudreve/v3/pkg/util"
 	"net/http"
 	"strings"
@@ -62,7 +61,7 @@ func (c *client) refreshDiscovery() error {
 	c.mu.Lock()
 	defer c.mu.Unlock()
 
-	cached, exist := cache.Get(DiscoverResponseCacheKey)
+	cached, exist := c.cache.Get(DiscoverResponseCacheKey)
 	if exist {
 		cachedDiscovery := cached.(WopiDiscovery)
 		c.discovery = &cachedDiscovery
@@ -70,14 +69,14 @@ func (c *client) refreshDiscovery() error {
 		res, err := c.http.Request("GET", c.config.discoveryEndpoint.String(), nil).
 			CheckHTTPResponse(http.StatusOK).GetResponse()
 		if err != nil {
-			return fmt.Errorf("failed to request discovery endpoint: %s", err)
+			return fmt.Errorf("failed to request discovery endpoint: %w", err)
 		}
 
 		if err := xml.Unmarshal([]byte(res), &c.discovery); err != nil {
-			return fmt.Errorf("failed to parse response discovery endpoint: %s", err)
+			return fmt.Errorf("failed to parse response discovery endpoint: %w", err)
 		}
 
-		if err := cache.Set(DiscoverResponseCacheKey, *c.discovery, DiscoverRefreshDuration); err != nil {
+		if err := c.cache.Set(DiscoverResponseCacheKey, *c.discovery, DiscoverRefreshDuration); err != nil {
 			return err
 		}
 	}

pkg/wopi/wopi.go

@@ -165,9 +165,9 @@ func (c *client) NewSession(user *model.User, file *model.File, action ActonType
 		UserID:      user.ID,
 		Action:      action,
 	}
-	err = cache.Set(SessionCachePrefix+sessionID.String(), *session, ttl)
+	err = c.cache.Set(SessionCachePrefix+sessionID.String(), *session, ttl)
 	if err != nil {
-		return nil, fmt.Errorf("failed to create document session: %s", err)
+		return nil, fmt.Errorf("failed to create document session: %w", err)
 	}
 
 	sessionRes := &Session{

pkg/wopi/wopi_test.go

@@ -0,0 +1,184 @@
+package wopi
+
+import (
+	"database/sql"
+	"errors"
+	"github.com/DATA-DOG/go-sqlmock"
+	model "github.com/cloudreve/Cloudreve/v3/models"
+	"github.com/cloudreve/Cloudreve/v3/pkg/cache"
+	"github.com/cloudreve/Cloudreve/v3/pkg/mocks/cachemock"
+	"github.com/cloudreve/Cloudreve/v3/pkg/mocks/requestmock"
+	"github.com/cloudreve/Cloudreve/v3/pkg/request"
+	"github.com/jinzhu/gorm"
+	"github.com/stretchr/testify/assert"
+	testMock "github.com/stretchr/testify/mock"
+	"net/url"
+	"testing"
+)
+
+var mock sqlmock.Sqlmock
+
+// TestMain 初始化数据库Mock
+func TestMain(m *testing.M) {
+	var db *sql.DB
+	var err error
+	db, mock, err = sqlmock.New()
+	if err != nil {
+		panic("An error was not expected when opening a stub database connection")
+	}
+	model.DB, _ = gorm.Open("mysql", db)
+	defer db.Close()
+	m.Run()
+}
+
+func TestNewSession(t *testing.T) {
+	a := assert.New(t)
+	endpoint, _ := url.Parse("http://localhost:8001/hosting/discovery")
+	client := &client{
+		cache: cache.NewMemoStore(),
+		config: config{
+			discoveryEndpoint: endpoint,
+		},
+	}
+
+	// Discovery failed
+	{
+		expectedErr := errors.New("error")
+		mockHttp := &requestmock.RequestMock{}
+		client.http = mockHttp
+		mockHttp.On(
+			"Request",
+			"GET",
+			endpoint.String(),
+			testMock.Anything,
+			testMock.Anything,
+		).Return(&request.Response{
+			Err: expectedErr,
+		})
+		res, err := client.NewSession(&model.User{}, &model.File{}, ActionPreview)
+		a.Nil(res)
+		a.ErrorIs(err, expectedErr)
+		mockHttp.AssertExpectations(t)
+	}
+
+	// not supported ext
+	{
+		client.discovery = &WopiDiscovery{}
+		client.actions = make(map[string]map[string]Action)
+		res, err := client.NewSession(&model.User{}, &model.File{}, ActionPreview)
+		a.Nil(res)
+		a.ErrorIs(err, ErrActionNotSupported)
+	}
+
+	// preferred action not supported
+	{
+		client.discovery = &WopiDiscovery{}
+		client.actions = map[string]map[string]Action{
+			".doc": {},
+		}
+		res, err := client.NewSession(&model.User{}, &model.File{Name: "1.doc"}, ActionPreview)
+		a.Nil(res)
+		a.ErrorIs(err, ErrActionNotSupported)
+	}
+
+	// src url cannot be parsed
+	{
+		client.discovery = &WopiDiscovery{}
+		client.actions = map[string]map[string]Action{
+			".doc": {
+				string(ActionPreviewFallback): Action{
+					Urlsrc: string([]byte{0x7f}),
+				},
+			},
+		}
+		res, err := client.NewSession(&model.User{}, &model.File{Name: "1.doc"}, ActionEdit)
+		a.Nil(res)
+		a.ErrorContains(err, "invalid control character in URL")
+	}
+
+	// all pass - default placeholder
+	{
+		client.discovery = &WopiDiscovery{}
+		client.actions = map[string]map[string]Action{
+			".doc": {
+				string(ActionPreviewFallback): Action{
+					Urlsrc: "https://doc.com/doc",
+				},
+			},
+		}
+		res, err := client.NewSession(&model.User{}, &model.File{Name: "1.doc"}, ActionEdit)
+		a.NotNil(res)
+		a.NoError(err)
+		resUrl := res.ActionURL.String()
+		a.Contains(resUrl, wopiSrcParamDefault)
+	}
+
+	// all pass - with placeholders
+	{
+		client.discovery = &WopiDiscovery{}
+		client.actions = map[string]map[string]Action{
+			".doc": {
+				string(ActionPreviewFallback): Action{
+					Urlsrc: "https://doc.com/doc?origin=preserved&<dc=DC_LLCC&><notsuported=DISABLE_ASYNC&><src=WOPI_SOURCE&>",
+				},
+			},
+		}
+		res, err := client.NewSession(&model.User{}, &model.File{Name: "1.doc"}, ActionEdit)
+		a.NotNil(res)
+		a.NoError(err)
+		resUrl := res.ActionURL.String()
+		a.Contains(resUrl, "origin=preserved")
+		a.Contains(resUrl, "dc=lng")
+		a.Contains(resUrl, "src=")
+		a.NotContains(resUrl, "notsuported")
+	}
+
+	// cache operation failed
+	{
+		mockCache := &cachemock.CacheClientMock{}
+		expectedErr := errors.New("error")
+		client.cache = mockCache
+		client.discovery = &WopiDiscovery{}
+		client.actions = map[string]map[string]Action{
+			".doc": {
+				string(ActionPreviewFallback): Action{
+					Urlsrc: "https://doc.com/doc",
+				},
+			},
+		}
+		mockCache.On("Set", testMock.Anything, testMock.Anything, testMock.Anything).Return(expectedErr)
+		res, err := client.NewSession(&model.User{}, &model.File{Name: "1.doc"}, ActionEdit)
+		a.Nil(res)
+		a.ErrorIs(err, expectedErr)
+	}
+}
+
+func TestInit(t *testing.T) {
+	a := assert.New(t)
+
+	// not enabled
+	{
+		a.Nil(Default)
+		Default = &client{}
+		Init()
+		a.Nil(Default)
+	}
+
+	// throw error
+	{
+		a.Nil(Default)
+		cache.Set("setting_wopi_enabled", "1", 0)
+		cache.Set("setting_wopi_endpoint", string([]byte{0x7f}), 0)
+		Init()
+		a.Nil(Default)
+	}
+
+	// all pass
+	{
+		a.Nil(Default)
+		cache.Set("setting_wopi_enabled", "1", 0)
+		cache.Set("setting_wopi_endpoint", "", 0)
+		Init()
+		a.NotNil(Default)
+	}
+}