7af90ebe47
- Remove trackers from torrenst/magnet URI --------- Co-authored-by: Mukhtar Akere <akeremukhtar10@gmail.com>
199 lines
8.0 KiB
Go
199 lines
8.0 KiB
Go
package utils
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/sirrobot01/decypharr/internal/testutil"
|
|
)
|
|
|
|
// checkMagnet is a helper function that verifies magnet properties
|
|
func checkMagnet(t *testing.T, magnet *Magnet, expectedInfoHash, expectedName, expectedLink string, expectedTrackerCount int, shouldBeTorrent bool) {
|
|
t.Helper() // This marks the function as a test helper
|
|
|
|
// Verify basic properties
|
|
if magnet.Name != expectedName {
|
|
t.Errorf("Expected name '%s', got '%s'", expectedName, magnet.Name)
|
|
}
|
|
if magnet.InfoHash != expectedInfoHash {
|
|
t.Errorf("Expected InfoHash '%s', got '%s'", expectedInfoHash, magnet.InfoHash)
|
|
}
|
|
if magnet.Link != expectedLink {
|
|
t.Errorf("Expected Link '%s', got '%s'", expectedLink, magnet.Link)
|
|
}
|
|
|
|
// Verify the magnet link contains the essential info hash
|
|
if !strings.Contains(magnet.Link, "xt=urn:btih:"+expectedInfoHash) {
|
|
t.Error("Magnet link should contain info hash")
|
|
}
|
|
|
|
// Verify tracker count
|
|
trCount := strings.Count(magnet.Link, "tr=")
|
|
if trCount != expectedTrackerCount {
|
|
t.Errorf("Expected %d tracker URLs, got %d", expectedTrackerCount, trCount)
|
|
}
|
|
}
|
|
|
|
// testMagnetFromFile is a helper function for tests that use GetMagnetFromFile with file operations
|
|
func testMagnetFromFile(t *testing.T, filePath string, rmTrackerUrls bool, expectedInfoHash, expectedName, expectedLink string, expectedTrackerCount int) {
|
|
t.Helper()
|
|
|
|
file, err := os.Open(filePath)
|
|
if err != nil {
|
|
t.Fatalf("Failed to open torrent file %s: %v", filePath, err)
|
|
}
|
|
defer file.Close()
|
|
|
|
magnet, err := GetMagnetFromFile(file, filepath.Base(filePath), rmTrackerUrls)
|
|
if err != nil {
|
|
t.Fatalf("GetMagnetFromFile failed: %v", err)
|
|
}
|
|
|
|
checkMagnet(t, magnet, expectedInfoHash, expectedName, expectedLink, expectedTrackerCount, true)
|
|
|
|
// Log the result
|
|
if rmTrackerUrls {
|
|
t.Logf("Generated clean magnet link: %s", magnet.Link)
|
|
} else {
|
|
t.Logf("Generated magnet link with trackers: %s", magnet.Link)
|
|
}
|
|
}
|
|
|
|
func TestGetMagnetFromFile_RealTorrentFile_StripTrue(t *testing.T) {
|
|
expectedInfoHash := "8a19577fb5f690970ca43a57ff1011ae202244b8"
|
|
expectedName := "ubuntu-25.04-desktop-amd64.iso"
|
|
expectedLink := "magnet:?xt=urn:btih:8a19577fb5f690970ca43a57ff1011ae202244b8&dn=ubuntu-25.04-desktop-amd64.iso"
|
|
expectedTrackerCount := 0 // Should be 0 when stripping trackers
|
|
|
|
torrentPath := testutil.GetTestTorrentPath()
|
|
testMagnetFromFile(t, torrentPath, true, expectedInfoHash, expectedName, expectedLink, expectedTrackerCount)
|
|
}
|
|
|
|
func TestGetMagnetFromFile_RealTorrentFile_StripFalse(t *testing.T) {
|
|
expectedInfoHash := "8a19577fb5f690970ca43a57ff1011ae202244b8"
|
|
expectedName := "ubuntu-25.04-desktop-amd64.iso"
|
|
expectedLink := "magnet:?xt=urn:btih:8a19577fb5f690970ca43a57ff1011ae202244b8&dn=ubuntu-25.04-desktop-amd64.iso&tr=https%3A%2F%2Ftorrent.ubuntu.com%2Fannounce&tr=https%3A%2F%2Fipv6.torrent.ubuntu.com%2Fannounce"
|
|
expectedTrackerCount := 2 // Should be 2 when preserving trackers
|
|
|
|
torrentPath := testutil.GetTestTorrentPath()
|
|
testMagnetFromFile(t, torrentPath, false, expectedInfoHash, expectedName, expectedLink, expectedTrackerCount)
|
|
}
|
|
|
|
func TestGetMagnetFromFile_MagnetFile_StripTrue(t *testing.T) {
|
|
expectedInfoHash := "8a19577fb5f690970ca43a57ff1011ae202244b8"
|
|
expectedName := "ubuntu-25.04-desktop-amd64.iso"
|
|
expectedLink := "magnet:?xt=urn:btih:8a19577fb5f690970ca43a57ff1011ae202244b8&dn=ubuntu-25.04-desktop-amd64.iso"
|
|
expectedTrackerCount := 0 // Should be 0 when stripping trackers
|
|
|
|
torrentPath := testutil.GetTestMagnetPath()
|
|
testMagnetFromFile(t, torrentPath, true, expectedInfoHash, expectedName, expectedLink, expectedTrackerCount)
|
|
}
|
|
|
|
func TestGetMagnetFromFile_MagnetFile_StripFalse(t *testing.T) {
|
|
expectedInfoHash := "8a19577fb5f690970ca43a57ff1011ae202244b8"
|
|
expectedName := "ubuntu-25.04-desktop-amd64.iso"
|
|
expectedLink := "magnet:?xt=urn:btih:8a19577fb5f690970ca43a57ff1011ae202244b8&dn=ubuntu-25.04-desktop-amd64.iso&tr=https%3A%2F%2Fipv6.torrent.ubuntu.com%2Fannounce&tr=https%3A%2F%2Ftorrent.ubuntu.com%2Fannounce"
|
|
expectedTrackerCount := 2
|
|
|
|
torrentPath := testutil.GetTestMagnetPath()
|
|
testMagnetFromFile(t, torrentPath, false, expectedInfoHash, expectedName, expectedLink, expectedTrackerCount)
|
|
}
|
|
|
|
func TestGetMagnetFromUrl_MagnetLink_StripTrue(t *testing.T) {
|
|
expectedInfoHash := "8a19577fb5f690970ca43a57ff1011ae202244b8"
|
|
expectedName := "ubuntu-25.04-desktop-amd64.iso"
|
|
expectedLink := "magnet:?xt=urn:btih:8a19577fb5f690970ca43a57ff1011ae202244b8&dn=ubuntu-25.04-desktop-amd64.iso"
|
|
expectedTrackerCount := 0
|
|
|
|
// Load the magnet URL from the test file
|
|
magnetUrl, err := testutil.GetTestMagnetContent()
|
|
if err != nil {
|
|
t.Fatalf("Failed to load magnet URL from test file: %v", err)
|
|
}
|
|
|
|
magnet, err := GetMagnetFromUrl(magnetUrl, true)
|
|
if err != nil {
|
|
t.Fatalf("GetMagnetFromUrl failed: %v", err)
|
|
}
|
|
|
|
checkMagnet(t, magnet, expectedInfoHash, expectedName, expectedLink, expectedTrackerCount, false)
|
|
t.Logf("Generated clean magnet link: %s", magnet.Link)
|
|
}
|
|
|
|
func TestGetMagnetFromUrl_MagnetLink_StripFalse(t *testing.T) {
|
|
expectedInfoHash := "8a19577fb5f690970ca43a57ff1011ae202244b8"
|
|
expectedName := "ubuntu-25.04-desktop-amd64.iso"
|
|
expectedLink := "magnet:?xt=urn:btih:8a19577fb5f690970ca43a57ff1011ae202244b8&dn=ubuntu-25.04-desktop-amd64.iso&tr=https%3A%2F%2Fipv6.torrent.ubuntu.com%2Fannounce&tr=https%3A%2F%2Ftorrent.ubuntu.com%2Fannounce"
|
|
expectedTrackerCount := 2
|
|
|
|
// Load the magnet URL from the test file
|
|
magnetUrl, err := testutil.GetTestMagnetContent()
|
|
if err != nil {
|
|
t.Fatalf("Failed to load magnet URL from test file: %v", err)
|
|
}
|
|
|
|
magnet, err := GetMagnetFromUrl(magnetUrl, false)
|
|
if err != nil {
|
|
t.Fatalf("GetMagnetFromUrl failed: %v", err)
|
|
}
|
|
|
|
checkMagnet(t, magnet, expectedInfoHash, expectedName, expectedLink, expectedTrackerCount, false)
|
|
t.Logf("Generated magnet link with trackers: %s", magnet.Link)
|
|
}
|
|
|
|
|
|
// testMagnetFromHttpTorrent is a helper function for tests that use GetMagnetFromUrl with HTTP torrent links
|
|
func testMagnetFromHttpTorrent(t *testing.T, torrentPath string, rmTrackerUrls bool, expectedInfoHash, expectedName, expectedLink string, expectedTrackerCount int) {
|
|
t.Helper()
|
|
|
|
// Read the torrent file content
|
|
torrentData, err := testutil.GetTestDataBytes(torrentPath)
|
|
if err != nil {
|
|
t.Fatalf("Failed to read torrent file: %v", err)
|
|
}
|
|
|
|
// Create a test HTTP server that serves the torrent file
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Content-Type", "application/x-bittorrent")
|
|
w.Write(torrentData)
|
|
}))
|
|
defer server.Close()
|
|
|
|
// Test the function with the mock server URL
|
|
magnet, err := GetMagnetFromUrl(server.URL, rmTrackerUrls)
|
|
if err != nil {
|
|
t.Fatalf("GetMagnetFromUrl failed: %v", err)
|
|
}
|
|
|
|
checkMagnet(t, magnet, expectedInfoHash, expectedName, expectedLink, expectedTrackerCount, true)
|
|
|
|
// Log the result
|
|
if rmTrackerUrls {
|
|
t.Logf("Generated clean magnet link from HTTP torrent: %s", magnet.Link)
|
|
} else {
|
|
t.Logf("Generated magnet link with trackers from HTTP torrent: %s", magnet.Link)
|
|
}
|
|
}
|
|
|
|
func TestGetMagnetFromUrl_TorrentLink_StripTrue(t *testing.T) {
|
|
expectedInfoHash := "8a19577fb5f690970ca43a57ff1011ae202244b8"
|
|
expectedName := "ubuntu-25.04-desktop-amd64.iso"
|
|
expectedLink := "magnet:?xt=urn:btih:8a19577fb5f690970ca43a57ff1011ae202244b8&dn=ubuntu-25.04-desktop-amd64.iso"
|
|
expectedTrackerCount := 0
|
|
|
|
testMagnetFromHttpTorrent(t, "ubuntu-25.04-desktop-amd64.iso.torrent", true, expectedInfoHash, expectedName, expectedLink, expectedTrackerCount)
|
|
}
|
|
|
|
func TestGetMagnetFromUrl_TorrentLink_StripFalse(t *testing.T) {
|
|
expectedInfoHash := "8a19577fb5f690970ca43a57ff1011ae202244b8"
|
|
expectedName := "ubuntu-25.04-desktop-amd64.iso"
|
|
expectedLink := "magnet:?xt=urn:btih:8a19577fb5f690970ca43a57ff1011ae202244b8&dn=ubuntu-25.04-desktop-amd64.iso&tr=https%3A%2F%2Ftorrent.ubuntu.com%2Fannounce&tr=https%3A%2F%2Fipv6.torrent.ubuntu.com%2Fannounce"
|
|
expectedTrackerCount := 2
|
|
|
|
testMagnetFromHttpTorrent(t, "ubuntu-25.04-desktop-amd64.iso.torrent", false, expectedInfoHash, expectedName, expectedLink, expectedTrackerCount)
|
|
}
|