Test: aria2 related

pkg/aria2/aria2_test.go

@@ -0,0 +1,89 @@
+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) {
+	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()
+		asserts.IsType(&DummyAria2{}, Instance)
+	}
+
+	// 无法解析服务器地址
+	{
+		cache.Set("setting_aria2_rpcurl", string(byte(0x7f)), 0)
+		Init()
+		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()
+		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()
+		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("?"))
+}