92e6c4fe1e
Extends DownloadSettings with 4 additional options: - Name: File name pattern for downloaded files (existed) - Disabled: Disables downloads - Originals: Only download files stored in "originals" folder - MediaRaw: Include RAW image files - MediaSidecar: Include metadata sidecar files (JSON, XMP, YAML)
153 lines
2.9 KiB
Go
153 lines
2.9 KiB
Go
package webdav
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/photoprism/photoprism/pkg/fs"
|
|
"github.com/photoprism/photoprism/pkg/rnd"
|
|
)
|
|
|
|
const (
|
|
testUrl = "http://dummy-webdav/"
|
|
testUser = "admin"
|
|
testPass = "photoprism"
|
|
)
|
|
|
|
func TestConnect(t *testing.T) {
|
|
c := New(testUrl, testUser, testPass, TimeoutLow)
|
|
|
|
assert.IsType(t, Client{}, c)
|
|
}
|
|
|
|
func TestClient_Files(t *testing.T) {
|
|
c := New(testUrl, testUser, testPass, TimeoutLow)
|
|
|
|
assert.IsType(t, Client{}, c)
|
|
|
|
files, err := c.Files("Photos")
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if len(files) == 0 {
|
|
t.Fatal("no files found")
|
|
}
|
|
}
|
|
|
|
func TestClient_Directories(t *testing.T) {
|
|
c := New(testUrl, testUser, testPass, TimeoutLow)
|
|
|
|
assert.IsType(t, Client{}, c)
|
|
|
|
t.Run("non-recursive", func(t *testing.T) {
|
|
dirs, err := c.Directories("", false, MaxRequestDuration)
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if len(dirs) == 0 {
|
|
t.Fatal("no directories found")
|
|
}
|
|
|
|
assert.IsType(t, fs.FileInfo{}, dirs[0])
|
|
assert.Equal(t, "Photos", dirs[0].Name)
|
|
assert.Equal(t, "/Photos", dirs[0].Abs)
|
|
assert.Equal(t, true, dirs[0].Dir)
|
|
assert.Equal(t, int64(0), dirs[0].Size)
|
|
})
|
|
|
|
t.Run("recursive", func(t *testing.T) {
|
|
dirs, err := c.Directories("", true, 0)
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if len(dirs) < 2 {
|
|
t.Fatal("at least 2 directories expected")
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestClient_Download(t *testing.T) {
|
|
c := New(testUrl, testUser, testPass, TimeoutDefault)
|
|
|
|
assert.IsType(t, Client{}, c)
|
|
|
|
files, err := c.Files("Photos")
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
tempDir := filepath.Join(os.TempDir(), rnd.UUID())
|
|
tempFile := tempDir + "/foo.jpg"
|
|
|
|
if len(files) == 0 {
|
|
t.Fatal("no files to download")
|
|
}
|
|
|
|
if err := c.Download(files[0].Abs, tempFile, false); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if !fs.FileExists(tempFile) {
|
|
t.Fatalf("%s does not exist", tempFile)
|
|
}
|
|
|
|
if err := os.RemoveAll(tempDir); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestClient_DownloadDir(t *testing.T) {
|
|
c := New(testUrl, testUser, testPass, TimeoutLow)
|
|
|
|
assert.IsType(t, Client{}, c)
|
|
|
|
t.Run("non-recursive", func(t *testing.T) {
|
|
tempDir := filepath.Join(os.TempDir(), rnd.UUID())
|
|
|
|
if errs := c.DownloadDir("Photos", tempDir, false, false); len(errs) > 0 {
|
|
t.Fatal(errs)
|
|
}
|
|
|
|
if err := os.RemoveAll(tempDir); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
})
|
|
|
|
t.Run("recursive", func(t *testing.T) {
|
|
tempDir := filepath.Join(os.TempDir(), rnd.UUID())
|
|
|
|
if errs := c.DownloadDir("Photos", tempDir, true, false); len(errs) > 0 {
|
|
t.Fatal(errs)
|
|
}
|
|
|
|
if err := os.RemoveAll(tempDir); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestClient_UploadAndDelete(t *testing.T) {
|
|
c := New(testUrl, testUser, testPass, TimeoutLow)
|
|
|
|
assert.IsType(t, Client{}, c)
|
|
|
|
tempName := rnd.UUID() + fs.ExtJPEG
|
|
|
|
if err := c.Upload("testdata/example.jpg", tempName); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := c.Delete(tempName); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|