056de22edb
* Feat: retrieve nodes from data table * Feat: master node ping slave node in REST API * Feat: master send scheduled ping request * Feat: inactive nodes recover loop * Modify: remove database operations from aria2 RPC caller implementation * Feat: init aria2 client in master node * Feat: Round Robin load balancer * Feat: create and monitor aria2 task in master node * Feat: salve receive and handle heartbeat * Fix: Node ID will be 0 in download record generated in older version * Feat: sign request headers with all `X-` prefix * Feat: API call to slave node will carry meta data in headers * Feat: call slave aria2 rpc method from master * Feat: get slave aria2 task status Feat: encode slave response data using gob * Feat: aria2 callback to master node / cancel or select task to slave node * Fix: use dummy aria2 client when caller initialize failed in master node * Feat: slave aria2 status event callback / salve RPC auth * Feat: prototype for slave driven filesystem * Feat: retry for init aria2 client in master node * Feat: init request client with global options * Feat: slave receive async task from master * Fix: competition write in request header * Refactor: dependency initialize order * Feat: generic message queue implementation * Feat: message queue implementation * Feat: master waiting slave transfer result * Feat: slave transfer file in stateless policy * Feat: slave transfer file in slave policy * Feat: slave transfer file in local policy * Feat: slave transfer file in OneDrive policy * Fix: failed to initialize update checker http client * Feat: list slave nodes for dashboard * Feat: test aria2 rpc connection in slave * Feat: add and save node * Feat: add and delete node in node pool * Fix: temp file cannot be removed when aria2 task fails * Fix: delete node in admin panel * Feat: edit node and get node info * Modify: delete unused settings
92 lines
2.3 KiB
Go
92 lines
2.3 KiB
Go
package aria2
|
||
|
||
import (
|
||
"database/sql"
|
||
"testing"
|
||
|
||
"github.com/DATA-DOG/go-sqlmock"
|
||
model "github.com/cloudreve/Cloudreve/v3/models"
|
||
"github.com/cloudreve/Cloudreve/v3/pkg/aria2/monitor"
|
||
"github.com/cloudreve/Cloudreve/v3/pkg/cache"
|
||
"github.com/jinzhu/gorm"
|
||
"github.com/stretchr/testify/assert"
|
||
)
|
||
|
||
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 TestDummyAria2(t *testing.T) {
|
||
asserts := assert.New(t)
|
||
instance := DummyAria2{}
|
||
asserts.Error(instance.CreateTask(nil, nil))
|
||
_, err := instance.Status(nil)
|
||
asserts.Error(err)
|
||
asserts.Error(instance.Cancel(nil))
|
||
asserts.Error(instance.Select(nil, nil))
|
||
}
|
||
|
||
func TestInit(t *testing.T) {
|
||
monitor.MAX_RETRY = 0
|
||
asserts := assert.New(t)
|
||
cache.Set("setting_aria2_token", "1", 0)
|
||
cache.Set("setting_aria2_call_timeout", "5", 0)
|
||
cache.Set("setting_aria2_options", `[]`, 0)
|
||
|
||
// 未指定RPC地址，跳过
|
||
{
|
||
cache.Set("setting_aria2_rpcurl", "", 0)
|
||
Init(false)
|
||
asserts.IsType(&DummyAria2{}, Instance)
|
||
}
|
||
|
||
// 无法解析服务器地址
|
||
{
|
||
cache.Set("setting_aria2_rpcurl", string(byte(0x7f)), 0)
|
||
Init(false)
|
||
asserts.IsType(&DummyAria2{}, Instance)
|
||
}
|
||
|
||
// 无法解析全局配置
|
||
{
|
||
Instance = &RPCService{}
|
||
cache.Set("setting_aria2_options", "?", 0)
|
||
cache.Set("setting_aria2_rpcurl", "ws://127.0.0.1:1234", 0)
|
||
Init(false)
|
||
asserts.IsType(&DummyAria2{}, Instance)
|
||
}
|
||
|
||
// 连接失败
|
||
{
|
||
cache.Set("setting_aria2_options", "{}", 0)
|
||
cache.Set("setting_aria2_rpcurl", "http://127.0.0.1:1234", 0)
|
||
cache.Set("setting_aria2_call_timeout", "1", 0)
|
||
cache.Set("setting_aria2_interval", "100", 0)
|
||
mock.ExpectQuery("SELECT(.+)").WillReturnRows(sqlmock.NewRows([]string{"g_id"}).AddRow("1"))
|
||
Init(false)
|
||
asserts.NoError(mock.ExpectationsWereMet())
|
||
asserts.IsType(&RPCService{}, Instance)
|
||
}
|
||
}
|
||
|
||
func TestGetStatus(t *testing.T) {
|
||
asserts := assert.New(t)
|
||
asserts.Equal(4, GetStatus("complete"))
|
||
asserts.Equal(1, GetStatus("active"))
|
||
asserts.Equal(0, GetStatus("waiting"))
|
||
asserts.Equal(2, GetStatus("paused"))
|
||
asserts.Equal(3, GetStatus("error"))
|
||
asserts.Equal(5, GetStatus("removed"))
|
||
asserts.Equal(6, GetStatus("?"))
|
||
}
|