29cf0a0126
Search, inspect, stream, and download torrents from the terminal. Replaces the entire *arr stack with a single binary.
92 lines
2.1 KiB
Go
92 lines
2.1 KiB
Go
package engine
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
func TestOrganizeDisabled(t *testing.T) {
|
|
r := &Result{FilePath: "/tmp/file.mkv", FileName: "file.mkv"}
|
|
task := &Task{Title: "Movie"}
|
|
path, err := organize(r, task, OrganizeConfig{Enabled: false})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if path != "/tmp/file.mkv" {
|
|
t.Errorf("path = %q, want original path when disabled", path)
|
|
}
|
|
}
|
|
|
|
func TestOrganizeMovie(t *testing.T) {
|
|
tmp := t.TempDir()
|
|
srcDir := filepath.Join(tmp, "src")
|
|
os.MkdirAll(srcDir, 0o755)
|
|
srcFile := filepath.Join(srcDir, "Movie.2023.1080p.mkv")
|
|
os.WriteFile(srcFile, []byte("data"), 0o644)
|
|
|
|
moviesDir := filepath.Join(tmp, "Movies")
|
|
|
|
r := &Result{FilePath: srcFile, FileName: "Movie.2023.1080p.mkv"}
|
|
task := &Task{Title: "Movie 2023"}
|
|
|
|
path, err := organize(r, task, OrganizeConfig{
|
|
Enabled: true,
|
|
MoviesDir: moviesDir,
|
|
})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Should be in Movies/Movie (2023)/
|
|
if path == srcFile {
|
|
t.Error("file should have moved")
|
|
}
|
|
if _, err := os.Stat(path); err != nil {
|
|
t.Errorf("organized file should exist at %s: %v", path, err)
|
|
}
|
|
}
|
|
|
|
func TestOrganizeTVShow(t *testing.T) {
|
|
tmp := t.TempDir()
|
|
srcFile := filepath.Join(tmp, "Show.S02E05.1080p.mkv")
|
|
os.WriteFile(srcFile, []byte("data"), 0o644)
|
|
|
|
tvDir := filepath.Join(tmp, "TV Shows")
|
|
|
|
r := &Result{FilePath: srcFile, FileName: "Show.S02E05.1080p.mkv"}
|
|
task := &Task{Title: "Show S02E05"}
|
|
|
|
path, err := organize(r, task, OrganizeConfig{
|
|
Enabled: true,
|
|
TVShowsDir: tvDir,
|
|
})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Should detect season from filename S02
|
|
if _, err := os.Stat(path); err != nil {
|
|
t.Errorf("organized file should exist at %s: %v", path, err)
|
|
}
|
|
}
|
|
|
|
func TestCleanTitle(t *testing.T) {
|
|
tests := []struct {
|
|
input string
|
|
want string
|
|
}{
|
|
{"The Matrix (1999)", "The Matrix"},
|
|
{"Oppenheimer 2023 1080p BluRay", "Oppenheimer 2023"},
|
|
{"Movie", "Movie"},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.input, func(t *testing.T) {
|
|
got := cleanTitle(tt.input)
|
|
if got != tt.want {
|
|
t.Errorf("cleanTitle(%q) = %q, want %q", tt.input, got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|