photoprism/internal/api/file_marker_test.go

63 lines
1.3 KiB
Go
Raw Normal View History

package api
import (
"encoding/json"
"fmt"
"net/http"
"testing"
"github.com/tidwall/gjson"
"github.com/stretchr/testify/assert"
)
func TestUpdateFileMarker(t *testing.T) {
t.Run("success", func(t *testing.T) {
app, router, _ := NewApiTest()
GetPhoto(router)
UpdateFileMarker(router)
r := PerformRequest(app, "GET", "/api/v1/photos/pt9jtdre2lvl0y11")
assert.Equal(t, http.StatusOK, r.Code)
photoUID := gjson.Get(r.Body.String(), "UID").String()
fileUID := gjson.Get(r.Body.String(), "Files.0.UID").String()
markerID := gjson.Get(r.Body.String(), "Files.0.Markers.0.ID").String()
assert.NotEmpty(t, photoUID)
assert.NotEmpty(t, fileUID)
assert.NotEmpty(t, markerID)
u := fmt.Sprintf("/api/v1/photos/%s/files/%s/markers/%s", photoUID, fileUID, markerID)
var m = struct {
RefUID string
RefSrc string
MarkerSrc string
MarkerType string
MarkerScore int
MarkerInvalid bool
MarkerLabel string
}{
RefUID: "3h59wvth837b5vyiub35",
RefSrc: "meta",
MarkerSrc: "image",
MarkerType: "Face",
MarkerScore: 100,
MarkerInvalid: true,
MarkerLabel: "Foo",
}
if b, err := json.Marshal(m); err != nil {
t.Fatal(err)
} else {
t.Logf("PUT %s", u)
r = PerformRequestWithBody(app, "PUT", u, string(b))
}
assert.Equal(t, http.StatusOK, r.Code)
})
}