photoprism/internal/query/faces_test.go

57 lines
980 B
Go

package query
import (
"testing"
"github.com/photoprism/photoprism/internal/entity"
"github.com/stretchr/testify/assert"
)
func TestFaces(t *testing.T) {
results, err := Faces(true)
if err != nil {
t.Fatal(err)
}
assert.GreaterOrEqual(t, len(results), 1)
for _, val := range results {
assert.IsType(t, entity.Face{}, val)
}
}
func TestMatchKnownFaces(t *testing.T) {
const faceFixtureId = uint(6)
if m, err := MarkerByID(faceFixtureId); err != nil {
t.Fatal(err)
} else {
assert.Empty(t, m.SubjectUID)
}
affected, err := MatchKnownFaces()
if err != nil {
t.Fatal(err)
}
assert.Equal(t, int64(1), affected)
if m, err := MarkerByID(faceFixtureId); err != nil {
t.Fatal(err)
} else {
assert.Equal(t, "rqu0xs11qekk9jx8", m.SubjectUID)
}
}
func TestPurgeAnonymousFaces(t *testing.T) {
assert.NoError(t, PurgeAnonymousFaces())
}
func TestCountNewFaceMarkers(t *testing.T) {
n := CountNewFaceMarkers()
assert.GreaterOrEqual(t, n, 1)
}