Test: aria2 related

pkg/aria2/caller_test.go

@@ -0,0 +1,51 @@
+package aria2
+
+import (
+	model "github.com/HFO4/cloudreve/models"
+	"github.com/HFO4/cloudreve/pkg/cache"
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+func TestRPCService_Init(t *testing.T) {
+	asserts := assert.New(t)
+	caller := &RPCService{}
+	asserts.Error(caller.Init("ws://", "", 1, nil))
+	asserts.NoError(caller.Init("http://127.0.0.1", "", 1, nil))
+}
+
+func TestRPCService_Status(t *testing.T) {
+	asserts := assert.New(t)
+	caller := &RPCService{}
+	asserts.NoError(caller.Init("http://127.0.0.1", "", 1, nil))
+
+	_, err := caller.Status(&model.Download{})
+	asserts.Error(err)
+}
+
+func TestRPCService_Cancel(t *testing.T) {
+	asserts := assert.New(t)
+	caller := &RPCService{}
+	asserts.NoError(caller.Init("http://127.0.0.1", "", 1, nil))
+
+	err := caller.Cancel(&model.Download{Parent: "test"})
+	asserts.Error(err)
+}
+
+func TestRPCService_Select(t *testing.T) {
+	asserts := assert.New(t)
+	caller := &RPCService{}
+	asserts.NoError(caller.Init("http://127.0.0.1", "", 1, nil))
+
+	err := caller.Select(&model.Download{Parent: "test"}, []int{1, 2, 3})
+	asserts.Error(err)
+}
+
+func TestRPCService_CreateTask(t *testing.T) {
+	asserts := assert.New(t)
+	caller := &RPCService{}
+	asserts.NoError(caller.Init("http://127.0.0.1", "", 1, nil))
+	cache.Set("setting_aria2_temp_path", "test", 0)
+	err := caller.CreateTask(&model.Download{Parent: "test"}, []interface{}{map[string]string{"1": "1"}})
+	asserts.Error(err)
+}