map_hoard_test.go

package ssp

import (
	"testing"
	"time"
)

func TestMapHoard(t *testing.T) {
	h := NewMapHoard()

	hoardCache := &HoardCache{}
	err := h.Save(Nut("nut"), hoardCache, time.Second)
	if err != nil {
		t.Fatalf("Failed save: %v", err)
	}

	val, err := h.Get(Nut("nut"))
	if err != nil {
		t.Fatalf("Failed get: %v", err)
	}

	if val != hoardCache {
		t.Fatalf("Wrong value get: %v", val)
	}

	// and again
	val, err = h.Get(Nut("nut"))
	if err != nil {
		t.Fatalf("Failed get: %v", err)
	}

	if val != hoardCache {
		t.Fatalf("Wrong value get: %v", val)
	}
}

func TestMapHoardGetAndDelete(t *testing.T) {
	h := NewMapHoard()

	hoardCache := &HoardCache{}
	err := h.Save(Nut("nut"), hoardCache, time.Second)
	if err != nil {
		t.Fatalf("Failed save: %v", err)
	}

	val, err := h.GetAndDelete(Nut("nut"))
	if err != nil {
		t.Fatalf("Failed get: %v", err)
	}

	if val != hoardCache {
		t.Fatalf("Wrong value get: %v", val)
	}

	// and again
	val, err = h.GetAndDelete(Nut("nut"))
	if err == nil {
		t.Fatalf("Succeeded when should have failed due to expiration")
	}

	if err != ErrNotFound {
		t.Fatalf("Wrong error: %v", err)
	}

	if val != nil {
		t.Fatalf("Value should be nil: %v", val)
	}
}

func TestMapHoardExpired(t *testing.T) {
	h := NewMapHoard()

	hoardCache := &HoardCache{}
	err := h.Save(Nut("nut"), hoardCache, 0)
	if err != nil {
		t.Fatalf("Failed save: %v", err)
	}
	time.Sleep(time.Microsecond)

	val, err := h.Get(Nut("nut"))
	if err == nil {
		t.Fatalf("Succeeded when should have failed due to expiration")
	}

	if err != ErrNotFound {
		t.Fatalf("Wrong error: %v", err)
	}

	if val != nil {
		t.Fatalf("Value should be nil: %v", val)
	}
}