gopeed/internal/protocol/bt/fetcher_test.go

package bt

import (
	"encoding/json"
	"github.com/GopeedLab/gopeed/internal/controller"
	"github.com/GopeedLab/gopeed/internal/fetcher"
	"github.com/GopeedLab/gopeed/internal/test"
	"github.com/GopeedLab/gopeed/pkg/base"
	"reflect"
	"testing"
)

func TestFetcher_Resolve_Torrent(t *testing.T) {
	doResolve(t, buildFetcher())
}

func TestFetcher_Config(t *testing.T) {
	doResolve(t, buildConfigFetcher())
}

func doResolve(t *testing.T, fetcher fetcher.Fetcher) {
	err := fetcher.Resolve(&base.Request{
		URL: "./testdata/ubuntu-22.04-live-server-amd64.iso.torrent",
	})
	if err != nil {
		panic(err)
	}

	want := &base.Resource{
		Name:    "ubuntu-22.04-live-server-amd64.iso",
		Size:    1466714112,
		Range:   true,
		RootDir: "ubuntu-22.04-live-server-amd64.iso",
		Files: []*base.FileInfo{
			{
				Name: "ubuntu-22.04-live-server-amd64.iso",
				Size: 1466714112,
			},
		},
		Hash: "8a55cfbd5ca5d11507364765936c4f9e55b253ed",
	}
	if !reflect.DeepEqual(want, fetcher.Meta().Res) {
		t.Errorf("Resolve() got = %v, want %v", fetcher.Meta().Res, want)
	}
}

func buildFetcher() fetcher.Fetcher {
	fetcher := new(FetcherBuilder).Build()
	fetcher.Setup(controller.NewController())
	return fetcher
}

func buildConfigFetcher() fetcher.Fetcher {
	fetcher := new(FetcherBuilder).Build()
	newController := controller.NewController()
	mockCfg := config{
		Trackers: []string{
			"udp://tracker.birkenwald.de:6969/announce",
			"udp://tracker.bitsearch.to:1337/announce",
		}}
	newController.GetConfig = func(v any) (bool, error) {
		if err := json.Unmarshal([]byte(test.ToJson(mockCfg)), v); err != nil {
			return false, err
		}
		return true, nil
	}
	fetcher.Setup(newController)
	return fetcher
}