90 lines
2.3 KiB
Go
90 lines
2.3 KiB
Go
package entity
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestNewLocation(t *testing.T) {
|
|
t.Run("new label", func(t *testing.T) {
|
|
l := NewGeo(1, 1)
|
|
l.GeoCategory = "restaurant"
|
|
l.GeoName = "LocationName"
|
|
l.Place = PlaceFixtures.Pointer("zinkwazi")
|
|
|
|
assert.Equal(t, "restaurant", l.Category())
|
|
assert.Equal(t, false, l.NoCategory())
|
|
assert.Equal(t, false, l.Unknown())
|
|
assert.Equal(t, "LocationName", l.Name())
|
|
assert.Equal(t, false, l.NoName())
|
|
assert.Equal(t, "KwaDukuza, KwaZulu-Natal, South Africa", l.Label())
|
|
assert.Equal(t, "KwaDukuza", l.City())
|
|
assert.Equal(t, false, l.LongCity())
|
|
assert.Equal(t, false, l.CityContains("xxx"))
|
|
assert.Equal(t, false, l.NoCity())
|
|
assert.Equal(t, "KwaZulu-Natal", l.State())
|
|
assert.Equal(t, false, l.NoState())
|
|
assert.Equal(t, "za", l.CountryCode())
|
|
assert.Equal(t, "South Africa", l.CountryName())
|
|
})
|
|
}
|
|
|
|
func TestLocation_Keywords(t *testing.T) {
|
|
t.Run("mexico", func(t *testing.T) {
|
|
m := GeoFixtures["mexico"]
|
|
r := m.Keywords()
|
|
assert.Equal(t, []string{"adosada", "ancient", "botanical", "garden", "mexico", "platform", "pyramid", "state-of-mexico", "teotihuacán"}, r)
|
|
})
|
|
t.Run("caravan park", func(t *testing.T) {
|
|
m := GeoFixtures["caravan park"]
|
|
r := m.Keywords()
|
|
assert.Equal(t, []string{"camping", "caravan", "kwazulu-natal", "lobotes", "mandeni", "park", "south-africa"}, r)
|
|
})
|
|
t.Run("place id empty", func(t *testing.T) {
|
|
m := &Geo{}
|
|
r := m.Keywords()
|
|
assert.Empty(t, r)
|
|
})
|
|
}
|
|
|
|
func TestLocation_Find(t *testing.T) {
|
|
t.Run("place in db", func(t *testing.T) {
|
|
m := GeoFixtures["mexico"]
|
|
r := m.Find("")
|
|
assert.Nil(t, r)
|
|
})
|
|
t.Run("invalid api", func(t *testing.T) {
|
|
l := NewGeo(2, 1)
|
|
err := l.Find("")
|
|
|
|
if err == nil {
|
|
t.Fatal("error expected")
|
|
}
|
|
|
|
assert.Equal(t, "maps: reverse lookup disabled", err.Error())
|
|
})
|
|
}
|
|
|
|
func TestFirstOrCreateLocation(t *testing.T) {
|
|
t.Run("id empty", func(t *testing.T) {
|
|
loc := &Geo{}
|
|
|
|
assert.Nil(t, FirstOrCreateGeo(loc))
|
|
})
|
|
t.Run("place id empty", func(t *testing.T) {
|
|
loc := &Geo{ID: "1234jhy"}
|
|
|
|
assert.Nil(t, FirstOrCreateGeo(loc))
|
|
})
|
|
t.Run("success", func(t *testing.T) {
|
|
loc := GeoFixtures.Pointer("caravan park")
|
|
|
|
result := FirstOrCreateGeo(loc)
|
|
|
|
if result == nil {
|
|
t.Fatal("result should not be nil")
|
|
}
|
|
assert.NotEmpty(t, result.ID)
|
|
})
|
|
}
|