mirror of
https://github.com/truewhile/MeBox.git
synced 2026-09-28 11:16:37 +08:00
truewhile
471 lines
16 KiB
Go
471 lines
16 KiB
Go
package service
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"go.uber.org/zap"
|
|
|
|
"github.com/truewhile/MeBox/internal/config"
|
|
"github.com/truewhile/MeBox/internal/model"
|
|
"github.com/truewhile/MeBox/internal/repository"
|
|
)
|
|
|
|
func TestListRecentSeriesCardsCountsAllEpisodesInSeries(t *testing.T) {
|
|
db := newServiceTestDB(t, &model.Library{}, &model.Media{})
|
|
repos := repository.New(db)
|
|
lib := model.Library{Name: "国漫", Path: "/media/anime", Type: "anime", Enabled: true}
|
|
if err := repos.Library.Create(t.Context(), &lib); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
now := time.Date(2026, 7, 2, 12, 0, 0, 0, time.UTC)
|
|
rows := make([]model.Media, 0, 40)
|
|
for i := 1; i <= 40; i++ {
|
|
created := now.Add(-48 * time.Hour)
|
|
if i > 23 {
|
|
created = now.Add(time.Duration(i) * time.Minute)
|
|
}
|
|
rows = append(rows, model.Media{
|
|
Base: model.Base{ID: fmt.Sprintf("recent-ep-%02d", i), CreatedAt: created, UpdatedAt: created},
|
|
LibraryID: lib.ID,
|
|
Title: "史上最强炼体老祖",
|
|
Path: fmt.Sprintf("/media/anime/国漫/史上最强炼体老祖/Season 01/史上最强炼体老祖.S01E%02d.mkv", i),
|
|
SeasonNum: 1,
|
|
EpisodeNum: i,
|
|
})
|
|
}
|
|
if err := repos.DB.Create(&rows).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
svc := NewMediaService(&config.Config{}, zap.NewNop(), repos)
|
|
|
|
cards, err := svc.ListRecentSeriesCards(t.Context(), 24, MediaVisibility{IncludeNSFW: true})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(cards) != 1 {
|
|
t.Fatalf("recent cards = %#v, want one series card", cards)
|
|
}
|
|
if cards[0].Count != 40 {
|
|
t.Fatalf("recent series count = %d, want full 40 episodes", cards[0].Count)
|
|
}
|
|
}
|
|
|
|
func TestMediaSeriesKeyCollapsesNestedSpecialFolders(t *testing.T) {
|
|
main := model.Media{
|
|
LibraryID: "lib-tv",
|
|
Path: `cloud://openlist/动漫/国漫/示例剧/Season 01/示例剧.S01E01.mkv`,
|
|
SeasonNum: 1,
|
|
EpisodeNum: 1,
|
|
}
|
|
special := model.Media{
|
|
LibraryID: "lib-tv",
|
|
Path: `cloud://openlist/动漫/国漫/示例剧/Extras/Season 01/示例剧.SP01.mkv`,
|
|
}
|
|
|
|
if got, want := mediaSeriesKey(special), mediaSeriesKey(main); got != want {
|
|
t.Fatalf("special key=%q, want main key=%q", got, want)
|
|
}
|
|
|
|
cards := groupMediaSeriesCards([]model.Media{main, special})
|
|
if len(cards) != 1 || cards[0].Count != 2 {
|
|
t.Fatalf("cards=%#v, want one merged series card with two items", cards)
|
|
}
|
|
}
|
|
|
|
func TestGroupMediaSeriesCardsSortsByLatestEpisodeTime(t *testing.T) {
|
|
now := time.Date(2026, 7, 2, 12, 0, 0, 0, time.UTC)
|
|
newerFirstEpisode := model.Media{
|
|
Base: model.Base{CreatedAt: now.Add(-72 * time.Hour), UpdatedAt: now.Add(-72 * time.Hour)},
|
|
LibraryID: "lib-tv",
|
|
Title: "更新合集",
|
|
Path: `F:\media\电视剧\国产剧\更新合集\Season 01\更新合集.S01E01.mkv`,
|
|
SeasonNum: 1,
|
|
EpisodeNum: 1,
|
|
}
|
|
newerLatestEpisode := model.Media{
|
|
Base: model.Base{CreatedAt: now, UpdatedAt: now},
|
|
LibraryID: "lib-tv",
|
|
Title: "更新合集",
|
|
Path: `F:\media\电视剧\国产剧\更新合集\Season 01\更新合集.S01E02.mkv`,
|
|
SeasonNum: 1,
|
|
EpisodeNum: 2,
|
|
}
|
|
olderSeries := model.Media{
|
|
Base: model.Base{CreatedAt: now.Add(-24 * time.Hour), UpdatedAt: now.Add(-24 * time.Hour)},
|
|
LibraryID: "lib-tv",
|
|
Title: "较早合集",
|
|
Path: `F:\media\电视剧\国产剧\较早合集\Season 01\较早合集.S01E01.mkv`,
|
|
SeasonNum: 1,
|
|
EpisodeNum: 1,
|
|
}
|
|
|
|
cards := groupMediaSeriesCards([]model.Media{olderSeries, newerFirstEpisode, newerLatestEpisode})
|
|
if len(cards) != 2 {
|
|
t.Fatalf("cards=%#v, want two series cards", cards)
|
|
}
|
|
if cards[0].Key != mediaSeriesKey(newerFirstEpisode) {
|
|
t.Fatalf("first card key=%q, want latest series key=%q", cards[0].Key, mediaSeriesKey(newerFirstEpisode))
|
|
}
|
|
if cards[0].Count != 2 {
|
|
t.Fatalf("latest series count=%d, want 2", cards[0].Count)
|
|
}
|
|
}
|
|
|
|
func TestMediaSeriesKeyCollapsesSpecialTitleSuffix(t *testing.T) {
|
|
main := model.Media{
|
|
LibraryID: "lib-tv",
|
|
Path: `cloud://openlist/电视剧/欧美剧/Example Show/Season 01/Example.Show.S01E01.mkv`,
|
|
SeasonNum: 1,
|
|
EpisodeNum: 1,
|
|
}
|
|
special := model.Media{
|
|
LibraryID: "lib-tv",
|
|
Path: `cloud://openlist/电视剧/欧美剧/Example Show Specials/Example.Show.Special.01.mkv`,
|
|
SeasonNum: 0,
|
|
EpisodeNum: 1,
|
|
}
|
|
chineseSpecial := model.Media{
|
|
LibraryID: "lib-tv",
|
|
Path: `cloud://openlist/动漫/国漫/示例剧 特别篇/示例剧.SP01.mkv`,
|
|
SeasonNum: 0,
|
|
EpisodeNum: 1,
|
|
}
|
|
chineseMain := model.Media{
|
|
LibraryID: "lib-tv",
|
|
Path: `cloud://openlist/动漫/国漫/示例剧/Season 01/示例剧.S01E01.mkv`,
|
|
SeasonNum: 1,
|
|
EpisodeNum: 1,
|
|
}
|
|
|
|
if got, want := mediaSeriesKey(special), mediaSeriesKey(main); got != want {
|
|
t.Fatalf("english special key=%q, want main key=%q", got, want)
|
|
}
|
|
if got, want := mediaSeriesKey(chineseSpecial), mediaSeriesKey(chineseMain); got != want {
|
|
t.Fatalf("chinese special key=%q, want main key=%q", got, want)
|
|
}
|
|
}
|
|
|
|
func TestMediaSeriesKeyCollapsesSeasonZeroAndSpecialAliases(t *testing.T) {
|
|
main := model.Media{
|
|
LibraryID: "lib-anime",
|
|
Path: `cloud://openlist/动漫/日番/宝可梦 (1997) {tmdb-60572}/Season 1/宝可梦.S01E01.mkv`,
|
|
SeasonNum: 1,
|
|
EpisodeNum: 1,
|
|
}
|
|
seasonZero := model.Media{
|
|
LibraryID: "lib-anime",
|
|
Path: `cloud://openlist/动漫/日番/宝可梦 (1997) {tmdb-60572}/Season 0/宝可梦.S00E34.mkv`,
|
|
SeasonNum: 0,
|
|
EpisodeNum: 34,
|
|
}
|
|
specialEpisode := model.Media{
|
|
LibraryID: "lib-anime",
|
|
Path: `cloud://openlist/动漫/日番/宝可梦 Special Episode/宝可梦.SP01.mkv`,
|
|
SeasonNum: 0,
|
|
EpisodeNum: 1,
|
|
}
|
|
extraEpisode := model.Media{
|
|
LibraryID: "lib-anime",
|
|
Path: `cloud://openlist/动漫/日番/宝可梦 番外篇/宝可梦.SP02.mkv`,
|
|
SeasonNum: 0,
|
|
EpisodeNum: 2,
|
|
}
|
|
|
|
want := mediaSeriesKey(main)
|
|
for name, item := range map[string]model.Media{
|
|
"season zero": seasonZero,
|
|
"special episode": specialEpisode,
|
|
"番外篇": extraEpisode,
|
|
} {
|
|
if got := mediaSeriesKey(item); got != want {
|
|
t.Fatalf("%s key=%q, want main key=%q", name, got, want)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestMediaSeriesKeyCollapsesNumberedSpecialSuffixes(t *testing.T) {
|
|
main := model.Media{
|
|
LibraryID: "lib-tv",
|
|
Path: `F:\media\电视剧\欧美剧\Example Show\Season 01\Example Show - S01E01.mkv`,
|
|
SeasonNum: 1,
|
|
EpisodeNum: 1,
|
|
}
|
|
chineseMain := model.Media{
|
|
LibraryID: "lib-tv",
|
|
Path: `F:\media\电视剧\欧美剧\示例剧\Season 01\示例剧.S01E01.mkv`,
|
|
SeasonNum: 1,
|
|
EpisodeNum: 1,
|
|
}
|
|
cases := map[string]struct {
|
|
item model.Media
|
|
want model.Media
|
|
}{
|
|
"sp number": {
|
|
item: model.Media{
|
|
LibraryID: "lib-tv",
|
|
Path: `F:\media\电视剧\欧美剧\Example Show SP01\Example Show.SP01.mkv`,
|
|
SeasonNum: 0,
|
|
EpisodeNum: 1,
|
|
},
|
|
want: main,
|
|
},
|
|
"ova number": {
|
|
item: model.Media{
|
|
LibraryID: "lib-tv",
|
|
Path: `F:\media\电视剧\欧美剧\Example Show OVA 1\Example Show.OVA.1.mkv`,
|
|
SeasonNum: 0,
|
|
EpisodeNum: 1,
|
|
},
|
|
want: main,
|
|
},
|
|
"season zero episode": {
|
|
item: model.Media{
|
|
LibraryID: "lib-tv",
|
|
Path: `F:\media\电视剧\欧美剧\Example Show S00E01\Example Show.S00E01.mkv`,
|
|
SeasonNum: 0,
|
|
EpisodeNum: 1,
|
|
},
|
|
want: main,
|
|
},
|
|
"wrapped special": {
|
|
item: model.Media{
|
|
LibraryID: "lib-tv",
|
|
Path: `F:\media\电视剧\欧美剧\Example Show [Special]\Example Show.Special.mkv`,
|
|
SeasonNum: 0,
|
|
EpisodeNum: 1,
|
|
},
|
|
want: main,
|
|
},
|
|
"chinese numbered special": {
|
|
item: model.Media{
|
|
LibraryID: "lib-tv",
|
|
Path: `F:\media\电视剧\欧美剧\示例剧 特别篇 第1集\示例剧.SP01.mkv`,
|
|
SeasonNum: 0,
|
|
EpisodeNum: 1,
|
|
},
|
|
want: chineseMain,
|
|
},
|
|
}
|
|
for name, tt := range cases {
|
|
want := mediaSeriesKey(tt.want)
|
|
if got := mediaSeriesKey(tt.item); got != want {
|
|
t.Fatalf("%s key=%q, want main key=%q", name, got, want)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestMediaSeriesKeyCleansReleaseNoiseFolders(t *testing.T) {
|
|
clean := model.Media{
|
|
LibraryID: "lib-variety",
|
|
Path: `F:\media\电视剧\综艺\Hntv Spring Festival Gala S01e (2026)\Season 1\Hntv Spring Festival Gala S01e - S01E202.ts`,
|
|
SeasonNum: 1,
|
|
EpisodeNum: 202,
|
|
}
|
|
dirty := model.Media{
|
|
LibraryID: "lib-variety",
|
|
Path: `F:\media\电视剧\综艺\Hntv Spring Festival Gala Fps Hlg Qhstudio S01e (2026)\Season 1\Hntv Spring Festival Gala Fps Hlg Qhstudio S01e - S01E202.ts`,
|
|
SeasonNum: 1,
|
|
EpisodeNum: 202,
|
|
}
|
|
if got, want := mediaSeriesKey(dirty), mediaSeriesKey(clean); got != want {
|
|
t.Fatalf("dirty folder key=%q, want clean folder key=%q", got, want)
|
|
}
|
|
|
|
noisyRelease := model.Media{
|
|
LibraryID: "lib-tv",
|
|
Path: `F:\media\电视剧\欧美剧\Motherhood Of Taihang Aac2 Mweb\Season 1\Motherhood Of Taihang Aac2 Mweb - S01E01-Aac2.Mweb.mkv`,
|
|
SeasonNum: 1,
|
|
EpisodeNum: 1,
|
|
}
|
|
cleanRelease := model.Media{
|
|
LibraryID: "lib-tv",
|
|
Path: `F:\media\电视剧\欧美剧\Motherhood Of Taihang\Season 1\Motherhood Of Taihang - S01E01.mkv`,
|
|
SeasonNum: 1,
|
|
EpisodeNum: 1,
|
|
}
|
|
if got, want := mediaSeriesKey(noisyRelease), mediaSeriesKey(cleanRelease); got != want {
|
|
t.Fatalf("release-noise folder key=%q, want clean key=%q", got, want)
|
|
}
|
|
}
|
|
|
|
func TestMediaSeriesKeyTreatsDomesticTelevisionFolderAsSeries(t *testing.T) {
|
|
main := model.Media{
|
|
LibraryID: "lib-domestic-tv",
|
|
Path: `/media/国产电视剧/人世间 (2022) [TMDBID-156568]/人世间.S01E01.mkv`,
|
|
SeasonNum: 1,
|
|
EpisodeNum: 1,
|
|
TMDbID: 156568,
|
|
}
|
|
weakEpisode := model.Media{
|
|
LibraryID: "lib-domestic-tv",
|
|
Path: `/media/国产电视剧/人世间 (2022) [TMDBID-156568]/人世间.S01E02.mkv`,
|
|
// Some local/cloud scans may miss S/E at first while local NFO or
|
|
// scraper metadata already carries an episode-level TMDb id.
|
|
TMDbID: 4375419,
|
|
}
|
|
folderRecord := model.Media{
|
|
LibraryID: "lib-domestic-tv",
|
|
Path: `/media/国产电视剧/人世间 (2022) [TMDBID-156568]`,
|
|
Title: "人世间",
|
|
TMDbID: 156568,
|
|
}
|
|
|
|
if got, want := mediaSeriesKey(weakEpisode), mediaSeriesKey(main); got != want {
|
|
t.Fatalf("domestic television folder key=%q, want main key=%q", got, want)
|
|
}
|
|
if got, want := mediaSeriesKey(folderRecord), mediaSeriesKey(main); got != want {
|
|
t.Fatalf("domestic television folder record key=%q, want main key=%q", got, want)
|
|
}
|
|
|
|
cards := groupMediaSeriesCards([]model.Media{main, weakEpisode, folderRecord})
|
|
if len(cards) != 1 || cards[0].Count != 3 {
|
|
t.Fatalf("cards=%#v, want one merged series card with three items", cards)
|
|
}
|
|
}
|
|
|
|
func TestMediaSeriesKeyUsesSeriesDirectoryExternalID(t *testing.T) {
|
|
episodeIDOnly := model.Media{
|
|
LibraryID: "lib-domestic-tv",
|
|
Path: `/media/电视剧/国产剧/人世间 (2022)/Season 01/人世间.S01E03.{tmdb-7129826}.mkv`,
|
|
SeasonNum: 1,
|
|
EpisodeNum: 3,
|
|
TMDbID: 7129826,
|
|
}
|
|
cleanFolder := model.Media{
|
|
LibraryID: "lib-domestic-tv",
|
|
Path: `/media/电视剧/国产剧/人世间 (2022)/Season 01/人世间.S01E04.mkv`,
|
|
SeasonNum: 1,
|
|
EpisodeNum: 4,
|
|
TMDbID: 156568,
|
|
}
|
|
if got, want := mediaSeriesKey(episodeIDOnly), mediaSeriesKey(cleanFolder); got != want {
|
|
t.Fatalf("episode filename tmdb id should not split clean folder key=%q, want %q", got, want)
|
|
}
|
|
}
|
|
|
|
func TestGroupMediaSeriesCardsMergesPollutedEpisodeFoldersBySharedShowID(t *testing.T) {
|
|
items := []model.Media{
|
|
{
|
|
LibraryID: "lib-variety",
|
|
Title: "脱口秀和Ta的朋友们",
|
|
Path: `F:\media\电视剧\综艺\脱口秀和Ta的朋友们 第01期\Season 01\show.S01E01.mkv`,
|
|
SeasonNum: 1,
|
|
EpisodeNum: 1,
|
|
TMDbID: 260001,
|
|
ScrapeStatus: "matched",
|
|
},
|
|
{
|
|
LibraryID: "lib-variety",
|
|
Title: "脱口秀和Ta的朋友们",
|
|
Path: `F:\media\电视剧\综艺\脱口秀和Ta的朋友们 第02期\Season 01\show.S01E02.mkv`,
|
|
SeasonNum: 1,
|
|
EpisodeNum: 2,
|
|
TMDbID: 260001,
|
|
ScrapeStatus: "matched",
|
|
},
|
|
}
|
|
|
|
cards := groupMediaSeriesCards(items)
|
|
if len(cards) != 1 || cards[0].Count != 2 {
|
|
t.Fatalf("cards=%#v, want one show card with two episodes", cards)
|
|
}
|
|
}
|
|
|
|
func TestGroupMediaSeriesCardsKeepsMovieVersionsAsOneMovie(t *testing.T) {
|
|
items := []model.Media{
|
|
{
|
|
Base: model.Base{ID: "movie-copy-a"},
|
|
LibraryID: "foreign-movies",
|
|
Title: "杀的就是你",
|
|
Path: `F:\media\电影\外语电影\They Will Kill You (2026)\movie-a.mkv`,
|
|
TMDbID: 1292695,
|
|
},
|
|
{
|
|
Base: model.Base{ID: "movie-copy-b"},
|
|
LibraryID: "western-movies",
|
|
Title: "杀的就是你",
|
|
Path: `F:\media\电影\欧美电影\They Will Kill You (2026)\movie-b.mkv`,
|
|
TMDbID: 1292695,
|
|
},
|
|
}
|
|
|
|
cards := groupMediaSeriesCards(items)
|
|
if len(cards) != 1 {
|
|
t.Fatalf("cards=%#v, want duplicate movie locations folded into one card", cards)
|
|
}
|
|
if cards[0].Count != 1 {
|
|
t.Fatalf("movie card count=%d, want 1 so versions are not shown as episodes", cards[0].Count)
|
|
}
|
|
}
|
|
|
|
func TestGroupMediaSeriesCardsDoesNotCollideMovieAndTVExternalIDs(t *testing.T) {
|
|
movie := model.Media{
|
|
Base: model.Base{ID: "movie"},
|
|
LibraryID: "movies",
|
|
Title: "同号电影",
|
|
Path: `/media/电影/同号电影 (2026)/movie.mkv`,
|
|
TMDbID: 12345,
|
|
}
|
|
episode := model.Media{
|
|
Base: model.Base{ID: "episode"},
|
|
LibraryID: "tv",
|
|
Title: "同号剧集",
|
|
Path: `/media/tv/同号剧集/episode.mkv`,
|
|
SeasonNum: 1,
|
|
EpisodeNum: 1,
|
|
TMDbID: 12345,
|
|
}
|
|
|
|
cards := groupMediaSeriesCards([]model.Media{movie, episode})
|
|
if len(cards) != 2 {
|
|
t.Fatalf("cards=%#v, want movie and TV item kept separate despite equal numeric TMDb IDs", cards)
|
|
}
|
|
}
|
|
|
|
func TestMediaSeriesKeyDoesNotUseGenericLibraryFolderAsSeriesTitle(t *testing.T) {
|
|
items := []model.Media{
|
|
{LibraryID: "tv", Path: `/media/电视剧/Alpha Show.S01E01.mkv`, SeasonNum: 1, EpisodeNum: 1},
|
|
{LibraryID: "tv", Path: `/media/电视剧/Beta Show.S01E01.mkv`, SeasonNum: 1, EpisodeNum: 1},
|
|
}
|
|
cards := groupMediaSeriesCards(items)
|
|
if len(cards) != 2 {
|
|
t.Fatalf("cards=%#v, want two independent series from a flat library root", cards)
|
|
}
|
|
}
|
|
|
|
func TestGroupMediaSeriesCardsBridgesReleaseFoldersByMatchedSeriesTitle(t *testing.T) {
|
|
items := []model.Media{
|
|
{LibraryID: "tv", Title: "同一部剧", ScrapeStatus: "matched", Path: `/media/电视剧/同一部剧 版本甲/Season 01/ep1.mkv`, SeasonNum: 1, EpisodeNum: 1},
|
|
{LibraryID: "tv", Title: "同一部剧", ScrapeStatus: "matched", Path: `/media/电视剧/同一部剧 版本乙/Season 01/ep2.mkv`, SeasonNum: 1, EpisodeNum: 2},
|
|
}
|
|
cards := groupMediaSeriesCards(items)
|
|
if len(cards) != 1 || cards[0].Count != 2 {
|
|
t.Fatalf("cards=%#v, want one series bridged by matched title", cards)
|
|
}
|
|
}
|
|
|
|
func TestGroupMediaSeriesCardsKeepsIndependentMoviesSeparateInSharedSubdirectory(t *testing.T) {
|
|
// 同一分类子目录下存放多部不同标题的独立电影，不应被强制折叠成 1 部
|
|
items := []model.Media{
|
|
{LibraryID: "movies", Title: "老师2024偷窥篇", Path: `/media/小姐姐/国产/nana/老师2024偷窥篇.strm`},
|
|
{LibraryID: "movies", Title: "紫光灯下的肉体诱惑", Path: `/media/小姐姐/国产/nana/紫光灯下的肉体诱惑.strm`},
|
|
{LibraryID: "movies", Title: "修洗衣机", Path: `/media/小姐姐/国产/nana/修洗衣机.strm`},
|
|
}
|
|
cards := groupMediaSeriesCards(items)
|
|
if len(cards) != 3 {
|
|
t.Fatalf("got %d cards, want 3 independent movie cards", len(cards))
|
|
}
|
|
|
|
// 但同一部电影的 CD1 和 CD2 仍应正确折叠为 1 部
|
|
cdItems := []model.Media{
|
|
{LibraryID: "movies", Title: "cd1", Path: `/media/电影/指环王 (2001)/cd1.mkv`},
|
|
{LibraryID: "movies", Title: "cd2", Path: `/media/电影/指环王 (2001)/cd2.mkv`},
|
|
}
|
|
cdCards := groupMediaSeriesCards(cdItems)
|
|
if len(cdCards) != 1 {
|
|
t.Fatalf("got %d cards for cd1/cd2, want 1 folded movie card", len(cdCards))
|
|
}
|
|
}
|