276 lines
6.2 KiB
Go
276 lines
6.2 KiB
Go
package maps
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/photoprism/photoprism/internal/maps/places"
|
|
"github.com/photoprism/photoprism/pkg/s2"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestLocation_QueryPlaces(t *testing.T) {
|
|
t.Run("U Berliner Rathaus", func(t *testing.T) {
|
|
lat := 52.51961810676184
|
|
lng := 13.40806264572578
|
|
id := s2.Token(lat, lng)
|
|
|
|
l := NewLocation(id)
|
|
|
|
if err := l.QueryPlaces(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
assert.Equal(t, "Alt-Berlin", l.LocName)
|
|
assert.Equal(t, "Berlin, Germany", l.LocLabel)
|
|
})
|
|
}
|
|
|
|
func TestLocation_Assign(t *testing.T) {
|
|
t.Run("BerlinFernsehturm", func(t *testing.T) {
|
|
lat := 52.5208
|
|
lng := 13.40953
|
|
id := s2.Token(lat, lng)
|
|
|
|
o, err := places.FindLocation(id)
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
assert.Equal(t, "Fernsehturm Berlin", o.Name())
|
|
assert.Equal(t, "Berlin", o.State())
|
|
assert.Equal(t, "de", o.CountryCode())
|
|
|
|
var l Location
|
|
|
|
if err := l.Assign(o); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
assert.Equal(t, "Fernsehturm Berlin", l.LocName)
|
|
assert.Equal(t, "Berlin, Germany", l.LocLabel)
|
|
})
|
|
|
|
t.Run("SantaMonica", func(t *testing.T) {
|
|
lat := 34.00909444444444
|
|
lng := -118.49700833333334
|
|
id := s2.Token(lat, lng)
|
|
|
|
o, err := places.FindLocation(id)
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
assert.False(t, o.Cached)
|
|
assert.Equal(t, "California", o.State())
|
|
assert.Equal(t, "us", o.CountryCode())
|
|
|
|
var l Location
|
|
|
|
if err := l.Assign(o); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
assert.Equal(t, "Santa Monica, California, USA", l.LocLabel)
|
|
})
|
|
|
|
t.Run("AirportZurich", func(t *testing.T) {
|
|
lat := 47.45401666666667
|
|
lng := 8.557494444444446
|
|
id := s2.Token(lat, lng)
|
|
|
|
o, err := places.FindLocation(id)
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
var l Location
|
|
|
|
if err := l.Assign(o); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
assert.Equal(t, "Airport", l.LocName)
|
|
assert.Equal(t, "Kloten, Zurich, Switzerland", l.LocLabel)
|
|
})
|
|
|
|
t.Run("AirportTegel", func(t *testing.T) {
|
|
lat := 52.559864397033024
|
|
lng := 13.28895092010498
|
|
id := s2.Token(lat, lng)
|
|
|
|
o, err := places.FindLocation(id)
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
var l Location
|
|
|
|
if err := l.Assign(o); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
assert.Equal(t, "Airport", l.LocName)
|
|
assert.Equal(t, "Berlin, Germany", l.LocLabel)
|
|
})
|
|
|
|
t.Run("SouthAfrica", func(t *testing.T) {
|
|
lat := -31.976301666666668
|
|
lng := 29.148046666666666
|
|
id := s2.Token(lat, lng)
|
|
|
|
o, err := places.FindLocation(id)
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
assert.False(t, o.Cached)
|
|
|
|
assert.Equal(t, "", o.Name())
|
|
assert.Equal(t, "Eastern Cape", o.State())
|
|
assert.Equal(t, "za", o.CountryCode())
|
|
|
|
var l Location
|
|
|
|
if err := l.Assign(o); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
assert.True(t, strings.HasPrefix(l.ID, "1e5e4205"))
|
|
assert.Equal(t, "", l.LocName)
|
|
assert.Equal(t, "Eastern Cape, South Africa", l.LocLabel)
|
|
})
|
|
|
|
t.Run("ocean", func(t *testing.T) {
|
|
lat := -21.976301666666668
|
|
lng := 49.148046666666666
|
|
id := s2.Token(lat, lng)
|
|
log.Printf("ID: %s", id)
|
|
o, err := places.FindLocation(id)
|
|
|
|
log.Printf("Output: %+v", o)
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
var l Location
|
|
|
|
if err := l.Assign(o); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
assert.Equal(t, "Indian Ocean", l.LocName)
|
|
assert.Equal(t, "", l.LocCategory)
|
|
assert.Equal(t, "", l.LocCity)
|
|
// TODO: Should be zz for international waters, fixed in places server
|
|
// assert.Equal(t, "", l.LocCountry)
|
|
})
|
|
}
|
|
|
|
func TestLocation_Unknown(t *testing.T) {
|
|
t.Run("true", func(t *testing.T) {
|
|
lat := 0.0
|
|
lng := 0.0
|
|
id := s2.Token(lat, lng)
|
|
|
|
l := NewLocation(id)
|
|
|
|
assert.Equal(t, true, l.Unknown())
|
|
})
|
|
t.Run("false", func(t *testing.T) {
|
|
lat := -31.976301666666668
|
|
lng := 29.148046666666666
|
|
id := s2.Token(lat, lng)
|
|
|
|
l := NewLocation(id)
|
|
|
|
assert.Equal(t, false, l.Unknown())
|
|
})
|
|
}
|
|
|
|
func TestLocation_place(t *testing.T) {
|
|
t.Run("unknown", func(t *testing.T) {
|
|
lat := 0.0
|
|
lng := 0.0
|
|
id := s2.Token(lat, lng)
|
|
|
|
l := NewLocation(id)
|
|
|
|
assert.Equal(t, "Unknown", l.label())
|
|
})
|
|
t.Run("Nürnberg, Bayern, Germany", func(t *testing.T) {
|
|
l := &Location{LocCountry: "de", LocCity: "Nürnberg", LocState: "Bayern"}
|
|
|
|
assert.Equal(t, "Unknown", l.label())
|
|
})
|
|
}
|
|
|
|
func TestLocation_Name(t *testing.T) {
|
|
t.Run("Christkindlesmarkt", func(t *testing.T) {
|
|
l := &Location{LocCountry: "de", LocCity: "Nürnberg", LocState: "Bayern", LocName: "Christkindlesmarkt"}
|
|
|
|
assert.Equal(t, "Christkindlesmarkt", l.Name())
|
|
})
|
|
}
|
|
|
|
func TestLocation_City(t *testing.T) {
|
|
t.Run("Nürnberg", func(t *testing.T) {
|
|
l := &Location{LocCountry: "de", LocCity: "Nürnberg", LocState: "Bayern", LocName: "Christkindlesmarkt"}
|
|
|
|
assert.Equal(t, "Nürnberg", l.City())
|
|
})
|
|
}
|
|
|
|
func TestLocation_State(t *testing.T) {
|
|
t.Run("Bayern", func(t *testing.T) {
|
|
l := &Location{LocCountry: "de", LocCity: "Nürnberg", LocState: "Bayern", LocName: "Christkindlesmarkt"}
|
|
|
|
assert.Equal(t, "Bayern", l.State())
|
|
})
|
|
}
|
|
|
|
func TestLocation_Category(t *testing.T) {
|
|
t.Run("test", func(t *testing.T) {
|
|
l := &Location{LocCategory: "test", LocCountry: "de", LocCity: "Nürnberg", LocState: "Bayern", LocName: "Christkindlesmarkt"}
|
|
|
|
assert.Equal(t, "test", l.Category())
|
|
})
|
|
}
|
|
|
|
func TestLocation_Source(t *testing.T) {
|
|
t.Run("source", func(t *testing.T) {
|
|
l := &Location{LocCategory: "test", LocCountry: "de", LocCity: "Nürnberg", LocState: "Bayern", LocName: "Christkindlesmarkt", LocSource: "source"}
|
|
|
|
assert.Equal(t, "source", l.Source())
|
|
})
|
|
}
|
|
|
|
func TestLocation_Place(t *testing.T) {
|
|
t.Run("test-label", func(t *testing.T) {
|
|
l := &Location{LocCategory: "test", LocCountry: "de", LocCity: "Nürnberg", LocLabel: "test-label", LocState: "Bayern", LocName: "Christkindlesmarkt"}
|
|
|
|
assert.Equal(t, "test-label", l.Label())
|
|
})
|
|
}
|
|
|
|
func TestLocation_CountryCode(t *testing.T) {
|
|
t.Run("de", func(t *testing.T) {
|
|
l := &Location{LocCategory: "test", LocCountry: "de", LocCity: "Nürnberg", LocLabel: "test-label", LocState: "Bayern", LocName: "Christkindlesmarkt"}
|
|
|
|
assert.Equal(t, "de", l.CountryCode())
|
|
})
|
|
}
|
|
|
|
func TestLocation_CountryName(t *testing.T) {
|
|
t.Run("Germany", func(t *testing.T) {
|
|
l := &Location{LocCategory: "test", LocCountry: "de", LocCity: "Nürnberg", LocLabel: "test-label", LocState: "Bayern", LocName: "Christkindlesmarkt"}
|
|
|
|
assert.Equal(t, "Germany", l.CountryName())
|
|
})
|
|
}
|