102 lines
2.4 KiB
Go
102 lines
2.4 KiB
Go
package entity
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestPhoto_GetTimeZone(t *testing.T) {
|
|
m := Photo{}
|
|
m.PhotoLat = 48.533905555
|
|
m.PhotoLng = 9.01
|
|
|
|
result := m.GetTimeZone()
|
|
|
|
if result != "Europe/Berlin" {
|
|
t.Fatalf("time zone should be Europe/Berlin: %s", result)
|
|
}
|
|
}
|
|
|
|
func TestPhoto_GetTakenAt(t *testing.T) {
|
|
m := Photo{}
|
|
m.PhotoLat = 48.533905555
|
|
m.PhotoLng = 9.01
|
|
m.TakenAt, _ = time.Parse(time.RFC3339, "2020-02-04T11:54:34Z")
|
|
m.TakenAtLocal, _ = time.Parse(time.RFC3339, "2020-02-04T11:54:34Z")
|
|
m.TimeZone = m.GetTimeZone()
|
|
|
|
if m.TimeZone != "Europe/Berlin" {
|
|
t.Fatalf("time zone should be Europe/Berlin: %s", m.TimeZone)
|
|
}
|
|
|
|
localTime := m.TakenAtLocal.Format("2006-01-02T15:04:05")
|
|
|
|
if localTime != "2020-02-04T11:54:34" {
|
|
t.Fatalf("local time should be 2020-02-04T11:54:34: %s", localTime)
|
|
}
|
|
|
|
utcTime := m.GetTakenAt().Format("2006-01-02T15:04:05")
|
|
|
|
if utcTime != "2020-02-04T10:54:34" {
|
|
t.Fatalf("utc time should be 2020-02-04T10:54:34: %s", utcTime)
|
|
}
|
|
}
|
|
|
|
func TestPhoto_CountryName(t *testing.T) {
|
|
t.Run("Unknown", func(t *testing.T) {
|
|
m := Photo{PhotoCountry: "xx"}
|
|
assert.Equal(t, "Unknown", m.CountryName())
|
|
})
|
|
t.Run("Germany", func(t *testing.T) {
|
|
m := Photo{PhotoCountry: "de"}
|
|
assert.Equal(t, "Germany", m.CountryName())
|
|
})
|
|
}
|
|
|
|
func TestUpdateLocation(t *testing.T) {
|
|
t.Run("estimate", func(t *testing.T) {
|
|
m := Photo{
|
|
PhotoName: "test_photo_2",
|
|
PhotoCountry: "zz",
|
|
PhotoLat: 0.0,
|
|
PhotoLng: 0.0,
|
|
PlaceID: "s2:85d1ea7d3278",
|
|
PlaceSrc: SrcEstimate,
|
|
}
|
|
|
|
assert.Equal(t, "Unknown", m.CountryName())
|
|
|
|
m.UpdateLocation()
|
|
|
|
assert.Equal(t, "Mexico", m.CountryName())
|
|
assert.Equal(t, "mx", m.PhotoCountry)
|
|
assert.Equal(t, float32(0.0), m.PhotoLat)
|
|
assert.Equal(t, float32(0.0), m.PhotoLng)
|
|
assert.Equal(t, "s2:85d1ea7d3278", m.PlaceID)
|
|
assert.Equal(t, SrcEstimate, m.PlaceSrc)
|
|
})
|
|
|
|
t.Run("change_estimate", func(t *testing.T) {
|
|
m := Photo{
|
|
PhotoName: "test_photo_1",
|
|
PhotoCountry: "de",
|
|
PhotoLat: 0.0,
|
|
PhotoLng: 0.0,
|
|
PlaceID: "s2:85d1ea7d3278",
|
|
PlaceSrc: SrcManual,
|
|
}
|
|
|
|
assert.Equal(t, "Germany", m.CountryName())
|
|
|
|
m.UpdateLocation()
|
|
|
|
assert.Equal(t, "Germany", m.CountryName())
|
|
assert.Equal(t, "de", m.PhotoCountry)
|
|
assert.Equal(t, float32(0.0), m.PhotoLat)
|
|
assert.Equal(t, float32(0.0), m.PhotoLng)
|
|
assert.Equal(t, "zz", m.PlaceID)
|
|
assert.Equal(t, SrcManual, m.PlaceSrc)
|
|
})
|
|
}
|