mirror of https://github.com/stashapp/stash.git
92 lines
2.9 KiB
Go
92 lines
2.9 KiB
Go
|
package manager
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"github.com/stashapp/stash/pkg/logger"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
var excludeTestFilenames = []string{
|
||
|
|
"/stash/videos/filename.mp4",
|
||
|
|
"/stash/videos/new filename.mp4",
|
||
|
|
"filename sample.mp4",
|
||
|
|
"/stash/videos/exclude/not wanted.webm",
|
||
|
|
"/stash/videos/exclude/not wanted2.webm",
|
||
|
|
"/somewhere/trash/not wanted.wmv",
|
||
|
|
"/disk2/stash/videos/exclude/!!wanted!!.avi",
|
||
|
|
"/disk2/stash/videos/xcl/not wanted.avi",
|
||
|
|
"/stash/videos/partial.file.001.webm",
|
||
|
|
"/stash/videos/partial.file.002.webm",
|
||
|
|
"/stash/videos/partial.file.003.webm",
|
||
|
|
"/stash/videos/sample file sample.mkv",
|
||
|
|
"/stash/videos/.ckRVp1/.still_encoding.mp4",
|
||
|
|
"c:\\stash\\videos\\exclude\\filename windows.mp4",
|
||
|
|
"c:\\stash\\videos\\filename windows.mp4",
|
||
|
|
"\\\\network\\videos\\filename windows network.mp4",
|
||
|
|
"\\\\network\\share\\windows network wanted.mp4",
|
||
|
|
"\\\\network\\share\\windows network wanted sample.mp4",
|
||
|
|
"\\\\network\\private\\windows.network.skip.mp4"}
|
||
|
|
|
||
|
|
var excludeTests = []struct {
|
||
|
|
testPattern []string
|
||
|
|
expected int
|
||
|
|
}{
|
||
|
|
{[]string{"sample\\.mp4$", "trash", "\\.[\\d]{3}\\.webm$"}, 6}, //generic
|
||
|
|
{[]string{"no_match\\.mp4"}, 0}, //no match
|
||
|
|
{[]string{"^/stash/videos/exclude/", "/videos/xcl/"}, 3}, //linux
|
||
|
|
{[]string{"/\\.[[:word:]]+/"}, 1}, //linux hidden dirs (handbrake unraid issue?)
|
||
|
|
{[]string{"c:\\\\stash\\\\videos\\\\exclude"}, 1}, //windows
|
||
|
|
{[]string{"\\/[/invalid"}, 0}, //invalid pattern
|
||
|
|
{[]string{"\\/[/invalid", "sample\\.[[:alnum:]]+$"}, 3}, //invalid pattern but continue
|
||
|
|
{[]string{"^\\\\\\\\network"}, 4}, //windows net share
|
||
|
|
{[]string{"\\\\private\\\\"}, 1}, //windows net share
|
||
|
|
{[]string{"\\\\private\\\\", "sample\\.mp4"}, 3}, //windows net share
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestExcludeFiles(t *testing.T) {
|
||
|
|
for _, test := range excludeTests {
|
||
|
|
err := runExclude(excludeTestFilenames, test.testPattern, test.expected)
|
||
|
|
if err != nil {
|
||
|
|
t.Error(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func runExclude(filenames []string, patterns []string, expCount int) error {
|
||
|
|
|
||
|
|
files, count := excludeFiles(filenames, patterns)
|
||
|
|
|
||
|
|
if count != expCount {
|
||
|
|
return fmt.Errorf("Was expecting %d, found %d", expCount, count)
|
||
|
|
}
|
||
|
|
if len(files) != len(filenames)-expCount {
|
||
|
|
return fmt.Errorf("Returned list should have %d files, not %d ", len(filenames)-expCount, len(files))
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestMatchFile(t *testing.T) {
|
||
|
|
for _, test := range excludeTests {
|
||
|
|
err := runMatch(excludeTestFilenames, test.testPattern, test.expected)
|
||
|
|
if err != nil {
|
||
|
|
t.Error(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func runMatch(filenames []string, patterns []string, expCount int) error {
|
||
|
|
count := 0
|
||
|
|
for _, file := range filenames {
|
||
|
|
if matchFile(file, patterns) {
|
||
|
|
logger.Infof("File \"%s\" matched pattern\n", file)
|
||
|
|
count++
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if count != expCount {
|
||
|
|
return fmt.Errorf("Was expecting %d, found %d", expCount, count)
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|