photoprism/internal/query/faces_test.go

package query

import (
	"testing"

	"github.com/photoprism/photoprism/internal/entity"

	"github.com/stretchr/testify/assert"
)

func TestFaces(t *testing.T) {
	t.Run("known", func(t *testing.T) {
		results, err := Faces(true, false)

		if err != nil {
			t.Fatal(err)
		}

		assert.GreaterOrEqual(t, len(results), 1)

		for _, val := range results {
			assert.IsType(t, entity.Face{}, val)
		}
	})

	t.Run("unmatched", func(t *testing.T) {
		results, err := Faces(false, true)

		if err != nil {
			t.Fatal(err)
		}

		assert.GreaterOrEqual(t, len(results), 1)

		for _, val := range results {
			assert.IsType(t, entity.Face{}, val)
		}
	})
}

func TestManuallyAddedFaces(t *testing.T) {
	t.Run("success", func(t *testing.T) {
		results, err := ManuallyAddedFaces()

		if err != nil {
			t.Fatal(err)
		}

		assert.GreaterOrEqual(t, len(results), 1)

		for _, val := range results {
			assert.IsType(t, entity.Face{}, val)
		}
	})
}

func TestMatchFaceMarkers(t *testing.T) {
	const faceFixtureId = uint(6)

	if m, err := MarkerByID(faceFixtureId); err != nil {
		t.Fatal(err)
	} else {
		assert.Empty(t, m.SubjectUID)
	}

	// Reset subject_uid.
	if err := Db().Model(&entity.Marker{}).
		Where("subject_src = ?", entity.SrcAuto).
		Where("subject_uid = ?", "jqu0xs11qekk9jx8").
		Updates(entity.Values{"SubjectUID": ""}).Error; err != nil {
		t.Fatal(err)
	}

	affected, err := MatchFaceMarkers()

	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, "jqu0xs11qekk9jx8", m.SubjectUID)
	}
}

func TestRemoveAnonymousFaceClusters(t *testing.T) {
	removed, err := RemoveAnonymousFaceClusters()

	if err != nil {
		t.Fatal(err)
	}

	assert.Equal(t, int64(2), removed)
}

func TestCountNewFaceMarkers(t *testing.T) {
	t.Run("all", func(t *testing.T) {
		assert.GreaterOrEqual(t, CountNewFaceMarkers(0, 0), 1)
	})
	t.Run("score 10", func(t *testing.T) {
		assert.GreaterOrEqual(t, CountNewFaceMarkers(0, 10), 1)
	})
	t.Run("size 160", func(t *testing.T) {
		assert.GreaterOrEqual(t, CountNewFaceMarkers(160, 0), 1)
	})
	t.Run("score 50 and size 160", func(t *testing.T) {
		assert.GreaterOrEqual(t, CountNewFaceMarkers(160, 50), 1)
	})
}