// +build osm package osm import ( "testing" "github.com/photoprism/photoprism/internal/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: "dipslay 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: "dipslay 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: "dipslay 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()) }