package places import ( "testing" "github.com/photoprism/photoprism/pkg/s2" "github.com/stretchr/testify/assert" ) 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) { var p = NewPlace("1", "", "", "", "de", "") location := NewLocation("1e95998417cc", 52.51961810676184, 13.40806264572578, "TestLocation", "test", p, 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) { var p = NewPlace("1", "testLabel", "berlin", "berlin", "de", "foobar") location := NewLocation("1e95998417cc", 52.51961810676184, 13.40806264572578, "TestLocation", "test", p, 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.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()) }) }