internal/entity/cell_test.go - develop - githubmirror/photoprism

internal/entity/cell_test.go
125 строк3 KB
Michael Mayer
Entity: Fix cell lookups
07 мар 2026, 13:48
07 мар 2026, 13:48168834a
package entity

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestNewCell(t *testing.T) {
	t.Run("NewCell", func(t *testing.T) {
		l := NewCell(1, 1)
		l.CellCategory = "restaurant"
		l.CellName = "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, true, l.NoStreet())
		assert.Equal(t, true, l.NoPostcode())
		assert.Equal(t, "", l.Postcode())
		assert.Equal(t, "za", l.CountryCode())
		assert.Equal(t, "South Africa", l.CountryName())
	})
}

func TestCell_Keywords(t *testing.T) {
	t.Run("Mexico", func(t *testing.T) {
		m := CellFixtures["mexico"]
		r := m.Keywords()
		assert.Equal(t, []string{"adosada", "ancient", "botanical", "garden", "mexico", "platform", "pyramid", "state-of-mexico", "teotihuacán"}, r)
	})
	t.Run("CaravanPark", func(t *testing.T) {
		m := CellFixtures["caravan park"]
		r := m.Keywords()
		assert.Equal(t, []string{"camping", "caravan", "kwazulu-natal", "lobotes", "mandeni", "park", "south-africa"}, r)
	})
	t.Run("CellIdEmpty", func(t *testing.T) {
		m := &Cell{}
		r := m.Keywords()
		assert.Empty(t, r)
	})
}

func TestCell_Find(t *testing.T) {
	t.Run("CellInDb", func(t *testing.T) {
		m := CellFixtures["mexico"]
		r := m.Find("")
		assert.Nil(t, r)
	})
	t.Run("RepairsMissingPlaceAssociation", func(t *testing.T) {
		fixture := CellFixtures["mexico"]
		fixture.Place = nil

		assert.NotPanics(t, func() {
			assert.Equal(t, "Teotihuacán", fixture.City())
		})
		assert.Equal(t, "State of Mexico", fixture.State())
		assert.Equal(t, "Mexico", fixture.CountryName())
	})
	t.Run("MissingPlaceFallsBackToUnknown", func(t *testing.T) {
		m := &Cell{ID: "s2:testmissingplace", PlaceID: "zz:missing"}

		assert.NotPanics(t, func() {
			assert.Equal(t, "Unknown", m.City())
		})
		assert.Equal(t, "Unknown", m.State())
		assert.Equal(t, "Unknown", m.CountryName())
	})
	t.Run("ApiEmpty", func(t *testing.T) {
		l := NewCell(2, 1)
		err := l.Find("")

		if err == nil {
			t.Fatal("error expected")
		}

		assert.Equal(t, "maps: location lookup disabled", err.Error())
	})
}

func TestFirstOrCreateCell(t *testing.T) {
	t.Run("IdEmpty", func(t *testing.T) {
		loc := &Cell{}

		assert.Nil(t, FirstOrCreateCell(loc))
	})
	t.Run("PlaceIdEmpty", func(t *testing.T) {
		loc := &Cell{ID: "1234jhy"}

		assert.Nil(t, FirstOrCreateCell(loc))
	})
	t.Run("Success", func(t *testing.T) {
		loc := CellFixtures.Pointer("caravan park")

		result := FirstOrCreateCell(loc)

		if result == nil {
			t.Fatal("result must not be nil")
		}
		assert.NotEmpty(t, result.ID)
	})
}

func TestCell_Refresh(t *testing.T) {
	t.Run("ApiEmpty", func(t *testing.T) {
		l := NewCell(2, 1)
		err := l.Refresh("")

		if err == nil {
			t.Fatal("error expected")
		}

		assert.Equal(t, "maps: location lookup disabled", err.Error())
	})
}