photoprism/internal/api/link_test.go

406 lines
13 KiB
Go
Raw Normal View History

package api
import (
"encoding/json"
"net/http"
"testing"
2020-11-21 18:08:41 +01:00
"github.com/tidwall/gjson"
"github.com/photoprism/photoprism/internal/entity"
"github.com/stretchr/testify/assert"
)
func TestCreateAlbumLink(t *testing.T) {
t.Run("create share link", func(t *testing.T) {
app, router, _ := NewApiTest()
var link entity.Link
CreateAlbumLink(router)
resp := PerformRequestWithBody(app, "POST", "/api/v1/albums/at9lxuqxpogaaba7/links", `{"Password": "foobar", "Expires": 0, "CanEdit": true}`)
if resp.Code != http.StatusOK {
t.Fatal(resp.Body.String())
}
if err := json.Unmarshal(resp.Body.Bytes(), &link); err != nil {
t.Fatal(err)
}
assert.NotEmpty(t, link.LinkUID)
assert.NotEmpty(t, link.ShareUID)
assert.NotEmpty(t, link.LinkToken)
assert.Equal(t, true, link.CanEdit)
assert.Equal(t, 0, link.LinkExpires)
assert.False(t, link.CanComment)
assert.True(t, link.CanEdit)
})
t.Run("album does not exist", func(t *testing.T) {
app, router, _ := NewApiTest()
CreateAlbumLink(router)
resp := PerformRequestWithBody(app, "POST", "/api/v1/albums/xxx/links", `{"Password": "foobar", "Expires": 0, "CanEdit": true}`)
if resp.Code != http.StatusNotFound {
t.Fatal(resp.Body.String())
}
val := gjson.Get(resp.Body.String(), "error")
2020-05-04 17:14:04 +02:00
assert.Equal(t, "Album not found", val.String())
})
t.Run("invalid request", func(t *testing.T) {
app, router, _ := NewApiTest()
CreateAlbumLink(router)
resp := PerformRequestWithBody(app, "POST", "/api/v1/albums/at9lxuqxpogaaba7/links", `{"Password": "foobar", "Expires": "abc", "CanEdit": true}`)
if resp.Code != http.StatusBadRequest {
t.Fatal(resp.Body.String())
}
2020-05-04 17:14:04 +02:00
})
}
func TestUpdateAlbumLink(t *testing.T) {
app, router, _ := NewApiTest()
CreateAlbumLink(router)
r := PerformRequestWithBody(app, "POST", "/api/v1/albums/at9lxuqxpogaaba7/links", `{"Password": "foobar", "Expires": 0, "CanEdit": true}`)
if r.Code != http.StatusOK {
t.Fatal(r.Body.String())
}
val2 := gjson.Get(r.Body.String(), "Expires")
assert.Equal(t, "0", val2.String())
uid := gjson.Get(r.Body.String(), "UID").String()
t.Run("successful request", func(t *testing.T) {
app, router, _ := NewApiTest()
UpdateAlbumLink(router)
r := PerformRequestWithBody(app, "PUT", "/api/v1/albums/at9lxuqxpogaaba7/links/"+uid, `{"Token": "newToken", "Expires": 8000, "Password": "1234nhfhfd"}`)
val := gjson.Get(r.Body.String(), "Token")
assert.Equal(t, "newtoken", val.String())
val2 := gjson.Get(r.Body.String(), "Expires")
assert.Equal(t, "8000", val2.String())
assert.Equal(t, http.StatusOK, r.Code)
})
t.Run("bad request", func(t *testing.T) {
app, router, _ := NewApiTest()
UpdateAlbumLink(router)
r := PerformRequestWithBody(app, "PUT", "/api/v1/albums/at9lxuqxpogaaba7/links/"+uid, `{"Token": "newToken", "Expires": "vgft", "xxx": "xxx"}`)
assert.Equal(t, http.StatusBadRequest, r.Code)
})
}
func TestDeleteAlbumLink(t *testing.T) {
app, router, _ := NewApiTest()
CreateAlbumLink(router)
r := PerformRequestWithBody(app, "POST", "/api/v1/albums/at9lxuqxpogaaba7/links", `{"Password": "foobar", "Expires": 0, "CanEdit": true}`)
if r.Code != http.StatusOK {
t.Fatal(r.Body.String())
}
uid := gjson.Get(r.Body.String(), "UID").String()
GetAlbumLinks(router)
r2 := PerformRequest(app, "GET", "/api/v1/albums/at9lxuqxpogaaba7/links")
len := gjson.Get(r2.Body.String(), "#")
t.Run("successful deletion", func(t *testing.T) {
app, router, _ := NewApiTest()
DeleteAlbumLink(router)
r := PerformRequest(app, "DELETE", "/api/v1/albums/at9lxuqxpogaaba7/links/"+uid)
assert.Equal(t, http.StatusOK, r.Code)
GetAlbumLinks(router)
r2 := PerformRequest(app, "GET", "/api/v1/albums/at9lxuqxpogaaba7/links")
len2 := gjson.Get(r2.Body.String(), "#")
assert.Greater(t, len.Int(), len2.Int())
})
}
func TestGetAlbumLinks(t *testing.T) {
t.Run("successful request", func(t *testing.T) {
app, router, _ := NewApiTest()
CreateAlbumLink(router)
r := PerformRequestWithBody(app, "POST", "/api/v1/albums/at9lxuqxpogaaba7/links", `{"Password": "foobar", "Expires": 0, "CanEdit": true}`)
if r.Code != http.StatusOK {
t.Fatal(r.Body.String())
}
GetAlbumLinks(router)
r2 := PerformRequest(app, "GET", "/api/v1/albums/at9lxuqxpogaaba7/links")
len := gjson.Get(r2.Body.String(), "#")
assert.GreaterOrEqual(t, len.Int(), int64(1))
assert.Equal(t, http.StatusOK, r2.Code)
})
t.Run("not found", func(t *testing.T) {
app, router, _ := NewApiTest()
GetAlbumLinks(router)
r := PerformRequest(app, "GET", "/api/v1/albums/xxx/links")
assert.Equal(t, http.StatusNotFound, r.Code)
})
}
func TestCreatePhotoLink(t *testing.T) {
2020-05-04 17:14:04 +02:00
t.Run("create share link", func(t *testing.T) {
app, router, _ := NewApiTest()
2020-05-04 17:14:04 +02:00
var link entity.Link
2020-05-04 17:14:04 +02:00
CreatePhotoLink(router)
2020-05-04 17:14:04 +02:00
resp := PerformRequestWithBody(app, "POST", "/api/v1/photos/pt9jtdre2lvl0yh7/links", `{"Password": "foobar", "Expires": 0, "CanEdit": true}`)
assert.Equal(t, http.StatusOK, resp.Code)
2020-05-04 17:14:04 +02:00
if err := json.Unmarshal(resp.Body.Bytes(), &link); err != nil {
2020-05-04 17:14:04 +02:00
t.Fatal(err)
}
assert.NotEmpty(t, link.LinkUID)
assert.NotEmpty(t, link.ShareUID)
assert.NotEmpty(t, link.LinkToken)
assert.Equal(t, 0, link.LinkExpires)
2020-05-04 17:14:04 +02:00
assert.False(t, link.CanComment)
assert.True(t, link.CanEdit)
})
t.Run("photo not found", func(t *testing.T) {
app, router, _ := NewApiTest()
2020-05-04 17:14:04 +02:00
CreatePhotoLink(router)
2020-05-04 17:14:04 +02:00
resp := PerformRequestWithBody(app, "POST", "/api/v1/photos/xxx/links", `{"Password": "foobar", "Expires": 0, "CanEdit": true}`)
2020-05-04 17:14:04 +02:00
if resp.Code != http.StatusNotFound {
t.Fatal(resp.Body.String())
2020-05-04 17:14:04 +02:00
}
})
t.Run("invalid request", func(t *testing.T) {
app, router, _ := NewApiTest()
CreatePhotoLink(router)
r := PerformRequestWithBody(app, "POST", "/api/v1/photos/pt9jtdre2lvl0yh7/links", `{"xxx": 123, "Expires": "abc", "CanEdit": "xxx"}`)
2020-05-04 17:14:04 +02:00
assert.Equal(t, http.StatusBadRequest, r.Code)
})
}
func TestUpdatePhotoLink(t *testing.T) {
app, router, _ := NewApiTest()
CreatePhotoLink(router)
r := PerformRequestWithBody(app, "POST", "/api/v1/photos/pt9jtdre2lvl0yh7/links", `{"Password": "foobar", "Expires": 0, "CanEdit": true}`)
if r.Code != http.StatusOK {
t.Fatal(r.Body.String())
}
val2 := gjson.Get(r.Body.String(), "Expires")
assert.Equal(t, "0", val2.String())
uid := gjson.Get(r.Body.String(), "UID").String()
t.Run("successful request", func(t *testing.T) {
app, router, _ := NewApiTest()
UpdatePhotoLink(router)
r := PerformRequestWithBody(app, "PUT", "/api/v1/photos/pt9jtdre2lvl0yh7/links/"+uid, `{"Token": "newToken", "Expires": 8000, "Password": "1234nhfhfd"}`)
val := gjson.Get(r.Body.String(), "Token")
assert.Equal(t, "newtoken", val.String())
val2 := gjson.Get(r.Body.String(), "Expires")
assert.Equal(t, "8000", val2.String())
assert.Equal(t, http.StatusOK, r.Code)
})
t.Run("bad request", func(t *testing.T) {
app, router, _ := NewApiTest()
UpdatePhotoLink(router)
r := PerformRequestWithBody(app, "PUT", "/api/v1/photos/pt9jtdre2lvl0yh7/links/"+uid, `{"Token": "newToken", "Expires": "vgft", "xxx": "xxx"}`)
assert.Equal(t, http.StatusBadRequest, r.Code)
})
}
//TODO Fully assert once functionality exists
func TestDeletePhotoLink(t *testing.T) {
app, router, _ := NewApiTest()
CreatePhotoLink(router)
r := PerformRequestWithBody(app, "POST", "/api/v1/photos/pt9jtdre2lvl0yh7/links", `{"Password": "foobar", "Expires": 0, "CanEdit": true}`)
if r.Code != http.StatusOK {
t.Fatal(r.Body.String())
}
uid := gjson.Get(r.Body.String(), "UID").String()
//GetPhotoLinks(router)
//r2 := PerformRequest(app, "GET", "/api/v1/photos/pt9jtdre2lvl0yh7/links")
//len := gjson.Get(r2.Body.String(), "#")
t.Run("successful deletion", func(t *testing.T) {
app, router, _ := NewApiTest()
DeletePhotoLink(router)
r := PerformRequest(app, "DELETE", "/api/v1/photos/pt9jtdre2lvl0yh7/links/"+uid)
assert.Equal(t, http.StatusOK, r.Code)
GetPhotoLinks(router)
r2 := PerformRequest(app, "GET", "/api/v1/photos/pt9jtdre2lvl0yh7/links")
t.Log(r2)
//len2 := gjson.Get(r2.Body.String(), "#")
//assert.Greater(t, len.Int(), len2.Int())
})
}
func TestGetPhotoLinks(t *testing.T) {
t.Run("successful request", func(t *testing.T) {
app, router, _ := NewApiTest()
CreatePhotoLink(router)
r := PerformRequestWithBody(app, "POST", "/api/v1/photos/pt9jtdre2lvl0yh7/links", `{"Password": "foobar", "Expires": 0, "CanEdit": true}`)
if r.Code != http.StatusOK {
t.Fatal(r.Body.String())
}
GetPhotoLinks(router)
r2 := PerformRequest(app, "GET", "/api/v1/photos/pt9jtdre2lvl0yh7/links")
//len := gjson.Get(r2.Body.String(), "#")
//assert.GreaterOrEqual(t, len.Int(), int64(1))
assert.Equal(t, http.StatusOK, r2.Code)
})
t.Run("not found", func(t *testing.T) {
app, router, _ := NewApiTest()
GetPhotoLinks(router)
r := PerformRequest(app, "GET", "/api/v1/photos/xxx/links")
assert.Equal(t, http.StatusNotFound, r.Code)
})
}
func TestCreateLabelLink(t *testing.T) {
2020-05-04 17:14:04 +02:00
t.Run("create share link", func(t *testing.T) {
app, router, _ := NewApiTest()
2020-05-04 17:14:04 +02:00
var link entity.Link
2020-05-04 17:14:04 +02:00
CreateLabelLink(router)
2020-05-04 17:14:04 +02:00
resp := PerformRequestWithBody(app, "POST", "/api/v1/labels/lt9k3pw1wowuy3c2/links", `{"Password": "foobar", "Expires": 0, "CanEdit": true}`)
assert.Equal(t, http.StatusOK, resp.Code)
2020-05-04 17:14:04 +02:00
if err := json.Unmarshal(resp.Body.Bytes(), &link); err != nil {
2020-05-04 17:14:04 +02:00
t.Fatal(err)
}
assert.NotEmpty(t, link.LinkUID)
assert.NotEmpty(t, link.ShareUID)
assert.NotEmpty(t, link.LinkToken)
assert.Equal(t, 0, link.LinkExpires)
2020-05-04 17:14:04 +02:00
assert.False(t, link.CanComment)
assert.True(t, link.CanEdit)
})
t.Run("label not found", func(t *testing.T) {
app, router, _ := NewApiTest()
CreateLabelLink(router)
resp := PerformRequestWithBody(app, "POST", "/api/v1/labels/xxx/links", `{"Password": "foobar", "Expires": 0, "CanEdit": true}`)
if resp.Code != http.StatusNotFound {
t.Fatal(resp.Body.String())
}
2020-05-04 17:14:04 +02:00
})
t.Run("invalid request", func(t *testing.T) {
app, router, _ := NewApiTest()
CreateLabelLink(router)
r := PerformRequestWithBody(app, "POST", "/api/v1/labels/lt9k3pw1wowuy3c2/links", `{"xxx": 123, "Expires": "abc", "CanEdit": "xxx"}`)
2020-05-04 17:14:04 +02:00
assert.Equal(t, http.StatusBadRequest, r.Code)
})
}
func TestUpdateLabelLink(t *testing.T) {
app, router, _ := NewApiTest()
CreateLabelLink(router)
r := PerformRequestWithBody(app, "POST", "/api/v1/labels/lt9k3pw1wowuy3c2/links", `{"Password": "foobar", "Expires": 0, "CanEdit": true}`)
if r.Code != http.StatusOK {
t.Fatal(r.Body.String())
}
val2 := gjson.Get(r.Body.String(), "Expires")
assert.Equal(t, "0", val2.String())
uid := gjson.Get(r.Body.String(), "UID").String()
t.Run("successful request", func(t *testing.T) {
app, router, _ := NewApiTest()
UpdateLabelLink(router)
r := PerformRequestWithBody(app, "PUT", "/api/v1/labels/lt9k3pw1wowuy3c2/links/"+uid, `{"Token": "newToken", "Expires": 8000, "Password": "1234nhfhfd"}`)
val := gjson.Get(r.Body.String(), "Token")
assert.Equal(t, "newtoken", val.String())
val2 := gjson.Get(r.Body.String(), "Expires")
assert.Equal(t, "8000", val2.String())
assert.Equal(t, http.StatusOK, r.Code)
})
t.Run("bad request", func(t *testing.T) {
app, router, _ := NewApiTest()
UpdateLabelLink(router)
r := PerformRequestWithBody(app, "PUT", "/api/v1/labels/lt9k3pw1wowuy3c2/links/"+uid, `{"Token": "newToken", "Expires": "vgft", "xxx": "xxx"}`)
assert.Equal(t, http.StatusBadRequest, r.Code)
})
}
func TestDeleteLabelLink(t *testing.T) {
app, router, _ := NewApiTest()
CreateLabelLink(router)
r := PerformRequestWithBody(app, "POST", "/api/v1/labels/lt9k3pw1wowuy3c2/links", `{"Password": "foobar", "Expires": 0, "CanEdit": true}`)
if r.Code != http.StatusOK {
t.Fatal(r.Body.String())
}
uid := gjson.Get(r.Body.String(), "UID").String()
//GetLabelLinks(router)
//r2 := PerformRequest(app, "GET", "/api/v1/labels/lt9k3pw1wowuy3c2/links")
//len := gjson.Get(r2.Body.String(), "#")
t.Run("successful deletion", func(t *testing.T) {
app, router, _ := NewApiTest()
DeleteLabelLink(router)
r := PerformRequest(app, "DELETE", "/api/v1/labels/lt9k3pw1wowuy3c2/links/"+uid)
assert.Equal(t, http.StatusOK, r.Code)
//GetLabelLinks(router)
//r2 := PerformRequest(app, "GET", "/api/v1/labels/lt9k3pw1wowuy3c2/links")
//len2 := gjson.Get(r2.Body.String(), "#")
//assert.Greater(t, len.Int(), len2.Int())
})
}
func TestGetLabelLinks(t *testing.T) {
t.Run("successful request", func(t *testing.T) {
app, router, _ := NewApiTest()
CreateLabelLink(router)
r := PerformRequestWithBody(app, "POST", "/api/v1/labels/lt9k3pw1wowuy3c2/links", `{"Password": "foobar", "Expires": 0, "CanEdit": true}`)
if r.Code != http.StatusOK {
t.Fatal(r.Body.String())
}
GetLabelLinks(router)
r2 := PerformRequest(app, "GET", "/api/v1/labels/lt9k3pw1wowuy3c2/links")
//len := gjson.Get(r2.Body.String(), "#")
//assert.GreaterOrEqual(t, len.Int(), int64(1))
assert.Equal(t, http.StatusOK, r2.Code)
})
t.Run("not found", func(t *testing.T) {
app, router, _ := NewApiTest()
GetLabelLinks(router)
r := PerformRequest(app, "GET", "/api/v1/labels/xxx/links")
assert.Equal(t, http.StatusNotFound, r.Code)
})
}