75 lines
2.1 KiB
Go
75 lines
2.1 KiB
Go
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())
|
|
})
|
|
|
|
}
|