116 lines
3.2 KiB
Go
116 lines
3.2 KiB
Go
package metainfo
|
|
|
|
import (
|
|
"encoding/hex"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
var (
|
|
exampleMagnetURI = `magnet:?xt=urn:btih:51340689c960f0778a4387aef9b4b52fd08390cd&dn=Shit+Movie+%281985%29+1337p+-+Eru&tr=http%3A%2F%2Fhttp.was.great%21&tr=udp%3A%2F%2Fanti.piracy.honeypot%3A6969`
|
|
exampleMagnet = Magnet{
|
|
DisplayName: "Shit Movie (1985) 1337p - Eru",
|
|
Trackers: []string{
|
|
"http://http.was.great!",
|
|
"udp://anti.piracy.honeypot:6969",
|
|
},
|
|
}
|
|
)
|
|
|
|
// Converting from our Magnet type to URL string.
|
|
func TestMagnetString(t *testing.T) {
|
|
hex.Decode(exampleMagnet.InfoHash[:], []byte("51340689c960f0778a4387aef9b4b52fd08390cd"))
|
|
s := exampleMagnet.String()
|
|
if s != exampleMagnetURI {
|
|
t.Fatalf("\nexpected:\n\t%q\nactual\n\t%q", exampleMagnetURI, s)
|
|
}
|
|
}
|
|
|
|
func TestParseMagnetURI(t *testing.T) {
|
|
var uri string
|
|
var m Magnet
|
|
var err error
|
|
|
|
// parsing the legit Magnet URI with btih-formatted xt should not return errors
|
|
uri = "magnet:?xt=urn:btih:ZOCMZQIPFFW7OLLMIC5HUB6BPCSDEOQU"
|
|
_, err = ParseMagnetURI(uri)
|
|
if err != nil {
|
|
t.Errorf("Attempting parsing the proper Magnet btih URI:\"%v\" failed with err: %v", uri, err)
|
|
}
|
|
|
|
// Checking if the magnet instance struct is built correctly from parsing
|
|
m, err = ParseMagnetURI(exampleMagnetURI)
|
|
if err != nil || !reflect.DeepEqual(exampleMagnet, m) {
|
|
t.Errorf("ParseMagnetURI(%s) returned %v, expected %v", uri, m, exampleMagnet)
|
|
}
|
|
|
|
// empty string URI case
|
|
_, err = ParseMagnetURI("")
|
|
if err == nil {
|
|
t.Errorf("Parsing empty string as URI should have returned an error but didn't")
|
|
}
|
|
|
|
// only BTIH (BitTorrent info hash)-formatted magnet links are currently supported
|
|
// must return error correctly when encountering other URN formats
|
|
uri = "magnet:?xt=urn:sha1:YNCKHTQCWBTRNJIV4WNAE52SJUQCZO5C"
|
|
_, err = ParseMagnetURI(uri)
|
|
if err == nil {
|
|
t.Errorf("Magnet URI with non-BTIH URNs (like \"%v\") are not supported and should return an error", uri)
|
|
}
|
|
|
|
// resilience to the broken hash
|
|
uri = "magnet:?xt=urn:btih:this hash is really broken"
|
|
_, err = ParseMagnetURI(uri)
|
|
if err == nil {
|
|
t.Errorf("Failed to detect broken Magnet URI: %v", uri)
|
|
}
|
|
|
|
}
|
|
|
|
func TestMagnetize(t *testing.T) {
|
|
mi, err := LoadFromFile("../testdata/bootstrap.dat.torrent")
|
|
require.NoError(t, err)
|
|
|
|
info, err := mi.UnmarshalInfo()
|
|
require.NoError(t, err)
|
|
m := mi.Magnet(info.Name, mi.HashInfoBytes())
|
|
|
|
assert.EqualValues(t, "bootstrap.dat", m.DisplayName)
|
|
|
|
ih := [20]byte{
|
|
54, 113, 155, 162, 206, 207, 159, 59, 215, 197,
|
|
171, 251, 122, 136, 233, 57, 97, 27, 83, 108,
|
|
}
|
|
|
|
if m.InfoHash != ih {
|
|
t.Errorf("Magnet infohash is incorrect")
|
|
}
|
|
|
|
trackers := []string{
|
|
"udp://tracker.openbittorrent.com:80",
|
|
"udp://tracker.openbittorrent.com:80",
|
|
"udp://tracker.publicbt.com:80",
|
|
"udp://coppersurfer.tk:6969/announce",
|
|
"udp://open.demonii.com:1337",
|
|
"http://bttracker.crunchbanglinux.org:6969/announce",
|
|
}
|
|
|
|
for _, expected := range trackers {
|
|
if !contains(m.Trackers, expected) {
|
|
t.Errorf("Magnet does not contain expected tracker: %s", expected)
|
|
}
|
|
}
|
|
}
|
|
|
|
func contains(haystack []string, needle string) bool {
|
|
for _, s := range haystack {
|
|
if s == needle {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|