Files

79 lines
2.2 KiB
Go

package magnet
import "testing"
func TestParse(t *testing.T) {
tests := []struct {
name string
raw string
infohash string
dn string
trackers int
}{
{
name: "btih hex (из BRIEF)",
raw: "magnet:?xt=urn:btih:541ADCFF3B6DD5DBA7088EA83317D9D6FAC331D6&tr=http%3A%2F%2Fbt.t-ru.org%2Fann%3Fmagnet&dn=rutracker-topic-6514485",
infohash: "541adcff3b6dd5dba7088ea83317d9d6fac331d6",
dn: "rutracker-topic-6514485",
trackers: 1,
},
{
name: "btih base32 (20 нулевых байт)",
raw: "magnet:?xt=urn:btih:AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA",
infohash: "0000000000000000000000000000000000000000",
},
{
name: "btmh v2 sha256",
raw: "magnet:?xt=urn:btmh:12200123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef",
infohash: "0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef",
},
}
for _, tc := range tests {
t.Run(tc.name, func(t *testing.T) {
got, err := Parse(tc.raw)
if err != nil {
t.Fatalf("Parse: %v", err)
}
if got.Infohash != tc.infohash {
t.Errorf("infohash = %q, want %q", got.Infohash, tc.infohash)
}
if got.DisplayName != tc.dn {
t.Errorf("dn = %q, want %q", got.DisplayName, tc.dn)
}
if len(got.Trackers) != tc.trackers {
t.Errorf("trackers = %d, want %d", len(got.Trackers), tc.trackers)
}
})
}
}
func TestParseErrors(t *testing.T) {
cases := []string{
"https://example.com/file.torrent", // не magnet
"magnet:?dn=no-infohash", // нет xt
"magnet:?xt=urn:btih:zzzz", // некорректный hash
"", // пусто
}
for _, raw := range cases {
if _, err := Parse(raw); err == nil {
t.Errorf("Parse(%q): ожидалась ошибка", raw)
}
}
}
func TestParseNormalisesCase(t *testing.T) {
lower := "magnet:?xt=urn:btih:541adcff3b6dd5dba7088ea83317d9d6fac331d6"
upper := "magnet:?xt=urn:btih:541ADCFF3B6DD5DBA7088EA83317D9D6FAC331D6"
a, err := Parse(lower)
if err != nil {
t.Fatal(err)
}
b, err := Parse(upper)
if err != nil {
t.Fatal(err)
}
if a.Infohash != b.Infohash {
t.Errorf("регистр влияет: %q != %q", a.Infohash, b.Infohash)
}
}