package api import ( "encoding/json" "net/http" "testing" "github.com/stretchr/testify/assert" "github.com/tidwall/gjson" "github.com/photoprism/photoprism/internal/entity" ) 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, 0, link.LinkExpires) }) 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") 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()) } }) } 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) { t.Run("create share link", func(t *testing.T) { app, router, _ := NewApiTest() var link entity.Link CreatePhotoLink(router) resp := PerformRequestWithBody(app, "POST", "/api/v1/photos/pt9jtdre2lvl0yh7/links", `{"Password":"foobar","Expires":0,"CanEdit":true}`) log.Debugf("BODY: %s", resp.Body.String()) assert.Equal(t, http.StatusOK, resp.Code) 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, 0, link.LinkExpires) assert.False(t, link.CanComment) assert.True(t, link.CanEdit) }) t.Run("photo not found", func(t *testing.T) { app, router, _ := NewApiTest() CreatePhotoLink(router) resp := PerformRequestWithBody(app, "POST", "/api/v1/photos/xxx/links", `{"Password": "foobar", "Expires": 0, "CanEdit": true}`) if resp.Code != http.StatusNotFound { t.Fatal(resp.Body.String()) } }) 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"}`) 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) { t.Run("create share link", func(t *testing.T) { app, router, _ := NewApiTest() var link entity.Link CreateLabelLink(router) resp := PerformRequestWithBody(app, "POST", "/api/v1/labels/lt9k3pw1wowuy3c2/links", `{"Password": "foobar", "Expires": 0, "CanEdit": true}`) assert.Equal(t, http.StatusOK, resp.Code) 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, 0, link.LinkExpires) 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()) } }) 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"}`) 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) }) } */