cloudreve/pkg/serializer/aria2_test.go

package serializer

import (
	model "github.com/HFO4/cloudreve/models"
	"github.com/HFO4/cloudreve/pkg/aria2/rpc"
	"github.com/HFO4/cloudreve/pkg/cache"
	"github.com/jinzhu/gorm"
	"github.com/stretchr/testify/assert"
	"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)
}