package search import ( "testing" "github.com/photoprism/photoprism/internal/entity" "github.com/photoprism/photoprism/pkg/txt" "github.com/stretchr/testify/assert" ) func TestLikeAny(t *testing.T) { t.Run("and_or_search", func(t *testing.T) { if w := LikeAny("k.keyword", "table spoon & usa | img json", true, false); len(w) != 2 { t.Fatal("two where conditions expected") } else { assert.Equal(t, "k.keyword LIKE 'spoon%' OR k.keyword LIKE 'table%'", w[0]) assert.Equal(t, "k.keyword LIKE 'json%' OR k.keyword LIKE 'usa'", w[1]) } }) t.Run(" exact and_or_search", func(t *testing.T) { if w := LikeAny("k.keyword", "table spoon & usa | img json", true, true); len(w) != 2 { t.Fatal("two where conditions expected") } else { assert.Equal(t, "k.keyword LIKE 'spoon' OR k.keyword LIKE 'table'", w[0]) assert.Equal(t, "k.keyword LIKE 'json' OR k.keyword LIKE 'usa'", w[1]) } }) t.Run("and_or_search_en", func(t *testing.T) { if w := LikeAny("k.keyword", "table spoon and usa or img json", true, false); len(w) != 2 { t.Fatal("two where conditions expected") } else { assert.Equal(t, "k.keyword LIKE 'spoon%' OR k.keyword LIKE 'table%'", w[0]) assert.Equal(t, "k.keyword LIKE 'json%' OR k.keyword LIKE 'usa'", w[1]) } }) t.Run("table spoon usa img json", func(t *testing.T) { if w := LikeAny("k.keyword", "table spoon usa img json", true, false); len(w) != 1 { t.Fatal("one where condition expected") } else { assert.Equal(t, "k.keyword LIKE 'json%' OR k.keyword LIKE 'spoon%' OR k.keyword LIKE 'table%' OR k.keyword LIKE 'usa'", w[0]) } }) t.Run("cat dog", func(t *testing.T) { if w := LikeAny("k.keyword", "cat dog", true, false); len(w) != 1 { t.Fatal("one where condition expected") } else { assert.Equal(t, "k.keyword LIKE 'cat' OR k.keyword LIKE 'dog'", w[0]) } }) t.Run("cats dogs", func(t *testing.T) { if w := LikeAny("k.keyword", "cats dogs", true, false); len(w) != 1 { t.Fatal("one where condition expected") } else { assert.Equal(t, "k.keyword LIKE 'cats%' OR k.keyword LIKE 'cat' OR k.keyword LIKE 'dogs%' OR k.keyword LIKE 'dog'", w[0]) } }) t.Run("spoon", func(t *testing.T) { if w := LikeAny("k.keyword", "spoon", true, false); len(w) != 1 { t.Fatal("one where condition expected") } else { assert.Equal(t, "k.keyword LIKE 'spoon%'", w[0]) } }) t.Run("img", func(t *testing.T) { if w := LikeAny("k.keyword", "img", true, false); len(w) > 0 { t.Fatal("no where condition expected") } }) t.Run("empty", func(t *testing.T) { if w := LikeAny("k.keyword", "", true, false); len(w) > 0 { t.Fatal("no where condition expected") } }) } func TestLikeAnyKeyword(t *testing.T) { t.Run("and_or_search", func(t *testing.T) { if w := LikeAnyKeyword("k.keyword", "table spoon & usa | img json"); len(w) != 2 { t.Fatal("two where conditions expected") } else { assert.Equal(t, "k.keyword LIKE 'spoon%' OR k.keyword LIKE 'table%'", w[0]) assert.Equal(t, "k.keyword LIKE 'json%' OR k.keyword LIKE 'usa'", w[1]) } }) t.Run("and_or_search_en", func(t *testing.T) { if w := LikeAnyKeyword("k.keyword", "table spoon and usa or img json"); len(w) != 2 { t.Fatal("two where conditions expected") } else { assert.Equal(t, "k.keyword LIKE 'spoon%' OR k.keyword LIKE 'table%'", w[0]) assert.Equal(t, "k.keyword LIKE 'json%' OR k.keyword LIKE 'usa'", w[1]) } }) } func TestLikeAnyWord(t *testing.T) { t.Run("and_or_search", func(t *testing.T) { if w := LikeAnyWord("k.keyword", "table spoon & usa | img json"); len(w) != 2 { t.Fatal("two where conditions expected") } else { assert.Equal(t, "k.keyword LIKE 'spoon%' OR k.keyword LIKE 'table%'", w[0]) assert.Equal(t, "k.keyword LIKE 'img%' OR k.keyword LIKE 'json%' OR k.keyword LIKE 'usa%'", w[1]) } }) t.Run("and_or_search_en", func(t *testing.T) { if w := LikeAnyWord("k.keyword", "table spoon and usa or img json"); len(w) != 2 { t.Fatal("two where conditions expected") } else { assert.Equal(t, "k.keyword LIKE 'spoon%' OR k.keyword LIKE 'table%'", w[0]) assert.Equal(t, "k.keyword LIKE 'img%' OR k.keyword LIKE 'json%' OR k.keyword LIKE 'usa%'", w[1]) } }) } func TestLikeAll(t *testing.T) { t.Run("keywords", func(t *testing.T) { if w := LikeAll("k.keyword", "Jo Mander 李", true, false); len(w) == 2 { assert.Equal(t, "k.keyword LIKE 'mander%'", w[0]) assert.Equal(t, "k.keyword LIKE '李'", w[1]) } else { t.Logf("wheres: %#v", w) t.Fatal("two where conditions expected") } }) t.Run("exact", func(t *testing.T) { if w := LikeAll("k.keyword", "Jo Mander 李", true, true); len(w) == 2 { assert.Equal(t, "k.keyword LIKE 'mander'", w[0]) assert.Equal(t, "k.keyword LIKE '李'", w[1]) } else { t.Logf("wheres: %#v", w) t.Fatal("two where conditions expected") } }) t.Run("string empty", func(t *testing.T) { w := LikeAll("k.keyword", "", true, true) assert.Empty(t, w) }) t.Run("0 words", func(t *testing.T) { w := LikeAll("k.keyword", "ab", true, true) assert.Empty(t, w) }) } func TestLikeAllKeywords(t *testing.T) { t.Run("keywords", func(t *testing.T) { if w := LikeAllKeywords("k.keyword", "Jo Mander 李"); len(w) == 2 { assert.Equal(t, "k.keyword LIKE 'mander%'", w[0]) assert.Equal(t, "k.keyword LIKE '李'", w[1]) } else { t.Logf("wheres: %#v", w) t.Fatal("two where conditions expected") } }) } func TestLikeAllWords(t *testing.T) { t.Run("keywords", func(t *testing.T) { if w := LikeAllWords("k.name", "Jo Mander 王"); len(w) == 3 { assert.Equal(t, "k.name LIKE 'jo%'", w[0]) assert.Equal(t, "k.name LIKE 'mander%'", w[1]) assert.Equal(t, "k.name LIKE '王%'", w[2]) } else { t.Logf("wheres: %#v", w) t.Fatal("two where conditions expected") } }) } func TestLikeAllNames(t *testing.T) { t.Run("MultipleNames", func(t *testing.T) { if w := LikeAllNames(Cols{"k.name"}, "j Mander 王"); len(w) == 1 { assert.Equal(t, "k.name LIKE 'j Mander 王%'", w[0]) } else { t.Fatalf("one where conditions expected: %#v", w) } }) t.Run("MultipleColumns", func(t *testing.T) { if w := LikeAllNames(Cols{"a.col1", "b.col2"}, "Mo Mander"); len(w) == 1 { assert.Equal(t, "a.col1 LIKE 'Mo Mander%' OR b.col2 LIKE 'Mo Mander%'", w[0]) } else { t.Fatalf("one where conditions expected: %#v", w) } }) t.Run("EmptyName", func(t *testing.T) { w := LikeAllNames(Cols{"k.name"}, "") assert.Empty(t, w) }) t.Run("NoWords", func(t *testing.T) { w := LikeAllNames(Cols{"k.name"}, "a") assert.Empty(t, w) }) t.Run("FullNames", func(t *testing.T) { if w := LikeAllNames(Cols{"j.name", "j.alias"}, "Bill & Melinda Gates"); len(w) == 2 { assert.Equal(t, "j.name LIKE 'Bill' OR j.name LIKE 'Bill %' OR j.alias LIKE 'Bill' OR j.alias LIKE 'Bill %'", w[0]) assert.Equal(t, "j.name LIKE 'Melinda Gates%' OR j.alias LIKE 'Melinda Gates%'", w[1]) } else { t.Fatalf("two where conditions expected: %#v", w) } }) t.Run("Plus", func(t *testing.T) { if w := LikeAllNames(Cols{"name"}, txt.NormalizeQuery("Paul + Paula")); len(w) == 2 { assert.Equal(t, "name LIKE 'paul' OR name LIKE 'paul %'", w[0]) assert.Equal(t, "name LIKE '%paula%'", w[1]) } else { t.Fatalf("two where conditions expected: %#v", w) } }) t.Run("Ane", func(t *testing.T) { if w := LikeAllNames(Cols{"name"}, txt.NormalizeQuery("Paul and Paula")); len(w) == 2 { assert.Equal(t, "name LIKE 'paul' OR name LIKE 'paul %'", w[0]) assert.Equal(t, "name LIKE '%paula%'", w[1]) } else { t.Fatalf("two where conditions expected: %#v", w) } }) t.Run("Or", func(t *testing.T) { if w := LikeAllNames(Cols{"name"}, txt.NormalizeQuery("Paul or Paula")); len(w) == 1 { assert.Equal(t, "name LIKE 'paul' OR name LIKE 'paul %' OR name LIKE '%paula%'", w[0]) } else { t.Fatalf("one where conditions expected: %#v", w) } }) } func TestAnySlug(t *testing.T) { t.Run("table spoon usa img json", func(t *testing.T) { where := AnySlug("custom_slug", "table spoon usa img json", " ") assert.Equal(t, "custom_slug = 'table' OR custom_slug = 'spoon' OR custom_slug = 'usa' OR custom_slug = 'img' OR custom_slug = 'json'", where) }) t.Run("cat dog", func(t *testing.T) { where := AnySlug("custom_slug", "cat dog", " ") assert.Equal(t, "custom_slug = 'cat' OR custom_slug = 'dog'", where) }) t.Run("cats dogs", func(t *testing.T) { where := AnySlug("custom_slug", "cats dogs", " ") assert.Equal(t, "custom_slug = 'cats' OR custom_slug = 'cat' OR custom_slug = 'dogs' OR custom_slug = 'dog'", where) }) t.Run("spoon", func(t *testing.T) { where := AnySlug("custom_slug", "spoon", " ") assert.Equal(t, "custom_slug = 'spoon'", where) }) t.Run("img", func(t *testing.T) { where := AnySlug("custom_slug", "img", " ") assert.Equal(t, "custom_slug = 'img'", where) }) t.Run("empty", func(t *testing.T) { where := AnySlug("custom_slug", "", " ") assert.Equal(t, "", where) }) t.Run("comma separated", func(t *testing.T) { where := AnySlug("custom_slug", "botanical-garden|landscape|bay", txt.Or) assert.Equal(t, "custom_slug = 'botanical-garden' OR custom_slug = 'landscape' OR custom_slug = 'bay'", where) }) t.Run("len = 0", func(t *testing.T) { where := AnySlug("custom_slug", " ", "") assert.Equal(t, "custom_slug = '' OR custom_slug = ''", where) }) } func TestAnyInt(t *testing.T) { t.Run("Empty", func(t *testing.T) { where := AnyInt("photos.photo_month", "", txt.Or, entity.UnknownMonth, txt.MonthMax) assert.Equal(t, "", where) }) t.Run("Range", func(t *testing.T) { where := AnyInt("photos.photo_month", "-3|0|10|9|11|12|13", txt.Or, entity.UnknownMonth, txt.MonthMax) assert.Equal(t, "photos.photo_month = 10 OR photos.photo_month = 9 OR photos.photo_month = 11 OR photos.photo_month = 12", where) }) t.Run("Chars", func(t *testing.T) { where := AnyInt("photos.photo_month", "a|b|c", txt.Or, entity.UnknownMonth, txt.MonthMax) assert.Equal(t, "", where) }) t.Run("CommaSeparated", func(t *testing.T) { where := AnyInt("photos.photo_month", "-3,10,9,11,12,13", ",", entity.UnknownMonth, txt.MonthMax) assert.Equal(t, "photos.photo_month = 10 OR photos.photo_month = 9 OR photos.photo_month = 11 OR photos.photo_month = 12", where) }) t.Run("Invalid", func(t *testing.T) { where := AnyInt("photos.photo_month", " , | ", ",", entity.UnknownMonth, txt.MonthMax) assert.Equal(t, "", where) }) } func TestOrLike(t *testing.T) { t.Run("Empty", func(t *testing.T) { where, values := OrLike("k.keyword", "") assert.Equal(t, "", where) assert.Equal(t, []interface{}{}, values) }) t.Run("OneTerm", func(t *testing.T) { where, values := OrLike("k.keyword", "bar") assert.Equal(t, "k.keyword LIKE ?", where) assert.Equal(t, []interface{}{"bar"}, values) }) t.Run("TwoTerms", func(t *testing.T) { where, values := OrLike("k.keyword", "foo*%|bar") assert.Equal(t, "k.keyword LIKE ? OR k.keyword LIKE ?", where) assert.Equal(t, []interface{}{"foo%", "bar"}, values) }) }