mirror of
https://github.com/truewhile/MeBox.git
synced 2026-09-28 11:16:37 +08:00
621 lines
21 KiB
Go
621 lines
21 KiB
Go
package handler
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"net/url"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/glebarez/sqlite"
|
|
"go.uber.org/zap"
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/truewhile/MeBox/internal/config"
|
|
"github.com/truewhile/MeBox/internal/middleware"
|
|
"github.com/truewhile/MeBox/internal/model"
|
|
"github.com/truewhile/MeBox/internal/repository"
|
|
"github.com/truewhile/MeBox/internal/service"
|
|
)
|
|
|
|
func TestQueueLibraryRootScanImportsNewLibraryMedia(t *testing.T) {
|
|
db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := db.AutoMigrate(&model.Library{}, &model.LibraryRoot{}, &model.Media{}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
rootPath := t.TempDir()
|
|
mediaPath := filepath.Join(rootPath, "Example Movie (2026).mkv")
|
|
if err := os.WriteFile(mediaPath, []byte("movie"), 0o644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
repos := repository.New(db)
|
|
lib := model.Library{Name: "电影", Path: rootPath, Type: "movie", Enabled: true}
|
|
root := model.LibraryRoot{Name: "电影", Path: rootPath, Enabled: true}
|
|
if err := repos.Library.CreateWithRoots(t.Context(), &lib, []model.LibraryRoot{root}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
libWithRoots, err := repos.Library.FindByID(t.Context(), lib.ID)
|
|
if err != nil || libWithRoots == nil || len(libWithRoots.Roots) != 1 {
|
|
t.Fatalf("library roots=%#v err=%v", libWithRoots, err)
|
|
}
|
|
scanner := service.NewScannerService(&config.Config{}, zap.NewNop(), repos, service.NewHub(zap.NewNop()), nil, nil)
|
|
svc := &service.Container{Repo: repos, Scan: scanner}
|
|
queueLibraryRootScan(svc, lib.ID, libWithRoots.Roots[0].ID)
|
|
|
|
deadline := time.Now().Add(3 * time.Second)
|
|
for time.Now().Before(deadline) {
|
|
var count int64
|
|
if err := repos.DB.Model(&model.Media{}).Where("library_id = ?", lib.ID).Count(&count).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if count == 1 {
|
|
return
|
|
}
|
|
time.Sleep(10 * time.Millisecond)
|
|
}
|
|
t.Fatal("initial library scan did not import the media file")
|
|
}
|
|
|
|
func TestListLibrariesHidesAdultDirectoriesUnlessAdminRequestsAll(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := db.AutoMigrate(&model.User{}, &model.Library{}, &model.Media{}, &model.Setting{}, &model.PlayProfile{}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
repos := repository.New(db)
|
|
viewer := &model.User{Username: "viewer", PasswordHash: "hash", Role: "admin", HideAdult: true}
|
|
if err := repos.User.Create(t.Context(), viewer); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
safe := model.Library{Name: "电影", Path: "/media/movie", Type: "movie", Enabled: true}
|
|
adult := model.Library{Name: "9KG", Path: "/media/9KG", Type: "movie", Enabled: true}
|
|
if err := repos.Library.Create(t.Context(), &safe); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := repos.Library.Create(t.Context(), &adult); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := repos.Setting.Set(t.Context(), service.AdultLibraryIDsSettingKey, `["`+adult.ID+`"]`); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := repos.Media.Upsert(t.Context(), &model.Media{LibraryID: safe.ID, Title: "误入普通库的成人条目", Path: "/media/movie/nsfw.mkv", NSFW: true}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
svc := &service.Container{
|
|
Repo: repos,
|
|
Media: service.NewMediaService(&config.Config{}, zap.NewNop(), repos),
|
|
}
|
|
|
|
visible := requestLibraries(t, svc, viewer.ID, "admin", "/api/libraries")
|
|
if len(visible) != 1 || visible[0].ID != safe.ID {
|
|
t.Fatalf("watching library list should hide adult directories, got %#v", visible)
|
|
}
|
|
|
|
all := requestLibraries(t, svc, viewer.ID, "admin", "/api/libraries?include_hidden=1")
|
|
if len(all) != 2 {
|
|
t.Fatalf("admin include_hidden list should keep management access, got %#v", all)
|
|
}
|
|
|
|
filtered := requestLibraries(t, svc, viewer.ID, "admin", "/api/libraries?include_hidden=1&ids="+safe.ID)
|
|
if len(filtered) != 1 || filtered[0].ID != safe.ID {
|
|
t.Fatalf("ids filter should return only requested library, got %#v", filtered)
|
|
}
|
|
}
|
|
|
|
func TestGetLibraryAllowsEmptyLibrary(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := db.AutoMigrate(&model.User{}, &model.Library{}, &model.Media{}, &model.Setting{}, &model.PlayProfile{}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
repos := repository.New(db)
|
|
lib := model.Library{Name: "空媒体库", Path: "/media/empty", Type: "movie", Enabled: true}
|
|
if err := repos.Library.Create(t.Context(), &lib); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
svc := &service.Container{
|
|
Repo: repos,
|
|
Media: service.NewMediaService(&config.Config{}, zap.NewNop(), repos),
|
|
}
|
|
|
|
got := requestLibrary(t, svc, "user-1", "user", "/api/libraries/"+lib.ID, lib.ID)
|
|
if got.ID != lib.ID || got.Name != "空媒体库" {
|
|
t.Fatalf("library detail = %#v, want empty library detail", got)
|
|
}
|
|
media := requestMediaList(t, svc, "/api/libraries/"+lib.ID+"/media", lib.ID)
|
|
if media.Total != 0 || len(media.Items) != 0 {
|
|
t.Fatalf("empty library media = %#v, want no items", media)
|
|
}
|
|
}
|
|
|
|
func TestListMediaGroupsMultipleVersionsByDefault(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := db.AutoMigrate(&model.User{}, &model.Library{}, &model.Media{}, &model.Setting{}, &model.PlayProfile{}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
repos := repository.New(db)
|
|
lib := model.Library{Name: "Movies", Path: "/media/movies", Type: "movie", Enabled: true}
|
|
if err := repos.Library.Create(t.Context(), &lib); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := repos.DB.Create(&[]model.Media{
|
|
{
|
|
Base: model.Base{ID: "movie-1080", CreatedAt: time.Now().Add(-time.Minute)},
|
|
LibraryID: lib.ID,
|
|
Title: "流浪地球",
|
|
Path: "/media/movies/The.Wandering.Earth.2019.1080p.mkv",
|
|
Year: 2019,
|
|
Width: 1920,
|
|
Height: 1080,
|
|
SizeBytes: 100,
|
|
},
|
|
{
|
|
Base: model.Base{ID: "movie-2160", CreatedAt: time.Now()},
|
|
LibraryID: lib.ID,
|
|
Title: "流浪地球",
|
|
Path: "cloud://openlist/Movies/The.Wandering.Earth.2019.2160p.mkv",
|
|
Year: 2019,
|
|
Width: 3840,
|
|
Height: 2160,
|
|
SizeBytes: 200,
|
|
},
|
|
}).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
svc := &service.Container{
|
|
Repo: repos,
|
|
Media: service.NewMediaService(&config.Config{}, zap.NewNop(), repos),
|
|
}
|
|
|
|
grouped := requestMediaList(t, svc, "/api/libraries/"+lib.ID+"/media", lib.ID)
|
|
if grouped.Total != 1 || len(grouped.Items) != 1 {
|
|
t.Fatalf("grouped response total=%d len=%d body=%#v", grouped.Total, len(grouped.Items), grouped)
|
|
}
|
|
if grouped.Items[0].ID != "movie-1080" {
|
|
t.Fatalf("primary id = %q, want local version to remain primary", grouped.Items[0].ID)
|
|
}
|
|
if len(grouped.Items[0].Versions) != 2 {
|
|
t.Fatalf("versions = %#v, want both versions", grouped.Items[0].Versions)
|
|
}
|
|
if grouped.Items[0].Versions[0].ID != "movie-1080" || grouped.Items[0].Versions[1].ID != "movie-2160" {
|
|
t.Fatalf("versions should keep local before cloud: %#v", grouped.Items[0].Versions)
|
|
}
|
|
|
|
raw := requestMediaList(t, svc, "/api/libraries/"+lib.ID+"/media?group_versions=0", lib.ID)
|
|
if raw.Total != 2 || len(raw.Items) != 2 {
|
|
t.Fatalf("raw response total=%d len=%d body=%#v", raw.Total, len(raw.Items), raw)
|
|
}
|
|
}
|
|
|
|
func TestListLibrarySeriesDoesNotTruncateLargeEpisodeLibraries(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := db.AutoMigrate(&model.User{}, &model.Library{}, &model.Media{}, &model.Setting{}, &model.PlayProfile{}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
repos := repository.New(db)
|
|
lib := model.Library{Name: "国漫", Path: "cloud://openlist/国漫", Type: "anime", Enabled: true}
|
|
if err := repos.Library.Create(t.Context(), &lib); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
rows := make([]model.Media, 0, 2001)
|
|
for i := 1; i <= 2001; i++ {
|
|
rows = append(rows, model.Media{
|
|
Base: model.Base{ID: fmt.Sprintf("ep-%04d", i), CreatedAt: time.Now().Add(time.Duration(i) * time.Second)},
|
|
LibraryID: lib.ID,
|
|
Title: "大剧",
|
|
Path: fmt.Sprintf("cloud://openlist/国漫/大剧 (2026) {tmdb-123}/Season 1/大剧.S01E%04d.mkv", i),
|
|
SeasonNum: 1,
|
|
EpisodeNum: i,
|
|
})
|
|
}
|
|
if err := repos.DB.CreateInBatches(rows, 500).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
svc := &service.Container{
|
|
Repo: repos,
|
|
Media: service.NewMediaService(&config.Config{}, zap.NewNop(), repos),
|
|
}
|
|
|
|
series := requestLibrarySeries(t, svc, "/api/libraries/"+lib.ID+"/series", lib.ID)
|
|
if series.Total != 1 || len(series.Items) != 1 {
|
|
t.Fatalf("series response total=%d len=%d body=%#v", series.Total, len(series.Items), series)
|
|
}
|
|
if series.Items[0].Count != 2001 {
|
|
t.Fatalf("series count = %d, want 2001", series.Items[0].Count)
|
|
}
|
|
if !strings.HasPrefix(series.Items[0].Key, "series:") ||
|
|
strings.Contains(series.Items[0].Key, "lib:") ||
|
|
strings.Contains(series.Items[0].Key, "show:") {
|
|
t.Fatalf("series key = %q, want compact non-raw key", series.Items[0].Key)
|
|
}
|
|
episodes := requestLibrarySeriesEpisodes(t, svc, "/api/libraries/"+lib.ID+"/series/episodes?key="+url.QueryEscape(series.Items[0].Key), lib.ID)
|
|
if episodes.Total != 2001 || len(episodes.Items) != 2001 {
|
|
t.Fatalf("episodes total=%d len=%d, want 2001", episodes.Total, len(episodes.Items))
|
|
}
|
|
if episodes.Items[0].EpisodeNum != 1 || episodes.Items[len(episodes.Items)-1].EpisodeNum != 2001 {
|
|
t.Fatalf("episode order first=%d last=%d", episodes.Items[0].EpisodeNum, episodes.Items[len(episodes.Items)-1].EpisodeNum)
|
|
}
|
|
}
|
|
|
|
func TestScrapeOptionsFromRequestPreservesEpisodeImagesFalse(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
c.Request = httptest.NewRequest(http.MethodPost, "/api/media/ep-1/scrape", bytes.NewBufferString(`{"episode_images":false,"refresh_matched":true}`))
|
|
c.Request.Header.Set("Content-Type", "application/json")
|
|
|
|
options, err := scrapeOptionsFromRequest(c, false)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if options.EpisodeArtwork == nil {
|
|
t.Fatal("EpisodeArtwork is nil, want explicit false")
|
|
}
|
|
if *options.EpisodeArtwork {
|
|
t.Fatal("EpisodeArtwork = true, want false")
|
|
}
|
|
if !options.IncludeMatched {
|
|
t.Fatal("IncludeMatched = false, want true from refresh_matched")
|
|
}
|
|
}
|
|
|
|
func requestLibraries(t *testing.T, svc *service.Container, userID, role, path string) []model.Library {
|
|
t.Helper()
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
c.Set(middleware.CtxUserID, userID)
|
|
c.Set(middleware.CtxUserRole, role)
|
|
c.Request = httptest.NewRequest(http.MethodGet, path, nil)
|
|
listLibrariesHandler(svc)(c)
|
|
if w.Code != http.StatusOK {
|
|
t.Fatalf("GET %s status = %d body=%s", path, w.Code, w.Body.String())
|
|
}
|
|
var libs []model.Library
|
|
if err := json.Unmarshal(w.Body.Bytes(), &libs); err != nil {
|
|
t.Fatalf("decode libraries: %v", err)
|
|
}
|
|
return libs
|
|
}
|
|
|
|
func requestLibrary(t *testing.T, svc *service.Container, userID, role, path, libraryID string) model.Library {
|
|
t.Helper()
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
c.Set(middleware.CtxUserID, userID)
|
|
c.Set(middleware.CtxUserRole, role)
|
|
c.Params = gin.Params{{Key: "id", Value: libraryID}}
|
|
c.Request = httptest.NewRequest(http.MethodGet, path, nil)
|
|
getLibraryHandler(svc)(c)
|
|
if w.Code != http.StatusOK {
|
|
t.Fatalf("GET %s status = %d body=%s", path, w.Code, w.Body.String())
|
|
}
|
|
var lib model.Library
|
|
if err := json.Unmarshal(w.Body.Bytes(), &lib); err != nil {
|
|
t.Fatalf("decode library: %v", err)
|
|
}
|
|
return lib
|
|
}
|
|
|
|
type mediaListResponse struct {
|
|
Items []service.MediaItem `json:"items"`
|
|
Total int64 `json:"total"`
|
|
}
|
|
|
|
type seriesListResponse struct {
|
|
Items []service.SeriesCard `json:"items"`
|
|
Total int64 `json:"total"`
|
|
}
|
|
|
|
type seriesEpisodesResponse struct {
|
|
Items []model.Media `json:"items"`
|
|
Total int64 `json:"total"`
|
|
}
|
|
|
|
func requestMediaList(t *testing.T, svc *service.Container, path, libraryID string) mediaListResponse {
|
|
t.Helper()
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
c.Set(middleware.CtxUserID, "user-1")
|
|
c.Set(middleware.CtxUserRole, "user")
|
|
c.Params = gin.Params{{Key: "id", Value: libraryID}}
|
|
c.Request = httptest.NewRequest(http.MethodGet, path, nil)
|
|
listMediaHandler(svc)(c)
|
|
if w.Code != http.StatusOK {
|
|
t.Fatalf("GET %s status = %d body=%s", path, w.Code, w.Body.String())
|
|
}
|
|
var payload mediaListResponse
|
|
if err := json.Unmarshal(w.Body.Bytes(), &payload); err != nil {
|
|
t.Fatalf("decode media list: %v", err)
|
|
}
|
|
return payload
|
|
}
|
|
|
|
func requestLibrarySeries(t *testing.T, svc *service.Container, path, libraryID string) seriesListResponse {
|
|
t.Helper()
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
c.Set(middleware.CtxUserID, "user-1")
|
|
c.Set(middleware.CtxUserRole, "user")
|
|
c.Params = gin.Params{{Key: "id", Value: libraryID}}
|
|
c.Request = httptest.NewRequest(http.MethodGet, path, nil)
|
|
listLibrarySeriesHandler(svc)(c)
|
|
if w.Code != http.StatusOK {
|
|
t.Fatalf("GET %s status = %d body=%s", path, w.Code, w.Body.String())
|
|
}
|
|
var payload seriesListResponse
|
|
if err := json.Unmarshal(w.Body.Bytes(), &payload); err != nil {
|
|
t.Fatalf("decode series list: %v", err)
|
|
}
|
|
return payload
|
|
}
|
|
|
|
func requestLibrarySeriesEpisodes(t *testing.T, svc *service.Container, path, libraryID string) seriesEpisodesResponse {
|
|
t.Helper()
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
c.Set(middleware.CtxUserID, "user-1")
|
|
c.Set(middleware.CtxUserRole, "user")
|
|
c.Params = gin.Params{{Key: "id", Value: libraryID}}
|
|
c.Request = httptest.NewRequest(http.MethodGet, path, nil)
|
|
listLibrarySeriesEpisodesHandler(svc)(c)
|
|
if w.Code != http.StatusOK {
|
|
t.Fatalf("GET %s status = %d body=%s", path, w.Code, w.Body.String())
|
|
}
|
|
var payload seriesEpisodesResponse
|
|
if err := json.Unmarshal(w.Body.Bytes(), &payload); err != nil {
|
|
t.Fatalf("decode series episodes: %v", err)
|
|
}
|
|
return payload
|
|
}
|
|
|
|
// 空库的 media / series 列表必须返回 "items":[](而不是 Go nil slice 序列化出的 null)。
|
|
// 前端 [].concat(null) 会得到 [null],随后在渲染期解引用 null 崩溃,导致「空库点进去
|
|
// 报错且无法返回」。这条测试钉死该 JSON 契约,防止再退化。
|
|
func TestEmptyLibraryListsReturnEmptyArraysNotNull(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := db.AutoMigrate(&model.Library{}, &model.Media{}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
repos := repository.New(db)
|
|
movie := model.Library{Name: "空电影库", Path: "/media/empty-movie", Type: "movie", Enabled: true}
|
|
tv := model.Library{Name: "空剧集库", Path: "/media/empty-tv", Type: "tv", Enabled: true}
|
|
for _, lib := range []*model.Library{&movie, &tv} {
|
|
if err := repos.Library.Create(t.Context(), lib); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
svc := &service.Container{
|
|
Repo: repos,
|
|
Media: service.NewMediaService(&config.Config{}, zap.NewNop(), repos),
|
|
}
|
|
|
|
cases := []struct {
|
|
name string
|
|
path string
|
|
lib string
|
|
handler gin.HandlerFunc
|
|
}{
|
|
{"media", "/api/libraries/" + movie.ID + "/media", movie.ID, listMediaHandler(svc)},
|
|
{"series", "/api/libraries/" + tv.ID + "/series", tv.ID, listLibrarySeriesHandler(svc)},
|
|
}
|
|
for _, tc := range cases {
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
c.Set(middleware.CtxUserID, "user-1")
|
|
c.Set(middleware.CtxUserRole, "user")
|
|
c.Params = gin.Params{{Key: "id", Value: tc.lib}}
|
|
c.Request = httptest.NewRequest(http.MethodGet, tc.path, nil)
|
|
tc.handler(c)
|
|
if w.Code != http.StatusOK {
|
|
t.Fatalf("%s: status = %d body=%s", tc.name, w.Code, w.Body.String())
|
|
}
|
|
body := w.Body.String()
|
|
if strings.Contains(body, `"items":null`) {
|
|
t.Fatalf("%s: empty library returned items:null (crashes frontend): %s", tc.name, body)
|
|
}
|
|
if !strings.Contains(body, `"items":[]`) {
|
|
t.Fatalf("%s: expected items:[] for empty library, got %s", tc.name, body)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestSearchMediaHandlerIncludesEmbyRemote(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
q := r.URL.Query()
|
|
if q.Get("SearchTerm") == "碧蓝之海" {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
_ = json.NewEncoder(w).Encode(map[string]any{
|
|
"TotalRecordCount": 1,
|
|
"Items": []map[string]any{
|
|
{
|
|
"Id": "156030",
|
|
"Name": "碧蓝之海",
|
|
"Type": "Series",
|
|
"ProductionYear": 2018,
|
|
},
|
|
},
|
|
})
|
|
return
|
|
}
|
|
w.Header().Set("Content-Type", "application/json")
|
|
_ = json.NewEncoder(w).Encode(map[string]any{
|
|
"TotalRecordCount": 0,
|
|
"Items": []map[string]any{},
|
|
})
|
|
}))
|
|
defer server.Close()
|
|
|
|
db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := db.AutoMigrate(&model.Library{}, &model.Media{}, &model.StrmAccount{}, &model.EmbyMount{}, &model.Setting{}, &model.User{}, &model.PlayProfile{}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
repos := repository.New(db)
|
|
adminUser := model.User{
|
|
Base: model.Base{ID: "user-1"},
|
|
Username: "admin",
|
|
Role: "admin",
|
|
}
|
|
_ = repos.DB.Create(&adminUser).Error
|
|
|
|
localLib := model.Library{Name: "本地电影", Path: "/media/movies", Type: "movie", Enabled: true}
|
|
if err := repos.Library.Create(t.Context(), &localLib); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
localMedia := model.Media{
|
|
Base: model.Base{ID: "local-1"},
|
|
LibraryID: localLib.ID,
|
|
Title: "流浪地球",
|
|
Year: 2019,
|
|
}
|
|
if err := repos.DB.Create(&localMedia).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
rawCfg, _ := json.Marshal(map[string]string{"url": server.URL, "token": "fake-token"})
|
|
acct := model.StrmAccount{
|
|
Base: model.Base{ID: "acct-1"},
|
|
Name: "远程Emby",
|
|
Provider: model.StrmProviderEmbyRemote,
|
|
Config: string(rawCfg),
|
|
Enabled: true,
|
|
}
|
|
if err := repos.StrmAccount.Create(t.Context(), &acct); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
mount := model.EmbyMount{
|
|
Base: model.Base{ID: "mount-1"},
|
|
AccountID: acct.ID,
|
|
RemoteViewID: "view-1",
|
|
RemoteViewName: "动漫",
|
|
CollectionType: "tvshows",
|
|
Enabled: true,
|
|
}
|
|
if err := repos.EmbyMount.Create(t.Context(), &mount); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
crypto := service.NewCryptoService("", zap.NewNop())
|
|
remoteSvc := service.NewEmbyRemoteService(&config.Config{}, zap.NewNop(), repos, crypto)
|
|
mediaSvc := service.NewMediaService(&config.Config{}, zap.NewNop(), repos)
|
|
|
|
svc := &service.Container{
|
|
Repo: repos,
|
|
Media: mediaSvc,
|
|
EmbyRemote: remoteSvc,
|
|
}
|
|
|
|
// 1. 搜索远程挂载媒体(碧蓝之海)
|
|
{
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
c.Set(middleware.CtxUserID, "user-1")
|
|
c.Set(middleware.CtxUserRole, "admin")
|
|
c.Request = httptest.NewRequest(http.MethodGet, "/api/media?q=碧蓝之海&limit=8", nil)
|
|
searchMediaHandler(svc)(c)
|
|
|
|
if w.Code != http.StatusOK {
|
|
t.Fatalf("search status=%d, body=%s", w.Code, w.Body.String())
|
|
}
|
|
var res struct {
|
|
Items []service.MediaItem `json:"items"`
|
|
}
|
|
if err := json.Unmarshal(w.Body.Bytes(), &res); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(res.Items) != 1 {
|
|
t.Fatalf("expected 1 item, got %d", len(res.Items))
|
|
}
|
|
if res.Items[0].Title != "碧蓝之海" {
|
|
t.Fatalf("expected Title '碧蓝之海', got %q", res.Items[0].Title)
|
|
}
|
|
expectedID := service.EncodeEmbyRemoteID("mount-1", "156030")
|
|
if res.Items[0].ID != expectedID {
|
|
t.Fatalf("expected ID %q, got %q", expectedID, res.Items[0].ID)
|
|
}
|
|
}
|
|
|
|
// 2. 搜索本地媒体(流浪地球)
|
|
{
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
c.Set(middleware.CtxUserID, "user-1")
|
|
c.Set(middleware.CtxUserRole, "admin")
|
|
c.Request = httptest.NewRequest(http.MethodGet, "/api/media?q=流浪地球&limit=8", nil)
|
|
searchMediaHandler(svc)(c)
|
|
|
|
if w.Code != http.StatusOK {
|
|
t.Fatalf("search status=%d, body=%s", w.Code, w.Body.String())
|
|
}
|
|
var res struct {
|
|
Items []service.MediaItem `json:"items"`
|
|
}
|
|
if err := json.Unmarshal(w.Body.Bytes(), &res); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(res.Items) != 1 {
|
|
t.Fatalf("expected 1 item, got %d", len(res.Items))
|
|
}
|
|
if res.Items[0].Title != "流浪地球" {
|
|
t.Fatalf("expected Title '流浪地球', got %q", res.Items[0].Title)
|
|
}
|
|
}
|
|
|
|
// 3. 搜索不存在的媒体
|
|
{
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
c.Set(middleware.CtxUserID, "user-1")
|
|
c.Set(middleware.CtxUserRole, "admin")
|
|
c.Request = httptest.NewRequest(http.MethodGet, "/api/media?q=不存在的影片&limit=8", nil)
|
|
searchMediaHandler(svc)(c)
|
|
|
|
if w.Code != http.StatusOK {
|
|
t.Fatalf("search status=%d, body=%s", w.Code, w.Body.String())
|
|
}
|
|
var res struct {
|
|
Items []service.MediaItem `json:"items"`
|
|
}
|
|
if err := json.Unmarshal(w.Body.Bytes(), &res); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(res.Items) != 0 {
|
|
t.Fatalf("expected 0 items, got %d", len(res.Items))
|
|
}
|
|
if strings.Contains(w.Body.String(), `"items":null`) {
|
|
t.Fatalf("expected items:[], got null: %s", w.Body.String())
|
|
}
|
|
}
|
|
}
|