photoprism/internal/entity/place_test.go

131 lines
2.9 KiB
Go
Raw Normal View History

2020-05-08 14:18:11 +02:00
package entity
import (
"testing"
"github.com/photoprism/photoprism/pkg/s2"
"github.com/stretchr/testify/assert"
2020-05-08 14:18:11 +02:00
)
func TestCreateUnknownPlace(t *testing.T) {
r := FirstOrCreatePlace(&UnknownPlace)
2020-05-08 14:18:11 +02:00
assert.True(t, r.Unknown())
}
func TestFindPlace(t *testing.T) {
t.Run("Holiday Park", func(t *testing.T) {
r := FindPlace("de:HFqPHxa2Hsol")
if r == nil {
t.Fatal("result should not be nil")
}
assert.Equal(t, "de", r.PlaceCountry)
2020-05-08 14:18:11 +02:00
})
t.Run("Mexico", func(t *testing.T) {
r := FindPlace("mx:VvfNBpFegSCr")
2020-05-15 13:15:10 +02:00
if r == nil {
t.Fatal("result should not be nil")
}
assert.Equal(t, "mx", r.PlaceCountry)
2020-05-15 13:15:10 +02:00
})
t.Run("KwaDukuza", func(t *testing.T) {
r := FindPlace("za:Rc1K7dTWRzBD")
if r == nil {
t.Fatal("result should not be nil")
}
assert.Equal(t, "za", r.PlaceCountry)
2020-05-08 14:18:11 +02:00
})
t.Run("not matching", func(t *testing.T) {
r := FindPlace("111")
if r != nil {
t.Fatal("result should be nil")
}
})
t.Run("not matching empty label", func(t *testing.T) {
r := FindPlace("111")
if r != nil {
t.Fatal("result should be nil")
}
2020-05-08 14:18:11 +02:00
})
}
func TestPlace_Find(t *testing.T) {
t.Run("record exists", func(t *testing.T) {
m := PlaceFixtures.Get("mexico")
if err := m.Find(); err != nil {
t.Fatal(err)
}
2020-05-08 14:18:11 +02:00
})
t.Run("record does not exist", func(t *testing.T) {
place := &Place{
ID: s2.TokenPrefix + "1110",
PlaceLabel: "test",
PlaceCity: "testCity",
PlaceState: "",
PlaceCountry: "",
PlaceKeywords: "",
PlaceFavorite: false,
PhotoCount: 0,
CreatedAt: TimeStamp(),
UpdatedAt: TimeStamp(),
}
err := place.Find()
assert.EqualError(t, err, "record not found")
2020-05-08 14:18:11 +02:00
})
}
func TestFirstOrCreatePlace(t *testing.T) {
t.Run("existing place", func(t *testing.T) {
m := PlaceFixtures.Pointer("zinkwazi")
r := FirstOrCreatePlace(m)
assert.Equal(t, "KwaDukuza, KwaZulu-Natal, South Africa", r.PlaceLabel)
})
t.Run("ID empty", func(t *testing.T) {
p := &Place{ID: ""}
assert.Nil(t, FirstOrCreatePlace(p))
})
t.Run("PlaceLabel empty", func(t *testing.T) {
p := &Place{ID: "abcde44", PlaceLabel: ""}
assert.Nil(t, FirstOrCreatePlace(p))
})
}
func TestPlace_LongCity(t *testing.T) {
t.Run("true", func(t *testing.T) {
p := Place{PlaceCity: "veryveryveryverylongcity"}
assert.True(t, p.LongCity())
})
t.Run("false", func(t *testing.T) {
p := Place{PlaceCity: "short"}
assert.False(t, p.LongCity())
})
}
func TestPlace_NoCity(t *testing.T) {
t.Run("true", func(t *testing.T) {
p := Place{PlaceCity: ""}
assert.True(t, p.NoCity())
})
t.Run("false", func(t *testing.T) {
p := Place{PlaceCity: "short"}
assert.False(t, p.NoCity())
})
}
func TestPlace_CityContains(t *testing.T) {
t.Run("true", func(t *testing.T) {
p := Place{PlaceCity: "Munich"}
assert.True(t, p.CityContains("Munich"))
})
t.Run("false", func(t *testing.T) {
p := Place{PlaceCity: "short"}
assert.False(t, p.CityContains("ich"))
})
2020-05-08 14:18:11 +02:00
}