mirror of
https://github.com/truewhile/MeBox.git
synced 2026-09-28 11:16:37 +08:00
ShukeBta
77 lines
2.7 KiB
Go
77 lines
2.7 KiB
Go
package service
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"go.uber.org/zap"
|
|
|
|
"github.com/ShukeBta/MediaStationGo/internal/config"
|
|
"github.com/ShukeBta/MediaStationGo/internal/model"
|
|
"github.com/ShukeBta/MediaStationGo/internal/repository"
|
|
)
|
|
|
|
func TestPrepareSubscriptionForRunFillsSeriesMetadata(t *testing.T) {
|
|
var searchedTV bool
|
|
upstream := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
switch r.URL.Path {
|
|
case "/search/tv":
|
|
searchedTV = true
|
|
_ = json.NewEncoder(w).Encode(map[string]any{
|
|
"results": []map[string]any{{
|
|
"id": 12345,
|
|
"name": "南部档案",
|
|
"original_name": "Archives The Nanyang Mystery",
|
|
"original_language": "zh",
|
|
"origin_country": []string{"CN"},
|
|
"first_air_date": "2026-01-01",
|
|
}},
|
|
})
|
|
case "/tv/12345":
|
|
_ = json.NewEncoder(w).Encode(map[string]any{"number_of_episodes": 33})
|
|
default:
|
|
http.NotFound(w, r)
|
|
}
|
|
}))
|
|
defer upstream.Close()
|
|
|
|
cfg := &config.Config{}
|
|
cfg.Secrets.TMDbAPIKey = "test-key"
|
|
cfg.Secrets.TMDbAPIProxy = upstream.URL
|
|
repos := repository.New(newServiceTestDB(t, &model.Subscription{}))
|
|
scraper := NewScraperService(cfg, zap.NewNop(), repos, NewTMDbProvider(cfg, zap.NewNop(), nil), nil, nil, nil, NewHub(zap.NewNop()))
|
|
svc := NewSubscriptionService(cfg, zap.NewNop(), repos, nil, nil, nil)
|
|
svc.SetScraper(scraper)
|
|
|
|
sub := model.Subscription{Name: "南部档案 自动订阅", FeedURL: "site-search://search?keyword=南部档案", Filter: "南部档案 2026", Enabled: true}
|
|
if err := repos.Subscription.Create(t.Context(), &sub); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
svc.prepareSubscriptionForRun(t.Context(), &sub)
|
|
if !searchedTV {
|
|
t.Fatal("blank media type subscription should try TV metadata before defaulting to movie")
|
|
}
|
|
if sub.MediaType != "tv" || sub.OriginalName != "Archives The Nanyang Mystery" || sub.Year != 2026 || sub.TotalEpisodes != 33 {
|
|
t.Fatalf("prepared subscription = %#v, want tv metadata with total episodes", sub)
|
|
}
|
|
|
|
var stored model.Subscription
|
|
if err := repos.DB.First(&stored, "id = ?", sub.ID).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if stored.MediaType != "tv" || stored.OriginalName != "Archives The Nanyang Mystery" || stored.Year != 2026 || stored.TotalEpisodes != 33 {
|
|
t.Fatalf("stored subscription = %#v, want persisted metadata", stored)
|
|
}
|
|
}
|
|
|
|
func TestSubscriptionMetadataLibraryTypesSearchesTVForBlankType(t *testing.T) {
|
|
got := subscriptionMetadataLibraryTypes("", "南部档案 2026")
|
|
if len(got) < 2 || got[0] != "tv" || got[1] != "movie" {
|
|
t.Fatalf("library types = %#v, want tv before movie for blank subscription type", got)
|
|
}
|
|
}
|