FedP2P/metainfo/nodes_test.go

package metainfo

import (
	"bytes"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"

	"github.com/anacrolix/torrent/bencode"
)

func testFileNodesMatch(t *testing.T, file string, nodes []Node) {
	mi, err := LoadFromFile(file)
	require.NoError(t, err)
	assert.EqualValues(t, nodes, mi.Nodes)
}

func TestNodesListStrings(t *testing.T) {
	testFileNodesMatch(t, "testdata/trackerless.torrent", []Node{
		"udp://tracker.openbittorrent.com:80",
		"udp://tracker.openbittorrent.com:80",
	})
}

func TestNodesListPairsBEP5(t *testing.T) {
	testFileNodesMatch(t, "testdata/issue_65a.torrent", []Node{
		"185.34.3.132:5680",
		"185.34.3.103:12340",
		"94.209.253.165:47232",
		"78.46.103.11:34319",
		"195.154.162.70:55011",
		"185.34.3.137:3732",
	})
	testFileNodesMatch(t, "testdata/issue_65b.torrent", []Node{
		"95.211.203.130:6881",
		"84.72.116.169:6889",
		"204.83.98.77:7000",
		"101.187.175.163:19665",
		"37.187.118.32:6881",
		"83.128.223.71:23865",
	})
}

func testMarshalMetainfo(t *testing.T, expected string, mi *MetaInfo) {
	b, err := bencode.Marshal(*mi)
	assert.NoError(t, err)
	assert.EqualValues(t, expected, string(b))
}

func TestMarshalMetainfoNodes(t *testing.T) {
	testMarshalMetainfo(t, "d4:infodee", &MetaInfo{InfoBytes: []byte("de")})
	testMarshalMetainfo(t, "d4:infod2:hi5:theree5:nodesl12:1.2.3.4:555514:not a hostportee", &MetaInfo{
		Nodes:     []Node{"1.2.3.4:5555", "not a hostport"},
		InfoBytes: []byte("d2:hi5:theree"),
	})
}

func TestUnmarshalBadMetainfoNodes(t *testing.T) {
	var mi MetaInfo
	// Should barf on the integer in the nodes list.
	err := bencode.Unmarshal([]byte("d5:nodesl1:ai42eee"), &mi)
	require.Error(t, err)
}

func TestMetainfoEmptyInfoBytes(t *testing.T) {
	var buf bytes.Buffer
	require.NoError(t, (&MetaInfo{
		// Include a non-empty field that comes after "info".
		UrlList: []string{"hello"},
	}).Write(&buf))
	var mi MetaInfo
	require.NoError(t, bencode.Unmarshal(buf.Bytes(), &mi))
}