2020-03-27 11:01:41 +01:00
|
|
|
package webdav
|
|
|
|
|
|
|
|
import (
|
|
|
|
"os"
|
2020-10-04 11:23:01 +02:00
|
|
|
"path/filepath"
|
2020-03-27 11:01:41 +01:00
|
|
|
"testing"
|
|
|
|
|
|
|
|
"github.com/photoprism/photoprism/pkg/fs"
|
|
|
|
"github.com/photoprism/photoprism/pkg/rnd"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
)
|
|
|
|
|
|
|
|
const (
|
2020-03-27 18:17:07 +01:00
|
|
|
testUrl = "http://webdav-dummy/"
|
2020-03-27 11:01:41 +01:00
|
|
|
testUser = "admin"
|
|
|
|
testPass = "photoprism"
|
|
|
|
)
|
|
|
|
|
|
|
|
func TestConnect(t *testing.T) {
|
2020-04-03 18:08:49 +02:00
|
|
|
c := New(testUrl, testUser, testPass)
|
2020-03-27 11:01:41 +01:00
|
|
|
|
|
|
|
assert.IsType(t, Client{}, c)
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestClient_Files(t *testing.T) {
|
2020-04-03 18:08:49 +02:00
|
|
|
c := New(testUrl, testUser, testPass)
|
2020-03-27 11:01:41 +01:00
|
|
|
|
|
|
|
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")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-03-27 11:56:24 +01:00
|
|
|
func TestClient_Directories(t *testing.T) {
|
2020-04-03 18:08:49 +02:00
|
|
|
c := New(testUrl, testUser, testPass)
|
2020-03-27 11:56:24 +01:00
|
|
|
|
|
|
|
assert.IsType(t, Client{}, c)
|
|
|
|
|
|
|
|
t.Run("non-recursive", func(t *testing.T) {
|
2020-12-05 01:24:33 +01:00
|
|
|
dirs, err := c.Directories("", false, SyncTimeout)
|
2020-03-27 11:56:24 +01:00
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(dirs) == 0 {
|
|
|
|
t.Fatal("no directories found")
|
|
|
|
}
|
|
|
|
|
2020-04-02 18:17:07 +02:00
|
|
|
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)
|
2020-03-27 11:56:24 +01:00
|
|
|
})
|
|
|
|
|
|
|
|
t.Run("recursive", func(t *testing.T) {
|
2020-12-05 01:24:33 +01:00
|
|
|
dirs, err := c.Directories("", true, SyncTimeout)
|
2020-03-27 11:56:24 +01:00
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(dirs) < 2 {
|
|
|
|
t.Fatal("at least 2 directories expected")
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2020-03-27 11:01:41 +01:00
|
|
|
func TestClient_Download(t *testing.T) {
|
2020-04-03 18:08:49 +02:00
|
|
|
c := New(testUrl, testUser, testPass)
|
2020-03-27 11:01:41 +01:00
|
|
|
|
|
|
|
assert.IsType(t, Client{}, c)
|
|
|
|
|
|
|
|
files, err := c.Files("Photos")
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
|
2020-10-04 11:23:01 +02:00
|
|
|
tempDir := filepath.Join(os.TempDir(), rnd.UUID())
|
2020-03-27 11:01:41 +01:00
|
|
|
tempFile := tempDir + "/foo.jpg"
|
|
|
|
|
|
|
|
if len(files) == 0 {
|
|
|
|
t.Fatal("no files to download")
|
|
|
|
}
|
|
|
|
|
2020-04-06 16:34:29 +02:00
|
|
|
if err := c.Download(files[0].Abs, tempFile, false); err != nil {
|
2020-03-27 11:01:41 +01:00
|
|
|
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) {
|
2020-04-03 18:08:49 +02:00
|
|
|
c := New(testUrl, testUser, testPass)
|
2020-03-27 11:01:41 +01:00
|
|
|
|
|
|
|
assert.IsType(t, Client{}, c)
|
|
|
|
|
|
|
|
t.Run("non-recursive", func(t *testing.T) {
|
2020-10-04 11:23:01 +02:00
|
|
|
tempDir := filepath.Join(os.TempDir(), rnd.UUID())
|
2020-03-27 11:01:41 +01:00
|
|
|
|
2020-04-06 16:34:29 +02:00
|
|
|
if errs := c.DownloadDir("Photos", tempDir, false, false); len(errs) > 0 {
|
2020-03-27 11:01:41 +01:00
|
|
|
t.Fatal(errs)
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := os.RemoveAll(tempDir); err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
|
|
|
|
t.Run("recursive", func(t *testing.T) {
|
2020-10-04 11:23:01 +02:00
|
|
|
tempDir := filepath.Join(os.TempDir(), rnd.UUID())
|
2020-03-27 11:01:41 +01:00
|
|
|
|
2020-04-06 16:34:29 +02:00
|
|
|
if errs := c.DownloadDir("Photos", tempDir, true, false); len(errs) > 0 {
|
2020-03-27 11:01:41 +01:00
|
|
|
t.Fatal(errs)
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := os.RemoveAll(tempDir); err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestClient_UploadAndDelete(t *testing.T) {
|
2020-04-03 18:08:49 +02:00
|
|
|
c := New(testUrl, testUser, testPass)
|
2020-03-27 11:01:41 +01:00
|
|
|
|
|
|
|
assert.IsType(t, Client{}, c)
|
|
|
|
|
2020-05-19 11:00:17 +02:00
|
|
|
tempName := rnd.UUID() + fs.JpegExt
|
2020-03-27 11:01:41 +01:00
|
|
|
|
|
|
|
if err := c.Upload("testdata/example.jpg", tempName); err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := c.Delete(tempName); err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
}
|