package service import ( "encoding/json" "errors" "net/http" "net/http/httptest" "os" "path/filepath" "strings" "sync/atomic" "testing" "github.com/glebarez/sqlite" "go.uber.org/zap" "gorm.io/gorm" "github.com/ShukeBta/MediaStationGo/internal/config" "github.com/ShukeBta/MediaStationGo/internal/model" "github.com/ShukeBta/MediaStationGo/internal/repository" ) 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 TestDownloadCompleteAutoOrganizesContentPath(t *testing.T) { root := t.TempDir() src := filepath.Join(root, "downloads", "国产剧", "狂飙.S01E01.2023.1080p.mkv") dest := filepath.Join(root, "media") if err := os.MkdirAll(filepath.Dir(src), 0o755); err != nil { t.Fatal(err) } if err := os.WriteFile(src, []byte("episode"), 0o644); err != nil { t.Fatal(err) } repos := newOrganizerTestRepo(t) for key, value := range map[string]string{ "organizer.auto_after_download": "true", "organize.target_dir": dest, "organize.transfer_mode": "copy", } { if err := repos.Setting.Set(t.Context(), key, value); err != nil { t.Fatal(err) } } org := NewOrganizerService(&config.Config{}, zap.NewNop(), repos) svc := NewDownloadService(zap.NewNop(), repos, NewHub(zap.NewNop()), org) svc.onTorrentComplete(t.Context(), QBitTorrent{ Hash: "done123", Name: "狂飙.S01E01.2023.1080p", Progress: 1, SavePath: filepath.Join(root, "downloads", "国产剧"), ContentPath: src, }) want := filepath.Join(dest, "电视剧", "国产剧", "狂飙", "Season 01", "狂飙 - S01E01.mkv") if _, err := os.Stat(want); err != nil { t.Fatalf("auto organized file missing at %q: %v", want, err) } if _, err := os.Stat(src); err != nil { t.Fatalf("copy mode should keep source: %v", err) } } func TestPublicDownloadTitleUsesMagnetDisplayName(t *testing.T) { got := publicDownloadTitle("magnet:?xt=urn:btih:abc&dn=%E6%B5%8B%E8%AF%95%E5%BD%B1%E7%89%87") if got != "测试影片" { t.Fatalf("publicDownloadTitle = %q, want %q", got, "测试影片") } } func configureTestDefaultQB(t *testing.T, repos *repository.Container, baseURL string) { t.Helper() if err := repos.DownloadClient.Create(t.Context(), &model.DownloadClient{ Name: "qB test", Type: "qbittorrent", Host: baseURL, Username: "admin", Password: "admin", IsDefault: true, Enabled: true, }); err != nil { t.Fatalf("create default qB client: %v", err) } if err := repos.Setting.Set(t.Context(), settingDownloadClientsManaged, "true"); err != nil { t.Fatalf("mark download clients managed: %v", err) } } func TestAddDownloadWithMetaSkipsExistingTaskBeforeQBAdd(t *testing.T) { var addCalls int32 qb := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { switch r.URL.Path { case "/api/v2/auth/login": _, _ = w.Write([]byte("Ok.")) case "/api/v2/torrents/info": _, _ = w.Write([]byte(`[]`)) case "/api/v2/torrents/add": atomic.AddInt32(&addCalls, 1) _, _ = w.Write([]byte("Ok.")) default: http.NotFound(w, r) } })) defer qb.Close() db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{}) if err != nil { t.Fatal(err) } if err := db.AutoMigrate(&model.DownloadTask{}, &model.Setting{}); err != nil { t.Fatal(err) } repos := repository.New(db) existing := &model.DownloadTask{ UserID: "u1", Source: "qbittorrent", URL: "https://pt.example/download?id=old&passkey=old", Title: "Some Show S01E01 1080p", SavePath: "/downloads/tv", Status: "completed", Progress: 1, } if err := repos.Download.Create(t.Context(), existing); err != nil { t.Fatal(err) } svc := NewDownloadService(zap.NewNop(), repos, NewHub(zap.NewNop()), nil) svc.qb.Configure(QBitConfig{BaseURL: qb.URL, Username: "admin", Password: "admin"}) task, err := svc.AddDownloadWithMeta(t.Context(), "u1", "https://pt.example/download?id=new&passkey=new", "/downloads/tv", DownloadTaskMeta{ Title: "Some Show S01E01 1080p", }) if !errors.Is(err, ErrDownloadAlreadyExists) { t.Fatalf("err = %v, want ErrDownloadAlreadyExists", err) } if task == nil || task.ID != existing.ID { t.Fatalf("task = %#v, want existing task %#v", task, existing) } if got := atomic.LoadInt32(&addCalls); got != 0 { t.Fatalf("qb add calls = %d, want 0", got) } } func TestAddDownloadWithMetaSkipsUserDeletedTaskBeforeQBAdd(t *testing.T) { var addCalls int32 qb := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { switch r.URL.Path { case "/api/v2/auth/login": _, _ = w.Write([]byte("Ok.")) case "/api/v2/torrents/info": _, _ = w.Write([]byte(`[]`)) case "/api/v2/torrents/add": atomic.AddInt32(&addCalls, 1) _, _ = w.Write([]byte("Ok.")) default: http.NotFound(w, r) } })) defer qb.Close() db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{}) if err != nil { t.Fatal(err) } if err := db.AutoMigrate(&model.DownloadTask{}, &model.Media{}, &model.Setting{}); err != nil { t.Fatal(err) } repos := repository.New(db) if err := repos.Setting.Set(t.Context(), "qbittorrent.url", qb.URL); err != nil { t.Fatal(err) } if err := repos.Setting.Set(t.Context(), "qbittorrent.username", "admin"); err != nil { t.Fatal(err) } if err := repos.Setting.Set(t.Context(), "qbittorrent.password", "admin"); err != nil { t.Fatal(err) } existing := &model.DownloadTask{ UserID: "u1", Source: "qbittorrent", URL: "https://pt.example/download?id=old&passkey=old", Title: "User Deleted Show S01E01 1080p", SavePath: "/downloads/tv", Status: "deleted", } if err := repos.Download.Create(t.Context(), existing); err != nil { t.Fatal(err) } svc := NewDownloadService(zap.NewNop(), repos, NewHub(zap.NewNop()), nil) task, err := svc.AddDownloadWithMeta(t.Context(), "u1", "https://pt.example/download?id=new&passkey=new", "/downloads/tv", DownloadTaskMeta{ Title: "User Deleted Show S01E01 1080p WEB-DL", }) if !errors.Is(err, ErrDownloadAlreadyExists) { t.Fatalf("err = %v, want ErrDownloadAlreadyExists", err) } if task == nil || task.ID != existing.ID { t.Fatalf("task = %#v, want existing task %#v", task, existing) } if got := atomic.LoadInt32(&addCalls); got != 0 { t.Fatalf("qb add calls = %d, want 0", got) } } func TestDeleteMarksMatchingDownloadTaskDeleted(t *testing.T) { const hash = "abc123" const title = "Delete Marker Show S01E01 1080p" var deleteCalls int32 qb := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { switch r.URL.Path { case "/api/v2/auth/login": _, _ = w.Write([]byte("Ok.")) case "/api/v2/torrents/info": _, _ = w.Write([]byte(`[{"hash":"abc123","name":"Delete Marker Show S01E01 1080p","state":"downloading","progress":0.5}]`)) case "/api/v2/torrents/delete": atomic.AddInt32(&deleteCalls, 1) _, _ = w.Write([]byte("Ok.")) default: http.NotFound(w, r) } })) defer qb.Close() db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{}) if err != nil { t.Fatal(err) } if err := db.AutoMigrate(&model.DownloadTask{}, &model.DownloadClient{}, &model.Setting{}); err != nil { t.Fatal(err) } repos := repository.New(db) configureTestDefaultQB(t, repos, qb.URL) task := &model.DownloadTask{ UserID: "u1", Source: "qbittorrent", URL: "https://pt.example/download?id=1", Title: title, SavePath: "/downloads/tv", Status: "downloading", Progress: 0.5, } if err := repos.Download.Create(t.Context(), task); err != nil { t.Fatal(err) } svc := NewDownloadService(zap.NewNop(), repos, NewHub(zap.NewNop()), nil) if err := svc.ReloadConfig(t.Context()); err != nil { t.Fatal(err) } if err := svc.Delete(t.Context(), hash, false); err != nil { t.Fatal(err) } if got := atomic.LoadInt32(&deleteCalls); got != 1 { t.Fatalf("delete calls = %d, want 1", got) } var updated model.DownloadTask if err := db.Where("id = ?", task.ID).First(&updated).Error; err != nil { t.Fatal(err) } if updated.Status != "deleted" { t.Fatalf("status = %q, want deleted", updated.Status) } } func TestAddDownloadWithMetaSkipsExistingLocalMovieBeforeQBAdd(t *testing.T) { db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{}) if err != nil { t.Fatal(err) } if err := db.AutoMigrate(&model.Media{}, &model.DownloadTask{}, &model.Setting{}); err != nil { t.Fatal(err) } repos := repository.New(db) if err := db.Create(&model.Media{ Title: "Inception", Path: "/media/movies/Inception (2010)/Inception (2010).mkv", }).Error; err != nil { t.Fatal(err) } svc := NewDownloadService(zap.NewNop(), repos, NewHub(zap.NewNop()), nil) task, err := svc.AddDownloadWithMeta(t.Context(), "u1", "magnet:?xt=urn:btih:cccccccccccccccccccccccccccccccccccccccc&dn=Inception+2010+1080p", "/downloads", DownloadTaskMeta{ Title: "Inception 2010 1080p WEB-DL", }) if !errors.Is(err, ErrMediaAlreadyInLibrary) { t.Fatalf("err = %v, want ErrMediaAlreadyInLibrary", err) } if task != nil { t.Fatalf("task = %#v, want nil because local media already exists", task) } rows, err := repos.Download.List(t.Context()) if err != nil { t.Fatal(err) } if len(rows) != 0 { t.Fatalf("download rows = %d, want 0", len(rows)) } } func TestAddDownloadWithMetaSkipsExistingLocalEpisodeBeforeQBAdd(t *testing.T) { db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{}) if err != nil { t.Fatal(err) } if err := db.AutoMigrate(&model.Media{}, &model.DownloadTask{}, &model.Setting{}); err != nil { t.Fatal(err) } repos := repository.New(db) if err := db.Create(&model.Media{ Title: "Some Show", Path: "/media/tv/Some Show/Season 01/Some Show - S01E01.mkv", SeasonNum: 1, EpisodeNum: 1, }).Error; err != nil { t.Fatal(err) } svc := NewDownloadService(zap.NewNop(), repos, NewHub(zap.NewNop()), nil) task, err := svc.AddDownloadWithMeta(t.Context(), "u1", "magnet:?xt=urn:btih:dddddddddddddddddddddddddddddddddddddddd&dn=Some+Show+S01E01", "/downloads", DownloadTaskMeta{ Title: "Some Show S01E01 2160p WEB-DL", }) if !errors.Is(err, ErrMediaAlreadyInLibrary) { t.Fatalf("err = %v, want ErrMediaAlreadyInLibrary", err) } if task != nil { t.Fatalf("task = %#v, want nil because local episode already exists", task) } rows, err := repos.Download.List(t.Context()) if err != nil { t.Fatal(err) } if len(rows) != 0 { t.Fatalf("download rows = %d, want 0", len(rows)) } } func TestReloadConfigDoesNotFallbackToLegacyAfterClientDeleted(t *testing.T) { var addCalls int32 qb := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { switch r.URL.Path { case "/api/v2/auth/login": _, _ = w.Write([]byte("Ok.")) case "/api/v2/torrents/info": if atomic.LoadInt32(&addCalls) > 0 { _, _ = w.Write([]byte(`[{"hash":"abc123","name":"Movie 2026 1080p","state":"downloading","progress":0.1}]`)) return } _, _ = w.Write([]byte(`[]`)) case "/api/v2/torrents/add": atomic.AddInt32(&addCalls, 1) _, _ = w.Write([]byte("Ok.")) default: http.NotFound(w, r) } })) defer qb.Close() db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{}) if err != nil { t.Fatal(err) } if err := db.AutoMigrate(&model.DownloadClient{}, &model.DownloadTask{}, &model.Setting{}); err != nil { t.Fatal(err) } repos := repository.New(db) if err := repos.Setting.Set(t.Context(), "qbittorrent.url", qb.URL); err != nil { t.Fatal(err) } if err := repos.Setting.Set(t.Context(), "qbittorrent.username", "admin"); err != nil { t.Fatal(err) } if err := repos.Setting.Set(t.Context(), "qbittorrent.password", "admin"); err != nil { t.Fatal(err) } client := &model.DownloadClient{Name: "qB", Type: "qbittorrent", Host: qb.URL, Username: "admin", Password: "admin", IsDefault: true, Enabled: true} if err := repos.DownloadClient.Create(t.Context(), client); err != nil { t.Fatal(err) } if err := repos.DownloadClient.Delete(t.Context(), client.ID); err != nil { t.Fatal(err) } svc := NewDownloadService(zap.NewNop(), repos, NewHub(zap.NewNop()), nil) if err := svc.ReloadConfig(t.Context()); err != nil { t.Fatal(err) } _, err = svc.AddDownloadWithMeta(t.Context(), "u1", "magnet:?xt=urn:btih:aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa&dn=Movie+2026+1080p", "/downloads", DownloadTaskMeta{ Title: "Movie 2026 1080p", }) if err == nil { t.Fatal("expected add to fail when the configured downloader was deleted") } if got := atomic.LoadInt32(&addCalls); got != 0 { t.Fatalf("qb add calls = %d, want 0", got) } } func TestAddDownloadWithMetaAutoClassifiesSavePathAndQBitCategory(t *testing.T) { var addCalls int32 var gotSavePath string var gotCategory string qb := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { switch r.URL.Path { case "/api/v2/auth/login": _, _ = w.Write([]byte("Ok.")) case "/api/v2/torrents/info": if atomic.LoadInt32(&addCalls) > 0 { _, _ = w.Write([]byte(`[{"hash":"auto123","name":"声生不息 S01E01","state":"downloading","progress":0.1}]`)) return } _, _ = w.Write([]byte(`[]`)) case "/api/v2/torrents/add": atomic.AddInt32(&addCalls, 1) if err := r.ParseMultipartForm(1024 * 1024); err != nil { http.Error(w, err.Error(), http.StatusBadRequest) return } gotSavePath = r.FormValue("savepath") gotCategory = r.FormValue("category") _, _ = w.Write([]byte("Ok.")) default: http.NotFound(w, r) } })) defer qb.Close() db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{}) if err != nil { t.Fatal(err) } if err := db.AutoMigrate(&model.DownloadClient{}, &model.DownloadTask{}, &model.Setting{}); err != nil { t.Fatal(err) } repos := repository.New(db) configureTestDefaultQB(t, repos, qb.URL) if err := repos.Setting.Set(t.Context(), "qbittorrent.savepath", "/downloads"); err != nil { t.Fatal(err) } svc := NewDownloadService(zap.NewNop(), repos, NewHub(zap.NewNop()), nil) task, err := svc.AddDownloadWithMeta(t.Context(), "u1", "magnet:?xt=urn:btih:eeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee&dn=%E5%A3%B0%E7%94%9F%E4%B8%8D%E6%81%AF+S01E01", "", DownloadTaskMeta{ Title: "声生不息 S01E01", SourceCategory: "综艺", }) if err != nil { t.Fatal(err) } wantPath := filepath.Join("/downloads", "综艺") if task.SavePath != wantPath { t.Fatalf("task save path = %q, want %q", task.SavePath, wantPath) } if gotSavePath != wantPath { t.Fatalf("qb savepath = %q, want %q", gotSavePath, wantPath) } if gotCategory != "综艺" { t.Fatalf("qb category = %q, want 综艺", gotCategory) } } func TestAddDownloadWithMetaCanDisableAutoClassifiedSavePath(t *testing.T) { var addCalls int32 var gotSavePath string var gotCategory string qb := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { switch r.URL.Path { case "/api/v2/auth/login": _, _ = w.Write([]byte("Ok.")) case "/api/v2/torrents/info": if atomic.LoadInt32(&addCalls) > 0 { _, _ = w.Write([]byte(`[{"hash":"auto456","name":"声生不息 S01E01","state":"downloading","progress":0.1}]`)) return } _, _ = w.Write([]byte(`[]`)) case "/api/v2/torrents/add": atomic.AddInt32(&addCalls, 1) if err := r.ParseMultipartForm(1024 * 1024); err != nil { http.Error(w, err.Error(), http.StatusBadRequest) return } gotSavePath = r.FormValue("savepath") gotCategory = r.FormValue("category") _, _ = w.Write([]byte("Ok.")) default: http.NotFound(w, r) } })) defer qb.Close() db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{}) if err != nil { t.Fatal(err) } if err := db.AutoMigrate(&model.DownloadClient{}, &model.DownloadTask{}, &model.Setting{}); err != nil { t.Fatal(err) } repos := repository.New(db) configureTestDefaultQB(t, repos, qb.URL) if err := repos.Setting.Set(t.Context(), "qbittorrent.savepath", "/downloads"); err != nil { t.Fatal(err) } if err := repos.Setting.Set(t.Context(), DownloadSmartClassifySettingKey, "false"); err != nil { t.Fatal(err) } svc := NewDownloadService(zap.NewNop(), repos, NewHub(zap.NewNop()), nil) task, err := svc.AddDownloadWithMeta(t.Context(), "u1", "magnet:?xt=urn:btih:ffffffffffffffffffffffffffffffffffffffff&dn=%E5%A3%B0%E7%94%9F%E4%B8%8D%E6%81%AF+S01E01", "", DownloadTaskMeta{ Title: "声生不息 S01E01", SourceCategory: "综艺", }) if err != nil { t.Fatal(err) } if task.SavePath != "/downloads" { t.Fatalf("task save path = %q, want /downloads", task.SavePath) } if gotSavePath != "/downloads" { t.Fatalf("qb savepath = %q, want /downloads", gotSavePath) } if gotCategory != "" { t.Fatalf("qb category = %q, want empty", gotCategory) } } func TestReloadConfigDoesNotFallbackToLegacyAfterClientDisabled(t *testing.T) { var addCalls int32 qb := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { switch r.URL.Path { case "/api/v2/auth/login": _, _ = w.Write([]byte("Ok.")) case "/api/v2/torrents/info": _, _ = w.Write([]byte(`[]`)) case "/api/v2/torrents/add": atomic.AddInt32(&addCalls, 1) _, _ = w.Write([]byte("Ok.")) default: http.NotFound(w, r) } })) defer qb.Close() db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{}) if err != nil { t.Fatal(err) } if err := db.AutoMigrate(&model.DownloadClient{}, &model.DownloadTask{}, &model.Setting{}); err != nil { t.Fatal(err) } repos := repository.New(db) if err := repos.Setting.Set(t.Context(), "qbittorrent.url", qb.URL); err != nil { t.Fatal(err) } if err := repos.Setting.Set(t.Context(), "qbittorrent.username", "admin"); err != nil { t.Fatal(err) } if err := repos.Setting.Set(t.Context(), "qbittorrent.password", "admin"); err != nil { t.Fatal(err) } client := &model.DownloadClient{Name: "qB", Type: "qbittorrent", Host: qb.URL, Username: "admin", Password: "admin", IsDefault: true, Enabled: true} if err := repos.DownloadClient.Create(t.Context(), client); err != nil { t.Fatal(err) } client.Enabled = false if err := repos.DownloadClient.Update(t.Context(), client); err != nil { t.Fatal(err) } svc := NewDownloadService(zap.NewNop(), repos, NewHub(zap.NewNop()), nil) if err := svc.ReloadConfig(t.Context()); err != nil { t.Fatal(err) } _, err = svc.AddDownloadWithMeta(t.Context(), "u1", "magnet:?xt=urn:btih:bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb&dn=Movie+2026+1080p", "/downloads", DownloadTaskMeta{ Title: "Movie 2026 1080p", }) if err == nil { t.Fatal("expected add to fail when the configured downloader was disabled") } if got := atomic.LoadInt32(&addCalls); got != 0 { t.Fatalf("qb add calls = %d, want 0", got) } } func TestAddDownloadWithMetaFailsClosedWhenNoDownloaderConfigured(t *testing.T) { db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{}) if err != nil { t.Fatal(err) } if err := db.AutoMigrate(&model.DownloadClient{}, &model.DownloadTask{}, &model.Setting{}); err != nil { t.Fatal(err) } repos := repository.New(db) svc := NewDownloadService(zap.NewNop(), repos, NewHub(zap.NewNop()), nil) task, err := svc.AddDownloadWithMeta(t.Context(), "u1", "magnet:?xt=urn:btih:cccccccccccccccccccccccccccccccccccccccc&dn=Movie+2026+1080p", "/downloads", DownloadTaskMeta{ Title: "Movie 2026 1080p", }) if err == nil { t.Fatal("expected no downloader configured error") } if task != nil { t.Fatalf("task = %#v, want nil", task) } rows, err := repos.Download.List(t.Context()) if err != nil { t.Fatal(err) } if len(rows) != 0 { t.Fatalf("download rows = %d, want 0", len(rows)) } } func TestReloadConfigManagedModeDoesNotFallbackToLegacyWithoutRows(t *testing.T) { var addCalls int32 qb := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { switch r.URL.Path { case "/api/v2/auth/login": _, _ = w.Write([]byte("Ok.")) case "/api/v2/torrents/info": _, _ = w.Write([]byte(`[]`)) case "/api/v2/torrents/add": atomic.AddInt32(&addCalls, 1) _, _ = w.Write([]byte("Ok.")) default: http.NotFound(w, r) } })) defer qb.Close() db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{}) if err != nil { t.Fatal(err) } if err := db.AutoMigrate(&model.DownloadClient{}, &model.DownloadTask{}, &model.Setting{}); err != nil { t.Fatal(err) } repos := repository.New(db) if err := repos.Setting.Set(t.Context(), "qbittorrent.url", qb.URL); err != nil { t.Fatal(err) } if err := repos.Setting.Set(t.Context(), "qbittorrent.username", "admin"); err != nil { t.Fatal(err) } if err := repos.Setting.Set(t.Context(), "qbittorrent.password", "admin"); err != nil { t.Fatal(err) } if err := repos.Setting.Set(t.Context(), settingDownloadClientsManaged, "true"); err != nil { t.Fatal(err) } svc := NewDownloadService(zap.NewNop(), repos, NewHub(zap.NewNop()), nil) _, err = svc.AddDownloadWithMeta(t.Context(), "u1", "magnet:?xt=urn:btih:dddddddddddddddddddddddddddddddddddddddd&dn=Movie+2026+1080p", "/downloads", DownloadTaskMeta{ Title: "Movie 2026 1080p", }) if err == nil { t.Fatal("expected managed mode to reject missing default downloader") } if got := atomic.LoadInt32(&addCalls); got != 0 { t.Fatalf("qb add calls = %d, want 0", got) } } func TestAddDownloadWithMetaSkipsExistingLocalEpisodeWithReleaseGroup(t *testing.T) { db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{}) if err != nil { t.Fatal(err) } if err := db.AutoMigrate(&model.Media{}, &model.DownloadTask{}, &model.Setting{}, &model.DownloadClient{}); err != nil { t.Fatal(err) } repos := repository.New(db) if err := db.Create(&model.Media{ Title: "凡人修仙传", Path: "/media/动漫/国漫/凡人修仙传/Season 01/凡人修仙传 - S01E146.mkv", SeasonNum: 1, EpisodeNum: 146, }).Error; err != nil { t.Fatal(err) } svc := NewDownloadService(zap.NewNop(), repos, NewHub(zap.NewNop()), nil) task, err := svc.AddDownloadWithMeta(t.Context(), "u1", "magnet:?xt=urn:btih:eeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee&dn=%5BMagicStar%5D+%E5%87%A1%E4%BA%BA%E4%BF%AE%E4%BB%99%E4%BC%A0+%E5%B9%B4%E7%95%AA+-+146+%5B1080p%5D", "/downloads", DownloadTaskMeta{ Title: "[MagicStar] 凡人修仙传 年番 - 146 [1080p][WEB-DL]", }) if !errors.Is(err, ErrMediaAlreadyInLibrary) { t.Fatalf("err = %v, want ErrMediaAlreadyInLibrary", err) } if task != nil { t.Fatalf("task = %#v, want nil", task) } rows, err := repos.Download.List(t.Context()) if err != nil { t.Fatal(err) } if len(rows) != 0 { t.Fatalf("download rows = %d, want 0", len(rows)) } }