125 lines
3.5 KiB
Go
125 lines
3.5 KiB
Go
// +build osm
|
|
|
|
package osm
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/photoprism/photoprism/pkg/s2"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestFindLocation(t *testing.T) {
|
|
t.Run("Fernsehturm Berlin 1", func(t *testing.T) {
|
|
lat := 52.5208
|
|
lng := 13.40953
|
|
id := s2.Token(lat, lng)
|
|
|
|
l, err := FindLocation(id)
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
assert.False(t, l.Cached)
|
|
assert.Equal(t, "Fernsehturm Berlin", l.LocName)
|
|
assert.Equal(t, "10178", l.Address.Postcode)
|
|
assert.Equal(t, "Berlin", l.Address.State)
|
|
assert.Equal(t, "de", l.Address.CountryCode)
|
|
assert.Equal(t, "Germany", l.Address.Country)
|
|
|
|
l.PlaceID = 123456
|
|
|
|
assert.Equal(t, 123456, l.PlaceID)
|
|
|
|
cached, err := FindLocation(id)
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
assert.True(t, cached.Cached)
|
|
assert.Equal(t, l.LocName, cached.LocName)
|
|
assert.Equal(t, l.Address.Postcode, cached.Address.Postcode)
|
|
assert.Equal(t, l.Address.State, cached.Address.State)
|
|
assert.Equal(t, l.Address.CountryCode, cached.Address.CountryCode)
|
|
assert.Equal(t, l.Address.Country, cached.Address.Country)
|
|
})
|
|
|
|
t.Run("Fernsehturm Berlin 2", func(t *testing.T) {
|
|
lat := 52.52057
|
|
lng := 13.40889
|
|
id := s2.Token(lat, lng)
|
|
|
|
l, err := FindLocation(id)
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
assert.False(t, l.Cached)
|
|
assert.Equal(t, "Menschen Museum", l.LocName)
|
|
assert.Equal(t, "10178", l.Address.Postcode)
|
|
assert.Equal(t, "Berlin", l.Address.State)
|
|
assert.Equal(t, "de", l.Address.CountryCode)
|
|
assert.Equal(t, "Germany", l.Address.Country)
|
|
})
|
|
|
|
t.Run("No Location", func(t *testing.T) {
|
|
lat := 0.0
|
|
lng := 0.0
|
|
id := s2.Token(lat, lng)
|
|
|
|
l, err := FindLocation(id)
|
|
|
|
if err == nil {
|
|
t.Fatal("err should not be nil")
|
|
}
|
|
|
|
assert.Equal(t, "osm: invalid location id", err.Error())
|
|
assert.False(t, l.Cached)
|
|
})
|
|
}
|
|
|
|
func TestOSM_State(t *testing.T) {
|
|
t.Run("Berlin", func(t *testing.T) {
|
|
|
|
a := Address{CountryCode: "de", City: "Berlin", State: "Berlin", HouseNumber: "63", Suburb: "Neukölln"}
|
|
l := &Location{LocCategory: "natural", LocName: "Nice title", LocType: "hill", LocDisplayName: "display name", Address: a}
|
|
assert.Equal(t, "Berlin", l.State())
|
|
})
|
|
}
|
|
|
|
func TestOSM_Suburb(t *testing.T) {
|
|
t.Run("Neukölln", func(t *testing.T) {
|
|
|
|
a := Address{CountryCode: "de", City: "Berlin", State: "Berlin", HouseNumber: "63", Suburb: "Neukölln"}
|
|
l := &Location{LocCategory: "natural", LocName: "Nice title", LocType: "hill", LocDisplayName: "display name", Address: a}
|
|
assert.Equal(t, "Neukölln", l.Suburb())
|
|
})
|
|
}
|
|
|
|
func TestOSM_CountryCode(t *testing.T) {
|
|
t.Run("de", func(t *testing.T) {
|
|
|
|
a := Address{CountryCode: "de", City: "Berlin", State: "Berlin", HouseNumber: "63", Suburb: "Neukölln"}
|
|
l := &Location{LocCategory: "natural", LocName: "Nice title", LocType: "hill", LocDisplayName: "display name", Address: a}
|
|
assert.Equal(t, "de", l.CountryCode())
|
|
})
|
|
}
|
|
|
|
func TestOSM_Keywords(t *testing.T) {
|
|
t.Run("cat", func(t *testing.T) {
|
|
|
|
a := Address{CountryCode: "de", City: "Berlin", State: "Berlin", HouseNumber: "63", Suburb: "Neukölln"}
|
|
l := &Location{LocCategory: "natural", LocName: "Nice title", LocType: "hill", LocDisplayName: "cat", Address: a}
|
|
assert.Equal(t, []string{"cat"}, l.Keywords())
|
|
})
|
|
}
|
|
|
|
func TestOSM_Source(t *testing.T) {
|
|
|
|
a := Address{CountryCode: "de", City: "Berlin", State: "Berlin", HouseNumber: "63", Suburb: "Neukölln"}
|
|
l := &Location{LocCategory: "natural", LocName: "Nice title", LocType: "hill", LocDisplayName: "cat", Address: a}
|
|
assert.Equal(t, "osm", l.Source())
|
|
}
|