photoprism/internal/maps/location_test.go

package maps

import (
	"testing"

	"github.com/stretchr/testify/assert"

	"github.com/photoprism/photoprism/pkg/s2"
)

func TestLocation_QueryPlaces(t *testing.T) {
	t.Run("BerlinerRathaus", func(t *testing.T) {
		lat := 52.51961810676184
		lng := 13.40806264572578
		id := s2.Token(lat, lng)

		l := Location{ID: id}

		if err := l.QueryPlaces(); err != nil {
			t.Fatal(err)
		}

		t.Logf("BerlinerRathaus: %#v", l)

		assert.Equal(t, "Mitte, Berlin, Germany", l.LocLabel)
	})
	t.Run("BerlinerFernsehturm", func(t *testing.T) {
		lat := 52.5208
		lng := 13.40953
		id := s2.Token(lat, lng)

		l := Location{ID: id}

		if err := l.QueryPlaces(); err != nil {
			t.Fatal(err)
		}
		t.Logf("BerlinerFernsehturm: %#v", l)

		assert.Equal(t, "Berliner Fernsehturm", l.LocName)
		assert.Equal(t, "Berlin", l.LocState)
		assert.Equal(t, "Berlin", l.LocCity)
		assert.Equal(t, "Panoramastraße", l.LocStreet)
		assert.Equal(t, "10178", l.LocPostcode)
		assert.Equal(t, "Mitte", l.LocDistrict)
		assert.Equal(t, "Mitte, Berlin, Germany", l.LocLabel)
	})
	t.Run("NorthAtlanticOcean", func(t *testing.T) {
		id := "0a3c25fcffad"

		l := Location{ID: id}

		if err := l.QueryPlaces(); err != nil {
			t.Fatal(err)
		}

		assert.Equal(t, "", l.LocName)
		assert.Equal(t, "", l.LocState)
		assert.Equal(t, "ocean", l.LocCategory)
		assert.Equal(t, "North Atlantic Ocean", l.LocDistrict)
		assert.Equal(t, "North Atlantic Ocean", l.LocLabel)
	})
	t.Run("SouthPacificOcean", func(t *testing.T) {
		id := "9aa986feefb4"

		l := Location{ID: id}

		if err := l.QueryPlaces(); err != nil {
			t.Fatal(err)
		}

		assert.Equal(t, "Puerto Velasco Ibarra", l.LocName)
		assert.Equal(t, "", l.LocCategory)
		assert.Equal(t, "Galápagos", l.LocState)
		assert.Equal(t, "", l.LocDistrict)
		assert.Equal(t, "ec", l.LocCountry)
		assert.Equal(t, "Galápagos, Ecuador", l.LocLabel)
		assert.Equal(t, "places", l.LocSource)
	})
}

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 := Location{ID: 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 := Location{ID: id}

		assert.Equal(t, false, l.Unknown())
	})
}

func TestLocation_S2Token(t *testing.T) {
	t.Run("123", func(t *testing.T) {
		l := Location{ID: "123"}

		assert.Equal(t, "123", l.S2Token())
	})
}

func TestLocation_PrefixedToken(t *testing.T) {
	t.Run("123", func(t *testing.T) {
		l := Location{ID: "123"}

		assert.Equal(t, s2.TokenPrefix+"123", l.PrefixedToken())
	})
}

func TestLocation_Name(t *testing.T) {
	t.Run("Christkindlesmarkt", func(t *testing.T) {
		l := Location{ID: "123", LocName: "Christkindlesmarkt"}

		assert.Equal(t, "Christkindlesmarkt", l.Name())
	})
}

func TestLocation_City(t *testing.T) {
	t.Run("Nürnberg", func(t *testing.T) {
		l := Location{ID: "123", LocCity: "Nürnberg"}

		assert.Equal(t, "Nürnberg", l.City())
	})
}

func TestLocation_State(t *testing.T) {
	t.Run("Bayern", func(t *testing.T) {
		l := Location{ID: "123", LocState: "Bayern"}

		assert.Equal(t, "Bayern", l.State())
	})
}

func TestLocation_Category(t *testing.T) {
	t.Run("test", func(t *testing.T) {
		l := Location{ID: "123", LocCategory: "test"}

		assert.Equal(t, "test", l.Category())
	})
}

func TestLocation_Source(t *testing.T) {
	t.Run("source", func(t *testing.T) {
		l := Location{ID: "123", LocSource: "mySource"}

		assert.Equal(t, "mySource", l.Source())
	})
}

func TestLocation_Place(t *testing.T) {
	t.Run("test-label", func(t *testing.T) {
		l := Location{ID: "123", LocLabel: "test-label"}

		assert.Equal(t, "test-label", l.Label())
	})
}

func TestLocation_CountryCode(t *testing.T) {
	t.Run("de", func(t *testing.T) {
		l := Location{ID: "123", LocCountry: "de"}

		assert.Equal(t, "de", l.CountryCode())
	})
}

func TestLocation_CountryName(t *testing.T) {
	t.Run("Germany", func(t *testing.T) {
		l := Location{ID: "123", LocCountry: "de"}

		assert.Equal(t, "Germany", l.CountryName())
	})
}

func TestLocation_QueryApi(t *testing.T) {
	l := Location{ID: "3", LocCountry: "de"}

	t.Run("xxx", func(t *testing.T) {
		api := l.QueryApi("xxx")
		assert.Error(t, api, "maps: reverse lookup disabled")
	})
	t.Run("places", func(t *testing.T) {
		api := l.QueryApi("places")
		assert.Error(t, api)
	})

}