torrent/metainfo/nodes_test.go

63 lines
1.6 KiB
Go

package metainfo
import (
"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:infod4:name0:12:piece lengthi0e6:piecesleee", &MetaInfo{})
testMarshalMetainfo(t, "d4:infod4:name0:12:piece lengthi0e6:pieceslee5:nodesl12:1.2.3.4:555514:not a hostportee", &MetaInfo{
Nodes: []Node{"1.2.3.4:5555", "not a hostport"},
})
}
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)
}