photoprism/internal/query/markers_test.go
2021-08-30 17:20:39 +02:00

192 lines
3.8 KiB
Go

package query
import (
"testing"
"time"
"github.com/photoprism/photoprism/internal/entity"
"github.com/stretchr/testify/assert"
)
func TestMarkers(t *testing.T) {
t.Run("find umatched", func(t *testing.T) {
results, err := Markers(3, 0, entity.MarkerFace, false, false, entity.TimeStamp())
if err != nil {
t.Fatal(err)
}
assert.GreaterOrEqual(t, len(results), 1)
for _, val := range results {
assert.IsType(t, entity.Marker{}, val)
}
})
t.Run("find all", func(t *testing.T) {
results, err := Markers(3, 0, entity.MarkerFace, false, false, time.Time{})
if err != nil {
t.Fatal(err)
}
assert.GreaterOrEqual(t, len(results), 1)
for _, val := range results {
assert.IsType(t, entity.Marker{}, val)
}
})
t.Run("find embeddings", func(t *testing.T) {
results, err := Markers(3, 0, entity.MarkerFace, true, false, time.Time{})
if err != nil {
t.Fatal(err)
}
assert.GreaterOrEqual(t, len(results), 1)
for _, val := range results {
assert.IsType(t, entity.Marker{}, val)
}
})
t.Run("find false", func(t *testing.T) {
results, err := Markers(3, 0, entity.MarkerFace, false, true, time.Time{})
if err != nil {
t.Fatal(err)
}
assert.Equal(t, 3, len(results))
})
}
func TestUnmatchedFaceMarkers(t *testing.T) {
t.Run("all", func(t *testing.T) {
results, err := UnmatchedFaceMarkers(3, 0, nil)
if err != nil {
t.Fatal(err)
}
assert.Equal(t, 3, len(results))
})
t.Run("before", func(t *testing.T) {
results, err := UnmatchedFaceMarkers(3, 0, entity.TimePointer())
if err != nil {
t.Fatal(err)
}
assert.Equal(t, 3, len(results))
})
}
func TestFaceMarkers(t *testing.T) {
t.Run("all", func(t *testing.T) {
results, err := FaceMarkers(3, 0)
if err != nil {
t.Fatal(err)
}
assert.Equal(t, 3, len(results))
})
}
func TestEmbeddings(t *testing.T) {
t.Run("all", func(t *testing.T) {
results, err := Embeddings(false, false, 0, 0)
if err != nil {
t.Fatal(err)
}
assert.GreaterOrEqual(t, len(results), 1)
for _, val := range results {
assert.IsType(t, entity.Embedding{}, val)
}
})
t.Run("size", func(t *testing.T) {
results, err := Embeddings(false, false, 230, 0)
if err != nil {
t.Fatal(err)
}
assert.Equal(t, len(results), 8)
for _, val := range results {
assert.IsType(t, entity.Embedding{}, val)
}
})
t.Run("score", func(t *testing.T) {
results, err := Embeddings(false, false, 0, 50)
if err != nil {
t.Fatal(err)
}
assert.GreaterOrEqual(t, len(results), 1)
for _, val := range results {
assert.IsType(t, entity.Embedding{}, val)
}
})
}
func TestRemoveInvalidMarkerReferences(t *testing.T) {
affected, err := RemoveInvalidMarkerReferences()
assert.NoError(t, err)
assert.GreaterOrEqual(t, affected, int64(0))
}
func TestRemoveNonExistentMarkerFaces(t *testing.T) {
affected, err := RemoveNonExistentMarkerFaces()
assert.NoError(t, err)
assert.GreaterOrEqual(t, affected, int64(1))
}
func TestRemoveNonExistentMarkerSubjects(t *testing.T) {
affected, err := RemoveNonExistentMarkerSubjects()
assert.NoError(t, err)
assert.GreaterOrEqual(t, affected, int64(1))
}
func TestFixMarkerReferences(t *testing.T) {
affected, err := FixMarkerReferences()
assert.NoError(t, err)
assert.GreaterOrEqual(t, affected, int64(0))
}
func TestMarkersWithNonExistentReferences(t *testing.T) {
f, s, err := MarkersWithNonExistentReferences()
assert.NoError(t, err)
assert.GreaterOrEqual(t, len(f), 0)
assert.GreaterOrEqual(t, len(s), 0)
}
func TestMarkersWithSubjectConflict(t *testing.T) {
m, err := MarkersWithSubjectConflict()
assert.NoError(t, err)
assert.GreaterOrEqual(t, len(m), 0)
}
func TestCountUnmatchedFaceMarkers(t *testing.T) {
n := CountUnmatchedFaceMarkers()
assert.GreaterOrEqual(t, n, 1)
}
func TestCountMarkers(t *testing.T) {
n := CountMarkers(entity.MarkerFace)
assert.GreaterOrEqual(t, n, 1)
}