mirror of
https://github.com/truewhile/MeBox.git
synced 2026-09-28 11:16:37 +08:00
ShukeBta
60 lines
2.4 KiB
Go
60 lines
2.4 KiB
Go
package service
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/ShukeBta/MediaStationGo/internal/model"
|
|
)
|
|
|
|
func TestSelectSiteSearchCandidatesAppliesSeederSizeAndFreeRules(t *testing.T) {
|
|
sub := &model.Subscription{
|
|
Name: "Some Show 自动订阅",
|
|
Filter: "Some Show",
|
|
MediaType: "tv",
|
|
MinSeeders: 10,
|
|
MaxSeeders: 100,
|
|
MinSizeGB: 1,
|
|
MaxSizeGB: 8,
|
|
FreeOnly: true,
|
|
}
|
|
results := []SearchResult{
|
|
{Title: "Some Show S01E01 1080p WEB-DL H264 AAC", DownloadURL: "https://pt/download/low-seed", Seeders: 3, Size: 2 * bytesPerGiB, Free: true},
|
|
{Title: "Some Show S01E01 1080p WEB-DL H264 AAC", DownloadURL: "https://pt/download/hot", Seeders: 500, Size: 2 * bytesPerGiB, Free: true},
|
|
{Title: "Some Show S01E01 1080p WEB-DL H264 AAC", DownloadURL: "https://pt/download/small", Seeders: 50, Size: bytesPerGiB / 2, Free: true},
|
|
{Title: "Some Show S01E01 1080p WEB-DL H264 AAC", DownloadURL: "https://pt/download/nonfree", Seeders: 50, Size: 2 * bytesPerGiB},
|
|
{Title: "Some Show S01E01 1080p WEB-DL H264 AAC", DownloadURL: "https://pt/download/right", Seeders: 50, Size: 2 * bytesPerGiB, Free: true},
|
|
}
|
|
|
|
got, stats := selectSiteSearchCandidatesWithStats(results, sub, nil, LocalAvailability{})
|
|
if len(got) != 1 || got[0].Download != "https://pt/download/right" {
|
|
t.Fatalf("selected %#v, want only torrent matching seed/size/free rules", got)
|
|
}
|
|
if stats.RuleMismatch != 4 || stats.Prepared != 1 || stats.Selected != 1 {
|
|
t.Fatalf("stats = %#v, want four rule mismatches and one selected", stats)
|
|
}
|
|
}
|
|
|
|
func TestSubscriptionTorrentRulesRecognizeFreeLabels(t *testing.T) {
|
|
sub := &model.Subscription{FreeOnly: true}
|
|
for _, item := range []SearchResult{
|
|
{Title: "Some Movie 2026 1080p WEB-DL FREE"},
|
|
{Title: "Some Movie 2026 1080p WEB-DL FreeLeech"},
|
|
{Title: "Some Movie 2026 1080p WEB-DL 免费"},
|
|
{Title: "Some Movie 2026 1080p WEB-DL", Free: true},
|
|
} {
|
|
if !matchesSubscriptionTorrentRules(sub, item) {
|
|
t.Fatalf("expected free rule to accept %#v", item)
|
|
}
|
|
}
|
|
if matchesSubscriptionTorrentRules(sub, SearchResult{Title: "Some Movie 2026 1080p WEB-DL"}) {
|
|
t.Fatal("free-only rule accepted non-free result")
|
|
}
|
|
}
|
|
|
|
func TestSubscriptionTorrentRulesRejectUnknownSizeWhenSizeRangeConfigured(t *testing.T) {
|
|
sub := &model.Subscription{MinSizeGB: 1}
|
|
if matchesSubscriptionTorrentRules(sub, SearchResult{Title: "Some Movie 2026 1080p WEB-DL"}) {
|
|
t.Fatal("size range accepted result without size metadata")
|
|
}
|
|
}
|