mirror of
https://github.com/truewhile/MeBox.git
synced 2026-09-29 11:36:36 +08:00
ShukeBta
83 lines
2.5 KiB
Go
83 lines
2.5 KiB
Go
package service
|
|
|
|
import (
|
|
"encoding/json"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/ShukeBta/MediaStationGo/internal/model"
|
|
)
|
|
|
|
func TestDownloadViewsDoNotExposePrivateURL(t *testing.T) {
|
|
rows := []model.DownloadTask{{
|
|
UserID: "u1",
|
|
Source: "qbittorrent",
|
|
URL: "https://tracker.example/download?id=1&passkey=private-token",
|
|
Title: "测试影片",
|
|
SavePath: "/downloads",
|
|
Status: "queued",
|
|
}}
|
|
|
|
tasks, torrents := DownloadViews(rows, nil)
|
|
data, err := json.Marshal(map[string]any{
|
|
"tasks": tasks,
|
|
"torrents": torrents,
|
|
})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
body := string(data)
|
|
if strings.Contains(body, "private-token") || strings.Contains(body, "passkey") || strings.Contains(body, "tracker.example") {
|
|
t.Fatalf("download views leaked private URL: %s", body)
|
|
}
|
|
if !strings.Contains(body, "测试影片") {
|
|
t.Fatalf("download views should keep public title: %s", body)
|
|
}
|
|
}
|
|
|
|
func TestDownloadCompleteNotificationPayloadUsesTaskMetadata(t *testing.T) {
|
|
body, data := downloadCompleteNotificationPayload(QBitTorrent{
|
|
Hash: "done123",
|
|
Name: "Release.Name.S01E02.1080p",
|
|
SavePath: "/downloads/show",
|
|
ContentPath: "/downloads/show/Release.Name.S01E02.1080p.mkv",
|
|
}, &model.DownloadTask{
|
|
Title: "正式标题",
|
|
PosterURL: "https://img.example/poster.jpg",
|
|
BackdropURL: "https://img.example/backdrop.jpg",
|
|
MediaType: "tv",
|
|
MediaCategory: "日番",
|
|
Overview: "简介",
|
|
OriginalName: "Original Title",
|
|
OriginalLanguage: "ja",
|
|
Year: 2026,
|
|
Rating: 8.7,
|
|
Genres: "动画,剧情",
|
|
})
|
|
|
|
if !strings.Contains(body, "任务：正式标题") {
|
|
t.Fatalf("body should prefer task title, got %q", body)
|
|
}
|
|
if !strings.Contains(body, "保存路径：/downloads/show/Release.Name.S01E02.1080p.mkv") {
|
|
t.Fatalf("body should include content path, got %q", body)
|
|
}
|
|
for key, want := range map[string]interface{}{
|
|
"resource_title": "Release.Name.S01E02.1080p",
|
|
"title": "正式标题",
|
|
"poster_url": "https://img.example/poster.jpg",
|
|
"backdrop_url": "https://img.example/backdrop.jpg",
|
|
"media_type": "tv",
|
|
"media_category": "日番",
|
|
"overview": "简介",
|
|
"original_title": "Original Title",
|
|
"original_language": "ja",
|
|
"year": 2026,
|
|
"rating": float32(8.7),
|
|
"genres": "动画,剧情",
|
|
} {
|
|
if got := data[key]; got != want {
|
|
t.Fatalf("data[%s] = %#v, want %#v", key, got, want)
|
|
}
|
|
}
|
|
}
|