photoprism/pkg/fs/directories_test.go

100 lines
2.2 KiB
Go

package fs
import (
"strings"
"testing"
"github.com/stretchr/testify/assert"
)
func TestDirs(t *testing.T) {
t.Run("recursive", func(t *testing.T) {
result, err := Dirs("testdata", true, true)
if err != nil {
t.Fatal(err)
}
assert.Len(t, result, 9)
assert.Contains(t, result, "/directory")
assert.Contains(t, result, "/directory/subdirectory")
assert.Contains(t, result, "/directory/subdirectory/animals")
assert.Contains(t, result, "/originals")
assert.NotContains(t, result, "/originals/storage")
assert.Contains(t, result, "/linked")
})
t.Run("recursive no-symlinks", func(t *testing.T) {
result, err := Dirs("testdata", true, false)
if err != nil {
t.Fatal(err)
}
assert.Contains(t, result, "/directory")
assert.Contains(t, result, "/directory/subdirectory")
assert.Contains(t, result, "/directory/subdirectory/animals")
assert.Contains(t, result, "/linked")
})
t.Run("non-recursive", func(t *testing.T) {
result, err := Dirs("testdata", false, true)
if err != nil {
t.Fatal(err)
}
assert.Contains(t, result, "/directory")
assert.Contains(t, result, "/linked")
})
t.Run("non-recursive no-symlinks", func(t *testing.T) {
result, err := Dirs("testdata/directory/subdirectory", false, false)
if err != nil {
t.Fatal(err)
}
assert.Len(t, result, 1)
assert.Contains(t, result, "/animals")
})
t.Run("non-recursive symlinks", func(t *testing.T) {
result, err := Dirs("testdata/linked", false, true)
if err != nil {
t.Fatal(err)
}
assert.Contains(t, result, "/photoprism")
assert.Contains(t, result, "/self")
})
t.Run("no-result", func(t *testing.T) {
result, err := Dirs("testdata/linked", false, false)
if err != nil {
t.Fatal(err)
}
assert.Empty(t, result)
})
}
func TestFindDir(t *testing.T) {
t.Run("Found", func(t *testing.T) {
result := FindDir([]string{"./testdata"})
assert.True(t, strings.HasSuffix(result, "/pkg/fs/testdata"))
})
t.Run("NotFound", func(t *testing.T) {
result := FindDir([]string{"/directory", "/directory/subdirectory", "/linked"})
assert.Equal(t, "", result)
})
}
func TestMkdirAll(t *testing.T) {
t.Run("Exists", func(t *testing.T) {
assert.NoError(t, MkdirAll("testdata"))
})
}