96 lines
3.8 KiB
Go
96 lines
3.8 KiB
Go
package entity
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestPhoto_EstimateCountry(t *testing.T) {
|
|
t.Run("uk", func(t *testing.T) {
|
|
m := Photo{PhotoName: "20200102_194030_9EFA9E5E", PhotoPath: "2000/05", OriginalName: "flickr import/changing-of-the-guard--buckingham-palace_7925318070_o.jpg"}
|
|
assert.Equal(t, UnknownCountry.ID, m.CountryCode())
|
|
assert.Equal(t, UnknownCountry.CountryName, m.CountryName())
|
|
m.EstimateCountry()
|
|
assert.Equal(t, "gb", m.CountryCode())
|
|
assert.Equal(t, "United Kingdom", m.CountryName())
|
|
})
|
|
|
|
t.Run("zz", func(t *testing.T) {
|
|
m := Photo{PhotoName: "20200102_194030_ADADADAD", PhotoPath: "2020/Berlin", OriginalName: "Zimmermannstrasse.jpg"}
|
|
assert.Equal(t, UnknownCountry.ID, m.CountryCode())
|
|
assert.Equal(t, UnknownCountry.CountryName, m.CountryName())
|
|
m.EstimateCountry()
|
|
assert.Equal(t, "de", m.CountryCode())
|
|
assert.Equal(t, "Germany", m.CountryName())
|
|
})
|
|
|
|
t.Run("de", func(t *testing.T) {
|
|
m := Photo{PhotoName: "Brauhaus", PhotoPath: "2020/Bayern", OriginalName: "München.jpg"}
|
|
assert.Equal(t, UnknownCountry.ID, m.CountryCode())
|
|
assert.Equal(t, UnknownCountry.CountryName, m.CountryName())
|
|
m.EstimateCountry()
|
|
assert.Equal(t, "de", m.CountryCode())
|
|
assert.Equal(t, "Germany", m.CountryName())
|
|
})
|
|
|
|
t.Run("ca", func(t *testing.T) {
|
|
m := Photo{PhotoTitle: "Port Lands / Gardiner Expressway / Toronto", PhotoPath: "2012/09", PhotoName: "20120910_231851_CA06E1AD", OriginalName: "demo/Toronto/port-lands--gardiner-expressway--toronto_7999515645_o.jpg"}
|
|
assert.Equal(t, UnknownCountry.ID, m.CountryCode())
|
|
assert.Equal(t, UnknownCountry.CountryName, m.CountryName())
|
|
m.EstimateCountry()
|
|
assert.Equal(t, "ca", m.CountryCode())
|
|
assert.Equal(t, "Canada", m.CountryName())
|
|
})
|
|
t.Run("photo has latlng", func(t *testing.T) {
|
|
m := Photo{
|
|
PhotoTitle: "Port Lands / Gardiner Expressway / Toronto",
|
|
PhotoLat: 13.333,
|
|
PhotoLng: 40.998,
|
|
PhotoCountry: "zz",
|
|
CellID: "161437aab90c",
|
|
PhotoName: "20120910_231851_CA06E1AD",
|
|
OriginalName: "demo/Toronto/port-lands--gardiner-expressway--toronto_7999515645_o.jpg",
|
|
}
|
|
m.EstimateCountry()
|
|
assert.Equal(t, "zz", m.CountryCode())
|
|
assert.Equal(t, "Unknown", m.CountryName())
|
|
})
|
|
|
|
}
|
|
|
|
func TestPhoto_EstimatePlace(t *testing.T) {
|
|
t.Run("photo already has location", func(t *testing.T) {
|
|
p := &Place{ID: "1000000001", PlaceCountry: "mx"}
|
|
m := Photo{PhotoName: "PhotoWithLocation", OriginalName: "demo/xyz.jpg", Place: p, PlaceID: "1000000001", PlaceSrc: SrcManual, PhotoCountry: "mx"}
|
|
assert.True(t, m.HasPlace())
|
|
assert.Equal(t, "mx", m.CountryCode())
|
|
assert.Equal(t, "Mexico", m.CountryName())
|
|
m.EstimatePlace()
|
|
assert.Equal(t, "mx", m.CountryCode())
|
|
assert.Equal(t, "Mexico", m.CountryName())
|
|
})
|
|
t.Run("recent photo has place", func(t *testing.T) {
|
|
m2 := Photo{PhotoName: "PhotoWithoutLocation", OriginalName: "demo/xyy.jpg", TakenAt: time.Date(2016, 11, 11, 8, 7, 18, 0, time.UTC)}
|
|
assert.Equal(t, "zz", m2.CountryCode())
|
|
m2.EstimatePlace()
|
|
assert.Equal(t, "mx", m2.CountryCode())
|
|
assert.Equal(t, "Mexico", m2.CountryName())
|
|
assert.Equal(t, SrcEstimate, m2.PlaceSrc)
|
|
})
|
|
t.Run("cant estimate - out of scope", func(t *testing.T) {
|
|
m2 := Photo{PhotoName: "PhotoWithoutLocation", OriginalName: "demo/xyy.jpg", TakenAt: time.Date(2016, 11, 13, 8, 7, 18, 0, time.UTC)}
|
|
assert.Equal(t, "zz", m2.CountryCode())
|
|
m2.EstimatePlace()
|
|
assert.Equal(t, "zz", m2.CountryCode())
|
|
})
|
|
/*t.Run("recent photo has country", func(t *testing.T) {
|
|
m2 := Photo{PhotoName: "PhotoWithoutLocation", OriginalName: "demo/zzz.jpg", TakenAt: time.Date(2001, 1, 1, 7, 20, 0, 0, time.UTC)}
|
|
assert.Equal(t, "zz", m2.CountryCode())
|
|
m2.EstimatePlace()
|
|
assert.Equal(t, "mx", m2.CountryCode())
|
|
assert.Equal(t, "Mexico", m2.CountryName())
|
|
assert.Equal(t, SrcEstimate, m2.PlaceSrc)
|
|
})*/
|
|
}
|