package search import ( "testing" "github.com/photoprism/photoprism/internal/entity" "github.com/photoprism/photoprism/internal/form" "github.com/stretchr/testify/assert" ) func TestSubjects(t *testing.T) { t.Run("FindAll", func(t *testing.T) { results, err := Subjects(form.SubjectSearch{Type: entity.SubjPerson}) assert.NoError(t, err) // t.Logf("Subjects: %#v", results) assert.LessOrEqual(t, 3, len(results)) }) t.Run("Find 2 subjects, sort by count", func(t *testing.T) { results, err := Subjects(form.SubjectSearch{Type: entity.SubjPerson, Count: 2, Order: "count"}) assert.NoError(t, err) //t.Logf("Subjects: %#v", results) assert.Greater(t, results[0].FileCount, results[1].FileCount) assert.Equal(t, 2, len(results)) }) t.Run("FindAll sort by name", func(t *testing.T) { results, err := Subjects(form.SubjectSearch{Type: entity.SubjPerson, Order: "name"}) assert.NoError(t, err) //t.Logf("Subjects: %#v", results) assert.Equal(t, "Actor A", results[0].SubjName) assert.LessOrEqual(t, 3, len(results)) }) t.Run("sort by added", func(t *testing.T) { results, err := Subjects(form.SubjectSearch{Type: entity.SubjPerson, Order: "added"}) assert.NoError(t, err) //t.Logf("Subjects: %#v", results) assert.Equal(t, "Jane Doe", results[0].SubjName) assert.LessOrEqual(t, 3, len(results)) }) t.Run("sort by relevance", func(t *testing.T) { results, err := Subjects(form.SubjectSearch{Type: entity.SubjPerson, Order: "relevance"}) assert.NoError(t, err) //t.Logf("Subjects: %#v", results) assert.Equal(t, "John Doe", results[0].SubjName) assert.LessOrEqual(t, 3, len(results)) }) t.Run("search favorite", func(t *testing.T) { results, err := Subjects(form.SubjectSearch{Type: entity.SubjPerson, Favorite: true}) assert.NoError(t, err) //t.Logf("Subjects: %#v", results) assert.Equal(t, "John Doe", results[0].SubjName) assert.LessOrEqual(t, 1, len(results)) }) t.Run("search private", func(t *testing.T) { results, err := Subjects(form.SubjectSearch{Type: entity.SubjPerson, Private: true}) assert.NoError(t, err) //t.Logf("Subjects: %#v", results) assert.Equal(t, 0, len(results)) }) t.Run("search excluded", func(t *testing.T) { results, err := Subjects(form.SubjectSearch{Type: entity.SubjPerson, Excluded: true}) assert.NoError(t, err) //t.Logf("Subjects: %#v", results) assert.Equal(t, 0, len(results)) }) t.Run("search file count >2", func(t *testing.T) { results, err := Subjects(form.SubjectSearch{Type: entity.SubjPerson, Files: 2}) assert.NoError(t, err) //t.Logf("Subjects: %#v", results) assert.LessOrEqual(t, 1, len(results)) }) t.Run("search for alias", func(t *testing.T) { results, err := Subjects(form.SubjectSearch{Type: entity.SubjPerson, Query: "Powell"}) assert.NoError(t, err) //t.Logf("Subjects: %#v", results) assert.Equal(t, "Dangling Subject", results[0].SubjName) assert.LessOrEqual(t, 1, len(results)) }) t.Run("search for ID", func(t *testing.T) { results, err := Subjects(form.SubjectSearch{Type: entity.SubjPerson, ID: "jqy3y652h8njw0sx"}) assert.NoError(t, err) //t.Logf("Subjects: %#v", results) assert.Equal(t, "Joe Biden", results[0].SubjName) assert.Equal(t, 1, len(results)) }) } func TestSubjectUIDs(t *testing.T) { t.Run("search for alias", func(t *testing.T) { results, _, _ := SubjectUIDs("Powell") //t.Logf("Subjects: %#v", results) //t.Logf("Names: %#v", names) assert.Equal(t, 1, len(results)) }) t.Run("search for not existing name", func(t *testing.T) { results, _, _ := SubjectUIDs("Anonymous") //t.Logf("Subjects: %#v", results) //t.Logf("Names: %#v", names) assert.Equal(t, 0, len(results)) }) t.Run("search with empty string", func(t *testing.T) { results, _, _ := SubjectUIDs("") //t.Logf("Subjects: %#v", results) //t.Logf("Names: %#v", names) assert.Equal(t, 0, len(results)) }) }