photoprism/internal/hub/places/location_test.go

141 lines
4.1 KiB
Go
Raw Normal View History

package places
import (
"testing"
"github.com/stretchr/testify/assert"
"github.com/photoprism/photoprism/pkg/s2"
)
func TestFindLocation(t *testing.T) {
t.Run("U Berliner Rathaus", func(t *testing.T) {
lat := 52.51961810676184
lng := 13.40806264572578
id := s2.Token(lat, lng)
l, err := FindLocation(id)
if err != nil {
t.Fatal(err)
}
assert.False(t, l.Cached)
assert.Equal(t, "Berlin", l.City())
assert.Equal(t, "de", l.CountryCode())
})
t.Run("wrong id", func(t *testing.T) {
l, err := FindLocation("2")
assert.Error(t, err, "places: skipping lat 0.000000, lng 0.000000")
t.Log(l)
})
t.Run("short id", func(t *testing.T) {
l, err := FindLocation("ab")
assert.Error(t, err, "places: skipping lat 0.000000, lng 0.000000")
t.Log(l)
})
t.Run("invalid id", func(t *testing.T) {
l, err := FindLocation("")
assert.Error(t, err, "places: invalid location id ")
t.Log(l)
})
t.Run("cached true", func(t *testing.T) {
location := Location{
ID: "1e95998417cc",
LocLat: 52.51961810676184,
LocLng: 13.40806264572578,
LocName: "TestLocation",
LocStreet: "",
LocPostcode: "",
LocCategory: "test",
Place: Place{PlaceID: "1"},
Cached: true,
}
l, err := FindLocation(location.ID)
if err != nil {
t.Fatal(err)
}
assert.Equal(t, false, l.Cached)
l2, err2 := FindLocation("1e95998417cc")
if err2 != nil {
t.Fatal(err2)
}
assert.Equal(t, true, l2.Cached)
})
}
func TestLocationGetters(t *testing.T) {
location := Location{
ID: "1e95998417cc",
LocLat: 52.51961810676184,
LocLng: 13.40806264572578,
LocName: "TestLocation",
LocStreet: "",
LocPostcode: "",
LocCategory: "test",
Place: Place{PlaceID: "1", LocLabel: "testLabel", LocDistrict: "Berlin", LocCity: "", LocState: "Berlin", LocCountry: "de", LocKeywords: "foobar"},
Cached: true,
}
t.Run("wrong id", func(t *testing.T) {
assert.Equal(t, "1e95998417cc", location.CellID())
assert.Equal(t, "TestLocation", location.Name())
assert.Equal(t, "test", location.Category())
assert.Equal(t, "testLabel", location.Label())
assert.Equal(t, "Berlin", location.State())
assert.Equal(t, "de", location.CountryCode())
assert.Equal(t, "Berlin", location.District())
assert.Equal(t, "", location.City())
assert.Equal(t, 52.51961810676184, location.Latitude())
assert.Equal(t, 13.40806264572578, location.Longitude())
assert.Equal(t, "places", location.Source())
assert.Equal(t, []string{"foobar"}, location.Keywords())
})
}
func TestLocation_State(t *testing.T) {
location := Location{
ID: "54903ee07f74",
LocLat: 47.6129432,
LocLng: -122.4821475,
LocName: "TestLocation",
LocStreet: "",
LocPostcode: "",
LocCategory: "test",
Place: Place{PlaceID: "549ed22c0434", LocLabel: "Seattle, WA", LocDistrict: "Berlin", LocCity: "Seattle", LocState: "WA", LocCountry: "us", LocKeywords: "foobar"},
Cached: true,
}
t.Run("Washington", func(t *testing.T) {
assert.Equal(t, "54903ee07f74", location.CellID())
assert.Equal(t, "Seattle, WA", location.Label())
assert.Equal(t, "Washington", location.State())
assert.Equal(t, "us", location.CountryCode())
assert.Equal(t, "Seattle", location.City())
assert.Equal(t, "places", location.Source())
})
}
func TestLocation_District(t *testing.T) {
location := Location{
ID: "54903ee07f74",
LocLat: 47.6129432,
LocLng: -122.4821475,
LocName: "TestLocation",
LocStreet: "",
LocPostcode: "",
LocCategory: "test",
Place: Place{PlaceID: "549ed22c0434", LocLabel: "Seattle, WA", LocDistrict: "Foo", LocCity: "Seattle", LocState: "WA", LocCountry: "us", LocKeywords: "foobar"},
Cached: true,
}
t.Run("Washington", func(t *testing.T) {
assert.Equal(t, "54903ee07f74", location.CellID())
assert.Equal(t, "Seattle, WA", location.Label())
assert.Equal(t, "Foo", location.District())
assert.Equal(t, "Washington", location.State())
assert.Equal(t, "us", location.CountryCode())
assert.Equal(t, "Seattle", location.City())
assert.Equal(t, "places", location.Source())
})
}