Test: Share related middleware

middleware/share_test.go

@@ -0,0 +1,145 @@
+package middleware
+
+import (
+	"github.com/DATA-DOG/go-sqlmock"
+	model "github.com/HFO4/cloudreve/models"
+	"github.com/HFO4/cloudreve/pkg/conf"
+	"github.com/gin-gonic/gin"
+	"github.com/jinzhu/gorm"
+	"github.com/stretchr/testify/assert"
+	"net/http/httptest"
+	"testing"
+)
+
+func TestShareAvailable(t *testing.T) {
+	asserts := assert.New(t)
+	rec := httptest.NewRecorder()
+	testFunc := ShareAvailable()
+
+	// 分享不存在
+	{
+		c, _ := gin.CreateTestContext(rec)
+		c.Params = []gin.Param{
+			{"id", "empty"},
+		}
+		testFunc(c)
+		asserts.True(c.IsAborted())
+	}
+
+	// 通过
+	{
+		conf.SystemConfig.HashIDSalt = ""
+		// 用户组
+		mock.ExpectQuery("SELECT(.+)groups(.+)").WillReturnRows(sqlmock.NewRows([]string{"id"}).AddRow(3))
+		mock.ExpectQuery("SELECT(.+)shares(.+)").
+			WillReturnRows(
+				sqlmock.NewRows(
+					[]string{"id", "remain_downloads", "source_id"}).
+					AddRow(1, 1, 2),
+			)
+		mock.ExpectQuery("SELECT(.+)files(.+)").WillReturnRows(sqlmock.NewRows([]string{"id"}).AddRow(2))
+		c, _ := gin.CreateTestContext(rec)
+		c.Params = []gin.Param{
+			{"id", "x9T4"},
+		}
+		testFunc(c)
+		asserts.NoError(mock.ExpectationsWereMet())
+		asserts.False(c.IsAborted())
+		asserts.NotNil(c.Get("user"))
+		asserts.NotNil(c.Get("share"))
+	}
+}
+
+func TestShareCanPreview(t *testing.T) {
+	asserts := assert.New(t)
+	rec := httptest.NewRecorder()
+	testFunc := ShareCanPreview()
+
+	// 无分享上下文
+	{
+		c, _ := gin.CreateTestContext(rec)
+		testFunc(c)
+		asserts.True(c.IsAborted())
+	}
+
+	// 可以预览
+	{
+		c, _ := gin.CreateTestContext(rec)
+		c.Set("share", &model.Share{PreviewEnabled: true})
+		testFunc(c)
+		asserts.False(c.IsAborted())
+	}
+
+	// 未开启预览
+	{
+		c, _ := gin.CreateTestContext(rec)
+		c.Set("share", &model.Share{PreviewEnabled: false})
+		testFunc(c)
+		asserts.True(c.IsAborted())
+	}
+}
+
+func TestCheckShareUnlocked(t *testing.T) {
+	asserts := assert.New(t)
+	rec := httptest.NewRecorder()
+	testFunc := CheckShareUnlocked()
+
+	// 无分享上下文
+	{
+		c, _ := gin.CreateTestContext(rec)
+		testFunc(c)
+		asserts.True(c.IsAborted())
+	}
+
+	// 无密码
+	{
+		c, _ := gin.CreateTestContext(rec)
+		c.Set("share", &model.Share{})
+		testFunc(c)
+		asserts.False(c.IsAborted())
+	}
+
+}
+
+func TestBeforeShareDownload(t *testing.T) {
+	asserts := assert.New(t)
+	rec := httptest.NewRecorder()
+	testFunc := BeforeShareDownload()
+
+	// 无分享上下文
+	{
+		c, _ := gin.CreateTestContext(rec)
+		testFunc(c)
+		asserts.True(c.IsAborted())
+
+		c, _ = gin.CreateTestContext(rec)
+		c.Set("share", &model.Share{})
+		testFunc(c)
+		asserts.True(c.IsAborted())
+	}
+
+	// 用户不能下载
+	{
+		c, _ := gin.CreateTestContext(rec)
+		c.Set("share", &model.Share{})
+		c.Set("user", &model.User{
+			Group: model.Group{OptionsSerialized: model.GroupOption{}},
+		})
+		testFunc(c)
+		asserts.True(c.IsAborted())
+	}
+
+	// 可以下载
+	{
+		c, _ := gin.CreateTestContext(rec)
+		c.Set("share", &model.Share{})
+		c.Set("user", &model.User{
+			Model: gorm.Model{ID: 1},
+			Group: model.Group{OptionsSerialized: model.GroupOption{
+				ShareDownloadEnabled: true,
+			}},
+		})
+		testFunc(c)
+		asserts.False(c.IsAborted())
+	}
+}