You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
73 lines
1.5 KiB
73 lines
1.5 KiB
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/zeebo/bencode"
|
|
)
|
|
|
|
func TestCreateTorrent(t *testing.T) {
|
|
fileName := "server.go"
|
|
var decoded Torrent
|
|
|
|
f, err := os.Open("server.go")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer f.Close()
|
|
|
|
encoded, err := createTorrent(fileName, f, nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
bencode.DecodeBytes(encoded, &decoded)
|
|
|
|
if decoded.Encoding != "UTF-8" {
|
|
t.Fatalf("Encoding was %s, expected UTF-8", decoded.Encoding)
|
|
}
|
|
|
|
if decoded.Info.Name != "server.go" {
|
|
t.Fatalf("Name was %s, expected server.go", decoded.Info.Name)
|
|
}
|
|
|
|
if decoded.Info.PieceLength <= 0 {
|
|
t.Fatal("Expected a piece length, got none")
|
|
}
|
|
|
|
if len(decoded.Info.Pieces) <= 0 {
|
|
t.Fatal("Expected at least one piece, got none")
|
|
}
|
|
|
|
if decoded.Info.Length <= 0 {
|
|
t.Fatal("Length was less than or equal to 0, expected more")
|
|
}
|
|
|
|
tracker := fmt.Sprintf("%s%s%s", Config.siteURL, Config.selifPath, fileName)
|
|
if decoded.UrlList[0] != tracker {
|
|
t.Fatalf("First entry in URL list was %s, expected %s", decoded.UrlList[0], tracker)
|
|
}
|
|
}
|
|
|
|
func TestCreateTorrentWithImage(t *testing.T) {
|
|
var decoded Torrent
|
|
|
|
f, err := os.Open("static/images/404.jpg")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer f.Close()
|
|
|
|
encoded, err := createTorrent("test.jpg", f, nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
bencode.DecodeBytes(encoded, &decoded)
|
|
|
|
if decoded.Info.Pieces != "r\x01\x80j\x99\x84\n\xd3dZ;1NX\xec;\x9d$+f" {
|
|
t.Fatal("Torrent pieces did not match expected pieces for image")
|
|
}
|
|
}
|