94 lines
2.5 KiB
Go
94 lines
2.5 KiB
Go
package layout
|
|
|
|
import "testing"
|
|
|
|
func TestSanitizeComponent(t *testing.T) {
|
|
tests := []struct {
|
|
in, want string
|
|
}{
|
|
{"Дюна Часть вторая", "Дюна Часть вторая"},
|
|
{"a/b\\c", "a b c"},
|
|
{"..", ""},
|
|
{"../../etc/passwd", "etc passwd"},
|
|
{" trailing. ", "trailing"},
|
|
{"name: sub*title?", "name sub title"},
|
|
{"multi space", "multi space"},
|
|
{"tab\tand\nnewline", "tab and newline"},
|
|
{".hidden", "hidden"},
|
|
{"a<b>c|d\"e", "a b c d e"},
|
|
}
|
|
for _, tt := range tests {
|
|
if got := sanitizeComponent(tt.in); got != tt.want {
|
|
t.Errorf("sanitizeComponent(%q) = %q, want %q", tt.in, got, tt.want)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestTitleYear(t *testing.T) {
|
|
got, err := titleYear("The Matrix", 1999)
|
|
if err != nil || got != "The Matrix (1999)" {
|
|
t.Errorf("got %q, %v", got, err)
|
|
}
|
|
got, err = titleYear("No Year", 0)
|
|
if err != nil || got != "No Year" {
|
|
t.Errorf("got %q, %v", got, err)
|
|
}
|
|
if _, err := titleYear("///", 2000); err == nil {
|
|
t.Error("want error for empty title after sanitize")
|
|
}
|
|
}
|
|
|
|
func TestFolderName(t *testing.T) {
|
|
if got := folderName("Dune (2024)", "tmdbid-693134"); got != "Dune (2024) [tmdbid-693134]" {
|
|
t.Errorf("got %q", got)
|
|
}
|
|
if got := folderName("Dune (2024)", ""); got != "Dune (2024)" {
|
|
t.Errorf("got %q", got)
|
|
}
|
|
}
|
|
|
|
func TestEpisodeStem(t *testing.T) {
|
|
if got := episodeStem("Fargo (2015)", 2, 1, 0); got != "Fargo (2015) S02E01" {
|
|
t.Errorf("got %q", got)
|
|
}
|
|
if got := episodeStem("Show", 1, 5, 6); got != "Show S01E05-E06" {
|
|
t.Errorf("got %q", got)
|
|
}
|
|
}
|
|
|
|
func TestSeasonFolder(t *testing.T) {
|
|
if got := seasonFolder(0); got != "Season 00" {
|
|
t.Errorf("got %q", got)
|
|
}
|
|
if got := seasonFolder(12); got != "Season 12" {
|
|
t.Errorf("got %q", got)
|
|
}
|
|
}
|
|
|
|
func TestSubtitleSuffix(t *testing.T) {
|
|
if got := subtitleSuffix("ru", nil); got != ".ru" {
|
|
t.Errorf("got %q", got)
|
|
}
|
|
if got := subtitleSuffix("RU", []string{"forced"}); got != ".ru.forced" {
|
|
t.Errorf("got %q", got)
|
|
}
|
|
if got := subtitleSuffix("", nil); got != "" {
|
|
t.Errorf("got %q", got)
|
|
}
|
|
}
|
|
|
|
func TestUnderRoot(t *testing.T) {
|
|
if !underRoot("/srv/media/movies", "/srv/media/movies/Film (2020)/Film (2020).mkv") {
|
|
t.Error("want true for nested path")
|
|
}
|
|
if underRoot("/srv/media/movies", "/srv/media/movies") {
|
|
t.Error("root itself is not a valid target")
|
|
}
|
|
if underRoot("/srv/media/movies", "/srv/media/series/x.mkv") {
|
|
t.Error("sibling dir must be rejected")
|
|
}
|
|
if underRoot("/srv/media/movies", "/srv/media/movies/../series/x.mkv") {
|
|
t.Error("traversal must be rejected after clean")
|
|
}
|
|
}
|