import (
"encoding/hex"
- "reflect"
"testing"
"github.com/stretchr/testify/assert"
}
)
+func init() {
+ hex.Decode(exampleMagnet.InfoHash[:], []byte("51340689c960f0778a4387aef9b4b52fd08390cd"))
+}
+
// 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)
- }
+ m, err := ParseMagnetUri(exampleMagnet.String())
+ require.NoError(t, err)
+ assert.EqualValues(t, exampleMagnet, m)
}
func TestParseMagnetURI(t *testing.T) {
// parsing the legit Magnet URI with btih-formatted xt should not return errors
uri = "magnet:?xt=urn:btih:ZOCMZQIPFFW7OLLMIC5HUB6BPCSDEOQU"
- _, err = ParseMagnetURI(uri)
+ _, 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)
- }
+ m, err = ParseMagnetUri(exampleMagnetURI)
+ assert.EqualValues(t, exampleMagnet, m)
+ assert.NoError(t, err)
// empty string URI case
- _, err = ParseMagnetURI("")
+ _, 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)
+ _, 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)
+ _, err = ParseMagnetUri(uri)
if err == nil {
t.Errorf("Failed to detect broken Magnet URI: %v", uri)
}
-
}
-func Test_Magnetize(t *testing.T) {
+func TestMagnetize(t *testing.T) {
mi, err := LoadFromFile("../testdata/bootstrap.dat.torrent")
require.NoError(t, err)
- m := mi.Magnet()
+ info, err := mi.UnmarshalInfo()
+ require.NoError(t, err)
+ m := mi.Magnet(nil, &info)
assert.EqualValues(t, "bootstrap.dat", m.DisplayName)