|
|
package aria2
|
|
|
|
|
|
import (
|
|
|
"database/sql"
|
|
|
"github.com/DATA-DOG/go-sqlmock"
|
|
|
model "github.com/HFO4/cloudreve/models"
|
|
|
"github.com/HFO4/cloudreve/pkg/cache"
|
|
|
"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 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) {
|
|
|
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("?"))
|
|
|
}
|
