Test: model.Tags / modifications

pkg/serializer/aria2_test.go

@@ -0,0 +1,92 @@
+package serializer
+
+import (
+	model "github.com/HFO4/cloudreve/models"
+	"github.com/HFO4/cloudreve/pkg/cache"
+	"github.com/jinzhu/gorm"
+	"github.com/stretchr/testify/assert"
+	"github.com/zyxar/argo/rpc"
+	"testing"
+)
+
+func TestBuildFinishedListResponse(t *testing.T) {
+	asserts := assert.New(t)
+	tasks := []model.Download{
+		{
+			StatusInfo: rpc.StatusInfo{
+				Files: []rpc.FileInfo{
+					{
+						Path: "/file/name.txt",
+					},
+				},
+			},
+			Task: &model.Task{
+				Model: gorm.Model{},
+				Error: "error",
+			},
+		},
+		{
+			StatusInfo: rpc.StatusInfo{
+				Files: []rpc.FileInfo{
+					{
+						Path: "/file/name1.txt",
+					},
+					{
+						Path: "/file/name2.txt",
+					},
+				},
+			},
+		},
+	}
+	tasks[1].StatusInfo.BitTorrent.Info.Name = "name.txt"
+	res := BuildFinishedListResponse(tasks).Data.([]FinishedListResponse)
+	asserts.Len(res, 2)
+	asserts.Equal("name.txt", res[1].Name)
+	asserts.Equal("name.txt", res[0].Name)
+	asserts.Equal("name.txt", res[0].Files[0].Path)
+	asserts.Equal("name1.txt", res[1].Files[0].Path)
+	asserts.Equal("name2.txt", res[1].Files[1].Path)
+	asserts.EqualValues(0, res[0].TaskStatus)
+	asserts.Equal("error", res[0].TaskError)
+}
+
+func TestBuildDownloadingResponse(t *testing.T) {
+	asserts := assert.New(t)
+	cache.Set("setting_aria2_interval", "10", 0)
+	tasks := []model.Download{
+		{
+			StatusInfo: rpc.StatusInfo{
+				Files: []rpc.FileInfo{
+					{
+						Path: "/file/name.txt",
+					},
+				},
+			},
+			Task: &model.Task{
+				Model: gorm.Model{},
+				Error: "error",
+			},
+		},
+		{
+			StatusInfo: rpc.StatusInfo{
+				Files: []rpc.FileInfo{
+					{
+						Path: "/file/name1.txt",
+					},
+					{
+						Path: "/file/name2.txt",
+					},
+				},
+			},
+		},
+	}
+	tasks[1].StatusInfo.BitTorrent.Info.Name = "name.txt"
+
+	res := BuildDownloadingResponse(tasks).Data.([]DownloadListResponse)
+	asserts.Len(res, 2)
+	asserts.Equal("name1.txt", res[1].Name)
+	asserts.Equal("name.txt", res[0].Name)
+	asserts.Equal("name.txt", res[0].Info.Files[0].Path)
+	asserts.Equal("name1.txt", res[1].Info.Files[0].Path)
+	asserts.Equal("name2.txt", res[1].Info.Files[1].Path)
+}