alice-lg/pkg/caches/neighbors_test.go

package caches

import (
	"github.com/alice-lg/alice-lg/pkg/api"

	"testing"
	"time"
)

/*
NeighborsCache Tests
*/

func TestNeighborsCacheSetGet(t *testing.T) {
	cache := NewNeighborsCache(false)

	response := &api.NeighborsResponse{
		Response: api.Response{
			Meta: &api.Meta{
				TTL: time.Now().UTC().Add(23 * time.Millisecond),
			},
		},
	}

	if cache.Get() != nil {
		t.Error("There should not be anything cached yet!")
	}

	cache.Set(response)

	fromCache := cache.Get()
	if fromCache != response {
		t.Error("Expected", response, "got", fromCache)
	}

	// Wait a bit
	time.Sleep(33 * time.Millisecond)

	fromCache = cache.Get()
	if fromCache != nil {
		t.Error("Expected empty cache result, got:", fromCache)
	}
}