You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
81 lines
2.0 KiB
81 lines
2.0 KiB
3 years ago
|
package middleware
|
||
|
|
|
||
|
|
import (
|
||
|
|
model "github.com/cloudreve/Cloudreve/v3/models"
|
||
|
|
"github.com/cloudreve/Cloudreve/v3/pkg/auth"
|
||
|
|
"github.com/cloudreve/Cloudreve/v3/pkg/cluster"
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
"github.com/jinzhu/gorm"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"net/http/httptest"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestMasterMetadata(t *testing.T) {
|
||
|
|
a := assert.New(t)
|
||
|
|
masterMetaDataFunc := MasterMetadata()
|
||
|
|
rec := httptest.NewRecorder()
|
||
|
|
c, _ := gin.CreateTestContext(rec)
|
||
|
|
c.Request = httptest.NewRequest("GET", "/", nil)
|
||
|
|
|
||
|
|
c.Request.Header = map[string][]string{
|
||
|
|
"X-Site-Id": {"expectedSiteID"},
|
||
|
|
"X-Site-Url": {"expectedSiteURL"},
|
||
|
|
"X-Cloudreve-Version": {"expectedMasterVersion"},
|
||
|
|
}
|
||
|
|
masterMetaDataFunc(c)
|
||
|
|
siteID, _ := c.Get("MasterSiteID")
|
||
|
|
siteURL, _ := c.Get("MasterSiteURL")
|
||
|
|
siteVersion, _ := c.Get("MasterVersion")
|
||
|
|
|
||
|
|
a.Equal("expectedSiteID", siteID.(string))
|
||
|
|
a.Equal("expectedSiteURL", siteURL.(string))
|
||
|
|
a.Equal("expectedMasterVersion", siteVersion.(string))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestSlaveRPCSignRequired(t *testing.T) {
|
||
|
|
a := assert.New(t)
|
||
|
|
np := &cluster.NodePool{}
|
||
|
|
np.Init()
|
||
|
|
slaveRPCSignRequiredFunc := SlaveRPCSignRequired(np)
|
||
|
|
rec := httptest.NewRecorder()
|
||
|
|
|
||
|
|
// id parse failed
|
||
|
|
{
|
||
|
|
c, _ := gin.CreateTestContext(rec)
|
||
|
|
c.Request = httptest.NewRequest("GET", "/", nil)
|
||
|
|
c.Request.Header.Set("X-Node-Id", "unknown")
|
||
|
|
slaveRPCSignRequiredFunc(c)
|
||
|
|
a.True(c.IsAborted())
|
||
|
|
}
|
||
|
|
|
||
|
|
// node id not exist
|
||
|
|
{
|
||
|
|
c, _ := gin.CreateTestContext(rec)
|
||
|
|
c.Request = httptest.NewRequest("GET", "/", nil)
|
||
|
|
c.Request.Header.Set("X-Node-Id", "38")
|
||
|
|
slaveRPCSignRequiredFunc(c)
|
||
|
|
a.True(c.IsAborted())
|
||
|
|
}
|
||
|
|
|
||
|
|
// success
|
||
|
|
{
|
||
|
|
authInstance := auth.HMACAuth{SecretKey: []byte("")}
|
||
|
|
np.Add(&model.Node{Model: gorm.Model{
|
||
|
|
ID: 38,
|
||
|
|
}})
|
||
|
|
|
||
|
|
c, _ := gin.CreateTestContext(rec)
|
||
|
|
c.Request = httptest.NewRequest("POST", "/", nil)
|
||
|
|
c.Request.Header.Set("X-Node-Id", "38")
|
||
|
|
c.Request = auth.SignRequest(authInstance, c.Request, 0)
|
||
|
|
slaveRPCSignRequiredFunc(c)
|
||
|
|
a.False(c.IsAborted())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestUseSlaveAria2Instance(t *testing.T) {
|
||
|
|
a := assert.New(t)
|
||
|
|
|
||
|
|
}
|