geomodel_test.go

package geomodel

import (
	"log"
	"sort"
	"testing"
)

/*type LocationCapable interface {
	Latitude() float64
	Longitude() float64
	Key() string
	Geocells() []string
}*/

type Place struct {
	lat, lon float64
	key      string
	geocells []string
}

func (p Place) Latitude() float64 {
	return p.lat
}

func (p Place) Longitude() float64 {
	return p.lon
}

func (p Place) Key() string {
	return p.key
}

func (p Place) Geocells() []string {
	return p.geocells
}

func TestGeoHash(t *testing.T) {
	log.Printf("GeoHash: %s", GeoCell(53.12869, 8.18976, 6))
}

func TestDecodeGeoHash(t *testing.T) {
	lat, lon := DecodeGeoHash(GeoHash(53.12869, 8.18976, 13))
	log.Printf("DecodeLat: %f, DecodeLon: %f", lat, lon)
}

func TestProximityFetch(t *testing.T) {

	// Example Places
	var places = []LocationCapable{Place{54, 8, "1", GeoCells(54, 8, 10)}, Place{50, 8, "2", GeoCells(50, 8, 10)}, Place{49, 8, "3", GeoCells(49, 8, 10)}, Place{48, 8, "4", GeoCells(48, 8, 10)}, Place{47, 8, "5", GeoCells(47, 8, 10)}}

	var result = ProximityFetch(50, 8, 20, 300, func(cells []string) []LocationCapable {
		// Implementing our own dirty thing here.
		var result []LocationCapable = make([]LocationCapable, 0)

		for _, place := range places {
			var added bool = false
			for _, c := range place.Geocells() {
				index := sort.SearchStrings(cells, c)
				if index < len(cells) {
					if place != nil {
						result = append(result, place)
					}
				}
			}
			if added {
				break
			}
		}

		return result
	}, 10)

	if len(result) > 1 {
		t.Fail()
	}

	if result[0].Latitude() != 50 || result[0].Longitude() != 8 {
		t.Fail()
	}

	// ProximityFetch(lat, lon float64, maxResults int, maxDistance float64, search RepositorySearch, maxResolution int) []LocationCapable
}