parent
416f4c1dd2
commit
6c9967b120
@ -0,0 +1,186 @@
|
|||||||
|
package cluster
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
model "github.com/cloudreve/Cloudreve/v3/models"
|
||||||
|
"github.com/cloudreve/Cloudreve/v3/pkg/aria2/rpc"
|
||||||
|
"github.com/cloudreve/Cloudreve/v3/pkg/serializer"
|
||||||
|
"github.com/cloudreve/Cloudreve/v3/pkg/util"
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
"os"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestMasterNode_Init(t *testing.T) {
|
||||||
|
a := assert.New(t)
|
||||||
|
m := &MasterNode{}
|
||||||
|
m.Init(&model.Node{Status: model.NodeSuspend})
|
||||||
|
a.Equal(model.NodeSuspend, m.DBModel().Status)
|
||||||
|
m.Init(&model.Node{Aria2Enabled: true})
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMasterNode_DummyMethods(t *testing.T) {
|
||||||
|
a := assert.New(t)
|
||||||
|
m := &MasterNode{
|
||||||
|
Model: &model.Node{},
|
||||||
|
}
|
||||||
|
|
||||||
|
m.Model.ID = 5
|
||||||
|
a.Equal(m.Model.ID, m.ID())
|
||||||
|
|
||||||
|
res, err := m.Ping(&serializer.NodePingReq{})
|
||||||
|
a.NoError(err)
|
||||||
|
a.NotNil(res)
|
||||||
|
|
||||||
|
a.True(m.IsActive())
|
||||||
|
a.True(m.IsMater())
|
||||||
|
|
||||||
|
m.SubscribeStatusChange(func(isActive bool, id uint) {})
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMasterNode_IsFeatureEnabled(t *testing.T) {
|
||||||
|
a := assert.New(t)
|
||||||
|
m := &MasterNode{
|
||||||
|
Model: &model.Node{},
|
||||||
|
}
|
||||||
|
|
||||||
|
a.False(m.IsFeatureEnabled("aria2"))
|
||||||
|
a.False(m.IsFeatureEnabled("random"))
|
||||||
|
m.Model.Aria2Enabled = true
|
||||||
|
a.True(m.IsFeatureEnabled("aria2"))
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMasterNode_AuthInstance(t *testing.T) {
|
||||||
|
a := assert.New(t)
|
||||||
|
m := &MasterNode{
|
||||||
|
Model: &model.Node{},
|
||||||
|
}
|
||||||
|
|
||||||
|
a.NotNil(m.MasterAuthInstance())
|
||||||
|
a.NotNil(m.SlaveAuthInstance())
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMasterNode_Kill(t *testing.T) {
|
||||||
|
m := &MasterNode{
|
||||||
|
Model: &model.Node{},
|
||||||
|
}
|
||||||
|
|
||||||
|
m.Kill()
|
||||||
|
|
||||||
|
caller, _ := rpc.New(context.Background(), "http://", "", 0, nil)
|
||||||
|
m.aria2RPC.Caller = caller
|
||||||
|
m.Kill()
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMasterNode_GetAria2Instance(t *testing.T) {
|
||||||
|
a := assert.New(t)
|
||||||
|
m := &MasterNode{
|
||||||
|
Model: &model.Node{},
|
||||||
|
aria2RPC: rpcService{},
|
||||||
|
}
|
||||||
|
|
||||||
|
m.aria2RPC.parent = m
|
||||||
|
|
||||||
|
a.NotNil(m.GetAria2Instance())
|
||||||
|
m.Model.Aria2Enabled = true
|
||||||
|
a.NotNil(m.GetAria2Instance())
|
||||||
|
m.aria2RPC.Initialized = true
|
||||||
|
a.NotNil(m.GetAria2Instance())
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestRpcService_Init(t *testing.T) {
|
||||||
|
a := assert.New(t)
|
||||||
|
m := &MasterNode{
|
||||||
|
Model: &model.Node{
|
||||||
|
Aria2OptionsSerialized: model.Aria2Option{
|
||||||
|
Options: "{",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
aria2RPC: rpcService{},
|
||||||
|
}
|
||||||
|
m.aria2RPC.parent = m
|
||||||
|
|
||||||
|
// failed to decode address
|
||||||
|
{
|
||||||
|
m.Model.Aria2OptionsSerialized.Server = string([]byte{0x7f})
|
||||||
|
a.Error(m.aria2RPC.Init())
|
||||||
|
}
|
||||||
|
|
||||||
|
// failed to decode options
|
||||||
|
{
|
||||||
|
m.Model.Aria2OptionsSerialized.Server = ""
|
||||||
|
a.Error(m.aria2RPC.Init())
|
||||||
|
}
|
||||||
|
|
||||||
|
// failed to initialized
|
||||||
|
{
|
||||||
|
m.Model.Aria2OptionsSerialized.Server = ""
|
||||||
|
m.Model.Aria2OptionsSerialized.Options = "{}"
|
||||||
|
caller, _ := rpc.New(context.Background(), "http://", "", 0, nil)
|
||||||
|
m.aria2RPC.Caller = caller
|
||||||
|
a.Error(m.aria2RPC.Init())
|
||||||
|
a.False(m.aria2RPC.Initialized)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func getTestRPCNode() *MasterNode {
|
||||||
|
m := &MasterNode{
|
||||||
|
Model: &model.Node{
|
||||||
|
Aria2OptionsSerialized: model.Aria2Option{},
|
||||||
|
},
|
||||||
|
aria2RPC: rpcService{
|
||||||
|
options: &clientOptions{
|
||||||
|
Options: map[string]interface{}{"1": "1"},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
m.aria2RPC.parent = m
|
||||||
|
caller, _ := rpc.New(context.Background(), "http://", "", 0, nil)
|
||||||
|
m.aria2RPC.Caller = caller
|
||||||
|
return m
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestRpcService_CreateTask(t *testing.T) {
|
||||||
|
a := assert.New(t)
|
||||||
|
m := getTestRPCNode()
|
||||||
|
|
||||||
|
res, err := m.aria2RPC.CreateTask(&model.Download{}, map[string]interface{}{"1": "1"})
|
||||||
|
a.Error(err)
|
||||||
|
a.Empty(res)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestRpcService_Status(t *testing.T) {
|
||||||
|
a := assert.New(t)
|
||||||
|
m := getTestRPCNode()
|
||||||
|
|
||||||
|
res, err := m.aria2RPC.Status(&model.Download{})
|
||||||
|
a.Error(err)
|
||||||
|
a.Empty(res)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestRpcService_Cancel(t *testing.T) {
|
||||||
|
a := assert.New(t)
|
||||||
|
m := getTestRPCNode()
|
||||||
|
|
||||||
|
a.Error(m.aria2RPC.Cancel(&model.Download{}))
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestRpcService_Select(t *testing.T) {
|
||||||
|
a := assert.New(t)
|
||||||
|
m := getTestRPCNode()
|
||||||
|
|
||||||
|
a.NotNil(m.aria2RPC.GetConfig())
|
||||||
|
a.Error(m.aria2RPC.Select(&model.Download{}, []int{1, 2, 3}))
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestRpcService_DeleteTempFile(t *testing.T) {
|
||||||
|
a := assert.New(t)
|
||||||
|
m := getTestRPCNode()
|
||||||
|
fdName := "TestRpcService_DeleteTempFile"
|
||||||
|
a.NoError(os.Mkdir(fdName, 0644))
|
||||||
|
|
||||||
|
a.NoError(m.aria2RPC.DeleteTempFile(&model.Download{Parent: fdName}))
|
||||||
|
time.Sleep(500 * time.Millisecond)
|
||||||
|
a.False(util.Exists(fdName))
|
||||||
|
}
|
@ -0,0 +1,17 @@
|
|||||||
|
package cluster
|
||||||
|
|
||||||
|
import (
|
||||||
|
model "github.com/cloudreve/Cloudreve/v3/models"
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestNewNodeFromDBModel(t *testing.T) {
|
||||||
|
a := assert.New(t)
|
||||||
|
a.IsType(&SlaveNode{}, NewNodeFromDBModel(&model.Node{
|
||||||
|
Type: model.SlaveNodeType,
|
||||||
|
}))
|
||||||
|
a.IsType(&MasterNode{}, NewNodeFromDBModel(&model.Node{
|
||||||
|
Type: model.MasterNodeType,
|
||||||
|
}))
|
||||||
|
}
|
@ -0,0 +1,64 @@
|
|||||||
|
package cluster
|
||||||
|
|
||||||
|
import (
|
||||||
|
"database/sql"
|
||||||
|
"errors"
|
||||||
|
"github.com/DATA-DOG/go-sqlmock"
|
||||||
|
model "github.com/cloudreve/Cloudreve/v3/models"
|
||||||
|
"github.com/jinzhu/gorm"
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
"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 TestInitFailed(t *testing.T) {
|
||||||
|
a := assert.New(t)
|
||||||
|
mock.ExpectQuery("SELECT(.+)").WillReturnError(errors.New("error"))
|
||||||
|
Init()
|
||||||
|
a.NoError(mock.ExpectationsWereMet())
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestInitSuccess(t *testing.T) {
|
||||||
|
a := assert.New(t)
|
||||||
|
mock.ExpectQuery("SELECT(.+)").WillReturnRows(sqlmock.NewRows([]string{"id", "aria2_enabled", "type"}).AddRow(1, true, model.MasterNodeType))
|
||||||
|
Init()
|
||||||
|
a.NoError(mock.ExpectationsWereMet())
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestNodePool_GetNodeByID(t *testing.T) {
|
||||||
|
a := assert.New(t)
|
||||||
|
p := &NodePool{}
|
||||||
|
p.Init()
|
||||||
|
mockNode := &nodeMock{}
|
||||||
|
|
||||||
|
// inactive
|
||||||
|
{
|
||||||
|
p.inactive[1] = mockNode
|
||||||
|
a.Equal(mockNode, p.GetNodeByID(1))
|
||||||
|
}
|
||||||
|
|
||||||
|
// active
|
||||||
|
{
|
||||||
|
delete(p.inactive, 1)
|
||||||
|
p.active[1] = mockNode
|
||||||
|
a.Equal(mockNode, p.GetNodeByID(1))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestNodePool_NodeStatusChange(t *testing.T) {
|
||||||
|
|
||||||
|
}
|
Loading…
Reference in new issue